Add habit checklist feature

src/Orbit.Api/Controllers/HabitsController.cs

@@ -5,6 +5,7 @@
 using Orbit.Application.Habits.Commands;
 using Orbit.Application.Habits.Queries;
 using Orbit.Domain.Enums;
+using Orbit.Domain.ValueObjects;
 
 namespace Orbit.Api.Controllers;
 
@@ -26,7 +27,8 @@ public record CreateHabitRequest(
         bool ReminderEnabled = false,
         int ReminderMinutesBefore = 15,
         bool SlipAlertEnabled = false,
-        IReadOnlyList<Guid>? TagIds = null);
+        IReadOnlyList<Guid>? TagIds = null,
+        IReadOnlyList<ChecklistItem>? ChecklistItems = null);
 
     public record UpdateHabitRequest(
         string Title,
@@ -39,7 +41,10 @@ public record UpdateHabitRequest(
         TimeOnly? DueTime = null,
         bool? ReminderEnabled = null,
         int? ReminderMinutesBefore = null,
-        bool? SlipAlertEnabled = null);
+        bool? SlipAlertEnabled = null,
+        IReadOnlyList<ChecklistItem>? ChecklistItems = null);
+
+    public record UpdateChecklistRequest(IReadOnlyList<ChecklistItem> ChecklistItems);
 
     public record LogHabitRequest(string? Note = null);
 
@@ -139,7 +144,8 @@ public async Task<IActionResult> CreateHabit(
             request.ReminderEnabled,
             request.ReminderMinutesBefore,
             request.SlipAlertEnabled,
-            request.TagIds);
+            request.TagIds,
+            request.ChecklistItems);
 
         var result = await mediator.Send(command, cancellationToken);
 
@@ -189,7 +195,26 @@ public async Task<IActionResult> UpdateHabit(
             request.DueTime,
             request.ReminderEnabled,
             request.ReminderMinutesBefore,
-            request.SlipAlertEnabled);
+            request.SlipAlertEnabled,
+            request.ChecklistItems);
+
+        var result = await mediator.Send(command, cancellationToken);
+
+        return result.IsSuccess
+            ? NoContent()
+            : BadRequest(new { error = result.Error });
+    }
+
+    [HttpPut("{id:guid}/checklist")]
+    public async Task<IActionResult> UpdateChecklist(
+        Guid id,
+        [FromBody] UpdateChecklistRequest request,
+        CancellationToken cancellationToken)
+    {
+        var command = new UpdateChecklistCommand(
+            HttpContext.GetUserId(),
+            id,
+            request.ChecklistItems);
 
         var result = await mediator.Send(command, cancellationToken);
 

src/Orbit.Application/Chat/Commands/ProcessUserChatCommand.cs

@@ -7,6 +7,7 @@
 using Orbit.Domain.Enums;
 using Orbit.Domain.Interfaces;
 using Orbit.Domain.Models;
+using Orbit.Domain.ValueObjects;
 
 namespace Orbit.Application.Chat.Commands;
 
@@ -386,7 +387,8 @@ public async Task<Result<ChatResponse>> Handle(
             isBadHabit: isBadHabit,
             dueDate: dueDate,
             dueTime: action.DueTime,
-            slipAlertEnabled: slipAlertEnabled);
+            slipAlertEnabled: slipAlertEnabled,
+            checklistItems: action.ChecklistItems);
 
         if (habitResult.IsFailure)
             return Result.Failure<(Guid? Id, string? Name)>(habitResult.Error);
@@ -483,7 +485,8 @@ public async Task<Result<ChatResponse>> Handle(
             action.Days,
             action.IsBadHabit ?? habit.IsBadHabit,
             action.DueDate ?? habit.DueDate,
-            dueTime: action.DueTime);
+            dueTime: action.DueTime,
+            checklistItems: action.ChecklistItems);
 
         if (result.IsFailure)
             return Result.Failure<(Guid? Id, string? Name)>(result.Error);

src/Orbit.Application/Habits/Commands/CreateHabitCommand.cs

@@ -5,6 +5,7 @@
 using Orbit.Domain.Entities;
 using Orbit.Domain.Enums;
 using Orbit.Domain.Interfaces;
+using Orbit.Domain.ValueObjects;
 
 namespace Orbit.Application.Habits.Commands;
 
@@ -22,7 +23,8 @@ public record CreateHabitCommand(
     bool ReminderEnabled = false,
     int ReminderMinutesBefore = 15,
     bool SlipAlertEnabled = false,
-    IReadOnlyList<Guid>? TagIds = null) : IRequest<Result<Guid>>;
+    IReadOnlyList<Guid>? TagIds = null,
+    IReadOnlyList<ChecklistItem>? ChecklistItems = null) : IRequest<Result<Guid>>;
 
 public class CreateHabitCommandHandler(
     IGenericRepository<Habit> habitRepository,
@@ -61,7 +63,8 @@ public async Task<Result<Guid>> Handle(CreateHabitCommand request, CancellationT
             dueTime: request.DueTime,
             reminderEnabled: request.ReminderEnabled,
             reminderMinutesBefore: request.ReminderMinutesBefore,
-            slipAlertEnabled: request.SlipAlertEnabled);
+            slipAlertEnabled: request.SlipAlertEnabled,
+            checklistItems: request.ChecklistItems);
 
         if (habitResult.IsFailure)
             return Result.Failure<Guid>(habitResult.Error);

src/Orbit.Application/Habits/Commands/DuplicateHabitCommand.cs

@@ -92,6 +92,7 @@ private static Result<Habit> CloneHabit(Habit source, Guid? parentHabitId, bool
             source.IsBadHabit,
             source.DueDate,
             source.DueTime,
-            parentHabitId);
+            parentHabitId,
+            checklistItems: source.ChecklistItems);
     }
 }

src/Orbit.Application/Habits/Commands/UpdateChecklistCommand.cs

@@ -0,0 +1,34 @@
+using MediatR;
+using Orbit.Application.Common;
+using Orbit.Domain.Common;
+using Orbit.Domain.Entities;
+using Orbit.Domain.Interfaces;
+using Orbit.Domain.ValueObjects;
+
+namespace Orbit.Application.Habits.Commands;
+
+public record UpdateChecklistCommand(
+    Guid UserId,
+    Guid HabitId,
+    IReadOnlyList<ChecklistItem> ChecklistItems) : IRequest<Result>;
+
+public class UpdateChecklistCommandHandler(
+    IGenericRepository<Habit> habitRepository,
+    IUnitOfWork unitOfWork) : IRequestHandler<UpdateChecklistCommand, Result>
+{
+    public async Task<Result> Handle(UpdateChecklistCommand request, CancellationToken cancellationToken)
+    {
+        var habit = await habitRepository.FindOneTrackedAsync(
+            h => h.Id == request.HabitId && h.UserId == request.UserId,
+            cancellationToken: cancellationToken);
+
+        if (habit is null)
+            return Result.Failure(ErrorMessages.HabitNotFound);
+
+        habit.UpdateChecklist(request.ChecklistItems);
+
+        await unitOfWork.SaveChangesAsync(cancellationToken);
+
+        return Result.Success();
+    }
+}

src/Orbit.Application/Habits/Commands/UpdateHabitCommand.cs

@@ -5,6 +5,7 @@
 using Orbit.Domain.Entities;
 using Orbit.Domain.Enums;
 using Orbit.Domain.Interfaces;
+using Orbit.Domain.ValueObjects;
 using System.Linq.Expressions;
 
 namespace Orbit.Application.Habits.Commands;
@@ -22,7 +23,8 @@ public record UpdateHabitCommand(
     TimeOnly? DueTime = null,
     bool? ReminderEnabled = null,
     int? ReminderMinutesBefore = null,
-    bool? SlipAlertEnabled = null) : IRequest<Result>;
+    bool? SlipAlertEnabled = null,
+    IReadOnlyList<ChecklistItem>? ChecklistItems = null) : IRequest<Result>;
 
 public class UpdateHabitCommandHandler(
     IGenericRepository<Habit> habitRepository,
@@ -51,7 +53,8 @@ public async Task<Result> Handle(UpdateHabitCommand request, CancellationToken c
             dueTime: request.DueTime,
             reminderEnabled: request.ReminderEnabled,
             reminderMinutesBefore: request.ReminderMinutesBefore,
-            slipAlertEnabled: request.SlipAlertEnabled);
+            slipAlertEnabled: request.SlipAlertEnabled,
+            checklistItems: request.ChecklistItems);
 
         if (result.IsFailure)
             return result;

src/Orbit.Application/Habits/Queries/GetHabitByIdQuery.cs

@@ -3,6 +3,7 @@
 using Orbit.Domain.Common;
 using Orbit.Domain.Entities;
 using Orbit.Domain.Interfaces;
+using Orbit.Domain.ValueObjects;
 
 namespace Orbit.Application.Habits.Queries;
 
@@ -18,6 +19,7 @@ public record HabitDetailResponse(
     TimeOnly? DueTime,
     IReadOnlyList<DayOfWeek> Days,
     int? Position,
+    IReadOnlyList<ChecklistItem> ChecklistItems,
     DateTime CreatedAtUtc,
     IReadOnlyList<HabitChildResponse> Children);
 
@@ -56,6 +58,7 @@ public async Task<Result<HabitDetailResponse>> Handle(GetHabitByIdQuery request,
             habit.DueTime,
             habit.Days.ToList(),
             habit.Position,
+            habit.ChecklistItems,
             habit.CreatedAtUtc,
             children));
     }
@@ -64,7 +67,7 @@ public async Task<Result<HabitDetailResponse>> Handle(GetHabitByIdQuery request,
         c.Id, c.Title, c.Description,
         c.FrequencyUnit, c.FrequencyQuantity, c.IsBadHabit, c.IsCompleted,
         c.Days.ToList(), c.DueDate, c.DueTime,
-        c.Position, MapChildren(c.Id, lookup));
+        c.Position, c.ChecklistItems, MapChildren(c.Id, lookup));
 
     private static List<HabitChildResponse> MapChildren(Guid parentId, ILookup<Guid?, Habit> lookup) =>
         lookup[parentId]

src/Orbit.Application/Habits/Queries/GetHabitScheduleQuery.cs

@@ -5,6 +5,7 @@
 using Orbit.Domain.Entities;
 using Orbit.Domain.Enums;
 using Orbit.Domain.Interfaces;
+using Orbit.Domain.ValueObjects;
 
 namespace Orbit.Application.Habits.Queries;
 
@@ -28,6 +29,7 @@ public record HabitScheduleItem(
     bool ReminderEnabled,
     int ReminderMinutesBefore,
     bool SlipAlertEnabled,
+    IReadOnlyList<ChecklistItem> ChecklistItems,
     IReadOnlyList<HabitTagItem> Tags,
     IReadOnlyList<HabitScheduleChildItem> Children);
 
@@ -43,6 +45,7 @@ public record HabitScheduleChildItem(
     DateOnly DueDate,
     TimeOnly? DueTime,
     int? Position,
+    IReadOnlyList<ChecklistItem> ChecklistItems,
     IReadOnlyList<HabitTagItem> Tags,
     IReadOnlyList<HabitScheduleChildItem> Children);
 
@@ -180,6 +183,7 @@ private static HabitScheduleItem MapToScheduleItem(
             h.ReminderEnabled,
             h.ReminderMinutesBefore,
             h.SlipAlertEnabled,
+            h.ChecklistItems,
             MapTags(h),
             MapChildren(h.Id, lookup, dateFrom, dateTo));
 
@@ -209,7 +213,7 @@ private static List<HabitScheduleChildItem> MapChildren(Guid parentId, ILookup<G
                 c.Id, c.Title, c.Description,
                 c.FrequencyUnit, c.FrequencyQuantity, c.IsBadHabit, c.IsCompleted,
                 c.Days.ToList(), c.DueDate, c.DueTime,
-                c.Position, MapTags(c), MapChildren(c.Id, lookup, dateFrom, dateTo)))
+                c.Position, c.ChecklistItems, MapTags(c), MapChildren(c.Id, lookup, dateFrom, dateTo)))
             .ToList();
 
     private static List<HabitTagItem> MapTags(Habit h) =>

src/Orbit.Application/Habits/Queries/GetHabitsQuery.cs

@@ -2,6 +2,7 @@
 using Orbit.Domain.Entities;
 using Orbit.Domain.Enums;
 using Orbit.Domain.Interfaces;
+using Orbit.Domain.ValueObjects;
 
 namespace Orbit.Application.Habits.Queries;
 
@@ -17,6 +18,7 @@ public record HabitResponse(
     TimeOnly? DueTime,
     IReadOnlyList<DayOfWeek> Days,
     int? Position,
+    IReadOnlyList<ChecklistItem> ChecklistItems,
     DateTime CreatedAtUtc,
     IReadOnlyList<HabitChildResponse> Children);
 
@@ -32,6 +34,7 @@ public record HabitChildResponse(
     DateOnly DueDate,
     TimeOnly? DueTime,
     int? Position,
+    IReadOnlyList<ChecklistItem> ChecklistItems,
     IReadOnlyList<HabitChildResponse> Children);
 
 public record GetHabitsQuery(
@@ -100,6 +103,7 @@ public async Task<IReadOnlyList<HabitResponse>> Handle(GetHabitsQuery request, C
         h.DueTime,
         h.Days.ToList(),
         h.Position,
+        h.ChecklistItems,
         h.CreatedAtUtc,
         MapChildren(h.Id, lookup));
 
@@ -111,6 +115,6 @@ private static List<HabitChildResponse> MapChildren(Guid parentId, ILookup<Guid?
                 c.Id, c.Title, c.Description,
                 c.FrequencyUnit, c.FrequencyQuantity, c.IsBadHabit, c.IsCompleted,
                 c.Days.ToList(), c.DueDate, c.DueTime,
-                c.Position, MapChildren(c.Id, lookup)))
+                c.Position, c.ChecklistItems, MapChildren(c.Id, lookup)))
             .ToList();
 }

src/Orbit.Domain/Entities/Habit.cs

@@ -1,5 +1,6 @@
 using Orbit.Domain.Common;
 using Orbit.Domain.Enums;
+using Orbit.Domain.ValueObjects;
 
 namespace Orbit.Domain.Entities;
 
@@ -18,6 +19,7 @@ public class Habit : Entity
     public bool ReminderEnabled { get; private set; }
     public int ReminderMinutesBefore { get; private set; } = 15;
     public bool SlipAlertEnabled { get; private set; }
+    public IReadOnlyList<ChecklistItem> ChecklistItems { get; private set; } = [];
     public int? Position { get; private set; }
     public DateTime CreatedAtUtc { get; private set; }
     public ICollection<System.DayOfWeek> Days { get; private set; } = [];
@@ -48,7 +50,8 @@ public static Result<Habit> Create(
         Guid? parentHabitId = null,
         bool reminderEnabled = false,
         int reminderMinutesBefore = 15,
-        bool slipAlertEnabled = false)
+        bool slipAlertEnabled = false,
+        IReadOnlyList<ChecklistItem>? checklistItems = null)
     {
         if (userId == Guid.Empty)
             return Result.Failure<Habit>("User ID is required.");
@@ -77,6 +80,7 @@ public static Result<Habit> Create(
             ReminderEnabled = reminderEnabled,
             ReminderMinutesBefore = reminderMinutesBefore,
             SlipAlertEnabled = slipAlertEnabled,
+            ChecklistItems = checklistItems ?? [],
             CreatedAtUtc = DateTime.UtcNow
         });
     }
@@ -104,6 +108,10 @@ public Result<HabitLog> Log(DateOnly date, string? note = null)
         {
             // Recurring habit: advance DueDate past today
             AdvanceDueDate(date);
+
+            // Reset checklist for next occurrence
+            if (ChecklistItems.Count > 0)
+                ChecklistItems = ChecklistItems.Select(i => i with { IsChecked = false }).ToList();
         }
 
         return Result.Success(log);
@@ -170,7 +178,8 @@ public Result Update(
         TimeOnly? dueTime = null,
         bool? reminderEnabled = null,
         int? reminderMinutesBefore = null,
-        bool? slipAlertEnabled = null)
+        bool? slipAlertEnabled = null,
+        IReadOnlyList<ChecklistItem>? checklistItems = null)
     {
         if (string.IsNullOrWhiteSpace(title))
             return Result.Failure("Title is required.");
@@ -199,10 +208,14 @@ public Result Update(
             ReminderMinutesBefore = reminderMinutesBefore.Value;
         if (slipAlertEnabled.HasValue)
             SlipAlertEnabled = slipAlertEnabled.Value;
+        if (checklistItems is not null)
+            ChecklistItems = checklistItems;
 
         return Result.Success();
     }
 
+    public void UpdateChecklist(IReadOnlyList<ChecklistItem> items) => ChecklistItems = items;
+
     public void SetPosition(int? position) => Position = position;
 
     public void SetParentHabitId(Guid? parentHabitId) => ParentHabitId = parentHabitId;

src/Orbit.Domain/Models/AiAction.cs

@@ -1,4 +1,5 @@
 using Orbit.Domain.Enums;
+using Orbit.Domain.ValueObjects;
 
 namespace Orbit.Domain.Models;
 
@@ -19,4 +20,5 @@ public record AiAction
     public List<AiAction>? SubHabits { get; init; }
     public List<AiAction>? SuggestedSubHabits { get; init; }
     public List<string>? TagNames { get; init; }
+    public List<ChecklistItem>? ChecklistItems { get; init; }
 }

src/Orbit.Domain/ValueObjects/ChecklistItem.cs

@@ -0,0 +1,3 @@
+namespace Orbit.Domain.ValueObjects;
+
+public record ChecklistItem(string Text, bool IsChecked);