batch import session data + auto match tracks + video

Hello everyone,
this is a script I'm working on to automate the process of importing loads of AAF to a session.

If you see better way to organise, or improve any part, would be great to hear from you and improve it.

One technical "pickle" I encountered is that the "Import Session Data" window must have all tracks visible so SF can see them all, there is no "scroll to track" for the menu, at least I don't know how. For me that means moving the window to a screen rotated 90º, fortunately my setup has one this way.

One thing I'm struggling with is the occasions where a AAF has a track that is not already present on the session. The ideal solution would be to set it to "New Track" but I'm failing miserably on figuring out where in the sequence this should be done. Usually a set of AAF to import will all have the same tracks, but occasionally a new track will be present.

As is, if an AAF has a new track, the script will fail and stop on the function selectPath(i, path) , witch makes cense since I'm asking to select a path that doesn't exist.
So I tried to do a catch (err) , this https://forum.soundflow.org/-348#post-11, was very helpful but for reasons I don't understand, there is an error already leaving the function, I see it only in the log file, and it needs to be there. If I catch the error, I'll be able to set the tracks but will loose the correct sequence of match track.

Any Ideas on what can I do to fix this?
If it becomes too complicated, I think an option could be to cancel the import, moving to the next one and write the name of the AAF on a text file as report of failure. But as I see it now, the place to do any action would be where its fails, and I'm unable to do any action without messing the sequence.

Here's a video of the script in action:
https://we.tl/t-fSZL72JCc4

Thank you all so much !

function letsStartWin() {
    const startWin = sf.interaction.displayDialog({
        defaultAnswer: "",
        buttons: ["Video Location & Go!", "Go!", "Cancel"],
        defaultButton: "Go!",
        cancelButton: "Cancel",
        title: "Batch Import Session Data",
        prompt: "Manualy Import the first AAF.\nThis will ensure a set of tracks is in place for eficient Match Track. \nCreate a new video track for the imported video.\n\nBE AWARE !!! \nVideo files will be added on the ONLINE VIDEO TRACK,\nSo make sure the NEW TRACK IS ONLINE! \nOtherwise the imported videos might go to the wrong track and will not end well !!! 😡 \n\n\By default, the script searches for video files in the session video folder, if you want a diferent location choose one.\n\nTo use \"Map Start Time To...\" enter the Time Code of the interval you want.\nLeave blank to use AAF start time.\n\nhh:mm:ss:ff"
    });
    return startWin
}

function alertSound() {
    sf.system.volumeSet({ volume: 0.5 });
    sf.system.exec({
        commandLine: `afplay /System/Library/Sounds/Glass.aiff`
    });
    sf.system.volumeSet({ volume: 1.0 });
}

function moveWindowandsize() {
    var screenIndex = 2;
    var screenFrame = sf.ui.screens.allScreens[screenIndex].frame;

    sf.ui.proTools.windows.whoseTitle.is('Import Session Data').first.windowMove({
        position: { x: screenFrame.x + 0, y: screenFrame.y + 22 },
        size: { x: screenFrame.x + 0, y: screenFrame.y + 3300 },
    });
}

function importWindow(importMenu, list) {

    sf.ui.proTools.appActivateMainWindow();

    sf.ui.proTools.appWaitForActive({
        timeout: -1
    });

    sf.ui.proTools.menuClick({
        menuPath: ["File", "Import", importMenu],
    });

    var win = sf.ui.frontmostApp.focusedWindow;

    sf.keyboard.type({ text: '/' });

    var sheet = win.sheets.first.elementWaitFor({ timeout: 500 }, 'Could not find "Go to" sheet in the Save/Open dialog').element;
    //// //////////////////////////Set Path
    sheet.comboBoxes.first.value.value = list;

    sheet.buttons.whoseTitle.is('Go').first.elementClick({}, 'Could not click "Go"');

    win.sheets.first.elementWaitFor({ waitForNoElement: true, timeout: 500 }, '"Go to" sheet didn\'t close in time');

    sf.ui.proTools.windows.whoseTitle.is('Open').first.buttons.whoseTitle.is('Open').first.elementClick();

    sf.ui.proTools.children.whoseRole.is("AXWindow").whoseTitle.is('Open').first.elementWaitFor({ waitType: "Disappear" });

    if (sf.ui.proTools.confirmationDialog.invalidate().buttons.whoseTitle.is('OK').first.exists) {
        sf.ui.proTools.confirmationDialog.buttons.whoseTitle.is('OK').first.elementClick();
        sf.ui.proTools.confirmationDialog.elementWaitFor({ waitType: "Disappear", onError: "Continue" });
    }
}

function dismissSessionNotes() {

    var sessionNotesWin = sf.ui.proTools.windows.whoseTitle.is('Session Notes')

    if (sessionNotesWin.exists) {

        sessionNotesWin.first.buttons.whoseTitle.is('No').first.elementClick();

        sessionNotesWin.first.elementWaitFor({
            waitType: "Disappear",
        });
    }

}

function closeImportDataWin() {

    var button = sf.ui.proTools.windows.whoseTitle.is('Import Session Data').first

    button.buttons.allItems[1].elementClick();

    button.elementWaitFor({ waitType: "Disappear" });

    dismissSessionNotes()


}

function getAafDetails() {
    const win = sf.ui.proTools.windows.whoseTitle.is('Import Session Data').first
    const sourcePanel = win.groups.whoseTitle.is('Source Properties:').invalidate().first;
    const tableData = sourcePanel.childrenByRole('AXTable').first.childrenByRole('AXRow')
        .map(fields => fields.childrenByRole('AXCell').first.children.first.title.invalidate().value);

    return tableData
}


function selectionSetAafStartTime(importingAaf) {    /////////////////   THIS WILL CLOSE IMPORT SESSION WINDOW ///////////////////////

    let aafDetails = getAafDetails();

    let aafStartTime = aafDetails.map(x => x.split("Start time:").slice(1)).join("")

    closeImportDataWin()

    sf.ui.proTools.selectionSetTimecode({ mainCounter: aafStartTime });

    var newMemLocWin = sf.ui.proTools.newMemoryLocationDialog
    sf.keyboard.press({ keys: "numpad enter" });
    newMemLocWin.elementWaitFor();
    newMemLocWin.textFields.allItems[1].elementSetTextFieldWithAreaValue({ value: importingAaf.join().split(".").slice(0, -1).join("") });
    newMemLocWin.radioButtons.first.checkboxSet({ targetValue: "Enable" });
    newMemLocWin.buttons.whoseTitle.is('OK').first.elementClick();
    newMemLocWin.elementWaitFor({ waitType: "Disappear" });
}

function getNextTcStart(intervalStartTime) {
  
    sf.ui.proTools.selectionSetTimecode({ selectionLength: intervalStartTime });
    var getEndTimecode = sf.ui.proTools.mainWindow.groups.whoseTitle.is('Counter Display Cluster').first.textFields.whoseTitle.is('Edit Selection End').first.invalidate().value.value;
    return getEndTimecode
}


function selectionSetUserInterval(intervalStartTime, importingAaf) {    /////////////////   THIS WILL CLOSE IMPORT SESSION WINDOW ///////////////////////

    let importSessionWin = sf.ui.proTools.windows.whoseTitle.is('Import Session Data').first

    importSessionWin.groups.whoseTitle.is('Timecode Mapping Options:').first.popupButtons.first.popupMenuSelect({
        menuPath: ["Map start timecode to"]
    });

    importSessionWin.mouseClickElement({ relativePosition: { "x": 390, "y": 97 } });

    sf.keyboard.type({ text: intervalStartTime });

    importSessionWin.mouseClickElement({ relativePosition: { "x": 530, "y": 100 } });

    closeImportDataWin()

    sf.ui.proTools.selectionSetTimecode({ mainCounter: intervalStartTime });

    var newMemLocWin = sf.ui.proTools.newMemoryLocationDialog
    sf.keyboard.press({ keys: "numpad enter" });
    newMemLocWin.elementWaitFor();
    newMemLocWin.textFields.allItems[1].elementSetTextFieldWithAreaValue({ value: importingAaf.join().split(".").slice(0, -1).join("") });
    newMemLocWin.radioButtons.first.checkboxSet({ targetValue: "Enable" });
    newMemLocWin.buttons.whoseTitle.is('OK').first.elementClick();
    newMemLocWin.elementWaitFor({ waitType: "Disappear" });
}

function setVideoImportOptions() {

    const videoImportOprionsWin = sf.ui.proTools.windows.whoseTitle.is('Video Import Options').first
    const radioButtons = videoImportOprionsWin.radioButtons
    const checkBoxes = videoImportOprionsWin.checkBoxes

    videoImportOprionsWin.elementWaitFor({ waitType: "Appear" });

    if (radioButtons.allItems[1].invalidate().isEnabled == true) {
        radioButtons.allItems[1].checkboxSet({ targetValue: "Enable" });

    } else if (radioButtons.allItems[1].invalidate().isEnabled == false) {
        radioButtons.allItems[2].checkboxSet({ targetValue: "Enable" });
    }

    if (videoImportOprionsWin.popupButtons.first.invalidate().value.value !== "Selection") {
        videoImportOprionsWin.popupButtons.first.popupMenuSelect({ menuPath: ["Selection"] });
    }

    checkBoxes.allItems[0].checkboxSet({ targetValue: "Disable" });

    checkBoxes.allItems[1].checkboxSet({ targetValue: "Disable" });

    checkBoxes.allItems[2].checkboxSet({ targetValue: "Disable" });

    checkBoxes.allItems[3].checkboxSet({ targetValue: "Disable" });

    videoImportOprionsWin.buttons.whoseTitle.is('OK').first.elementClick();

    videoImportOprionsWin.elementWaitFor({ waitType: "Disappear" });

}

function setDataImportOptions() {
    let importSessionButtons = sf.ui.proTools.windows.whoseTitle.is('Import Session Data').first.groups.whoseTitle.is('Session Data').first.checkBoxes

    moveWindowandsize()

    sf.ui.proTools.windows.whoseTitle.is('Import Session Data').first.groups.whoseTitle.is('Tracks').first.buttons.whoseTitle.is('Match Tracks').first.elementClick();

    importSessionButtons.allItems[3].checkboxSet({ targetValue: "Disable" });

    importSessionButtons.first.checkboxSet({ targetValue: "Disable" });

    importSessionButtons.allItems[2].checkboxSet({ targetValue: "Disable" });

    importSessionButtons.allItems[4].checkboxSet({ targetValue: "Enable" });

    importSessionButtons.allItems[1].checkboxSet({ targetValue: "Disable" });

    sf.ui.proTools.windows.whoseTitle.is('Import Session Data').first.radioButtons.allItems[1].checkboxSet({ targetValue: "Enable" });

    var trackDataToImportWin = sf.ui.proTools.windows.whoseTitle.is('Track Data to Import').first

    var checkBoxes = trackDataToImportWin.groups.whoseTitle.is('Track Data').first.checkBoxes

    sf.ui.proTools.windows.whoseTitle.is('Import Session Data').first.groups.whoseTitle.is('Session Data').first.buttons.whoseTitle.is('Choose...').first.elementClick();

    trackDataToImportWin.elementWaitFor({ waitType: "Appear" });

    checkBoxes.allItems[0].checkboxSet({ targetValue: "Enable" });

    checkBoxes.allItems[1].checkboxSet({ targetValue: "Enable" });

    checkBoxes.allItems[2].checkboxSet({ targetValue: "Disable" });

    checkBoxes.allItems[3].checkboxSet({ targetValue: "Disable" });

    checkBoxes.allItems[4].checkboxSet({ targetValue: "Disable" });

    checkBoxes.allItems[5].checkboxSet({ targetValue: "Disable" });

    checkBoxes.allItems[6].checkboxSet({ targetValue: "Disable" });

    trackDataToImportWin.buttons.whoseTitle.is('OK').first.elementClick();

    trackDataToImportWin.elementWaitFor({ waitType: "Disappear" });
}

function selectPath(i, path) {
    try {
        let list = sf.ui.proTools.windows.whoseTitle.is('Import Session Data').first.groups.whoseTitle.is('Tracks').first.tables.whoseTitle.is('Source Destination').first.children.whoseRole.is("AXRow")
        let poppupMenu = list.allItems[i].children.whoseRole.is("AXCell").allItems[1].children.whoseRole.is("AXMenuButton").first
        poppupMenu.popupMenuSelect({
            menuPath: [path],
            targetValue: "Enable"
        });
    } catch (err) {
        selectPath(i, "New Track")
    }
}


function isMonoMenu() {
    let tracksTable = sf.ui.proTools.windows.whoseTitle.is('Import Session Data').first.groups.whoseTitle.is('Tracks').first.tables.whoseTitle.is('Source Destination').first.children.whoseRole.is("AXRow")
    let numberOfTracks = tracksTable.allItems.count
    for (var i = 0; i < numberOfTracks; i++) {
        let stereoOrMono = tracksTable.allItems[i].children.whoseRole.is("AXCell").first.children.whoseRole.is("AXStaticText").first.invalidate().value.value.split(" (").splice(1).join()

        if (stereoOrMono == "Mono audio)") {
            var monoDestinationMenu = tracksTable.allItems[i].children.whoseRole.is("AXCell").allItems[1].children.whoseRole.is("AXMenuButton").first.popupMenuFetchAllItems().menuItems.map(mi => mi.path.join(' -> '));
            sf.ui.proTools.appActivateMainWindow();
            return monoDestinationMenu
        }
    }
}

function isStereoMenu() {
    let tracksTable = sf.ui.proTools.windows.whoseTitle.is('Import Session Data').first.groups.whoseTitle.is('Tracks').first.tables.whoseTitle.is('Source Destination').first.children.whoseRole.is("AXRow")
    let numberOfTracks = tracksTable.allItems.count
    for (var i = 0; i < numberOfTracks; i++) {
        let stereoOrMono = tracksTable.allItems[i].children.whoseRole.is("AXCell").first.children.whoseRole.is("AXStaticText").first.invalidate().value.value.split(" (").splice(1).join()

        if (stereoOrMono == "Stereo audio)") {
            var stereoDestinationMenu = tracksTable.allItems[i].children.whoseRole.is("AXCell").allItems[1].children.whoseRole.is("AXMenuButton").first.popupMenuFetchAllItems().menuItems.map(mi => mi.path.join(' -> '));
            sf.ui.proTools.appActivateMainWindow();
            return stereoDestinationMenu
        }
    }
}


function autoMatchNext(firstIndex, nameFromSource, numberOfDups) {

    let destinationbutton = sf.ui.proTools.windows.whoseTitle.is('Import Session Data').first.groups.whoseTitle.is('Tracks').first.tables.whoseTitle.is('Source Destination').first.children.whoseRole.is("AXRow")

    const duplicates = numberOfDups.filter(x => x.startsWith(nameFromSource)).length - 1

    for (var i = 0; i < duplicates; i++) {

        let dotNumber = 1 + i

        let n = firstIndex + i

        let nameFromSourcePlusDot = nameFromSource + "." + dotNumber

        selectPath(n, nameFromSourcePlusDot);
    }
}


function autoMatchFirst() {
    let tracksTable = sf.ui.proTools.windows.whoseTitle.is('Import Session Data').first.groups.whoseTitle.is('Tracks').first.tables.whoseTitle.is('Source Destination').first.children.whoseRole.is("AXRow")
    let numberOfTracks = tracksTable.allItems.count
    let destinationList
    let nameFromSource

    let stereoMenu = isStereoMenu()
    let monoMenu = isMonoMenu()
    let destinationMenu = monoMenu.concat(stereoMenu)

    let numberOfDups = tracksTable.allItems.slice().map(l => l.children.whoseRole.is("AXCell").first.children.whoseRole.is("AXStaticText").first.invalidate().value.value.split(" (").slice(0, -1).join())


    for (var i = 0; i < numberOfTracks; i++) {
        var firstIndex = i + 1

        destinationList = tracksTable.allItems[i].children.whoseRole.is("AXCell").allItems[1].children.whoseRole.is("AXMenuButton").first.invalidate().value.value

        nameFromSource = tracksTable.allItems[i].children.whoseRole.is("AXCell").first.children.whoseRole.is("AXStaticText").first.invalidate().value.value.split(" (").slice(0, -1).join();

        var sourcePlusDot = nameFromSource + ".1"

        if (destinationList === "(none)" && destinationMenu.indexOf(nameFromSource) !== -1) {
            selectPath(i, nameFromSource)
        }
        if (destinationList === "(none)" && destinationMenu.indexOf(sourcePlusDot) !== -1) {
            autoMatchNext(firstIndex, nameFromSource, numberOfDups)
        }
        if (destinationList === "(none)" && destinationMenu.indexOf(nameFromSource) == -1) {
            selectPath(i, "New Track")
        }
    }
}


function sessionVideoFolder() {


    try {
        var sessionVFolder = sf.ui.proTools.mainWindow.sessionPath.split('/').slice(0, -1).join('/') + '/Video Files';

        var sessionVideoFolder = sf.file.directoryGetFiles({
            path: sessionVFolder,
            isRecursive: true,
        }).paths.filter(p => p.match(/\.(mov|mp4|mxf)$/i));

        return sessionVideoFolder

    } catch (err) {
        alert("Whoops something went wrong!")
        throw 0
    }
}


function altVideoFolder() {

    var sessionVFolder = sf.ui.proTools.mainWindow.sessionPath.split('/').slice(0, -1).join('/') + '/Video Files';

    const videoFolderPath = sessionVideoFolder()

    var folderPath = sf.interaction.selectFolder({
        defaultLocation: sessionVFolder,
        prompt: 'Select Video Files Folder',
    }).path

    try {
        var altVideoFolder = sf.file.directoryGetFiles({
            path: folderPath,
            isRecursive: true,
        }).paths.filter(p => p.match(/\.(mov|mp4|mxf)$/i));

        return altVideoFolder

    } catch (err) {
        log("No Files on Selected Folder")
    }
}


////////////////////////////////////////////////////
///////////////////     Main   /////////////////////


function batchImport() {

    //// Start Prompt
    const startWinAnswer = letsStartWin()


    //// Check if user interval is valid
    /////https://gist.github.com/allensarkisyan/5441601
    var isValidSMPTETimeCode = /(^(?:(?:[0-1][0-9]|[0-2][0-3]):)(?:[0-5][0-9]:){2}(?:[0-2][0-9])$)/;
    if (isValidSMPTETimeCode.test(startWinAnswer.text) == false && startWinAnswer.text.length > 1) {
        alert("Time Code Not Valid!")
        throw 0
    }


    /// decide video location
    let videoFilePath
    if (startWinAnswer.button == "Go!") {
        videoFilePath = sessionVideoFolder()

    }
    if (startWinAnswer.button == "Video Location & Go!") {
        videoFilePath = altVideoFolder()
    }

    /////////////////////////////////////////////////////////////
    sf.ui.proTools.appActivateMainWindow();
    sf.ui.proTools.invalidate();
    /////////////////////////////////////////////////////////////
    var aafPath = sf.ui.finder.selectedPaths;
    var importSessionData = "Session Data..."
    var ImportVideo = "Video..."





    for (let i = 0; i < aafPath.length; i++) {

        ////////////////   AAF   ////////////////
        let aafList = aafPath[i]

        let importingAaf = aafList.split("/").slice(-1)

        log("Import Session Data: \n" + importingAaf)

        //////////////////   VIDEO  ///////////////
        var getVideoFile = videoFilePath.filter(x => x.includes(aafList.split("/").slice(-1).join("/").split(".").shift()))

        if (getVideoFile.length == 0) {
            log("No Video Found.")
        }

        //////////////////   Map Start Time To  //////////////      
        let intervalStartTime
        if (isValidSMPTETimeCode.test(startWinAnswer.text) == true) {
            intervalStartTime = getNextTcStart(startWinAnswer.text)
        }

        //=================================================================//
        importWindow(importSessionData, aafList)

        setDataImportOptions()

        autoMatchFirst()

        if (isValidSMPTETimeCode.test(startWinAnswer.text) == true) {
            selectionSetUserInterval(intervalStartTime, importingAaf)

        } else {
            selectionSetAafStartTime(importingAaf)
        }

        if (getVideoFile.length == 1) {
            importWindow(ImportVideo, getVideoFile.join())

            setVideoImportOptions()
        }

        sf.ui.proTools.menuClick({ menuPath: ["File", "Save"] });
    }
    alertSound()
    alert("Done!")

}


batchImport()