Implement whitelist

config.js

@@ -20,6 +20,7 @@ const config = {
   http: {
     agent: process.env.HTTP_AGENT,
   },
+  whitelist: process.env.WHITELIST ? process.env.WHITELIST.split(",") : [],
 };
 
 const envConfFile = path.resolve(

handler/fetch.js

@@ -10,23 +10,33 @@ if (!config.http || !config.http.agent) {
   process.exit();
 }
 
+const isWhitelisted = (ctx) => {
+  return (
+    config.whitelist.length == 0 ||
+    config.whitelist.includes(ctx.message.from.username)
+  );
+};
+
 // Domain checks
 const command = (ctx) => {
-  if (ctx.message.text) {
-    let urls = getUrls(ctx.message.text);
-
-    urls.forEach((url) => {
-      if (url.match(/https??:\/\/(vm\.)??tiktok\.com\/(\w|\W|\d)+/)) {
-        if (ctx.session) {
-          ctx.session.lookups++;
-        }
+  if (!ctx.message.text) return;
+  if (!isWhitelisted(ctx)) {
+    return ctx.reply(ctx.i18n.t("errors.whitelist"));
+  }
 
-        ctx.replyWithChatAction("typing");
+  let urls = getUrls(ctx.message.text);
 
-        download(ctx, url);
+  urls.forEach((url) => {
+    if (url.match(/https??:\/\/(vm\.)??tiktok\.com\/(\w|\W|\d)+/)) {
+      if (ctx.session) {
+        ctx.session.lookups++;
       }
-    });
-  }
+
+      ctx.replyWithChatAction("typing");
+
+      download(ctx, url);
+    }
+  });
 };
 
 // Lookup process

locales/en.yaml

@@ -20,3 +20,4 @@ errors:
   url: I can download only TikTok videos and this doesn't look like a valid URL.
   http: Couldn't fetch shared URL. Please check if it's available and try again later.
   stream: No valid streams were found. Please try again later!
+  whitelist: You need to be whitelisted to be able to use the bot. Contact administrator.