deviceController.js

const { execSync, spawn } = require('child_process');
const { promisify } = require('util');
const Client = require('ssh2').Client;

const execute = promisify(require('child_process').exec);

let deviceList = [];
let iproxyList = {};
let clientSshPort = 2222;

const getConnectedDevices = async () => {
    try {
        const { stdout, stderr } = await execute('idevice_id -l');

        if (stderr) {
            throw new Error(stderr);
        }

        let devices = stdout.trim().split('\n');
        if (devices.length === 1 && devices[0] === '') {
            devices = [];
        } else {
            devices = await Promise.all(devices.map(async (id) => ({
                id,
                status: 'uninitialized',
                isJailbroken: await isJailbroken(id),
                ssh: await getSsh(id)
            })));
        }

        return devices;
    } catch (error) {
        console.error(`Error executing command: ${error}`);
        throw error;
    }
};

const reloadDevices = async () => {
    deviceList = await getConnectedDevices();
    await routineDeviceCheck();
};

const getDevices = () => deviceList;

const isJailbroken = async (id) => {
    try {
        const output = execSync(`ideviceinstaller -u ${id} -l -o list_system`);
        return output.includes('Cydia');
    } catch {
        return false;
    }
};

const getSsh = async (id) => {
    if (!iproxyList[id]) await setupSsh(id);
    return iproxyList[id];
};

const setupSsh = async (id) => {
    const port = clientSshPort++;
    const iproxy = spawn('iproxy', [port, 22, '-u', id]); // if you're having problems with iproxy, try omitting '-u'
    iproxyList[id] = { iproxy_pid: iproxy.pid, port };
};

const routineDeviceCheck = () => {
    for (const device of deviceList) {
        const conn = new Client();
        const port = device.ssh.port;

        const connectionParams = {
            host: '127.0.0.1',
            port,
            username: process.env.SSH_OVERRIDE_USERNAME || 'root',
            password: process.env.SSH_OVERRIDE_PASSWORD || 'alpine',
        };

        conn.on('ready', () => {
            device.status = 'ready';
            device.isJailbroken = true;
            conn.end();
        });

        conn.on('error', () => {
            if (device.status != 'offline') {
                device.status = 'offline';
                sendWebhook(`[${device.id}]: Device is offline or not jailbroken! Go to http://${process.env.HOST}/device/${device.id}/start-jailbreak when your device is in DFU mode.`);
            }
        });

        conn.connect(connectionParams);
    }
};

const performJailbreak = async(id) => {
    // TODO: Implement jailbreaking logic
    return Promise.resolve();
};

const enterRecovery = async(id) => await execute(`ideviceenterrecovery ${id}`);

module.exports = {
    getDevices,
    reloadDevices,
    routineDeviceCheck,
    performJailbreak,
    enterRecovery
};