obsidian-sample-plugin/src/LinkThumbnailWidgetParams.ts

84 lines
3.3 KiB
TypeScript

import { requestUrl } from "obsidian";
import { decode } from 'iconv-lite';
interface ogData {
"ogTitle": string,
"ogDescription": string,
"ogImage": string,
"ogImageAlt": string,
"ogUrl": string
}
// url 정규식
const urlRegex = new RegExp("^(http:\\/\\/www\\.|https:\\/\\/www\\.|http:\\/\\/|https:\\/\\/)?[a-z0-9]+([\\-.]{1}[a-z0-9]+)*\\.[a-z]{2,5}(:[0-9]{1,5})?(\\/.*)?$");
async function getOgData(url: string) {
const urlArr = urlRegex.exec(url);
if (urlArr) {
try {
const response = await requestUrl(url);
if (response && response.headers && response.headers['content-type'] && !response.headers['content-type']?.includes('text/')) {
throw new Error('Page must return a header content-type with text/');
}
// 인코딩 문제 해결
const bodyArrayBuffer = response.arrayBuffer;
const contentType = response.headers["content-type"].toLowerCase().trim();
const charset = contentType.substring(contentType.indexOf("charset=") + 8, contentType.length);
let body;
if (charset === "utf-8") {
body = Buffer.from(bodyArrayBuffer).toString('utf-8');
} else {
body = decode(Buffer.from(bodyArrayBuffer), charset);
}
const parser = new DOMParser();
const document = parser.parseFromString(body, 'text/html');
const ogTitle = document.querySelector("meta[property='og:title']")?.getAttribute("content") || document.querySelector("title")?.textContent || "";
const ogDescription = document.querySelector("meta[property='og:description']")?.getAttribute("content") || "";
let ogImage = document.querySelector("meta[property='og:image']")?.getAttribute("content") || "";
if (ogImage.startsWith("/")) {
const baseUrl = url.replace(urlArr[4], "");
ogImage = baseUrl + ogImage;
}
const ogImageAlt = document.querySelector("meta[property='og:image:alt']")?.getAttribute("content") || "";
const ogUrl = document.querySelector("meta[property='og:url']")?.getAttribute("content") || url;
const data: ogData = {
"ogTitle": ogTitle,
"ogDescription": ogDescription,
"ogImage": ogImage,
"ogImageAlt": ogImageAlt,
"ogUrl": ogUrl
}
return data;
} catch (error) {
console.error(error);
}
}
}
function renderOgData(data:ogData) {
return `
<div class="openGraphPreview">
${(data?.ogImage === "")? "" : `<div class="se-oglink-thumbnail"><img src="${data?.ogImage}" alt="${data?.ogImageAlt}" class="se-oglink-thumbnail-resource"></img></div>`}
<div class="se-oglink-info-container">
<strong class="se-oglink-info">${data?.ogTitle}</strong>
<description class="se-oglink-summary">${data?.ogDescription}</description>
<p class="se-oglink-url">${data?.ogUrl}</p>
</div>
</div>
`;
}
export async function LinkThumbnailWidgetParams(url: string) {
const data = await getOgData(url);
if (data) {
const result = renderOgData(data);
return result;
}
return null;
}