2023-02-09 17:47:59 +00:00
|
|
|
import fs from 'unstorage/drivers/fs'
|
|
|
|
import memory from 'unstorage/drivers/memory'
|
2022-12-22 02:12:37 +00:00
|
|
|
|
2023-01-01 19:30:39 +00:00
|
|
|
import { stringifyQuery } from 'ufo'
|
2022-11-23 23:12:25 +00:00
|
|
|
|
2022-12-18 23:13:01 +00:00
|
|
|
import { $fetch } from 'ofetch'
|
2022-11-23 23:12:25 +00:00
|
|
|
import type { Storage } from 'unstorage'
|
|
|
|
|
2023-02-08 20:00:43 +00:00
|
|
|
import cached from '../cache-driver'
|
|
|
|
import kv from '../cloudflare-driver'
|
2022-11-23 23:12:25 +00:00
|
|
|
|
2023-01-29 14:18:27 +00:00
|
|
|
// @ts-expect-error virtual import
|
|
|
|
import { env } from '#build-info'
|
2023-01-29 15:52:01 +00:00
|
|
|
// @ts-expect-error virtual import
|
|
|
|
import { driver } from '#storage-config'
|
2023-01-29 14:18:27 +00:00
|
|
|
|
2022-11-15 15:48:23 +00:00
|
|
|
import type { AppInfo } from '~/types'
|
2022-11-23 23:12:25 +00:00
|
|
|
import { APP_NAME } from '~/constants'
|
2022-11-15 14:29:46 +00:00
|
|
|
|
2022-11-23 23:12:25 +00:00
|
|
|
const storage = useStorage() as Storage
|
|
|
|
|
2023-01-29 15:52:01 +00:00
|
|
|
if (driver === 'fs') {
|
|
|
|
const config = useRuntimeConfig()
|
2022-11-30 21:17:20 +00:00
|
|
|
storage.mount('servers', fs({ base: config.storage.fsBase }))
|
2022-11-23 23:12:25 +00:00
|
|
|
}
|
2023-01-29 15:52:01 +00:00
|
|
|
else if (driver === 'cloudflare') {
|
|
|
|
const config = useRuntimeConfig()
|
2022-11-23 23:12:25 +00:00
|
|
|
storage.mount('servers', cached(kv({
|
|
|
|
accountId: config.cloudflare.accountId,
|
|
|
|
namespaceId: config.cloudflare.namespaceId,
|
|
|
|
apiToken: config.cloudflare.apiToken,
|
|
|
|
})))
|
|
|
|
}
|
2023-01-29 15:52:01 +00:00
|
|
|
else if (driver === 'memory') {
|
2023-01-16 21:52:50 +00:00
|
|
|
storage.mount('servers', memory())
|
|
|
|
}
|
2022-11-30 21:17:20 +00:00
|
|
|
|
2023-01-01 19:30:39 +00:00
|
|
|
export function getRedirectURI(origin: string, server: string) {
|
|
|
|
return `${origin}/api/${server}/oauth?${stringifyQuery({ origin })}`
|
2022-11-29 22:55:53 +00:00
|
|
|
}
|
2022-11-23 23:12:25 +00:00
|
|
|
|
2023-01-01 19:30:39 +00:00
|
|
|
async function fetchAppInfo(origin: string, server: string) {
|
2022-11-23 23:12:25 +00:00
|
|
|
const app: AppInfo = await $fetch(`https://${server}/api/v1/apps`, {
|
|
|
|
method: 'POST',
|
|
|
|
body: {
|
2023-01-29 14:18:27 +00:00
|
|
|
client_name: APP_NAME + (env !== 'release' ? ` (${env})` : ''),
|
2022-11-29 22:55:53 +00:00
|
|
|
website: 'https://elk.zone',
|
2023-01-01 19:30:39 +00:00
|
|
|
redirect_uris: getRedirectURI(origin, server),
|
2022-11-23 23:12:25 +00:00
|
|
|
scopes: 'read write follow push',
|
|
|
|
},
|
2022-11-15 15:48:23 +00:00
|
|
|
})
|
2022-11-23 23:12:25 +00:00
|
|
|
return app
|
|
|
|
}
|
|
|
|
|
2023-01-01 19:30:39 +00:00
|
|
|
export async function getApp(origin: string, server: string) {
|
2023-01-17 10:44:15 +00:00
|
|
|
const host = origin.replace(/^https?:\/\//, '').replace(/[^\w\d]/g, '-')
|
|
|
|
const key = `servers:v2:${server}:${host}.json`.toLowerCase()
|
2022-11-23 23:12:25 +00:00
|
|
|
|
|
|
|
try {
|
2022-11-27 18:22:39 +00:00
|
|
|
if (await storage.hasItem(key))
|
|
|
|
return await storage.getItem(key) as Promise<AppInfo>
|
2023-01-01 19:30:39 +00:00
|
|
|
const appInfo = await fetchAppInfo(origin, server)
|
2022-11-23 23:12:25 +00:00
|
|
|
await storage.setItem(key, appInfo)
|
|
|
|
return appInfo
|
|
|
|
}
|
|
|
|
catch {
|
|
|
|
return null
|
|
|
|
}
|
2022-11-15 14:29:46 +00:00
|
|
|
}
|
2022-12-24 00:07:38 +00:00
|
|
|
|
2023-01-15 09:34:17 +00:00
|
|
|
export async function deleteApp(server: string) {
|
2023-01-17 10:44:15 +00:00
|
|
|
const keys = (await storage.getKeys(`servers:v2:${server}:`))
|
2023-01-15 09:34:17 +00:00
|
|
|
for (const key of keys)
|
|
|
|
await storage.removeItem(key)
|
|
|
|
}
|
|
|
|
|
2022-12-24 00:07:38 +00:00
|
|
|
export async function listServers() {
|
2023-01-17 10:44:15 +00:00
|
|
|
const keys = await storage.getKeys('servers:v2:')
|
2022-12-24 00:07:38 +00:00
|
|
|
const servers = new Set<string>()
|
|
|
|
for await (const key of keys) {
|
2023-01-17 10:44:15 +00:00
|
|
|
const id = key.split(':')[2]
|
2022-12-24 00:07:38 +00:00
|
|
|
if (id)
|
2022-12-24 00:51:45 +00:00
|
|
|
servers.add(id.toLocaleLowerCase())
|
2022-12-24 00:07:38 +00:00
|
|
|
}
|
|
|
|
return Array.from(servers).sort()
|
|
|
|
}
|