Fix sidebar new task button not setting folder

Author: charlesvien

apps/code/src/renderer/features/sidebar/components/TaskListView.tsx

@@ -318,6 +318,8 @@ export function TaskListView({
                   f.remoteUrl?.toLowerCase() === group.id.toLowerCase() ||
                   f.path === group.id,
               );
+              const groupFolderId =
+                folder?.id ?? group.tasks.find((t) => t.folderId)?.folderId;
               return (
                 <DraggableFolder key={group.id} id={group.id} index={index}>
                   <SidebarSection
@@ -335,8 +337,8 @@ export function TaskListView({
                     addSpacingBefore={false}
                     tooltipContent={folder?.path ?? group.id}
                     onNewTask={() => {
-                      if (folder) {
-                        navigateToTaskInput(folder.id);
+                      if (groupFolderId) {
+                        navigateToTaskInput(groupFolderId);
                       } else {
                         navigateToTaskInput();
                       }

apps/code/src/renderer/features/sidebar/hooks/useSidebarData.ts

@@ -25,6 +25,7 @@ export interface TaskData {
   isPinned: boolean;
   needsPermission: boolean;
   repository: TaskRepositoryInfo | null;
+  folderId?: string;
   taskRunStatus?:
     | "started"
     | "in_progress"
@@ -203,6 +204,7 @@ export function useSidebarData({
         isPinned: pinnedTaskIds.has(task.id),
         needsPermission: (session?.pendingPermissions?.size ?? 0) > 0,
         repository: getRepositoryInfo(task, workspace?.folderPath),
+        folderId: workspace?.folderId || undefined,
         taskRunStatus: task.latest_run?.status,
         taskRunEnvironment: task.latest_run?.environment,
       };