76 lines
2.9 KiB
TypeScript
76 lines
2.9 KiB
TypeScript
/*
|
|
* Copyright (C) 2026 Fluxer Contributors
|
|
*
|
|
* This file is part of Fluxer.
|
|
*
|
|
* Fluxer is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Fluxer is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Affero General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with Fluxer. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
import {selectOne} from 'css-select';
|
|
import type {Document, Element, Text} from 'domhandler';
|
|
import {parseDocument} from 'htmlparser2';
|
|
import type {MessageEmbedResponse} from '~/channel/EmbedTypes';
|
|
import {Logger} from '~/Logger';
|
|
import {BaseResolver} from '~/unfurler/resolvers/BaseResolver';
|
|
|
|
interface TenorJsonLd {
|
|
image?: {thumbnailUrl?: string};
|
|
video?: {contentUrl?: string};
|
|
}
|
|
|
|
export class TenorResolver extends BaseResolver {
|
|
match(url: URL, mimeType: string, _content: Uint8Array): boolean {
|
|
return mimeType.startsWith('text/html') && url.hostname === 'tenor.com';
|
|
}
|
|
|
|
async resolve(url: URL, content: Uint8Array, isNSFWAllowed: boolean = false): Promise<Array<MessageEmbedResponse>> {
|
|
const document = parseDocument(Buffer.from(content).toString('utf-8'));
|
|
const jsonLdContent = this.extractJsonLdContent(document);
|
|
if (!jsonLdContent) {
|
|
return [];
|
|
}
|
|
const {thumbnailURL, videoURL} = this.extractURLsFromJsonLd(jsonLdContent);
|
|
const thumbnail = thumbnailURL ? await this.resolveMediaURL(url, thumbnailURL, isNSFWAllowed) : undefined;
|
|
const video = videoURL ? await this.resolveMediaURL(url, videoURL, isNSFWAllowed) : undefined;
|
|
const embed: MessageEmbedResponse = {
|
|
type: 'gifv',
|
|
url: url.href,
|
|
provider: {name: 'Tenor', url: 'https://tenor.com'},
|
|
thumbnail: thumbnail ?? undefined,
|
|
video: video ?? undefined,
|
|
};
|
|
return [embed];
|
|
}
|
|
|
|
private extractJsonLdContent(document: Document): TenorJsonLd | null {
|
|
const scriptElement = selectOne('script.dynamic[type="application/ld+json"]', document) as Element | null;
|
|
if (scriptElement && scriptElement.children.length > 0) {
|
|
const scriptContentNode = scriptElement.children[0] as Text;
|
|
const scriptContent = scriptContentNode.data;
|
|
try {
|
|
return JSON.parse(scriptContent) as TenorJsonLd;
|
|
} catch (error) {
|
|
Logger.error({error}, 'Failed to parse JSON-LD content');
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
private extractURLsFromJsonLd(jsonLdContent: TenorJsonLd): {thumbnailURL?: string; videoURL?: string} {
|
|
const thumbnailUrl = jsonLdContent.image?.thumbnailUrl;
|
|
const videoUrl = jsonLdContent.video?.contentUrl;
|
|
return {thumbnailURL: thumbnailUrl, videoURL: videoUrl};
|
|
}
|
|
}
|