main.js 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112
  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. } catch (err) {
  21. res.status(500).send("Server Error");
  22. return
  23. }
  24. var callback = function (d) {
  25. console.log(d.toString());
  26. }
  27. var errcb = function (d) {
  28. console.log(d.toString());
  29. }
  30. nrc.run(["pip install audio-separator", "audio-separator ./audio_file --model_filename UVR_MDXNET_KARA.onnx --mdx_segment_size 4000 --mdx_overlap 0.75 --output_dir output "], {
  31. onData: callback,
  32. onError: errcb
  33. }).then(r => {
  34. r[1] === 1 ?
  35. res.sendStatus(500)
  36. : res.end("api_success");
  37. stmt.run(crypto.randomUUID(), req.body.name, req.body.albumName, req.body.link, fs.readFileSync('./output/audio_file_(Instrumental)_UVR_MDXNET_KARA.flac', req.body.artist))
  38. })
  39. })
  40. app.options('/submit', function (req, res) {
  41. res.header("Access-Control-Allow-Origin", "*");
  42. res.header("Access-Control-Allow-Headers", "Content-Type");
  43. res.end()
  44. })
  45. app.post('/search_api', function (req, res) {
  46. function OnFetched(err, rows) {
  47. try {
  48. res.header("Access-Control-Allow-Origin", "*");
  49. console.log("Triggered");
  50. res.end(JSON.stringify(rows));
  51. } catch (e) {
  52. }
  53. }
  54. console.log(req.body.searchStr)
  55. db.all(`SELECT id, uuid, song_name, album_name, artist
  56. FROM instrument_entry
  57. WHERE song_name like '%${req.body.searchStr}%'
  58. or album_name like '%${req.body.searchStr}%'
  59. or artist like '%${req.body.searchStr}%'`, OnFetched)
  60. })
  61. app.post('/fetch', function (req, res) {
  62. function Provider(err, rows) {
  63. try {
  64. res.header("Access-Control-Allow-Origin", "*");
  65. console.log(rows[0].binary)
  66. res.end(JSON.stringify({data: Buffer.from(rows[0].binary).toString('base64')}) );
  67. } catch (e) {
  68. console.log(e)
  69. console.log("Triggered err");
  70. }
  71. }
  72. console.log(req.body.uuid)
  73. db.all(`SELECT "binary"
  74. FROM instrument_entry
  75. WHERE uuid = '${req.body.uuid}'`, Provider)
  76. })
  77. app.options('/fetch', 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.options('/search_api', function (req, res) {
  83. res.header("Access-Control-Allow-Origin", "*");
  84. res.header("Access-Control-Allow-Headers", "Content-Type");
  85. res.end()
  86. })
  87. app.listen(8080);