main.js 2.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192
  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 FROM instrument_entry WHERE song_name like '%${req.body.searchStr}%' or album_name like '%${req.body.searchStr}%' or artist like '%${req.body.searchStr}%'`, OnFetched)
  56. })
  57. app.post('/fetch', function (req, res) {
  58. function Provider(err, rows){
  59. try{
  60. res.header("Access-Control-Allow-Origin", "*");
  61. console.log("Triggered");
  62. res.end(rows[0]);
  63. }catch(e){
  64. }
  65. }
  66. db.all(`SELECT binary FROM instrument_entry WHERE uuid=${req.body.uuid}`, Provider)
  67. })
  68. app.options('/search_api', function (req, res) {
  69. res.header("Access-Control-Allow-Origin", "*");
  70. res.header("Access-Control-Allow-Headers", "Content-Type");
  71. res.end()
  72. })
  73. app.listen(8080);