import { createUnionType, Field, Int, ObjectType, registerEnumType } from "@nestjs/graphql";
import { BaseDTO } from "../utils/base.dto";
import { DescriptionChecklistObjectType } from "./description-types/description-checklist/description-checklist.dto";
import { DescriptionTextObjectType } from "./description-types/description-text/description-text.dto";

export enum DescriptionType {
  Text = 'Text',
  Checklist = 'Checklist'
}

registerEnumType(DescriptionType, {
  name: 'DescriptionType',
});

@ObjectType()
export class DescriptionObjectType extends BaseDTO {
  @Field(() => Int)
  description_id: number;

  @Field(() => DescriptionType)
  description_type: DescriptionType;

  @Field(() => Int)
  task_id: number;
}

export const DescriptionableUnion = createUnionType({
  name: 'DescriptionableUnion',
  types: () => [DescriptionTextObjectType, DescriptionChecklistObjectType] as const,
  resolveType(value) {
    if (value['__UnionDescriminator__'] === DescriptionType.Text) {
      return DescriptionTextObjectType;
    } else {
      return DescriptionChecklistObjectType;
    }
  },
});