feat: adds pushes to telegram

Author: FriOne

.env.example

@@ -1 +1,3 @@
 MONGODB_URI=
+BOT_API_TOKEN=
+TELEGRAM_CHAT_ID=

global.d.ts

@@ -0,0 +1,9 @@
+declare global {
+  namespace NodeJs {
+    interface ProcessEnv {
+      MONGODB_URI?: string;
+      BOT_API_TOKEN?: string;
+      TELEGRAM_CHAT_ID?: string;
+    }
+  }
+}

scripts/commands/handlers/updateUsageData.ts

@@ -1,16 +1,32 @@
 import { FeatureModel } from '@/server/models/features';
 import { FeatureUpdateModel } from '@/server/models/featureUpdates';
 import { FeatureSchemaNoTimeType } from '@/server/schemas/featureSchema';
-import { FeatureUpdateType } from '@/server/schemas/featureUpdateSchema';
+import {
+  FeatureUpdateType,
+  FeatureUpdateSchemaNoTimeType,
+} from '@/server/schemas/featureUpdateSchema';
 
 import { getLastUsageDataVersion } from '../utils/getLastUsageDataVersion';
+import { sendMessageToTelegram } from '../utils/sendMessageToTelegram';
 
-export async function updateUsageData(initial: boolean) {
+type NotificationUpdate = FeatureUpdateSchemaNoTimeType;
+
+type Params = {
+  initial: boolean;
+  sendToTelegram: boolean;
+};
+
+export async function updateUsageData({ initial, sendToTelegram }: Params) {
   if (initial) {
     await FeatureModel.deleteMany({});
     await FeatureUpdateModel.deleteMany({});
   }
 
+  const notificationUpdates: Record<FeatureUpdateType, NotificationUpdate[]> = {
+    twoPercent: [],
+    threePercent: [],
+  };
+
   for await (const lastFeatureData of getLastUsageDataVersion()) {
     const dbFeatureData = await FeatureModel.findById(lastFeatureData._id).exec();
 
@@ -38,14 +54,36 @@ export async function updateUsageData(initial: boolean) {
         continue;
       }
 
-      await FeatureUpdateModel.create({
+      const featureUpdate: FeatureUpdateSchemaNoTimeType = {
         key: lastFeatureData._id,
         title: lastFeatureData.title,
         updateType: lastDataUpdateType,
-      });
+      };
+
+      if (sendToTelegram) {
+        notificationUpdates[lastDataUpdateType].push(featureUpdate);
+      }
+
+      await FeatureUpdateModel.create(featureUpdate);
       console.info(`Feature "${lastFeatureData._id}" updated with "${lastDataUpdateType}" type`);
     }
   }
+
+  if (sendToTelegram && notificationUpdates.threePercent.length > 0) {
+    const cutUpdates = notificationUpdates.threePercent.slice(0, 100);
+    const chunkSize = 10;
+    for (let i = 0; i < cutUpdates.length; i += chunkSize) {
+      const chunk = notificationUpdates.threePercent.slice(i, i + chunkSize);
+      const message = formatTelegramMessage(chunk);
+      await sendMessageToTelegram(message);
+    }
+  }
+}
+
+function formatTelegramMessage(featureUpdates: NotificationUpdate[]) {
+  return featureUpdates
+    .map((featureUpdate) => `[${featureUpdate.title}](https://caniuse.com/${featureUpdate.key})`)
+    .join(`\n`);
 }
 
 function getNewFeatureUpdateType(lastFeatureData: FeatureSchemaNoTimeType) {

scripts/commands/updateUsageData.ts

@@ -12,17 +12,25 @@ export const command = {
 };
 
 function builder(yargs: Argv) {
-  return yargs.option('initial', {
-    type: 'boolean',
-    description:
-      'Treat all data as new, those features that pass check will be added as new update',
-    default: false,
-  });
+  return yargs
+    .option('initial', {
+      type: 'boolean',
+      description: 'Removes all feature data from db and add everything from scratch',
+      default: false,
+    })
+    .option('telegram', {
+      type: 'boolean',
+      description: 'Send new updates to telegram chat',
+      default: false,
+    });
 }
 
 async function handler(argv: Record<string, unknown>) {
   try {
-    await updateUsageData(!!argv.initial);
+    await updateUsageData({
+      initial: !!argv.initial,
+      sendToTelegram: !!argv.telegram,
+    });
   } catch (error) {
     console.error('Error while updating usage data', error);
   }

scripts/commands/utils/sendMessageToTelegram.ts

@@ -0,0 +1,21 @@
+import { URLSearchParams } from 'url';
+
+export async function sendMessageToTelegram(message: string) {
+  const token = process.env.BOT_API_TOKEN;
+  const chatId = process.env.TELEGRAM_CHAT_ID;
+
+  if (!token) {
+    throw new Error(`Telegram bot api token is not set`);
+  }
+  if (!chatId) {
+    throw new Error(`Telegram chat id is not set`);
+  }
+
+  const url = `https://api.telegram.org/bot${token}/sendMessage`;
+  const body = new URLSearchParams({ chat_id: chatId, text: message, parse_mode: 'markdown' });
+  const response = await fetch(url, { method: 'POST', body });
+
+  if (!response.ok || response.status !== 200) {
+    throw new Error(`Can't send telegram message "${response.statusText}"`);
+  }
+}

src/server/schemas/featureUpdateSchema.ts

@@ -11,4 +11,8 @@ export const FeatureUpdateSchema = new Schema(
 );
 
 export type FeatureUpdateSchemaType = InferSchemaType<typeof FeatureUpdateSchema>;
+export type FeatureUpdateSchemaNoTimeType = Omit<
+  FeatureUpdateSchemaType,
+  'createdAt' | 'updatedAt'
+>;
 export type FeatureUpdateType = FeatureUpdateSchemaType['updateType'];