ElectronicBabylonianLiterature/ebl-frontend

import React, { useEffect, useState } from 'react'
import { immerable } from 'immer'
import { ChapterDisplay } from 'corpus/domain/chapter'
import WordService from 'dictionary/application/WordService'
import WordDownloadButton from 'common/WordDownloadButton'
import Download from 'common/Download'
import { wordExport } from 'corpus/ui/WordExport'
import Promise from 'bluebird'
import { Document } from 'docx'
import TextService from 'corpus/application/TextService'
import { RowsContextService, useRowsContext } from 'corpus/ui/RowsContext'
import {
  TranslationContextService,
  useTranslationContext,
} from 'corpus/ui/TranslationContext'

type DowndloadChapterProps = {
  chapter: ChapterDisplay
  wordService: WordService
  textService: TextService
}

export default function DownloadChapter({
  chapter,
  wordService,
  textService,
}: DowndloadChapterProps): JSX.Element {
  const baseFileName = chapter.uniqueIdentifier
  const [json, setJson] = useState<string>()
  const [atf, setAtf] = useState<string>()
  const pdfDownloadButton = <span key="pdfDownloadButton"></span>
  const rowsContext = useRowsContext(chapter.lines.length, true, true, true)
  const translationContext = useTranslationContext()
  const wordDownloadButton = (
    <WordDownloadButton
      context={
        new CorpusWordExportContext(
          chapter,
          wordService,
          textService,
          rowsContext,
          translationContext
        )
      }
      baseFileName={baseFileName}
      getWordDoc={getWordDoc}
      key="wordDownload"
    >
      Download as Word
    </WordDownloadButton>
  )
  useEffect(() => {
    const jsonUrl = URL.createObjectURL(
      new Blob([JSON.stringify(chapter, null, 2)], {
        type: 'application/json',
      })
    )
    setJson(jsonUrl)

    const atfUrl = URL.createObjectURL(
      new Blob([chapter.atf], {
        type: 'text/plain',
      })
    )
    setAtf(atfUrl)

    return (): void => {
      URL.revokeObjectURL(atfUrl)
      URL.revokeObjectURL(jsonUrl)
    }
  }, [chapter])
  return (
    <Download
      baseFileName={baseFileName}
      pdfDownloadButton={pdfDownloadButton}
      wordDownloadButton={wordDownloadButton}
      atfUrl={atf}
      jsonUrl={json}
    />
  )
}

export class CorpusWordExportContext {
  readonly [immerable] = true

  constructor(
    readonly chapter: ChapterDisplay,
    readonly wordService: WordService,
    readonly textService: TextService,
    readonly rowsContext: RowsContextService,
    readonly translationContext: TranslationContextService
  ) {}
}

function getWordDoc(
  this: CorpusWordExportContext,
  jQueryRef: JQuery
): Promise<Document> {
  return new Promise((resolve) => {
    resolve(
      wordExport(
        this.chapter,
        {
          wordService: this.wordService,
          textService: this.textService,
          rowsContext: this.rowsContext,
          translationContext: this.translationContext,
        },
        jQueryRef
      )
    )
  })
}