Compare commits
8 Commits
v0.2.11
...
8ba7242861
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
8ba7242861 | ||
|
|
b595411a07 | ||
|
|
08b633400d | ||
|
|
860bf9295f | ||
|
|
fd79bfa3ea | ||
| 119c9b8fd9 | |||
|
|
7f7f8b1085 | ||
| 1a523a805a |
@@ -481,6 +481,13 @@ write_index() {
|
|||||||
local issues_json="$3"
|
local issues_json="$3"
|
||||||
local temp_file="$INDEX_FILE.tmp.$$"
|
local temp_file="$INDEX_FILE.tmp.$$"
|
||||||
printf '{"base": %s, "pm_agent": %s, "issues": %s}\n' "$base" "$pm_agent" "$issues_json" > "$temp_file"
|
printf '{"base": %s, "pm_agent": %s, "issues": %s}\n' "$base" "$pm_agent" "$issues_json" > "$temp_file"
|
||||||
|
|
||||||
|
if ! python3 -c "import json; json.load(open('$temp_file'))" 2>/dev/null; then
|
||||||
|
echo "Error: write_index would create malformed JSON, aborting. base=$base, pm_agent=$pm_agent, issues_json=$issues_json" >&2
|
||||||
|
rm -f "$temp_file"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
|
||||||
mv "$temp_file" "$INDEX_FILE"
|
mv "$temp_file" "$INDEX_FILE"
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -53,5 +53,9 @@ load_agent_env() {
|
|||||||
set -a
|
set -a
|
||||||
source "$ENV_DIR/default.env"
|
source "$ENV_DIR/default.env"
|
||||||
set +a
|
set +a
|
||||||
|
elif [ -f "$ENV_DIR/pm-agent.env" ]; then
|
||||||
|
set -a
|
||||||
|
source "$ENV_DIR/pm-agent.env"
|
||||||
|
set +a
|
||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -15,6 +15,13 @@ write_index() {
|
|||||||
local issues_json="$3"
|
local issues_json="$3"
|
||||||
local temp_file="$INDEX_FILE.tmp.$$"
|
local temp_file="$INDEX_FILE.tmp.$$"
|
||||||
printf '{"base": %s, "pm_agent": %s, "issues": %s}\n' "$base" "$pm_agent" "$issues_json" > "$temp_file"
|
printf '{"base": %s, "pm_agent": %s, "issues": %s}\n' "$base" "$pm_agent" "$issues_json" > "$temp_file"
|
||||||
|
|
||||||
|
if ! python3 -c "import json; json.load(open('$temp_file'))" 2>/dev/null; then
|
||||||
|
echo "Error: write_index would create malformed JSON, aborting. base=$base, pm_agent=$pm_agent, issues_json=$issues_json" >&2
|
||||||
|
rm -f "$temp_file"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
|
||||||
mv "$temp_file" "$INDEX_FILE"
|
mv "$temp_file" "$INDEX_FILE"
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -109,7 +109,7 @@ process_task() {
|
|||||||
|
|
||||||
source "$SCRIPT_DIR/kugetsu-session.sh"
|
source "$SCRIPT_DIR/kugetsu-session.sh"
|
||||||
|
|
||||||
if worktree_exists "$issue_ref" "$HOME/.kugetsu-worktrees" || [ -f "$SESSIONS_DIR/$(issue_ref_to_filename "$issue_ref").json" ]; then
|
if worktree_exists "$issue_ref" "$WORKTREES_DIR" || [ -f "$SESSIONS_DIR/$(issue_ref_to_filename "$issue_ref").json" ]; then
|
||||||
log_file="$LOGS_DIR/delegate-$(date +%s).log"
|
log_file="$LOGS_DIR/delegate-$(date +%s).log"
|
||||||
if cmd_continue "$issue_ref" "$message" >> "$log_file" 2>&1; then
|
if cmd_continue "$issue_ref" "$message" >> "$log_file" 2>&1; then
|
||||||
sleep 1
|
sleep 1
|
||||||
|
|||||||
@@ -58,6 +58,25 @@ EOF
|
|||||||
echo "Created config file: $KUGETSU_DIR/config"
|
echo "Created config file: $KUGETSU_DIR/config"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
mkdir -p "$ENV_DIR"
|
||||||
|
if [ ! -f "$ENV_DIR/default.env" ]; then
|
||||||
|
cat > "$ENV_DIR/default.env" << 'EOF'
|
||||||
|
# Environment variables for agents
|
||||||
|
# Copy this file to <agent-type>.env (e.g., pm-agent.env, dev.env)
|
||||||
|
# and set your tokens and configuration
|
||||||
|
|
||||||
|
# Required: Gitea token for API access
|
||||||
|
# GITEA_TOKEN=your_gitea_token_here
|
||||||
|
|
||||||
|
# Optional: GitHub token (if using GitHub)
|
||||||
|
# GITHUB_TOKEN=your_github_token_here
|
||||||
|
|
||||||
|
# Optional: GitLab token (if using GitLab)
|
||||||
|
# GITLAB_TOKEN=your_gitlab_token_here
|
||||||
|
EOF
|
||||||
|
echo "Created env template: $ENV_DIR/default.env"
|
||||||
|
fi
|
||||||
|
|
||||||
local existing_base=$(get_base_session_id)
|
local existing_base=$(get_base_session_id)
|
||||||
local existing_pm=$(get_pm_agent_session_id)
|
local existing_pm=$(get_pm_agent_session_id)
|
||||||
|
|
||||||
@@ -185,18 +204,52 @@ cmd_delegate() {
|
|||||||
return
|
return
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# No issue ref detected — delegate directly to PM agent (legacy path)
|
# No issue ref detected — fork a new session from base session
|
||||||
local pm_session=$(get_pm_agent_session_id)
|
local base_session=$(get_base_session_id)
|
||||||
if [ -z "$pm_session" ] || [ "$pm_session" = "null" ] || [ "$pm_session" = "None" ]; then
|
if [ -z "$base_session" ] || [ "$base_session" = "null" ]; then
|
||||||
echo "Error: PM agent session not found. Run 'kugetsu init' first." >&2
|
echo "Error: Base session not found. Run 'kugetsu init' first." >&2
|
||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
mkdir -p "$LOGS_DIR"
|
mkdir -p "$LOGS_DIR"
|
||||||
local log_file="$LOGS_DIR/delegate-$(date +%s).log"
|
local log_file="$LOGS_DIR/delegate-$(date +%s).log"
|
||||||
nohup sh -c "GITEA_TOKEN='***' opencode run '$message' --continue --session '$pm_session' >> '$log_file' 2>&1" > /dev/null 2>&1 &
|
load_agent_env "pm-agent"
|
||||||
disown
|
|
||||||
echo "Delegated to PM agent (logged to $(basename "$log_file"))"
|
local new_session=$(create_session "$base_session")
|
||||||
|
if [ -z "$new_session" ]; then
|
||||||
|
echo "Error: Failed to create session" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
nohup sh -c "GITEA_TOKEN='${GITEA_TOKEN:-}' opencode run '$message' --session '$new_session'" >> "$log_file" 2>&1 &
|
||||||
|
echo "Delegated to new session (logged to $(basename "$log_file"))"
|
||||||
|
}
|
||||||
|
|
||||||
|
create_session() {
|
||||||
|
local base_session="${1:-$base_session_id}"
|
||||||
|
|
||||||
|
if [ -z "$base_session" ] || [ "$base_session" = "null" ]; then
|
||||||
|
echo "Error: base session not found. Run 'kugetsu init' first." >&2
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
local before_json=$(opencode session list --format=json 2>/dev/null)
|
||||||
|
local before_ids=$(echo "$before_json" | python3 -c "import sys,json; sessions=json.load(sys.stdin); print(' '.join(s['id'] for s in sessions))" 2>/dev/null || echo "")
|
||||||
|
|
||||||
|
opencode run --fork --session "$base_session" "new session" 2>/dev/null
|
||||||
|
|
||||||
|
local after_json=$(opencode session list --format=json 2>/dev/null)
|
||||||
|
local after_ids=$(echo "$after_json" | python3 -c "import sys,json; sessions=json.load(sys.stdin); print(' '.join(s['id'] for s in sessions))" 2>/dev/null || echo "")
|
||||||
|
|
||||||
|
local new_session_id=""
|
||||||
|
for sess in $after_ids; do
|
||||||
|
if [[ ! " $before_ids " =~ " $sess " ]] && [[ "$sess" != "$base_session" ]]; then
|
||||||
|
new_session_id="$sess"
|
||||||
|
break
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
|
echo "$new_session_id"
|
||||||
}
|
}
|
||||||
|
|
||||||
build_dev_agent_message() {
|
build_dev_agent_message() {
|
||||||
@@ -256,12 +309,6 @@ cmd_start() {
|
|||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
local pm_agent_session_id=$(get_pm_agent_session_id)
|
|
||||||
if [ -z "$pm_agent_session_id" ] || [ "$pm_agent_session_id" = "null" ]; then
|
|
||||||
echo "Error: PM agent session not found. Run 'kugetsu init' first." >&2
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
if worktree_exists "$issue_ref"; then
|
if worktree_exists "$issue_ref"; then
|
||||||
echo "Issue '$issue_ref' already has a worktree. Use 'kugetsu continue' instead."
|
echo "Issue '$issue_ref' already has a worktree. Use 'kugetsu continue' instead."
|
||||||
exit 1
|
exit 1
|
||||||
@@ -282,22 +329,12 @@ cmd_start() {
|
|||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
local before_sessions=$(opencode session list 2>/dev/null | grep -oP '^ses_\w+' | sort)
|
|
||||||
local before_set="|$before_sessions|"
|
|
||||||
|
|
||||||
create_worktree "$issue_ref" "$WORKTREES_DIR"
|
create_worktree "$issue_ref" "$WORKTREES_DIR"
|
||||||
|
|
||||||
local after_sessions=$(opencode session list 2>/dev/null | grep -oP '^ses_\w+' | sort)
|
local new_session_id=$(create_session "$base_session_id")
|
||||||
local new_session_id=""
|
|
||||||
while IFS= read -r sess; do
|
|
||||||
if [[ ! "$before_set" =~ \|${sess}\| ]] && [[ "$sess" != "$base_session_id" ]] && [[ "$sess" != "$pm_agent_session_id" ]]; then
|
|
||||||
new_session_id="$sess"
|
|
||||||
break
|
|
||||||
fi
|
|
||||||
done <<< "$after_sessions"
|
|
||||||
|
|
||||||
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 create session" >&2
|
||||||
remove_worktree_for_issue "$issue_ref"
|
remove_worktree_for_issue "$issue_ref"
|
||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
@@ -313,7 +350,8 @@ cmd_start() {
|
|||||||
|
|
||||||
load_agent_env "dev"
|
load_agent_env "dev"
|
||||||
|
|
||||||
(cd "$worktree_path" && nohup sh -c "GITEA_TOKEN='$GITEA_TOKEN' opencode run '$dev_message' --continue --session '$new_session_id'" >> "$LOGS_DIR/dev-$new_session_id.log" 2>&1 &) && disown
|
cd "$worktree_path"
|
||||||
|
nohup sh -c "GITEA_TOKEN='${GITEA_TOKEN:-}' opencode run '$dev_message' --session '$new_session_id'" >> "$LOGS_DIR/dev-$new_session_id.log" 2>&1 &
|
||||||
|
|
||||||
echo "Session started for '$issue_ref': $new_session_id"
|
echo "Session started for '$issue_ref': $new_session_id"
|
||||||
echo "Worktree: $worktree_path"
|
echo "Worktree: $worktree_path"
|
||||||
@@ -367,9 +405,10 @@ cmd_continue() {
|
|||||||
fi
|
fi
|
||||||
|
|
||||||
if [ -n "$worktree_path" ] && [ -d "$worktree_path" ]; then
|
if [ -n "$worktree_path" ] && [ -d "$worktree_path" ]; then
|
||||||
(cd "$worktree_path" && nohup sh -c "GITEA_TOKEN='$GITEA_TOKEN' opencode run '$message' --continue --session '$opencode_session_id'" >> "$LOGS_DIR/dev-$opencode_session_id.log" 2>&1 &) && disown
|
cd "$worktree_path"
|
||||||
|
nohup sh -c "GITEA_TOKEN='${GITEA_TOKEN:-}' opencode run '$message' --session '$opencode_session_id'" >> "$LOGS_DIR/dev-$opencode_session_id.log" 2>&1 &
|
||||||
else
|
else
|
||||||
(nohup sh -c "GITEA_TOKEN='$GITEA_TOKEN' opencode run '$message' --continue --session '$opencode_session_id'" >> "$LOGS_DIR/dev-$opencode_session_id.log" 2>&1 &) && disown
|
nohup sh -c "GITEA_TOKEN='${GITEA_TOKEN:-}' opencode run '$message' --session '$opencode_session_id'" >> "$LOGS_DIR/dev-$opencode_session_id.log" 2>&1 &
|
||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -521,7 +560,7 @@ cmd_destroy() {
|
|||||||
local target="${1:-}"
|
local target="${1:-}"
|
||||||
local force=false
|
local force=false
|
||||||
|
|
||||||
if [ "$2" = "-y" ]; then
|
if [ "${2:-}" = "-y" ]; then
|
||||||
force=true
|
force=true
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
|||||||
181
skills/kugetsu/tests/test-create-session.sh
Normal file
181
skills/kugetsu/tests/test-create-session.sh
Normal file
@@ -0,0 +1,181 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
# Tests for create_session function
|
||||||
|
#
|
||||||
|
# Run with: bash skills/kugetsu/tests/test-create-session.sh
|
||||||
|
#
|
||||||
|
# NOTE: These tests MUST be run sequentially (not in parallel)
|
||||||
|
# to avoid exhausting memory with too many opencode sessions.
|
||||||
|
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
|
source "$SCRIPT_DIR/../scripts/kugetsu-config.sh"
|
||||||
|
source "$SCRIPT_DIR/../scripts/kugetsu-index.sh"
|
||||||
|
source "$SCRIPT_DIR/../scripts/kugetsu-session.sh"
|
||||||
|
|
||||||
|
PASS=0
|
||||||
|
FAIL=0
|
||||||
|
RUN=0
|
||||||
|
|
||||||
|
pass() {
|
||||||
|
echo "PASS: $1"
|
||||||
|
PASS=$((PASS + 1))
|
||||||
|
}
|
||||||
|
|
||||||
|
fail() {
|
||||||
|
echo "FAIL: $1"
|
||||||
|
echo " Expected: $2"
|
||||||
|
echo " Got: $3"
|
||||||
|
FAIL=$((FAIL + 1))
|
||||||
|
}
|
||||||
|
|
||||||
|
run_test() {
|
||||||
|
local name="$1"
|
||||||
|
local test_func="$2"
|
||||||
|
RUN=$((RUN + 1))
|
||||||
|
echo ""
|
||||||
|
echo "=== Test $RUN: $name ==="
|
||||||
|
echo "--- $name ---"
|
||||||
|
$test_func
|
||||||
|
}
|
||||||
|
|
||||||
|
echo "=== create_session Test Suite ==="
|
||||||
|
echo "NOTE: Running sequentially to avoid memory exhaustion"
|
||||||
|
echo ""
|
||||||
|
|
||||||
|
# Test 1: create_session requires base session
|
||||||
|
test_create_session_requires_base() {
|
||||||
|
local base_id=$(get_base_session_id)
|
||||||
|
if [ -z "$base_id" ] || [ "$base_id" = "null" ]; then
|
||||||
|
skip "Base session not initialized - run 'kugetsu init' first"
|
||||||
|
return
|
||||||
|
fi
|
||||||
|
|
||||||
|
local result=$(create_session "$base_id")
|
||||||
|
if [ -n "$result" ] && [[ "$result" =~ ^ses_ ]]; then
|
||||||
|
pass "create_session returns valid session ID"
|
||||||
|
else
|
||||||
|
fail "create_session returns valid session ID" "ses_xxx" "$result"
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# Test 2: create_session creates a NEW session (different from base)
|
||||||
|
test_create_session_is_new() {
|
||||||
|
local base_id=$(get_base_session_id)
|
||||||
|
if [ -z "$base_id" ] || [ "$base_id" = "null" ]; then
|
||||||
|
skip "Base session not initialized - run 'kugetsu init' first"
|
||||||
|
return
|
||||||
|
fi
|
||||||
|
|
||||||
|
local new_id=$(create_session "$base_id")
|
||||||
|
if [ "$new_id" != "$base_id" ]; then
|
||||||
|
pass "create_session returns NEW session (not same as base)"
|
||||||
|
else
|
||||||
|
fail "create_session returns NEW session" "different from base_id" "$new_id"
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# Test 3: create_session can be called multiple times (creates different sessions)
|
||||||
|
test_create_session_multiple_calls() {
|
||||||
|
local base_id=$(get_base_session_id)
|
||||||
|
if [ -z "$base_id" ] || [ "$base_id" = "null" ]; then
|
||||||
|
skip "Base session not initialized - run 'kugetsu init' first"
|
||||||
|
return
|
||||||
|
fi
|
||||||
|
|
||||||
|
local id1=$(create_session "$base_id")
|
||||||
|
sleep 1
|
||||||
|
local id2=$(create_session "$base_id")
|
||||||
|
|
||||||
|
if [ "$id1" != "$id2" ]; then
|
||||||
|
pass "create_session creates different sessions on each call"
|
||||||
|
else
|
||||||
|
fail "create_session creates different sessions" "$id1 != $id2" "both equal: $id1"
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# Test 4: JSON session list parsing
|
||||||
|
test_session_json_parsing() {
|
||||||
|
local json='[{"id": "ses_abc123", "title": "test"}, {"id": "ses_def456", "title": "test2"}]'
|
||||||
|
local ids=$(echo "$json" | python3 -c "import sys,json; sessions=json.load(sys.stdin); print(' '.join(s['id'] for s in sessions))" 2>/dev/null)
|
||||||
|
|
||||||
|
if [ "$ids" = "ses_abc123 ses_def456" ]; then
|
||||||
|
pass "JSON session list parsing extracts IDs correctly"
|
||||||
|
else
|
||||||
|
fail "JSON session list parsing" "ses_abc123 ses_def456" "$ids"
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# Test 5: Session ID format validation
|
||||||
|
test_session_id_format() {
|
||||||
|
local json='[{"id": "ses_2b4814406ffe3AxcpbrP7FknDr", "title": "test"}]'
|
||||||
|
local ids=$(echo "$json" | python3 -c "import sys,json; sessions=json.load(sys.stdin); print(' '.join(s['id'] for s in sessions))" 2>/dev/null)
|
||||||
|
|
||||||
|
if [[ "$ids" =~ ^ses_ ]]; then
|
||||||
|
pass "Session ID format is valid (starts with ses_)"
|
||||||
|
else
|
||||||
|
fail "Session ID format" "ses_xxx" "$ids"
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# Test 6: create_session accepts optional base session parameter
|
||||||
|
test_create_session_with_param() {
|
||||||
|
local base_id=$(get_base_session_id)
|
||||||
|
if [ -z "$base_id" ] || [ "$base_id" = "null" ]; then
|
||||||
|
skip "Base session not initialized - run 'kugetsu init' first"
|
||||||
|
return
|
||||||
|
fi
|
||||||
|
|
||||||
|
local result=$(create_session "$base_id")
|
||||||
|
if [ -n "$result" ] && [[ "$result" =~ ^ses_ ]]; then
|
||||||
|
pass "create_session accepts base session parameter"
|
||||||
|
else
|
||||||
|
fail "create_session accepts base session parameter" "ses_xxx" "$result"
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# Test 7: Verify session appears in opencode session list after creation
|
||||||
|
test_session_visible_in_list() {
|
||||||
|
local base_id=$(get_base_session_id)
|
||||||
|
if [ -z "$base_id" ] || [ "$base_id" = "null" ]; then
|
||||||
|
skip "Base session not initialized - run 'kugetsu init' first"
|
||||||
|
return
|
||||||
|
fi
|
||||||
|
|
||||||
|
local new_id=$(create_session "$base_id")
|
||||||
|
sleep 1
|
||||||
|
|
||||||
|
local all_sessions=$(opencode session list --format=json 2>/dev/null)
|
||||||
|
if echo "$all_sessions" | grep -q "$new_id"; then
|
||||||
|
pass "Created session appears in opencode session list"
|
||||||
|
else
|
||||||
|
fail "Created session appears in session list" "should contain $new_id" "$all_sessions"
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
skip() {
|
||||||
|
echo "SKIP: $1"
|
||||||
|
}
|
||||||
|
|
||||||
|
# Run tests sequentially
|
||||||
|
run_test "create_session requires base session" test_create_session_requires_base
|
||||||
|
run_test "create_session creates NEW session" test_create_session_is_new
|
||||||
|
run_test "create_session creates different sessions on multiple calls" test_create_session_multiple_calls
|
||||||
|
run_test "JSON session list parsing" test_session_json_parsing
|
||||||
|
run_test "Session ID format validation" test_session_id_format
|
||||||
|
run_test "create_session accepts optional base session parameter" test_create_session_with_param
|
||||||
|
run_test "Created session visible in opencode session list" test_session_visible_in_list
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
echo "=== Test Results ==="
|
||||||
|
echo "Passed: $PASS"
|
||||||
|
echo "Failed: $FAIL"
|
||||||
|
echo "Total: $RUN"
|
||||||
|
|
||||||
|
if [ $FAIL -eq 0 ]; then
|
||||||
|
echo "All tests passed!"
|
||||||
|
exit 0
|
||||||
|
else
|
||||||
|
echo "Some tests failed!"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
Reference in New Issue
Block a user