elk/composables/cache.ts

96 lines
2.6 KiB
TypeScript
Raw Normal View History

2022-11-24 05:47:14 +00:00
import LRU from 'lru-cache'
import type { Account, Status } from 'masto'
const cache = new LRU<string, any>({
max: 1000,
})
if (process.dev && process.client)
2022-11-24 05:47:14 +00:00
// eslint-disable-next-line no-console
console.log({ cache })
2022-11-24 07:53:27 +00:00
export function setCached(key: string, value: any, override = false) {
if (override || !cache.has(key))
cache.set(key, value)
2022-11-24 05:47:14 +00:00
}
function removeCached(key: string) {
cache.delete(key)
}
2022-11-24 05:47:14 +00:00
2022-12-01 07:24:35 +00:00
export function fetchStatus(id: string, force = false): Promise<Status> {
const server = currentServer.value
const key = `${server}:status:${id}`
2022-11-24 05:47:14 +00:00
const cached = cache.get(key)
2022-12-01 07:24:35 +00:00
if (cached && !force)
2022-11-24 05:47:14 +00:00
return cached
const promise = useMasto().statuses.fetch(id)
2022-11-24 05:47:14 +00:00
.then((status) => {
cacheStatus(status)
return status
})
cache.set(key, promise)
return promise
}
2022-12-06 23:38:00 +00:00
export function fetchAccountById(id?: string | null): Promise<Account | null> {
if (!id)
return Promise.resolve(null)
const server = currentServer.value
const key = `${server}:account:${id}`
2022-11-24 05:47:14 +00:00
const cached = cache.get(key)
if (cached)
return cached
const uri = currentInstance.value?.uri
const promise = useMasto().accounts.fetch(id)
.then((r) => {
2022-12-28 16:01:13 +00:00
if (r.acct && !r.acct.includes('@') && uri)
r.acct = `${r.acct}@${uri}`
cacheAccount(r, server, true)
return r
2022-11-24 05:47:14 +00:00
})
cache.set(key, promise)
return promise
}
2022-11-30 07:08:10 +00:00
export async function fetchAccountByHandle(acct: string): Promise<Account> {
const server = currentServer.value
const key = `${server}:account:${acct}`
2022-11-24 05:47:14 +00:00
const cached = cache.get(key)
if (cached)
return cached
const uri = currentInstance.value?.uri
const account = useMasto().accounts.lookup({ acct })
2022-11-24 05:47:14 +00:00
.then((r) => {
2022-12-28 16:01:13 +00:00
if (r.acct && !r.acct.includes('@') && uri)
r.acct = `${r.acct}@${uri}`
cacheAccount(r, server, true)
2022-11-24 05:47:14 +00:00
return r
})
cache.set(key, account)
return account
}
2022-11-30 07:08:10 +00:00
export function useAccountByHandle(acct: string) {
return useAsyncState(() => fetchAccountByHandle(acct), null).state
}
2022-12-06 23:38:00 +00:00
export function useAccountById(id?: string | null) {
2022-11-30 07:08:10 +00:00
return useAsyncState(() => fetchAccountById(id), null).state
}
export function cacheStatus(status: Status, server = currentServer.value, override?: boolean) {
setCached(`${server}:status:${status.id}`, status, override)
2022-11-24 05:47:14 +00:00
}
export function removeCachedStatus(id: string, server = currentServer.value) {
removeCached(`${server}:status:${id}`)
}
export function cacheAccount(account: Account, server = currentServer.value, override?: boolean) {
setCached(`${server}:account:${account.id}`, account, override)
setCached(`${server}:account:${account.acct}`, account, override)
2022-11-24 05:47:14 +00:00
}