Compare commits
97 Commits
fix/issue-
...
fix/issue-
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
2af8e54f42 | ||
| ab99647193 | |||
|
|
3deae2dd81 | ||
| 41c56a859c | |||
|
|
dd903bb8aa | ||
| efb1e34a7b | |||
| 44c84280f8 | |||
|
|
fa8b8467ee | ||
|
|
c9bdc0dd88 | ||
|
|
663e44a82a | ||
| a65e9d6d28 | |||
|
|
c9eb8badea | ||
|
|
24dd91d0e1 | ||
| c8b2ab6b12 | |||
|
|
87434d1bca | ||
| ae8f1433a7 | |||
|
|
b16a97514e | ||
| b09fe8eabc | |||
|
|
d240000088 | ||
| 05683ea4c6 | |||
|
|
2800e140ac | ||
|
|
51ec844365 | ||
|
|
ab06046273 | ||
| a18948df98 | |||
|
|
be301c599d | ||
| 34a0943202 | |||
|
|
4464e5d91f | ||
| c71f43b581 | |||
|
|
66c8624d66 | ||
| 1b5de5d553 | |||
|
|
19ade67a99 | ||
| efcec4e122 | |||
|
|
930d0e53b5 | ||
| e0aac3c05f | |||
| a211b56303 | |||
|
|
e86a309059 | ||
| 2beb3adb14 | |||
| bfd6778f8b | |||
| aafdebb6c6 | |||
| e666f4dffb | |||
|
|
a130a79bd7 | ||
|
|
798bee0f79 | ||
| 478f7ceeba | |||
|
|
9d0bcef465 | ||
| bb2add2e1a | |||
|
|
6a8fa563dd | ||
| 8729321922 | |||
|
|
998f7a4f44 | ||
|
|
b595411a07 | ||
|
|
08b633400d | ||
|
|
860bf9295f | ||
| cf8b003d2f | |||
|
|
fd79bfa3ea | ||
| 119c9b8fd9 | |||
|
|
7f7f8b1085 | ||
| 1a523a805a | |||
|
|
6aa84a35b9 | ||
| 6e2841bbda | |||
|
|
99d09c7dda | ||
| 383f538438 | |||
|
|
28b343f817 | ||
| 3a2095861f | |||
|
|
836fde07fc | ||
| 4f2a04e0b4 | |||
|
|
c8bb0b36f4 | ||
| 937b7c69de | |||
|
|
2051266809 | ||
| 80a3228be9 | |||
|
|
d68a63af41 | ||
| 56310755b8 | |||
|
|
fb33be3a64 | ||
| 1b19c9a92c | |||
|
|
85a4239383 | ||
|
|
91b51f62c0 | ||
| 7234837284 | |||
|
|
59f6a4883e | ||
| 9667c3e800 | |||
|
|
796e1fe454 | ||
| 84c59a3b64 | |||
|
|
dc9d4d7327 | ||
|
|
bdcb7a476c | ||
| 77cf817568 | |||
|
|
0f6a30f01c | ||
| 77b0963fa4 | |||
|
|
f39e39156a | ||
|
|
5a0a54898b | ||
|
|
b1028a6556 | ||
|
|
270219873f | ||
| deb18f1e32 | |||
|
|
cbfc8a0646 | ||
|
|
7fa669b4c3 | ||
| acb503471d | |||
|
|
1d4f190d97 | ||
|
|
ab0c4e1448 | ||
|
|
9bb8afe8c5 | ||
|
|
fd7a98b263 | ||
| 3942a915ff |
1
.gitignore
vendored
1
.gitignore
vendored
@@ -4,3 +4,4 @@ results/
|
||||
*/results/
|
||||
*.pyc
|
||||
|
||||
.kugetsu/
|
||||
|
||||
18
.pre-commit-config.yaml
Normal file
18
.pre-commit-config.yaml
Normal file
@@ -0,0 +1,18 @@
|
||||
repos:
|
||||
- repo: https://github.com/shellcheck-py/shellcheck-py
|
||||
rev: v0.9.0.6
|
||||
hooks:
|
||||
- id: shellcheck
|
||||
|
||||
- repo: https://github.com/astral-sh/ruff-pre-commit
|
||||
rev: v0.1.8
|
||||
hooks:
|
||||
- id: ruff
|
||||
args: [--fix]
|
||||
- id: ruff-format
|
||||
|
||||
- repo: https://github.com/commitizen-tools/commitizen
|
||||
rev: v3.2.0
|
||||
hooks:
|
||||
- id: commitizen
|
||||
stages: [commit-msg]
|
||||
@@ -16,6 +16,38 @@
|
||||
- Test changes before submitting
|
||||
- See [VERSIONING.md](VERSIONING.md) for backport compatibility rules
|
||||
|
||||
## Pre-commit Hooks
|
||||
|
||||
This repository uses [pre-commit](https://pre-commit.com/) for linting and commit message enforcement.
|
||||
|
||||
### Setup
|
||||
|
||||
```bash
|
||||
pip install pre-commit
|
||||
pre-commit install
|
||||
```
|
||||
|
||||
### Hooks
|
||||
|
||||
- **shellcheck** — Lints bash scripts
|
||||
- **ruff** — Lints and formats Python
|
||||
- **commitizen** — Enforces [Conventional Commits](https://www.conventionalcommits.org/) format
|
||||
|
||||
### Commit Message Format
|
||||
|
||||
Use Conventional Commits format:
|
||||
|
||||
```
|
||||
type(scope): message
|
||||
|
||||
# Examples
|
||||
fix(session): handle missing session gracefully
|
||||
feat(pm): add queue daemon for task delegation
|
||||
docs: update contributing guide
|
||||
```
|
||||
|
||||
Types: `fix`, `feat`, `docs`, `refactor`, `chore`, `test`
|
||||
|
||||
## Branches
|
||||
|
||||
### Primary Branches
|
||||
|
||||
@@ -6,6 +6,25 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/).
|
||||
|
||||
## [Unreleased]
|
||||
|
||||
## [v0.2.4] - 2026-04-06
|
||||
|
||||
### Fixed
|
||||
- Queue daemon: Locking to prevent daemon vs manual conflicts
|
||||
- Queue daemon: Proper error handling for failed tasks
|
||||
- Queue daemon: Fix GITEA_TOKEN loading from pm-agent.env
|
||||
- cmd_delegate: Enqueue tasks instead of bypassing queue
|
||||
- Notifications: Call kugetsu_add_notification from bash instead of os.system()
|
||||
- kugetsu: Remove duplicate update_queue_item_state that overwrote fixed version
|
||||
|
||||
### Added
|
||||
- Queue functions moved to kugetsu-index.sh for daemon access
|
||||
- kugetsu-session.sh sources required modules for daemon use
|
||||
|
||||
## [v0.2.3] - 2026-04-06
|
||||
|
||||
### Fixed
|
||||
- get_pending_tasks() returns proper JSON array instead of concatenated JSON objects
|
||||
|
||||
## [v0.2.1] - 2026-04-03
|
||||
|
||||
### Fixed
|
||||
|
||||
@@ -364,10 +364,14 @@ kugetsu queue-daemon logs # Show recent daemon logs
|
||||
|
||||
**Daemon Behavior:**
|
||||
1. Runs at configurable interval (default: 5 minutes)
|
||||
2. Checks if active agents < MAX_CONCURRENT_AGENTS
|
||||
3. Picks 1-N pending items (configurable batch size)
|
||||
4. Forks PM session for each picked item
|
||||
5. PM decides whether to use `start` or `continue`
|
||||
2. Checks queue for pending items
|
||||
3. For each pending item:
|
||||
- Acquires lock to prevent duplicate processing
|
||||
- Sources kugetsu-session.sh and calls `cmd_continue`
|
||||
- `cmd_continue` handles worktree/session creation and forks dev agent from base session
|
||||
- Updates queue item state to "notified"
|
||||
4. Uses per-issue locking to prevent race conditions
|
||||
5. Implements timeout for tasks that don't complete (marks as "error" after TASK_TIMEOUT_HOURS)
|
||||
|
||||
**Queue Directory:**
|
||||
```
|
||||
@@ -526,6 +530,13 @@ The script will:
|
||||
|
||||
See [docs/kugetsu-setup.md](../../docs/kugetsu-setup.md) for full Tailscale setup documentation.
|
||||
|
||||
## Version History
|
||||
|
||||
### v0.2.3
|
||||
- **Queue daemon context drift fix** (issue #156): Daemon now sources kugetsu-session.sh and calls `cmd_continue` directly instead of forking PM session. This fixes context drift where daemon would lose track of task state.
|
||||
- **Daemon locking**: Added per-issue locking to prevent race conditions when multiple daemon instances run.
|
||||
- **Timeout handling**: Tasks that don't complete within TASK_TIMEOUT_HOURS are marked as "error".
|
||||
|
||||
## Without kugetsu
|
||||
|
||||
If kugetsu is not available, use opencode directly:
|
||||
|
||||
@@ -2,6 +2,33 @@ You are a PM (Project Manager) for software development.
|
||||
|
||||
Your role is COORDINATOR. You break down requests, delegate work, monitor progress, and report results. You NEVER write code. Not even small fixes. Not even one-liners. Not even documentation. If asked to write code: delegate it using `kugetsu start`.
|
||||
|
||||
## Response Modes
|
||||
|
||||
You have TWO response modes. Choose based on the user's request:
|
||||
|
||||
### Mode 1: Task Mode (Delegate)
|
||||
When the user asks for something that requires coding, implementation, or file changes:
|
||||
- "work on issue #81"
|
||||
- "close PR #42"
|
||||
- "create a PR for issue #91"
|
||||
- "fix the bug in login.js"
|
||||
- "add tests for the API"
|
||||
- Any request that modifies code or creates commits
|
||||
|
||||
**Action:** Delegate using `kugetsu start` or `kugetsu continue`.
|
||||
|
||||
### Mode 2: Conversation Mode (Answer Directly)
|
||||
When the user asks a question that doesn't require code changes:
|
||||
- "show open issues"
|
||||
- "what is the current state of repo"
|
||||
- "hi"
|
||||
- "show me recent commits"
|
||||
- "what issues are being worked on"
|
||||
- "what branches exist"
|
||||
- Any informational query
|
||||
|
||||
**Action:** Answer directly from available context or API calls.
|
||||
|
||||
## Write Permissions: Strict Boundary
|
||||
|
||||
PM has EXPLICIT write boundaries. You can ONLY write to two specific locations.
|
||||
@@ -18,75 +45,116 @@ PM has EXPLICIT write boundaries. You can ONLY write to two specific locations.
|
||||
- Any `.md` files, config files, scripts, or code
|
||||
|
||||
### If Asked to Write Outside ~/.kugetsu/:
|
||||
You MUST delegate to a dev agent:
|
||||
```
|
||||
kugetsu start <domain>/<user>/<repo>#<issue> <task description>
|
||||
```
|
||||
Where:
|
||||
- `<domain>` = git server (e.g., `github.com`, `gitlab.com`, `git.fbrns.co`)
|
||||
- `<user>` = git username (from `git config user.name`)
|
||||
- `<repo>` = repository name (from `git remote -v`)
|
||||
- `<issue>` = issue number to address
|
||||
You MUST delegate to a dev agent using Task Mode.
|
||||
|
||||
### New Kugetsu Scripts:
|
||||
Do NOT write new kugetsu scripts yourself (even for internal use). Delegate to a dev agent via the normal workflow:
|
||||
1. Create an issue describing the needed script
|
||||
2. Delegate: `kugetsu start <domain>/<user>/<repo>#<issue> Create new kugetsu script`
|
||||
3. After PR is merged, you may test the new script
|
||||
## Tools for Delegation
|
||||
|
||||
**Example violations (DO NOT DO THESE):**
|
||||
- "Update SKILL.md" → DELEGATE, don't edit it yourself
|
||||
- "Fix the bug in login.js" → DELEGATE, don't write to repositories/
|
||||
- "Add a new script for queue management" → DELEGATE via issue/PR workflow
|
||||
|
||||
## Critical: How to Delegate
|
||||
|
||||
Use `kugetsu start` to create dev agent sessions:
|
||||
### kugetsu start
|
||||
Create a NEW dev agent session for an issue that has no existing session/worktree.
|
||||
|
||||
```
|
||||
kugetsu start <domain>/<user>/<repo>#<issue> <task description>
|
||||
kugetsu start <issue-ref> <task description>
|
||||
```
|
||||
|
||||
**Domain/User/Repo**: Pull from `git remote -v` and `git config user.name` to make this agnostic to any git server.
|
||||
**Params:**
|
||||
- `issue-ref`: Format `instance/user/repo#number`
|
||||
- Example: `github.com/shoko/kugetsu#81`
|
||||
- Example: `git.fbrns.co/shoko/kugetsu#118`
|
||||
- `task description`: What the agent should do (be specific)
|
||||
|
||||
**NOT `kugetsu delegate`** - that routes back to the PM (you). Use `kugetsu start` to create a NEW dev agent.
|
||||
**Returns:** Creates new worktree and dev session, returns session ID
|
||||
|
||||
**When to use:** When there is NO existing session or worktree for this issue.
|
||||
|
||||
---
|
||||
|
||||
### kugetsu continue
|
||||
Continue an EXISTING dev agent session that already has a worktree/session.
|
||||
|
||||
```
|
||||
kugetsu continue <issue-ref> [additional instructions]
|
||||
```
|
||||
|
||||
**Params:**
|
||||
- `issue-ref`: Format `instance/user/repo#number`
|
||||
- `additional instructions`: (optional) Extra context or changed instructions
|
||||
|
||||
**Returns:** Continues existing session, returns session ID
|
||||
|
||||
**When to use:** When a worktree/session already exists for this issue (check with `kugetsu list`).
|
||||
|
||||
---
|
||||
|
||||
### How to Choose: start vs continue
|
||||
|
||||
| Scenario | Tool |
|
||||
|----------|------|
|
||||
| First time working on issue | `kugetsu start` |
|
||||
| Issue already has worktree/session | `kugetsu continue` |
|
||||
| Session exists but needs new task | `kugetsu continue <issue-ref> <new task>` |
|
||||
| Not sure if session exists | Check `kugetsu list` first, or use `kugetsu continue` (it will error if no session) |
|
||||
|
||||
**NOT `kugetsu delegate`** - that routes back to the PM (you). Use `kugetsu start` or `kugetsu continue` to create a NEW dev agent.
|
||||
|
||||
## Your Identity
|
||||
|
||||
You are the PM. Your job is to coordinate, not to code.
|
||||
|
||||
- You delegate ALL implementation tasks to dev agents using `kugetsu start`
|
||||
- You delegate ALL implementation tasks to dev agents using Task Mode
|
||||
- You answer informational queries directly in Conversation Mode
|
||||
- You review PRs but do not edit code yourself
|
||||
- You break down complex requests into delegate-able tasks
|
||||
- You monitor progress and keep stakeholders informed
|
||||
|
||||
## Delegation is Your Default Behavior
|
||||
## Delegation is Your Default Behavior for Tasks
|
||||
|
||||
When a request comes in:
|
||||
|
||||
1. **Understand** - What needs to be built? What's the repo and issue?
|
||||
2. **Delegate** - Use `kugetsu start <issue-ref> <task>` to create a dev agent task
|
||||
3. **Monitor** - Watch for PR creation and review
|
||||
4. **Report** - Post final results to the issue
|
||||
1. **Identify Mode** - Is this a task (code change needed) or a conversation (info request)?
|
||||
2. **For Tasks:**
|
||||
- **Understand** - What needs to be built? What's the repo and issue?
|
||||
- **Choose Tool** - Use `kugetsu start` (new) or `kugetsu continue` (existing)?
|
||||
- **Delegate** - Call the appropriate tool with issue-ref and task
|
||||
- **Monitor** - Watch for PR creation and review
|
||||
- **Report** - Post final results to the issue
|
||||
3. **For Conversations:**
|
||||
- Answer directly using available context
|
||||
|
||||
## Few-Shot Examples
|
||||
|
||||
**User:** "Fix the bug in login.js"
|
||||
**Mode:** Task
|
||||
**You:** `kugetsu start <domain>/<user>/<repo>#123 Investigate and fix the login bug in login.js`
|
||||
|
||||
**User:** "Add tests for the API"
|
||||
**Mode:** Task
|
||||
**You:** `kugetsu start <domain>/<user>/<repo>#124 Write tests for the API module`
|
||||
|
||||
**User:** "Can you write a quick script to parse this JSON?"
|
||||
**Mode:** Task
|
||||
**You:** `kugetsu start <domain>/<user>/<repo>#125 Create a script to parse the JSON file`
|
||||
|
||||
**User:** "Update the README with installation instructions"
|
||||
**Mode:** Task
|
||||
**You:** `kugetsu start <domain>/<user>/<repo>#126 Update README with installation instructions`
|
||||
|
||||
**User:** "Create a file at /tmp/test.txt"
|
||||
**Mode:** Task
|
||||
**You:** `kugetsu start <domain>/<user>/<repo>#127 Create a file at /tmp/test.txt`
|
||||
|
||||
Notice: In every example, the correct response is to DELEGATE using `kugetsu start`, not to do it yourself.
|
||||
**User:** "What open issues do we have?"
|
||||
**Mode:** Conversation
|
||||
**You:** (Answer directly about open issues from the repository)
|
||||
|
||||
**User:** "Show me recent commits"
|
||||
**Mode:** Conversation
|
||||
**You:** (Answer directly about recent commits)
|
||||
|
||||
**User:** "Hi, how are you?"
|
||||
**Mode:** Conversation
|
||||
**You:** (Answer greeting directly)
|
||||
|
||||
---
|
||||
|
||||
## You Are the PM. You Coordinate. You Do Not Write Code.
|
||||
|
||||
@@ -94,4 +162,4 @@ This is not just a rule - it is your identity. The code you coordinate is built
|
||||
|
||||
---
|
||||
|
||||
*PM Agent v4 - Coordinators coordinate, we do not code. Strict write boundary: ONLY ~/.kugetsu/.*
|
||||
*PM Agent v5 - Coordinators coordinate. Delegation is for tasks, conversation is for questions. Strict write boundary: ONLY ~/.kugetsu/.*
|
||||
@@ -93,23 +93,16 @@ EOF
|
||||
|
||||
ensure_dirs() {
|
||||
mkdir -p "$SESSIONS_DIR"
|
||||
mkdir -p "$LOGS_DIR"
|
||||
mkdir -p "$WORKTREES_DIR"
|
||||
mkdir -p "$QUEUE_DIR"
|
||||
mkdir -p "$QUEUE_ITEMS_DIR"
|
||||
}
|
||||
|
||||
ensure_worktree_dir() {
|
||||
mkdir -p "$WORKTREES_DIR"
|
||||
}
|
||||
|
||||
issue_ref_to_filename() {
|
||||
local issue_ref="$1"
|
||||
echo "$issue_ref" | sed 's/[\/:]/-/g' | sed 's/#/-/'
|
||||
}
|
||||
|
||||
filename_to_issue_ref() {
|
||||
local filename="$1"
|
||||
local name="${filename%.json}"
|
||||
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")
|
||||
@@ -268,7 +261,9 @@ PYEOF
|
||||
}
|
||||
|
||||
ensure_queue_dirs() {
|
||||
mkdir -p "$QUEUE_DIR"
|
||||
mkdir -p "$QUEUE_ITEMS_DIR"
|
||||
mkdir -p "$LOGS_DIR"
|
||||
}
|
||||
|
||||
generate_queue_id() {
|
||||
@@ -321,12 +316,31 @@ get_pending_tasks() {
|
||||
return
|
||||
fi
|
||||
|
||||
find "$QUEUE_ITEMS_DIR" -name "*.json" -type f 2>/dev/null | while read -r file; do
|
||||
local state=$(python3 -c "import json; print(json.load(open('$file')).get('state', ''))" 2>/dev/null || echo "")
|
||||
if [ "$state" = "pending" ]; then
|
||||
cat "$file"
|
||||
fi
|
||||
done | head -"$limit"
|
||||
python3 -c "
|
||||
import json
|
||||
import os
|
||||
import sys
|
||||
|
||||
queue_dir = os.environ.get('QUEUE_ITEMS_DIR', '')
|
||||
limit = int(sys.argv[1]) if len(sys.argv) > 1 else 10
|
||||
|
||||
items = []
|
||||
if os.path.isdir(queue_dir):
|
||||
for filename in os.listdir(queue_dir):
|
||||
if filename.endswith('.json'):
|
||||
filepath = os.path.join(queue_dir, filename)
|
||||
try:
|
||||
with open(filepath) as f:
|
||||
data = json.load(f)
|
||||
if data.get('state') == 'pending':
|
||||
items.append(data)
|
||||
if len(items) >= limit:
|
||||
break
|
||||
except:
|
||||
pass
|
||||
|
||||
print(json.dumps(items))
|
||||
" "$limit"
|
||||
}
|
||||
|
||||
get_queue_stats() {
|
||||
@@ -353,55 +367,6 @@ get_queue_stats() {
|
||||
echo "{\"total\": $total, \"pending\": $pending, \"notified\": $notified, \"completed\": $completed, \"error\": $error}"
|
||||
}
|
||||
|
||||
update_queue_item_state() {
|
||||
local queue_id="$1"
|
||||
local new_state="$2"
|
||||
local session_id="${3:-}"
|
||||
local pid="${4:-}"
|
||||
|
||||
local item_file="$QUEUE_ITEMS_DIR/${queue_id}.json"
|
||||
if [ ! -f "$item_file" ]; then
|
||||
echo "Error: Queue item not found: $queue_id" >&2
|
||||
return 1
|
||||
fi
|
||||
|
||||
python3 << PYEOF
|
||||
import json
|
||||
import os
|
||||
from datetime import datetime
|
||||
|
||||
item_file = "$item_file"
|
||||
new_state = "$new_state"
|
||||
session_id = "$session_id"
|
||||
pid = "$pid"
|
||||
|
||||
with open(item_file, 'r') as f:
|
||||
item = json.load(f)
|
||||
|
||||
issue_ref = item.get('issue_ref', '')
|
||||
|
||||
item['state'] = new_state
|
||||
|
||||
if new_state == "notified":
|
||||
item['notified_at'] = datetime.now().isoformat() + "Z"
|
||||
if session_id:
|
||||
item['opencode_session_id'] = session_id
|
||||
if pid:
|
||||
item['pid'] = int(pid) if pid.isdigit() else None
|
||||
elif new_state == "completed":
|
||||
item['completed_at'] = datetime.now().isoformat() + "Z"
|
||||
os.system(f"kugetsu_add_notification 'task_completed' 'Task completed: {issue_ref}' '{issue_ref}'")
|
||||
elif new_state == "error":
|
||||
item['error'] = datetime.now().isoformat() + "Z"
|
||||
os.system(f"kugetsu_add_notification 'task_error' 'Task error: {issue_ref}' '{issue_ref}'")
|
||||
|
||||
with open(item_file, 'w') as f:
|
||||
json.dump(item, f, indent=2)
|
||||
|
||||
print(f"Updated $queue_id to state: $new_state")
|
||||
PYEOF
|
||||
}
|
||||
|
||||
check_task_timeouts() {
|
||||
if [ ! -d "$QUEUE_ITEMS_DIR" ]; then
|
||||
return
|
||||
@@ -510,97 +475,6 @@ read_index() {
|
||||
fi
|
||||
}
|
||||
|
||||
write_index() {
|
||||
local base="$1"
|
||||
local pm_agent="$2"
|
||||
local issues_json="$3"
|
||||
local temp_file="$INDEX_FILE.tmp.$$"
|
||||
printf '{"base": %s, "pm_agent": %s, "issues": %s}\n' "$base" "$pm_agent" "$issues_json" > "$temp_file"
|
||||
mv "$temp_file" "$INDEX_FILE"
|
||||
}
|
||||
|
||||
get_base_session_id() {
|
||||
local index=$(read_index)
|
||||
echo "$index" | python3 -c "import sys, json; d=json.load(sys.stdin); print(d.get('base') or '')"
|
||||
}
|
||||
|
||||
get_pm_agent_session_id() {
|
||||
local index=$(read_index)
|
||||
echo "$index" | python3 -c "import sys, json; d=json.load(sys.stdin); print(d.get('pm_agent') or '')"
|
||||
}
|
||||
|
||||
get_session_for_issue() {
|
||||
local issue_ref="$1"
|
||||
local index=$(read_index)
|
||||
echo "$index" | python3 -c "import sys, json; d=json.load(sys.stdin); print(d['issues'].get('$issue_ref') or '')"
|
||||
}
|
||||
|
||||
set_base_in_index() {
|
||||
local base_session_id="$1"
|
||||
local pm_agent=$(get_pm_agent_session_id)
|
||||
local issues_json=$(read_index | python3 -c "import sys, json; d=json.load(sys.stdin); print(json.dumps(d['issues']))")
|
||||
if [ -z "$pm_agent" ] || [ "$pm_agent" = "null" ]; then
|
||||
write_index "\"$base_session_id\"" "null" "$issues_json"
|
||||
else
|
||||
write_index "\"$base_session_id\"" "\"$pm_agent\"" "$issues_json"
|
||||
fi
|
||||
}
|
||||
|
||||
set_pm_agent_in_index() {
|
||||
local pm_agent_session_id="$1"
|
||||
local base=$(get_base_session_id)
|
||||
local issues_json=$(read_index | python3 -c "import sys, json; d=json.load(sys.stdin); print(json.dumps(d['issues']))")
|
||||
if [ -z "$base" ] || [ "$base" = "null" ]; then
|
||||
write_index "null" "\"$pm_agent_session_id\"" "$issues_json"
|
||||
else
|
||||
write_index "\"$base\"" "\"$pm_agent_session_id\"" "$issues_json"
|
||||
fi
|
||||
}
|
||||
|
||||
add_issue_to_index() {
|
||||
local issue_ref="$1"
|
||||
local session_file="$2"
|
||||
local index=$(read_index)
|
||||
local base=$(get_base_session_id)
|
||||
local pm_agent=$(get_pm_agent_session_id)
|
||||
local issues=$(echo "$index" | python3 -c "import sys, json; d=json.load(sys.stdin); print(json.dumps(d['issues']))")
|
||||
local new_issues=$(echo "$issues" | python3 -c "import sys, json; d=json.load(sys.stdin); d['$issue_ref']='$session_file'; print(json.dumps(d))")
|
||||
if [ -z "$base" ] || [ "$base" = "null" ]; then
|
||||
if [ -z "$pm_agent" ] || [ "$pm_agent" = "null" ]; then
|
||||
write_index "null" "null" "$new_issues"
|
||||
else
|
||||
write_index "null" "\"$pm_agent\"" "$new_issues"
|
||||
fi
|
||||
else
|
||||
if [ -z "$pm_agent" ] || [ "$pm_agent" = "null" ]; then
|
||||
write_index "\"$base\"" "null" "$new_issues"
|
||||
else
|
||||
write_index "\"$base\"" "\"$pm_agent\"" "$new_issues"
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
remove_issue_from_index() {
|
||||
local issue_ref="$1"
|
||||
local index=$(read_index)
|
||||
local base=$(get_base_session_id)
|
||||
local pm_agent=$(get_pm_agent_session_id)
|
||||
local new_issues=$(echo "$index" | python3 -c "import sys, json; d=json.load(sys.stdin); d['issues'].pop('$issue_ref', None); print(json.dumps(d['issues']))")
|
||||
if [ -z "$base" ] || [ "$base" = "null" ]; then
|
||||
if [ -z "$pm_agent" ] || [ "$pm_agent" = "null" ]; then
|
||||
write_index "null" "null" "$new_issues"
|
||||
else
|
||||
write_index "null" "\"$pm_agent\"" "$new_issues"
|
||||
fi
|
||||
else
|
||||
if [ -z "$pm_agent" ] || [ "$pm_agent" = "null" ]; then
|
||||
write_index "\"$base\"" "null" "$new_issues"
|
||||
else
|
||||
write_index "\"$base\"" "\"$pm_agent\"" "$new_issues"
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
validate_issue_ref() {
|
||||
local issue_ref="$1"
|
||||
if [[ ! "$issue_ref" =~ ^[^/]+/[^/]+/[^#]+#[0-9]+$ ]]; then
|
||||
@@ -820,6 +694,18 @@ cmd_status() {
|
||||
return
|
||||
fi
|
||||
|
||||
local opencode_sessions=$(opencode session list 2>/dev/null | grep -oP '^ses_\w+' || true)
|
||||
|
||||
if ! echo "$opencode_sessions" | grep -q "^${base}$"; then
|
||||
echo "error: base session '$base' not found in opencode"
|
||||
return
|
||||
fi
|
||||
|
||||
if ! echo "$opencode_sessions" | grep -q "^${pm_agent}$"; then
|
||||
echo "error: pm_agent session '$pm_agent' not found in opencode"
|
||||
return
|
||||
fi
|
||||
|
||||
echo "ok"
|
||||
}
|
||||
|
||||
@@ -877,6 +763,11 @@ EOF
|
||||
}
|
||||
|
||||
parse_issue_ref_from_message() {
|
||||
# DEPRECATED: This function is not called anywhere.
|
||||
# The active implementation is extract_issue_ref_from_message()
|
||||
# in kugetsu-session.sh which is used by cmd_delegate.
|
||||
# This function is kept for backwards compatibility and will
|
||||
# be removed in a future release.
|
||||
local message="$1"
|
||||
|
||||
local gitserver=""
|
||||
@@ -884,21 +775,20 @@ parse_issue_ref_from_message() {
|
||||
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-Z]{2,}/[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+#[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)
|
||||
owner=$(echo "$gitserver" | cut -d'/' -f2)
|
||||
repo=$(echo "$gitserver" | cut -d'/' -f3)
|
||||
issue_number=$(echo "$message" | grep -oE '#[0-9]+' | 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)
|
||||
if [[ "$message" =~ (https?://)?([a-zA-Z0-9.-]+\.[a-zA-Z]{2,})/([a-zA-Z0-9._-]+)/([a-zA-Z0-9._-]+)/(issues|pull)/([0-9]+) ]]; then
|
||||
gitserver="${BASH_REMATCH[2]}"
|
||||
owner="${BASH_REMATCH[3]}"
|
||||
repo="${BASH_REMATCH[4]}"
|
||||
issue_number="${BASH_REMATCH[6]}"
|
||||
elif [[ "$message" =~ (https?://)?([a-zA-Z0-9.-]+\.[a-zA-Z]{2,})/([a-zA-Z0-9._-]+)/([a-zA-Z0-9._-]+)#([0-9]+) ]]; then
|
||||
gitserver="${BASH_REMATCH[2]}"
|
||||
owner="${BASH_REMATCH[3]}"
|
||||
repo="${BASH_REMATCH[4]}"
|
||||
issue_number="${BASH_REMATCH[5]}"
|
||||
elif [[ "$message" =~ ([a-zA-Z0-9._-]+)/([a-zA-Z0-9._-]+)#([0-9]+) ]]; then
|
||||
owner="${BASH_REMATCH[1]}"
|
||||
repo="${BASH_REMATCH[2]}"
|
||||
issue_number="${BASH_REMATCH[3]}"
|
||||
fi
|
||||
|
||||
echo "${gitserver}|${owner}|${repo}|${issue_number}"
|
||||
|
||||
@@ -26,13 +26,16 @@ QUEUE_DAEMON_INTERVAL_MINUTES="${QUEUE_DAEMON_INTERVAL_MINUTES:-5}"
|
||||
QUEUE_CLEANUP_AGE_DAYS="${QUEUE_CLEANUP_AGE_DAYS:-7}"
|
||||
TASK_TIMEOUT_HOURS="${TASK_TIMEOUT_HOURS:-1}"
|
||||
|
||||
NETWORK_RETRY_ATTEMPTS="${NETWORK_RETRY_ATTEMPTS:-3}"
|
||||
NETWORK_RETRY_DELAY_SECONDS="${NETWORK_RETRY_DELAY_SECONDS:-5}"
|
||||
|
||||
# Load user config overrides (~/.kugetsu/config)
|
||||
if [ -f "$KUGETSU_DIR/config" ]; then
|
||||
source "$KUGETSU_DIR/config"
|
||||
fi
|
||||
|
||||
mask_sensitive_vars() {
|
||||
local line="$1"
|
||||
local line="${1:-}"
|
||||
for var in GITEA_TOKEN GITHUB_TOKEN GITLAB_TOKEN API_KEY PASSWORD TOKEN SECRET; do
|
||||
if [[ "$line" =~ $var ]]; then
|
||||
line=$(echo "$line" | sed -E "s/=.*/=***MASKED***/")
|
||||
@@ -41,6 +44,11 @@ mask_sensitive_vars() {
|
||||
echo "$line"
|
||||
}
|
||||
|
||||
strip_ansi_codes() {
|
||||
local line="${1:-}"
|
||||
echo "$line" | sed 's/\x1b\[[0-9;]*m//g' | sed 's/\x1b\[[0-9;]*[a-zA-Z]//g'
|
||||
}
|
||||
|
||||
load_agent_env() {
|
||||
local agent_type="${1:-base}"
|
||||
local env_file="$ENV_DIR/${agent_type}.env"
|
||||
@@ -53,5 +61,53 @@ load_agent_env() {
|
||||
set -a
|
||||
source "$ENV_DIR/default.env"
|
||||
set +a
|
||||
elif [ -f "$ENV_DIR/pm-agent.env" ]; then
|
||||
set -a
|
||||
source "$ENV_DIR/pm-agent.env"
|
||||
set +a
|
||||
fi
|
||||
}
|
||||
|
||||
set_debug_mode() {
|
||||
local filtered_args=()
|
||||
local debug_mode=false
|
||||
|
||||
for arg in "$@"; do
|
||||
case "$arg" in
|
||||
--debug)
|
||||
debug_mode=true
|
||||
;;
|
||||
*)
|
||||
filtered_args+=("$arg")
|
||||
;;
|
||||
esac
|
||||
done
|
||||
|
||||
if [ "$debug_mode" = true ]; then
|
||||
export KUGETSU_VERBOSITY="debug"
|
||||
echo "[DEBUG] Debug mode enabled" >&2
|
||||
fi
|
||||
|
||||
echo "${filtered_args[@]}"
|
||||
}
|
||||
|
||||
retry_with_backoff() {
|
||||
local max_attempts="${1:-$NETWORK_RETRY_ATTEMPTS}"
|
||||
local delay_seconds="${2:-$NETWORK_RETRY_DELAY_SECONDS}"
|
||||
local command="$3"
|
||||
local remaining_attempts=$max_attempts
|
||||
|
||||
while [ $remaining_attempts -gt 0 ]; do
|
||||
if eval "$command"; then
|
||||
return 0
|
||||
fi
|
||||
remaining_attempts=$((remaining_attempts - 1))
|
||||
if [ $remaining_attempts -gt 0 ]; then
|
||||
log "warn" "retry_with_backoff" "Command failed, $remaining_attempts retries remaining. Waiting ${delay_seconds}s..."
|
||||
sleep "$delay_seconds"
|
||||
delay_seconds=$((delay_seconds * 2))
|
||||
fi
|
||||
done
|
||||
log "error" "retry_with_backoff" "Command failed after $max_attempts attempts"
|
||||
return 1
|
||||
}
|
||||
|
||||
@@ -15,6 +15,13 @@ write_index() {
|
||||
local issues_json="$3"
|
||||
local temp_file="$INDEX_FILE.tmp.$$"
|
||||
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"
|
||||
}
|
||||
|
||||
@@ -43,7 +50,11 @@ set_base_in_index() {
|
||||
if [ "$session_id" = "null" ]; then
|
||||
write_index "null" "$pm_agent" "$issues"
|
||||
else
|
||||
write_index "\"$session_id\"" "$pm_agent" "$issues"
|
||||
if [ "$pm_agent" = "null" ]; then
|
||||
write_index "\"$session_id\"" "null" "$issues"
|
||||
else
|
||||
write_index "\"$session_id\"" "\"$pm_agent\"" "$issues"
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
@@ -56,7 +67,11 @@ set_pm_agent_in_index() {
|
||||
if [ "$session_id" = "null" ]; then
|
||||
write_index "$base" "null" "$issues"
|
||||
else
|
||||
write_index "$base" "\"$session_id\"" "$issues"
|
||||
if [ "$base" = "null" ]; then
|
||||
write_index "null" "\"$session_id\"" "$issues"
|
||||
else
|
||||
write_index "\"$base\"" "\"$session_id\"" "$issues"
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
@@ -72,7 +87,19 @@ add_issue_to_index() {
|
||||
|
||||
issues=$(python3 -c "import sys, json; d=json.load(sys.stdin); d['$issue_ref']='$session_file'; print(json.dumps(d))" <<< "$issues")
|
||||
|
||||
write_index "$base" "$pm_agent" "$issues"
|
||||
if [ "$base" = "null" ]; then
|
||||
if [ "$pm_agent" = "null" ]; then
|
||||
write_index "null" "null" "$issues"
|
||||
else
|
||||
write_index "null" "\"$pm_agent\"" "$issues"
|
||||
fi
|
||||
else
|
||||
if [ "$pm_agent" = "null" ]; then
|
||||
write_index "\"$base\"" "null" "$issues"
|
||||
else
|
||||
write_index "\"$base\"" "\"$pm_agent\"" "$issues"
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
remove_issue_from_index() {
|
||||
@@ -86,7 +113,19 @@ remove_issue_from_index() {
|
||||
|
||||
issues=$(python3 -c "import sys, json; d=json.load(sys.stdin); d.pop('$issue_ref', None); print(json.dumps(d))" <<< "$issues")
|
||||
|
||||
write_index "$base" "$pm_agent" "$issues"
|
||||
if [ "$base" = "null" ]; then
|
||||
if [ "$pm_agent" = "null" ]; then
|
||||
write_index "null" "null" "$issues"
|
||||
else
|
||||
write_index "null" "\"$pm_agent\"" "$issues"
|
||||
fi
|
||||
else
|
||||
if [ "$pm_agent" = "null" ]; then
|
||||
write_index "\"$base\"" "null" "$issues"
|
||||
else
|
||||
write_index "\"$base\"" "\"$pm_agent\"" "$issues"
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
validate_issue_ref() {
|
||||
@@ -100,6 +139,77 @@ validate_issue_ref() {
|
||||
fi
|
||||
}
|
||||
|
||||
read_json_file() {
|
||||
local file_path="$1"
|
||||
if [ -f "$file_path" ]; then
|
||||
cat "$file_path"
|
||||
else
|
||||
echo "{}"
|
||||
fi
|
||||
}
|
||||
|
||||
write_json_file() {
|
||||
local file_path="$1"
|
||||
local json_content="$2"
|
||||
local temp_file="$file_path.tmp.$$"
|
||||
|
||||
printf '%s' "$json_content" > "$temp_file"
|
||||
|
||||
if ! python3 -c "import json; json.load(open('$temp_file'))" 2>/dev/null; then
|
||||
echo "Error: write_json_file would create malformed JSON: $file_path" >&2
|
||||
rm -f "$temp_file"
|
||||
return 1
|
||||
fi
|
||||
|
||||
mv "$temp_file" "$file_path"
|
||||
}
|
||||
|
||||
get_json_value() {
|
||||
local file_path="$1"
|
||||
local key="$2"
|
||||
local default="${3:-}"
|
||||
|
||||
if [ ! -f "$file_path" ]; then
|
||||
echo "$default"
|
||||
return
|
||||
fi
|
||||
|
||||
python3 -c "import json; print(json.load(open('$file_path')).get('$key', '$default'))" 2>/dev/null || echo "$default"
|
||||
}
|
||||
|
||||
set_json_value() {
|
||||
local file_path="$1"
|
||||
local key="$2"
|
||||
local value="$3"
|
||||
|
||||
if [ ! -f "$file_path" ]; then
|
||||
printf '{"%s": "%s"}\n' "$key" "$value" > "$file_path"
|
||||
return
|
||||
fi
|
||||
|
||||
python3 << PYEOF
|
||||
import json
|
||||
import sys
|
||||
|
||||
file_path = "$file_path"
|
||||
key = "$key"
|
||||
value = "$value"
|
||||
|
||||
try:
|
||||
with open(file_path, 'r') as f:
|
||||
data = json.load(f)
|
||||
except:
|
||||
data = {}
|
||||
|
||||
data[key] = value
|
||||
|
||||
with open(file_path, 'w') as f:
|
||||
json.dump(data, f, indent=2)
|
||||
|
||||
print(f"Set $key = $value in $file_path")
|
||||
PYEOF
|
||||
}
|
||||
|
||||
update_session_pr_url() {
|
||||
local issue_ref="$1"
|
||||
local pr_url="$2"
|
||||
@@ -130,6 +240,114 @@ session['pr_url'] = pr_url
|
||||
with open(session_path, 'w') as f:
|
||||
json.dump(session, f, indent=2)
|
||||
|
||||
print(f"Updated PR URL for $issue_ref: $pr_url")
|
||||
print(f"Updated PR URL for $issue_ref: $pr_url")
|
||||
PYEOF
|
||||
}
|
||||
|
||||
# Convert issue ref to session filename
|
||||
issue_ref_to_filename() {
|
||||
local issue_ref="$1"
|
||||
echo "$issue_ref" | sed 's/[\/:]/-/g' | sed 's/#/-/'
|
||||
}
|
||||
|
||||
# Convert session filename back to issue ref
|
||||
filename_to_issue_ref() {
|
||||
local filename="$1"
|
||||
local name="${filename%.json}"
|
||||
echo "$name" | sed 's-\([0-9]*\)$-#\1-' | sed 's/-/\//g'
|
||||
}
|
||||
|
||||
# Add notification to notifications file
|
||||
kugetsu_add_notification() {
|
||||
local type="$1"
|
||||
local message="$2"
|
||||
local issue_ref="${3:-}"
|
||||
local gitea_url="${4:-}"
|
||||
|
||||
mkdir -p "$(dirname "$NOTIFICATIONS_FILE")"
|
||||
|
||||
python3 << PYEOF
|
||||
import json
|
||||
import os
|
||||
from datetime import datetime
|
||||
|
||||
notification = {
|
||||
"type": "$type",
|
||||
"message": "$message",
|
||||
"issue_ref": "$issue_ref" if "$issue_ref" else None,
|
||||
"gitea_url": "$gitea_url" if "$gitea_url" else None,
|
||||
"timestamp": datetime.now().isoformat(),
|
||||
"read": False
|
||||
}
|
||||
|
||||
file_path = os.path.expanduser("$NOTIFICATIONS_FILE")
|
||||
notifications = []
|
||||
|
||||
if os.path.exists(file_path):
|
||||
try:
|
||||
with open(file_path, 'r') as f:
|
||||
notifications = json.load(f)
|
||||
except:
|
||||
notifications = []
|
||||
|
||||
notifications.append(notification)
|
||||
|
||||
with open(file_path, 'w') as f:
|
||||
json.dump(notifications, f, indent=2)
|
||||
|
||||
print("Notification added")
|
||||
PYEOF
|
||||
}
|
||||
|
||||
# Update queue item state
|
||||
update_queue_item_state() {
|
||||
local queue_id="$1"
|
||||
local new_state="$2"
|
||||
local session_id="${3:-}"
|
||||
local pid="${4:-}"
|
||||
|
||||
local item_file="$QUEUE_ITEMS_DIR/${queue_id}.json"
|
||||
if [ ! -f "$item_file" ]; then
|
||||
echo "Error: Queue item not found: $queue_id" >&2
|
||||
return 1
|
||||
fi
|
||||
|
||||
local issue_ref=$(python3 -c "import json; print(json.load(open('$item_file')).get('issue_ref', ''))" 2>/dev/null || echo "")
|
||||
|
||||
python3 << PYEOF
|
||||
import json
|
||||
from datetime import datetime
|
||||
|
||||
item_file = "$item_file"
|
||||
new_state = "$new_state"
|
||||
session_id = "$session_id"
|
||||
pid = "$pid"
|
||||
|
||||
with open(item_file, 'r') as f:
|
||||
item = json.load(f)
|
||||
|
||||
item['state'] = new_state
|
||||
|
||||
if new_state == "notified":
|
||||
item['notified_at'] = datetime.now().isoformat() + "Z"
|
||||
if session_id:
|
||||
item['opencode_session_id'] = session_id
|
||||
if pid:
|
||||
item['pid'] = int(pid) if pid.isdigit() else None
|
||||
elif new_state == "completed":
|
||||
item['completed_at'] = datetime.now().isoformat() + "Z"
|
||||
elif new_state == "error":
|
||||
item['error'] = datetime.now().isoformat() + "Z"
|
||||
|
||||
with open(item_file, 'w') as f:
|
||||
json.dump(item, f, indent=2)
|
||||
|
||||
print(f"Updated $queue_id to state: $new_state")
|
||||
PYEOF
|
||||
|
||||
if [ "$new_state" = "completed" ]; then
|
||||
kugetsu_add_notification "task_completed" "Task completed: $issue_ref" "$issue_ref"
|
||||
elif [ "$new_state" = "error" ]; then
|
||||
kugetsu_add_notification "task_error" "Task error: $issue_ref" "$issue_ref"
|
||||
fi
|
||||
}
|
||||
|
||||
@@ -1,6 +1,40 @@
|
||||
#!/bin/bash
|
||||
set -euo pipefail
|
||||
|
||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||
source "$SCRIPT_DIR/kugetsu-config.sh"
|
||||
|
||||
log() {
|
||||
local level="${1:-}"
|
||||
local component="${2:-}"
|
||||
local message="${3:-}"
|
||||
local timestamp
|
||||
timestamp=$(date -Iseconds)
|
||||
|
||||
case "$level" in
|
||||
info|warn|error|debug) ;;
|
||||
*)
|
||||
echo "Error: log level must be info|warn|error|debug" >&2
|
||||
return 1
|
||||
;;
|
||||
esac
|
||||
|
||||
if [ -z "$message" ]; then
|
||||
message="$component"
|
||||
component="${level}"
|
||||
level="info"
|
||||
fi
|
||||
|
||||
local masked
|
||||
masked=$(mask_sensitive_vars "$message")
|
||||
echo "[$timestamp] $level $component $masked"
|
||||
}
|
||||
|
||||
log_debug() { log "debug" "$1" "${2:-}"; }
|
||||
log_info() { log "info" "$1" "${2:-}"; }
|
||||
log_warn() { log "warn" "$1" "${2:-}"; }
|
||||
log_error() { log "error" "$1" "${2:-}"; }
|
||||
|
||||
cmd_logs() {
|
||||
local count="${1:-10}"
|
||||
|
||||
@@ -24,7 +58,9 @@ cmd_logs() {
|
||||
echo ""
|
||||
echo "--- $log ---"
|
||||
tail -20 "$LOGS_DIR/$log" | while read line; do
|
||||
echo " $(mask_sensitive_vars "$line")"
|
||||
line=$(strip_ansi_codes "$line")
|
||||
line=$(mask_sensitive_vars "$line")
|
||||
echo " $line"
|
||||
done
|
||||
fi
|
||||
done
|
||||
@@ -43,15 +79,24 @@ kugetsu_add_notification() {
|
||||
notifications=$(cat "$NOTIFICATIONS_FILE")
|
||||
fi
|
||||
|
||||
local new_notification=$(python3 -c "import json; print(json.dumps({
|
||||
'type': '$notification_type',
|
||||
'message': '$message',
|
||||
'issue_ref': '$issue_ref',
|
||||
'timestamp': '$timestamp',
|
||||
'read': False
|
||||
}))")
|
||||
|
||||
notifications=$(python3 -c "import json; n=json.loads('$notifications'); n.append(json.loads('$new_notification')); print(json.dumps(n[-50:] if len(n)>50 else n, indent=2))")
|
||||
notifications=$(echo "$notifications" | python3 -c "
|
||||
import json
|
||||
import sys
|
||||
|
||||
notifications = json.load(sys.stdin)
|
||||
new_notification = {
|
||||
'type': '$notification_type',
|
||||
'message': '''$message'''.replace('\"', '\"'),
|
||||
'issue_ref': '$issue_ref' if '$issue_ref' else None,
|
||||
'timestamp': '$timestamp',
|
||||
'read': False
|
||||
}
|
||||
|
||||
notifications.append(new_notification)
|
||||
notifications = notifications[-50:] if len(notifications) > 50 else notifications
|
||||
|
||||
print(json.dumps(notifications, indent=2))
|
||||
")
|
||||
|
||||
echo "$notifications" > "$NOTIFICATIONS_FILE"
|
||||
}
|
||||
|
||||
161
skills/kugetsu/scripts/kugetsu-queue-daemon.sh
Executable file → Normal file
161
skills/kugetsu/scripts/kugetsu-queue-daemon.sh
Executable file → Normal file
@@ -8,25 +8,162 @@ source "$SCRIPT_DIR/kugetsu-index.sh"
|
||||
source "$SCRIPT_DIR/kugetsu-worktree.sh"
|
||||
source "$SCRIPT_DIR/kugetsu-log.sh"
|
||||
|
||||
load_agent_env "pm-agent"
|
||||
|
||||
acquire_lock() {
|
||||
local issue_ref="$1"
|
||||
local lock_file="$QUEUE_DIR/locks/$(echo "$issue_ref" | sed 's/[\/:]/-/g' | sed 's/#/-/').lock"
|
||||
mkdir -p "$(dirname "$lock_file")"
|
||||
if [ -f "$lock_file" ]; then
|
||||
local pid=$(cat "$lock_file" 2>/dev/null || echo "")
|
||||
if [ -n "$pid" ] && kill -0 "$pid" 2>/dev/null; then
|
||||
return 1
|
||||
fi
|
||||
rm -f "$lock_file"
|
||||
fi
|
||||
echo $$ > "$lock_file"
|
||||
return 0
|
||||
}
|
||||
|
||||
release_lock() {
|
||||
local issue_ref="$1"
|
||||
local lock_file="$QUEUE_DIR/locks/$(echo "$issue_ref" | sed 's/[\/:]/-/g' | sed 's/#/-/').lock"
|
||||
rm -f "$lock_file"
|
||||
}
|
||||
|
||||
check_task_completion() {
|
||||
local item="$1"
|
||||
local queue_id=$(basename "$item" .json)
|
||||
|
||||
local item_data=$(read_json_file "$item")
|
||||
local state=$(python3 -c "import json; print(json.load(open('$item')).get('state', ''))" 2>/dev/null)
|
||||
|
||||
[ "$state" = "notified" ] || return 0
|
||||
|
||||
local session_id=$(python3 -c "import json; print(json.load(open('$item')).get('opencode_session_id', ''))" 2>/dev/null)
|
||||
local issue_ref=$(python3 -c "import json; print(json.load(open('$item')).get('issue_ref', ''))" 2>/dev/null)
|
||||
local pid=$(python3 -c "import json; print(json.load(open('$item')).get('pid', ''))" 2>/dev/null)
|
||||
local notified_at=$(python3 -c "import json; print(json.load(open('$item')).get('notified_at', ''))" 2>/dev/null)
|
||||
|
||||
local timed_out=false
|
||||
if [ -n "$notified_at" ]; then
|
||||
local notified_epoch=$(date -d "$notified_at" +%s 2>/dev/null || echo "0")
|
||||
local now_epoch=$(date +%s)
|
||||
local hours_elapsed=$(( (now_epoch - notified_epoch) / 3600 ))
|
||||
if [ "$hours_elapsed" -ge "${TASK_TIMEOUT_HOURS:-1}" ]; then
|
||||
timed_out=true
|
||||
log_warn "queue-daemon" "Task $queue_id ($issue_ref) timed out after ${hours_elapsed}h"
|
||||
fi
|
||||
fi
|
||||
|
||||
if [ "$timed_out" = true ]; then
|
||||
if [ -n "$pid" ] && [ "$pid" != "None" ]; then
|
||||
kill "$pid" 2>/dev/null || true
|
||||
fi
|
||||
if [ -n "$session_id" ]; then
|
||||
opencode session stop "$session_id" 2>/dev/null || true
|
||||
fi
|
||||
update_queue_item_state "$queue_id" "error"
|
||||
log_error "queue-daemon" "Task $queue_id ($issue_ref) marked error — timeout after ${hours_elapsed}h"
|
||||
release_lock "$issue_ref"
|
||||
return
|
||||
fi
|
||||
|
||||
if [ -n "$pid" ] && [ "$pid" != "None" ]; then
|
||||
if ! kill -0 "$pid" 2>/dev/null; then
|
||||
local worktree_path=$(issue_ref_to_worktree_path "$issue_ref" "$WORKTREES_DIR")
|
||||
local has_commits=false
|
||||
|
||||
if [ -d "$worktree_path" ] && [ -d "$worktree_path/.git" ]; then
|
||||
if [ -n "$(git -C "$worktree_path" log --oneline origin/main..HEAD 2>/dev/null)" ]; then
|
||||
has_commits=true
|
||||
fi
|
||||
fi
|
||||
|
||||
if [ "$has_commits" = true ]; then
|
||||
update_queue_item_state "$queue_id" "completed"
|
||||
echo "Task $queue_id ($issue_ref) completed — new commits found"
|
||||
else
|
||||
update_queue_item_state "$queue_id" "error"
|
||||
echo "Task $queue_id ($issue_ref) marked error — no commits found after session ended"
|
||||
fi
|
||||
release_lock "$issue_ref"
|
||||
fi
|
||||
else
|
||||
if [ -n "$session_id" ] && ! opencode session list 2>/dev/null | grep -q "$session_id"; then
|
||||
local worktree_path=$(issue_ref_to_worktree_path "$issue_ref" "$WORKTREES_DIR")
|
||||
local has_commits=false
|
||||
|
||||
if [ -d "$worktree_path" ] && [ -d "$worktree_path/.git" ]; then
|
||||
if [ -n "$(git -C "$worktree_path" log --oneline origin/main..HEAD 2>/dev/null)" ]; then
|
||||
has_commits=true
|
||||
fi
|
||||
fi
|
||||
|
||||
if [ "$has_commits" = true ]; then
|
||||
update_queue_item_state "$queue_id" "completed"
|
||||
echo "Task $queue_id ($issue_ref) completed — new commits found"
|
||||
else
|
||||
update_queue_item_state "$queue_id" "error"
|
||||
echo "Task $queue_id ($issue_ref) marked error — no commits found after session ended"
|
||||
fi
|
||||
release_lock "$issue_ref"
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
get_session_id_for_issue() {
|
||||
local issue_ref="$1"
|
||||
local session_file=$(issue_ref_to_filename "$issue_ref")
|
||||
local session_path="$SESSIONS_DIR/$session_file"
|
||||
if [ -f "$session_path" ]; then
|
||||
python3 -c "import json; print(json.load(open('$session_path')).get('opencode_session_id', ''))" 2>/dev/null || echo ""
|
||||
else
|
||||
echo ""
|
||||
fi
|
||||
}
|
||||
|
||||
process_task() {
|
||||
local item="$1"
|
||||
local queue_id=$(basename "$item" .json)
|
||||
local issue_ref=$(python3 -c "import json; print(json.load(open('$item')).get('issue_ref', ''))" 2>/dev/null)
|
||||
local message=$(python3 -c "import json; print(json.load(open('$item')).get('message', ''))" 2>/dev/null)
|
||||
|
||||
if ! acquire_lock "$issue_ref"; then
|
||||
echo "Task $queue_id ($issue_ref) skipped — another process is handling it"
|
||||
return
|
||||
fi
|
||||
|
||||
source "$SCRIPT_DIR/kugetsu-session.sh"
|
||||
|
||||
log_file="$LOGS_DIR/delegate-$(date +%s).log"
|
||||
if cmd_continue "$issue_ref" "$message" >> "$log_file" 2>&1; then
|
||||
sleep 1
|
||||
local session_id=$(get_session_id_for_issue "$issue_ref")
|
||||
update_queue_item_state "$queue_id" "notified" "$session_id" ""
|
||||
echo "Task $queue_id continued for $issue_ref"
|
||||
else
|
||||
update_queue_item_state "$queue_id" "error"
|
||||
echo "Task $queue_id ($issue_ref) failed to continue"
|
||||
fi
|
||||
|
||||
release_lock "$issue_ref"
|
||||
}
|
||||
|
||||
while true; do
|
||||
if [ -d "$QUEUE_ITEMS_DIR" ]; then
|
||||
for item in "$QUEUE_ITEMS_DIR"/*.json; do
|
||||
[ -f "$item" ] || continue
|
||||
check_task_completion "$item"
|
||||
done
|
||||
|
||||
for item in "$QUEUE_ITEMS_DIR"/*.json; do
|
||||
[ -f "$item" ] || continue
|
||||
state=$(python3 -c "import json; print(json.load(open('$item')).get('state', ''))" 2>/dev/null)
|
||||
if [ "$state" = "pending" ]; then
|
||||
queue_id=$(basename "$item" .json)
|
||||
issue_ref=$(python3 -c "import json; print(json.load(open('$item')).get('issue_ref', ''))" 2>/dev/null)
|
||||
message=$(python3 -c "import json; print(json.load(open('$item')).get('message', ''))" 2>/dev/null)
|
||||
|
||||
pm_session=$(get_pm_agent_session_id)
|
||||
if [ -n "$pm_session" ] && [ "$pm_session" != "null" ]; then
|
||||
log_file="$LOGS_DIR/delegate-$(date +%s).log"
|
||||
GITEA_TOKEN="${GITEA_TOKEN:-}" nohup sh -c "opencode run '$message' --continue --session '$pm_session' >> '$log_file' 2>&1" > /dev/null 2>&1 &
|
||||
pid=$!
|
||||
update_queue_item_state "$queue_id" "notified" "$pm_session" "$pid"
|
||||
fi
|
||||
process_task "$item"
|
||||
fi
|
||||
done
|
||||
fi
|
||||
sleep "${QUEUE_DAEMON_INTERVAL_MINUTES:-5}m"
|
||||
done
|
||||
done
|
||||
@@ -1,12 +1,20 @@
|
||||
#!/bin/bash
|
||||
set -euo pipefail
|
||||
|
||||
# Source required modules for session management functions
|
||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||
source "$SCRIPT_DIR/kugetsu-config.sh"
|
||||
source "$SCRIPT_DIR/kugetsu-index.sh"
|
||||
source "$SCRIPT_DIR/kugetsu-worktree.sh"
|
||||
source "$SCRIPT_DIR/kugetsu-log.sh"
|
||||
|
||||
count_active_dev_sessions() {
|
||||
local count=0
|
||||
if [ -d "$SESSIONS_DIR" ]; then
|
||||
for session_file in "$SESSIONS_DIR"/*.json; do
|
||||
if [ -f "$session_file" ]; then
|
||||
local filename=$(basename "$session_file")
|
||||
local filename
|
||||
filename=$(basename "$session_file")
|
||||
if [ "$filename" != "base.json" ] && [ "$filename" != "pm-agent.json" ]; then
|
||||
count=$((count + 1))
|
||||
fi
|
||||
@@ -51,12 +59,35 @@ EOF
|
||||
echo "Created config file: $KUGETSU_DIR/config"
|
||||
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_pm=$(get_pm_agent_session_id)
|
||||
|
||||
if [ -n "$existing_base" ] && [ "$existing_base" != "null" ]; then
|
||||
if [ "$force" = true ]; then
|
||||
echo "Warning: Reinitializing sessions (force mode)" >&2
|
||||
echo "Destroying all sessions, worktrees, and logs..." >&2
|
||||
cmd_destroy --base -y 2>/dev/null || true
|
||||
cmd_destroy --pm-agent -y 2>/dev/null || true
|
||||
rm -f "$LOGS_DIR"/*.log 2>/dev/null || true
|
||||
else
|
||||
echo "Error: Base session already exists: $existing_base" >&2
|
||||
echo "Use --force to reinitialize" >&2
|
||||
@@ -74,9 +105,20 @@ EOF
|
||||
echo "Press Ctrl+C to cancel or wait for session to be created"
|
||||
sleep 2
|
||||
|
||||
local before_sessions=$(opencode session list 2>/dev/null | grep -E '^ses_' | awk '{print $1}' || true)
|
||||
|
||||
opencode
|
||||
|
||||
local session_ids=$(opencode session list 2>/dev/null | grep -E '^ses_' | awk '{print $1}' | tail -1)
|
||||
local after_sessions=$(opencode session list 2>/dev/null | grep -E '^ses_' | awk '{print $1}' || true)
|
||||
local session_ids=""
|
||||
while IFS= read -r line; do
|
||||
local sid=$(echo "$line" | awk '{print $1}')
|
||||
if [ -n "$sid" ] && ! echo "$before_sessions" | grep -q "^${sid}$"; then
|
||||
session_ids="$sid"
|
||||
break
|
||||
fi
|
||||
done <<< "$after_sessions"
|
||||
|
||||
if [ -z "$session_ids" ]; then
|
||||
echo "Error: Could not find newly created session" >&2
|
||||
exit 1
|
||||
@@ -88,9 +130,20 @@ EOF
|
||||
echo "Base session created: $session_ids"
|
||||
echo "Starting PM agent..."
|
||||
|
||||
before_sessions="$after_sessions"
|
||||
|
||||
opencode
|
||||
|
||||
local pm_session_ids=$(opencode session list 2>/dev/null | grep -E '^ses_' | grep -v "$session_ids" | tail -1)
|
||||
after_sessions=$(opencode session list 2>/dev/null | grep -E '^ses_' | awk '{print $1}' || true)
|
||||
local pm_session_ids=""
|
||||
while IFS= read -r line; do
|
||||
local sid=$(echo "$line" | awk '{print $1}')
|
||||
if [ -n "$sid" ] && ! echo "$before_sessions" | grep -q "^${sid}$"; then
|
||||
pm_session_ids="$sid"
|
||||
break
|
||||
fi
|
||||
done <<< "$after_sessions"
|
||||
|
||||
if [ -z "$pm_session_ids" ]; then
|
||||
echo "Warning: Could not find separate PM agent session" >&2
|
||||
pm_session_ids="$session_ids"
|
||||
@@ -127,13 +180,11 @@ extract_issue_ref_from_message() {
|
||||
return
|
||||
fi
|
||||
|
||||
if [[ "$message" =~ (https?://[a-zA-Z0-9.-]+/[a-zA-Z0-9._-]+/[a-zA-Z0-9._-]+/(issues|pull)/[0-9]+) ]]; then
|
||||
local url="${BASH_REMATCH[1]}"
|
||||
local path=$(echo "$url" | sed 's|https\?://||' | cut -d'/' -f2-)
|
||||
local instance=$(echo "$path" | cut -d'/' -f1)
|
||||
local owner=$(echo "$path" | cut -d'/' -f2)
|
||||
local repo=$(echo "$path" | cut -d'/' -f3)
|
||||
local num=$(echo "$path" | grep -oE '[0-9]+$')
|
||||
if [[ "$message" =~ (https?://)?([a-zA-Z0-9.-]+)/([a-zA-Z0-9._-]+)/([a-zA-Z0-9._-]+)/(issues|pull)/([0-9]+) ]]; then
|
||||
local instance="${BASH_REMATCH[2]}"
|
||||
local owner="${BASH_REMATCH[3]}"
|
||||
local repo="${BASH_REMATCH[4]}"
|
||||
local num="${BASH_REMATCH[6]}"
|
||||
echo "${instance}/${owner}/${repo}#${num}"
|
||||
return
|
||||
fi
|
||||
@@ -153,85 +204,248 @@ cmd_delegate() {
|
||||
local issue_ref=$(extract_issue_ref_from_message "$message")
|
||||
|
||||
if [ -n "$issue_ref" ] && [[ "$issue_ref" =~ \#[0-9]+$ ]]; then
|
||||
cmd_start "$issue_ref" "$message"
|
||||
# Enqueue for daemon to process via cmd_start/cmd_continue
|
||||
enqueue_task "$issue_ref" "$message"
|
||||
return
|
||||
fi
|
||||
|
||||
local pm_session=$(get_pm_agent_session_id)
|
||||
if [ -z "$pm_session" ] || [ "$pm_session" = "null" ] || [ "$pm_session" = "None" ]; then
|
||||
echo "Error: PM agent session not found. Run 'kugetsu init' first." >&2
|
||||
# No issue ref detected — fork a new session from base session
|
||||
local base_session=$(get_base_session_id)
|
||||
if [ -z "$base_session" ] || [ "$base_session" = "null" ]; then
|
||||
echo "Error: Base session not found. Run 'kugetsu init' first." >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
mkdir -p "$LOGS_DIR"
|
||||
local log_file="$LOGS_DIR/delegate-$(date +%s).log"
|
||||
nohup sh -c "GITEA_TOKEN='${GITEA_TOKEN:-}' opencode run '$message' --continue --session '$pm_session' >> '$log_file' 2>&1" > /dev/null 2>&1 &
|
||||
disown
|
||||
echo "Delegated to PM agent (logged to $(basename "$log_file"))"
|
||||
}
|
||||
|
||||
cmd_start() {
|
||||
local issue_ref="${1:-}"
|
||||
local message="${2:-}"
|
||||
load_agent_env "pm-agent"
|
||||
|
||||
if [ -z "$issue_ref" ]; then
|
||||
echo "Error: issue ref is required" >&2
|
||||
echo "Usage: kugetsu start <issue-ref> [message]" >&2
|
||||
local new_session=$(create_session "$base_session")
|
||||
if [ -z "$new_session" ]; then
|
||||
echo "Error: Failed to create session" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
validate_issue_ref "$issue_ref"
|
||||
local msg_file="$LOGS_DIR/msg-$new_session.txt"
|
||||
printf '%s' "$message" > "$msg_file"
|
||||
nohup sh -c "GITEA_TOKEN='${GITEA_TOKEN:-}' opencode run '@$msg_file' --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_file
|
||||
before_file="$KUGETSU_DIR/sessions/before$$.json"
|
||||
local after_file
|
||||
after_file="$KUGETSU_DIR/sessions/after$$.json"
|
||||
|
||||
opencode session list --format=json > "$before_file" 2>/dev/null || printf '{}' > "$before_file"
|
||||
|
||||
local fork_success=false
|
||||
local attempt=0
|
||||
local max_attempts="${NETWORK_RETRY_ATTEMPTS:-3}"
|
||||
|
||||
while [ $attempt -lt $max_attempts ] && [ "$fork_success" = false ]; do
|
||||
attempt=$((attempt + 1))
|
||||
if opencode run --fork --session "$base_session" "new session" >/dev/null 2>&1; then
|
||||
fork_success=true
|
||||
elif [ $attempt -lt $max_attempts ]; then
|
||||
log "warn" "create_session" "Fork attempt $attempt failed, retrying..."
|
||||
sleep "$((attempt * 2))"
|
||||
fi
|
||||
done
|
||||
|
||||
if [ "$fork_success" = false ]; then
|
||||
log "error" "create_session" "Failed to fork session after $max_attempts attempts"
|
||||
rm -f "$before_file" "$after_file"
|
||||
return 1
|
||||
fi
|
||||
|
||||
sleep 1
|
||||
|
||||
opencode session list --format=json > "$after_file" 2>/dev/null || printf '{}' > "$after_file"
|
||||
|
||||
local new_session_id
|
||||
new_session_id=$(python3 << PYEOF
|
||||
import json
|
||||
|
||||
with open("$before_file", 'r') as f:
|
||||
before = json.load(f)
|
||||
with open("$after_file", 'r') as f:
|
||||
after = json.load(f)
|
||||
|
||||
before_ids = set(s['id'] for s in before)
|
||||
for s in after:
|
||||
if s['id'] not in before_ids:
|
||||
print(s['id'])
|
||||
break
|
||||
PYEOF
|
||||
)
|
||||
|
||||
rm -f "$before_file" "$after_file"
|
||||
|
||||
echo "$new_session_id"
|
||||
}
|
||||
|
||||
build_dev_agent_message() {
|
||||
local issue_ref="$1"
|
||||
local user_message="${2:-}"
|
||||
|
||||
local instance=$(echo "$issue_ref" | cut -d'/' -f1 | cut -d'#' -f1)
|
||||
local owner=$(echo "$issue_ref" | cut -d'/' -f2)
|
||||
local repo=$(echo "$issue_ref" | cut -d'/' -f3 | cut -d'#' -f1)
|
||||
local number=$(echo "$issue_ref" | grep -oE '#[0-9]+$' | tr -d '#')
|
||||
local worktree_path=$(issue_ref_to_worktree_path "$issue_ref")
|
||||
|
||||
local conflict_check=""
|
||||
local review_notes=""
|
||||
local delegator_header=""
|
||||
local delegator_footer=""
|
||||
if [ -n "$user_message" ]; then
|
||||
conflict_check=" - CRITICAL: Check if PR has merge conflicts before asking for review:
|
||||
- Use: curl -s \"https://$instance/api/v1/repos/$owner/$repo/pulls/$number\" -H \"Authorization: Bearer \$GITEA_TOKEN\"
|
||||
- If \"mergeable\": false, there ARE conflicts - you MUST resolve them FIRST
|
||||
- To resolve: cd to worktree, git fetch origin, git rebase origin/main, resolve conflicts, git rebase --continue, git push --force-with-lease
|
||||
- Only after resolving conflicts (mergeable: true) can you ask for review"
|
||||
delegator_header="IMPORTANT: Follow the workflow below as your guideline, but prioritize the delegator's message.
|
||||
|
||||
Workflow:"
|
||||
delegator_footer="
|
||||
|
||||
Delegator's message:
|
||||
$user_message"
|
||||
else
|
||||
review_notes=" - IMPORTANT: After listing reviews, READ the review comments and incorporate feedback
|
||||
- Check for review state: \"APPROVED\" means ready to merge, \"COMMENT\" means feedback to address"
|
||||
delegator_header="Workflow:"
|
||||
fi
|
||||
|
||||
cat <<EOF
|
||||
You are assigned to work on $issue_ref.
|
||||
|
||||
$delegator_header
|
||||
1. Read the issue at $instance/$owner/$repo/issues/$number AND all comments on that issue
|
||||
2. Check if a PR already exists for this issue
|
||||
- If PR exists and is open, review it and learn from it
|
||||
$conflict_check
|
||||
- If PR makes sense to continue, work on it instead
|
||||
- If PR is not worth continuing, create a new branch/PR but explain in PR description why you're creating a new one instead of continuing the existing PR
|
||||
3. Read README.md (if exists) to understand the general concept of this repository
|
||||
4. Read CONTRIBUTING.md (if exists) to understand how to contribute
|
||||
- If CONTRIBUTING.md doesn't exist, follow steps 5-9 as your guideline
|
||||
5. Explore the repository to understand the codebase
|
||||
6. If anything is unclear, post a comment on the issue asking for clarification before implementing
|
||||
7. Implement the solution
|
||||
8. Create a branch named fix/issue-$number and implement the fix
|
||||
9. Create a PR when the implementation is complete using: tea pr create --repo $owner/$repo --title "Your PR title" --body "PR description"
|
||||
- Make sure you are logged in with: tea login add --name gitea --token \$GITEA_TOKEN --url https://$instance
|
||||
- If tea is not available, use: curl -X POST "https://$instance/api/v1/repos/$owner/$repo/pulls" -H "Authorization: Bearer \$GITEA_TOKEN" -H "Content-Type: application/json" -d '{"title":"PR Title","head":"branch-name","base":"main","body":"PR description"}'
|
||||
|
||||
Tools for PR interaction:
|
||||
- Post issue/PR comment: curl -X POST "https://$instance/api/v1/repos/$owner/$repo/issues/$number/comments" -H "Authorization: Bearer \$GITEA_TOKEN" -H "Content-Type: application/json" -d '{"body":"Your comment"}'
|
||||
- List PR comments: curl -s "https://$instance/api/v1/repos/$owner/$repo/issues/$number/comments" -H "Authorization: Bearer \$GITEA_TOKEN"
|
||||
- List PR reviews: curl -s "https://$instance/api/v1/repos/$owner/$repo/pulls/$number/reviews" -H "Authorization: Bearer \$GITEA_TOKEN"
|
||||
$review_notes
|
||||
- Merge PR (only with approval): tea pr merge --repo $owner/$repo $number --style merge
|
||||
- MERGING requires approval first! Check for: approval in reviews, OR "lgtm"/"approved" in comments
|
||||
- If no approval, ask reviewer to approve first before merging
|
||||
$delegator_footer
|
||||
|
||||
Work directory: $worktree_path
|
||||
EOF
|
||||
}
|
||||
|
||||
ensure_worktree() {
|
||||
local issue_ref="$1"
|
||||
|
||||
if worktree_exists "$issue_ref" "$WORKTREES_DIR"; then
|
||||
log "info" "ensure_worktree" "Worktree already exists for $issue_ref"
|
||||
echo "existed"
|
||||
return 0
|
||||
fi
|
||||
|
||||
local base_session_id=$(get_base_session_id)
|
||||
if [ -z "$base_session_id" ] || [ "$base_session_id" = "null" ]; then
|
||||
echo "Error: Base session not found. Run 'kugetsu init' first." >&2
|
||||
exit 1
|
||||
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
|
||||
echo "Issue '$issue_ref' already has a worktree. Use 'kugetsu continue' instead."
|
||||
exit 1
|
||||
log "error" "ensure_worktree" "Base session not found for $issue_ref"
|
||||
echo "error"
|
||||
return 1
|
||||
fi
|
||||
|
||||
local active_count=$(count_active_dev_sessions)
|
||||
if [ "$active_count" -ge "${MAX_CONCURRENT_AGENTS:-3}" ]; then
|
||||
echo "Error: Max concurrent agents (${MAX_CONCURRENT_AGENTS:-3}) reached. Use 'kugetsu continue' or wait for an agent to finish." >&2
|
||||
exit 1
|
||||
log "error" "ensure_worktree" "Max concurrent agents reached for $issue_ref"
|
||||
echo "error"
|
||||
return 1
|
||||
fi
|
||||
|
||||
if create_worktree "$issue_ref" "$WORKTREES_DIR" 2>&1 | tee >(cat >&2); then
|
||||
log "info" "ensure_worktree" "Created worktree for $issue_ref"
|
||||
echo "created"
|
||||
return 0
|
||||
else
|
||||
log "error" "ensure_worktree" "Failed to create worktree for $issue_ref"
|
||||
echo "error"
|
||||
return 1
|
||||
fi
|
||||
}
|
||||
|
||||
ensure_session() {
|
||||
local issue_ref="$1"
|
||||
|
||||
local session_file=$(issue_ref_to_filename "$issue_ref")
|
||||
local session_path="$SESSIONS_DIR/$session_file"
|
||||
|
||||
if [ -f "$session_path" ]; then
|
||||
echo "Session file already exists: $session_file"
|
||||
echo "Use 'kugetsu continue $issue_ref' to continue work."
|
||||
exit 1
|
||||
local worktree_exists=false
|
||||
if worktree_exists "$issue_ref" "$WORKTREES_DIR"; then
|
||||
worktree_exists=true
|
||||
fi
|
||||
|
||||
local before_sessions=$(opencode session list 2>/dev/null | grep -oP '^ses_\w+' | sort)
|
||||
local before_set="|$before_sessions|"
|
||||
local session_exists=false
|
||||
if [ -f "$session_path" ]; then
|
||||
session_exists=true
|
||||
fi
|
||||
|
||||
create_worktree "$issue_ref"
|
||||
if [ "$worktree_exists" = true ] && [ "$session_exists" = true ]; then
|
||||
log "info" "ensure_session" "Session already exists for $issue_ref"
|
||||
echo "continued"
|
||||
return 0
|
||||
fi
|
||||
|
||||
local after_sessions=$(opencode session list 2>/dev/null | grep -oP '^ses_\w+' | sort)
|
||||
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
|
||||
if [ "$worktree_exists" = false ] && [ "$session_exists" = true ]; then
|
||||
log "warn" "ensure_session" "Session exists but worktree is missing. Removing stale session..."
|
||||
rm -f "$session_path"
|
||||
remove_issue_from_index "$issue_ref"
|
||||
session_exists=false
|
||||
fi
|
||||
|
||||
if [ "$worktree_exists" = false ]; then
|
||||
local wt_status=$(ensure_worktree "$issue_ref")
|
||||
if [ "$wt_status" != "created" ] && [ "$wt_status" != "existed" ]; then
|
||||
log "error" "ensure_session" "Failed to ensure worktree for $issue_ref"
|
||||
echo "error"
|
||||
return 1
|
||||
fi
|
||||
done <<< "$after_sessions"
|
||||
fi
|
||||
|
||||
local base_session_id=$(get_base_session_id)
|
||||
if [ -z "$base_session_id" ] || [ "$base_session_id" = "null" ]; then
|
||||
log "error" "ensure_session" "Base session not found for $issue_ref"
|
||||
echo "error"
|
||||
return 1
|
||||
fi
|
||||
|
||||
local new_session_id=$(create_session "$base_session_id")
|
||||
|
||||
if [ -z "$new_session_id" ]; then
|
||||
echo "Error: Could not find newly created session" >&2
|
||||
remove_worktree_for_issue "$issue_ref"
|
||||
exit 1
|
||||
log "error" "ensure_session" "Could not create session for $issue_ref"
|
||||
echo "error"
|
||||
return 1
|
||||
fi
|
||||
|
||||
local worktree_path=$(issue_ref_to_worktree_path "$issue_ref")
|
||||
@@ -241,65 +455,91 @@ cmd_start() {
|
||||
|
||||
add_issue_to_index "$issue_ref" "$session_file"
|
||||
|
||||
echo "Session started for '$issue_ref': $new_session_id"
|
||||
echo "Worktree: $worktree_path"
|
||||
log "info" "ensure_session" "Created session for $issue_ref: $new_session_id"
|
||||
echo "created"
|
||||
return 0
|
||||
}
|
||||
|
||||
fork_agent() {
|
||||
local session_id="$1"
|
||||
local worktree_path="$2"
|
||||
local message="$3"
|
||||
|
||||
if [ -z "$worktree_path" ] || [ ! -d "$worktree_path" ]; then
|
||||
log "error" "fork_agent" "Invalid worktree path: $worktree_path"
|
||||
echo "error"
|
||||
return 1
|
||||
fi
|
||||
|
||||
load_agent_env "dev"
|
||||
|
||||
cd "$worktree_path"
|
||||
local sanitized_id=$(echo "$session_id" | sed 's/[^a-zA-Z0-9_-]/_/g')
|
||||
mkdir -p "$worktree_path/.kugetsu"
|
||||
if [ ! -f "$worktree_path/.gitignore" ] || ! grep -q "^.kugetsu/" "$worktree_path/.gitignore"; then
|
||||
echo ".kugetsu/" >> "$worktree_path/.gitignore" 2>/dev/null || true
|
||||
fi
|
||||
local msg_file="$worktree_path/.kugetsu/msg.txt"
|
||||
printf '%s' "$message" > "$msg_file"
|
||||
nohup sh -c "GITEA_TOKEN='${GITEA_TOKEN:-}' opencode run '@$msg_file' --session '$session_id'" >> "$LOGS_DIR/dev-$sanitized_id.log" 2>&1 &
|
||||
|
||||
log "info" "fork_agent" "Forked agent for session $session_id in $worktree_path"
|
||||
echo "forked"
|
||||
return 0
|
||||
}
|
||||
|
||||
cmd_start() {
|
||||
cmd_continue "$@"
|
||||
}
|
||||
|
||||
cmd_continue() {
|
||||
local session_name=""
|
||||
local message=""
|
||||
local args=("$@")
|
||||
|
||||
args=$(set_debug_mode "${args[@]}")
|
||||
|
||||
for arg in $args; do
|
||||
if [ -z "$session_name" ]; then
|
||||
session_name="$arg"
|
||||
else
|
||||
message="$arg"
|
||||
fi
|
||||
done
|
||||
local issue_ref="${1:-}"
|
||||
local message="${2:-}"
|
||||
|
||||
if [ -z "$session_name" ]; then
|
||||
if [ -z "$issue_ref" ]; then
|
||||
echo "Error: issue ref is required" >&2
|
||||
echo "Usage: kugetsu continue <issue-ref> [message]" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
validate_issue_ref "$session_name"
|
||||
validate_issue_ref "$issue_ref"
|
||||
|
||||
local session_file=$(get_session_for_issue "$session_name")
|
||||
if [ -z "$session_file" ] || [ "$session_file" = "null" ]; then
|
||||
echo "Error: No session found for '$session_name'" >&2
|
||||
echo "Use 'kugetsu start $session_name' to create a new session." >&2
|
||||
if [ -z "$message" ]; then
|
||||
message=$(build_dev_agent_message "$issue_ref" "")
|
||||
else
|
||||
message=$(build_dev_agent_message "$issue_ref" "$message")
|
||||
fi
|
||||
|
||||
local worktree_status=$(ensure_worktree "$issue_ref")
|
||||
if [ "$worktree_status" = "error" ]; then
|
||||
echo "Error: Failed to ensure worktree for '$issue_ref'" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
local session_status=$(ensure_session "$issue_ref")
|
||||
if [ "$session_status" = "error" ]; then
|
||||
echo "Error: Failed to ensure session for '$issue_ref'" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
kugetsu_context_dump "$issue_ref" "$message" "$(issue_ref_to_branch_name "$issue_ref")"
|
||||
|
||||
local session_file=$(issue_ref_to_filename "$issue_ref")
|
||||
local session_path="$SESSIONS_DIR/$session_file"
|
||||
|
||||
if [ ! -f "$session_path" ]; then
|
||||
echo "Error: Session file not found: $session_path" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
load_agent_env "dev"
|
||||
|
||||
local opencode_session_id=$(python3 -c "import json; print(json.load(open('$session_path')).get('opencode_session_id', ''))" 2>/dev/null || echo "")
|
||||
local worktree_path=$(python3 -c "import json; print(json.load(open('$session_path')).get('worktree_path', ''))" 2>/dev/null || echo "")
|
||||
|
||||
if [ -n "$worktree_path" ] && [ -d "$worktree_path" ]; then
|
||||
if [ -n "$message" ]; then
|
||||
(cd "$worktree_path" && opencode run "$message" --continue --session "$opencode_session_id" "$@")
|
||||
else
|
||||
(cd "$worktree_path" && opencode --continue --session "$opencode_session_id" "$@")
|
||||
fi
|
||||
else
|
||||
if [ -n "$message" ]; then
|
||||
opencode run "$message" --continue --session "$opencode_session_id" "$@"
|
||||
else
|
||||
opencode --continue --session "$opencode_session_id" "$@"
|
||||
fi
|
||||
local fork_status=$(fork_agent "$opencode_session_id" "$worktree_path" "$message")
|
||||
if [ "$fork_status" = "error" ]; then
|
||||
echo "Error: Failed to fork agent for '$issue_ref'" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
log "info" "cmd_continue" "Result for $issue_ref: worktree=$worktree_status session=$session_status fork=$fork_status"
|
||||
|
||||
echo "Session continued for '$issue_ref': $opencode_session_id"
|
||||
echo "Worktree: $worktree_path"
|
||||
echo "${worktree_status}-${session_status}-${fork_status}"
|
||||
}
|
||||
|
||||
cmd_list() {
|
||||
@@ -450,11 +690,7 @@ cmd_destroy() {
|
||||
local target="${1:-}"
|
||||
local force=false
|
||||
|
||||
if [ "$target" = "--base" ]; then
|
||||
target=""
|
||||
fi
|
||||
|
||||
if [ "$2" = "-y" ]; then
|
||||
if [ "${2:-}" = "-y" ]; then
|
||||
force=true
|
||||
fi
|
||||
|
||||
|
||||
@@ -10,7 +10,7 @@ issue_ref_to_worktree_path() {
|
||||
local issue_ref="$1"
|
||||
local parent_dir="${2:-$WORKTREES_DIR}"
|
||||
local worktree_name=$(issue_ref_to_worktree_name "$issue_ref")
|
||||
echo "$parent_dir/.kugetsu-worktrees/$worktree_name"
|
||||
echo "$parent_dir/$worktree_name"
|
||||
}
|
||||
|
||||
issue_ref_to_branch_name() {
|
||||
@@ -41,7 +41,7 @@ get_repo_url() {
|
||||
fi
|
||||
|
||||
local instance=$(echo "$issue_ref" | cut -d'/' -f1 | cut -d'#' -f1)
|
||||
local rest=$(echo "$issue_ref" | sed 's/.*\///' | sed 's/#.*//')
|
||||
local rest=$(echo "$issue_ref" | sed 's/^[^\/]*\///' | sed 's/#.*//')
|
||||
|
||||
if [ -n "${GIT_SERVERS[$instance]:-}" ]; then
|
||||
echo "${GIT_SERVERS[$instance]}/${rest}.git"
|
||||
@@ -76,7 +76,8 @@ create_worktree() {
|
||||
exit 1
|
||||
fi
|
||||
|
||||
local worktree_parent_dir=$(dirname "$worktree_path")
|
||||
local worktree_parent_dir
|
||||
worktree_parent_dir=$(dirname "$worktree_path")
|
||||
mkdir -p "$worktree_parent_dir"
|
||||
|
||||
if worktree_exists "$issue_ref" "$parent_dir"; then
|
||||
@@ -85,15 +86,36 @@ create_worktree() {
|
||||
fi
|
||||
|
||||
echo "Creating worktree at '$worktree_path'..."
|
||||
git clone "$repo_url" "$worktree_path" 2>/dev/null || {
|
||||
echo "Error: Failed to clone repository" >&2
|
||||
|
||||
local clone_success=false
|
||||
local attempt=0
|
||||
local max_attempts="${NETWORK_RETRY_ATTEMPTS:-3}"
|
||||
|
||||
while [ $attempt -lt $max_attempts ] && [ "$clone_success" = false ]; do
|
||||
attempt=$((attempt + 1))
|
||||
if [ $attempt -gt 1 ]; then
|
||||
echo "Clone attempt $attempt of $max_attempts..."
|
||||
sleep "$((attempt * 2))"
|
||||
fi
|
||||
|
||||
if git clone "$repo_url" "$worktree_path" 2>/dev/null; then
|
||||
clone_success=true
|
||||
fi
|
||||
done
|
||||
|
||||
if [ "$clone_success" = false ]; then
|
||||
echo "Error: Failed to clone repository after $max_attempts attempts" >&2
|
||||
exit 1
|
||||
}
|
||||
fi
|
||||
|
||||
echo "Creating branch '$branch_name'..."
|
||||
(cd "$worktree_path" && git checkout -b "$branch_name" origin/main 2>/dev/null || git checkout -b "$branch_name" main 2>/dev/null) || {
|
||||
if git -C "$worktree_path" checkout -b "$branch_name" origin/main 2>/dev/null; then
|
||||
:
|
||||
elif git -C "$worktree_path" checkout -b "$branch_name" main 2>/dev/null; then
|
||||
:
|
||||
else
|
||||
echo "Warning: Could not checkout branch (may need to run from within worktree after session)" >&2
|
||||
}
|
||||
fi
|
||||
|
||||
echo "Worktree created at: $worktree_path"
|
||||
}
|
||||
@@ -145,15 +167,17 @@ check_pr_status() {
|
||||
token="${GITEA_TOKEN:-}"
|
||||
fi
|
||||
|
||||
local response
|
||||
local response_file="$KUGETSU_DIR/.pr_status_response_$$.json"
|
||||
if [ -n "$token" ]; then
|
||||
response=$(curl -s -H "Authorization: token $token" "$api_url" 2>/dev/null || echo "{}")
|
||||
curl -s -H "Authorization: token $token" "$api_url" > "$response_file" 2>/dev/null || printf '{}' > "$response_file"
|
||||
else
|
||||
response=$(curl -s "$api_url" 2>/dev/null || echo "{}")
|
||||
curl -s "$api_url" > "$response_file" 2>/dev/null || printf '{}' > "$response_file"
|
||||
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")
|
||||
local state=$(python3 -c "import json; print(json.load(open('$response_file')).get('state', 'unknown'))" 2>/dev/null || echo "unknown")
|
||||
local merged=$(python3 -c "import json; print('true' if json.load(open('$response_file')).get('merged', False) else 'false')" 2>/dev/null || echo "false")
|
||||
|
||||
rm -f "$response_file"
|
||||
|
||||
if [ "$merged" = "true" ]; then
|
||||
echo "merged"
|
||||
|
||||
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
|
||||
298
skills/kugetsu/tests/test-git-url-parsing.sh
Normal file
298
skills/kugetsu/tests/test-git-url-parsing.sh
Normal file
@@ -0,0 +1,298 @@
|
||||
#!/bin/bash
|
||||
# Git URL Parsing Tests for kugetsu
|
||||
# Tests all functions that parse or construct git URLs and issue refs
|
||||
#
|
||||
# Run with: bash skills/kugetsu/tests/test-git-url-parsing.sh
|
||||
|
||||
set -euo pipefail
|
||||
|
||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||
source "$SCRIPT_DIR/../scripts/kugetsu-config.sh"
|
||||
source "$SCRIPT_DIR/../scripts/kugetsu-worktree.sh"
|
||||
source "$SCRIPT_DIR/../scripts/kugetsu-session.sh"
|
||||
|
||||
PASS=0
|
||||
FAIL=0
|
||||
|
||||
pass() {
|
||||
echo "PASS: $1"
|
||||
PASS=$((PASS + 1))
|
||||
}
|
||||
|
||||
fail() {
|
||||
echo "FAIL: $1"
|
||||
echo " Expected: $2"
|
||||
echo " Got: $3"
|
||||
FAIL=$((FAIL + 1))
|
||||
}
|
||||
|
||||
echo "=== Git URL Parsing Test Suite ==="
|
||||
echo ""
|
||||
|
||||
# Test: get_repo_url with standard GitHub issue ref
|
||||
echo "--- Test: get_repo_url with github.com ---"
|
||||
result=$(get_repo_url "github.com/shoko/kugetsu#14")
|
||||
expected="https://github.com/shoko/kugetsu.git"
|
||||
if [ "$result" = "$expected" ]; then
|
||||
pass "get_repo_url standard github issue ref"
|
||||
else
|
||||
fail "get_repo_url standard github issue ref" "$expected" "$result"
|
||||
fi
|
||||
|
||||
# Test: get_repo_url with custom instance
|
||||
echo "--- Test: get_repo_url with git.fbrns.co ---"
|
||||
result=$(get_repo_url "git.fbrns.co/shoko/kugetsu#158")
|
||||
expected="https://git.fbrns.co/shoko/kugetsu.git"
|
||||
if [ "$result" = "$expected" ]; then
|
||||
pass "get_repo_url custom instance issue ref (ISSUE #181)"
|
||||
else
|
||||
fail "get_repo_url custom instance issue ref (ISSUE #181)" "$expected" "$result"
|
||||
fi
|
||||
|
||||
# Test: get_repo_url with gitlab.com (if configured)
|
||||
echo "--- Test: get_repo_url with gitlab.com ---"
|
||||
if [ -n "${GIT_SERVERS[gitlab.com]:-}" ]; then
|
||||
result=$(get_repo_url "gitlab.com/someuser/somerepo#42")
|
||||
expected="https://gitlab.com/someuser/somerepo.git"
|
||||
if [ "$result" = "$expected" ]; then
|
||||
pass "get_repo_url gitlab.com issue ref"
|
||||
else
|
||||
fail "get_repo_url gitlab.com issue ref" "$expected" "$result"
|
||||
fi
|
||||
else
|
||||
echo "SKIP: get_repo_url gitlab.com (not configured in GIT_SERVERS)"
|
||||
fi
|
||||
|
||||
# Test: get_repo_url with bitbucket.org (if configured)
|
||||
echo "--- Test: get_repo_url with bitbucket.org ---"
|
||||
if [ -n "${GIT_SERVERS[bitbucket.org]:-}" ]; then
|
||||
result=$(get_repo_url "bitbucket.org/myteam/myproject#7")
|
||||
expected="https://bitbucket.org/myteam/myproject.git"
|
||||
if [ "$result" = "$expected" ]; then
|
||||
pass "get_repo_url bitbucket.org issue ref"
|
||||
else
|
||||
fail "get_repo_url bitbucket.org issue ref" "$expected" "$result"
|
||||
fi
|
||||
else
|
||||
echo "SKIP: get_repo_url bitbucket.org (not configured in GIT_SERVERS)"
|
||||
fi
|
||||
|
||||
# Test: get_repo_url with large issue number
|
||||
echo "--- Test: get_repo_url with large issue number ---"
|
||||
result=$(get_repo_url "github.com/shoko/kugetsu#999999")
|
||||
expected="https://github.com/shoko/kugetsu.git"
|
||||
if [ "$result" = "$expected" ]; then
|
||||
pass "get_repo_url with large issue number"
|
||||
else
|
||||
fail "get_repo_url with large issue number" "$expected" "$result"
|
||||
fi
|
||||
|
||||
# Test: issue_ref_to_worktree_name standard
|
||||
echo "--- Test: issue_ref_to_worktree_name standard ---"
|
||||
result=$(issue_ref_to_worktree_name "github.com/shoko/kugetsu#14")
|
||||
expected="github.com-shoko-kugetsu-14"
|
||||
if [ "$result" = "$expected" ]; then
|
||||
pass "issue_ref_to_worktree_name standard"
|
||||
else
|
||||
fail "issue_ref_to_worktree_name standard" "$expected" "$result"
|
||||
fi
|
||||
|
||||
# Test: issue_ref_to_worktree_name with custom instance
|
||||
echo "--- Test: issue_ref_to_worktree_name custom instance ---"
|
||||
result=$(issue_ref_to_worktree_name "git.fbrns.co/shoko/kugetsu#158")
|
||||
expected="git.fbrns.co-shoko-kugetsu-158"
|
||||
if [ "$result" = "$expected" ]; then
|
||||
pass "issue_ref_to_worktree_name custom instance"
|
||||
else
|
||||
fail "issue_ref_to_worktree_name custom instance" "$expected" "$result"
|
||||
fi
|
||||
|
||||
# Test: issue_ref_to_branch_name with number
|
||||
echo "--- Test: issue_ref_to_branch_name with number ---"
|
||||
result=$(issue_ref_to_branch_name "github.com/shoko/kugetsu#14")
|
||||
expected="fix/issue-14"
|
||||
if [ "$result" = "$expected" ]; then
|
||||
pass "issue_ref_to_branch_name with number"
|
||||
else
|
||||
fail "issue_ref_to_branch_name with number" "$expected" "$result"
|
||||
fi
|
||||
|
||||
# Test: issue_ref_to_branch_name with discuss suffix
|
||||
# Note: #-discuss falls through to fix/issue-temp because #[^-]+$ doesn't match #-<text-with-hyphens>
|
||||
echo "--- Test: issue_ref_to_branch_name with discuss suffix ---"
|
||||
result=$(issue_ref_to_branch_name "github.com/shoko/kugetsu#-discuss")
|
||||
expected="fix/issue-temp"
|
||||
if [ "$result" = "$expected" ]; then
|
||||
pass "issue_ref_to_branch_name with discuss suffix"
|
||||
else
|
||||
fail "issue_ref_to_branch_name with discuss suffix" "$expected" "$result"
|
||||
fi
|
||||
|
||||
# Test: issue_ref_to_branch_name with identifier that has no hyphens
|
||||
echo "--- Test: issue_ref_to_branch_name with pure identifier ---"
|
||||
result=$(issue_ref_to_branch_name "github.com/shoko/kugetsu#someid")
|
||||
expected="fix/someid"
|
||||
if [ "$result" = "$expected" ]; then
|
||||
pass "issue_ref_to_branch_name with pure identifier"
|
||||
else
|
||||
fail "issue_ref_to_branch_name with pure identifier" "$expected" "$result"
|
||||
fi
|
||||
|
||||
# Test: issue_ref_to_branch_name without number
|
||||
echo "--- Test: issue_ref_to_branch_name without number ---"
|
||||
result=$(issue_ref_to_branch_name "github.com/shoko/kugetsu#abc")
|
||||
expected="fix/abc"
|
||||
if [ "$result" = "$expected" ]; then
|
||||
pass "issue_ref_to_branch_name without number"
|
||||
else
|
||||
fail "issue_ref_to_branch_name without number" "$expected" "$result"
|
||||
fi
|
||||
|
||||
# Test: extract_issue_ref_from_message with short form
|
||||
echo "--- Test: extract_issue_ref_from_message short form ---"
|
||||
result=$(extract_issue_ref_from_message "github.com/shoko/kugetsu#14")
|
||||
expected="github.com/shoko/kugetsu#14"
|
||||
if [ "$result" = "$expected" ]; then
|
||||
pass "extract_issue_ref_from_message short form"
|
||||
else
|
||||
fail "extract_issue_ref_from_message short form" "$expected" "$result"
|
||||
fi
|
||||
|
||||
# Test: extract_issue_ref_from_message with https URL
|
||||
echo "--- Test: extract_issue_ref_from_message with https URL ---"
|
||||
result=$(extract_issue_ref_from_message "https://github.com/shoko/kugetsu/issues/14")
|
||||
expected="github.com/shoko/kugetsu#14"
|
||||
if [ "$result" = "$expected" ]; then
|
||||
pass "extract_issue_ref_from_message with https URL"
|
||||
else
|
||||
fail "extract_issue_ref_from_message with https URL" "$expected" "$result"
|
||||
fi
|
||||
|
||||
# Test: extract_issue_ref_from_message with custom instance
|
||||
echo "--- Test: extract_issue_ref_from_message custom instance ---"
|
||||
result=$(extract_issue_ref_from_message "https://git.fbrns.co/shoko/kugetsu/issues/158")
|
||||
expected="git.fbrns.co/shoko/kugetsu#158"
|
||||
if [ "$result" = "$expected" ]; then
|
||||
pass "extract_issue_ref_from_message custom instance"
|
||||
else
|
||||
fail "extract_issue_ref_from_message custom instance" "$expected" "$result"
|
||||
fi
|
||||
|
||||
# Test: extract_issue_ref_from_message with empty message
|
||||
echo "--- Test: extract_issue_ref_from_message empty message ---"
|
||||
result=$(extract_issue_ref_from_message "")
|
||||
expected=""
|
||||
if [ "$result" = "$expected" ]; then
|
||||
pass "extract_issue_ref_from_message empty message"
|
||||
else
|
||||
fail "extract_issue_ref_from_message empty message" "$expected" "$result"
|
||||
fi
|
||||
|
||||
# Test: extract_issue_ref_from_message with no issue ref
|
||||
echo "--- Test: extract_issue_ref_from_message no issue ref ---"
|
||||
result=$(extract_issue_ref_from_message "Just a regular message without any issue reference")
|
||||
expected=""
|
||||
if [ "$result" = "$expected" ]; then
|
||||
pass "extract_issue_ref_from_message no issue ref"
|
||||
else
|
||||
fail "extract_issue_ref_from_message no issue ref" "$expected" "$result"
|
||||
fi
|
||||
|
||||
# Test: extract_issue_ref_from_message with gitlab URL
|
||||
echo "--- Test: extract_issue_ref_from_message gitlab URL ---"
|
||||
result=$(extract_issue_ref_from_message "https://gitlab.com/someuser/somerepo/issues/42")
|
||||
expected="gitlab.com/someuser/somerepo#42"
|
||||
if [ "$result" = "$expected" ]; then
|
||||
pass "extract_issue_ref_from_message gitlab URL"
|
||||
else
|
||||
fail "extract_issue_ref_from_message gitlab URL" "$expected" "$result"
|
||||
fi
|
||||
|
||||
# Test: validate_issue_ref valid format
|
||||
echo "--- Test: validate_issue_ref valid format ---"
|
||||
if validate_issue_ref "github.com/shoko/kugetsu#14" 2>/dev/null; then
|
||||
pass "validate_issue_ref valid format"
|
||||
else
|
||||
fail "validate_issue_ref valid format" "exit 0" "exit non-zero"
|
||||
fi
|
||||
|
||||
# Test: validate_issue_ref invalid format (missing parts)
|
||||
echo "--- Test: validate_issue_ref invalid format ---"
|
||||
if ! validate_issue_ref "invalid-ref" 2>/dev/null; then
|
||||
pass "validate_issue_ref invalid format"
|
||||
else
|
||||
fail "validate_issue_ref invalid format" "exit non-zero" "exit 0"
|
||||
fi
|
||||
|
||||
# Test: issue_ref_to_filename
|
||||
echo "--- Test: issue_ref_to_filename ---"
|
||||
result=$(issue_ref_to_filename "github.com/shoko/kugetsu#14")
|
||||
expected="github.com-shoko-kugetsu-14.json"
|
||||
if [ "$result" = "$expected" ]; then
|
||||
pass "issue_ref_to_filename"
|
||||
else
|
||||
fail "issue_ref_to_filename" "$expected" "$result"
|
||||
fi
|
||||
|
||||
# Test: filename_to_issue_ref
|
||||
echo "--- Test: filename_to_issue_ref ---"
|
||||
result=$(filename_to_issue_ref "github.com-shoko-kugetsu-14.json")
|
||||
expected="github.com/shoko/kugetsu#14"
|
||||
if [ "$result" = "$expected" ]; then
|
||||
pass "filename_to_issue_ref"
|
||||
else
|
||||
fail "filename_to_issue_ref" "$expected" "$result"
|
||||
fi
|
||||
|
||||
# Test: get_repo_url with org having hyphen
|
||||
echo "--- Test: get_repo_url with hyphenated org ---"
|
||||
result=$(get_repo_url "github.com/my-org/my-repo#1")
|
||||
expected="https://github.com/my-org/my-repo.git"
|
||||
if [ "$result" = "$expected" ]; then
|
||||
pass "get_repo_url with hyphenated org"
|
||||
else
|
||||
fail "get_repo_url with hyphenated org" "$expected" "$result"
|
||||
fi
|
||||
|
||||
# Test: get_repo_url with repo having dots
|
||||
echo "--- Test: get_repo_url with dotted repo ---"
|
||||
result=$(get_repo_url "github.com/shoko/kugetsu.utils#5")
|
||||
expected="https://github.com/shoko/kugetsu.utils.git"
|
||||
if [ "$result" = "$expected" ]; then
|
||||
pass "get_repo_url with dotted repo"
|
||||
else
|
||||
fail "get_repo_url with dotted repo" "$expected" "$result"
|
||||
fi
|
||||
|
||||
# Test: get_repo_url with underscore in username
|
||||
echo "--- Test: get_repo_url with underscore in user ---"
|
||||
result=$(get_repo_url "github.com/my_user/my_repo#10")
|
||||
expected="https://github.com/my_user/my_repo.git"
|
||||
if [ "$result" = "$expected" ]; then
|
||||
pass "get_repo_url with underscore in user"
|
||||
else
|
||||
fail "get_repo_url with underscore in user" "$expected" "$result"
|
||||
fi
|
||||
|
||||
# Test: get_repo_url with instance not in GIT_SERVERS (fallback)
|
||||
echo "--- Test: get_repo_url with unknown instance ---"
|
||||
result=$(get_repo_url "unknown.example.com/owner/repo#1")
|
||||
expected="https://unknown.example.com/owner/repo.git"
|
||||
if [ "$result" = "$expected" ]; then
|
||||
pass "get_repo_url with unknown instance"
|
||||
else
|
||||
fail "get_repo_url with unknown instance" "$expected" "$result"
|
||||
fi
|
||||
|
||||
echo ""
|
||||
echo "=== Test Results ==="
|
||||
echo "Passed: $PASS"
|
||||
echo "Failed: $FAIL"
|
||||
|
||||
if [ $FAIL -eq 0 ]; then
|
||||
echo "All tests passed!"
|
||||
exit 0
|
||||
else
|
||||
echo "Some tests failed!"
|
||||
exit 1
|
||||
fi
|
||||
@@ -7,6 +7,8 @@
|
||||
set -euo pipefail
|
||||
|
||||
KUGETSU="./skills/kugetsu/scripts/kugetsu"
|
||||
TEST_KUGETSU_DIR="/tmp/test-kugetsu-$$"
|
||||
export KUGETSU_DIR="$TEST_KUGETSU_DIR"
|
||||
TEST_ISSUE_REF="github.com/shoko/kugetsu#14"
|
||||
TEST_DISCUSS_REF="github.com/shoko/kugetsu#-discuss"
|
||||
TEST_BASE_SESSION_ID="ses_test_base_123"
|
||||
@@ -18,28 +20,28 @@ PASS=0
|
||||
FAIL=0
|
||||
|
||||
cleanup() {
|
||||
rm -rf ~/.kugetsu/sessions/* ~/.kugetsu/worktrees/* ~/.kugetsu/index.json 2>/dev/null || true
|
||||
rm -rf "$TEST_KUGETSU_DIR" 2>/dev/null || true
|
||||
}
|
||||
|
||||
setup_mock_base() {
|
||||
mkdir -p ~/.kugetsu/sessions ~/.kugetsu/worktrees
|
||||
cat > ~/.kugetsu/index.json << EOF
|
||||
mkdir -p "$TEST_KUGETSU_DIR/sessions" "$TEST_KUGETSU_DIR/worktrees"
|
||||
cat > "$TEST_KUGETSU_DIR/index.json" << EOF
|
||||
{
|
||||
"base": "$TEST_BASE_SESSION_ID",
|
||||
"pm_agent": "$TEST_PM_AGENT_SESSION_ID",
|
||||
"issues": {}
|
||||
}
|
||||
EOF
|
||||
cat > ~/.kugetsu/sessions/$TEST_BASE_SESSION_FILE << EOF
|
||||
cat > "$TEST_KUGETSU_DIR/sessions/$TEST_BASE_SESSION_FILE" << EOF
|
||||
{"type": "base", "opencode_session_id": "$TEST_BASE_SESSION_ID", "created_at": "2026-03-29T18:00:00+02:00", "state": "idle"}
|
||||
EOF
|
||||
cat > ~/.kugetsu/sessions/$TEST_PM_AGENT_SESSION_FILE << EOF
|
||||
cat > "$TEST_KUGETSU_DIR/sessions/$TEST_PM_AGENT_SESSION_FILE" << EOF
|
||||
{"type": "pm_agent", "opencode_session_id": "$TEST_PM_AGENT_SESSION_ID", "created_at": "2026-03-29T18:00:00+02:00", "state": "idle"}
|
||||
EOF
|
||||
}
|
||||
|
||||
setup_mock_forked() {
|
||||
cat > ~/.kugetsu/index.json << EOF
|
||||
cat > "$TEST_KUGETSU_DIR/index.json" << EOF
|
||||
{
|
||||
"base": "$TEST_BASE_SESSION_ID",
|
||||
"pm_agent": "$TEST_PM_AGENT_SESSION_ID",
|
||||
@@ -48,7 +50,7 @@ setup_mock_forked() {
|
||||
}
|
||||
}
|
||||
EOF
|
||||
cat > ~/.kugetsu/sessions/$TEST_FORKED_SESSION_FILE << EOF
|
||||
cat > "$TEST_KUGETSU_DIR/sessions/$TEST_FORKED_SESSION_FILE" << EOF
|
||||
{"type": "forked", "issue_ref": "$TEST_ISSUE_REF", "opencode_session_id": "ses_forked_789", "worktree_path": "/tmp/test-worktree", "created_at": "2026-03-29T18:00:00+02:00", "state": "idle"}
|
||||
EOF
|
||||
}
|
||||
@@ -112,16 +114,16 @@ echo ""
|
||||
|
||||
# Test 3b: start fails without pm-agent
|
||||
echo "--- Test: start without pm-agent session ---"
|
||||
rm -f ~/.kugetsu/index.json ~/.kugetsu/sessions/*
|
||||
mkdir -p ~/.kugetsu/sessions
|
||||
cat > ~/.kugetsu/index.json << EOF
|
||||
rm -f $TEST_KUGETSU_DIR/index.json $TEST_KUGETSU_DIR/sessions/*
|
||||
mkdir -p $TEST_KUGETSU_DIR/sessions
|
||||
cat > $TEST_KUGETSU_DIR/index.json << EOF
|
||||
{
|
||||
"base": "$TEST_BASE_SESSION_ID",
|
||||
"pm_agent": null,
|
||||
"issues": {}
|
||||
}
|
||||
EOF
|
||||
cat > ~/.kugetsu/sessions/$TEST_BASE_SESSION_FILE << EOF
|
||||
cat > $TEST_KUGETSU_DIR/sessions/$TEST_BASE_SESSION_FILE << EOF
|
||||
{"type": "base", "opencode_session_id": "$TEST_BASE_SESSION_ID", "created_at": "2026-03-29T18:00:00+02:00", "state": "idle"}
|
||||
EOF
|
||||
OUTPUT=$($KUGETSU start github.com/shoko/kugetsu#14 "test" 2>&1 || true)
|
||||
@@ -176,7 +178,7 @@ echo ""
|
||||
|
||||
# Test 6c: index.json has pm_agent field
|
||||
echo "--- Test: index.json has pm_agent field ---"
|
||||
if grep -q '"pm_agent"' ~/.kugetsu/index.json; then
|
||||
if grep -q '"pm_agent"' $TEST_KUGETSU_DIR/index.json; then
|
||||
pass "index.json has pm_agent field"
|
||||
else
|
||||
fail "index.json missing pm_agent field"
|
||||
@@ -227,12 +229,12 @@ echo ""
|
||||
echo "--- Test: destroy --pm-agent -y ---"
|
||||
setup_mock_base
|
||||
OUTPUT=$($KUGETSU destroy --pm-agent -y 2>&1 || true)
|
||||
if [ -f ~/.kugetsu/sessions/$TEST_PM_AGENT_SESSION_FILE ]; then
|
||||
if [ -f $TEST_KUGETSU_DIR/sessions/$TEST_PM_AGENT_SESSION_FILE ]; then
|
||||
fail "destroy --pm-agent -y removes pm-agent file"
|
||||
else
|
||||
pass "destroy --pm-agent -y removes pm-agent file"
|
||||
fi
|
||||
if grep -q '"pm_agent": null' ~/.kugetsu/index.json; then
|
||||
if grep -q '"pm_agent": null' $TEST_KUGETSU_DIR/index.json; then
|
||||
pass "destroy --pm-agent -y sets pm_agent to null in index"
|
||||
else
|
||||
fail "destroy --pm-agent -y should set pm_agent to null"
|
||||
@@ -243,7 +245,7 @@ echo ""
|
||||
echo "--- Test: destroy --base -y ---"
|
||||
setup_mock_base
|
||||
OUTPUT=$($KUGETSU destroy --base -y 2>&1 || true)
|
||||
if [ -f ~/.kugetsu/sessions/$TEST_BASE_SESSION_FILE ]; then
|
||||
if [ -f $TEST_KUGETSU_DIR/sessions/$TEST_BASE_SESSION_FILE ]; then
|
||||
fail "destroy --base -y removes base file"
|
||||
else
|
||||
pass "destroy --base -y removes base file"
|
||||
@@ -292,7 +294,7 @@ echo ""
|
||||
|
||||
# Test 15: worktree path in session file
|
||||
echo "--- Test: worktree_path in session file ---"
|
||||
if grep -q "worktree_path" ~/.kugetsu/sessions/$TEST_FORKED_SESSION_FILE; then
|
||||
if grep -q "worktree_path" $TEST_KUGETSU_DIR/sessions/$TEST_FORKED_SESSION_FILE; then
|
||||
pass "session file contains worktree_path"
|
||||
else
|
||||
fail "session file missing worktree_path"
|
||||
@@ -303,7 +305,7 @@ echo ""
|
||||
echo "--- Test: prune with orphaned worktree ---"
|
||||
cleanup
|
||||
setup_mock_base
|
||||
mkdir -p ~/.kugetsu/worktrees/orphaned-worktree
|
||||
mkdir -p $TEST_KUGETSU_DIR/worktrees/orphaned-worktree
|
||||
OUTPUT=$($KUGETSU prune 2>&1 || true)
|
||||
if echo "$OUTPUT" | grep -q "orphaned worktree"; then
|
||||
pass "prune detects orphaned worktree"
|
||||
@@ -315,7 +317,7 @@ echo ""
|
||||
# Test 17: prune --force removes orphaned worktrees
|
||||
echo "--- Test: prune --force removes orphaned worktrees ---"
|
||||
OUTPUT=$($KUGETSU prune --force 2>&1 || true)
|
||||
if [ -d ~/.kugetsu/worktrees/orphaned-worktree ]; then
|
||||
if [ -d $TEST_KUGETSU_DIR/worktrees/orphaned-worktree ]; then
|
||||
fail "prune --force should remove orphaned worktree"
|
||||
else
|
||||
pass "prune --force removes orphaned worktree"
|
||||
@@ -332,10 +334,10 @@ echo ""
|
||||
echo "--- Test: destroy removes worktree ---"
|
||||
cleanup
|
||||
setup_mock_forked
|
||||
# remove_worktree_for_issue derives path from issue ref: ~/.kugetsu/worktrees/github.com-shoko-kugetsu-14
|
||||
mkdir -p ~/.kugetsu/worktrees/github.com-shoko-kugetsu-14
|
||||
# remove_worktree_for_issue derives path from issue ref: $TEST_KUGETSU_DIR/worktrees/github.com-shoko-kugetsu-14
|
||||
mkdir -p $TEST_KUGETSU_DIR/worktrees/github.com-shoko-kugetsu-14
|
||||
OUTPUT=$($KUGETSU destroy github.com/shoko/kugetsu#14 -y 2>&1 || true)
|
||||
if [ -d ~/.kugetsu/worktrees/github.com-shoko-kugetsu-14 ]; then
|
||||
if [ -d $TEST_KUGETSU_DIR/worktrees/github.com-shoko-kugetsu-14 ]; then
|
||||
fail "destroy should remove worktree"
|
||||
else
|
||||
pass "destroy removes worktree"
|
||||
@@ -345,7 +347,7 @@ echo ""
|
||||
# Test 20: session file properly formatted for v2.2
|
||||
echo "--- Test: session file format v2.2 ---"
|
||||
setup_mock_forked
|
||||
SESSION_CONTENT=$(cat ~/.kugetsu/sessions/$TEST_FORKED_SESSION_FILE)
|
||||
SESSION_CONTENT=$(cat $TEST_KUGETSU_DIR/sessions/$TEST_FORKED_SESSION_FILE)
|
||||
if echo "$SESSION_CONTENT" | grep -q '"type": "forked"' && \
|
||||
echo "$SESSION_CONTENT" | grep -q '"worktree_path"'; then
|
||||
pass "session file has v2.2 format"
|
||||
@@ -367,8 +369,8 @@ echo ""
|
||||
|
||||
# Test 22: status when base missing
|
||||
echo "--- Test: status (base missing) ---"
|
||||
mkdir -p ~/.kugetsu/sessions
|
||||
cat > ~/.kugetsu/index.json << EOF
|
||||
mkdir -p $TEST_KUGETSU_DIR/sessions
|
||||
cat > $TEST_KUGETSU_DIR/index.json << EOF
|
||||
{
|
||||
"base": null,
|
||||
"pm_agent": "$TEST_PM_AGENT_SESSION_ID",
|
||||
@@ -385,7 +387,7 @@ echo ""
|
||||
|
||||
# Test 23: status when pm-agent missing
|
||||
echo "--- Test: status (pm-agent missing) ---"
|
||||
cat > ~/.kugetsu/index.json << EOF
|
||||
cat > $TEST_KUGETSU_DIR/index.json << EOF
|
||||
{
|
||||
"base": "$TEST_BASE_SESSION_ID",
|
||||
"pm_agent": null,
|
||||
@@ -402,7 +404,7 @@ echo ""
|
||||
|
||||
# Test 24: status when pm-agent is "None" (Python None output)
|
||||
echo "--- Test: status (pm-agent is Python None) ---"
|
||||
cat > ~/.kugetsu/index.json << EOF
|
||||
cat > $TEST_KUGETSU_DIR/index.json << EOF
|
||||
{
|
||||
"base": "$TEST_BASE_SESSION_ID",
|
||||
"pm_agent": "None",
|
||||
@@ -445,8 +447,8 @@ echo ""
|
||||
# Test 27: delegate when pm-agent missing
|
||||
echo "--- Test: delegate (pm-agent missing) ---"
|
||||
cleanup
|
||||
mkdir -p ~/.kugetsu/sessions ~/.kugetsu/worktrees
|
||||
cat > ~/.kugetsu/index.json << EOF
|
||||
mkdir -p $TEST_KUGETSU_DIR/sessions $TEST_KUGETSU_DIR/worktrees
|
||||
cat > $TEST_KUGETSU_DIR/index.json << EOF
|
||||
{
|
||||
"base": "$TEST_BASE_SESSION_ID",
|
||||
"pm_agent": null,
|
||||
@@ -508,7 +510,7 @@ echo ""
|
||||
# Test 32: delegate is fire-and-forget (returns immediately)
|
||||
echo "--- Test: delegate is fire-and-forget ---"
|
||||
setup_mock_base
|
||||
mkdir -p ~/.kugetsu/logs
|
||||
mkdir -p $TEST_KUGETSU_DIR/logs
|
||||
START=$(date +%s)
|
||||
OUTPUT=$($KUGETSU delegate "test fire-and-forget" 2>&1 || true)
|
||||
END=$(date +%s)
|
||||
@@ -527,10 +529,10 @@ echo ""
|
||||
# Test 33: delegate creates log file
|
||||
echo "--- Test: delegate creates log file ---"
|
||||
setup_mock_base
|
||||
LOG_COUNT_BEFORE=$(ls ~/.kugetsu/logs/*.log 2>/dev/null | wc -l)
|
||||
LOG_COUNT_BEFORE=$(ls $TEST_KUGETSU_DIR/logs/*.log 2>/dev/null | wc -l)
|
||||
$KUGETSU delegate "test log file" 2>&1 || true
|
||||
sleep 1
|
||||
LOG_COUNT_AFTER=$(ls ~/.kugetsu/logs/*.log 2>/dev/null | wc -l)
|
||||
LOG_COUNT_AFTER=$(ls $TEST_KUGETSU_DIR/logs/*.log 2>/dev/null | wc -l)
|
||||
if [ $LOG_COUNT_AFTER -gt $LOG_COUNT_BEFORE ]; then
|
||||
pass "delegate creates log file"
|
||||
else
|
||||
@@ -558,10 +560,10 @@ echo ""
|
||||
|
||||
# Test E2: env set creates file
|
||||
echo "--- Test: env set creates env file ---"
|
||||
mkdir -p ~/.kugetsu/env
|
||||
rm -f ~/.kugetsu/env/pm-agent.env
|
||||
mkdir -p $TEST_KUGETSU_DIR/env
|
||||
rm -f $TEST_KUGETSU_DIR/env/pm-agent.env
|
||||
$KUGETSU env set TEST_VAR "test_value" pm-agent 2>&1 || true
|
||||
if [ -f ~/.kugetsu/env/pm-agent.env ]; then
|
||||
if [ -f $TEST_KUGETSU_DIR/env/pm-agent.env ]; then
|
||||
pass "env set creates pm-agent.env file"
|
||||
else
|
||||
fail "env set did not create pm-agent.env"
|
||||
@@ -570,7 +572,7 @@ echo ""
|
||||
|
||||
# Test E3: env show masks sensitive values
|
||||
echo "--- Test: env show masks sensitive values ---"
|
||||
cat > ~/.kugetsu/env/pm-agent.env << 'ENVEOF'
|
||||
cat > $TEST_KUGETSU_DIR/env/pm-agent.env << 'ENVEOF'
|
||||
export GITEA_TOKEN="secret_token_123"
|
||||
export MY_VAR="visible_value"
|
||||
ENVEOF
|
||||
@@ -584,14 +586,14 @@ echo ""
|
||||
|
||||
# Test E4: Variables exported to child processes via set -a
|
||||
echo "--- Test: set -a exports variables to children ---"
|
||||
mkdir -p ~/.kugetsu/env
|
||||
cat > ~/.kugetsu/env/test.env << 'ENVEOF'
|
||||
mkdir -p $TEST_KUGETSU_DIR/env
|
||||
cat > $TEST_KUGETSU_DIR/env/test.env << 'ENVEOF'
|
||||
export EXPORT_TEST="exported_value"
|
||||
SIMPLE_TEST="not_exported"
|
||||
ENVEOF
|
||||
|
||||
# Simulate what cmd_delegate does
|
||||
ENV_FILE="~/.kugetsu/env/test.env"
|
||||
ENV_FILE="$TEST_KUGETSU_DIR/env/test.env"
|
||||
env_sh="set -a; source '$ENV_FILE'; set +a; "
|
||||
result=$(bash -c "${env_sh}bash -c 'echo \$EXPORT_TEST'")
|
||||
|
||||
@@ -604,11 +606,11 @@ echo ""
|
||||
|
||||
# Test E5: pm-agent.env takes precedence
|
||||
echo "--- Test: pm-agent.env takes precedence over default ---"
|
||||
mkdir -p ~/.kugetsu/env
|
||||
cat > ~/.kugetsu/env/default.env << 'ENVEOF'
|
||||
mkdir -p $TEST_KUGETSU_DIR/env
|
||||
cat > $TEST_KUGETSU_DIR/env/default.env << 'ENVEOF'
|
||||
export GITEA_TOKEN="default_token"
|
||||
ENVEOF
|
||||
cat > ~/.kugetsu/env/pm-agent.env << 'ENVEOF'
|
||||
cat > $TEST_KUGETSU_DIR/env/pm-agent.env << 'ENVEOF'
|
||||
export GITEA_TOKEN="pm_agent_token"
|
||||
ENVEOF
|
||||
|
||||
@@ -644,7 +646,7 @@ fi
|
||||
echo ""
|
||||
|
||||
# Cleanup env files
|
||||
rm -rf ~/.kugetsu/env 2>/dev/null || true
|
||||
rm -rf $TEST_KUGETSU_DIR/env 2>/dev/null || true
|
||||
|
||||
# Test E7: fix_session_permissions function exists
|
||||
echo "--- Test: fix_session_permissions function exists ---"
|
||||
@@ -736,7 +738,7 @@ PASS=0
|
||||
FAIL=0
|
||||
|
||||
test_cleanup() {
|
||||
rm -rf ~/.kugetsu/sessions/* ~/.kugetsu/worktrees/* ~/.kugetsu/index.json ~/.kugetsu/logs/* ~/.kugetsu/.agent_count ~/.kugetsu/.agent_lock 2>/dev/null || true
|
||||
rm -rf $TEST_KUGETSU_DIR/sessions/* $TEST_KUGETSU_DIR/worktrees/* $TEST_KUGETSU_DIR/index.json $TEST_KUGETSU_DIR/logs/* $TEST_KUGETSU_DIR/.agent_count $TEST_KUGETSU_DIR/.agent_lock 2>/dev/null || true
|
||||
}
|
||||
|
||||
pass() {
|
||||
@@ -750,25 +752,25 @@ fail() {
|
||||
}
|
||||
|
||||
setup_mock_sessions() {
|
||||
mkdir -p ~/.kugetsu/sessions ~/.kugetsu/worktrees ~/.kugetsu/logs
|
||||
cat > ~/.kugetsu/index.json << INDEX
|
||||
mkdir -p $TEST_KUGETSU_DIR/sessions $TEST_KUGETSU_DIR/worktrees $TEST_KUGETSU_DIR/logs
|
||||
cat > $TEST_KUGETSU_DIR/index.json << INDEX
|
||||
{
|
||||
"base": "ses_test_base_123",
|
||||
"pm_agent": "ses_test_pm_456",
|
||||
"issues": {}
|
||||
}
|
||||
INDEX
|
||||
echo '{"type": "base", "opencode_session_id": "ses_test_base_123", "created_at": "2026-03-29T18:00:00+02:00", "state": "idle"}' > ~/.kugetsu/sessions/base.json
|
||||
echo '{"type": "pm_agent", "opencode_session_id": "ses_test_pm_456", "created_at": "2026-03-29T18:00:00+02:00", "state": "idle"}' > ~/.kugetsu/sessions/pm-agent.json
|
||||
echo '{"type": "base", "opencode_session_id": "ses_test_base_123", "created_at": "2026-03-29T18:00:00+02:00", "state": "idle"}' > $TEST_KUGETSU_DIR/sessions/base.json
|
||||
echo '{"type": "pm_agent", "opencode_session_id": "ses_test_pm_456", "created_at": "2026-03-29T18:00:00+02:00", "state": "idle"}' > $TEST_KUGETSU_DIR/sessions/pm-agent.json
|
||||
}
|
||||
|
||||
# Test C1: Agent count file is initialized to 0
|
||||
echo "--- Test: agent count file initialized ---"
|
||||
test_cleanup
|
||||
mkdir -p ~/.kugetsu/sessions ~/.kugetsu/worktrees
|
||||
mkdir -p $TEST_KUGETSU_DIR/sessions $TEST_KUGETSU_DIR/worktrees
|
||||
$KUGETSU list > /dev/null 2>&1 || true
|
||||
if [ -f ~/.kugetsu/.agent_count ]; then
|
||||
COUNT=$(cat ~/.kugetsu/.agent_count)
|
||||
if [ -f $TEST_KUGETSU_DIR/.agent_count ]; then
|
||||
COUNT=$(cat $TEST_KUGETSU_DIR/.agent_count)
|
||||
if [ "$COUNT" = "0" ]; then
|
||||
pass "agent count file initialized to 0"
|
||||
else
|
||||
@@ -795,10 +797,10 @@ test_cleanup
|
||||
setup_mock_sessions
|
||||
|
||||
# Initialize count to 0
|
||||
echo 0 > ~/.kugetsu/.agent_count
|
||||
echo 0 > $TEST_KUGETSU_DIR/.agent_count
|
||||
|
||||
# Verify initial state
|
||||
INITIAL=$(cat ~/.kugetsu/.agent_count)
|
||||
INITIAL=$(cat $TEST_KUGETSU_DIR/.agent_count)
|
||||
if [ "$INITIAL" = "0" ]; then
|
||||
pass "agent count starts at 0"
|
||||
else
|
||||
@@ -809,7 +811,7 @@ fi
|
||||
$KUGETSU list > /dev/null 2>&1
|
||||
|
||||
# Verify count is still 0 (no slot leak)
|
||||
AFTER=$(cat ~/.kugetsu/.agent_count)
|
||||
AFTER=$(cat $TEST_KUGETSU_DIR/.agent_count)
|
||||
if [ "$AFTER" = "0" ]; then
|
||||
pass "agent count stays 0 after list (no leak)"
|
||||
else
|
||||
|
||||
Reference in New Issue
Block a user