minispin.js

const SSAPI = require('ssapi-node');
const api = new SSAPI();

async function lookup(query, by) {
    // If arg is a number string
    if (query.match(/^\d+$/)) {
        const song = await api.getOpenData(`song/${query}`);
        if (song && song.id) {
            return song;
        }
        throw `${query} not found`;
    }
    const matches = query.match(/^https:\/\/spinsha.re\/song\/(\d+)/);
    if (matches) {
        const song = await api.getOpenData(`song/${matches[1]}`);
        if (song && song.id) {
            return song;
        }
        throw `${query} not found`;
    }
    const data = await api.postOpenData("searchCharts", {
        searchQuery: query,
        showExplicit: true
    });
    if (data.length === 0) {
        throw `${query} not found`;
    } else if (data.length > 1) {
        if (by) {
            const filtered = data.filter(song =>
                song.artist.toLowerCase() == by.toLowerCase() ||
                song.charter.toLowerCase() == by.toLowerCase());
            if (filtered.length == 0) {
                throw `Could not find such a chart by ${by}`;
            }
            if (filtered.length == 1) {
                return filtered[0];
            }
        }
        throw `Too many results, be more specific`;
    } else {
        return data[0];
    }
}

async function open(chatClient, channel, db) {
    const res = await db.query('UPDATE mini_vanilla SET open = true WHERE channel = $1', [channel]);
    chatClient.say(channel, `Requests are now open`);
}

async function close(chatClient, channel, db) {
    const res = await db.query('UPDATE mini_vanilla SET open = false WHERE channel = $1', [channel]);
    chatClient.say(channel, `Requests are now closed`);
}

async function request(chatClient, channel, db, user, args) {
    const { rows } = await db.query('SELECT open from mini_vanilla WHERE channel = $1', [channel]);
    if (rows[0].open === false) {
        chatClient.say(channel, `Sorry, ${user}, requests are closed`);
        return;
    }
    const query = args.filter(w => !w.startsWith("by:")).join(' ');
    const byArg = args.filter(w => w.startsWith("by:"))[0];
    let by = null;
    if (byArg) {
        by = byArg.slice(3);
    }
    const song = await lookup(query, by).catch(err => {
        chatClient.say(channel, err);
    });
    // Insert song into queue
    if (song) {
        const text = `#${song.id}: ${song.title} - ${song.charter} (${song.XDDifficulty})`;
        const { rows } = await db.query('SELECT count(1) from minirequests where channel = $1 and added_by = $2', [channel, user]);
        const priority = rows[0].count;
        const res = await db.query('INSERT INTO minirequests (spin_id, channel, title, added_by, priority, done) VALUES ($1, $2, $3, $4, $5, $6)',
            [song.id, channel, text, user, priority, false]);
        console.log(`Inserting ${res.rowCount} song: ${song.id}`);
        chatClient.say(channel, `Adding #${song.id}: ${song.title} - ${song.charter} (${song.XDDifficulty})`);
    }
}

async function done(channel, db, id) {
    if (id) {
        const res = await db.query('UPDATE minirequests SET done = true WHERE spin_id = $1 AND channel = $2', [id, channel]);
        console.log(`Done ${res.rowCount} song: ${id}`);
    } else {
        const res = await db.query('UPDATE minirequests SET done = true WHERE id = (' +
            'SELECT id FROM minirequests ' +
            'WHERE channel=$1 AND DONE=false ' +
            'ORDER BY priority, id LIMIT 1)', [channel]);
        console.log(`Done ${res.rowCount} song`);
    }
}

async function clear(channel, db) {
    await db.query('DELETE FROM minirequests WHERE channel = $1', [channel]);
}

async function queue(channel, db) {
    const { rows } = await db.query(
        'SELECT title, added_by FROM minirequests ' +
        'WHERE channel=$1 AND done=false ' +
        'ORDER BY priority, id', [channel]);
    return rows;
}

module.exports = {
    clear,
    done,
    queue,
    request,
    open,
    close,
};