doccano/frontend/services/application/tasks/textClassification/textClassificationApplicationService.ts

import { Category } from '@/domain/models/tasks/category'
import { AnnotationApplicationService } from '../annotationApplicationService'
import { TextClassificationDTO } from './textClassificationData'

export class TextClassificationService extends AnnotationApplicationService<Category> {
  public async list(projectId: string, docId: number): Promise<TextClassificationDTO[]> {
    const items = await this.repository.list(projectId, docId)
    return items.map((item) => new TextClassificationDTO(item))
  }

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