export const getParts = query({
  args: {
    locationId: v.id("locations"),
    factoryId: v.id("factories"),
    machineClassId: v.id("machineClasses"),
    name: v.string(),
  },
  async handler(ctx, args) {
    // Required filter by locationId
    let partsQuery = ctx.db
      .query("parts")
      .withIndex("by_locationId", (q) => q.eq("locationId", args.locationId));

    // Optional filter by machineClassId
    if (args.machineClassId) {
      partsQuery = partsQuery.filter((q) =>
        q.eq(q.field("machineClassId"), args.machineClassId)
      );
    }

    // Optional filter by factoryId
    if (args.factoryId) {
      partsQuery = partsQuery.filter((q) =>
        q.eq(q.field("factoryId"), args.factoryId)
      );
    }

    // Optional search by name
    if (args.name) {
      partsQuery = partsQuery
        .withSearchIndex("by_name", (q) => q.match("name", args.name), {
          searchField: "name",
        });
    }

    const parts = await partsQuery.collect();

    return parts;
  },
});

export const getParts = query({
  args: {
    locationId: v.id("locations"),
    factoryId: v.id("factories"),
    machineClassId: v.id("machineClasses"),
    name: v.string(),
  },
  async handler(ctx, args) {
    // Required filter by locationId
    let partsQuery = ctx.db
      .query("parts")
      .withIndex("by_locationId", (q) => q.eq("locationId", args.locationId));

    // Optional filter by machineClassId
    if (args.machineClassId) {
      partsQuery = partsQuery.filter((q) =>
        q.eq(q.field("machineClassId"), args.machineClassId)
      );
    }

    // Optional filter by factoryId
    if (args.factoryId) {
      partsQuery = partsQuery.filter((q) =>
        q.eq(q.field("factoryId"), args.factoryId)
      );
    }

    // Optional search by name
    if (args.name) {
      partsQuery = partsQuery
        .withSearchIndex("by_name", (q) => q.match("name", args.name), {
          searchField: "name",
        });
    }

    const parts = await partsQuery.collect();

    return parts;
  },
});