api.js 5.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108
  1. const mime = require("mime-types");
  2. const fs = require('fs');
  3. const Path = require('path');
  4. const Security = require('../src/security.js');
  5. const MediaService = require('../model/mediaService.js');
  6. const MediaFileMetaModel = require('../model/mediaItemMeta.js').MediaFileMetaModel;
  7. const { AccessModel, ACCESS_TYPE, ACCESS_GRANT } = require('../model/access.js');
  8. module.exports = { register: app => {
  9. app.router.get("/api/access/list", (req, res) => {
  10. app.routerUtils.onApiRequest(req, res);
  11. app.routerUtils.jsonResponse(res, req.sessionObj?.accessList || {});
  12. });
  13. app.router.post("/api/access/link", async (req, res) => { // /api/access/link, post: { linkIds: [string] (JSON) }
  14. app.routerUtils.onApiRequest(req, res);
  15. if (!req.post?.linkIds?.length)
  16. return app.routerUtils.httpResponse(res, 400, "Missing argument");
  17. for (let i of JSON.parse(req.post.linkIds)) {
  18. if (await app.databaseHelper.findOne(AccessModel, { type: ACCESS_TYPE.link, typeData: i }))
  19. Security.addLinkToSession(req, i);
  20. }
  21. app.routerUtils.jsonResponse(res, req.sessionObj.accessList);
  22. });
  23. app.router.del("/api/access/:id", (req, res) => {
  24. app.routerUtils.onApiRequest(req, res);
  25. const result = Security.removeFromSession(req, req.params.id);
  26. app.routerUtils.jsonResponse(res, result);
  27. });
  28. app.router.patch("/api/media/:id/meta/:key", async (req, res) => {
  29. app.routerUtils.onApiRequest(req, res);
  30. if (!req.params.id ||!req.params.key || !req.body)
  31. return app.routerUtils.onBadRequest(res);
  32. let data = await MediaService.fetchOne(app, req.params.id, req.sessionObj?.accessList);
  33. if (!data || data.accessType !== ACCESS_GRANT.write)
  34. return app.routerUtils.onPageNotFound(res);
  35. let newMediaItemMedia = new MediaFileMetaModel(data.md5sum, req.params.key, req.body.value, false);
  36. await app.databaseHelper.upsertOne(newMediaItemMedia);
  37. res.end();
  38. });
  39. app.router.get("/api/media/list", async (req, res) => {
  40. app.routerUtils.onApiRequest(req, res);
  41. let first = undefined,
  42. last = undefined;
  43. if (req.body?.chronology !== undefined) {
  44. let range = await MediaService.getMediaRange(app);
  45. first = range[0];
  46. last = range[1];
  47. }
  48. let fromDate = parseInt(req.body?.from);
  49. let count = parseInt(req.body?.count);
  50. app.routerUtils.jsonResponse(res, {
  51. data: await MediaService.fetchMediasWithAccess(
  52. app,
  53. isNaN(fromDate) ? 0 : fromDate,
  54. isNaN(count) ? 25 : Math.min(75, count),
  55. req.sessionObj?.accessList),
  56. first: first,
  57. last: last
  58. });
  59. });
  60. app.router.get("/api/media/:md5sum", async (req, res) => {
  61. app.routerUtils.onApiRequest(req, res);
  62. let data = await MediaService.fetchOne(app, req.params.md5sum, req.sessionObj?.accessList);
  63. if (!data)
  64. return app.routerUtils.onPageNotFound(res);
  65. app.routerUtils.jsonResponse(res, data);
  66. });
  67. app.router.get("/api/media/thumbnail/:md5sum.jpg", async (req, res) => {
  68. app.routerUtils.onApiRequest(req, res);
  69. let data = await MediaService.fetchOne(app, req.params.md5sum, req.sessionObj?.accessList);
  70. if (!data)
  71. return app.routerUtils.onPageNotFound(res);
  72. try {
  73. let thumbnail = null;
  74. try {
  75. thumbnail = await (await app.libraryManager.findMedia(data.path))?.createThumbnail(req.body?.w || 0, req.body?.h || 0, req.body?.q || 6);
  76. } catch (err) {
  77. return app.routerUtils.apiError(res);
  78. }
  79. if (!thumbnail)
  80. return app.routerUtils.onPageNotFound(res);
  81. res.setHeader("Content-Type", "image/jpeg");
  82. res.setHeader("Content-Length", fs.statSync(thumbnail.name)?.size || undefined);
  83. res.setHeader("Cache-Control", "private, max-age=2630000"); // 1 month cache
  84. let rd = fs.createReadStream(thumbnail.name);
  85. rd.once('end', () => thumbnail.removeCallback());
  86. rd.pipe(res);
  87. }
  88. catch (err) {
  89. console.error(err);
  90. app.routerUtils.onPageNotFound(res);
  91. }
  92. });
  93. app.router.get("/api/media/original/:md5sum", async (req, res) => {
  94. app.routerUtils.onApiRequest(req, res);
  95. let data = await MediaService.fetchOne(app, req.params.md5sum, req.sessionObj?.accessList);
  96. if (!data)
  97. return app.routerUtils.onPageNotFound(res);
  98. const fileName = Path.basename(data.path);
  99. res.setHeader("Content-Type", mime.lookup(data.path));
  100. res.setHeader("Content-Length", fs.statSync(data.path)?.size || undefined);
  101. res.setHeader("Content-Disposition", `attachment; filename="${fileName}"`);
  102. res.setHeader("Cache-Control", "private, max-age=2630000"); // 1 month cache
  103. fs.createReadStream(data.path).pipe(res);
  104. });
  105. }};