/**
 * Allows passing arguments to the function executed by startParallelShell.
 */
export const funWithArgs = (fn, ...args) => {
    const result = `(${fn.toString()})(${args.map((x) => tojson(x)).reduce((x, y) => `${x}, ${y}`)})`;
    if (fn.constructor.name === "AsyncFunction") {
        return `await ${result}`;
    }

    return result;
};

/**
 * Internal function used by _doAssertCommandInParallelShell().
 */
export const _parallelShellRunCommand = function (dbName, cmdObj, expectedCode, checkResultFn, checkResultArgs) {
    const expectedCodeStr = expectedCode === undefined ? "" : "; expectedCode: " + tojson(expectedCode);
    jsTestLog(
        "Starting command in parallel shell - host: " +
            db.getMongo().host +
            "; db: " +
            dbName +
            "; command: " +
            tojson(cmdObj) +
            expectedCodeStr,
    );
    const testDB = db.getSiblingDB(dbName);
    const result = testDB.runCommand(cmdObj);
    jsTestLog(
        "Finished running command in parallel shell - host: " +
            db.getMongo().host +
            "; db: " +
            dbName +
            "; command: " +
            tojson(cmdObj) +
            expectedCodeStr +
            "; result: " +
            tojson(result),
    );
    if (expectedCode === undefined) {
        assert.commandWorked(result);
    } else {
        assert.commandFailedWithCode(result, expectedCode);
    }
    if (!checkResultFn) {
        return;
    }
    jsTestLog(
        "Checking command result in parallel shell - host: " +
            db.getMongo().host +
            "; db: " +
            dbName +
            "; command: " +
            tojson(cmdObj) +
            expectedCodeStr +
            "; result: " +
            tojson(result) +
            "; checkResultFn: " +
            checkResultFn +
            "; checkResultArgs: " +
            tojson(checkResultArgs),
    );
    checkResultFn(result, checkResultArgs);
    jsTestLog(
        "Successfully verified command result in parallel shell - host: " +
            db.getMongo().host +
            "; db: " +
            dbName +
            "; command: " +
            tojson(cmdObj) +
            expectedCodeStr +
            "; result: " +
            tojson(result),
    );
};

/**
 * Internal function used by assertCommandWorkedInParallelShell() and
 * assertCommandFailedWithCodeInParallelShell().
 */
export const _doAssertCommandInParallelShell = function (
    conn,
    db,
    cmdObj,
    expectedCode,
    checkResultFn,
    checkResultArgs,
) {
    // Return joinable object to caller.
    return startParallelShell(
        funWithArgs(_parallelShellRunCommand, db.getName(), cmdObj, expectedCode, checkResultFn, checkResultArgs),
        conn.port,
    );
};

/**
 * Starts command in a parallel shell.
 * Provides similar behavior to assert.commandWorked().
 */
export const assertCommandWorkedInParallelShell = function (conn, db, cmdObj, checkResultFn, checkResultArgs) {
    return _doAssertCommandInParallelShell(conn, db, cmdObj, checkResultFn, checkResultArgs);
};

/**
 * Starts command in a parallel shell.
 * Provides similar behavior to assert.commandFailedWithCode().
 */
export const assertCommandFailedWithCodeInParallelShell = function (
    conn,
    db,
    cmdObj,
    expectedCode,
    checkResultFn,
    checkResultArgs,
) {
    assert(
        expectedCode,
        "expected error code(s) must be provided to run command in parallel shell - host: " +
            db.getMongo().host +
            "; db: " +
            db.getName() +
            "; command: " +
            tojson(cmdObj),
    );
    return _doAssertCommandInParallelShell(conn, db, cmdObj, expectedCode, checkResultFn, checkResultArgs);
};