import { Filter, UpdateFilter } from 'mongodb';

interface Foo {
  optional?: string;
  optionalNested?: {
    path: string;
  };
  required: string;
  requiredNested: {
    path: string;
  };
}

const queryOptionalOK: Filter<Foo> = { optional: "foo", "optionalNested.path": "foo" };
// ^ OK
const queryOptionalFail: Filter<Foo> = { optional: 1, "optionalNested.path": 1 };
// ^ Fail - TS flags only `optional`, but `optionalNested.path` should also be flagged
const queryRequiredOK: Filter<Foo> = { required: "foo", "requiredNested.path": "foo" };
// ^ OK
const queryRequiredFail: Filter<Foo> = { required: 1, "requiredNested.path": 1 };
// ^ OK - Both fields are flagged

const updateOptionalOK: UpdateFilter<Foo> = { $set: { optional: "foo", "optionalNested.path": "foo" } };
// ^ OK - Expected
const updateOptionalFail: UpdateFilter<Foo> = { $set: { optional: 1, "optionalNested.path": 1 } };
// ^ Fail - TS flags only `optional`, but `optionalNested.path` should also be flagged
const updateRequiredOK: UpdateFilter<Foo> = { $set: { required: "foo", "requiredNested.path": "foo" } };
// ^ OK - Expected
const updateRequiredFail: UpdateFilter<Foo> = { $set: { required: 1, "requiredNested.path": 1 } };
// ^ OK - Both fields are flagged