import { FromApiSeq2seqRepository } from '@/repositories/tasks/seq2seq/api'
import { Seq2seqLabel } from '@/models/tasks/seq2seq'
import { AnnotationApplicationService } from './annotationService'
import annotation from '~/i18n/de/projects/annotation'

export class Seq2seqDTO {
  id: number
  text: string
  user: number

  constructor(item: Seq2seqLabel) {
    this.id = item.id
    this.text = item.text
    this.user = item.user
  }
}

export class Seq2seqApplicationService extends AnnotationApplicationService<Seq2seqLabel> {
  constructor(
    readonly repository: FromApiSeq2seqRepository
  ) {
    super(new FromApiSeq2seqRepository())
  }

  public async list(projectId: string, docId: number): Promise<Seq2seqDTO[]> {
    const items = await this.repository.list(projectId, docId)
    return items.map(item => new Seq2seqDTO(item))
  }

  public async create(projectId: string, docId: number, text: string): Promise<void> {
    const item = new Seq2seqLabel(0, text, 0)
    await this.repository.create(projectId, docId, item)
  }

  public async changeText(projectId: string, docId: number, annotationId: number, text: string): Promise<void> {
    await this.repository.update(projectId, docId, annotationId, text)
  }
}