ElectronicBabylonianLiterature/ebl-frontend

import AfoRegisterRecord, {
  AfoRegisterRecordSuggestion,
} from 'afo-register/domain/Record'
import Bluebird from 'bluebird'
import Promise from 'bluebird'
import FragmentService from 'fragmentarium/application/FragmentService'
import ApiClient from 'http/ApiClient'

function createAfoRegisterRecord(data): AfoRegisterRecord {
  return new AfoRegisterRecord(data)
}

function createAfoRegisterRecordSuggestion(data): AfoRegisterRecordSuggestion {
  return new AfoRegisterRecordSuggestion(data)
}

function injectFragmentReferencesToRecords(
  records: AfoRegisterRecord[],
  fragmentService: FragmentService
): Promise<AfoRegisterRecord[]> {
  const traditionalReferences = records.map(
    (record: AfoRegisterRecord) => record.id
  )
  return fragmentService
    .queryByTraditionalReferences(traditionalReferences)
    .then((result) =>
      records.map((record) => {
        const match = result.items.find(
          (item) => item.traditionalReference === record.id
        )
        if (match) {
          return record.setFragmentNumbers(match.fragmentNumbers)
        }
        return record
      })
    )
}

export default class AfoRegisterRepository {
  private readonly apiClient: ApiClient

  constructor(apiClient: ApiClient) {
    this.apiClient = apiClient
  }

  search(
    query: string,
    fragmentService?: FragmentService
  ): Promise<AfoRegisterRecord[]> {
    return this.apiClient
      .fetchJson(`/afo-register?${query}`, false)
      .then((result) => result.map(createAfoRegisterRecord))
      .then((records) => {
        if (fragmentService) {
          return Bluebird.all(
            injectFragmentReferencesToRecords(records, fragmentService)
          )
        } else {
          return records
        }
      })
  }

  searchTextsAndNumbers(
    query: readonly string[]
  ): Promise<AfoRegisterRecord[]> {
    return this.apiClient
      .postJson(`/afo-register/texts-numbers`, query, false)
      .then((result) => result.map(createAfoRegisterRecord))
  }

  searchSuggestions(query: string): Promise<AfoRegisterRecordSuggestion[]> {
    return this.apiClient
      .fetchJson(`/afo-register/suggestions?text_query=${query}`, false)
      .then((result) => result.map(createAfoRegisterRecordSuggestion))
  }
}