add create subscription

.env.sample

@@ -1,2 +1,3 @@
 TOKEN=123
-MONGO=mongodb://localhost:27017/tlgcoin
+MONGO=mongodb://localhost:27017/tlgcoin
+WEB3_PROVIDER=http://localhost:27017

.prettierrc.json

@@ -0,0 +1,6 @@
+{
+ "trailingComma": "es5",
+ "tabWidth": 2,
+ "semi": false,
+ "singleQuote": true
+}

README.md

@@ -16,6 +16,7 @@ And you should be good to go! Feel free to fork and submit pull requests. Thanks
 
 - `TOKEN` — Telegram bot token
 - `MONGO`— URL of the mongo database
+- `WEB3_PROVIDER` - Your local/remote Ethereum node
 
 Also, please, consider looking at `.env.sample`.
 

locales/en.yaml

@@ -5,3 +5,4 @@ language: Please, select the language.
 language_selected: <b>Bot</b> now speaks English.
 notifications_on: Notifications enabled
 notifications_off: Notifications disabled
+subscription_message: In order to subscribe to this group, you need to transfer 1 ETH to ${subscriptionAddress}.

locales/ru.yaml

@@ -44,3 +44,4 @@ start_group_admin: Администратор группы подтвержде
 start_group_not_admin: Простите, эта ссылка существует только для администраторов сообществ.
 notifications_on: Уведомления включены
 notifications_off: Уведомления отключены
+subscription_message: Чтобы подписаться на этоу группу, вам нужно перевести 1 ETH по адресу ${subscriptionAddress}.

package.json

@@ -27,7 +27,8 @@
  "mongoose": "5.10.18",
  "telegraf": "^4.3.0",
  "telegraf-i18n": "^6.6.0",
- "typescript": "^4.3.2"
+ "typescript": "^4.3.2",
+ "web3": "^1.4.0"
  },
  "_moduleAliases": {
  "@": "dist"

src/app.ts

@@ -15,6 +15,7 @@ import { attachChat } from '@/middlewares/attachChat'
 import { handleMyChatMember } from '@/handlers/handleMyChatMember'
 import { toggleNotifications } from '@/handlers/toggleNotifications'
 import { stopIfPrivate } from '@/middlewares/stopIfPrivate'
+import { runMongo } from '@/models/index'
 
 // Middlewares
 bot.use(ignoreOldMessageUpdates)
@@ -32,6 +33,9 @@ bot.on('my_chat_member', stopIfPrivate, handleMyChatMember)
 // Errors
 bot.catch(console.error)
 // Start bot
+runMongo().then(() => {
+ console.log('Mongo connected')
+})
 bot.launch().then(() => {
  console.info(`Bot ${bot.botInfo.username} is up and running`)
 })

src/handlers/handleStart.ts

@@ -1,5 +1,6 @@
 import { Context } from 'telegraf'
 import { MessageAfterLanguage, sendLanguage } from '@/handlers/language'
+import { getOrCreateSubscription } from '@/models/Subscription'
 
 export function handleStart(ctx: Context) {
  return sendStart(ctx)
@@ -20,11 +21,25 @@ export async function sendStart(ctx: Context) {
 
  const allowedStatuses = ['administrator', 'creator']
  if (allowedStatuses.includes(chatMemberInfo.status)) {
- return ctx.reply(ctx.i18n.t('start_group_admin'))
+ ctx.reply(ctx.i18n.t('start_group_admin'))
+ const subscription = await getOrCreateSubscription(
+ ctx.from.id,
+ ctx.chat.id
+ )
+ ctx.reply(
+ ctx.i18n.t('subscription_message', {
+ subscriptionAddress: subscription.accounts.eth.address,
+ })
+ )
  } else {
  return ctx.reply(ctx.i18n.t('start_group_not_admin'))
  }
  }
 
- return ctx.replyWithHTML(ctx.i18n.t('help'))
+ const subscription = await getOrCreateSubscription(ctx.from.id, ctx.chat.id)
+ ctx.reply(
+ ctx.i18n.t('subscription_message', {
+ subscriptionAddress: subscription.accounts.eth.address,
+ })
+ )
 }

src/helpers/web3.ts

@@ -0,0 +1,3 @@
+import Web3 from 'web3'
+
+export const web3 = new Web3(Web3.givenProvider || process.env.WEB3_PROVIDER)

src/models/Chat.ts

@@ -10,9 +10,7 @@ export class Chat {
 }
 
 // Get Chat model
-const ChatModel = getModelForClass(Chat, {
- schemaOptions: { timestamps: true },
-})
+const ChatModel = getModelForClass(Chat)
 
 // Get or create chat
 export async function findChat(id: number) {

src/models/Subscription.ts

@@ -0,0 +1,56 @@
+import {
+ prop,
+ getModelForClass,
+ Severity,
+ setGlobalOptions,
+} from '@typegoose/typegoose'
+import { web3 } from '@/helpers/web3'
+
+interface Accounts {
+ eth: {
+ address: string
+ privateKey: string
+ }
+}
+
+interface Prices {
+ monthly: { eth: number }
+}
+
+setGlobalOptions({
+ schemaOptions: { timestamps: true },
+ options: { allowMixed: Severity.ALLOW },
+})
+
+export class Subscription {
+ @prop({ required: true })
+ userId: number
+ @prop({ required: true })
+ chatId: number
+ @prop({ required: true, unique: true })
+ accounts: Accounts
+ @prop()
+ prices?: Prices
+}
+
+export const SubscriptionModel = getModelForClass(Subscription)
+
+export async function getOrCreateSubscription(userId: number, chatId: number) {
+ let subscription = await SubscriptionModel.findOne({
+ chatId,
+ })
+
+ if (subscription) {
+ return subscription
+ }
+
+ const ethAccount = web3.eth.accounts.create()
+
+ subscription = await SubscriptionModel.create({
+ userId,
+ chatId,
+ accounts: { eth: ethAccount },
+ })
+
+ return subscription
+}

src/models/index.ts

@@ -1,9 +1,12 @@
-import * as mongoose from 'mongoose'
+import mongoose from 'mongoose'
 
-mongoose.connect(process.env.MONGO, {
- useNewUrlParser: true,
- useUnifiedTopology: true,
-})
-mongoose.set('useCreateIndex', true)
+export function runMongo(mongoUrl = process.env.MONGO) {
+ return mongoose.connect(mongoUrl, {
+ useNewUrlParser: true,
+ useUnifiedTopology: true,
+ useCreateIndex: true,
+ useFindAndModify: true,
+ })
+}
 
 export * from './Chat'

tsconfig.json

@@ -11,7 +11,9 @@
  "@/*": ["*"]
  },
  "emitDecoratorMetadata": true,
- "experimentalDecorators": true
+ "experimentalDecorators": true,
+ "allowSyntheticDefaultImports": true,
+ "esModuleInterop": true
  },
  "include": ["src/**/*"]
 }