//
// On error inserting documents, traces back and shows where the document was dropped
//

export function traceMissingDoc(coll, doc, mongos) {
    if (mongos) coll = mongos.getCollection(coll + "");
    else mongos = coll.getMongo();

    let config = mongos.getDB("config");
    let shards = config.shards.find().toArray();
    for (var i = 0; i < shards.length; i++) {
        shards[i].conn = new Mongo(shards[i].host);
    }

    let shardKeyPatt = config.collections.findOne({_id: coll + ""}).key;

    // Project out the shard key
    let shardKey = {};
    for (let k in shardKeyPatt) {
        if (doc[k] == undefined) {
            jsTest.log(
                "Shard key " +
                    tojson(shardKey) +
                    " not found in doc " +
                    tojson(doc) +
                    ", falling back to _id search...",
            );
            shardKeyPatt = {_id: 1};
            shardKey = {_id: doc["_id"]};
            break;
        }
        shardKey[k] = doc[k];
    }

    if (doc["_id"] == undefined) {
        jsTest.log("Id not found in doc " + tojson(doc) + " cannot trace oplog entries.");
        return;
    }

    jsTest.log("Using shard key : " + tojson(shardKey));

    let allOps = [];
    for (var i = 0; i < shards.length; i++) {
        let oplog = shards[i].conn.getCollection("local.oplog.rs");

        if (!oplog.findOne()) {
            jsTest.log("No oplog was found on shard " + shards[i]._id);
            continue;
        }

        let addKeyQuery = function (query, prefix) {
            for (let k in shardKey) {
                query[prefix + "." + k] = shardKey[k];
            }
            return query;
        };

        let addToOps = function (cursor) {
            cursor.forEach(function (doc) {
                doc.shard = shards[i]._id;
                doc.realTime = new Date(doc.ts.getTime() * 1000);
                allOps.push(doc);
            });
        };

        // Find ops
        addToOps(oplog.find(addKeyQuery({op: "i"}, "o")));
        let updateQuery = {$or: [addKeyQuery({op: "u"}, "o2"), {op: "u", "o2._id": doc["_id"]}]};
        addToOps(oplog.find(updateQuery));
        addToOps(oplog.find({op: "d", "o._id": doc["_id"]}));
    }

    let compareOps = function (opA, opB) {
        return bsonWoCompare(opA.ts, opB.ts);
    };

    allOps.sort(compareOps);

    jsTest.log.info("Ops found on each shard", {doc});
    for (var i = 0; i < allOps.length; i++) {
        jsTest.log.info({ops: allOps[i]});
    }

    return allOps;
}