main.js 4.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143
  1. var express = require('express');
  2. var app = express();
  3. var fs = require("fs");
  4. const sqlite3 = require('sqlite3').verbose();
  5. const db = new sqlite3.Database('./database.db');
  6. const jwt = require("jsonwebtoken");
  7. const bodyParser = require("body-parser");
  8. const {spawn} = require("child_process");
  9. const nrc = require('node-run-cmd')
  10. const https = require("node:https");
  11. const {encode} = require("html-entities");
  12. const querystring = require("node:querystring");
  13. // var key = fs.readFileSync('./andyxie.cn.key')
  14. // var cert = fs.readFileSync('./andyxie.cn.pem')
  15. console.log(crypto.randomUUID())
  16. app.use(bodyParser.json({"limit": "200mb"}));
  17. app.use(express.json());
  18. const stmt = db.prepare("INSERT INTO instrument_entry (uuid, song_name, album_name, link_to, binary, artist) VALUES (?,?,?,?,?,?)")
  19. app.post('/submit', function (req, res) {
  20. res.header("Access-Control-Allow-Origin", "*");
  21. try {
  22. fetch(req.body.file).then(res => {
  23. res.arrayBuffer().then(r => fs.writeFileSync("./audio_file", Buffer.from(r)));
  24. })
  25. res.end("api_success");
  26. } catch (err) {
  27. res.status(500).send("Server Error");
  28. return
  29. }
  30. var callback = function (d) {
  31. console.log(d.toString());
  32. }
  33. var errcb = function (d) {
  34. console.log(d.toString());
  35. }
  36. nrc.run(["pip install audio-separator", "audio-separator ./audio_file --model_filename UVR_MDXNET_KARA.onnx --mdx_segment_size 4000 --mdx_overlap 0.85 --output_format MP3 --mdx_batch_size 20 --output_dir output "], {
  37. onData: callback,
  38. onError: errcb
  39. }).then(r => {
  40. stmt.run(crypto.randomUUID(), req.body.name, req.body.albumName, req.body.link, fs.readFileSync('./output/audio_file_(Instrumental)_UVR_MDXNET_KARA.mp3'), req.body.artist)
  41. })
  42. })
  43. app.options('/submit', function (req, res) {
  44. res.header("Access-Control-Allow-Origin", "*");
  45. res.header("Access-Control-Allow-Headers", "Content-Type");
  46. res.end()
  47. })
  48. app.post('/search_api', function (req, res) {
  49. function OnFetched(err, rows) {
  50. try {
  51. res.header("Access-Control-Allow-Origin", "*");
  52. console.log("Triggered");
  53. res.end(JSON.stringify(rows));
  54. } catch (e) {
  55. }
  56. }
  57. console.log(req.body.searchStr)
  58. db.all(`SELECT id, uuid, song_name, album_name, artist
  59. FROM instrument_entry
  60. WHERE song_name like '%${req.body.searchStr}%'
  61. or album_name like '%${req.body.searchStr}%'
  62. or artist like '%${req.body.searchStr}%'`, OnFetched)
  63. })
  64. app.get('/:uuid', function (req, res) {
  65. async function Provider(err, rows) {
  66. try {
  67. res.header("Access-Control-Allow-Origin", "*");
  68. /** @type {ArrayBuffer}*/
  69. var buffer = rows[0].binary
  70. res.setHeader('Content-Length', buffer.byteLength);
  71. db.all(`SELECT "song_name"
  72. FROM instrument_entry
  73. WHERE uuid = '${req.params.uuid}'`, (err, row) => {
  74. res.setHeader('Content-Disposition', 'attachment; filename=' + encodeURI(row[0].song_name)+'.mp3' );
  75. res.send(buffer)
  76. res.end()
  77. })
  78. } catch (e) {
  79. console.log(e)
  80. console.log("Triggered err");
  81. }
  82. }
  83. db.all(`SELECT "binary"
  84. FROM instrument_entry
  85. WHERE uuid = '${req.params.uuid}'`, Provider)
  86. })
  87. app.post('/fetch', function (req, res) {
  88. function Provider(err, rows) {
  89. try {
  90. res.header("Access-Control-Allow-Origin", "*");
  91. console.log(rows[0].binary)
  92. var str = URL.createObjectURL(new Blob(rows[0].binary))
  93. console.log(str)
  94. res.end(JSON.stringify({data: Buffer.from(rows[0].binary).toString('base64')}));
  95. } catch (e) {
  96. console.log(e)
  97. console.log("Triggered err");
  98. }
  99. }
  100. console.log(req.body.uuid)
  101. db.all(`SELECT "binary"
  102. FROM instrument_entry
  103. WHERE uuid = '${req.body.uuid}'`, Provider)
  104. })
  105. app.options('/fetch', function (req, res) {
  106. res.header("Access-Control-Allow-Origin", "*");
  107. res.header("Access-Control-Allow-Headers", "Content-Type");
  108. res.end()
  109. })
  110. app.options('/search_api', function (req, res) {
  111. res.header("Access-Control-Allow-Origin", "*");
  112. res.header("Access-Control-Allow-Headers", "Content-Type");
  113. res.end()
  114. })
  115. app.listen(8080)