@brimveyn/aimux-config 0.6.8 → 0.6.10

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@brimveyn/aimux-config",
-  "version": "0.6.8",
+  "version": "0.6.10",
   "description": "TypeScript configuration API for aimux — keymaps, themes, and backends with a fluent builder.",
   "keywords": [
     "aimux",

package/src/actions.ts

@@ -195,7 +195,7 @@ export const launchSelectedAssistant: ActionFn = (ctx: ModeContext) => {
     ctx.state.modal.step === 'worktree' &&
     ctx.state.modal.createWorktree
   ) {
-    return r([{ type: 'enter-new-tab-worktree-create' }])
+    return r([{ type: 'enter-new-tab-worktree-create' }], [{ type: 'load-new-tab-base-branches' }])
   }
   return r([], [{ type: 'launch-selected-assistant' }])
 }
@@ -210,19 +210,49 @@ export const deleteSelectedWorktree: ActionFn = (ctx: ModeContext) => {
   const session = ctx.state.sessions.find((entry) => entry.id === sessionId)
   const worktree = session?.worktrees?.[modal.selectedIndex]
   if (!worktree) return null
+  // First attempt is never forced — a recoverable failure opens the confirm
+  // dialog (see set-new-tab-worktree-delete-prompt), where confirmDeleteWorktree
+  // retries with force.
   return r(
-    [
-      { index: modal.selectedIndex, type: 'set-modal-selection-index' },
-      { message: null, type: 'set-new-tab-worktree-delete-state' },
-    ],
+    [{ index: modal.selectedIndex, type: 'set-modal-selection-index' }],
+    [{ force: false, sessionId, type: 'delete-worktree', worktreeId: worktree.id }]
+  )
+}
+
+export const confirmDeleteWorktree: ActionFn = (ctx: ModeContext) => {
+  const modal = ctx.state.modal
+  const sessionId = ctx.state.currentSessionId
+  if (modal.type !== 'new-tab' || modal.worktreeDeletePrompt == null) return null
+  if (!(sessionId != null && sessionId !== '')) return null
+  const { worktreeId } = modal.worktreeDeletePrompt
+  return r(
+    [{ prompt: null, type: 'set-new-tab-worktree-delete-prompt' }],
+    [{ force: true, sessionId, type: 'delete-worktree', worktreeId }],
+    'modal.new-tab.command-edit'
+  )
+}
+
+export const cancelDeleteWorktree: KeyResult = r(
+  [{ prompt: null, type: 'set-new-tab-worktree-delete-prompt' }],
+  [],
+  'modal.new-tab.command-edit'
+)
+
+export const confirmWorktreeDeleteModal: ActionFn = (ctx: ModeContext) => {
+  const modal = ctx.state.modal
+  if (modal.type !== 'worktree-delete-confirm') return null
+  return r(
+    [{ type: 'close-modal' }],
     [
       {
-        force: modal.worktreeDeleteConfirmId === worktree.id,
-        sessionId,
+        closeTabs: modal.closeTabs,
+        force: modal.force,
+        sessionId: modal.sessionId,
         type: 'delete-worktree',
-        worktreeId: worktree.id,
+        worktreeId: modal.worktreeId,
       },
-    ]
+    ],
+    'navigation'
   )
 }
 

package/src/defaults.ts

@@ -204,6 +204,17 @@ export function getDefaultKeymapConfig(): ResolvedKeymapConfig {
         .map('<CR>', actions.confirmWorktreeMove, 'Move')
     )
 
+    // -----------------------------------------------------------------------
+    // Modal: standalone worktree delete confirmation (sidebar "Remove worktree")
+    // -----------------------------------------------------------------------
+    .mode('modal.worktree-delete-confirm', (m) =>
+      m
+        .map('<Esc>', actions.closeModal, 'Cancel')
+        .map('n', actions.closeModal, 'Cancel')
+        .map('<CR>', actions.confirmWorktreeDeleteModal, 'Delete worktree')
+        .map('y', actions.confirmWorktreeDeleteModal, 'Delete worktree')
+    )
+
     // -----------------------------------------------------------------------
     // Modal: rename-tab
     // -----------------------------------------------------------------------
@@ -239,6 +250,17 @@ export function getDefaultKeymapConfig(): ResolvedKeymapConfig {
         .passthrough()
     )
 
+    // -----------------------------------------------------------------------
+    // Modal: new-tab worktree delete confirmation dialog
+    // -----------------------------------------------------------------------
+    .mode('modal.new-tab.worktree-delete-confirm', (m) =>
+      m
+        .map('<Esc>', actions.cancelDeleteWorktree, 'Cancel')
+        .map('n', actions.cancelDeleteWorktree, 'Cancel')
+        .map('<CR>', actions.confirmDeleteWorktree, 'Delete worktree')
+        .map('y', actions.confirmDeleteWorktree, 'Delete worktree')
+    )
+
     // -----------------------------------------------------------------------
     // Modal: session-picker (always in filter mode via Picker)
     // -----------------------------------------------------------------------

package/src/index.ts

@@ -87,4 +87,7 @@ export type {
   StatusBarConfig,
   StatusBarSeparator,
   TabSession,
+  WorktreeTemplate,
+  WorktreeTemplatePane,
+  WorktreeTemplateTab,
 } from './types'

package/src/resolver.ts

@@ -48,6 +48,7 @@ export function resolveConfig(userConfig: AimuxUserConfig): ResolvedConfig {
     snippetTriggerChar: resolveSnippetTriggerChar(userConfig.snippetTriggerChar),
     statusBar: userConfig.statusBar ?? {},
     theme: resolveTheme(userConfig.theme),
+    worktreeTemplates: userConfig.worktreeTemplates ?? [],
   }
 }
 

package/src/types.ts

@@ -14,6 +14,8 @@ export type ModeId =
   | 'git-mode'
   | 'modal.new-tab.command-edit'
   | 'modal.new-tab.editing-command'
+  | 'modal.new-tab.worktree-delete-confirm'
+  | 'modal.worktree-delete-confirm'
   | 'modal.session-picker.filtering'
   | 'modal.session-name'
   | 'modal.create-session'
@@ -124,6 +126,13 @@ export interface WorkspaceSnapshotV1 {
   layoutTree?: LayoutNode
   layoutTrees?: Record<string, LayoutNode>
   tabGroupMap?: Record<string, string>
+  /**
+   * Last viewed tab per worktree, keyed by worktree id. Optional and additive
+   * (no version bump): older builds ignore it, newer builds tolerate its
+   * absence. Written now so the data accrues, but restoring it at startup is
+   * gated off until a future change flips RESTORE_LAST_ACTIVE_TAB_BY_WORKTREE.
+   */
+  lastActiveTabByWorktree?: Record<string, string>
 }
 
 export type WorktreeSource = 'primary' | 'aimux-temp' | 'external'
@@ -289,16 +298,21 @@ export interface ModalClosed extends ModalBase {
 export interface ModalNewTab extends ModalBase {
   type: 'new-tab'
   editingCommand: AssistantId | null
-  activeField: 'assistant' | 'branch-name' | 'target-worktree' | 'worktree-name'
+  activeField: 'assistant' | 'branch-name' | 'target-worktree' | 'worktree-name' | 'base'
   branchError: string | null
   branchName: string
   createWorktree: boolean
   selectedAssistantId: AssistantId | null
-  step: 'assistant' | 'worktree' | 'worktree-create'
+  step: 'assistant' | 'worktree' | 'worktree-create' | 'template'
   targetWorktreeIndex: number
-  worktreeDeleteConfirmId: string | null
-  worktreeDeleteMessage: string | null
+  worktreeDeletePrompt: { worktreeId: string; reason: string } | null
   worktreeName: string
+  /** Filter text typed into the "Base" picker on the worktree-create step. */
+  baseQuery: string
+  /** Resolved base ref the new worktree is forked from (branch of a worktree or a local branch). */
+  baseRef: string
+  /** Local branches available as base refs, loaded when the create step opens. */
+  baseBranches: string[]
 }
 export interface ModalSessionPicker extends ModalBase {
   type: 'session-picker'
@@ -366,6 +380,22 @@ export interface ModalWorktreeMove extends ModalBase {
   deleteSource: boolean
 }
 
+/**
+ * Standalone confirmation for a recoverable worktree delete failure triggered
+ * outside the new-tab picker (e.g. the sidebar's "Remove worktree"). Carries the
+ * params needed to re-run the delete with force once confirmed.
+ */
+export interface ModalWorktreeDeleteConfirm extends ModalBase {
+  type: 'worktree-delete-confirm'
+  sessionId: string
+  worktreeId: string
+  worktreeLabel: string
+  reason: string
+  closeTabs: boolean
+  /** Whether confirming force-deletes — true only after a recoverable failure. */
+  force: boolean
+}
+
 export type ModalState =
   | ModalClosed
   | ModalNewTab
@@ -382,6 +412,7 @@ export type ModalState =
   | ModalUpdateAvailable
   | ModalAIUsage
   | ModalWorktreeMove
+  | ModalWorktreeDeleteConfirm
 
 export interface LayoutState {
   terminalCols: number
@@ -446,7 +477,9 @@ export interface AppState {
   autoCommit: AutoCommitState
   multiRepo: MultiRepoState
   worktreeDivergence: Record<string, BranchDivergence>
+  lastActiveTabByWorktree: Record<string, string>
   pendingChords: string[] | null
+  worktreeTemplates: WorktreeTemplate[]
 }
 
 // ─── AppAction union ──────────────────────────────────────────────────────────
@@ -455,11 +488,13 @@ export type ModalAction =
   | { type: 'open-new-tab-modal' }
   | { type: 'set-new-tab-branch-error'; message: string | null }
   | {
-      type: 'set-new-tab-worktree-delete-state'
-      confirmWorktreeId?: string | null
-      message: string | null
+      type: 'set-new-tab-worktree-delete-prompt'
+      prompt: { worktreeId: string; reason: string } | null
     }
+  | { type: 'set-new-tab-base-branches'; branches: string[] }
   | { type: 'enter-new-tab-worktree-create' }
+  | { type: 'enter-new-tab-template-pick' }
+  | { type: 'enter-new-tab-template-shortcut' }
   | { type: 'select-new-tab-assistant'; assistantId?: AssistantId }
   | { type: 'toggle-new-tab-worktree'; assistantId?: AssistantId }
   | { type: 'open-help-modal'; scope?: ModeId }
@@ -491,6 +526,15 @@ export type ModalAction =
   | { type: 'open-edit-custom-command'; assistantId: AssistantId }
   | { type: 'open-worktree-move-modal'; sourceWorktreeId: string }
   | { type: 'toggle-worktree-move-delete' }
+  | {
+      type: 'open-worktree-delete-confirm'
+      sessionId: string
+      worktreeId: string
+      worktreeLabel: string
+      reason: string
+      closeTabs: boolean
+      force: boolean
+    }
 
 export type SessionAction =
   | { type: 'load-session'; sessionId: string; workspaceSnapshot?: WorkspaceSnapshotV1 }
@@ -502,7 +546,6 @@ export type SessionAction =
   | { type: 'reorder-active-session'; delta: number }
   | { type: 'set-session-status'; sessionId: string; status: SessionStatus }
   | { type: 'add-worktree-record'; sessionId: string; worktree: WorktreeRecord; activate?: boolean }
-  | { type: 'remove-worktree-record'; sessionId: string; worktreeId: string }
   | { type: 'set-active-worktree'; sessionId: string; worktreeId: string }
   | {
       type: 'update-worktree-record'
@@ -590,6 +633,26 @@ export interface BranchDivergence {
   behind: number
 }
 
+export interface WorktreeTemplatePane {
+  id: string
+  assistant: string
+  splitFrom?: string
+  direction?: SplitDirection
+  ratio?: number
+  send?: string
+}
+
+export interface WorktreeTemplateTab {
+  panes: WorktreeTemplatePane[]
+}
+
+export interface WorktreeTemplate {
+  id: string
+  name: string
+  description?: string
+  tabs: WorktreeTemplateTab[]
+}
+
 export type GitPanelAction =
   | { type: 'git-refresh-success'; payload: GitRefreshPayload }
   | { type: 'git-refresh-error'; kind: GitPanelError }
@@ -678,6 +741,7 @@ export type AppAction =
 export type SideEffect =
   | { type: 'quit'; state: AppState }
   | { type: 'launch-selected-assistant' }
+  | { type: 'load-new-tab-base-branches' }
   | { type: 'edit-selected-assistant' }
   | { type: 'confirm-selected-session' }
   | { type: 'delete-selected-session' }
@@ -717,7 +781,18 @@ export type SideEffect =
   | { type: 'cycle-sidebar-item'; direction: 1 | -1 }
   | { type: 'switch-tab-by-index'; index: number }
   | { type: 'delete-session'; sessionId: string }
-  | { type: 'delete-worktree'; sessionId: string; worktreeId: string; force?: boolean }
+  | {
+      type: 'delete-worktree'
+      sessionId: string
+      worktreeId: string
+      // Force the git worktree removal (discards uncommitted changes in the
+      // worktree). Also implies closing the worktree's tabs.
+      force?: boolean
+      // Close the worktree's tabs without forcing the git removal. Lets the
+      // sidebar "Remove worktree" clean up tabs (avoiding orphans) while still
+      // refusing to discard uncommitted work in a temp worktree.
+      closeTabs?: boolean
+    }
   | {
       type: 'move-worktree'
       sessionId: string
@@ -1038,6 +1113,11 @@ export interface AimuxUserConfig {
   statusBar?: StatusBarConfig
   externalEditor?: ExternalEditorConfig
   integrations?: AimuxIntegrationsConfig
+  /**
+   * Worktree templates: layouts (multi-pane + initial commands) applied at
+   * worktree creation. Selected from a picker in the new-tab modal.
+   */
+  worktreeTemplates?: WorktreeTemplate[]
 }
 
 // ─── Resolved config (internal) ───────────────────────────────────────────────
@@ -1114,4 +1194,5 @@ export interface ResolvedConfig {
   integrations: {
     claudeHooks: boolean
   }
+  worktreeTemplates: WorktreeTemplate[]
 }