Dimitrie Stefanescu
94 lines
2.6 KiB
JavaScript
94 lines
2.6 KiB
JavaScript
/* 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
|
|
}
|
|
}
|