main.js 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116
  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. var key = fs.readFileSync('./andyxie.cn.key')
  12. var cert = fs.readFileSync('./andyxie.cn.pem')
  13. console.log(crypto.randomUUID())
  14. app.use(bodyParser.json({"limit": "200mb"}));
  15. app.use(express.json());
  16. const stmt = db.prepare("INSERT INTO instrument_entry (uuid, song_name, album_name, link_to, binary, artist) VALUES (?,?,?,?,?,?)")
  17. app.post('/submit', function (req, res) {
  18. res.header("Access-Control-Allow-Origin", "*");
  19. try {
  20. fetch(req.body.file).then(res => {
  21. res.arrayBuffer().then(r => fs.writeFileSync("./audio_file", Buffer.from(r)));
  22. })
  23. res.end("api_success");
  24. } catch (err) {
  25. res.status(500).send("Server Error");
  26. return
  27. }
  28. var callback = function (d) {
  29. console.log(d.toString());
  30. }
  31. var errcb = function (d) {
  32. console.log(d.toString());
  33. }
  34. 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 "], {
  35. onData: callback,
  36. onError: errcb
  37. }).then(r => {
  38. 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)
  39. })
  40. })
  41. app.options('/submit', function (req, res) {
  42. res.header("Access-Control-Allow-Origin", "*");
  43. res.header("Access-Control-Allow-Headers", "Content-Type");
  44. res.end()
  45. })
  46. app.post('/search_api', function (req, res) {
  47. function OnFetched(err, rows) {
  48. try {
  49. res.header("Access-Control-Allow-Origin", "*");
  50. console.log("Triggered");
  51. res.end(JSON.stringify(rows));
  52. } catch (e) {
  53. }
  54. }
  55. console.log(req.body.searchStr)
  56. db.all(`SELECT id, uuid, song_name, album_name, artist
  57. FROM instrument_entry
  58. WHERE song_name like '%${req.body.searchStr}%'
  59. or album_name like '%${req.body.searchStr}%'
  60. or artist like '%${req.body.searchStr}%'`, OnFetched)
  61. })
  62. app.post('/fetch', function (req, res) {
  63. function Provider(err, rows) {
  64. try {
  65. res.header("Access-Control-Allow-Origin", "*");
  66. console.log(rows[0].binary)
  67. res.end(JSON.stringify({data: Buffer.from(rows[0].binary).toString('base64')}));
  68. } catch (e) {
  69. console.log(e)
  70. console.log("Triggered err");
  71. }
  72. }
  73. console.log(req.body.uuid)
  74. db.all(`SELECT "binary"
  75. FROM instrument_entry
  76. WHERE uuid = '${req.body.uuid}'`, Provider)
  77. })
  78. app.options('/fetch', function (req, res) {
  79. res.header("Access-Control-Allow-Origin", "*");
  80. res.header("Access-Control-Allow-Headers", "Content-Type");
  81. res.end()
  82. })
  83. app.options('/search_api', function (req, res) {
  84. res.header("Access-Control-Allow-Origin", "*");
  85. res.header("Access-Control-Allow-Headers", "Content-Type");
  86. res.end()
  87. })
  88. https.createServer({
  89. key:key, cert: cert,
  90. }, app).listen(8080)