added models

Author: dennohpeter

src/index.ts

@@ -8,6 +8,7 @@ const chalkTable = require('chalk-table');
 import BigNumber from "bignumber.js";
 import { buildTradeMsg, flat } from "./utils";
 import { MONITORED_TOKENS } from "./data/token";
+import { Approve } from "./models";
 
 const Main = async () => {
     const oneInch = new OneInch()

src/models/approve.ts

@@ -0,0 +1,43 @@
+import { Model, Document, model, Schema } from "mongoose";
+import { Token } from "../types";
+
+// An interface that describes attributes that a transaction should have
+interface ApproveAttrs {
+    token: Token;
+}
+
+// An interface that describes what attributes a transaction model should have
+interface ApproveModel extends Model<ApproveDoc> {
+    build(attrs: ApproveAttrs): ApproveDoc;
+}
+
+// An interface that descibes single transaction properties
+interface ApproveDoc extends Document {
+    token: Token;
+}
+
+// Creating transaction schema
+const approveSchema = new Schema(
+    {
+        token: {
+            type: {
+                name: String,
+                symbol: String,
+                address: String
+            }, unique: true
+        },
+    },
+    {
+        timestamps: true,
+    }
+);
+
+// Statics
+approveSchema.statics.build = (attrs: ApproveAttrs) => {
+    return new Approve(attrs);
+};
+
+// Creating transaction model
+const Approve = model<ApproveDoc & ApproveModel>("Approve", approveSchema);
+
+export { Approve, ApproveDoc };

src/models/index.ts

@@ -0,0 +1,2 @@
+export * from './approve'
+export * from './user'

src/models/user.ts

@@ -0,0 +1,51 @@
+import { Schema, model, Model } from 'mongoose';
+
+// An interface that describes attributes that a user should have
+interface UserAttrs {
+    tg_id: number;
+    first_name?: string;
+    last_name?: string;
+    username?: string;
+    is_bot: boolean;
+    is_active?: boolean;
+    bot_name?: string;
+}
+
+// An interface that describes what attributes a user model should have
+interface UserModel extends Model<UserDoc> {
+    build(attrs: UserAttrs): UserDoc
+}
+
+// An interface that descibes single user properties
+interface UserDoc extends Document {
+    tg_id: number;
+    first_name?: string;
+    last_name?: string;
+    username?: string;
+    is_bot: boolean;
+    is_active?: boolean;
+    last_action?: string;
+    created_at?: Date;
+    bot_name?: string;
+}
+
+// Creating user schema
+const userSchema = new Schema({
+    tg_id: { type: Number },
+    is_bot: { type: Boolean },
+    first_name: { type: String },
+    last_name: { type: String },
+    username: { type: String },
+    bot_name: { type: String },
+    is_active: { type: Boolean, default: false },
+    last_action: { type: String },
+    created_at: { type: Date, default: Date.now }
+
+})
+// Statics
+userSchema.static('build', (attrs: UserAttrs) => { return new User(attrs) })
+
+// Creating user model
+const User = model<UserDoc & UserModel>('User', userSchema)
+
+export { User, UserAttrs, UserDoc }

src/types/1inch.ts

@@ -2,7 +2,7 @@ export interface Token {
     symbol: string,
     name: string,
     address: string,
-    decimals: number,
+    decimals?: number,
     logoURI?: string
 }
 export interface Quote {