instrunet_backend/main.js

let argv = require('minimist')(process.argv.slice(2));
const fs = require('fs');
const express = require('express');
const app = express();
const mysql = require('mysql2')
const SendSeekable = require('send-seekable');
const pool = mysql.createPool({
    keepAliveInitialDelay: 0, enableKeepAlive: true,
})
const db = mysql.createConnection({
    host: 'mc.andyxie.cn',
    user: 'instrunet',
    password: 'Moyingren2015',
    database: "instrunet_data",
    pool: pool,
    enableKeepAlive: true,
    keepAliveInitialDelay: 0,
})
const nodemailer = require('nodemailer')

let Queue = require('js-queue');


let queue = new Queue();
queue.autoRun = false;

const bodyParser = require("body-parser");
const nrc = require('node-run-cmd')
const https = require("node:https");
const {OpenCC} = require("opencc");
const converters2t = new OpenCC('s2t.json')
const convertert2s = new OpenCC('t2s.json')
app.use(bodyParser.json({"limit": "200mb"}));
app.use(express.json());
app.use(SendSeekable)


const transporter = nodemailer.createTransport({
    host: 'smtp.qq.com', port: 465, secure: true, auth: {
        user: '3095864740@qq.com', pass: 'caemyuagapsadfff',
    }
})

/// TODO
let ncmAPIUrl = "http://localhost:5999";

let res_o = null;
let req_o = null;
let currentTask = [];
function Submit() {

    let reqLocal = req_o;
    let resLocal = res_o;

        let uuid = crypto.randomUUID()
        db.connect(function (err) {
            if (err) {
                console.log(err)
            }
        })


        let albumCover = null;
        try {
            fetch(reqLocal.body.file).then(res => {
                res.arrayBuffer().then(r => fs.writeFileSync("./" + uuid, Buffer.from(r)));
            })
            fetch(reqLocal.body.albumCover).then(res => {
                res.arrayBuffer().then(r => {
                    albumCover = r
                })
            })



        } catch (err) {

                console.log(err)

            return
        }

        const callback = function (d) {
            console.log(d.toString());
        }
        const errcb = function (d) {
            console.log(d.toString());
        }
        let kind_of = [];

        switch (reqLocal.body.kind) {
            case 0:
                kind_of[0] = `audio-separator ./${uuid} --model_filename UVR-MDX-NET-Inst_HQ_5.onnx --mdx_enable_denoise  --mdx_segment_size 4000 --mdx_overlap 0.75  --output_format mp3 --mdx_batch_size 300 --output_dir output`
                kind_of[1] = `./output/${uuid}_(Instrumental)_UVR-MDX-NET-Inst_HQ_5.mp3`
                kind_of[2] = `${uuid}_(Instrumental)_UVR-MDX-NET-Inst_HQ_5.mp3`
                kind_of[3] = `./output/${uuid}_(Vocals)_UVR-MDX-NET-Inst_HQ_5.mp3`
                break;
            case 1:
                kind_of[0] = `audio-separator ./${uuid} --model_filename UVR_MDXNET_KARA.onnx --mdx_enable_denoise  --mdx_segment_size 4000 --mdx_overlap 0.75 --output_format mp3 --mdx_batch_size 300 --output_dir output`
                kind_of[1] = `./output/${uuid}_(Instrumental)_UVR_MDXNET_KARA.mp3`
                kind_of[2] = `${uuid}_(Instrumental)_UVR_MDXNET_KARA.mp3`
                kind_of[3] = `./output/${uuid}_(Vocals)_UVR_MDXNET_KARA.mp3`
                break;
            case 3:
                kind_of[0] = `audio-separator ./${uuid} --model_filename kuielab_a_bass.onnx --mdx_enable_denoise  --mdx_segment_size 4000 --mdx_overlap 0.75 --output_format mp3 --mdx_batch_size 300 --output_dir output`
                kind_of[1] = `./output/${uuid}_(Bass)_kuielab_a_bass.mp3`
                kind_of[2] = `${uuid}_(Bass)_kuielab_a_bass.mp3`
                kind_of[3] = `./output/${uuid}_(No Bass)_kuielab_a_bass.mp3`
                break;
            case 4:
                kind_of[0] = `audio-separator ./${uuid} --model_filename kuielab_a_drums.onnx --mdx_enable_denoise  --mdx_segment_size 4000 --mdx_overlap 0.75 --output_format mp3 --mdx_batch_size 300 --output_dir output`
                kind_of[1] = `./output/${uuid}_(Drums)_kuielab_a_drums.mp3`
                kind_of[2] = `${uuid}_(Drums)_kuielab_a_drums.mp3`
                kind_of[3] = `./output/${uuid}_(No Drums)_kuielab_a_drums.mp3`
                break;


        }
        nrc.run([kind_of[0]], {
            onData: callback, onError: errcb
        }).then(() => {

            db.execute(("INSERT INTO instrunet_entry (uuid, song_name, album_name, link_to, databinary, artist,kind, albumcover) VALUES (?,?,?,?,?,?,?,?)"), [uuid, reqLocal.body.name, reqLocal.body.albumName, reqLocal.body.link, fs.readFileSync(kind_of[1]), reqLocal.body.artist, reqLocal.body.kind, albumCover])
            db.unprepare(() => {

            })
            fs.rm(kind_of[1], (err) => {
                if (err) {
                    console.log(err);

                }
            })
            fs.rm(kind_of[3], (err) => {
                if (err) {
                    console.log(err);

                }
            })
            fs.rm(uuid, (err) => {
                if (err) {
                    console.log(err);

                }


            })
            if (reqLocal.body.email !== undefined) {
                console.log(reqLocal.body.email)
                transporter.sendMail({
                    from: '"xiey0" <xiey0@qq.com>',
                    to: reqLocal.body.email,
                    subject: "你的音频已处理完成。",
                    text: "你的音频已处理完成。",
                    html: fs.readFileSync("./Template.html", "utf8").toString().replace("{song_name}", reqLocal.body.name).replace("{href_link}", "https://andyxie.cn:4000/player?play=" + uuid)
                }).then((result) => {
                    console.log("Message sent: %s", result.messageId)

                })
            }
            currentTask.shift();
            queue.next()
        })


}


app.post('/submit', SubmitWrapper)

function SubmitWrapper(req, res) {
    req_o = req;
    res_o = res;
    if (req.body.file !== undefined) {
        queue.add(Submit)
        currentTask.push({
            name: req.body.name,
            kind: req.body.kind,
        })
    }
    console.log(queue.contents)
    console.log(currentTask)
    if (currentTask.length === 1) {
        queue.next()
    }
    res.header("Access-Control-Allow-Origin", "*");
    res.end("api_success")


}


app.options('/submit', function (req, res) {
    res.header("Access-Control-Allow-Origin", "*");
    res.header("Access-Control-Allow-Headers", "Content-Type");
    res.end()
})
app.post('/search_api', async function (req, res) {


    db.execute(`SELECT uuid, song_name, album_name, artist, kind
                FROM instrunet_entry
                WHERE song_name like '%${await converters2t.convertPromise(req.body.searchStr)}%'
                   or album_name like '%${await converters2t.convertPromise(req.body.searchStr)}%'
                   or artist like '%${await converters2t.convertPromise(req.body.searchStr)}%'`, async (err, rowsT) => {
        db.execute(`SELECT uuid, song_name, album_name, artist, kind
                    FROM instrunet_entry
                    WHERE song_name like '%${await convertert2s.convertPromise(req.body.searchStr)}%'
                       or album_name like '%${await convertert2s.convertPromise(req.body.searchStr)}%'
                       or artist like '%${await convertert2s.convertPromise(req.body.searchStr)}%'`, (err, rowsS) => {
            try {
                let prepare = [];
                rowsT.forEach(row => {
                    prepare = prepare.concat(row);
                })
                rowsS.forEach(row => {
                    if (JSON.stringify(prepare).indexOf(JSON.stringify(row)) === -1) {
                        prepare = prepare.concat(row);
                    }
                })
                res.header("Access-Control-Allow-Origin", "*");

                res.end(JSON.stringify(prepare));


            } catch (e) {
                console.log(e)
            }
        });
    })


})
app.options('/search_api', function (req, res) {
    res.header("Access-Control-Allow-Origin", "*");
    res.header("Access-Control-Allow-Headers", "Content-Type");
    res.end()
})
app.get("/getSingle", function (req, res) {
    if (req.query.id) {
        db.execute(`SELECT song_name, album_name, artist, kind, albumcover
                    FROM instrunet_entry
                    WHERE uuid = "${req.query.id}"`, function (err, rows) {
            if (err) {
                console.log(err);
            }
            res.contentType("application/json");
            res.header("Access-Control-Allow-Origin", "*");
            res.header("Access-Control-Allow-Headers", "Content-Type");
            res.end(JSON.stringify(rows[0]));
        })

    } else {
        res.contentType("application/json");
        res.header("Access-Control-Allow-Origin", "*");
        res.header("Access-Control-Allow-Headers", "Content-Type");
        res.end("{}")

    }
})

// 163
app.options('/ncm/url', function (req, res) {
    res.header("Access-Control-Allow-Origin", "*");
    res.header("Access-Control-Allow-Headers", "Content-Type");
    res.end()
})
app.post("/ncm/url", function (req, res) {
    if (req.body.id) {
        let id = req.body.id
        fetch(ncmAPIUrl + "/song/download/url/v1?id=" + id + "&level=hires", {
            headers: {
                Cookie: "MUSIC_U=0087F9D8E102A1C1661EBE1792412F3351DA64D1BD3D862BA77E45E9024524725F3A1983345D9B5A4014C725D19C069DD71081F6FE3659F9E1FD412DC427FB809FAF7789AEEA10E9DE6F06C58D1959BA209D2A83C3FA753261036C4CFD0D143B6C7748B8A6D2DD5C2E96E75D1E847E4AAE035CB2C86B175D9AFC6A164C522ED76E24AE654740AB6BAF5B29597F7E3B0158B2EC1C37F2688279871873FA7ADAEF8280A059E84C4BBFB9E4F225F9A2065DF652247D5496587A7B1E3D35DB0CD3F825C06FE5BFE5CFEF1770847099704360504B73C9B396E37CECE4F9DDEE6001588C3C4F5B2861D9ADF339FC47DD480858CA800620785EA032215B63B81025304DB3331F384793FF8EE681247E34C7931176F2F618B66C122F0602F1EA15F963E422DEC79C257F3577A197BECE71E316C751C3B9F5F3CD07BFDC0270A287A1BB6576"
            }
        }).then(async result => {
            let result_json = await result.json()
            if (result_json.data.url !== null) {
                let infos = await (await fetch(ncmAPIUrl + "/song/detail?ids=" + id)).json();
                req.body.file = "data:audio/flac;base64," + Buffer.from(await (await fetch(result_json.data.url)).arrayBuffer()).toString("base64")
                req.body.name = infos.songs[0].name;
                req.body.albumName = infos.songs[0].al.name
                req.body.albumCover = infos.songs[0].al.picUrl
                req.body.link = result_json.data.url
                req.body.artist = infos.songs[0].ar[0].name
                /// Complete


                SubmitWrapper(req, res)


            } else {

                res.status(404).header("Access-Control-Allow-Origin", "*").send("未找到")
            }

        })

    } else {
        res.contentType("application/json");
        res.header("Access-Control-Allow-Origin", "*");
        res.header("Access-Control-Allow-Headers", "Content-Type");
        res.end("{}")
    }
})

app.get("/favicon.ico", function (req, res) {
    res.end("");
})
let availCache = {};
setInterval(() => {
    availCache = {};
}, 1800000)

// Fetch
app.get('/:uuid', function (req, res) {

    let uuid = req.params.uuid;

    function Provider(err, rows) {
        try {
            res.contentType("audio/mp3");
            res.header("Access-Control-Allow-Origin", "*");
            res.header("Access-Control-Allow-Headers", "Content-Type");
            res.header("Access-Control-Allow-Origin", "*");
            res.header('Content-Disposition', `attachment; filename="${encodeURI(rows[0].song_name)}"`);
            /** @type {ArrayBuffer}*/
            availCache[uuid] = rows[0].databinary
            res.sendSeekable(availCache[uuid])


        } catch (e) {
            console.log(e)
            console.log("Triggered err");

        }
    }

    if (availCache[uuid] !== undefined) {
        db.execute(`SELECT song_name
                    FROM instrunet_entry
                    WHERE uuid = '${uuid}'`, (err, rows) => {
            res.contentType("audio/mp3");
            res.header("Access-Control-Allow-Origin", "*");
            res.header("Access-Control-Allow-Headers", "Content-Type");
            res.header("Access-Control-Allow-Origin", "*");
            res.header('Content-Disposition', `attachment; filename="${encodeURI(rows[0].song_name)}"`);
            res.sendSeekable(availCache[uuid])
        })

    } else {
        db.execute(`SELECT song_name, databinary
                    FROM instrunet_entry
                    WHERE uuid = '${uuid}'`, Provider)
        db.unprepare()
    }


})


if (argv.https === "true") {
    https.createServer({
        key: fs.readFileSync('andyxie.cn.key'), cert: fs.readFileSync('andyxie.cn.pem')
    }, app).listen(8080)
    console.log("Listening on port 8080 with TLS")
} else {
    app.listen(8080)
    console.log("Listening on port 8080")
}