[Substance Painter] P4 functions for Painter

function getInfo() {
    var projectUrl = alg.project.url();
    var filePath = projectUrl.slice(8);

    var checkoutput = alg.subprocess.check_output("p4 fstat " + filePath);
    var test = checkoutput.split ("\n");

    var fileInfo = new Object();

    for (var i = 0; i < test.length; ++i) {
        var temp = test[i].split(" ");
        try {
            fileInfo[temp[1]] =  temp[2].replace("\r", "");
        } catch(e) {
            //pass
        }

}

    return fileInfo

}

function statusColor() {
    var check = getInfo()
    var color;
    if ((check.workRev == check.headRev) || (check.haveRev == check.headRev)) {
        return color = "green";
    } else {
        return color = "red";
    }
}


function addFile() {

    var projectUrl = alg.project.url();
    var filePath = projectUrl.slice(8);

    alg.log.info ("Attempting to add file to default changelist...")
    alg.subprocess.call("p4 add " + filePath);

    var checkChangeList = getChangeList();
    if ( checkChangeList != "None" ) {

    alg.log.info("File has been checked out to default changelist:" + checkChangeList); }
    else {
        alg.log.info("File is not in any changelist... try again or add manually")
    }
}

function editFile() {

    alg.log.info("Opening file for edit...")

    var check = getInfo();
    alg.log.info(check);

    if ( check.isMapped != undefined ) {

        alg.subprocess.check_output("p4 edit " + check.clientFile)
        alg.log.info("File checked out to changelist: " + check.change)
    }
        else {
            alg.log.info("File does not exist in depot")
            addFile()
        }

}

function getChangeList() {
    var projectUrl = alg.project.url();
    var filePath = projectUrl.slice(8);

    var check = getInfo();

    alg.log.info(check.headAction)
    alg.log.info(check.headChange)

    if (check.headAction == 'edit') {

        if (check.headChange == "") {
            return check.change;
        }
    }
    else {
        return undefined;
    }

}