edge.js

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

/**
 * A JXA script and an Alfred Workflow for controlling Google Chrome (Javascript for Automation). 
 * Also see my "How I Navigate Hundreds of Tabs on Chrome with JXA and Alfred" article at [1]
 * if you're interested in learning how I created the workflow.
 * [1] https://medium.com/@bit2pixel/how-i-navigate-hundreds-of-tabs-on-chrome-with-jxa-and-alfred-9bbf971af02b  
 */

ObjC.import('stdlib')
ObjC.import('Foundation')

const chrome = Application('Microsoft Edge')
chrome.includeStandardAdditions = true

// Mode flags
const MODE_CLI = 0    // Ask questions in command line
const MODE_UI = 1     // Ask questions with Chrome dialogs
const MODE_YES = 2    // Answer all questions with `yes`
let MODE = MODE_CLI   // Default mode is command line

// Print the usage message
function usage() {
    println('\n--------------')
    println('Chrome Control')
    println('--------------\n')
    println('list                        List all open tabs in all Chrome windows          usage: ./chrome.js list')
    println('dedup                       Close duplicate tabs                              usage: ./chrome.js dedup')
    println('close <winIdx,tabIdx>       Close a specific tab in a specific window         usage: ./chrome.js close 0,13')
    println('close --title <string(s)>   Close all tabs with titles containing strings     usage: ./chrome.js close --title Inbox "iphone - apple"')
    println('close --url <string(s)>     Close all tabs with URLs containing strings       usage: ./chrome.js close --url mail.google apple')
    println('focus <winIdx,tabIdx>       Focus on a specific tab in a specific window      usage: ./chrome.js focus 0,13')
    println('--ui                        If set, use Chrome to show messages               usage  ./chrome.js close --title inbox --ui')
    println('--yes                       If set, all questions will be anwered with "y"    usage  ./chrome.js close --title inbox --yes')
    $.exit(1)
}

// Run Chrome Control and catch all exceptions
function run(argv) {
    try {
        chromeControl(argv)
    } catch (e) {
        console.log(e)
        $.exit(1)
    }
}

// Chrome Control
function chromeControl(argv) {
    if (argv.length < 1) { usage() }

    // --ui flag will cause the questions to be asked using a 
    // Chrome dialog instead of text in command line.
    let uiFlagIdx = argv.indexOf('--ui')
    if (uiFlagIdx > -1) {
        MODE = MODE_UI
        argv.splice(uiFlagIdx, 1)
    }

    // --yes flag will cause no questions to be asked to the user.
    // It'll close all tabs straight away so use it with caution.
    let yesFlagIdx = argv.indexOf('--yes')
    if (yesFlagIdx > -1) {
        MODE = MODE_YES
        argv.splice(yesFlagIdx, 1)
    }

    const cmd = argv[0]
    if (cmd === 'list') {
        list('all')
    } else if (cmd === 'dedup') {
        dedup()
    } else if (cmd === 'close') {
        if (argv.length == 1) { usage() }
        if (argv.length == 2) {
            const arg = argv[1]
            closeTab(arg)
            $.exit(0)
        }
        const subcmd = argv[1]
        const keywords = argv.slice(2, argv.length)
        closeByKeyword(subcmd, keywords)
    } else if (cmd === 'focus') {
        if (argv.length !== 2) { usage() }
        const arg = argv[1]
        focus(arg)
    } else if (cmd === 'new') {
        create(parseInt(argv[1], 10), argv[2])
    } else {
        usage()
    }

    $.exit(0)
}

// Create new tab in existing window or new window
function create(windowId, url) {
    if (windowId <= 0) {
        //let currentWindowCount = chrome.windows.length
        let newWindow = chrome.Window().make({ "url": url })
        let newTab = chrome.Tab({ "url": url })
        newWindow.tabs.push(newTab)
        newWindow.tabs[0].close()
        println(newWindow.id())
        //chrome.windows.push(targetWindow)
    } else {
        let windows = chrome.windows
        for (let i = 0; i < windows.length; i++) {
            let window = windows[i]
            if (window.id() === windowId) {
                let newTab = chrome.Tab({ "url": url })
                window.tabs.push(newTab)
            }
        }
    }
}

/**
 * Commands
 */

// List all open tabs
function list() {
    // Iterate all tabs in all windows
    // Double entries arrays matching windows/tabs indexes (Using this improves a lot the performances)
    // let allTabsTitle = chrome.windows.tabs.title()
    // let allTabsUrls = chrome.windows.tabs.url()
    // var titleToUrl = {}
    // for (var winIdx = 0; winIdx < allTabsTitle.length; winIdx++) {
    //     for (var tabIdx = 0; tabIdx < allTabsTitle[winIdx].length; tabIdx++) {
    //         let title = allTabsTitle[winIdx][tabIdx]
    //         let url = allTabsUrls[winIdx][tabIdx]

    //         titleToUrl[title] = {
    //             'title': title || 'No Title',
    //             'url': url,
    //             'winIdx': winIdx,
    //             'tabIdx': tabIdx,

    //             // Alfred specific properties
    //             'arg': `${winIdx},${tabIdx}`,
    //             'subtitle': url,
    //         }
    //     }
    // }

    let allWindows = chrome.windows
    //let allTabsUrls = chrome.windows.tabs.url()
    var titleToUrl = {}
    for (var winIdx = 0; winIdx < allWindows.length; winIdx++) {
        let window = allWindows[winIdx]
        let winId = window.id()
        for (var tabIdx = 0; tabIdx < window.tabs.length; tabIdx++) {
            let tab = window.tabs[tabIdx]
            let title = tab.title()
            let url = tab.url()

            titleToUrl[title] = {
                'title': title || 'No Title',
                'url': url,
                'winIdx': winIdx,
                'tabIdx': tabIdx,
                'winId': winId,
                // Alfred specific properties
                'arg': `${winId},${tabIdx}`,
                'subtitle': url,
            }
        }
    }

    // Generate output
    out = { 'items': [] }
    Object.keys(titleToUrl).sort().forEach(title => {
        out.items.push(titleToUrl[title])
    })

    // Print output
    println(JSON.stringify(out))
}

// Close a specific tab
function closeTab(arg) {
    let { winIdx, tabIdx } = parseWinTabIdx(arg)
    for (let i = 0; i < chrome.windows.length; i++) {
        let window = chrome.windows[i]
        if (window.id() === winIdx) {
            let tabToClose = window.tabs[tabIdx]

            // Ask the user before closing tab
            areYouSure([tabToClose], 'Close this tab?', 'Couldn\'t find any matching tabs')

            tabToClose.close()
        }
    }
    // let tabToClose = chrome.windows[winIdx].tabs[tabIdx]

    // // Ask the user before closing tab
    // areYouSure([tabToClose], 'Close this tab?', 'Couldn\'t find any matching tabs')

    // tabToClose.close()
}

// Close a tab if strings are found in the title or URL
function closeByKeyword(cmd, keywords) {
    if (cmd === '--title') {
        getProperty = function (tab) { return tab.title() }
    }
    else if (cmd === '--url') {
        getProperty = function (tab) { return tab.url() }
    } else {
        usage()
    }

    let tabsToClose = []

    // Iterate all tabs in all windows and compare the property returned
    // by `getProperty` to the given keywords
    chrome.windows().forEach(window => {
        window.tabs().forEach(tab => {
            keywords.forEach(keyword => {
                if (getProperty(tab).toLowerCase().includes(keyword.toLowerCase())) {
                    tabsToClose.push(tab)
                }
            })
        })
    })

    // Ask the user before closing tabs
    areYouSure(tabsToClose, 'Close these tabs?', 'Couldn\'t find any matching tabs')

    // Close tabs
    tabsToClose.forEach(tab => { tab.close() })
}


function setWinFront(n) {
    let system = Application('System Events')
    let process = system.processes['Microsoft Edge']
    process.windows[n].actions['AXRaise'].perform()
}
// Focus on a specific tab
function focus(arg) {
    let { winIdx, tabIdx } = parseWinTabIdx(arg)
    for (let i = 0; i < chrome.windows.length; i++) {
        let window = chrome.windows[i]
        if (window.id() === winIdx) {
            //window.visible = false
            window.visible = true
            window.activeTabIndex = tabIdx + 1 // Focous on tab
            window.index = 1 // Focus on this specific Chrome window
            chrome.activate()
        }
    }

}

// Close duplicate tabs
function dedup() {
    let urls = {}
    let dups = []

    chrome.windows().forEach(window => {
        window.tabs().forEach(tab => {
            const url = tab.url();
            if (urls[url] === undefined) {
                urls[url] = null
            } else {
                dups.push(tab)
            }
        })
    })

    // Ask the user before closing tabs
    areYouSure(dups, 'Close these duplicates?', 'No duplicates found')

    // Close tabs
    dups.forEach(tab => { tab.close() })
}

/**
 * Helpers
 */

// Show a message box in Chrome
const alert = function (msg) {
    if (MODE === MODE_YES) {
        return
    }
    chrome.activate()
    chrome.displayAlert(msg)
}

// Grab input from the command line and return it
const prompt = function (msg) {
    if (MODE === MODE_YES) {
        return 'y'
    } else if (MODE === MODE_UI) {
        chrome.activate()
        chrome.displayDialog(msg)
        return
    }
    println(`\n${msg} (y/N)`)
    return $.NSString.alloc.initWithDataEncoding(
        $.NSFileHandle.fileHandleWithStandardInput.availableData,
        $.NSUTF8StringEncoding
    ).js.trim()
}

// JXA always prints to stderr, so we need this custom print function
const print = function (msg) {
    $.NSFileHandle.fileHandleWithStandardOutput.writeData(
        $.NSString.alloc.initWithString(String(msg))
            .dataUsingEncoding($.NSUTF8StringEncoding)
    )
}

// Print with a new line at the end
const println = function (msg) {
    print(msg + '\n')
}

// Ask the user before closing tabs
function areYouSure(tabsToClose, promptMsg, emptyMsg) {
    // Give user feedback if no matching tabs were found
    if (tabsToClose.length === 0) {
        if (MODE == MODE_CLI) {
            println(emptyMsg)
        } else {
            alert(emptyMsg)
        }

        $.exit(0)
    }

    // Grab the titles to show to the user
    let titles = []
    tabsToClose.forEach(tab => {
        titles.push(tab.title())
    })

    // Focus on Chrome and ask user if they really want to close these tabs
    if (MODE == MODE_CLI) {
        println(`\n${titles.join('\n\n')}`)
        if (prompt(promptMsg) !== 'y') {
            println('Canceled')
            $.exit(0)
        }
    } else {
        prompt(`${promptMsg}\n\n${titles.join('\n\n')}`)
    }

}

// Get winIdx and tabIdx from arg
function parseWinTabIdx(arg) {
    const s = arg.split(',')
    if (s.length !== 2) {
        println('\nInvalid window and tab index. Example: 0,13\n')
        usage()
    }

    let winIdx = parseInt(s[0])
    let tabIdx = parseInt(s[1])

    if (isNaN(winIdx) || isNaN(tabIdx)) {
        throw ("Error: winIdx and tabIdx must be integers")
    }

    return { winIdx, tabIdx }
}