fixing some bugs

apps/app/src/components/views/board-view/components/worktree-selector.tsx

@@ -198,13 +198,15 @@ export function WorktreeSelector({
   // Initialize selection to main if not set
   useEffect(() => {
     if (worktrees.length > 0 && currentWorktree === undefined) {
-      setCurrentWorktree(projectPath, null); // null = main worktree
+      const mainWorktree = worktrees.find(w => w.isMain);
+      const mainBranch = mainWorktree?.branch || "main";
+      setCurrentWorktree(projectPath, null, mainBranch); // null = main worktree
     }
   }, [worktrees, currentWorktree, projectPath, setCurrentWorktree]);
 
   const handleSelectWorktree = async (worktree: WorktreeInfo) => {
-    // Simply select the worktree in the UI
-    setCurrentWorktree(projectPath, worktree.isMain ? null : worktree.path);
+    // Simply select the worktree in the UI with both path and branch
+    setCurrentWorktree(projectPath, worktree.isMain ? null : worktree.path, worktree.branch);
   };
 
   const handleStartDevServer = async (worktree: WorktreeInfo) => {
@@ -454,19 +456,20 @@ export function WorktreeSelector({
   };
 
   // The "selected" worktree is based on UI state, not git's current branch
-  // currentWorktree is null for main, or the worktree path for others
-  const selectedWorktree = currentWorktree
-    ? worktrees.find((w) => w.path === currentWorktree)
+  // currentWorktree.path is null for main, or the worktree path for others
+  const currentWorktreePath = currentWorktree?.path ?? null;
+  const selectedWorktree = currentWorktreePath
+    ? worktrees.find((w) => w.path === currentWorktreePath)
     : worktrees.find((w) => w.isMain);
 
 
   // Render a worktree tab with branch selector (for main) and actions dropdown
   const renderWorktreeTab = (worktree: WorktreeInfo) => {
     // Selection is based on UI state, not git's current branch
-    // Default to main selected if currentWorktree is null or undefined
+    // Default to main selected if currentWorktree is null/undefined or path is null
     const isSelected = worktree.isMain
-      ? currentWorktree === null || currentWorktree === undefined
-      : worktree.path === currentWorktree;
+      ? currentWorktree === null || currentWorktree === undefined || currentWorktree.path === null
+      : worktree.path === currentWorktreePath;
 
     const isRunning = hasRunningFeatures(worktree);
 

apps/app/src/components/views/board-view/dialogs/add-feature-dialog.tsx

@@ -69,6 +69,7 @@ interface AddFeatureDialogProps {
   categorySuggestions: string[];
   branchSuggestions: string[];
   defaultSkipTests: boolean;
+  defaultBranch?: string;
   isMaximized: boolean;
   showProfilesOnly: boolean;
   aiProfiles: AIProfile[];
@@ -81,6 +82,7 @@ export function AddFeatureDialog({
   categorySuggestions,
   branchSuggestions,
   defaultSkipTests,
+  defaultBranch = "main",
   isMaximized,
   showProfilesOnly,
   aiProfiles,
@@ -109,15 +111,16 @@ export function AddFeatureDialog({
   // Get enhancement model from store
   const { enhancementModel } = useAppStore();
 
-  // Sync skipTests default when dialog opens
+  // Sync defaults when dialog opens
   useEffect(() => {
     if (open) {
       setNewFeature((prev) => ({
         ...prev,
         skipTests: defaultSkipTests,
+        branchName: defaultBranch,
       }));
     }
-  }, [open, defaultSkipTests]);
+  }, [open, defaultSkipTests, defaultBranch]);
 
   const handleAdd = () => {
     if (!newFeature.description.trim()) {
@@ -155,7 +158,7 @@ export function AddFeatureDialog({
       model: "opus",
       priority: 2,
       thinkingLevel: "none",
-      branchName: "main",
+      branchName: defaultBranch,
     });
     setNewFeaturePreviewMap(new Map());
     setShowAdvancedOptions(false);

apps/app/src/components/views/board-view/dialogs/create-worktree-dialog.tsx

@@ -16,11 +16,16 @@ import { GitBranch, Loader2 } from "lucide-react";
 import { getElectronAPI } from "@/lib/electron";
 import { toast } from "sonner";
 
+interface CreatedWorktreeInfo {
+  path: string;
+  branch: string;
+}
+
 interface CreateWorktreeDialogProps {
   open: boolean;
   onOpenChange: (open: boolean) => void;
   projectPath: string;
-  onCreated: (worktreePath: string) => void;
+  onCreated: (worktree: CreatedWorktreeInfo) => void;
 }
 
 export function CreateWorktreeDialog({
@@ -68,7 +73,7 @@ export function CreateWorktreeDialog({
               : "Using existing branch",
           }
         );
-        onCreated(result.worktree.path);
+        onCreated({ path: result.worktree.path, branch: result.worktree.branch });
         onOpenChange(false);
         setBranchName("");
       } else {

apps/app/src/components/views/board-view/hooks/use-board-column-features.ts

@@ -43,7 +43,12 @@ export function useBoardColumnFeatures({
     // Determine the effective worktree path and branch for filtering
     // If currentWorktreePath is null, we're on the main worktree
     const effectiveWorktreePath = currentWorktreePath || projectPath;
-    const effectiveBranch = currentWorktreeBranch || "main";
+    // Use the branch name from the selected worktree
+    // If we're selecting main (currentWorktreePath is null), currentWorktreeBranch
+    // should contain the main branch's actual name, defaulting to "main"
+    // If we're selecting a non-main worktree but can't find it, currentWorktreeBranch is null
+    // In that case, we can't do branch-based filtering, so we'll handle it specially below
+    const effectiveBranch = currentWorktreeBranch;
 
     filteredFeatures.forEach((f) => {
       // If feature has a running agent, always show it in "in_progress"
@@ -62,6 +67,11 @@ export function useBoardColumnFeatures({
       } else if (f.worktreePath) {
         // Has worktreePath - match by path
         matchesWorktree = f.worktreePath === effectiveWorktreePath;
+      } else if (effectiveBranch === null) {
+        // We're selecting a non-main worktree but can't determine its branch yet
+        // (worktrees haven't loaded). Don't show branch-only features until we know.
+        // This prevents showing wrong features during loading.
+        matchesWorktree = false;
       } else {
         // Has branchName but no worktreePath - match by branch name
         matchesWorktree = featureBranch === effectiveBranch;
@@ -76,10 +86,12 @@ export function useBoardColumnFeatures({
         // Otherwise, use the feature's status (fallback to backlog for unknown statuses)
         const status = f.status as ColumnId;
 
-        // Backlog items are always visible (they have no worktree assigned)
-        // For other statuses, filter by worktree
+        // Filter all items by worktree, including backlog
+        // This ensures backlog items with a branch assigned only show in that branch
         if (status === "backlog") {
-          map.backlog.push(f);
+          if (matchesWorktree) {
+            map.backlog.push(f);
+          }
         } else if (map[status]) {
           // Only show if matches current worktree or has no worktree assigned
           if (matchesWorktree) {