main.js 4.3 KB

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