filter.js

#!/usr/bin/env osascript -l JavaScript

function getenv(name, default_value) {
    var env = $.NSProcessInfo.processInfo.environment.js;
    return (name in env) ? env[name].js : default_value;
}

function sessionToObj(windowId, tabIndex, query) {
    return (session) => ({
        id:      [windowId, tabIndex, session.id()].join(" "),
        title:   session.name(),
        tty:     session.tty(),
        output:  session.isProcessing(),
        profile: session.profileName(),
        query:   query
    });
}

function objToItem(obj) {
    var description = obj.tty + (obj.output?" *":"");
    return {
        title: obj.title,
        subtitle: ["Select", description].join(" "),
        arg: ["select", obj.id].join(" "),
        uid: [obj.title, obj.profile].join("/"),
        icon: { path: "icon.png" },
        mods: {
            alt: {
                arg: ["close", obj.id].join(" "),
                subtitle: ["Close", description].join(" ")
            },
            cmd: {
                arg: ["ssh", obj.query].join(" "),
                title: ["⇌ ssh", obj.query].join(" "),
                subtitle: "Switch to ssh workflow"
            }
        }
    };
}

function allSessionObjs(app, query) {
    var windows = app.windows;
    var results = new Array();
    var windowIds = windows.id();
    for (var i=0; i < windowIds.length; ++i) {
        var windowId = windowIds[i];
        var tabs = windows.byId(windowId).tabs;
        for (var tabIndex=0; tabIndex < tabs.length; ++tabIndex) {
            results.push(
                ...tabs[tabIndex].sessions().map(
                    sessionToObj(windowId, tabIndex, query)
                )
            );
        }
    }
    return results;
}

function titleFilter(pattern) {
    var re = new RegExp(pattern.replace(/./g, "$&.*?"), "i");
    return (obj) => re.exec(obj.title);
}

function ttyFilter(ttyNum) {
    if (ttyNum.length == 0) {
        return () => true;
    } else {
        var re = new RegExp(ttyNum.padStart(3, "0") + "$");
        return (obj) => re.exec(obj.tty);
    }
}

function run(args) {
    args = args.join(" ").split(" ");
    var query = String(args[0] || "");
    var tty = String(args[1] || "");
    var sessionObjs = new Array();
    var app = Application(getenv("iterm_application", "com.googlecode.iterm2"));
    app.strictPropertyScope = true;
    app.strictCommandScope  = true;
    app.strictParameterType = true;
    if (app.running()) {
        sessionObjs = allSessionObjs(app, query)
            .filter(titleFilter(query))
            .filter(ttyFilter(tty));
    }
    if (sessionObjs.length == 0) {
        return JSON.stringify({
            items: [
                {
                    title: ["⇌ ssh", query].join(" "),
                    subtitle: "No matches! Switch to ssh workflow?",
                    arg: ["ssh", query].join(" "),
                    icon: { path: "icon.png" },
                }
            ]
        });
    } else {
        return JSON.stringify({
            items: sessionObjs.map(objToItem)
        });
    }
}