This commit is contained in:
Luke Berndt
2023-02-11 10:15:23 -05:00
parent 0c1c00c79a
commit b226b685c7
6 changed files with 226 additions and 53 deletions

View File

@@ -1,9 +1,32 @@
var ObjectID = require('mongodb').ObjectID;
const https = require('https');
var db = require('../db');
var mongoose = require("mongoose");
var Event = require("../models/event");
const AdmZip = require("adm-zip");
var { callModel: Call, callSchema } = require("../models/call");
var frontend_server = process.env['REACT_APP_FRONTEND_SERVER'] != null ? process.env['REACT_APP_FRONTEND_SERVER'] : 'https://openmhz.com';
const os = require('os');
var fs = require('fs');
const s3 = require('aws-sdk');
const { events } = require('../models/event');
var s3_endpoint = process.env['S3_ENDPOINT'] != null ? process.env['S3_ENDPOINT'] : 's3.us-west-1.wasabisys.com';
var s3_bucket = process.env['S3_BUCKET'] != null ? process.env['S3_BUCKET'] : 'openmhz-west';
var s3_profile = process.env['S3_PROFILE'] != null ? process.env['S3_PROFILE'] : 'wasabi-account';
const s3Endpoint = new s3.Endpoint(s3_endpoint);
var s3Credentials = new s3.SharedIniFileCredentials({
profile: s3_profile
});
s3.config.credentials = s3Credentials;
const s3S3 = new s3.S3({
endpoint: s3Endpoint,
maxRetries: 2
});
exports.getEvent = async function (req, res, next) {
@@ -23,7 +46,7 @@ exports.getEvent = async function (req, res, next) {
exports.getEvents = async function (req, res, next) {
try {
let events = await Event.find({},["title", "description", "startTime", "endTime", "expireTime", "numCalls", "createdAt", "shortNames"]).exec();
let events = await Event.find({}, ["title", "description", "startTime", "endTime", "expireTime", "numCalls", "createdAt", "shortNames"]).exec();
res.contentType('json');
res.send(JSON.stringify(events));
}
@@ -34,6 +57,124 @@ exports.getEvents = async function (req, res, next) {
}
}
async function downloadFile(folderName, url) {
return new Promise((resolve, reject) => {
https.get(url, (response) => {
const filename = new URL(url).pathname.split('/').pop();
const filePath = `${folderName}/${filename}`;
console.log("Downloading to filename " + filename + " and filePath " + filePath);
let stream = fs.createWriteStream(filePath);
response.on("error", reject);
stream.on("finish", () => {
stream.close();
console.log("Download Completed");
resolve();
}).on("error", reject);
response.pipe(stream);
}).on("error", reject);
});
}
const exportEventJson = (eventFolder, event) => {
const filename = eventFolder + "/event.json";
fs.writeFileSync(filename, JSON.stringify(event));
}
const createEventZip = async (folder, filename, zipFolder) => {
try {
const zip = new AdmZip();
zip.addLocalFolder(folder, zipFolder);
zip.writeZip(filename);
console.log(`Created ${filename} successfully for event from ${folder}`);
} catch (e) {
console.error("Event Error trying to zip: " + filename + " error: " + e);
}
}
const uploadEventZip = (zipFile, s3File) => {
// If not a pro plan, use S3 storage
var s3Src = fs.createReadStream(zipFile);
return new Promise((resolve, reject) => {
const s3Params = {
Bucket: s3_bucket,
Key: s3File,
Body: s3Src,
ACL: 'public-read'
};
s3S3.upload(s3Params, function (err, data) {
s3Src.destroy();
if (err) {
console.error("Event S3 Upload Error " + err);
reject(err);
} else {
resolve("done");
}
});
});
}
const cleanupEvent = (folder, filename) => {
fs.rmSync(folder, { recursive: true, force: true });
fs.rmSync(filename);
}
const packageEvent = (eventId) => {
const event = "tst"
const tmpdir = os.tmpdir();
return new Promise(async (resolve, reject) => {
var objectId = eventId;
const event = await Event.findById(objectId).exec();
if (!event) {
reject("Unable to load Event " + eventId);
}
let eventFolder = "event_" + Date.now();
event.shortNames.forEach((shortName) => {
eventFolder = eventFolder + "_" + shortName;
});
const tmpEventFolder = tmpdir + "/" + eventFolder
const zipFile = tmpdir + "/" + eventFolder + ".zip"
const s3File = "events/" + eventFolder + ".zip"
console.log("Using tmp " + tmpEventFolder);
try {
if (!fs.existsSync(tmpEventFolder)) {
fs.mkdirSync(tmpEventFolder);
}
for (const call of event.calls) {
await downloadFile(tmpEventFolder, call.url)
}
exportEventJson(tmpEventFolder, event);
createEventZip(tmpEventFolder, zipFile, eventFolder);
await uploadEventZip(zipFile, s3File);
var url = 'https://' + s3_endpoint + "/" + s3_bucket + "/" + s3File;
event.downloadUrl = url;
await event.save();
} catch (e) {
console.error("Failed processing event " + e);
reject(e);
}
cleanupEvent(tmpEventFolder, zipFile);
resolve('resolved');
});
};
const compareCalls = (a, b) => {
const aTimestamp = new Date(a.time).getTime()
const bTimestamp = new Date(b.time).getTime()
@@ -59,7 +200,7 @@ exports.addNewEvent = async function (req, res, next) {
callIds.forEach((id) => {
objectIds.push(new ObjectID(id));
})
event.title = title.replace(/[^a-zA-Z0-9 \.\-\_]/g, "");
event.description = description.replace(/[^a-zA-Z0-9 \.\-\_]/g, "");
try {
@@ -80,11 +221,12 @@ exports.addNewEvent = async function (req, res, next) {
});
event.calls = calls;
event.numCalls = calls.length;
await event.save();
const url = "/events/" + event._id;
const savedEvent = await event.save();
console.log("Event ID " + savedEvent._id);
await packageEvent(savedEvent._id);
const url = "/events/" + event._id;
res.contentType('json');
res.send(JSON.stringify({url:url}));
res.send(JSON.stringify({ url: url }));
}
catch (err) {
console.error(err);
@@ -94,4 +236,6 @@ exports.addNewEvent = async function (req, res, next) {
})
};
};