capture-set.ts

import { snakeCaseToSentenceCase } from 'utils/snakeCaseToSentenceCase'
import { UserPermission } from 'utils/user/types'
import { Entity } from './entity'
import { Job } from './job'

export type ServerCaptureSet = any // TODO: Update this type

export class CaptureSet extends Entity {
  private readonly _jobs: Job[] = []

  public constructor(entity: ServerCaptureSet) {
    super(entity)

    if (this._data.jobs) {
      this._jobs = this._data.jobs.map((job: any) => new Job(job))
    }
  }

  get canPopulate(): boolean {
    return (
      this._data.user_permissions.includes(UserPermission.Populate) &&
      this._data.method !== 'starred'
    )
  }

  get currentJob(): Job | undefined {
    if (!this._jobs.length) {
      return
    }

    return this._jobs.sort((j1: Job, j2: Job) => {
      const date1 = new Date(j1.updatedAt as string)
      const date2 = new Date(j2.updatedAt as string)

      return date2.getTime() - date1.getTime()
    })[0]
  }

  get hasJobInProgress(): boolean {
    return this._jobs.some(
      (job) =>
        job.status.code === 'CREATED' ||
        job.status.code === 'PENDING' ||
        job.status.code === 'STARTED'
    )
  }

  get kwargs(): { [key: string]: string | number } {
    return this._data.kwargs || {}
  }

  get method(): string {
    return this._data.method
  }

  get methodNameDisplay(): string {
    return snakeCaseToSentenceCase(this._data.method)
  }

  get methodDetailsDisplay(): string[] {
    return Object.entries(this._data.kwargs).map(
      ([key, value]) => `${snakeCaseToSentenceCase(key)}: ${value}`
    )
  }

  get name(): string {
    return this._data.name
  }

  get numImages(): number | undefined {
    return this._data.source_images_count
  }

  get numImagesWithDetections(): number | undefined {
    return this._data.source_images_with_detections_count
  }

  get numImagesProcessed(): number | undefined {
    return this._data.source_images_processed_count
  }

  get numImagesWithDetectionsLabel(): string {
    const pct =
      this.numImagesWithDetections && this.numImages
        ? (this.numImagesWithDetections / this.numImages) * 100
        : 0

    return `${this.numImagesWithDetections?.toLocaleString()} (${pct.toFixed(
      0
    )}%)`
  }

  get numImagesProcessedLabel(): string {
    const numProcessed = this.numImagesProcessed ?? 0
    const pct =
      this.numImages && this.numImages > 0
        ? (numProcessed / this.numImages) * 100
        : 0

    return `${numProcessed.toLocaleString()} (${pct.toFixed(0)}%)`
  }

  get numJobs(): number | undefined {
    return this._data.jobs?.length
  }

  get numOccurrences(): number | undefined {
    return this._data.occurrences_count ?? undefined
  }

  get numTaxa(): number | undefined {
    return this._data.taxa_count ?? undefined
  }

  get settingsDisplay(): string {
    return snakeCaseToSentenceCase(this.method)
  }

  get settingsDetailsDisplay(): string[] {
    return Object.entries(this._data.kwargs).map(
      ([key, value]) => `${snakeCaseToSentenceCase(key)}: ${value}`
    )
  }
}