visible/visible

import { Column, Entity, ManyToOne, PrimaryColumn } from 'typeorm';

import {
  Difficulty,
  Impact,
  Location,
  Outcome,
  Report,
} from '../../../domain/models';
import { RuleDBEntity } from '../rule';
import { SourceDBEntity } from './source-db-entity';

@Entity('report')
export class ReportDBEntity {
  @PrimaryColumn('uuid')
  id!: string;

  @Column('varchar', { length: 255 })
  outcome!: Outcome;

  @Column('varchar', { length: 255, nullable: true })
  impact?: Impact;

  @Column('varchar', { length: 255, nullable: true })
  difficulty?: Difficulty;

  @Column('text', { nullable: true })
  target?: string;

  @Column('varchar', { length: 255, nullable: true })
  message?: string;

  @Column('uuid')
  sourceId!: string;

  @Column('uuid')
  ruleId!: string;

  @Column('json', { nullable: true })
  location?: Location;

  @Column('text', { nullable: true })
  diffHunk?: string;

  @Column('text', { nullable: true })
  screenshot?: string;

  @ManyToOne(() => SourceDBEntity, { onDelete: 'CASCADE' })
  readonly source?: SourceDBEntity;

  @ManyToOne(() => RuleDBEntity)
  readonly rule?: RuleDBEntity;

  static fromDomain(report: Report): ReportDBEntity {
    const entity = new ReportDBEntity();
    entity.id = report.id;
    entity.outcome = report.outcome;
    entity.difficulty = report.difficulty;
    entity.impact = report.impact;
    entity.target = report.target;
    entity.message = report.message;
    entity.sourceId = report.sourceId;
    entity.ruleId = report.ruleId;
    entity.diffHunk = report.diffHunk;
    entity.screenshot = report.screenshot;
    entity.location = report.location;
    return entity;
  }

  toDomain(): Report {
    return Report.from({
      id: this.id,
      outcome: this.outcome,
      impact: this.impact,
      difficulty: this.difficulty,
      ruleId: this.ruleId,
      target: this.target,
      message: this.message,
      sourceId: this.sourceId,
      diffHunk: this.diffHunk,
      screenshot: this.screenshot,
      location: this.location,
    });
  }
}