chore: added put

This commit is contained in:
Junseo Park 2021-02-17 15:15:04 +09:00
parent 5769f90fbb
commit 16d440cf69
5 changed files with 16 additions and 10 deletions

View File

@ -205,7 +205,7 @@ const Bots: NextPage<BotsProps> = ({ data, date, user, theme, setTheme }) => {
} }
export const getServerSideProps = async (ctx: Context) => { export const getServerSideProps = async (ctx: Context) => {
const parsed = parseCookie(ctx) const parsed = parseCookie(ctx.req)
const data = await get.bot.load(ctx.query.id) ?? { id: '' } const data = await get.bot.load(ctx.query.id) ?? { id: '' }
const user = await get.Authorization(parsed?.token) const user = await get.Authorization(parsed?.token)
return { return {

View File

@ -1,15 +1,16 @@
import { NextPage, NextPageContext } from 'next' import { NextPage } from 'next'
import Link from 'next/link' import Link from 'next/link'
import dynamic from 'next/dynamic' import dynamic from 'next/dynamic'
import { useRouter } from 'next/router' import { useRouter } from 'next/router'
import { Bot, User } from '@types' import { Bot, CsrfContext, User } from '@types'
import { get } from '@utils/Query' import { get } from '@utils/Query'
import { makeBotURL, parseCookie, checkBotFlag } from '@utils/Tools' import { makeBotURL, parseCookie, checkBotFlag } from '@utils/Tools'
import { ParsedUrlQuery } from 'querystring' import { ParsedUrlQuery } from 'querystring'
import NotFound from 'pages/404' import NotFound from 'pages/404'
import { getToken } from '@utils/Csrf'
const Container = dynamic(() => import('@components/Container')) const Container = dynamic(() => import('@components/Container'))
@ -24,7 +25,8 @@ const Advertisement = dynamic(() => import('@components/Advertisement'))
const Tooltip = dynamic(() => import('@components/Tooltip')) const Tooltip = dynamic(() => import('@components/Tooltip'))
const Markdown = dynamic(() => import ('@components/Markdown')) const Markdown = dynamic(() => import ('@components/Markdown'))
const VoteBot: NextPage<VoteBotProps> = ({ data }) => { const VoteBot: NextPage<VoteBotProps> = ({ data, user, csrfToken }) => {
console.log(csrfToken)
const router = useRouter() const router = useRouter()
if(!data?.id) return <NotFound /> if(!data?.id) return <NotFound />
if((checkBotFlag(data.flags, 'trusted') || checkBotFlag(data.flags, 'partnered')) && data.vanity && data.vanity !== router.query.id) router.push(`/bots/${data.vanity}`) if((checkBotFlag(data.flags, 'trusted') || checkBotFlag(data.flags, 'partnered')) && data.vanity && data.vanity !== router.query.id) router.push(`/bots/${data.vanity}`)
@ -54,11 +56,13 @@ const VoteBot: NextPage<VoteBotProps> = ({ data }) => {
} }
export const getServerSideProps = async (ctx: Context) => { export const getServerSideProps = async (ctx: Context) => {
const parsed = parseCookie(ctx) const parsed = parseCookie(ctx.req)
const data = await get.bot.load(ctx.query.id) const data = await get.bot.load(ctx.query.id)
const user = await get.Authorization(parsed?.token) const user = await get.Authorization(parsed?.token)
return { return {
props: { props: {
csrfToken: getToken(ctx.req, ctx.res),
data, data,
user: await get.user.load(user || '') user: await get.user.load(user || '')
}, },
@ -66,12 +70,13 @@ export const getServerSideProps = async (ctx: Context) => {
} }
interface VoteBotProps { interface VoteBotProps {
csrfToken: string
vote: boolean vote: boolean
data: Bot data: Bot
user: User user: User
} }
interface Context extends NextPageContext { interface Context extends CsrfContext {
query: URLQuery query: URLQuery
} }

View File

@ -43,7 +43,7 @@ const DiscordCallback:NextPage<DiscordCallbackProps> = ({ data }) => {
} }
export const getServerSideProps = async(ctx: NextPageContext) => { export const getServerSideProps = async(ctx: NextPageContext) => {
const parsed = parseCookie(ctx) const parsed = parseCookie(ctx.req)
const token = verify(parsed?.token ?? '') const token = verify(parsed?.token ?? '')
if(!token) return { props: { data: null } } if(!token) return { props: { data: null } }

View File

@ -46,7 +46,7 @@ const Panel:NextPage<PanelProps> = ({ logged, user, submits }) => {
} }
export const getServerSideProps = async (ctx: NextPageContext) => { export const getServerSideProps = async (ctx: NextPageContext) => {
const parsed = parseCookie(ctx) const parsed = parseCookie(ctx.req)
const user = await get.Authorization(parsed?.token) || '' const user = await get.Authorization(parsed?.token) || ''
const submits = await get.botSubmits.load(user) const submits = await get.botSubmits.load(user)

View File

@ -5,6 +5,7 @@ import cookie from 'cookie'
import { BotFlags, ImageOptions, UserFlags } from '@types' import { BotFlags, ImageOptions, UserFlags } from '@types'
import { KoreanbotsEndPoints, Oauth } from './Constants' import { KoreanbotsEndPoints, Oauth } from './Constants'
import { NextRouter } from 'next/router' import { NextRouter } from 'next/router'
import { IncomingMessage } from 'http'
export function formatNumber(value: number):string { export function formatNumber(value: number):string {
const suffixes = ['', '만', '억', '조','해'] const suffixes = ['', '만', '억', '조','해']
@ -95,8 +96,8 @@ export function bufferToStream(binary: Buffer) {
return readableInstanceStream return readableInstanceStream
} }
export function parseCookie(ctx: NextPageContext): { [key: string]: string } { export function parseCookie(req: { headers: { cookie?: string }}): { [key: string]: string } {
return cookie.parse(ctx.req.headers.cookie || '') return cookie.parse(req.headers.cookie || '')
} }
export function redirectTo(router: NextRouter, to: string) { export function redirectTo(router: NextRouter, to: string) {