Merge pull request #206 from GeneralMagicio/add-vesting-table

added vesting table

Author: lovelgeorge99

migration/1746211907434-AddVestingTablemigration.ts

@@ -0,0 +1,41 @@
+import { MigrationInterface, QueryRunner } from 'typeorm';
+
+export class AddVestingTableMigration1746211907434
+  implements MigrationInterface
+{
+  name = 'Migration1746211907434';
+
+  public async up(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `CREATE TABLE "vesting_data" ("id" SERIAL NOT NULL, "status" text NOT NULL DEFAULT 'pending', "walletAddress" character varying, "paymentToken" character varying NOT NULL, "amount" bigint NOT NULL, "rewardStreamStart" TIMESTAMP, "cliff" double precision, "rewardStreamEnd" TIMESTAMP, "createdAt" TIMESTAMP NOT NULL DEFAULT now(), "projectId" integer, "userId" integer, CONSTRAINT "PK_c8abbf61fa05ae0eec4048b6408" PRIMARY KEY ("id"))`,
+    );
+    await queryRunner.query(
+      `CREATE INDEX "IDX_2826daca8f0317c5e209df4de5" ON "vesting_data" ("projectId") `,
+    );
+    await queryRunner.query(
+      `CREATE INDEX "IDX_b81d2052f429a1bce90dc68e2e" ON "vesting_data" ("userId") `,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "vesting_data" ADD CONSTRAINT "FK_2826daca8f0317c5e209df4de55" FOREIGN KEY ("projectId") REFERENCES "project"("id") ON DELETE NO ACTION ON UPDATE NO ACTION`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "vesting_data" ADD CONSTRAINT "FK_b81d2052f429a1bce90dc68e2ee" FOREIGN KEY ("userId") REFERENCES "user"("id") ON DELETE CASCADE ON UPDATE NO ACTION`,
+    );
+  }
+
+  public async down(queryRunner: QueryRunner): Promise<void> {
+    await queryRunner.query(
+      `ALTER TABLE "vesting_data" DROP CONSTRAINT "FK_b81d2052f429a1bce90dc68e2ee"`,
+    );
+    await queryRunner.query(
+      `ALTER TABLE "vesting_data" DROP CONSTRAINT "FK_2826daca8f0317c5e209df4de55"`,
+    );
+    await queryRunner.query(
+      `DROP INDEX "public"."IDX_b81d2052f429a1bce90dc68e2e"`,
+    );
+    await queryRunner.query(
+      `DROP INDEX "public"."IDX_2826daca8f0317c5e209df4de5"`,
+    );
+    await queryRunner.query(`DROP TABLE "vesting_data"`);
+  }
+}

src/entities/entities.ts

@@ -39,6 +39,7 @@ import { AnkrState } from './ankrState';
 import { SwapTransaction } from './swapTransaction';
 import { QaccPointsHistory } from './qaccPointsHistory';
 import { UserRankMaterializedView } from './userRanksMaterialized';
+import { VestingData } from './vestingData';
 
 export const getEntities = (): DataSourceOptions['entities'] => {
   return [
@@ -91,5 +92,6 @@ export const getEntities = (): DataSourceOptions['entities'] => {
     AnkrState,
     SwapTransaction,
     UserRankMaterializedView,
+    VestingData,
   ];
 };

src/entities/vestingData.ts

@@ -0,0 +1,66 @@
+import { Field, ID, ObjectType } from 'type-graphql';
+import {
+  BaseEntity,
+  Column,
+  CreateDateColumn,
+  Entity,
+  Index,
+  ManyToOne,
+  PrimaryGeneratedColumn,
+} from 'typeorm';
+import { User } from './user';
+import { Project } from './project';
+
+export const VESTING_STATUS = {
+  PENDING: 'pending',
+  FINAL: 'FINAL',
+};
+
+@ObjectType()
+@Entity('vesting_data')
+export class VestingData extends BaseEntity {
+  @Field(_type => ID)
+  @PrimaryGeneratedColumn()
+  readonly id: number;
+
+  @Index()
+  @Field(_type => Project)
+  @ManyToOne(_type => Project, { eager: true })
+  project: Project;
+
+  @Field()
+  @Column('text', { default: VESTING_STATUS.PENDING })
+  status: string;
+
+  @Index()
+  @Field(_type => User, { nullable: true })
+  @ManyToOne(() => User, { eager: true, onDelete: 'CASCADE' })
+  user: User;
+
+  @Field(_type => String, { nullable: true })
+  @Column({ nullable: true })
+  walletAddress?: string;
+
+  @Field()
+  @Column()
+  paymentToken: string;
+
+  @Field()
+  @Column({ type: 'bigint' })
+  amount: string;
+
+  @Field({ nullable: true })
+  @Column({ nullable: true })
+  rewardStreamStart?: Date;
+
+  @Field({ nullable: true })
+  @Column({ type: 'float', nullable: true })
+  cliff?: number;
+
+  @Field({ nullable: true })
+  @Column({ nullable: true })
+  rewardStreamEnd?: Date;
+
+  @CreateDateColumn()
+  createdAt: Date;
+}