Closing all FOLDER tracks in Pro Tools

There still seems to be issues with the selectAllTracks logic. It works if the first track in the session is a folder, but not another type of track. Hmmmm

I'm going to come back to this and refine it a bunch. I just figured out another problem with it: it won't work if the tracks are small enough to hide the folder button, and I'll try and fix the track logic so it doesn't need to be a folder track at the top first :)

OK here's something. I think I've made the select all tracks a bit more reliable (no longer need a folder track at the top of the session?) and to speed things up, this track specifically looks for he name folder in the track name of all folder tracks (aka you must have the word 'folder' in your folder track names) and then only selects those tracks to try and collapse. Checking for the button on all tracks took too long when the first say, 50 tracks weren't folders. Maybe this is good organizationally in the long run? (time will tell :) )
(EDIT: I made it so it restores the track selection when it's finished)

function selectAllTracks() {
    var trackHeader = sf.ui.proTools.trackGetAllTracks().trackListItems[0].header;
    trackHeader.trackSelect();
    trackHeader.trackScrollToView();
    sf.ui.proTools.trackDeselectAll();
    if (sf.ui.proTools.mainWindow.groups.allItems[7].popupButtons.first.exists) {
        sf.ui.proTools.mainWindow.groups.allItems[7].popupButtons.first.mouseClickElement({ isOption: true });
    } else if (sf.ui.proTools.trackListItems[0].children.allItems[1].buttons.first.exists) {
        sf.ui.proTools.trackListItems[0].children.allItems[1].buttons.first.mouseClickElement({ isOption: true });
    } else {
        alert(`Couldn't select all tracks for some reason`)
        throw 0;
    }

}

//this function here is all Christian. He's literally the best.
function trackSize() {

    var size = 'small';

    var f = sf.ui.proTools.selectedTrack.frame;
    var popupMenu = sf.ui.proTools.selectedTrack.popupMenuOpenFromElement({
        relativePosition: { x: f.w - 10, y: 5 },
        isOption: true,
        isShift: true,
    }).popupMenu;
    popupMenu.menuClickPopupMenu({
        menuPath: [size]
    });

}

sf.ui.proTools.appActivateMainWindow();

sf.ui.proTools.mainWindow.invalidate();

var originalSelection = sf.ui.proTools.selectedTrackNames;

selectAllTracks();

var selectedTracks = sf.ui.proTools.selectedTrackNames;

var folderTracks = selectedTracks.filter(n => n.match(/Folder/i))

trackSize();

for (var i = 0; i < folderTracks.length; i++) {
    sf.ui.proTools.trackSelectByName({
        deselectOthers: true,
        names: [folderTracks[i]]
    });

    if (sf.ui.proTools.selectedTrack.children.whoseRole.is("AXDisclosureTriangle").whoseTitle.is('Close Folder').first.exists) {
        sf.ui.proTools.selectedTrack.trackScrollToView();
        sf.ui.proTools.selectedTrack.children.whoseRole.is("AXDisclosureTriangle").first.mouseClickElement({
            isOption: true,
        });
        break;
    }

}

sf.ui.proTools.trackSelectByName({
    deselectOthers: true,
    names: originalSelection
});