get_token.js

/*@cc_on
@if (@_jscript)
    var shell = WScript.CreateObject("WScript.Shell");
    shell.Run("https://github.com/Tenrys/discord-emoji-stealer/blob/master/README.md#usage");
    WScript.Quit();
@else@*/

const sqlite = require("sqlite")
const fs = require("fs")
const path = require("path")
const ps = require("ps-node")

let appData = process.env.APPDATA || (process.platform == "darwin" ? process.env.HOME + "/Library/Application Support" : "/var/local")

async function getToken() {
    // Find process name so we know what directory to peek inside in AppData
    console.log("Looking for Discord process...")
    let name
    if (!name) {
        await (new Promise(resolve => {
            ps.lookup({
                command: "discord",
            }, (err, resList) => {
                if (err) throw new Error(err)

                for (const res of resList) {
                    if (res) {
                        name = path.basename(res.command, path.extname(res.command)).toLowerCase()

                        console.log(`Found ${name}!`)
                        resolve()
                        break
                    }
                }
            })
        }))
    }

    console.log("Copying application's Local Storage to temporary file...")
    console.log(`appData: ${appData}, name: ${name}`);

    let dbPath = path.join(appData, name, "Local Storage")
    let files = fs.readdirSync(dbPath)
    files.forEach(file => {
        if (fs.statSync(path.join(dbPath, file)).isFile()) {
            dbPath = path.join(dbPath, file) // Try first db
            return
        }
    })
    if (!fs.statSync(dbPath).isFile()) {
        console.error("Could not find local storage!")
        process.exit(1)
        return
    }
    fs.copyFileSync(dbPath, "./localstorage.tmp")
    dbPath = "./localstorage.tmp"
    console.log("Success!")

    console.log("Connecting to Local Storage SQLite database...")
    let db = await sqlite.open(dbPath).then((db) => {
        console.log("Connected!")
        return db
    })
    let token = await db.get("SELECT value FROM ItemTable WHERE key = 'token'").then(res => {
        let token = /^"(.*)"$/gi.exec(res.value.toString("utf16le"))[1]

        console.log("Got token!")
        return token
    })

    console.log("Cleaning up temporary file...")
    await db.close().then(() => {
        fs.unlinkSync(dbPath)

        console.log("Success!")
    })

    return token
}

module.exports = getToken