Merge pull request #5 from lastarc/refactor-embed-proxy

feat(bot): add emded-proxy v2 support with v1 as fallback
lastarc · Mar 31, 2024 · 67accb5 · 67accb5
2 parents 62133dd + e03d632
commit 67accb5
Showing 1 changed file with 20 additions and 1 deletion.
diff --git a/bot/commands/media/tiktok.js b/bot/commands/media/tiktok.js
@@ -107,7 +107,26 @@ module.exports = {
 		}
 
 		console.log(vurl, vwidth, vheight);
-		const embeddableUrl = `${EMBED_PROXY_URL}/?src=${encodeURIComponent(vurl)}&width=${vwidth}&height=${vheight}`;
+		let embeddableUrl = '';
+
+		const v2Check = await fetch(`${EMBED_PROXY_URL}/v2/healthz`);
+		if (v2Check.ok) {
+			const v2Res = await fetch(`${EMBED_PROXY_URL}/v2/add`, {
+				method: 'POST',
+				headers: {
+					'Content-Type': 'application/json',
+				},
+				body: JSON.stringify({
+					src: vurl,
+					width: vwidth,
+					height: vheight,
+				}),
+			});
+			const v2Json = await v2Res.json();
+			embeddableUrl = `${EMBED_PROXY_URL}/v2/~/${v2Json.slug}`;
+		} else {
+			embeddableUrl = `${EMBED_PROXY_URL}/?src=${encodeURIComponent(vurl)}&width=${vwidth}&height=${vheight}`;
+		}
 		await replyMsg.edit(`Done ${embeddableUrl}`);
 	},
 };