/**
 * $lookup from a view whose definition contains a bind-sensitive extension stage binds the view
 * onto the view's own definition stage at execution time, so the stage rejects the view. Reading
 * the view directly or through $unionWith binds an empty namespace to the definition stages and
 * succeeds.
 *
 * @tags: [featureFlagExtensionsAPI]
 */
import {
    checkPlatformCompatibleWithExtensions,
    deleteExtensionConfigs,
    generateExtensionConfigs,
    getExtensionConfDir,
} from "jstests/noPassthrough/libs/extension_helpers.js";

checkPlatformCompatibleWithExtensions();
const extNames = generateExtensionConfigs(["libdisallow_view_mongo_extension.so"]);
const conn = MongoRunner.runMongod({
    loadExtensions: extNames,
    extensionsConfigPath: getExtensionConfDir(),
    extensionsSignaturePublicKeyPath:
        "src/mongo/db/extension/test_examples/test_extensions_signing_keys/test_extensions_signing_public_key.asc",
});
const db = conn.getDB(jsTestName());

assert.commandWorked(db.c.insertMany([{_id: 1, a: 1}, {_id: 2, a: 2}]));
assert.commandWorked(db.l.insert({_id: 1, a: 1}));
// $disallowViews is a pass-through stage that uasserts 11507700 when bound to a view.
assert.commandWorked(db.createView("v", "c", [{$disallowViews: {}}]));
assert.commandWorked(db.createView("vMatch", "c", [{$match: {a: {$gte: 0}}}]));

function run(label, fn) {
    try {
        const res = fn();
        jsTest.log.info("B run", {label, res});
        return {res};
    } catch (e) {
        jsTest.log.info("B run", {label, code: e.code, err: String(e.message).slice(0, 200)});
        return {code: e.code};
    }
}
const ids = (docs) => docs.map((d) => d._id).sort();

const direct = run("aggregate on v", () => ids(db.v.aggregate([]).toArray()));
const union = run("$unionWith v", () =>
    ids(db.l.aggregate([{$match: {_id: -1}}, {$unionWith: {coll: "v", pipeline: []}}]).toArray()));
const lookupMatchView = run("$lookup lf/ff from vMatch", () =>
    ids(db.l.aggregate([{$lookup: {from: "vMatch", localField: "a", foreignField: "a", as: "j"}}]).toArray()[0].j));
const onView = run("$disallowViews in a user pipeline on vMatch", () => db.vMatch.aggregate([{$disallowViews: {}}]).toArray());
run("$lookup pipeline from v", () =>
    ids(db.l.aggregate([{$lookup: {from: "v", pipeline: [], as: "j"}}]).toArray()[0].j));
const lookupV = run("$lookup lf/ff from v", () =>
    ids(db.l.aggregate([{$lookup: {from: "v", localField: "a", foreignField: "a", as: "j"}}]).toArray()[0].j));

// Controls: the view definition stage runs fine on the backing collection.
assert.eq(direct.res, [1, 2], "aggregate on v");
assert.eq(union.res, [1, 2], "$unionWith v");
assert.eq(lookupMatchView.res, [1], "$lookup from a plain view");
// Control: the stage does reject a view when it is a user stage running on one.
assert.eq(onView.code, 11507700, "$disallowViews on a view");
// Fails at ba06f31d06c: error 11507700 "Attempted to use in view: v".
assert.eq(lookupV.res, [1], "$lookup lf/ff from v");

MongoRunner.stopMongod(conn);
deleteExtensionConfigs(extNames);