instrunet_backend/main.js

import {Worker} from 'worker_threads'
import Minimist from 'minimist'

let argv = Minimist(process.argv.slice(2));
import fs from 'fs';
import express from 'express';
import ffmpeg from 'fluent-ffmpeg';

const app = express();
import mysql from 'mysql2';
import SendSeekable from 'send-seekable'

const pool = mysql.createPool({
    keepAliveInitialDelay: 0, enableKeepAlive: true,
})
import 'cnchar'
import 'sort-array'

const Kind = {
    0: "去和声伴奏",
    1: "和声伴奏",
    2: "人声",
    3: "贝斯",
    4: "鼓",
    5: "其他",
}

const db = mysql.createConnection({
    host: 'mc.andyxie.cn',
    user: 'instrunet',
    password: 'Moyingren2015',
    database: "instrunet_data",
    pool: pool,
    enableKeepAlive: true,
    keepAliveInitialDelay: 0,
})
import nodemailer from 'nodemailer'
import Queue from 'js-queue'


import webp from 'webp-converter'

import sortArray from "sort-array";

import pkg from 'opencc'
const {OpenCC} = pkg

import https from "node:https";

import nrc from "node-run-cmd";

import bodyParser from "body-parser";

webp.grant_permission()
let queue = new Queue();
queue.autoRun = false;
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',
    }
})


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


let currentTask = [];

async function Submit(req) {

    let uuid = crypto.randomUUID()


    db.connect(function (err) {
        if (err) {
            console.log(err)
        }
    })


    let albumCover = null;
    try {
        let resource = await fetch(req.body.file)
        if (!resource.ok) {
            GC()
            currentTask.shift();
            console.log(queue.contents)
            console.log(currentTask)
            queue.next()
            return
        }
        let ab = await resource.arrayBuffer()
        fs.writeFileSync("./" + uuid, Buffer.from(ab))


        if (URL.canParse(req.body.albumCover)) {
            console.log("can parse ")
            fetch(req.body.albumCover).then(res => {
                res.arrayBuffer().then(async r => {
                    if (r.byteLength === 0) {
                        GC()
                        currentTask.shift();
                        console.log(queue.contents)
                        console.log(currentTask)
                        queue.next()
                        return
                    }
                    if (!fs.existsSync("tmp")) {
                        fs.mkdirSync("tmp")
                    }
                    fs.writeFileSync("tmp/" + uuid, new Buffer.from(r));
                    await webp.cwebp("tmp/" + uuid, "tmp/" + uuid + ".webp", "-q 50 -mt ", "-v")
                    try {
                        albumCover = fs.readFileSync("tmp/" + uuid + ".webp")
                    } catch (e) {
                        console.log(e)
                        fs.rmSync("tmp/" + uuid);
                        albumCover = null;
                    }
                })
            })
        }


    } catch (err) {
        GC()
        currentTask.shift();
        console.log(queue.contents)
        console.log(currentTask)
        queue.next()
        return
    }

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

    switch (req.body.kind) {
        case 0:
            kind_of[0] = `audio-separator ./${uuid} --model_filename UVR-MDX-NET-Inst_HQ_4.onnx  --mdx_enable_denoise  --mdx_segment_size 4000 --mdx_overlap 0.85 --mdx_batch_size 300  --output_format mp3 --output_dir output`
            kind_of[1] = `./output/${uuid}_(Instrumental)_UVR-MDX-NET-Inst_HQ_4.mp3`
            kind_of[2] = `${uuid}_(Instrumental)_UVR-MDX-NET-Inst_HQ_4.mp3`
            kind_of[3] = `./output/${uuid}_(Vocals)_UVR-MDX-NET-Inst_HQ_4.mp3`


            break;
        case 1:
            kind_of[0] = `audio-separator ./${uuid} --model_filename UVR_MDXNET_KARA_2.onnx  --mdx_enable_denoise  --mdx_segment_size 4000 --mdx_overlap 0.85 --mdx_batch_size 300  --output_format mp3  --output_dir output`
            kind_of[1] = `./output/${uuid}_(Instrumental)_UVR_MDXNET_KARA_2.mp3`
            kind_of[2] = `${uuid}_(Instrumental)_UVR_MDXNET_KARA_2.mp3`
            kind_of[3] = `./output/${uuid}_(Vocals)_UVR_MDXNET_KARA_2.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 --sample_rate=48000 --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 --sample_rate=48000 --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;


    }

    function GC() {


        try {
            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);

                }


            })
            fs.rmSync("tmp/" + uuid)
            fs.rmSync("tmp/" + uuid + ".webp")
        } catch (err) {

        }
    }

    try {
        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, email) VALUES (?,?,?,?,?,?,?,?,?)"), [uuid, req.body.name, req.body.albumName, req.body.link, fs.readFileSync(kind_of[1]), req.body.artist, req.body.kind, albumCover === null ? new Buffer.from([]) : albumCover, (req.body.email === null || req.body.email === undefined || req.body.email === "") ? "" : req.body.email])
            GC();
            if (req.body.email !== undefined && req.body.email !== "") {
                console.log(req.body.email)
                transporter.sendMail({
                    from: '"xiey0" <xiey0@qq.com>',
                    to: req.body.email,
                    subject: "你的音频已处理完成。",
                    text: "你的音频已处理完成。",
                    html: fs.readFileSync("./Template.html", "utf8").toString().replace("{song_name}", req.body.name).replace("{href_link}", "https://andyxie.cn:4000/player?play=" + uuid).replace("{kind}", Kind[req.body.kind]).replace("{album}", req.body.albumName)
                }).then((result) => {
                    console.log("Message sent: %s", result.messageId)

                })
            }
            currentTask.shift();
            console.log(queue.contents)
            console.log(currentTask)
            queue.next()

        })
    } catch (e) {
        GC()
        console.error(e)
        return
    }


}


app.get('/queue', (req, res) => {
    res.header("Access-Control-Allow-Origin", "*");

    res.end(JSON.stringify(currentTask));
})

app.post('/submit', SubmitWrapper)

async function SubmitWrapper(req, res) {
    // TODO
    //  May extract logic for dupe check in the future.
    if (req.body.file === undefined || req.body.file === null || req.body.file === "" || req.body.kind === undefined || req.body.kind === null || req.body.kind === "") {
        res.status(500).header("Access-Control-Allow-Origin", "*").end("想都别想");

        return
    }
    if (req.body.artist === undefined || req.body.albumName === undefined || req.body.artist === null || req.body.albumName === null || req.body.artist === "" || req.body.albumName === "") {
        req.body.artist = "未知艺术家"
        req.body.albumName = "未知专辑"
    }
    if (req.body.name === undefined || req.body.name === null || req.body.name === "") {
        res.status(500).header("Access-Control-Allow-Origin", "*").end("想都别想");
        return
    }
    if (!URL.canParse(req.body.file)) {
        res.status(500).header("Access-Control-Allow-Origin", "*").end("想都别想");
        return
    }
    db.execute(("SELECT uuid, song_name, album_name, artist, kind FROM instrunet_entry WHERE song_name = ? and artist = ? and kind = ?"), [await converters2t.convertPromise(req.body.name), await converters2t.convertPromise(req.body.artist), req.body.kind], async (err, rowsT) => {
        db.execute("SELECT uuid, song_name, album_name, artist, kind FROM instrunet_entry WHERE song_name = ? and artist = ? and kind = ?", [await convertert2s.convertPromise(req.body.name), await convertert2s.convertPromise(req.body.artist), req.body.kind], (err, rowsS) => {
            let dedupe = [];
            rowsT.forEach(row => {
                dedupe = dedupe.concat(row);
            })
            rowsS.forEach(row => {
                if (JSON.stringify(dedupe).indexOf(JSON.stringify(row)) === -1) {
                    dedupe = dedupe.concat(row);
                }
            })
            if (dedupe.length === 0) {
                // Verify
                if (req.body.file.substring(0, 5) !== "data:") {
                    res.status(500).header("Access-Control-Allow-Origin", "*").end("想都别想");
                    return
                }
                for (const item of currentTask) {
                    if (req.body.name === item.name) {
                        if (req.body.albumName === item.albumName) {
                            if (req.body.kind === item.kind) {

                                res.header("Access-Control-Allow-Origin", "*");
                                res.statusCode = 500
                                res.end("傻逼，重复了。请在盲目上传之前看看库里有没有好么傻逼？")
                                return

                            }
                        }
                    }
                }
                for (let prop of [req.body.name, req.body.albumName, req.body.artist]) {
                    if (prop === undefined || prop === "" || prop === null) {
                        res.status(500).header("Access-Control-Allow-Origin", "*").end("想都别想");
                        return
                    }
                }


                if (req.body.link === undefined) {
                    req.body.link = ""
                }


                queue.add(() => {
                    Submit(req)
                })
                currentTask.push({
                    name: req.body.name,
                    albumName: req.body.albumName,
                    kind: req.body.kind,
                    artist: req.body.artist,
                    email: req.body.email !== undefined ? req.body.email : null,
                })

                console.log(queue.contents)
                console.log(currentTask)
                if (currentTask.length === 1) {
                    queue.next()
                }
                res.header("Access-Control-Allow-Origin", "*");
                res.end("api_success")
            } else {
                res.header("Access-Control-Allow-Origin", "*");
                res.statusCode = 500
                res.end("傻逼，重复了。请在盲目上传之前看看库里有没有好么傻逼？")
            }

        })
    })


}

app.post('/lyric', async (req, res) => {
    let name = req.body.name
    let artist = req.body.artist;
    let album = req.body.albumName;
    let lrc = null;
    await fetchThatShit()

    async function fetchThatShit() {
        try {
            let fetched = (await (await fetch(`http://192.168.1.166:28883/jsonapi?title=${convertert2s.convertSync(name)}&artist=${convertert2s.convertSync(artist)}&album=${convertert2s.convertSync(album)}`)).text())

            lrc = fetched


        } catch (err) {
            await fetchThatShit()
        }
    }


    res.header("Access-Control-Allow-Origin", "*");
    res.end(lrc)
})
app.options('/lyric', async (req, res) => {
    res.header("Access-Control-Allow-Origin", "*");
    res.header("Access-Control-Allow-Headers", "Content-Type");
    res.end()
})


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 ? or album_name like ? or artist like ?", [`%${await converters2t.convertPromise(req.body.searchStr)}%`, `%${await converters2t.convertPromise(req.body.searchStr)}%`, `%${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 ? or album_name like ? or artist like ?", [`%${await convertert2s.convertPromise(req.body.searchStr)}%`, `%${await convertert2s.convertPromise(req.body.searchStr)}%`, `%${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);
                    }
                })
                for (let obj of prepare) {
                    obj.stroke = obj.song_name[0].stroke() // Problematic as shit. Do not use.
                    obj.spell = obj.song_name.spell()
                }
                sortArray(prepare, {
                    by: 'spell'
                })

                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) {
    let uuid = crypto.randomUUID()
    if (req.query.id) {
        if (req.query.albumcover === "true") {
            db.execute("SELECT albumcover FROM instrunet_entry WHERE uuid = ?", [req.query.id], async 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

            db.execute("SELECT song_name, album_name, artist, kind FROM instrunet_entry WHERE uuid = ?", [req.query.id], async 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 (Number.isNaN(Number(req.body.id))) {
        res.status(500).header("Access-Control-Allow-Origin", "*").end("格式错误");
        return
    }

    if (req.body.id !== undefined && req.body.kind !== undefined) {
        try {
            let id = req.body.id
            fetch(ncmAPIUrl + "/song/download/url/v1?id=" + id + "&level=hires", {
                headers: {
                    Cookie: "MUSIC_U=0087F9D8E102A1C1661EBE1792412F3351DA64D1BD3D862BA77E45E9024524725F3A1983345D9B5A4014C725D19C069DD71081F6FE3659F9E1FD412DC427FB809FAF7789AEEA10E9DE6F06C58D1959BA209D2A83C3FA753261036C4CFD0D143B6C7748B8A6D2DD5C2E96E75D1E847E4AAE035CB2C86B175D9AFC6A164C522ED76E24AE654740AB6BAF5B29597F7E3B0158B2EC1C37F2688279871873FA7ADAEF8280A059E84C4BBFB9E4F225F9A2065DF652247D5496587A7B1E3D35DB0CD3F825C06FE5BFE5CFEF1770847099704360504B73C9B396E37CECE4F9DDEE6001588C3C4F5B2861D9ADF339FC47DD480858CA800620785EA032215B63B81025304DB3331F384793FF8EE681247E34C7931176F2F618B66C122F0602F1EA15F963E422DEC79C257F3577A197BECE71E316C751C3B9F5F3CD07BFDC0270A287A1BB6576"
                }
            }).then(async result => {
                if (result.status === 200 || result.status === 304) {
                    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(500).header("Access-Control-Allow-Origin", "*").end("不存在");
                    }


                } else {
                    res.status(404).header("Access-Control-Allow-Origin", "*").send("未找到或出现错误")
                }


            })
        } catch (err) {
            res.status(500).header("Access-Control-Allow-Origin", "*");
            console.log(err)
        }


    } 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', async function (req, res) {

    let uuid = req.params.uuid;
    let pitch = req.query.pitch;

    async function Provider(b) {

        try {

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

                const worker = new Worker('./pitched.js', {
                    workerData: {
                        binary: availCache[uuid].databinary, pitch: pitch
                    }
                })


                worker.on("message", message => {
                    res.contentType("audio/wav");
                    res.header("Access-Control-Allow-Origin", "*");
                    res.header("Access-Control-Allow-Headers", "Content-Type");
                    res.header("Access-Control-Allow-Origin", "*");
                    res.sendSeekable(Buffer.from(message))
                })



            }


        } catch (e) {
            res.contentType("text/plain");
            res.status(500).header("Access-Control-Allow-Origin", "*").end("Err.");
            console.log(e)
            console.log("Triggered err");

        }


    }

    if (availCache[uuid]) {
        await Provider();

    } else {
        db.execute("SELECT song_name, databinary FROM instrunet_entry WHERE uuid = ?", [uuid], (err, rows) => {
            availCache[uuid] = rows[0]
            Provider()
        })

    }


})


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")
}