Compare commits

...

7 Commits

Author SHA1 Message Date
shokollm
66bbe53d3d Fix typo: KUETSU_DIR -> KUGETSU_DIR 2026-04-05 04:13:59 +00:00
shokollm
68f989d75e Merge feat/context-dump-load with feat/worktree-lifecycle changes
Combines context dump/load feature with worktree lifecycle features:
- CONTEXT_DIR and ENABLE_CONTEXT_DUMP config
- Context functions: kugetsu_context_load/dump/update_message
- WORKTREE_CHECK_PR_STATUS config
- check_pr_status() and update_session_pr_url() functions
- set-pr and context commands
2026-04-05 04:12:31 +00:00
shokollm
b72bf8e3c4 feat(context): add context dump/load for session isolation
- Add CONTEXT_DIR and ENABLE_CONTEXT_DUMP config options
- Add issue_ref_to_context_file() to derive context file path
- Add kugetsu_context_load() to load previous context
- Add kugetsu_context_dump() to save context on start
- Add kugetsu_context_update_message() to append to history
- Integrate context loading into cmd_start and cmd_continue
- Add kugetsu context command to view context for issue

Context is stored in ~/.kugetsu/context/<issue-ref>.json with:
- issue_ref, current_branch, updated_at
- last_message summary
- conversation_history (last 20 messages)

Closes #136
2026-04-05 02:54:55 +00:00
5d12f6ca42 Merge pull request 'feat(kugetsu): smart delegate with worktree awareness' (#130) from feature/smart-delegate-worktree-awareness into main 2026-04-03 16:31:30 +02:00
shokollm
91505345a2 feat(kugetsu): smart delegate with worktree awareness
- Parse issue refs from message (gitserver.com/owner/repo/issues/123 or owner/repo#123)
- Find existing worktrees/sessions by issue number
- Ask user to confirm which worktree to use, or delegate anyway
- Inject missing info context to PM agent
- Inject selected worktree context to PM agent

Fixes #128
2026-04-03 14:20:48 +00:00
f7fe22de25 Merge pull request 'fix(kugetsu): wrap cmd_continue in subshell with cd for correct worktree dir' (#129) from fix/cmd-continue-worktree-dir into main 2026-04-03 15:58:19 +02:00
shokollm
3ce43ffa65 fix(kugetsu): wrap cmd_continue in subshell with cd for correct worktree dir
The --dir flag only sets directory for the subprocess, not the session's
stored directory in opencode's SQLite DB. This was already fixed for
cmd_start in v0.1.10, but cmd_continue still had the bug.

Fixes #127
2026-04-03 13:06:02 +00:00

View File

@@ -13,6 +13,9 @@ VERBOSITY_DIR="$KUGETSU_DIR/verbosity"
MAX_CONCURRENT_AGENTS="${MAX_CONCURRENT_AGENTS:-3}"
KUGETSU_VERBOSITY="${KUGETSU_VERBOSITY:-default}"
CONTEXT_DIR="${CONTEXT_DIR:-$KUGETSU_DIR/context}"
ENABLE_CONTEXT_DUMP="${ENABLE_CONTEXT_DUMP:-true}"
WORKTREE_CHECK_PR_STATUS="${WORKTREE_CHECK_PR_STATUS:-true}"
# Load user config overrides (~/.kugetsu/config)
if [ -f "$KUGETSU_DIR/config" ]; then
@@ -77,6 +80,8 @@ Usage:
kugetsu destroy <issue-ref> [-y] Delete session for issue
kugetsu destroy --pm-agent [-y] Delete pm-agent session (not recommended)
kugetsu destroy --base [-y] Delete base session
kugetsu set-pr <issue-ref> <pr-url> Set PR URL for session (for PR tracking)
kugetsu context <issue-ref> Show context for issue
kugetsu help Show this help
Issue Ref Format:
@@ -267,6 +272,251 @@ filename_to_issue_ref() {
echo "$name" | sed 's/-\([0-9]*\)$/#\1' | sed 's/-/\//g'
}
issue_ref_to_context_file() {
local issue_ref="$1"
local context_filename=$(issue_ref_to_filename "$issue_ref")
echo "$CONTEXT_DIR/${context_filename}.json"
}
kugetsu_context_load() {
local issue_ref="$1"
if [ "$ENABLE_CONTEXT_DUMP" != "true" ]; then
echo ""
return
fi
local context_file=$(issue_ref_to_context_file "$issue_ref")
if [ ! -f "$context_file" ]; then
echo ""
return
fi
python3 << PYEOF
import json
import sys
context_file = "$context_file"
try:
with open(context_file, 'r') as f:
ctx = json.load(f)
lines = []
lines.append("## PREVIOUS CONTEXT")
lines.append(f"Issue: {ctx.get('issue_ref', 'unknown')}")
lines.append(f"Last updated: {ctx.get('updated_at', 'unknown')}")
lines.append(f"Current branch: {ctx.get('current_branch', 'unknown')}")
lines.append("")
lines.append("### Previous work summary:")
lines.append(ctx.get('last_message', '(no previous message)'))
lines.append("")
history = ctx.get('conversation_history', [])
if history:
lines.append("### Conversation history:")
for msg in history[-5:]:
role = msg.get('role', 'unknown')
content = msg.get('content', '')
ts = msg.get('timestamp', '')
lines.append(f"- [{ts}] {role}: {content[:200]}...")
print('\n'.join(lines))
except Exception as e:
print(f"Warning: Failed to load context: {e}", file=sys.stderr)
print("", file=sys.stderr)
PYEOF
}
kugetsu_context_dump() {
local issue_ref="$1"
local message="$2"
local branch_name="${3:-}"
if [ "$ENABLE_CONTEXT_DUMP" != "true" ]; then
return
fi
local context_file=$(issue_ref_to_context_file "$issue_ref")
mkdir -p "$CONTEXT_DIR"
python3 << PYEOF
import json
import os
from datetime import datetime
context_file = "$context_file"
issue_ref = "$issue_ref"
message = """$message"""
branch_name = "$branch_name"
context = {
"issue_ref": issue_ref,
"current_branch": branch_name,
"updated_at": datetime.now().isoformat() + "Z",
"last_message": message[:500] if message else "",
"conversation_history": []
}
if os.path.exists(context_file):
try:
with open(context_file, 'r') as f:
existing = json.load(f)
history = existing.get('conversation_history', [])
history.append({
"role": "user",
"content": message[:1000] if message else "",
"timestamp": datetime.now().isoformat() + "Z"
})
history = history[-20:]
context["conversation_history"] = history
context["created_at"] = existing.get("created_at", context["updated_at"])
except:
context["created_at"] = datetime.now().isoformat() + "Z"
else:
context["created_at"] = datetime.now().isoformat() + "Z"
with open(context_file, 'w') as f:
json.dump(context, f, indent=2)
PYEOF
}
kugetsu_context_update_message() {
local issue_ref="$1"
local message="$2"
if [ "$ENABLE_CONTEXT_DUMP" != "true" ]; then
return
fi
local context_file=$(issue_ref_to_context_file "$issue_ref")
if [ ! -f "$context_file" ]; then
return
fi
python3 << PYEOF
import json
from datetime import datetime
context_file = "$context_file"
message = """$message"""
try:
with open(context_file, 'r') as f:
ctx = json.load(f)
history = ctx.get('conversation_history', [])
history.append({
"role": "assistant",
"content": message[:1000] if message else "",
"timestamp": datetime.now().isoformat() + "Z"
})
history = history[-20:]
ctx["conversation_history"] = history
ctx["last_message"] = message[:500] if message else ""
ctx["updated_at"] = datetime.now().isoformat() + "Z"
with open(context_file, 'w') as f:
json.dump(ctx, f, indent=2)
except Exception as e:
pass
PYEOF
}
update_session_pr_url() {
local issue_ref="$1"
local pr_url="$2"
if [ -z "$issue_ref" ] || [ -z "$pr_url" ]; then
echo "Error: update_session_pr_url requires <issue-ref> and <pr-url>" >&2
return 1
fi
local session_file=$(get_session_for_issue "$issue_ref")
if [ -z "$session_file" ] || [ "$session_file" = "null" ]; then
echo "Error: No session found for '$issue_ref'" >&2
return 1
fi
local session_path="$SESSIONS_DIR/$session_file"
if [ ! -f "$session_path" ]; then
echo "Error: Session file not found: $session_path" >&2
return 1
fi
python3 << PYEOF
import json
session_path = "$session_path"
pr_url = "$pr_url"
with open(session_path, 'r') as f:
session = json.load(f)
session['pr_url'] = pr_url
with open(session_path, 'w') as f:
json.dump(session, f, indent=2)
print(f"Updated pr_url to: {pr_url}")
PYEOF
}
check_pr_status() {
local pr_url="$1"
if [ -z "$pr_url" ]; then
echo "no_pr_url"
return 1
fi
local hostname=$(echo "$pr_url" | sed -E 's|https://([^/]+)/.*|\1|')
local server_base="${GIT_SERVERS[$hostname]:-}"
if [ -z "$server_base" ]; then
echo "unknown_server"
return 1
fi
local api_base="${server_base}/api/v1"
local api_url=$(echo "$pr_url" | sed -E 's|https://[^/]+/([^/]+)/([^/]+)/(pulls|merge_requests)/([0-9]+)|'"${api_base}"'/repos/\1/\2/\3/\4|')
local token=""
if [[ "$hostname" == "github.com" ]]; then
token="${GITHUB_TOKEN:-}"
else
token="${GITEA_TOKEN:-}"
fi
local response
if [ -n "$token" ]; then
response=$(curl -s -H "Authorization: token $token" "$api_url" 2>/dev/null || echo "{}")
else
response=$(curl -s "$api_url" 2>/dev/null || echo "{}")
fi
local state=$(echo "$response" | python3 -c "import json, sys; d=json.load(sys.stdin); print(d.get('state', 'unknown'))" 2>/dev/null || echo "unknown")
local merged=$(echo "$response" | python3 -c "import json, sys; d=json.load(sys.stdin); print('true' if d.get('merged', False) else 'false')" 2>/dev/null || echo "false")
if [ "$merged" = "true" ]; then
echo "merged"
elif [ "$state" = "closed" ]; then
echo "closed"
elif [ "$state" = "open" ]; then
echo "open"
else
echo "unknown"
fi
}
read_index() {
if [ -f "$INDEX_FILE" ]; then
cat "$INDEX_FILE"
@@ -641,6 +891,99 @@ EOF
fi
}
parse_issue_ref_from_message() {
local message="$1"
local gitserver=""
local owner=""
local repo=""
local issue_number=""
if echo "$message" | grep -qE '[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}/[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+/(issues|pull)/[0-9]+'; then
gitserver=$(echo "$message" | grep -oE '[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}/[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+' | head -1 | sed 's/\/[^/]*\/[^/]*$//')
local full_path=$(echo "$message" | grep -oE '[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}/[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+/(issues|pull)/[0-9]+' | head -1)
owner=$(echo "$full_path" | cut -d'/' -f2)
repo=$(echo "$full_path" | cut -d'/' -f3)
issue_number=$(echo "$full_path" | grep -oE '[0-9]+$' | head -1)
elif echo "$message" | grep -qE '[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+#([0-9]+)'; then
owner=$(echo "$message" | grep -oE '[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+#' | sed 's/#$//' | cut -d'/' -f1)
repo=$(echo "$message" | grep -oE '[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+#' | sed 's/#$//' | cut -d'/' -f2)
issue_number=$(echo "$message" | grep -oE '#[0-9]+' | grep -oE '[0-9]+' | head -1)
fi
echo "${gitserver}|${owner}|${repo}|${issue_number}"
}
get_missing_info() {
local parsed="$1"
local gitserver=$(echo "$parsed" | cut -d'|' -f1)
local owner=$(echo "$parsed" | cut -d'|' -f2)
local repo=$(echo "$parsed" | cut -d'|' -f3)
local issue_number=$(echo "$parsed" | cut -d'|' -f4)
local missing=""
[ -z "$gitserver" ] && missing="${missing}git server, "
[ -z "$owner" ] && missing="${missing}owner, "
[ -z "$repo" ] && missing="${missing}repository, "
[ -z "$issue_number" ] && missing="${missing}issue number, "
echo "$missing" | sed 's/, $//'
}
build_missing_info_context() {
local missing="$1"
if [ -n "$missing" ]; then
echo ""
echo "NOTE: This task delegation has no information about: ${missing}."
echo "We need them if user wants to work on a specific issue. Otherwise we don't need it."
fi
}
find_worktrees_by_issue_number() {
local issue_number="$1"
local results=""
if [ ! -d "$WORKTREES_DIR/.kugetsu-worktrees" ]; then
echo ""
return
fi
for wt in "$WORKTREES_DIR/.kugetsu-worktrees"/*; do
if [ -d "$wt" ]; then
local wt_issue_number=$(echo "$wt" | grep -oE '#?[0-9]+$' | grep -oE '[0-9]+' | head -1)
if [ "$wt_issue_number" = "$issue_number" ]; then
results="${results}${wt}:worktree
"
fi
fi
done
echo "$results"
}
find_sessions_by_issue_number() {
local issue_number="$1"
local results=""
if [ ! -d "$SESSIONS_DIR" ]; then
echo ""
return
fi
for session_file in "$SESSIONS_DIR"/*.json; do
if [ -f "$session_file" ]; then
local session_issue_ref=$(basename "$session_file" .json | sed 's/_/\//g')
local session_issue_number=$(echo "$session_issue_ref" | grep -oE '#?[0-9]+$' | grep -oE '[0-9]+' | head -1)
if [ "$session_issue_number" = "$issue_number" ]; then
results="${results}${session_file}:session
"
fi
fi
done
echo "$results"
}
cmd_delegate() {
local message="${1:-}"
local verbosity="${KUGETSU_VERBOSITY:-default}"
@@ -660,6 +1003,70 @@ cmd_delegate() {
mkdir -p "$LOGS_DIR"
local log_file="$LOGS_DIR/delegate-$(date +%s).log"
local parsed=$(parse_issue_ref_from_message "$message")
local gitserver=$(echo "$parsed" | cut -d'|' -f1)
local owner=$(echo "$parsed" | cut -d'|' -f2)
local repo=$(echo "$parsed" | cut -d'|' -f3)
local issue_number=$(echo "$parsed" | cut -d'|' -f4)
local missing_info=$(get_missing_info "$parsed")
local context_injection=""
if [ -n "$missing_info" ]; then
context_injection=$(build_missing_info_context "$missing_info")
echo "NOTE: Delegation missing information: ${missing_info}"
fi
local candidates=""
local candidate_count=0
if [ -n "$issue_number" ]; then
local worktrees=$(find_worktrees_by_issue_number "$issue_number")
local sessions=$(find_sessions_by_issue_number "$issue_number")
while IFS=: read -r path type; do
if [ -n "$path" ]; then
candidate_count=$((candidate_count + 1))
candidates="${candidates}${candidate_count}) ${path} (${type})
"
fi
done <<< "$worktrees"
while IFS=: read -r path type; do
if [ -n "$path" ]; then
candidate_count=$((candidate_count + 1))
candidates="${candidates}${candidate_count}) ${path} (${type})
"
fi
done <<< "$sessions"
fi
local use_worktree=""
if [ $candidate_count -gt 0 ]; then
echo "Found $candidate_count existing worktree(s)/session(s) for issue #${issue_number}:"
echo "$candidates"
echo "r) Delegate anyway (without routing)"
echo "Which one to use? [1-${candidate_count}/r]: "
read -r choice
if [ "$choice" = "r" ] || [ -z "$choice" ]; then
use_worktree=""
elif [ "$choice" -ge 1 ] && [ "$choice" -le "$candidate_count" ]; then
local selected=$(echo "$candidates" | sed -n "${choice}p")
use_worktree=$(echo "$selected" | sed 's/) .*//')
fi
fi
local final_message="${message}${context_injection}"
if [ -n "$use_worktree" ]; then
if [ -d "$use_worktree" ]; then
echo "Using worktree: $use_worktree"
final_message="${final_message}
NOTE: Worktree selected: ${use_worktree}"
fi
fi
local temp_dir="${KUGETSU_TEMP_DIR:-$HOME/.local/share/opencode/tool-output}"
mkdir -p "$ENV_DIR"
@@ -671,7 +1078,7 @@ cmd_delegate() {
fi
env_sh="${env_sh}set +a; "
nohup sh -c "${env_sh}opencode run '$message' --continue --session '$pm_session' >> '$log_file' 2>&1" > /dev/null 2>&1 &
nohup sh -c "${env_sh}opencode run '${final_message}' --continue --session '$pm_session' >> '$log_file' 2>&1" > /dev/null 2>&1 &
disown
echo "Delegated to PM agent (logged to $(basename "$log_file"))"
echo "Verbosity: $verbosity"
@@ -1300,13 +1707,16 @@ cmd_start() {
exit 1
fi
local fork_log="$SESSIONS_DIR/$session_file.fork.log"
local opencode_db="${OPENCODE_DB:-$HOME/.local/share/opencode/opencode.db}"
local fork_log="$SESSIONS_DIR/$session_file.fork.log"
> "$fork_log"
local fork_context=$(kugetsu_get_fork_context "$issue_ref")
local previous_context=$(kugetsu_context_load "$issue_ref")
local branch_name=$(issue_ref_to_branch_name "$issue_ref")
local full_message="${fork_context}
${previous_context}
## YOUR TASK
$message"
@@ -1391,6 +1801,8 @@ for row in cursor.fetchall():
add_issue_to_index "$issue_ref" "$session_file"
kugetsu_context_dump "$issue_ref" "$message" "$branch_name"
echo "Session started for '$issue_ref': $new_session_id"
echo "Worktree: $worktree_path"
}
@@ -1438,21 +1850,31 @@ cmd_continue() {
local worktree_path=$(python3 -c "import json; print(json.load(open('$session_path')).get('worktree_path', ''))" 2>/dev/null || echo "")
echo "Continuing session for '$session_name'..."
local previous_context=$(kugetsu_context_load "$session_name")
local full_message="${previous_context}
## CONTINUE TASK
$message"
# Note: --continue always allowed (existing sessions don't count toward limit)
# Wrap in subshell with cd to ensure worktree directory is set correctly in session DB
if [ -n "$worktree_path" ] && [ -d "$worktree_path" ]; then
echo "Using worktree: $worktree_path"
if [ "$DEBUG_MODE" = true ]; then
opencode run "$message" --continue --session "$opencode_session_id" --dir "$worktree_path" 2>&1 | tee "$session_path.debug.log" &
(cd "$worktree_path" && opencode run "$full_message" --continue --session "$opencode_session_id" --dir "$worktree_path" 2>&1) | tee "$session_path.debug.log" &
else
opencode run "$message" --continue --session "$opencode_session_id" --dir "$worktree_path" 2>&1 &
(cd "$worktree_path" && opencode run "$full_message" --continue --session "$opencode_session_id" --dir "$worktree_path" 2>&1) &
fi
else
if [ "$DEBUG_MODE" = true ]; then
opencode run "$message" --continue --session "$opencode_session_id" 2>&1 | tee "$session_path.debug.log" &
opencode run "$full_message" --continue --session "$opencode_session_id" 2>&1 | tee "$session_path.debug.log" &
else
opencode run "$message" --continue --session "$opencode_session_id" 2>&1 &
opencode run "$full_message" --continue --session "$opencode_session_id" 2>&1 &
fi
fi
kugetsu_context_update_message "$session_name" "$message"
}
cmd_list() {
@@ -1729,6 +2151,33 @@ main() {
destroy)
cmd_destroy "$@"
;;
set-pr)
local issue_ref="${1:-}"
local pr_url="${2:-}"
if [ -z "$issue_ref" ] || [ -z "$pr_url" ]; then
echo "Usage: kugetsu set-pr <issue-ref> <pr-url>" >&2
echo "Example: kugetsu set-pr github.com/shoko/kugetsu#14 https://github.com/shoko/kugetsu/pulls/123" >&2
exit 1
fi
validate_issue_ref "$issue_ref"
update_session_pr_url "$issue_ref" "$pr_url"
;;
context)
local issue_ref="${1:-}"
if [ -z "$issue_ref" ]; then
echo "Usage: kugetsu context <issue-ref>" >&2
echo "Show context for an issue" >&2
exit 1
fi
validate_issue_ref "$issue_ref"
local context_file=$(issue_ref_to_context_file "$issue_ref")
if [ -f "$context_file" ]; then
cat "$context_file"
else
echo "No context found for '$issue_ref'" >&2
exit 1
fi
;;
*)
echo "Error: unknown command '$command'" >&2
usage