export const durationAggregateByAuthorAndTime = new TableAggregate<{
  Key: [Id<"users">, number]; 
  DataModel: DataModel;
  TableName: "tasks";
}>(components.taskDuration, { 
  sortKey: (doc) => [doc.authorId, doc._creationTime], 
  sumValue: (doc) => doc.duration, 
});

export const durationAggregateByAuthorAndTime = new TableAggregate<{
  Key: [Id<"users">, number]; 
  DataModel: DataModel;
  TableName: "tasks";
}>(components.taskDuration, { 
  sortKey: (doc) => [doc.authorId, doc._creationTime], 
  sumValue: (doc) => doc.duration, 
});

// Same as before, but less efficient for this case than namespacing
export const totalFocusTimeByUserWithKey = query({
  args: {
    userId: v.id("users"),
  },
  handler: async (ctx, args) => {
    const totalFocusTime = await durationAggregateByAuthorAndTime.sum(ctx, {
      // Use 'prefix' to bound the query to keys starting with the userId
      prefix: [args.userId],
    });
    return totalFocusTime;
  },
});

// Same as before, but less efficient for this case than namespacing
export const totalFocusTimeByUserWithKey = query({
  args: {
    userId: v.id("users"),
  },
  handler: async (ctx, args) => {
    const totalFocusTime = await durationAggregateByAuthorAndTime.sum(ctx, {
      // Use 'prefix' to bound the query to keys starting with the userId
      prefix: [args.userId],
    });
    return totalFocusTime;
  },
});

// Another example: Get the total duration across ALL users (system-wide)
export const totalSystemFocusTime = query({
  args: {},
  handler: async (ctx) => {
    // Calling .sum() without bounds/prefix sums across the entire aggregate
    const totalDuration = await durationAggregateByAuthorAndTime.sum(ctx);
    return totalDuration;
  },
});

// Another example: Get the total duration across ALL users (system-wide)
export const totalSystemFocusTime = query({
  args: {},
  handler: async (ctx) => {
    // Calling .sum() without bounds/prefix sums across the entire aggregate
    const totalDuration = await durationAggregateByAuthorAndTime.sum(ctx);
    return totalDuration;
  },
});

export const totalFocusTimeByUser = query({
  args: {
    userId: v.id("users"),
  },
  handler: async (ctx, args) => {
    // Optional: Check if user exists
    await getDocumentOrThrow(ctx, "users", args.userId);

    // Use the 'namespace' option to get the sum for the specific user
    const totalFocusTime = await durationAggregateByUser.sum(ctx, {
      namespace: args.userId,
    });
    return totalFocusTime;
  },
});

export const totalFocusTimeByUser = query({
  args: {
    userId: v.id("users"),
  },
  handler: async (ctx, args) => {
    // Optional: Check if user exists
    await getDocumentOrThrow(ctx, "users", args.userId);

    // Use the 'namespace' option to get the sum for the specific user
    const totalFocusTime = await durationAggregateByUser.sum(ctx, {
      namespace: args.userId,
    });
    return totalFocusTime;
  },
});

export const totalFocusTimeByUserForWeek = query({
  args: {
    userId: v.id("users"),
    weekStartTimestamp: v.number(), // Pass the start timestamp of the week
    weekEndTimestamp: v.number(),   // Pass the end timestamp of the week
  },
  handler: async (ctx, args) => {
    await getDocumentOrThrow(ctx, "users", args.userId);

    // Define the bounds based on _creationTime (the key)
    const bounds = {
        lower: { key: args.weekStartTimestamp, inclusive: true },
        upper: { key: args.weekEndTimestamp, inclusive: false } // exclusive upper bound usually makes sense for time ranges, play around with this
    };

    // Get the sum within the specified namespace and time bounds
    const totalFocusTimeByWeek = await durationAggregateByUser.sum(ctx, {
      namespace: args.userId,
      bounds: bounds,
    });
    return totalFocusTimeByWeek;
  },
});

export const totalFocusTimeByUserForWeek = query({
  args: {
    userId: v.id("users"),
    weekStartTimestamp: v.number(), // Pass the start timestamp of the week
    weekEndTimestamp: v.number(),   // Pass the end timestamp of the week
  },
  handler: async (ctx, args) => {
    await getDocumentOrThrow(ctx, "users", args.userId);

    // Define the bounds based on _creationTime (the key)
    const bounds = {
        lower: { key: args.weekStartTimestamp, inclusive: true },
        upper: { key: args.weekEndTimestamp, inclusive: false } // exclusive upper bound usually makes sense for time ranges, play around with this
    };

    // Get the sum within the specified namespace and time bounds
    const totalFocusTimeByWeek = await durationAggregateByUser.sum(ctx, {
      namespace: args.userId,
      bounds: bounds,
    });
    return totalFocusTimeByWeek;
  },
});

export const durationAggregateByUser = new TableAggregate<{
  Namespace: Id<"tasks">; 
  Key: number; 
  DataModel: DataModel;
  TableName: "tasks";
}>(components.taskDuration, { 
  namespace: (doc) => doc.authorId, 
  sortKey: (doc) => doc._creationTime,
  sumValue: (doc) => doc.duration, 
});

export const durationAggregateByUser = new TableAggregate<{
  Namespace: Id<"tasks">; 
  Key: number; 
  DataModel: DataModel;
  TableName: "tasks";
}>(components.taskDuration, { 
  namespace: (doc) => doc.authorId, 
  sortKey: (doc) => doc._creationTime,
  sumValue: (doc) => doc.duration, 
});

npx convex deploy --cmd 'npm run build' --preview-run 'system/dummy:seedForPreviewDeployments' && npx convex env set VERCEL_URL $VERCEL_URL

npx convex deploy --cmd 'npm run build' --preview-run 'system/dummy:seedForPreviewDeployments' && npx convex env set VERCEL_URL $VERCEL_URL