LjFileHelper.js 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465
  1. var request = require('request');
  2. var fs = require('fs');
  3. var ffmpeg = require('fluent-ffmpeg');
  4. var ljConst = require("../common/LjConst.js");
  5. var path = require('path');
  6. var logHelper = require("./LjLogHelper");
  7. /**
  8. * The helper of file
  9. */
  10. class LjFileHelper
  11. {
  12. /**
  13. * Download file
  14. * @param fileUrl the file url
  15. * @returns {boolean}
  16. */
  17. static downloadFile(fileUrl)
  18. {
  19. //获取文件属性对象
  20. var filePropObj = this.getFileProperty(fileUrl);
  21. if (filePropObj == null)
  22. {
  23. return false;
  24. }
  25. //视频处理
  26. if (filePropObj.fileType == ljConst.VIDEO_TYPE_M3U8)
  27. {
  28. //未加密视频本地存储路径
  29. var localFileDir = filePropObj.folderDir + filePropObj.fileName + ".mp4"
  30. //M3U8视频转换MP4
  31. ffmpeg(fileUrl).format(ljConst.VIDEO_TYPE_MP4)
  32. .on('start', function (err)
  33. {
  34. logHelper.info("Starting down video:" + fileUrl);
  35. })
  36. .on('error', function (err)
  37. {
  38. logHelper.error("Succeeded to down video[" + fileUrl + "]: " + err.message);
  39. })
  40. .on('end', function ()
  41. {
  42. //加密视频存储路径
  43. let localFileDirNew = filePropObj.folderDir + filePropObj.fileName + "_new" + ".mp4"
  44. logHelper.info("Succeeded to down video:" + localFileDirNew);
  45. //加密存储
  46. LjFileHelper.writeEncryptedStream(localFileDir, localFileDirNew);
  47. logHelper.info("Succeeded to encrypt video:" + localFileDirNew);
  48. //删除文件(加密存储完成后删除未加密文件)
  49. fs.unlink(localFileDir, function (err)
  50. {
  51. if (err)
  52. {
  53. //删除失败
  54. logHelper.info("Failed to delete video[" + err.message + "]:" + localFileDir);
  55. throw err;
  56. }
  57. else
  58. {
  59. logHelper.info('Succeeded to delete video:' + localFileDir)
  60. }
  61. })
  62. })
  63. .save(localFileDir);
  64. }//图片处理
  65. else
  66. {
  67. //未加密图片本地存储地址
  68. var localFileDir = filePropObj.folderDir + filePropObj.fileName + "." + filePropObj.fileType;
  69. //读取图片
  70. request(fileUrl).pipe(fs.createWriteStream(localFileDir)).on('close', function ()
  71. {
  72. //加密图片存储地址
  73. var localFileDirNew = filePropObj.folderDir + filePropObj.fileName + "_new." + filePropObj.fileType;
  74. logHelper.info("Succeeded to down image:" + localFileDirNew);
  75. //加密图片
  76. LjFileHelper.writeEncryptedStream(localFileDir, localFileDirNew);
  77. logHelper.info("Succeeded to image image:" + localFileDirNew);
  78. //删除文件(加密存储完成后删除未加密文件)
  79. fs.unlink(localFileDir, function (err)
  80. {
  81. if (err)
  82. {
  83. //删除失败
  84. logHelper.erro("Failed to delete image[" + err.message + "]:" + localFileDir);
  85. throw err;
  86. }
  87. else
  88. {
  89. logHelper.info('Succeeded to delete image:' + localFileDir)
  90. }
  91. })
  92. });
  93. }
  94. }
  95. /**
  96. * Delete file
  97. * @param fileUrl the file url
  98. * @returns {boolean}
  99. */
  100. static delFile(fileUrl)
  101. {
  102. //获取文件属性对象
  103. var filePropObj = this.getFileProperty(fileUrl);
  104. if (filePropObj == null)
  105. {
  106. return false;
  107. }
  108. //视频处理
  109. if (filePropObj.fileType == ljConst.VIDEO_TYPE_M3U8)
  110. {
  111. //加密视频存储路径
  112. let localFileDirNew = filePropObj.folderDir + filePropObj.fileName + "_new" + ".mp4"
  113. logHelper.info("Succeeded to down video:" + localFileDirNew);
  114. //删除加密文件
  115. fs.unlink(localFileDirNew, function (err)
  116. {
  117. if (err)
  118. {
  119. //删除失败
  120. logHelper.info("Failed to delete video[" + err.message + "]:" + localFileDir);
  121. throw err;
  122. }
  123. else
  124. {
  125. logHelper.info('Succeeded to delete video:' + localFileDir)
  126. }
  127. })
  128. }//图片处理
  129. else
  130. {
  131. //加密图片存储地址
  132. var localFileDirNew = filePropObj.folderDir + filePropObj.fileName + "_new." + filePropObj.fileType;
  133. logHelper.info("Succeeded to down image:" + localFileDirNew);
  134. //删除加密文件
  135. fs.unlink(localFileDirNew, function (err)
  136. {
  137. if (err)
  138. {
  139. //删除失败
  140. logHelper.erro("Failed to delete image[" + err.message + "]:" + localFileDir);
  141. throw err;
  142. }
  143. else
  144. {
  145. logHelper.info('Succeeded to delete image:' + localFileDir)
  146. }
  147. })
  148. }
  149. }
  150. /**
  151. * Gets file directory
  152. * @param fileName the file name
  153. * @returns {*|string} the file directory
  154. */
  155. static getFileDir(fileName)
  156. {
  157. var fileType = fileName.split(".")[1]
  158. var newFileDir = "";
  159. if (fileType == "m3u8")
  160. {
  161. fileName = fileName.replace(".m3u8", "_new.mp4")
  162. newFileDir = ljConst.BASE_DIR + fileName.replace(/\*/g, '\\');
  163. }
  164. else
  165. {
  166. fileName = fileName.replace(".", "_new.")
  167. newFileDir = ljConst.BASE_DIR + fileName.replace(/\*/g, '\\');
  168. }
  169. return newFileDir;
  170. }
  171. /**
  172. * Makes directory
  173. * @param dirname the directory name
  174. * @returns {boolean} true or false
  175. */
  176. static mkdirsSync(dirname)
  177. {
  178. if (fs.existsSync(dirname))
  179. {
  180. return true;
  181. }
  182. else
  183. {
  184. if (this.mkdirsSync(path.dirname(dirname)))
  185. {
  186. fs.mkdirSync(dirname);
  187. return true;
  188. }
  189. }
  190. }
  191. /**
  192. * Gets the file property object
  193. * @param fileUrl the file URL
  194. * @returns {null}
  195. */
  196. static getFileProperty(fileUrl)
  197. {
  198. /*--文件夹目录-Begin--*/
  199. var folderDir = ljConst.BASE_DIR;
  200. var splitArr = fileUrl.split('/')
  201. var arrLen = splitArr.length;
  202. for (var i = 3; i < arrLen - 1; i++)
  203. {
  204. folderDir += splitArr[i] + "\\";
  205. }
  206. //创建文件夹目录
  207. this.mkdirsSync(folderDir);
  208. var filePropObj = {};
  209. filePropObj.folderDir = folderDir;
  210. //获取文件名称
  211. filePropObj.fileName = splitArr[arrLen - 1].split(".")[0];
  212. //获取文件类型
  213. filePropObj.fileType = splitArr[arrLen - 1].split(".")[1];
  214. /*--文件夹目录-Begin--*/
  215. return filePropObj;
  216. }
  217. static Encrypted(c)
  218. {
  219. var b = '';
  220. for (var i = 0; i < c.length; i++)
  221. {
  222. var ch = c[i];
  223. if (ch == '0')
  224. {
  225. ch = 'f';
  226. }
  227. else if (ch == 'f')
  228. {
  229. ch = '0';
  230. }
  231. b = b + ch;
  232. //console.log(ch);
  233. }
  234. return b;
  235. }
  236. static Decrypted(c)
  237. {
  238. var b = '';
  239. for (var i = 0; i < c.length; i++)
  240. {
  241. var ch = c[i];
  242. if (ch == 'f')
  243. {
  244. ch = '0';
  245. }
  246. else if (ch == '0')
  247. {
  248. ch = 'f';
  249. }
  250. b = b + ch;
  251. //console.log(ch);
  252. }
  253. return b;
  254. }
  255. /**
  256. * Writes encrypted content, which is read from given source file, into given dest file
  257. * @param fSrc the source file
  258. * @param fDest the dest file
  259. */
  260. static writeEncryptedStream(fSrc, fDest)
  261. {
  262. var rs = fs.createReadStream(fSrc);
  263. var ws = fs.createWriteStream(fDest);
  264. rs.on('data', function (chunk)
  265. {
  266. let b = new Buffer(chunk, 'hex');
  267. //console.log(b);
  268. let c = b.toString('hex');
  269. c = LjFileHelper.encryptedContent(c);
  270. let d = Buffer.from(c, 'hex');
  271. ws.write(d);
  272. });
  273. rs.on('end', function ()
  274. {
  275. ws.end();
  276. logHelper.info('Succeeded in writing into encrypted stream[' + fDest + '].');
  277. });
  278. rs.on('error', function (err)
  279. {
  280. console.log(err.stack);
  281. });
  282. }
  283. /**
  284. * Encrypts the content
  285. * @param c the content to be encrypted
  286. * @return the encrypted content
  287. */
  288. static encryptedContent(c)
  289. {
  290. var b = '';
  291. for (var i = 0; i < c.length; i++)
  292. {
  293. var ch = c[i];
  294. if (ch == '0')
  295. {
  296. ch = 'f';
  297. }
  298. else if (ch == 'f')
  299. {
  300. ch = '0';
  301. }
  302. else if (ch == '1')
  303. {
  304. ch = 'e';
  305. }
  306. else if (ch == 'e')
  307. {
  308. ch = '1';
  309. }
  310. else if (ch == '2')
  311. {
  312. ch = 'd';
  313. }
  314. else if (ch == 'd')
  315. {
  316. ch = '2';
  317. }
  318. else if (ch == '3')
  319. {
  320. ch = 'c';
  321. }
  322. else if (ch == 'c')
  323. {
  324. ch = '3';
  325. }
  326. b = b + ch;
  327. //console.log(ch);
  328. }
  329. return b;
  330. }
  331. /**
  332. * Decrypts the encrypted content
  333. * @param c the encrypted content
  334. * @return the decrypted content
  335. */
  336. static decryptedContent(c)
  337. {
  338. var b = '';
  339. for (var i = 0; i < c.length; i++)
  340. {
  341. var ch = c[i];
  342. if (ch == 'f')
  343. {
  344. ch = '0';
  345. }
  346. else if (ch == '0')
  347. {
  348. ch = 'f';
  349. }
  350. else if (ch == 'e')
  351. {
  352. ch = '1';
  353. }
  354. else if (ch == '1')
  355. {
  356. ch = 'e';
  357. }
  358. else if (ch == 'd')
  359. {
  360. ch = '2';
  361. }
  362. else if (ch == '2')
  363. {
  364. ch = 'd';
  365. }
  366. else if (ch == 'c')
  367. {
  368. ch = '3';
  369. }
  370. else if (ch == '3')
  371. {
  372. ch = 'c';
  373. }
  374. b = b + ch;
  375. //console.log(ch);
  376. }
  377. return b;
  378. }
  379. /**
  380. * Reads decrypted content, which is read from given encrypted file, into HTTP response stream
  381. * @param res the HTTP response stream
  382. * @param fDest the encrypted file
  383. */
  384. static readDecryptedStream(res, fDest, contentType)
  385. {
  386. res.writeHead(200, {'Content-Type': contentType});
  387. var rs = fs.createReadStream(fDest);
  388. rs.on('data', function (chunk)
  389. {
  390. let b = new Buffer(chunk, 'hex');
  391. let c = b.toString('hex');
  392. c = LjFileHelper.decryptedContent(c);
  393. let d = Buffer.from(c, 'hex');
  394. res.write(d);
  395. });
  396. rs.on('end', function ()
  397. {
  398. res.end();
  399. console.log('Succeeded in reading from decrypted stream[' + fDest + '].');
  400. });
  401. rs.on('error', function (err)
  402. {
  403. console.log(err.stack);
  404. });
  405. }
  406. /**
  407. * Is video
  408. * @param dir the directory
  409. * @returns {boolean}
  410. */
  411. isVideo(dir)
  412. {
  413. if (dir == "")
  414. {
  415. return false;
  416. }
  417. var fileType = dir.split(".")[1];
  418. if ("m3u8" == fileType)
  419. {
  420. return true;
  421. }
  422. return false;
  423. }
  424. }
  425. module.exports = LjFileHelper;