import fs from 'unstorage/drivers/fs' import memory from 'unstorage/drivers/memory' import kv from 'unstorage/drivers/cloudflare-kv-http' import vercelKVDriver from 'unstorage/drivers/vercel-kv' import { $fetch } from 'ofetch' import cached from '../cache-driver' // @ts-expect-error virtual import import { env } from '#build-info' // @ts-expect-error virtual import import { driver } from '#storage-config' import type { AppInfo } from '~/types' import { APP_NAME } from '~/constants' const storage = useStorage() if (driver === 'fs') { const config = useRuntimeConfig() storage.mount('servers', fs({ base: config.storage.fsBase })) } else if (driver === 'cloudflare') { const config = useRuntimeConfig() storage.mount('servers', cached(kv({ accountId: config.cloudflare.accountId, namespaceId: config.cloudflare.namespaceId, apiToken: config.cloudflare.apiToken, }))) } else if (driver === 'vercel') { const config = useRuntimeConfig() storage.mount('servers', cached(vercelKVDriver({ url: config.vercel.url, token: config.vercel.token, env: config.vercel.env, base: config.vercel.base, }))) } else if (driver === 'memory') { storage.mount('servers', memory()) } export function getRedirectURI(origin: string, server: string) { origin = origin.replace(/\?.*$/, '') return `${origin}/api/${server}/oauth/${encodeURIComponent(origin)}` } async function fetchAppInfo(origin: string, server: string) { const app: AppInfo = await $fetch(`https://${server}/api/v1/apps`, { method: 'POST', body: { client_name: APP_NAME + (env !== 'release' ? ` (${env})` : ''), website: 'https://elk.zone', redirect_uris: getRedirectURI(origin, server), scopes: 'read write follow push', }, }) return app } export async function getApp(origin: string, server: string) { const host = origin.replace(/^https?:\/\//, '').replace(/[^\w\d]/g, '-').replace(/\?.*$/, '') const key = `servers:v3:${server}:${host}.json`.toLowerCase() try { if (await storage.hasItem(key)) return (storage.getItem(key, {}) as Promise) const appInfo = await fetchAppInfo(origin, server) await storage.setItem(key, appInfo) return appInfo } catch { return null } } export async function deleteApp(server: string) { const keys = (await storage.getKeys(`servers:v3:${server}:`)) for (const key of keys) await storage.removeItem(key) } export async function listServers() { const keys = await storage.getKeys('servers:v3:') const servers = new Set() for await (const key of keys) { const id = key.split(':')[2] if (id) servers.add(id.toLocaleLowerCase()) } return Array.from(servers).sort() }