Rename refreshToken to appToken and add fields (#4691)

packages/twenty-server/src/engine/core-modules/app-token/app-token.auto-resolver-opts.ts

@@ -0,0 +1,39 @@
+import {
+  AutoResolverOpts,
+  PagingStrategies,
+  ReadResolverOpts,
+} from '@ptc-org/nestjs-query-graphql';
+
+import { AppToken } from 'src/engine/core-modules/app-token/app-token.entity';
+import { CreateAppTokenInput } from 'src/engine/core-modules/app-token/dtos/create-app-token.input';
+import { JwtAuthGuard } from 'src/engine/guards/jwt.auth.guard';
+
+export const appTokenAutoResolverOpts: AutoResolverOpts<
+  any,
+  any,
+  unknown,
+  unknown,
+  ReadResolverOpts<any>,
+  PagingStrategies
+>[] = [
+  {
+    EntityClass: AppToken,
+    DTOClass: AppToken,
+    CreateDTOClass: CreateAppTokenInput,
+    enableTotalCount: true,
+    pagingStrategy: PagingStrategies.CURSOR,
+    read: {
+      many: { disabled: true },
+      one: { disabled: true },
+    },
+    create: {
+      many: { disabled: true },
+    },
+    update: {
+      many: { disabled: true },
+      one: { disabled: true },
+    },
+    delete: { many: { disabled: true }, one: { disabled: true } },
+    guards: [JwtAuthGuard],
+  },
+];

packages/twenty-server/src/engine/core-modules/app-token/app-token.entity.ts

@@ -0,0 +1,71 @@
+import { Field, ID, ObjectType } from '@nestjs/graphql';
+
+import {
+  Entity,
+  Column,
+  PrimaryGeneratedColumn,
+  ManyToOne,
+  JoinColumn,
+  CreateDateColumn,
+  UpdateDateColumn,
+} from 'typeorm';
+import { BeforeCreateOne, IDField } from '@ptc-org/nestjs-query-graphql';
+
+import { BeforeCreateOneAppToken } from 'src/engine/core-modules/app-token/hooks/before-create-one-app-token.hook';
+import { User } from 'src/engine/core-modules/user/user.entity';
+import { Workspace } from 'src/engine/core-modules/workspace/workspace.entity';
+export enum AppTokenType {
+  RefreshToken = 'REFRESH_TOKEN',
+}
+
+@Entity({ name: 'appToken', schema: 'core' })
+@ObjectType('AppToken')
+@BeforeCreateOne(BeforeCreateOneAppToken)
+export class AppToken {
+  @IDField(() => ID)
+  @PrimaryGeneratedColumn('uuid')
+  id: string;
+
+  @ManyToOne(() => User, (user) => user.appTokens, {
+    onDelete: 'CASCADE',
+  })
+  @JoinColumn({ name: 'userId' })
+  user: User;
+
+  @Column()
+  userId: string;
+
+  @ManyToOne(() => Workspace, (workspace) => workspace.appTokens, {
+    onDelete: 'CASCADE',
+  })
+  @JoinColumn({ name: 'workspaceId' })
+  workspace: Workspace;
+
+  @Column({ nullable: true })
+  workspaceId: string;
+
+  @Field()
+  @Column({ nullable: false, type: 'text', default: AppTokenType.RefreshToken })
+  type: AppTokenType;
+
+  @Column({ nullable: true, type: 'text' })
+  value: string;
+
+  @Field()
+  @Column('timestamp with time zone')
+  expiresAt: Date;
+
+  @Column('timestamp with time zone', { nullable: true })
+  deletedAt: Date | null;
+
+  @Column('timestamp with time zone', { nullable: true })
+  revokedAt: Date | null;
+
+  @Field()
+  @CreateDateColumn({ type: 'timestamp with time zone' })
+  createdAt: Date;
+
+  @Field()
+  @UpdateDateColumn({ type: 'timestamp with time zone' })
+  updatedAt: Date;
+}

packages/twenty-server/src/engine/core-modules/app-token/app-token.module.ts

@@ -0,0 +1,19 @@
+import { Module } from '@nestjs/common';
+
+import { NestjsQueryGraphQLModule } from '@ptc-org/nestjs-query-graphql';
+import { NestjsQueryTypeOrmModule } from '@ptc-org/nestjs-query-typeorm';
+
+import { AppToken } from 'src/engine/core-modules/app-token/app-token.entity';
+import { appTokenAutoResolverOpts } from 'src/engine/core-modules/app-token/app-token.auto-resolver-opts';
+import { AppTokenService } from 'src/engine/core-modules/app-token/services/app-token.service';
+
+@Module({
+  imports: [
+    NestjsQueryGraphQLModule.forFeature({
+      imports: [NestjsQueryTypeOrmModule.forFeature([AppToken], 'core')],
+      services: [AppTokenService],
+      resolvers: appTokenAutoResolverOpts,
+    }),
+  ],
+})
+export class AppTokenModule {}

packages/twenty-server/src/engine/core-modules/app-token/dtos/create-app-token.input.ts

@@ -0,0 +1,11 @@
+import { Field, InputType } from '@nestjs/graphql';
+
+import { IsDate, IsNotEmpty } from 'class-validator';
+
+@InputType()
+export class CreateAppTokenInput {
+  @IsDate()
+  @IsNotEmpty()
+  @Field()
+  expiresAt: Date;
+}

packages/twenty-server/src/engine/core-modules/app-token/hooks/before-create-one-app-token.hook.ts

@@ -0,0 +1,25 @@
+import {
+  BeforeCreateOneHook,
+  CreateOneInputType,
+} from '@ptc-org/nestjs-query-graphql';
+import { v4 as uuidv4 } from 'uuid';
+
+import { AppToken } from 'src/engine/core-modules/app-token/app-token.entity';
+
+export class BeforeCreateOneAppToken<T extends AppToken>
+  implements BeforeCreateOneHook<T, any>
+{
+  async run(
+    instance: CreateOneInputType<T>,
+    context: any,
+  ): Promise<CreateOneInputType<T>> {
+    const userId = context?.req?.user?.user?.id;
+
+    instance.input.userId = userId;
+    // FIXME: These fields should be autogenerated, we need to run a migration for this
+    instance.input.id = uuidv4();
+    instance.input.updatedAt = new Date();
+
+    return instance;
+  }
+}

packages/twenty-server/src/engine/core-modules/app-token/services/app-token.service.spec.ts

@@ -0,0 +1,27 @@
+import { Test, TestingModule } from '@nestjs/testing';
+import { getRepositoryToken } from '@nestjs/typeorm';
+
+import { AppTokenService } from 'src/engine/core-modules/app-token/services/app-token.service';
+import { AppToken } from 'src/engine/core-modules/app-token/app-token.entity';
+
+describe('AppTokenService', () => {
+  let service: AppTokenService;
+
+  beforeEach(async () => {
+    const module: TestingModule = await Test.createTestingModule({
+      providers: [
+        AppTokenService,
+        {
+          provide: getRepositoryToken(AppToken),
+          useValue: {},
+        },
+      ],
+    }).compile();
+
+    service = module.get<AppTokenService>(AppTokenService);
+  });
+
+  it('should be defined', () => {
+    expect(service).toBeDefined();
+  });
+});

packages/twenty-server/src/engine/core-modules/app-token/services/app-token.service.ts

@@ -0,0 +1,5 @@
+import { TypeOrmQueryService } from '@ptc-org/nestjs-query-typeorm';
+
+import { AppToken } from 'src/engine/core-modules/app-token/app-token.entity';
+
+export class AppTokenService extends TypeOrmQueryService<AppToken> {}