42 lines
1.2 KiB
TypeScript
42 lines
1.2 KiB
TypeScript
import path from 'node:path';
|
||
|
||
import { config } from '@/config';
|
||
import { art } from '@/utils/render';
|
||
|
||
export function renderDesc(post, link, quality: 'sample' | 'orig') {
|
||
const { id, source, owner, file_url: fileUrl, tags, score } = post;
|
||
const isHttp = /^https?:\/\//.test(source);
|
||
const sourceHost = isHttp ? new URL(source).host : source || 'unknown';
|
||
const imgQualityMap = { sample: 'sample_url', orig: 'file_url' };
|
||
|
||
// 判断是否是视频链接
|
||
const videoExtList = ['mp4', 'webm'];
|
||
const fileExt = fileUrl.slice(fileUrl.lastIndexOf('.') + 1);
|
||
const isVideo = videoExtList.includes(fileExt);
|
||
// 如果是视频则始终使用 fileUrl（原文件）
|
||
let contentURL = post[imgQualityMap[quality]] || fileUrl;
|
||
if (isVideo) {
|
||
contentURL = fileUrl;
|
||
}
|
||
|
||
return art(path.join(__dirname, 'templates/description.art'), {
|
||
id,
|
||
source,
|
||
owner,
|
||
tags,
|
||
link,
|
||
isHttp,
|
||
sourceHost,
|
||
contentURL,
|
||
isVideo,
|
||
score: score || 0,
|
||
});
|
||
}
|
||
|
||
export function getAPIKeys() {
|
||
return {
|
||
apiKey: config.gelbooru.apiKey || '',
|
||
userId: config.gelbooru.userId || '',
|
||
};
|
||
}
|