elk/server/api/og-image.ts

51 lines
1.3 KiB
TypeScript
Raw Normal View History

2022-12-11 01:09:11 +03:00
const inMemoryCache = new Map<string, { url: string; lastUsed: number }>()
export default defineEventHandler(async (event) => {
const { cardUrl } = getQuery(event)
if (!cardUrl) {
throw createError({
statusCode: 422,
statusMessage: 'Missing cardUrl.',
})
}
if (typeof cardUrl !== 'string') {
throw createError({
statusCode: 422,
statusMessage: 'cardUrl must be string.',
})
}
if (inMemoryCache.has(cardUrl)) {
const { url } = inMemoryCache.get(cardUrl)!
await send(event, url)
2022-12-11 03:04:37 +03:00
inMemoryCache.set(cardUrl, { url, lastUsed: Date.now() })
// Remove some oldest entries if cache gets to big
2022-12-11 01:09:11 +03:00
if (inMemoryCache.size > 5000) {
2022-12-11 03:04:37 +03:00
// Remove 10% of the oldest entries
const entries = Array.from(inMemoryCache.entries()).sort(
(a, b) => a[1].lastUsed - b[1].lastUsed,
2022-12-11 01:09:11 +03:00
)
2022-12-11 03:04:37 +03:00
const entriesToRemove = Math.floor(entries.length * 0.1)
for (let i = 0; i < entriesToRemove; i++)
inMemoryCache.delete(entries[i][0])
2022-12-11 01:09:11 +03:00
}
return
}
const result = await $fetch<string>(cardUrl)
let ogImageUrl: string | null | undefined = null
const match = result.match(/<meta property="og:image" content="([^"]+)" \/>/)
ogImageUrl = match?.[1] ?? ''
inMemoryCache.set(cardUrl, { url: ogImageUrl, lastUsed: Date.now() })
await send(event, ogImageUrl)
})