feat: wasabi vista user (#49)

* feat: wasabi vista user

* chore: check code

* fix: remove code from log

* chore: updates

Author: hmbanan666

apps/web-app/server/services/telegram/wasabi-vista.ts

@@ -1,3 +1,4 @@
+import { repository } from '@roll-stack/database'
 import { Bot } from 'grammy'
 
 const logger = useLogger('telegram:wasabi-vista')
@@ -12,27 +13,50 @@ export async function useCreateWasabiVistaBot() {
     // const locale = ctx.message.from.language_code
 
     if (ctx.hasCommand('start')) {
-      // Welcome message with buttons
-      await ctx.reply(
-        `Ключ доступа: ${generateAccessCode()}`,
-        // {
-        //   reply_markup: {
-        //     inline_keyboard: [
-        //       [{ text: dictionary(locale).bots.woodland.title, url: woodlandsBotUrl }],
-        //       [{ text: dictionary(locale).bots.subscribeToChannel, url: gameChannelUrl }],
-        //       [{ text: dictionary(locale).bots.chatgame.playingOnTwitch, url: twitchUrl }],
-        //     ],
-        //   },
-        // },
-      )
+      // Not private chat?
+      if (ctx.message.chat.type !== 'private') {
+        await ctx.reply('Я пока не умею отвечать на групповые сообщения.')
+        return
+      }
 
+      // Find user
+      const wasabiVistaUser = await repository.wasabiVista.findUserByTelegramId(ctx.message.from.id.toString())
+      if (!wasabiVistaUser) {
+        const accessKey = await generateAccessCode()
+
+        const createdUser = await repository.wasabiVista.createUser({
+          telegramId: ctx.message.from.id.toString(),
+          accessKey,
+          firstName: ctx.message.from.first_name,
+          lastName: ctx.message.from.last_name,
+          username: ctx.message.from.username,
+        })
+
+        logger.log('new user', createdUser.id, ctx.message.from.id, ctx.message.text)
+
+        await ctx.reply(`Ключ доступа: ${accessKey}`)
+
+        return
+      }
+
+      if (!wasabiVistaUser.user) {
+        await ctx.reply('Нет доступа. Используйте ранее полученный Ключ доступа. Или передайте его в службу поддержки.')
+        return
+      }
+
+      await ctx.reply('Вы уже авторизованы.')
       return
     }
 
     logger.log(ctx.message.from.id, ctx.message.text)
     ctx.reply('Я пока не умею отвечать на сообщения.')
   })
 
+  // Somebody invited bot to a group
+  bot.on('my_chat_member', async (ctx) => {
+    logger.log(ctx.chat, ctx.chatMember, ctx.message)
+  })
+
   try {
     await bot.start()
     logger.info('Wasabi Vista bot started successfully')
@@ -54,6 +78,17 @@ export async function notifyWasabiVistaAdmin(message: string) {
   return useWasabiVistaBot().api.sendMessage(telegram.adminId, message)
 }
 
-function generateAccessCode(): string {
-  return getRandInteger(100000, 999999).toString()
+async function generateAccessCode(): Promise<string> {
+  let selectedCode
+
+  // Code should be unique
+  while (!selectedCode) {
+    const code = getRandInteger(100000, 999999).toString()
+    const wasabiVistaUser = await repository.wasabiVista.findUserByKey(code)
+    if (!wasabiVistaUser) {
+      selectedCode = code
+    }
+  }
+
+  return selectedCode
 }

packages/database/src/repository/index.ts

@@ -19,6 +19,7 @@ import { Print } from './print'
 import { Product } from './product'
 import { Task } from './task'
 import { User } from './user'
+import { WasabiVista } from './wasabi-vista'
 
 class Repository {
   readonly channel = Channel
@@ -41,6 +42,7 @@ class Repository {
   readonly product = Product
   readonly task = Task
   readonly user = User
+  readonly wasabiVista = WasabiVista
 
   async checkHealth(): Promise<boolean> {
     await useDatabase().query.users.findFirst()

packages/database/src/repository/wasabi-vista.ts

@@ -0,0 +1,48 @@
+import type { WasabiVistaUserDraft } from '../types'
+import { eq, sql } from 'drizzle-orm'
+import { useDatabase } from '../database'
+import { wasabiVistaUsers } from '../tables'
+
+export class WasabiVista {
+  static async findUser(id: string) {
+    return useDatabase().query.wasabiVistaUsers.findFirst({
+      where: (users, { eq }) => eq(users.id, id),
+    })
+  }
+
+  static async findUserByKey(key: string) {
+    return useDatabase().query.wasabiVistaUsers.findFirst({
+      where: (users, { eq }) => eq(users.accessKey, key),
+    })
+  }
+
+  static async findUserByTelegramId(telegramId: string) {
+    return useDatabase().query.wasabiVistaUsers.findFirst({
+      where: (users, { eq }) => eq(users.telegramId, telegramId),
+      with: {
+        user: true,
+      },
+    })
+  }
+
+  static async createUser(data: WasabiVistaUserDraft) {
+    const [user] = await useDatabase().insert(wasabiVistaUsers).values(data).returning()
+    return user
+  }
+
+  static async updateUser(id: string, data: Partial<WasabiVistaUserDraft>) {
+    const [user] = await useDatabase()
+      .update(wasabiVistaUsers)
+      .set({
+        ...data,
+        updatedAt: sql`now()`,
+      })
+      .where(eq(wasabiVistaUsers.id, id))
+      .returning()
+    return user
+  }
+
+  static async deleteUser(id: string) {
+    return useDatabase().delete(wasabiVistaUsers).where(eq(wasabiVistaUsers.id, id))
+  }
+}

packages/database/src/tables.ts

@@ -589,6 +589,21 @@ export const networkMetrics = pgTable('network_metrics', {
   averageCheck: numeric('average_check', { mode: 'number' }).notNull().default(0),
 })
 
+export const wasabiVistaUsers = pgTable('wasabi_vista_users', {
+  id: cuid2('id').defaultRandom().primaryKey(),
+  createdAt: timestamp('created_at', { precision: 3, withTimezone: true, mode: 'string' }).notNull().defaultNow(),
+  updatedAt: timestamp('updated_at', { precision: 3, withTimezone: true, mode: 'string' }).notNull().defaultNow(),
+  accessKey: varchar('access_key').notNull().unique(),
+  telegramId: varchar('telegram_id').notNull().unique(),
+  firstName: varchar('first_name'),
+  lastName: varchar('last_name'),
+  username: varchar('username'),
+  userId: cuid2('user_id').references(() => users.id, {
+    onDelete: 'cascade',
+    onUpdate: 'cascade',
+  }),
+})
+
 export const userRelations = relations(users, ({ many, one }) => ({
   chatMessages: many(chatMessages),
   chatMembers: many(chatMembers),
@@ -601,6 +616,7 @@ export const userRelations = relations(users, ({ many, one }) => ({
   }),
   postLikes: many(postLikes),
   postComments: many(postComments),
+  wasabiVistaUsers: many(wasabiVistaUsers),
 }))
 
 export const partnerRelations = relations(partners, ({ many, one }) => ({
@@ -943,3 +959,10 @@ export const clientReviewRelations = relations(clientReviews, ({ one }) => ({
     references: [kitchens.id],
   }),
 }))
+
+export const wasabiVistaUserRelations = relations(wasabiVistaUsers, ({ one }) => ({
+  user: one(users, {
+    fields: [wasabiVistaUsers.userId],
+    references: [users.id],
+  }),
+}))

packages/database/src/types.ts

@@ -128,3 +128,6 @@ export type ClientReviewDraft = InferInsertModel<typeof tables.clientReviews>
 
 export type NetworkMetrics = InferSelectModel<typeof tables.networkMetrics>
 export type NetworkMetricsDraft = InferInsertModel<typeof tables.networkMetrics>
+
+export type WasabiVistaUser = InferSelectModel<typeof tables.wasabiVistaUsers>
+export type WasabiVistaUserDraft = InferInsertModel<typeof tables.wasabiVistaUsers>