feat: Adding load-attachments endpoints

This commit is contained in:
2023-10-07 00:03:53 -06:00
parent e538e229f3
commit 964a94eec8
7 changed files with 88 additions and 7 deletions

View File

@@ -1,6 +1,19 @@
"use strict";
const { ROOT_PATH, LIB_PATH, MODELS_PATH, HANDLERS_PATH } = process.env;
const { ROOT_PATH, LIB_PATH, MODELS_PATH, API_CONFIG } = process.env;
const { getPagination , getPage } = require( `${ROOT_PATH}/${LIB_PATH}/Misc.js` );
const apiConfig = require( `${ROOT_PATH}/${API_CONFIG}` );
const { S3Client, PutObjectCommand } = require('@aws-sdk/client-s3');
const s3Client = new S3Client({
region : apiConfig.S3.region,
credentials : {
accessKeyId : apiConfig.S3.accessKeyId,
secretAccessKey : apiConfig.S3.secretAccessKey
}
});
const s3Bucket = apiConfig.S3.bucket;
const s3BucketKey = apiConfig.S3.key;
const Model = require( `${ROOT_PATH}/${MODELS_PATH}/load-attachments.model.js` );
const UserModel = require( `${ROOT_PATH}/${MODELS_PATH}/users.model.js` );
const LoadsModel = require( `${ROOT_PATH}/${MODELS_PATH}/loads.model.js` );
@@ -17,7 +30,7 @@ async function getAuthorizationFilter( userId ){
}
const getAttachment = async(req, res) => {
const attachmentId = req.query.id;
const attachmentId = req.params.id;
const CompanyAccessFilter = await getAuthorizationFilter( req.JWT.payload.sub );
const filter = {
$and : [
@@ -37,8 +50,9 @@ const getAttachmentList = async(req, res) => {
};
const getLoadAttachmentList = async(req, res) => {
const loadId = req.query.id;
const loadId = req.params.id;
const CompanyAccessFilter = await getAuthorizationFilter( req.JWT.payload.sub );
console.log( loadId );
const filter = {
$and : [
{ load : loadId },
@@ -62,7 +76,7 @@ async function getLoadById( loadId , companyId ){
}
]
};
return await Model.findOne( filter ) || null;
return await LoadsModel.findOne( filter ) || null;
}
async function createLoadAttachment( type , userId , loadId ){
@@ -83,7 +97,7 @@ async function createLoadAttachment( type , userId , loadId ){
await attachment.save();
}
else if( load && prevAttachment ){
prevAttachment.updatedAt = new Date.now();
prevAttachment.updatedAt = Date.now();
await prevAttachment.save();
attachment = prevAttachment;
}else{
@@ -95,10 +109,22 @@ async function createLoadAttachment( type , userId , loadId ){
return attachment;
}
async function uploadFile( bucket, key, file , obj_id ){
const params = {
Bucket: bucket,
Key : `${key}/${obj_id}`,
ContentType : file.mimetype,
Body : file.data
};
const s3resp = await s3Client.send( new PutObjectCommand( params ) );
return s3resp;
}
const postLoadingAttachment = async(req, res) => {
const loadId = req.params.id;
const attachment = await createLoadAttachment( "Loading", req.JWT.payload.sub , loadId );
if( attachment ){
const file = req.files.attachment;
if( attachment && file ){
const s3resp = await uploadFile( s3Bucket, s3BucketKey, file , attachment._id );
res.send( attachment );
}else{
res.status(401).send({error:"Unauthorized",code:401});
@@ -108,7 +134,9 @@ const postLoadingAttachment = async(req, res) => {
const postDownloadingAttachment = async(req, res) => {
const loadId = req.params.id;
const attachment = await createLoadAttachment( "Downloading", req.JWT.payload.sub , loadId );
const file = req.files.attachment;
if( attachment ){
const s3resp = await uploadFile( s3Bucket, s3BucketKey, file , attachment._id );
res.send( attachment );
}else{
res.status(401).send({error:"Unauthorized",code:401});