twenty/packages/twenty-server/src/core/feature-flag/feature-flag.entity.ts

import { Field, ID, ObjectType } from '@nestjs/graphql';

import {
  Entity,
  Unique,
  PrimaryGeneratedColumn,
  Column,
  CreateDateColumn,
  UpdateDateColumn,
  ManyToOne,
} from 'typeorm';
import { IDField } from '@ptc-org/nestjs-query-graphql';

import { Workspace } from 'src/core/workspace/workspace.entity';

export enum FeatureFlagKeys {
  IsRelationFieldTypeEnabled = 'IS_RELATION_FIELD_TYPE_ENABLED',
  IsMessagingEnabled = 'IS_MESSAGING_ENABLED',
  IsNoteCreateImagesEnabled = 'IS_NOTE_CREATE_IMAGES_ENABLED',
  IsSelectFieldTypeEnabled = 'IS_SELECT_FIELD_TYPE_ENABLED',
  IsRatingFieldTypeEnabled = 'IS_RATING_FIELD_TYPE_ENABLED',
  IsWorkspaceCleanable = 'IS_WORKSPACE_CLEANABLE',
}

@Entity({ name: 'featureFlag', schema: 'core' })
@ObjectType('FeatureFlag')
@Unique('IndexOnKeyAndWorkspaceIdUnique', ['key', 'workspaceId'])
export class FeatureFlagEntity {
  @IDField(() => ID)
  @PrimaryGeneratedColumn('uuid')
  id: string;

  @Field()
  @Column({ nullable: false, type: 'text' })
  key: FeatureFlagKeys;

  @Field()
  @Column({ nullable: false, type: 'uuid' })
  workspaceId: string;

  @ManyToOne(() => Workspace, (workspace) => workspace.featureFlags, {
    onDelete: 'CASCADE',
  })
  workspace: Workspace;

  @Field()
  @Column({ nullable: false })
  value: boolean;

  @CreateDateColumn()
  createdAt: Date;

  @UpdateDateColumn()
  updatedAt: Date;
}