syhily/yufan.me
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add avatar fetch for cdn with origin limit.

Browse Source
This commit is contained in:
Yufan Sheng 2024-06-25 21:08:44 +08:00
parent 2414040c4c
commit b467df07d2
Signed by: syhily
GPG Key ID: 9D18A22A7DCD5A9B
1 changed files with 16 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/pages/avatar/[hash].png.ts
View File

@ -1,7 +1,7 @@
import { queryEmail } from '@/helpers/db/query';
import { encodedEmail, urlJoin } from '@/helpers/tools';
import options from '@/options';
import type { APIRoute } from 'astro';
import type { APIRoute, ValidRedirectStatus } from 'astro';
const defaultAvatar = (): string => {
return urlJoin(options.assetsPrefix(), '/images/default-avatar.png');
@ -11,12 +11,22 @@ function isNumeric(str: string) {
return !Number.isNaN(str) && !Number.isNaN(Number.parseFloat(str));
}
const avatarImage = async (hash: string): Promise<Response> => {
const avatarImage = async (
hash: string,
redirect: (path: string, status?: ValidRedirectStatus) => Response,
): Promise<Response> => {
const defaultAvatarLink = defaultAvatar();
const link = urlJoin(
options.settings.comments.avatar.mirror,
`${hash}.png?s=${options.settings.comments.avatar.size}&d=${defaultAvatar()}`,
`${hash}.png?s=${options.settings.comments.avatar.size}&d=${defaultAvatarLink}`,
);
return new Response(Buffer.from(await (await fetch(link)).arrayBuffer()), {
const resp = await fetch(link, { redirect: 'manual', headers: { Referer: options.website } });
if (resp.headers.get('location') === defaultAvatarLink) {
return redirect(defaultAvatarLink, 302);
}
return new Response(Buffer.from(await resp.arrayBuffer()), {
headers: { 'Content-Type': 'image/png' },
});
};
@ -33,8 +43,8 @@ export const GET: APIRoute = async ({ params, redirect }) => {
if (email === null) {
return redirect(defaultAvatar());
}
return avatarImage(encodedEmail(email));
return avatarImage(encodedEmail(email), redirect);
}
return avatarImage(hash);
return avatarImage(hash, redirect);
};