Update media-service.js
This commit is contained in:
@@ -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();
|
const chunks = []
|
||||||
let 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) => {
|
||||||
|
const url = req._.req.path
|
||||||
|
if (url.includes('content')) {
|
||||||
|
const id = req.data.id
|
||||||
|
const mediaObj = mediaDB.get(id)
|
||||||
|
if (!mediaObj) {
|
||||||
|
req.reject(404, 'Media not found for the ID')
|
||||||
|
return
|
||||||
|
}
|
||||||
|
const decodedMedia = new Buffer(
|
||||||
|
mediaObj.media.split(';base64,').pop(),
|
||||||
|
'base64'
|
||||||
|
)
|
||||||
|
return _formatResult(decodedMedia)
|
||||||
|
} else return next() //> delegate to next/default handlers
|
||||||
|
})
|
||||||
|
|
||||||
srv.on("READ", "Media", (req, next) => {
|
srv.on('DELETE', 'Media', (req, next) => {
|
||||||
try {
|
const id = req.data.id
|
||||||
const url = req._.req.path;
|
mediaDB
|
||||||
if (url.indexOf("content") > 0) {
|
.chain()
|
||||||
const id = req.data.id;
|
.find({ $loki: id })
|
||||||
if (mediaDB.get(id) == null) {
|
.remove()
|
||||||
req.reject(404, "Media not found for the ID");
|
return next() //> delegate to next/default handlers
|
||||||
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
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
srv.on("DELETE", "Media", (req, next) => {
|
|
||||||
try {
|
|
||||||
const id = req.data.id;
|
|
||||||
const obj = mediaDB.get(id);
|
|
||||||
if (obj != null) {
|
|
||||||
mediaDB.remove(obj)
|
|
||||||
}
|
|
||||||
return next() //> delegate to next/default handlers
|
|
||||||
} catch (error) {
|
|
||||||
req.reject(404, "Media DB error ")
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
function _formatResult(decodedMedia) {
|
|
||||||
const readable = new Readable
|
|
||||||
const result = new Array()
|
|
||||||
readable.push(decodedMedia)
|
|
||||||
readable.push(null)
|
|
||||||
result.push({ value: readable })
|
|
||||||
return result
|
|
||||||
}
|
|
||||||
|
|
||||||
|
function _formatResult (decodedMedia) {
|
||||||
|
const readable = new Readable()
|
||||||
|
const result = new Array()
|
||||||
|
readable.push(decodedMedia)
|
||||||
|
readable.push(null)
|
||||||
|
result.push({ value: readable })
|
||||||
|
return result
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user