import { GLOBAL_KEY_SYMBOL, RETRY_KEY_SYMBOL } from '@sigi/core'
import serialize from 'serialize-javascript'

const ScriptId = 'sigi-persisted-data'

export class StateToPersist<T> {
  constructor(
    private readonly dataToPersist: T,
    private readonly actionsToRetry: { [index: string]: string[] },
  ) {}

  extractToScriptString(withScriptTag = true) {
    if (this.dataToPersist == null) {
      return ''
    }
    const contentScript = this.serialize()
    return withScriptTag ? `<script id="${ScriptId}">${contentScript}</script>` : contentScript
  }

  renderToJSX() {
    if (this.dataToPersist == null) {
      return null
    }
    return <script id={ScriptId} dangerouslySetInnerHTML={{ __html: this.serialize() }} />
  }

  renderToDocument(doc: string) {
    if (this.dataToPersist == null) {
      return doc
    }
    const endBodyPosition = doc.indexOf('</body>')
    if (endBodyPosition === -1) {
      return doc
    }
    return doc.substr(0, endBodyPosition) + this.extractToScriptString() + doc.substring(endBodyPosition)
  }

  extractToJSONScriptString() {
    if (this.dataToPersist == null) {
      return ''
    }
    const stateContent = serialize(this.dataToPersist, { isJSON: true })
    return `<script id="${GLOBAL_KEY_SYMBOL}" type="application/json">${stateContent}</script><script id="${RETRY_KEY_SYMBOL}" type="application/json">${JSON.stringify(
      this.actionsToRetry,
    )}</script>`
  }

  private serialize() {
    const content = serialize(this.dataToPersist, { isJSON: true })
    return `window['${GLOBAL_KEY_SYMBOL}']=${content};window['${RETRY_KEY_SYMBOL}']=${JSON.stringify(
      this.actionsToRetry,
    )}`
  }
}