export const dbRules: Rules<AuthQueryCtx | AuthMutationCtx, DataModel> = {
  firestoreEdits: {
    insert: async (ctx, edit) => {
      const role = await _getExistingRoleFromUserAndOrgWorkspaceIds(ctx, {
        userId: ctx.user._id,
        workspaceId: ctx.workspace._id,
      });
      console.log("role", role);

      if (role?.role === "admin") {
        return true;
      }
      return false;
    },
...

export const dbRules: Rules<AuthQueryCtx | AuthMutationCtx, DataModel> = {
  firestoreEdits: {
    insert: async (ctx, edit) => {
      const role = await _getExistingRoleFromUserAndOrgWorkspaceIds(ctx, {
        userId: ctx.user._id,
        workspaceId: ctx.workspace._id,
      });
      console.log("role", role);

      if (role?.role === "admin") {
        return true;
      }
      return false;
    },
...

export const zAuthenticatedMutation = zCustomMutation(
  mutationWithTriggers, // The base function we're extending
  {
    args: {
      // This type makes sure we can't just forget to pass the clerkOrgId
      // clerkOrgId: v.union(v.string(), v.null()),
      clerkOrgId: v.string(),
      userTokenIdentifier: v.optional(v.string()),
    },
    input: async (ctx, args) => {
      const { user, workspace } = await enhanceClientAuthContext(
        ctx,
        args.clerkOrgId,
        args.userTokenIdentifier
      );

      // RLS wrapper
      const safeDB: DatabaseWriter = wrapDatabaseWriter(
        { user, workspace, ...ctx },
        ctx.db,
        dbRules
      );

      return { ctx: { user, workspace }, db: safeDB, args: { ...args } };
    },
  }
);

export const zAuthenticatedMutation = zCustomMutation(
  mutationWithTriggers, // The base function we're extending
  {
    args: {
      // This type makes sure we can't just forget to pass the clerkOrgId
      // clerkOrgId: v.union(v.string(), v.null()),
      clerkOrgId: v.string(),
      userTokenIdentifier: v.optional(v.string()),
    },
    input: async (ctx, args) => {
      const { user, workspace } = await enhanceClientAuthContext(
        ctx,
        args.clerkOrgId,
        args.userTokenIdentifier
      );

      // RLS wrapper
      const safeDB: DatabaseWriter = wrapDatabaseWriter(
        { user, workspace, ...ctx },
        ctx.db,
        dbRules
      );

      return { ctx: { user, workspace }, db: safeDB, args: { ...args } };
    },
  }
);

export const addEdit = zAuthenticatedMutation({
  args: {
    edit: zAddEditArgs,
  },
  handler: async (ctx, args) => {
...

export const addEdit = zAuthenticatedMutation({
  args: {
    edit: zAddEditArgs,
  },
  handler: async (ctx, args) => {
...

triggers.register("blocks", async (ctx, change) => {
  if (change.operation === "insert") {
    const newBlockFid = parseFid(change.newDoc.fid, "blocks");
    // Try to fetch block with same fid - which will throw if there's multiple. Note that at the time of running this query the new doc is returned by the query
    await _getBlockByFid(ctx, {
      blockFid: newBlockFid,
    });
  } else if (change.operation === "update") {
    // Throw an error if the fid is being updated
    if (change.oldDoc.fid !== change.newDoc.fid) {
      throw new ConvexError("Block FID cannot be updated");
    }
  }
});

triggers.register("blocks", async (ctx, change) => {
  if (change.operation === "insert") {
    const newBlockFid = parseFid(change.newDoc.fid, "blocks");
    // Try to fetch block with same fid - which will throw if there's multiple. Note that at the time of running this query the new doc is returned by the query
    await _getBlockByFid(ctx, {
      blockFid: newBlockFid,
    });
  } else if (change.operation === "update") {
    // Throw an error if the fid is being updated
    if (change.oldDoc.fid !== change.newDoc.fid) {
      throw new ConvexError("Block FID cannot be updated");
    }
  }
});

Error fetching POST  https://sensible-salmon-524.convex.cloud/api/push_config 400 Bad Request: InvalidModules: Hit an error while pushing:
Loading the pushed modules encountered the following
    error:
Function execution timed out (maximum duration: 1s)

Error fetching POST  https://sensible-salmon-524.convex.cloud/api/push_config 400 Bad Request: InvalidModules: Hit an error while pushing:
Loading the pushed modules encountered the following
    error:
Function execution timed out (maximum duration: 1s)

// Helper function to prefix object keys for the Convex validator
function prefixValidatorKeys<T extends Record<string, any>>(obj: T, prefix: z.infer<typeof zBlockType>) {
  return Object.fromEntries(
    Object.entries(obj).map(([key, value]) => [`${prefix}_${key}`, value])
  );
}

// Create the Geopoint block properties validator
export const vGeopointBlockProperties = v.object({
  ...vCommonProps.fields,
  ...prefixValidatorKeys(vGeopointSpecificProps.fields, "geopoint"),
});

// Create a type that includes both common and prefixed specific properties
type PrefixProperties<T, P extends z.infer<typeof zBlockType>> = {
  [K in keyof T as `${P}_${string & K}`]: T[K]
};

export type GeopointBlockProperties = 
  Infer<typeof vCommonProps> & 
  PrefixProperties<Infer<typeof vGeopointSpecificProps>, 'geopoint'>;

const testGeopointBlock: GeopointBlockProperties = {
  gridLayout: { x: 0, y: 0 },
  geopoint_mapType: "roadmap",
  geopoint_zoom: 10,
};

// Helper function to prefix object keys for the Convex validator
function prefixValidatorKeys<T extends Record<string, any>>(obj: T, prefix: z.infer<typeof zBlockType>) {
  return Object.fromEntries(
    Object.entries(obj).map(([key, value]) => [`${prefix}_${key}`, value])
  );
}

// Create the Geopoint block properties validator
export const vGeopointBlockProperties = v.object({
  ...vCommonProps.fields,
  ...prefixValidatorKeys(vGeopointSpecificProps.fields, "geopoint"),
});

// Create a type that includes both common and prefixed specific properties
type PrefixProperties<T, P extends z.infer<typeof zBlockType>> = {
  [K in keyof T as `${P}_${string & K}`]: T[K]
};

export type GeopointBlockProperties = 
  Infer<typeof vCommonProps> & 
  PrefixProperties<Infer<typeof vGeopointSpecificProps>, 'geopoint'>;

const testGeopointBlock: GeopointBlockProperties = {
  gridLayout: { x: 0, y: 0 },
  geopoint_mapType: "roadmap",
  geopoint_zoom: 10,
};

export function useAuthMutation<Mutation extends AuthFunction<"mutation">>(
  name: Mutation
) {
  const organization = useOrganization();
  const clerkOrgId = organization?.organization?.id;
  const originalMutation = useMutation(name);

  return useCallback(
    async (
      ...args: AuthArgsArray<Mutation>
    ): Promise<FunctionReturnType<Mutation>> => {
      const newArgs = {
        ...(args[0] ?? {}),
        clerkOrgId,
      } as FunctionArgs<Mutation>;

      return originalMutation(...([newArgs] as OptionalRestArgs<Mutation>));
    },
    [clerkOrgId, originalMutation]
  );
}

export function useAuthMutation<Mutation extends AuthFunction<"mutation">>(
  name: Mutation
) {
  const organization = useOrganization();
  const clerkOrgId = organization?.organization?.id;
  const originalMutation = useMutation(name);

  return useCallback(
    async (
      ...args: AuthArgsArray<Mutation>
    ): Promise<FunctionReturnType<Mutation>> => {
      const newArgs = {
        ...(args[0] ?? {}),
        clerkOrgId,
      } as FunctionArgs<Mutation>;

      return originalMutation(...([newArgs] as OptionalRestArgs<Mutation>));
    },
    [clerkOrgId, originalMutation]
  );
}