add support for reds key prefix

This commit is contained in:
gzzchh 2022-03-26 19:50:09 +08:00 committed by myl7
parent d6aa1d6154
commit f409c3666d
2 changed files with 8 additions and 5 deletions

View file

@ -13,6 +13,9 @@ module.exports = {
// /public directory of your GitHub project (not your OneDrive folder!), and referenced here by its relative path to /public.
icon: '/icons/128.png',
// Prefix for KV Storage
kvPrefix: process.env.KV_PREFIX || 'spwoo',
// The name of your website. Present alongside your icon.
title: "Spencer's OneDrive",

View file

@ -1,12 +1,12 @@
import Redis from 'ioredis'
import siteConfig from '../config/site.config'
// Persistent key-value store is provided by Redis, hosted on Upstash
// https://vercel.com/integrations/upstash
const kv = new Redis(process.env.REDIS_URL)
export async function getOdAuthTokens(): Promise<{ accessToken: unknown; refreshToken: unknown }> {
const accessToken = await kv.get('access_token')
const refreshToken = await kv.get('refresh_token')
const accessToken = await kv.get(`${siteConfig.kvPrefix}_access_token`)
const refreshToken = await kv.get(`${siteConfig.kvPrefix}_refresh_token`)
return {
accessToken,
@ -23,6 +23,6 @@ export async function storeOdAuthTokens({
accessTokenExpiry: number
refreshToken: string
}): Promise<void> {
await kv.set('access_token', accessToken, 'ex', accessTokenExpiry)
await kv.set('refresh_token', refreshToken)
await kv.set(`${siteConfig.kvPrefix}_access_token`, accessToken, 'ex', accessTokenExpiry)
await kv.set(`${siteConfig.kvPrefix}_refresh_token`, refreshToken)
}