import { query } from "../_generated/server";
import { customQuery, customCtx } from "convex-helpers/server/customFunctions";

// Use `userQuery` instead of `query` to add this behavior.
const userQuery = customQuery(
  query, // The base function we're extending
  // Here we're using a `customCtx` helper because our modification
  // only modifies the `ctx` argument to the function.
  customCtx(async (ctx) => {
    // Look up the logged in user
    const identity = await ctx.auth.getUserIdentity();
    if (!identity) throw new Error("Unauthorized");
    // Pass in a user to use in evaluating rules,
    // which validate data access at access / write time.
    const db = wrapDatabaseReader({ identity }, ctx.db, rules);
    // This new ctx will be applied to the function's.
    // The user is a new field, the db replaces ctx.db
    return { identity, db };
  }),
);

import { query } from "../_generated/server";
import { customQuery, customCtx } from "convex-helpers/server/customFunctions";

// Use `userQuery` instead of `query` to add this behavior.
const userQuery = customQuery(
  query, // The base function we're extending
  // Here we're using a `customCtx` helper because our modification
  // only modifies the `ctx` argument to the function.
  customCtx(async (ctx) => {
    // Look up the logged in user
    const identity = await ctx.auth.getUserIdentity();
    if (!identity) throw new Error("Unauthorized");
    // Pass in a user to use in evaluating rules,
    // which validate data access at access / write time.
    const db = wrapDatabaseReader({ identity }, ctx.db, rules);
    // This new ctx will be applied to the function's.
    // The user is a new field, the db replaces ctx.db
    return { identity, db };
  }),
);

export const getByIdInfo = query({
  args: {
    diagramId: v.optional(v.id("diagrams")),
  },
  handler: async (ctx, args) => {
    const identity = await ctx.auth.getUserIdentity();

    if (!args.diagramId) {
      return; // No diagramId provided
    }

    const document = await ctx.db.get(args.diagramId);

    if (!document) {
      throw new Error("Document not found or unauthorized access");
    }

    if (!identity || document.userId !== identity.subject) {
      throw new Error("Unauthorized access");
    }

    return { ...document }; // Return the document information
  },
});

export const getByIdInfo = query({
  args: {
    diagramId: v.optional(v.id("diagrams")),
  },
  handler: async (ctx, args) => {
    const identity = await ctx.auth.getUserIdentity();

    if (!args.diagramId) {
      return; // No diagramId provided
    }

    const document = await ctx.db.get(args.diagramId);

    if (!document) {
      throw new Error("Document not found or unauthorized access");
    }

    if (!identity || document.userId !== identity.subject) {
      throw new Error("Unauthorized access");
    }

    return { ...document }; // Return the document information
  },
});