report_archive_controller.js 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296
  1. 'use strict';
  2. /**
  3. * Created by Tony on 2021/3/31.
  4. */
  5. const path = require('path');
  6. const uuidV1 = require('uuid').v1;
  7. const fs = require('fs');
  8. const MAX_ARCHIVE = 3;
  9. const tenderMenu = require('../../config/menu').tenderMenu;
  10. const measureType = require('../const/tender').measureType;
  11. const fsUtil = require('../public/js/fsUtil');
  12. const auditConst = require('../const/audit');
  13. module.exports = app => {
  14. class ReportArchiveController extends app.BaseController {
  15. async index(ctx) {
  16. const tender = ctx.tender;
  17. const stage = ctx.stage;
  18. let stage_id = -1;
  19. let stage_order = -1;
  20. let stage_times = -1;
  21. let stage_status = -1;
  22. const treeNodes = await ctx.service.rptTreeNode.getNodesByProjectId([-1, tender.data.project_id]);
  23. const custTreeNodes = await ctx.service.rptTreeNodeCust.getCustFoldersByUserId(this.ctx.session.sessionUser.accountId);
  24. const stageList = await ctx.service.stage.getValidStagesShort(tender.id);
  25. //
  26. // 。。。
  27. let archiveList = [];
  28. // console.log('tender.data.project_id: ' + tender.data.project_id);
  29. if (stage) {
  30. // console.log('ctx.stage.id: ' + ctx.stage.id);
  31. const archives = await ctx.service.rptArchive.getPrjStgArchive(tender.data.project_id, ctx.stage.id);
  32. stage_id = stage.id;
  33. stage_order = stage.order;
  34. stage_times = stage.times;
  35. stage_status = stage.status;
  36. if (archives.length > 0) {
  37. archiveList = JSON.parse(archives[0].content);
  38. }
  39. } else {
  40. // console.log('stageList[0].id: ' + stageList[0].id);
  41. const archives = await ctx.service.rptArchive.getPrjStgArchive(tender.data.project_id, stageList[stageList.length - 1].id);
  42. // stage_id = stageList[0].id;
  43. // stage_order = stageList[0].order;
  44. // stage_times = stageList[0].times;
  45. // stage_status = stageList[0].status;
  46. if (archives && archives.length > 0) {
  47. archiveList = JSON.parse(archives[0].content);
  48. }
  49. }
  50. let rpt_tpl_items = '{ customize: [], common: [] }';
  51. if (custTreeNodes.length > 0) {
  52. rpt_tpl_items = custTreeNodes[0].rpt_tpl_items;
  53. }
  54. //
  55. const renderData = {
  56. tender: tender.data,
  57. rpt_tpl_data: JSON.stringify(treeNodes),
  58. cust_tpl_data: rpt_tpl_items,
  59. project_id: tender.data.project_id,
  60. tender_id: tender.id,
  61. stg_id: stage_id,
  62. stg_order: stage_order,
  63. stg_times: stage_times,
  64. stg_status: stage_status,
  65. stage_list: JSON.stringify(stageList),
  66. tenderMenu,
  67. measureType,
  68. jsFiles: this.app.jsFiles.common.concat(this.app.jsFiles.report.main),
  69. stages: stageList,
  70. auditConst: auditConst.stage,
  71. archiveList,
  72. };
  73. await this.layout('report/index_archive.ejs', renderData);
  74. }
  75. async getReportArchive(ctx) {
  76. const params = JSON.parse(ctx.request.body.params);
  77. // ctx.body = await this._getReport(ctx, params);
  78. const archives = await ctx.service.rptArchive.getPrjStgArchive(params.prjId, params.stgId);
  79. let archiveList = [];
  80. if (archives.length > 0) {
  81. archiveList = JSON.parse(archives[0].content);
  82. }
  83. ctx.body = {
  84. data: archiveList,
  85. };
  86. }
  87. async addReportArchive(ctx) {
  88. try {
  89. const stream = await ctx.getFileStream();
  90. const prjId = ctx.params.prjId;
  91. const stgId = ctx.params.stgId;
  92. const rptId = ctx.params.rptId;
  93. const newName = uuidV1();
  94. const fileName = newName + '.PDF';
  95. // console.log('adding fileName: ' + fileName);
  96. // await ctx.helper.saveStreamFile(stream, path.join(this.app.baseDir, 'app', 'public/archive', fileName));
  97. const oss_result = await ctx.oss.put('archive/' + fileName, stream);
  98. if (!(oss_result && oss_result.url && oss_result.res.status === 200)) {
  99. throw '上传文件失败';
  100. }
  101. // const url = await ctx.oss.delete('archive/52d3e7f0-c7fb-11eb-b8c2-51b890b95d23.PDF');
  102. // console.log(url);
  103. // const flag = true;
  104. // if (flag) {
  105. // throw 'ok';
  106. // }
  107. const updateDate = new Date();
  108. const montStr = (updateDate.getMonth() + 1) < 10 ? ('0' + (updateDate.getMonth() + 1)) : ((updateDate.getMonth() + 1));
  109. const dateStr = (updateDate.getDate()) < 10 ? ('0' + (updateDate.getDate())) : ((updateDate.getDate()));
  110. const dtStr = updateDate.getFullYear() + '-' + montStr + '-' + dateStr;
  111. const orgArchiveList = await ctx.service.rptArchive.getPrjStgArchive(prjId, stgId);
  112. if (orgArchiveList.length > 0) {
  113. const contentArr = JSON.parse(orgArchiveList[0].content);
  114. let hasArchive = false;
  115. for (const item of contentArr) {
  116. if (item.rpt_id === rptId) {
  117. hasArchive = true;
  118. if (item.items.length >= MAX_ARCHIVE) {
  119. // 超出界限,需要删除时间最旧的那个
  120. let rmIdx = 0;
  121. for (let idx = 1; idx < item.items.length; idx++) {
  122. if (item.items[rmIdx].updateDate_time < item.items[idx].updateDate_time) {
  123. rmIdx = idx;
  124. }
  125. }
  126. // 同时删除oss文件
  127. await ctx.oss.delete('archive/' + item.items[rmIdx].uuid + '.PDF');
  128. item.items.splice(rmIdx, 1);
  129. }
  130. const newItem = { uuid: newName, updateDate_time: dtStr };
  131. item.items.push(newItem);
  132. break;
  133. }
  134. }
  135. if (!hasArchive) {
  136. // 表示有新的模板需要添加
  137. contentArr.push({ rpt_id: rptId, items: [{ uuid: newName, updateDate_time: dtStr }] });
  138. }
  139. const updatedRst = await ctx.service.rptArchive.updateArchive(orgArchiveList[0].id, prjId, stgId, contentArr);
  140. // console.log(updatedRst);
  141. ctx.body = { err: 0, msg: '', data: { fileName, updateDate, addedRst: contentArr } };
  142. } else {
  143. // 需要增加
  144. const archiveArr = [];
  145. archiveArr.push({ rpt_id: rptId, items: [{ uuid: newName, updateDate_time: dtStr }] });
  146. const addedRst = await ctx.service.rptArchive.createArchive(prjId, stgId, archiveArr);
  147. // console.log(addedRst);
  148. ctx.body = { err: 0, msg: '', data: { fileName, updateDate, addedRst: archiveArr } };
  149. }
  150. } catch (err) {
  151. this.log(err);
  152. ctx.body = { err: 1, msg: err.toString(), data: null };
  153. }
  154. }
  155. async updateReportArchive(ctx) {
  156. try {
  157. const stream = await ctx.getFileStream();
  158. const prjId = ctx.params.prjId;
  159. const stgId = ctx.params.stgId;
  160. const rptId = ctx.params.rptId;
  161. const orgName = ctx.params.orgName;
  162. const fileName = orgName + '.PDF';
  163. console.log('updating fileName: ' + fileName);
  164. // await ctx.helper.saveStreamFile(stream, path.join(this.app.baseDir, 'app', 'public/archive', fileName));
  165. const oss_result = await ctx.oss.put('archive/' + fileName, stream);
  166. if (!(oss_result && oss_result.url && oss_result.res.status === 200)) {
  167. throw '上传文件失败';
  168. }
  169. const updateDate = new Date();
  170. const montStr = (updateDate.getMonth() + 1) < 10 ? ('0' + (updateDate.getMonth() + 1)) : ((updateDate.getMonth() + 1));
  171. const dateStr = (updateDate.getDate()) < 10 ? ('0' + (updateDate.getDate())) : ((updateDate.getDate()));
  172. const dtStr = updateDate.getFullYear() + '-' + montStr + '-' + dateStr;
  173. const orgArchiveList = await ctx.service.rptArchive.getPrjStgArchive(prjId, stgId);
  174. if (orgArchiveList.length > 0) {
  175. const contentArr = JSON.parse(orgArchiveList[0].content);
  176. for (const item of contentArr) {
  177. if (item.rpt_id === rptId) {
  178. if (item.items && item.items.length > 0) {
  179. for (const subItem of item.items) {
  180. if (subItem.uuid === orgName) {
  181. subItem.updateDate_time = dtStr;
  182. break;
  183. }
  184. }
  185. } else {
  186. item.items = [{ uuid: orgName, updateDate_time: dtStr }];
  187. }
  188. break;
  189. }
  190. }
  191. const updatedRst = await ctx.service.rptArchive.updateArchive(orgArchiveList[0].id, prjId, stgId, contentArr);
  192. // console.log(updatedRst);
  193. ctx.body = { err: 0, msg: '', data: { fileName, updateDate, updatedRst: contentArr } };
  194. } else {
  195. // 需要增加
  196. const archiveArr = [];
  197. archiveArr.push({ rpt_id: rptId, items: [{ uuid: orgName, updateDate_time: dtStr }] });
  198. const updatedRst = await ctx.service.rptArchive.createArchive(prjId, stgId, archiveArr);
  199. // console.log(updatedRst);
  200. ctx.body = { err: 0, msg: '', data: { fileName, updateDate, updatedRst: archiveArr } };
  201. }
  202. } catch (err) {
  203. this.log(err);
  204. ctx.body = { err: 1, msg: err.toString(), data: null };
  205. }
  206. }
  207. async removeReportArchive(ctx) {
  208. try {
  209. const prjId = ctx.params.prjId;
  210. const stgId = ctx.params.stgId;
  211. const rptId = ctx.params.rptId;
  212. const orgName = ctx.params.orgName;
  213. const fileName = orgName + '.PDF';
  214. console.log('removing fileName: ' + fileName);
  215. const fullName = path.join(this.app.baseDir, 'app', 'public/archive', fileName);
  216. // await ctx.helper.saveStreamFile(stream, path.join(this.app.baseDir, 'app', 'public/archive', fileName));
  217. // fs.stat(fullName, function(err, data) {
  218. // if (err) {
  219. // console.log(err);
  220. // } else {
  221. // fs.unlink(fullName, function(err) {
  222. // if (err) {
  223. // console.log(err);
  224. // }
  225. // });
  226. // }
  227. // });
  228. const oss_result = await ctx.oss.delete('archive/' + fileName);
  229. if (!(oss_result && oss_result.res.status === 204)) {
  230. throw '删除归档文件失败';
  231. }
  232. const orgArchiveList = await ctx.service.rptArchive.getPrjStgArchive(prjId, stgId);
  233. if (orgArchiveList.length > 0) {
  234. const contentArr = JSON.parse(orgArchiveList[0].content);
  235. for (const item of contentArr) {
  236. if (item.rpt_id === rptId) {
  237. if (item.items && item.items.length > 0) {
  238. for (const subIdx in item.items) {
  239. if (item.items[subIdx].uuid === orgName) {
  240. item.items.splice(subIdx, 1);
  241. break;
  242. }
  243. }
  244. }
  245. break;
  246. }
  247. }
  248. const updatedRst = await ctx.service.rptArchive.updateArchive(prjId, stgId, contentArr);
  249. ctx.body = { err: 0, msg: '', data: { fileName, updatedRst } };
  250. } else {
  251. ctx.body = { err: 0, msg: '', data: { fileName, updatedRst: null } };
  252. }
  253. } catch (err) {
  254. this.log(err);
  255. ctx.body = { err: 1, msg: err.toString(), data: null };
  256. }
  257. }
  258. async addMultiReportArchive(ctx, params) {
  259. // 暂时不支持
  260. }
  261. async getArchivedFileByUUID(ctx) {
  262. // console.log('downloading : ' + ctx.params.uuid);
  263. const uuid = ctx.params.uuid;
  264. const rptName = ctx.params.rptName;
  265. const suffix = '.PDF';
  266. try {
  267. const rptNameURI = encodeURI(rptName);
  268. const filePath = this.app.baseDir + '/app/public/archive/';
  269. // console.log('filePath: ' + filePath);
  270. // await this.ctx.helper.recursiveMkdirSync(this.app.baseDir + '/app/public/download');
  271. ctx.set({
  272. 'Content-Type': 'application/vnd.openxmlformats',
  273. 'Content-Disposition': 'attachment; filename="' + rptNameURI + suffix + "\"; filename*=utf-8''" + rptNameURI + suffix,
  274. });
  275. ctx.body = await fs.readFileSync(filePath + uuid + suffix);
  276. } catch (e) {
  277. console.log(e);
  278. }
  279. }
  280. }
  281. return ReportArchiveController;
  282. };