Cristian Balas
107 lines
2.7 KiB
JavaScript
107 lines
2.7 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 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())
|
|
const Bucket = process.env.S3_BUCKET
|
|
|
|
try {
|
|
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 }) {
|
|
const fileInfo = await FileUploads().where({ id: fileId }).select('*').first()
|
|
return fileInfo
|
|
},
|
|
|
|
async getStreamFileUploads({ streamId }) {
|
|
const fileInfos = await FileUploads()
|
|
.where({ streamId })
|
|
.select('*')
|
|
.orderBy([{ column: 'uploadDate', order: 'desc' }])
|
|
return fileInfos
|
|
},
|
|
|
|
async getFileStream({ fileId }) {
|
|
const s3 = new S3(getS3Config())
|
|
const Bucket = process.env.S3_BUCKET
|
|
const Key = `files/${fileId}`
|
|
|
|
const fileStream = s3.getObject({ Key, Bucket }).createReadStream()
|
|
return fileStream
|
|
},
|
|
|
|
async startUploadFile({
|
|
streamId,
|
|
branchName,
|
|
userId,
|
|
fileName,
|
|
fileType,
|
|
fileStream
|
|
}) {
|
|
// Create ID and db entry
|
|
const fileId = crs({ length: 10 })
|
|
const dbFile = {
|
|
id: fileId,
|
|
streamId,
|
|
branchName,
|
|
userId,
|
|
fileName,
|
|
fileType
|
|
}
|
|
await FileUploads().insert(dbFile)
|
|
|
|
// Upload stream
|
|
const s3 = new S3(getS3Config())
|
|
const Bucket = process.env.S3_BUCKET
|
|
// TODO: error if missing
|
|
const Key = `files/${fileId}`
|
|
|
|
await s3.upload({ Bucket, Key, Body: fileStream }).promise()
|
|
|
|
return fileId
|
|
},
|
|
|
|
async finishUploadFile({ fileId }) {
|
|
const s3 = new S3(getS3Config())
|
|
const Bucket = process.env.S3_BUCKET
|
|
// TODO: error if missing
|
|
const Key = `files/${fileId}`
|
|
|
|
// Get file size and update db entry
|
|
const headResponse = await s3.headObject({ Key, Bucket }).promise()
|
|
const fileSize = headResponse.ContentLength
|
|
|
|
await FileUploads().where({ id: fileId }).update({ uploadComplete: true, fileSize })
|
|
}
|
|
}
|