elk/server/shared.ts
2022-11-25 23:15:54 +08:00

70 lines
1.8 KiB
TypeScript

// @ts-expect-error unstorage needs to provide backwards-compatible subpath types
import _fs from 'unstorage/drivers/fs'
// @ts-expect-error unstorage needs to provide backwards-compatible subpath types
import _kv from 'unstorage/drivers/cloudflare-kv-http'
import { $fetch } from 'ohmyfetch'
import type { Storage } from 'unstorage'
import cached from './cache-driver'
import type { AppInfo } from '~/types'
import { APP_NAME } from '~/constants'
const fs = _fs as typeof import('unstorage/dist/drivers/fs')['default']
const kv = _kv as typeof import('unstorage/dist/drivers/cloudflare-kv-http')['default']
const storage = useStorage() as Storage
if (process.dev) {
storage.mount('servers', fs({ base: 'node_modules/.cache/servers' }))
}
else {
const config = useRuntimeConfig()
storage.mount('servers', cached(kv({
accountId: config.cloudflare.accountId,
namespaceId: config.cloudflare.namespaceId,
apiToken: config.cloudflare.apiToken,
})))
}
const KNOWN_DOMAINS = [
'http://localhost:5314',
'https://elk.netlify.app',
'https://elk.zone',
]
async function fetchAppInfo(server: string) {
const redirect_uris = [
'urn:ietf:wg:oauth:2.0:oob',
...KNOWN_DOMAINS.map(d => `${d}/api/${server}/oauth`),
].join('\n')
const app: AppInfo = await $fetch(`https://${server}/api/v1/apps`, {
method: 'POST',
body: {
client_name: APP_NAME,
redirect_uris,
scopes: 'read write follow push',
},
})
return app
}
const serverKey = (server: string) => `servers:${server}.json`
export async function getApp(server: string) {
const key = serverKey(server)
if (await storage.hasItem(key))
return storage.getItem(key) as Promise<AppInfo>
try {
const appInfo = await fetchAppInfo(server)
await storage.setItem(key, appInfo)
return appInfo
}
catch {
return null
}
}