Merge pull request #10 from hker9527/feat-9

feat: Auto-generate Twitter embeds

Author: hker9527

README.md

@@ -39,3 +39,4 @@ yarn start
 |[saucenao](https://saucenao.com/)|Reverse image search|
 |[exchangerate.host](https://exchangerate.host/)|Currency conversion|
 |[Google Safebrowsing](https://safebrowsing.google.com/)|Detect malicious URLs|
+|[VXTwitter](https://github.com/dylanpdx/BetterTwitFix)|Generate embeds from Twitter links|

src/modules/index.ts

@@ -2,9 +2,11 @@ import { StealthModule } from "@type/StealthModule";
 import { moebooru } from "./moebooru";
 import { pixiv } from "./pixiv";
 import { scam } from "./scam";
+import { twitter } from "./twitter";
 
 export const modules: StealthModule[] = [
 	moebooru,
 	pixiv,
-	scam
+	scam,
+	twitter
 ]

src/modules/twitter.ts

@@ -0,0 +1,50 @@
+import { StealthModule } from "@type/StealthModule";
+import { ZAPIVXTwitter } from "@type/api/VXTwitter";
+import { APIEmbed } from "discord.js";
+
+export const twitter: StealthModule = {
+	name: "twitter",
+	event: "messageCreate",
+	pattern: /https?:\/\/(?:www\.)?twitter\.com\/(?:#!\/)?(\w+)\/status\/(\d+)/,
+	action: async (obj) => {
+		const statusId = obj.matches!.pop();
+		const response = await (await fetch(`https://api.vxtwitter.com/i/status/${statusId}`)).text();
+
+		try {
+			const json = JSON.parse(response);
+			if (ZAPIVXTwitter.check(json)) {
+				await obj.message.suppressEmbeds();
+				return {
+					type: "send",
+					result: {
+						embeds: ([
+							{
+								author: {
+									name: `${json.user_name} (@${json.user_screen_name})`,
+									icon_url: "https://cdn-icons-png.flaticon.com/512/179/179342.png",
+									url: `https://twitter.com/${json.user_screen_name}`
+								},
+								color: 0x1DA1F2,
+								// Remove https://t.co/... links
+								description: json.text.replace(/https:\/\/t\.co\/\w+/g, ""),
+								timestamp: new Date(json.date_epoch * 1000).toISOString(),
+								image: {
+									url: json.mediaURLs[0]
+								},
+								url: json.tweetURL
+							}
+						] as APIEmbed[]).concat(json.mediaURLs.splice(1).map((url: string) => ({
+							image: {
+								url
+							},
+							url: json.tweetURL
+						}))) 
+					}
+				};
+			}
+			return false;
+		} catch (e) {
+			return false;
+		}
+	}
+};

src/types/api/VXTwitter.ts

@@ -0,0 +1,44 @@
+import { z } from "zod";
+import { Zod } from "@type/Zod";
+
+const SizeSchema = z.object({
+	height: z.number(),
+	width: z.number()
+});
+
+const BaseMediaSchema = z.object({
+	altText: z.string().nullable(),
+	size: SizeSchema,
+	thumbnail_url: z.string(),
+	url: z.string()
+});
+
+const ImageMediaSchema = BaseMediaSchema.extend({
+	type: z.literal("image")
+});
+
+const VideoMediaSchema = BaseMediaSchema.extend({
+	duration_millis: z.number(),
+	type: z.literal("video")
+});
+
+const VXTwitterSchema = z.object({
+	conversationID: z.string(),
+	date: z.string(),
+	date_epoch: z.number(),
+	hashtags: z.array(z.string()),
+	likes: z.number(),
+	mediaURLs: z.array(z.string()),
+	media_extended: z.array(z.union([ImageMediaSchema, VideoMediaSchema])),
+	replies: z.number(),
+	retweets: z.number(),
+	text: z.string(),
+	tweetID: z.string(),
+	tweetURL: z.string(),
+	user_name: z.string(),
+	user_screen_name: z.string()
+});
+
+export const ZAPIVXTwitter = new Zod(VXTwitterSchema);
+
+export type APIVXTwitter = z.infer<typeof VXTwitterSchema>;