Update media-service.js

This commit is contained in:
Lakshmi C Rajeev
2020-01-03 14:07:17 +05:30
committed by GitHub
parent 630bb2b19c
commit cd808c76dd

View File

@@ -1,82 +1,68 @@
module.exports = (srv) => { const loki = require('lokijs')
const loki = require('lokijs'); const db = new loki('DB')
const db = new loki('DB'); const mediaDB = db.addCollection('Media')
const mediaDB = db.addCollection("Media");
const { Readable, PassThrough } = require('stream') const { Readable, PassThrough } = require('stream')
srv.before('CREATE', 'Media', (req) => { module.exports = srv => {
const obj = mediaDB.insert({ media: "" }); srv.before('CREATE', 'Media', req => {
req.data.id = obj.$loki; const obj = mediaDB.insert({ media: '' })
req.data.id = obj.$loki
}) })
srv.on('UPDATE', 'Media', (req, next) => { srv.on('UPDATE', 'Media', (req, next) => {
try{ const url = req._.req.path
const url = req._.req.path; if (url.includes('content')) {
if (url.indexOf("content")>0) { const id = req.data.id
const id = req.data.id; const obj = mediaDB.get(id)
const obj = mediaDB.get(id); if (!obj) {
if(obj==null) { req.reject(404, 'No record found for the ID')
req.reject(404,"No record found for the ID"); return
return;
} }
const stream = new PassThrough(); const stream = new PassThrough()
let chunks = []; const chunks = []
stream.on('data', (chunk) => { stream.on('data', chunk => {
chunks.push(chunk); chunks.push(chunk)
}); })
stream.on('end', () => { stream.on('end', () => {
obj.media = Buffer.concat(chunks).toString('base64'); obj.media = Buffer.concat(chunks).toString('base64')
mediaDB.update(obj); mediaDB.update(obj)
}); })
req.data.content.pipe(stream); req.data.content.pipe(stream)
} else return next() } else return next()
}catch(error){
req.reject(404,"Media DB error")
console.log(error)
return
}
}) })
srv.on("READ", "Media", (req, next) => { srv.on('READ', 'Media', (req, next) => {
try { const url = req._.req.path
const url = req._.req.path; if (url.includes('content')) {
if (url.indexOf("content") > 0) { const id = req.data.id
const id = req.data.id; const mediaObj = mediaDB.get(id)
if (mediaDB.get(id) == null) { if (!mediaObj) {
req.reject(404, "Media not found for the ID"); req.reject(404, 'Media not found for the ID')
return;
}
const decodedMedia = new Buffer(mediaDB.get(id).media.split(';base64,').pop(), 'base64');
return _formatResult(decodedMedia);
}
else return next() //> delegate to next/default handlers
} catch (error) {
req.reject(404, "Media DB error")
return return
} }
const decodedMedia = new Buffer(
mediaObj.media.split(';base64,').pop(),
'base64'
)
return _formatResult(decodedMedia)
} else return next() //> delegate to next/default handlers
}) })
srv.on("DELETE", "Media", (req, next) => { srv.on('DELETE', 'Media', (req, next) => {
try { const id = req.data.id
const id = req.data.id; mediaDB
const obj = mediaDB.get(id); .chain()
if (obj != null) { .find({ $loki: id })
mediaDB.remove(obj) .remove()
}
return next() //> delegate to next/default handlers return next() //> delegate to next/default handlers
} catch (error) {
req.reject(404, "Media DB error ")
}
}) })
function _formatResult (decodedMedia) { function _formatResult (decodedMedia) {
const readable = new Readable const readable = new Readable()
const result = new Array() const result = new Array()
readable.push(decodedMedia) readable.push(decodedMedia)
readable.push(null) readable.push(null)
result.push({ value: readable }) result.push({ value: readable })
return result return result
} }
} }