Compare commits
16 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| c4c3556247 | |||
|
|
4342347ac6 | ||
| 7888a34bd9 | |||
|
|
e2a37cdbb9 | ||
|
|
6e9472b5e2 | ||
|
|
775f73348a | ||
| 2e9081f4f5 | |||
|
|
f7ac2f35fe | ||
| 97d7511e56 | |||
|
|
cd12a0cda8 | ||
| ffdf5e34c8 | |||
|
|
b3ac73a283 | ||
|
|
1128b3dfa8 | ||
| 90f46a778a | |||
|
|
ede47439b0 | ||
| a690788498 |
@@ -146,8 +146,9 @@ issue_ref_to_worktree_name() {
|
|||||||
|
|
||||||
issue_ref_to_worktree_path() {
|
issue_ref_to_worktree_path() {
|
||||||
local issue_ref="$1"
|
local issue_ref="$1"
|
||||||
|
local parent_dir="${2:-$WORKTREES_DIR}"
|
||||||
local worktree_name=$(issue_ref_to_worktree_name "$issue_ref")
|
local worktree_name=$(issue_ref_to_worktree_name "$issue_ref")
|
||||||
echo "$WORKTREES_DIR/$worktree_name"
|
echo "$parent_dir/.kugetsu-worktrees/$worktree_name"
|
||||||
}
|
}
|
||||||
|
|
||||||
issue_ref_to_branch_name() {
|
issue_ref_to_branch_name() {
|
||||||
@@ -195,13 +196,15 @@ get_repo_url() {
|
|||||||
|
|
||||||
worktree_exists() {
|
worktree_exists() {
|
||||||
local issue_ref="$1"
|
local issue_ref="$1"
|
||||||
local worktree_path=$(issue_ref_to_worktree_path "$issue_ref")
|
local parent_dir="${2:-$PWD}"
|
||||||
|
local worktree_path=$(issue_ref_to_worktree_path "$issue_ref" "$parent_dir")
|
||||||
[ -d "$worktree_path" ]
|
[ -d "$worktree_path" ]
|
||||||
}
|
}
|
||||||
|
|
||||||
create_worktree() {
|
create_worktree() {
|
||||||
local issue_ref="$1"
|
local issue_ref="$1"
|
||||||
local worktree_path=$(issue_ref_to_worktree_path "$issue_ref")
|
local parent_dir="${2:-$PWD}"
|
||||||
|
local worktree_path=$(issue_ref_to_worktree_path "$issue_ref" "$parent_dir")
|
||||||
local branch_name=$(issue_ref_to_branch_name "$issue_ref")
|
local branch_name=$(issue_ref_to_branch_name "$issue_ref")
|
||||||
local repo_url=$(get_repo_url "$issue_ref")
|
local repo_url=$(get_repo_url "$issue_ref")
|
||||||
|
|
||||||
@@ -211,9 +214,10 @@ create_worktree() {
|
|||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
ensure_worktree_dir
|
local worktree_parent_dir=$(dirname "$worktree_path")
|
||||||
|
mkdir -p "$worktree_parent_dir"
|
||||||
|
|
||||||
if worktree_exists "$issue_ref"; then
|
if worktree_exists "$issue_ref" "$parent_dir"; then
|
||||||
echo "Removing existing worktree at '$worktree_path'..."
|
echo "Removing existing worktree at '$worktree_path'..."
|
||||||
git worktree remove "$worktree_path" 2>/dev/null || rm -rf "$worktree_path"
|
git worktree remove "$worktree_path" 2>/dev/null || rm -rf "$worktree_path"
|
||||||
fi
|
fi
|
||||||
@@ -234,9 +238,10 @@ create_worktree() {
|
|||||||
|
|
||||||
remove_worktree_for_issue() {
|
remove_worktree_for_issue() {
|
||||||
local issue_ref="$1"
|
local issue_ref="$1"
|
||||||
local worktree_path=$(issue_ref_to_worktree_path "$issue_ref")
|
local parent_dir="${2:-$PWD}"
|
||||||
|
local worktree_path=$(issue_ref_to_worktree_path "$issue_ref" "$parent_dir")
|
||||||
|
|
||||||
if worktree_exists "$issue_ref"; then
|
if worktree_exists "$issue_ref" "$parent_dir"; then
|
||||||
echo "Removing worktree at '$worktree_path'..."
|
echo "Removing worktree at '$worktree_path'..."
|
||||||
git worktree remove "$worktree_path" 2>/dev/null || rm -rf "$worktree_path"
|
git worktree remove "$worktree_path" 2>/dev/null || rm -rf "$worktree_path"
|
||||||
fi
|
fi
|
||||||
@@ -871,7 +876,7 @@ cmd_doctor() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fix_session_permissions() {
|
fix_session_permissions() {
|
||||||
local opencode_db="${OPENCODE_DB:-$HOME/.opencode/opencode.db}"
|
local opencode_db="${OPENCODE_DB:-$HOME/.local/share/opencode/opencode.db}"
|
||||||
|
|
||||||
if [ ! -f "$opencode_db" ]; then
|
if [ ! -f "$opencode_db" ]; then
|
||||||
echo "[ERROR] opencode database not found: $opencode_db"
|
echo "[ERROR] opencode database not found: $opencode_db"
|
||||||
@@ -1102,6 +1107,22 @@ EOF
|
|||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
local cwd_files=$(ls -A "$PWD" 2>/dev/null | wc -l)
|
||||||
|
local cwd_git=$(git rev-parse --is-inside-work-tree 2>/dev/null || echo "false")
|
||||||
|
if [ "$cwd_files" -gt 0 ] || [ "$cwd_git" = "true" ]; then
|
||||||
|
echo "Warning: Current directory is not empty: $PWD" >&2
|
||||||
|
echo "This may cause project context to contaminate the base session." >&2
|
||||||
|
echo "Consider running kugetsu init from an empty directory." >&2
|
||||||
|
echo "" >&2
|
||||||
|
echo "Files in current directory: $cwd_files" >&2
|
||||||
|
if [ "$cwd_git" = "true" ]; then
|
||||||
|
echo "Git repository detected: $(git rev-parse --show-toplevel 2>/dev/null || echo 'unknown')" >&2
|
||||||
|
fi
|
||||||
|
echo "" >&2
|
||||||
|
echo "Press Ctrl+C to cancel or wait 5 seconds to continue anyway..." >&2
|
||||||
|
sleep 5
|
||||||
|
fi
|
||||||
|
|
||||||
echo "Starting TUI to create base session..."
|
echo "Starting TUI to create base session..."
|
||||||
echo "Press Ctrl+C to cancel or wait for session to be created"
|
echo "Press Ctrl+C to cancel or wait for session to be created"
|
||||||
sleep 2
|
sleep 2
|
||||||
@@ -1215,17 +1236,12 @@ cmd_start() {
|
|||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
local worktree_path=$(issue_ref_to_worktree_path "$issue_ref")
|
local parent_dir="$PWD"
|
||||||
create_worktree "$issue_ref"
|
local worktree_path=$(issue_ref_to_worktree_path "$issue_ref" "$parent_dir")
|
||||||
|
create_worktree "$issue_ref" "$parent_dir"
|
||||||
|
|
||||||
local session_file="$(issue_ref_to_filename "$issue_ref").json"
|
local session_file="$(issue_ref_to_filename "$issue_ref").json"
|
||||||
|
|
||||||
# Get list of sessions before fork to compare against after
|
|
||||||
declare -a before_sessions=()
|
|
||||||
while IFS= read -r sess; do
|
|
||||||
before_sessions+=("$sess")
|
|
||||||
done < <(opencode session list 2>/dev/null | grep -oP '^ses_\w+')
|
|
||||||
|
|
||||||
echo "Forking session for '$issue_ref'..."
|
echo "Forking session for '$issue_ref'..."
|
||||||
|
|
||||||
# Session-counting: count actual dev sessions, reject if at limit
|
# Session-counting: count actual dev sessions, reject if at limit
|
||||||
@@ -1233,48 +1249,90 @@ cmd_start() {
|
|||||||
if [ "$active_count" -ge "$MAX_CONCURRENT_AGENTS" ]; then
|
if [ "$active_count" -ge "$MAX_CONCURRENT_AGENTS" ]; then
|
||||||
echo "Error: Max concurrent agents ($MAX_CONCURRENT_AGENTS) reached" >&2
|
echo "Error: Max concurrent agents ($MAX_CONCURRENT_AGENTS) reached" >&2
|
||||||
echo "Active sessions: $active_count" >&2
|
echo "Active sessions: $active_count" >&2
|
||||||
remove_worktree_for_issue "$issue_ref"
|
remove_worktree_for_issue "$issue_ref" "$parent_dir"
|
||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
local fork_log="$SESSIONS_DIR/$session_file.fork.log"
|
||||||
|
local opencode_db="${OPENCODE_DB:-$HOME/.local/share/opencode/opencode.db}"
|
||||||
|
|
||||||
|
> "$fork_log"
|
||||||
|
|
||||||
|
fix_session_permissions
|
||||||
|
|
||||||
if [ "$DEBUG_MODE" = true ]; then
|
if [ "$DEBUG_MODE" = true ]; then
|
||||||
opencode run "$message" --fork --session "$base_session_id" --dir "$worktree_path" 2>&1 | tee "$SESSIONS_DIR/$session_file.debug.log" &
|
(cd "$worktree_path" && opencode run "$message" --fork --session "$base_session_id" --dir "$worktree_path" 2>&1) | tee "$fork_log" &
|
||||||
else
|
else
|
||||||
opencode run "$message" --fork --session "$base_session_id" --dir "$worktree_path" 2>&1 &
|
(cd "$worktree_path" && opencode run "$message" --fork --session "$base_session_id" --dir "$worktree_path" 2>&1) >> "$fork_log" &
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# Wait briefly for session to be created
|
local fork_pid=$!
|
||||||
sleep 1
|
|
||||||
|
|
||||||
# Find the new session by comparing before/after lists
|
local max_attempts=10
|
||||||
# Skip any session that existed before the fork and skip base/pm-agent
|
local attempt=1
|
||||||
local new_session_id=""
|
local new_session_id=""
|
||||||
while IFS= read -r sess; do
|
local fork_log_output=""
|
||||||
# Skip base and pm-agent
|
|
||||||
[ "$sess" = "$base_session_id" ] && continue
|
|
||||||
[ "$sess" = "$pm_agent_session_id" ] && continue
|
|
||||||
|
|
||||||
# Check if this session existed before
|
while [ $attempt -le $max_attempts ]; do
|
||||||
local existed_before=false
|
sleep 1
|
||||||
for before_sess in "${before_sessions[@]}"; do
|
|
||||||
if [ "$sess" = "$before_sess" ]; then
|
|
||||||
existed_before=true
|
|
||||||
break
|
|
||||||
fi
|
|
||||||
done
|
|
||||||
|
|
||||||
if [ "$existed_before" = false ]; then
|
new_session_id=$(python3 -c "
|
||||||
new_session_id="$sess"
|
import sqlite3
|
||||||
|
conn = sqlite3.connect('$opencode_db')
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute(\"SELECT id FROM session WHERE directory = '$worktree_path' ORDER BY time_created DESC LIMIT 1\")
|
||||||
|
result = cursor.fetchone()
|
||||||
|
if result:
|
||||||
|
print(result[0])
|
||||||
|
" 2>/dev/null || echo "")
|
||||||
|
|
||||||
|
if [ -n "$new_session_id" ] && [ "$new_session_id" != "$base_session_id" ] && [ "$new_session_id" != "$pm_agent_session_id" ]; then
|
||||||
break
|
break
|
||||||
fi
|
fi
|
||||||
done < <(opencode session list 2>/dev/null | grep -oP '^ses_\w+')
|
|
||||||
|
if ! kill -0 $fork_pid 2>/dev/null; then
|
||||||
|
fork_log_output=$(tail -20 "$fork_log" 2>/dev/null || echo "(log empty or unavailable)")
|
||||||
|
break
|
||||||
|
fi
|
||||||
|
|
||||||
|
attempt=$((attempt + 1))
|
||||||
|
done
|
||||||
|
|
||||||
if [ -z "$new_session_id" ]; then
|
if [ -z "$new_session_id" ]; then
|
||||||
echo "Error: Could not find newly created session" >&2
|
echo "Error: Could not find newly created session after ${max_attempts}s" >&2
|
||||||
|
if [ -n "$fork_log_output" ]; then
|
||||||
|
echo "Fork log output:" >&2
|
||||||
|
echo "$fork_log_output" >&2
|
||||||
|
fi
|
||||||
remove_worktree_for_issue "$issue_ref"
|
remove_worktree_for_issue "$issue_ref"
|
||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
echo "Updating permissions for new session: $new_session_id"
|
||||||
|
python3 -c "
|
||||||
|
import sqlite3
|
||||||
|
conn = sqlite3.connect('$opencode_db')
|
||||||
|
cursor = conn.cursor()
|
||||||
|
PERMISSION_JSON = '[{\"permission\":\"question\",\"pattern\":\"*\",\"action\":\"deny\"},{\"permission\":\"plan_enter\",\"pattern\":\"*\",\"action\":\"deny\"},{\"permission\":\"plan_exit\",\"pattern\":\"*\",\"action\":\"deny\"},{\"permission\":\"external_directory\",\"pattern\":\"*\",\"action\":\"allow\"}]'
|
||||||
|
cursor.execute('UPDATE session SET permission = ? WHERE id = ?', (PERMISSION_JSON, '$new_session_id'))
|
||||||
|
conn.commit()
|
||||||
|
print('[OK] Session permissions updated')
|
||||||
|
"
|
||||||
|
|
||||||
|
if [ "$DEBUG_MODE" = true ]; then
|
||||||
|
echo "[DEBUG] Forked session permissions check:"
|
||||||
|
python3 -c "
|
||||||
|
import sqlite3
|
||||||
|
conn = sqlite3.connect('$opencode_db')
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute(\"SELECT id, directory, permission FROM session WHERE id = '$new_session_id'\")
|
||||||
|
for row in cursor.fetchall():
|
||||||
|
print(' ID:', row[0])
|
||||||
|
print(' Directory:', row[1])
|
||||||
|
print(' Permission:', row[2])
|
||||||
|
" 2>/dev/null || echo " (failed to query DB)"
|
||||||
|
fi
|
||||||
|
|
||||||
printf '{"type": "forked", "issue_ref": "%s", "opencode_session_id": "%s", "worktree_path": "%s", "created_at": "%s", "state": "idle"}\n' \
|
printf '{"type": "forked", "issue_ref": "%s", "opencode_session_id": "%s", "worktree_path": "%s", "created_at": "%s", "state": "idle"}\n' \
|
||||||
"$issue_ref" "$new_session_id" "$worktree_path" "$(date -Iseconds)" > "$SESSIONS_DIR/$session_file"
|
"$issue_ref" "$new_session_id" "$worktree_path" "$(date -Iseconds)" > "$SESSIONS_DIR/$session_file"
|
||||||
|
|
||||||
@@ -1490,6 +1548,7 @@ cmd_destroy() {
|
|||||||
|
|
||||||
if [ "$target" = "base" ]; then
|
if [ "$target" = "base" ]; then
|
||||||
if [ "$force" = true ]; then
|
if [ "$force" = true ]; then
|
||||||
|
local base_session_id=$(get_base_session_id)
|
||||||
rm -f "$SESSIONS_DIR/base.json"
|
rm -f "$SESSIONS_DIR/base.json"
|
||||||
local pm_agent=$(get_pm_agent_session_id)
|
local pm_agent=$(get_pm_agent_session_id)
|
||||||
if [ -n "$pm_agent" ] && [ "$pm_agent" != "null" ]; then
|
if [ -n "$pm_agent" ] && [ "$pm_agent" != "null" ]; then
|
||||||
@@ -1498,6 +1557,10 @@ cmd_destroy() {
|
|||||||
else
|
else
|
||||||
echo '{"base": null, "pm_agent": null, "issues": {}}' > "$INDEX_FILE"
|
echo '{"base": null, "pm_agent": null, "issues": {}}' > "$INDEX_FILE"
|
||||||
fi
|
fi
|
||||||
|
if [ -n "$base_session_id" ] && [ "$base_session_id" != "null" ]; then
|
||||||
|
echo "Deleting opencode session: $base_session_id"
|
||||||
|
opencode session delete "$base_session_id" 2>/dev/null || echo "Warning: Could not delete session from opencode (may already be deleted)"
|
||||||
|
fi
|
||||||
echo "Base session destroyed"
|
echo "Base session destroyed"
|
||||||
else
|
else
|
||||||
echo "Error: destroying base session requires --base -y" >&2
|
echo "Error: destroying base session requires --base -y" >&2
|
||||||
@@ -1508,6 +1571,7 @@ cmd_destroy() {
|
|||||||
|
|
||||||
if [ "$target" = "pm-agent" ]; then
|
if [ "$target" = "pm-agent" ]; then
|
||||||
if [ "$force" = true ]; then
|
if [ "$force" = true ]; then
|
||||||
|
local pm_session_id=$(get_pm_agent_session_id)
|
||||||
rm -f "$SESSIONS_DIR/pm-agent.json"
|
rm -f "$SESSIONS_DIR/pm-agent.json"
|
||||||
local base=$(get_base_session_id)
|
local base=$(get_base_session_id)
|
||||||
if [ -n "$base" ] && [ "$base" != "null" ]; then
|
if [ -n "$base" ] && [ "$base" != "null" ]; then
|
||||||
@@ -1515,6 +1579,10 @@ cmd_destroy() {
|
|||||||
else
|
else
|
||||||
write_index "null" "null" "{}"
|
write_index "null" "null" "{}"
|
||||||
fi
|
fi
|
||||||
|
if [ -n "$pm_session_id" ] && [ "$pm_session_id" != "null" ]; then
|
||||||
|
echo "Deleting opencode session: $pm_session_id"
|
||||||
|
opencode session delete "$pm_session_id" 2>/dev/null || echo "Warning: Could not delete session from opencode (may already be deleted)"
|
||||||
|
fi
|
||||||
echo "PM agent session destroyed"
|
echo "PM agent session destroyed"
|
||||||
else
|
else
|
||||||
echo "Error: destroying pm-agent session requires --pm-agent -y" >&2
|
echo "Error: destroying pm-agent session requires --pm-agent -y" >&2
|
||||||
|
|||||||
Reference in New Issue
Block a user