import path from 'path'
import formidable from './formidable/index.js'
import fs from 'fs/promises'
import { HttpError } from 'flaska'

export default class IngestRoutes {
  constructor(opts = {}) {
    Object.assign(this, {
    })
  }

  uploadPart(org, ctx) {
    let form = formidable.IncomingForm()
    form.uploadDir = org.uploadDir
    form.maxFileSize = org.maxFileSize || 8 * 1024 * 1024
    form.maxFieldsSize = org.maxFieldsSize || 10 * 1024
    form.maxFields = org.maxFields || 50

    let filename = ctx.params.path.split('/').slice(-1)[0]
    let target = path.join(org.uploadDir, filename)

    return new Promise(function(res, rej) {
      form.parse(ctx.req, function(err, fields, files) {
        if (err) return rej(new HttpError(400, err.message))

        ctx.req.body = fields
        ctx.req.files = files
        ctx.req.file = null

        if (!ctx.req.files) {
          return res()
        }

        let keys = Object.keys(files).filter(key => Boolean(ctx.req.files[key]))

        Promise.all(
          keys.map(key => {
            return fs.rm(target, { force: true })
            .then(() => {
              return fs.rename(ctx.req.files[key].path, target)
            })
            .then(function() {
              ctx.req.files[key].path = target
              ctx.req.files[key].filename = filename
            })
          })
        )
        .then(() => {
          if (keys.length === 1 && keys[0] === 'file') {
            ctx.req.file = ctx.req.files.file
          }
          res()
        }, rej)
      })
    })
  }

  register(server) {
    server.flaska.put('/ingest/::path', [
      this.uploadPart.bind(this, {
        maxFileSize: 100 * 1024 * 1024,
        uploadDir: './upload',
      }),
    ], this.ingest.bind(this))
    server.flaska.post('/ingest/::path', [
      this.uploadPart.bind(this, {
        maxFileSize: 100 * 1024 * 1024,
        uploadDir: './upload',
      }),
    ], this.ingest.bind(this))
  }

  /** POST: /api/ingest */
  async ingest(ctx) {
    ctx.log.info(`Got part ${ctx.req.file.filename} size ${ctx.req.file.size}`)

    ctx.body = {
      success: true,
    }
  }
}