/* istanbul ignore file */
'use strict'

const appRoot = require( 'app-root-path' )
const crs = require( 'crypto-random-string' )
const knex = require( `${appRoot}/db/knex` )
const S3 = require( 'aws-sdk/clients/s3' )
const stream = require( 'stream' )

const FileUploads = ( ) => knex( 'file_uploads' )

function getS3Config()
{
  return {
    accessKeyId: process.env.S3_ACCESS_KEY || 'minioadmin',
    secretAccessKey: process.env.S3_SECRET_KEY || 'minioadmin',
    endpoint: process.env.S3_ENDPOINT || 'http://127.0.0.1:9000' ,
    s3ForcePathStyle: true,
    signatureVersion: 'v4'
  }
}


module.exports = {

  async checkBucket() {
    const s3 = new S3( getS3Config() )
    let Bucket = process.env.S3_BUCKET
    
    try {
      let data = await s3.headBucket( { Bucket } ).promise()
      return
    } catch ( err ) {
      if ( err.statusCode === 403 ) {
        throw new Error( 'Access denied to S3 bucket ' )
      }
      if ( process.env.S3_CREATE_BUCKET === 'true' ) {
        await s3.createBucket( { Bucket } ).promise()
      } else {
        throw new Error( `Can't open S3 bucket '${Bucket}': ${err.toString()}` )
      }
    }
  },

  async getFileInfo( { fileId } ) {
    let fileInfo = await FileUploads( ).where( { id: fileId } ).select( '*' ).first( )
    return fileInfo
  },

  async getStreamFileUploads( { streamId } ) {
    let fileInfos = await FileUploads().where( { streamId: streamId } ).select( '*' ).orderBy( [ { column: 'uploadDate', order: 'desc' } ] )
    return fileInfos
  },

  async getFileStream( { fileId } ) {
    const s3 = new S3( getS3Config() )
    let Bucket = process.env.S3_BUCKET
    let Key = `files/${fileId}`
    
    let fileStream = s3.getObject( { Key, Bucket } ).createReadStream()
    return fileStream
  },

  async uploadFile( { streamId, branchName, userId, fileName, fileType, fileStream } ) {
    // Create ID and db entry
    let fileId = crs( { length: 10 } )
    let dbFile = {
      id: fileId,
      streamId,
      branchName,
      userId,
      fileName,
      fileType,
    }
    await FileUploads( ).insert( dbFile )
    
    // Upload stream
    const s3 = new S3( getS3Config() )
    let Bucket = process.env.S3_BUCKET
    // TODO: error if missing
    let Key = `files/${fileId}`
    
    let uploadResponse = await s3.upload( { Bucket, Key, Body: fileStream } ).promise()

    // Get file size and update db entry
    let headResponse = await s3.headObject( { Key, Bucket } ).promise()
    let fileSize = headResponse.ContentLength

    await FileUploads().where( { id: fileId } ).update( { uploadComplete: true, fileSize } )

    return fileId
  }
}