summaryrefslogtreecommitdiff
path: root/dotfiles/common/.local/bin/aix
blob: b9d9e69d4b00859a2e154fd468e47a2fe01db661 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
#!/bin/bash
# Launch tmux session with an LLM in fzf-selected project directories

SESSION="ai"

# Switch to or attach to the session
attach_session() {
  if [ -n "$TMUX" ]; then
    tmux switch-client -t "$SESSION"
  else
    tmux attach-session -t "$SESSION"
  fi
}

# Dependency checks
for cmd in fzf tmux claude; do
  if ! command -v "$cmd" &>/dev/null; then
    echo "Error: $cmd is not installed" >&2
    exit 1
  fi
done

# Claude command — separated for safe editing
AI_CMD="claude"
AI_INSTRUCTIONS='Read docs/protocols.org and follow all instructions.'

# Create a window in the session and launch claude; prints window ID
create_window() {
  local dir="$1" name="$2" wid
  wid=$(tmux new-window -t "$SESSION" -n "$name" -c "$dir" -P -F '#{window_id}')
  sleep 0.1
  tmux send-keys -t "$wid" "$AI_CMD \"$AI_INSTRUCTIONS\"" Enter
  echo "$wid"
}

# Read fzf selections into the 'selected' array
read_selections() {
  selected=()
  while IFS= read -r line; do
    selected+=("$line")
  done <<<"$1"
}

# Build candidate directory list
build_candidates() {
  candidates=()
  [ -d "$HOME/.emacs.d" ] && candidates+=("~/.emacs.d")
  if [ -d "$HOME/code" ]; then
    while IFS= read -r d; do
      candidates+=("~/${d#"$HOME"/}")
    done < <(find "$HOME/code" -maxdepth 1 -mindepth 1 -type d | sort)
  fi
  if [ -d "$HOME/projects" ]; then
    while IFS= read -r d; do
      candidates+=("~/${d#"$HOME"/}")
    done < <(find "$HOME/projects" -maxdepth 1 -mindepth 1 -type d | sort)
  fi
}

# Sort windows: shells at base-index, projects alphabetically after
sort_windows() {
  local windows shells projects base_idx
  base_idx=$(tmux show-option -gv base-index 2>/dev/null || echo 0)
  windows=$(tmux list-windows -t "$SESSION" -F '#{window_name}'$'\t''#{window_id}')
  shells=$(echo "$windows" | grep -iE '^(bash|zsh|fish|sh)'$'\t' | sort -t$'\t' -k1,1f)
  projects=$(echo "$windows" | grep -viE '^(bash|zsh|fish|sh)'$'\t' | sort -t$'\t' -k1,1f)
  local all
  all=$(printf '%s\n' "$shells" "$projects" | sed '/^$/d')

  # Move all to high temp indices to avoid collisions
  local i=900
  while IFS=$'\t' read -r _name wid; do
    tmux move-window -s "$wid" -t "$SESSION:$i"
    ((i++))
  done <<<"$all"

  # Place shells at base-index, projects contiguously after
  i=$base_idx
  if [ -n "$shells" ]; then
    while IFS=$'\t' read -r _name wid; do
      tmux move-window -s "$wid" -t "$SESSION:$i"
      ((i++))
    done <<<"$shells"
  fi
  if [ -n "$projects" ]; then
    while IFS=$'\t' read -r _name wid; do
      tmux move-window -s "$wid" -t "$SESSION:$i"
      ((i++))
    done <<<"$projects"
  fi
}

# Attach directly to existing session
if [ "$1" = "--attach" ]; then
  if ! tmux has-session -t "$SESSION" 2>/dev/null; then
    echo "No $SESSION session to attach to." >&2
    exit 1
  fi
  sort_windows
  attach_session
  exit 0
fi

# If session exists, add new windows to it
if tmux has-session -t "$SESSION" 2>/dev/null; then
  # Get existing window names to filter duplicates
  existing=$(tmux list-windows -t "$SESSION" -F '#{window_name}')

  build_candidates

  # Filter out candidates that already have a window open
  filtered=()
  for c in "${candidates[@]}"; do
    name="$(basename "${c/#\~/$HOME}")"
    if ! echo "$existing" | grep -qxF "$name"; then
      filtered+=("$c")
    fi
  done

  if [ ${#filtered[@]} -eq 0 ]; then
    echo "All projects already have windows open."
  else
    selections=$(printf '%s\n' "${filtered[@]}" | fzf --multi --height=70% --reverse)

    if [ -n "$selections" ]; then
      read_selections "$selections"

      first_wid=""
      for entry in "${selected[@]}"; do
        dir="${entry/#\~/$HOME}"
        name="$(basename "$dir")"
        wid=$(create_window "$dir" "$name")
        [ -z "$first_wid" ] && first_wid="$wid"
      done
    fi
  fi

  sort_windows
  [ -n "$first_wid" ] && tmux select-window -t "$first_wid"
  attach_session
  exit 0
fi

# New session: select projects and create session
build_candidates

selections=$(printf '%s\n' "${candidates[@]}" | fzf --multi --height=70% --reverse)
[ -z "$selections" ] && exit 0

read_selections "$selections"

# Create session with first selection
first="${selected[0]}"
dir="${first/#\~/$HOME}"
name="$(basename "$dir")"
first_wid=$(tmux new-session -d -s "$SESSION" -n "$name" -c "$dir" -P -F '#{window_id}')
tmux send-keys -t "$first_wid" "$AI_CMD \"$AI_INSTRUCTIONS\"" Enter

# Create remaining windows
for entry in "${selected[@]:1}"; do
  dir="${entry/#\~/$HOME}"
  name="$(basename "$dir")"
  create_window "$dir" "$name" > /dev/null
done

# Sort windows and select first project
sort_windows
tmux select-window -t "$first_wid"
attach_session