main.js 3.2 KB

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