|
|
// plugins/cookie.server.ts
|
|
|
import { defineNuxtPlugin, useRequestHeaders, useRequestEvent } from '#app'
|
|
|
import { parse, serialize } from 'cookie'
|
|
|
import type { SerializeOptions } from 'cookie'
|
|
|
|
|
|
export default defineNuxtPlugin(() => {
|
|
|
const event = useRequestEvent()
|
|
|
const headers = useRequestHeaders(['cookie'])
|
|
|
|
|
|
// 解析请求中的Cookie
|
|
|
const requestCookies = headers.cookie ? parse(headers.cookie) : {}
|
|
|
|
|
|
return {
|
|
|
provide: {
|
|
|
serverCookie: {
|
|
|
// 获取Cookie
|
|
|
get: (name: string) => {
|
|
|
return requestCookies[name]
|
|
|
},
|
|
|
|
|
|
// 获取所有Cookie
|
|
|
getAll: () => {
|
|
|
return { ...requestCookies }
|
|
|
},
|
|
|
|
|
|
// 设置Cookie（响应头）
|
|
|
set: (name: string, value: string, options: Partial<SerializeOptions> = {}) => {
|
|
|
if (event?.node?.res) {
|
|
|
const cookie = serialize(name, value, {
|
|
|
path: '/',
|
|
|
sameSite: 'lax',
|
|
|
...options
|
|
|
} as SerializeOptions)
|
|
|
|
|
|
// 处理Set-Cookie头
|
|
|
const existingCookies = event.node.res.getHeader('Set-Cookie')
|
|
|
let cookiesArray: string[] = []
|
|
|
|
|
|
if (Array.isArray(existingCookies)) {
|
|
|
cookiesArray = [...existingCookies]
|
|
|
} else if (typeof existingCookies === 'string') {
|
|
|
cookiesArray = [existingCookies]
|
|
|
}
|
|
|
|
|
|
cookiesArray.push(cookie)
|
|
|
event.node.res.setHeader('Set-Cookie', cookiesArray)
|
|
|
}
|
|
|
},
|
|
|
|
|
|
// 删除Cookie
|
|
|
remove: (name: string, options: Partial<SerializeOptions> = {}) => {
|
|
|
if (event?.node?.res) {
|
|
|
const cookie = serialize(name, '', {
|
|
|
...options,
|
|
|
maxAge: 0,
|
|
|
expires: new Date(0)
|
|
|
} as SerializeOptions)
|
|
|
|
|
|
const existingCookies = event.node.res.getHeader('Set-Cookie')
|
|
|
let cookiesArray: string[] = []
|
|
|
|
|
|
if (Array.isArray(existingCookies)) {
|
|
|
cookiesArray = [...existingCookies]
|
|
|
} else if (typeof existingCookies === 'string') {
|
|
|
cookiesArray = [existingCookies]
|
|
|
}
|
|
|
|
|
|
cookiesArray.push(cookie)
|
|
|
event.node.res.setHeader('Set-Cookie', cookiesArray)
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
}) |