import { AnnotationApplicationService } from '../annotationApplicationService'
import { RelationDTO } from './relationData'
import { SpanDTO } from './sequenceLabelingData'
import { APISequenceLabelingRepository } from '~/repositories/tasks/sequenceLabeling/apiSequenceLabeling'
import { Span } from '~/domain/models/tasks/sequenceLabeling'
import { RelationRepository } from '~/domain/models/tasks/relationRepository'
import { RelationItem } from '~/domain/models/tasks/relation'

export class SequenceLabelingApplicationService extends AnnotationApplicationService<Span> {
  constructor(
    readonly repository: APISequenceLabelingRepository,
    readonly relationRepository: RelationRepository
  ) {
    super(new APISequenceLabelingRepository())
  }

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

  public async create(
    projectId: string,
    docId: number,
    labelId: number,
    startOffset: number,
    endOffset: number
  ): Promise<void> {
    const item = new Span(0, labelId, 0, startOffset, endOffset)
    try {
      await this.repository.create(projectId, docId, item)
    } catch (e: any) {
      console.log(e.response.data.detail)
    }
  }

  public async changeLabel(
    projectId: string,
    docId: number,
    annotationId: number,
    labelId: number
  ): Promise<void> {
    try {
      await this.repository.update(projectId, docId, annotationId, labelId)
    } catch (e: any) {
      console.log(e.response.data.detail)
    }
  }

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

  public async createRelation(
    projectId: string,
    docId: number,
    fromId: number,
    toId: number,
    typeId: number
  ): Promise<void> {
    const relation = new RelationItem(0, fromId, toId, typeId)
    await this.relationRepository.create(projectId, docId, relation)
  }

  public async deleteRelation(projectId: string, docId: number, relationId: number): Promise<void> {
    await this.relationRepository.delete(projectId, docId, relationId)
  }

  public async updateRelation(
    projectId: string,
    docId: number,
    relationId: number,
    typeId: number
  ): Promise<void> {
    await this.relationRepository.update(projectId, docId, relationId, typeId)
  }
}