doccano/frontend/domain/models/statistics/statistics.ts

export type Label = {[key: string]: number}

export type User = {[key: string]: number}

export type ConfirmedCount = {[key: string]: number}

export class Statistics {
  constructor(
    public label:      Label,
    public userLabel:  User,
    public total:      number,
    public remaining:  number,
    public user:       User,
    public confirmedCount: ConfirmedCount,
  ) {}

  static valueOf(
    { label, user_label, total, remaining, user, confirmed_count }:
    {
      label:      Label,
      user_label: User,
      total:      number,
      remaining:  number,
      user:       User,
      confirmed_count: ConfirmedCount,
    }
  ): Statistics {
    return new Statistics(label, user_label, total, remaining, user, confirmed_count)
  }

  private makeData(object: Label | User, label: string) {
    const labels = object ? Object.keys(object) : []
    const counts = object ? Object.values(object) : []
    return {
      labels,
      datasets: [{
        label,
        backgroundColor: '#00d1b2',
        data: counts
      }]
    }
  }

  public labelStats(label: string) {
    return this.makeData(this.label, label)
  }

  public userStats(label: string) {
    return this.makeData(this.user, label)
  }

  public progress(labels: string[]) {
    const complete = this.total - this.remaining
    const incomplete = this.remaining
    return {
      datasets: [{
        data: [complete, incomplete],
        backgroundColor: ['#00d1b2', '#ffdd57']
      }],
      labels
    }
  }

  private makeProgressData(roleName: string, labels: string[]) {
    const confirmed = this.confirmedCount[roleName]
    const unconfirmed = this.total - confirmed
    return {
      datasets: [{
        data: [confirmed, unconfirmed],
        backgroundColor: ['#00d1b2', '#ffdd57']
      }],
      labels
    }
  }

  public annotatorProgress(labels: string[]) {
    return this.makeProgressData('annotator', labels)
  }

  public approverProgress(labels: string[]) {
    return this.makeProgressData('annotation_approver', labels)
  }

  public adminProgress(labels: string[]) {
    return this.makeProgressData('project_admin', labels)
  }
}