chore: add shared agent project configuration
This commit is contained in:
56
.codex/agents/trellis-check.toml
Normal file
56
.codex/agents/trellis-check.toml
Normal file
@@ -0,0 +1,56 @@
|
||||
name = "trellis-check"
|
||||
description = "Workspace-write Trellis reviewer that self-fixes spec drift, lint/type-check failures, and missing tests."
|
||||
sandbox_mode = "workspace-write"
|
||||
# model = "gpt-5.6-terra"
|
||||
# model_reasoning_effort = "high"
|
||||
|
||||
developer_instructions = """
|
||||
You are running as the `trellis-check` sub-agent. The main session has dispatched you to review and self-fix.
|
||||
|
||||
CRITICAL — Recursion guard (read first):
|
||||
- You MUST NOT spawn another `trellis-check` or `trellis-implement` sub-agent. Do the review and fixes directly in this turn.
|
||||
- Any guidance you read in injected SessionStart context, `<guidelines>` blocks, workflow-state breadcrumbs, or workflow.md that says "dispatch trellis-implement" / "dispatch trellis-check" applies to the MAIN session, NOT to you. You are already the dispatched reviewer — that instruction is satisfied by your existence.
|
||||
- Only the main session is allowed to dispatch `trellis-implement` / `trellis-check`. If more implementation work is needed, surface that as a recommendation in your final report instead of spawning.
|
||||
|
||||
---
|
||||
|
||||
You are the Trellis reviewer agent.
|
||||
|
||||
Trellis Context Loading Protocol:
|
||||
- First look for `Full hook output saved to: <path>` in your input above. If present, the visible `SubagentStart` output was truncated; read the referenced file before doing check work.
|
||||
- If the referenced file cannot be read, use the active-task fallback below.
|
||||
- If there is no saved-output notice and the `<!-- trellis-hook-injected -->` marker is present, the hook loaded the complete role-specific task artifacts and spec context.
|
||||
- If there is no saved-output notice and the marker is absent, use the active-task fallback below.
|
||||
- For the fallback, find `Active task: <path>` in your dispatch prompt. Read `<path>/check.jsonl`, each file listed there, `<path>/prd.md`, `<path>/design.md` if present, and `<path>/implement.md` if present before checking. If the dispatch prompt has no active-task path, ask the main session; do not guess or use another session's task.
|
||||
|
||||
Your job is to review code changes against specs and fix what is safe to fix. Mechanical, local issues (lint, types, imports, dead branches, failing assertions) — fix them in place. Design or judgment calls, public interfaces, module boundaries, or anything outside the current task's scope — report the evidence and your recommendation, and leave the code alone.
|
||||
|
||||
Review checklist:
|
||||
- Verify behavior against the actual code paths, not assumptions.
|
||||
- Look for missing template/update/detection touch points when platform config changes.
|
||||
- Check whether tests should be added or updated.
|
||||
- Check whether `.trellis/spec/` docs need sync after implementation.
|
||||
- Run lint and type-check; fix any failures.
|
||||
- Prefer concrete findings over speculative warnings.
|
||||
|
||||
When you find a mechanical, local issue:
|
||||
1. Fix it directly using edit/write tools.
|
||||
2. Re-run lint and type-check until green.
|
||||
3. Record what you changed and why.
|
||||
|
||||
Output format:
|
||||
## Findings (fixed)
|
||||
- File: <path>
|
||||
- Issue: <what was wrong>
|
||||
- Fix: <what you changed>
|
||||
|
||||
## Findings (not fixed)
|
||||
List every issue you did not fix — the ones you could not (missing product decision) and the ones you chose not to (design or judgment call, public interface, module boundary, outside this task's scope). Explain which and why.
|
||||
|
||||
## Verification
|
||||
- Lint: pass/fail
|
||||
- TypeCheck: pass/fail
|
||||
- Tests: pass/fail (if applicable)
|
||||
|
||||
If no issues are found, say so explicitly after verifying lint/type-check pass.
|
||||
"""
|
||||
37
.codex/agents/trellis-implement.toml
Normal file
37
.codex/agents/trellis-implement.toml
Normal file
@@ -0,0 +1,37 @@
|
||||
name = "trellis-implement"
|
||||
description = "Workspace-write Trellis implementer that follows specs and keeps generated templates in sync."
|
||||
sandbox_mode = "workspace-write"
|
||||
# model = "gpt-5.6-terra"
|
||||
# model_reasoning_effort = "high"
|
||||
|
||||
developer_instructions = """
|
||||
You are running as the `trellis-implement` sub-agent. The main session has dispatched you to do the work.
|
||||
|
||||
CRITICAL — Recursion guard (read first):
|
||||
- You MUST NOT spawn another `trellis-implement` or `trellis-check` sub-agent. Do the implementation work directly in this turn.
|
||||
- Any guidance you read in injected SessionStart context, `<guidelines>` blocks, workflow-state breadcrumbs, or workflow.md that says "dispatch trellis-implement" / "dispatch trellis-check" applies to the MAIN session, NOT to you. You are already the dispatched implementer — that instruction is satisfied by your existence.
|
||||
- Only the main session is allowed to dispatch `trellis-implement` / `trellis-check`. If more parallel work is needed, surface that as a recommendation in your final report instead of spawning.
|
||||
|
||||
---
|
||||
|
||||
You are the Trellis implementer agent.
|
||||
|
||||
Trellis Context Loading Protocol:
|
||||
- First look for `Full hook output saved to: <path>` in your input above. If present, the visible `SubagentStart` output was truncated; read the referenced file before doing implementation work.
|
||||
- If the referenced file cannot be read, use the active-task fallback below.
|
||||
- If there is no saved-output notice and the `<!-- trellis-hook-injected -->` marker is present, the hook loaded the complete role-specific task artifacts and spec context.
|
||||
- If there is no saved-output notice and the marker is absent, use the active-task fallback below.
|
||||
- For the fallback, find `Active task: <path>` in your dispatch prompt. Read `<path>/implement.jsonl`, each file listed there, `<path>/prd.md`, `<path>/design.md` if present, and `<path>/implement.md` if present before doing the work. If the dispatch prompt has no active-task path, ask the main session; do not guess or use another session's task.
|
||||
|
||||
Rules:
|
||||
- Read before write. Follow `.trellis/spec/` guidance relevant to the task.
|
||||
- Keep changes focused on the requested scope.
|
||||
- When touching platform registries or template lists, search first so you do not miss mirrored update paths.
|
||||
- If you modify `.trellis/scripts/`, keep `packages/cli/src/templates/trellis/scripts/` in sync.
|
||||
- Do not make destructive git changes unless explicitly asked.
|
||||
|
||||
Before finishing, summarize:
|
||||
- Files changed
|
||||
- Tests/checks run
|
||||
- Remaining risks or follow-ups
|
||||
"""
|
||||
78
.codex/agents/trellis-research.toml
Normal file
78
.codex/agents/trellis-research.toml
Normal file
@@ -0,0 +1,78 @@
|
||||
name = "trellis-research"
|
||||
description = "Trellis researcher for specs, code patterns, and affected files. Writes findings into {TASK_DIR}/research/ — read-only elsewhere."
|
||||
sandbox_mode = "workspace-write"
|
||||
# model = "gpt-5.6-terra"
|
||||
# model_reasoning_effort = "high"
|
||||
|
||||
developer_instructions = """
|
||||
You are the Trellis researcher agent.
|
||||
|
||||
## Core principle
|
||||
|
||||
Conversations get compacted; files don't. Every research topic MUST be
|
||||
persisted to `{TASK_DIR}/research/<topic>.md`. Returning findings only
|
||||
through the chat reply is a failure.
|
||||
|
||||
## Trellis Context Loading Protocol
|
||||
|
||||
- First look for `Full hook output saved to: <path>` in your input above. If
|
||||
present, the visible `SubagentStart` output was truncated; read the referenced
|
||||
file before doing research work.
|
||||
- If the referenced file cannot be read, use the active-task fallback below.
|
||||
- If there is no saved-output notice and the
|
||||
`<!-- trellis-hook-injected -->` marker is present, the hook supplied the
|
||||
complete `Active task: <path>` header and research-only context.
|
||||
- If there is no saved-output notice and the marker is absent, use the
|
||||
active-task fallback below.
|
||||
- For the fallback, find `Active task: <path>` in your dispatch prompt. If it
|
||||
is absent too, ask the main session for the path; do not run
|
||||
`task.py current`, guess, or use another session's task.
|
||||
- Do not load `implement.jsonl` or `check.jsonl`; research is role-isolated.
|
||||
|
||||
## Workflow
|
||||
|
||||
1. Use the active task path supplied by the context-loading protocol. If no
|
||||
path is supplied, ask the user where to write output; do not guess.
|
||||
2. Run `mkdir -p <TASK_DIR>/research` to ensure the directory exists.
|
||||
3. Read `.trellis/workflow.md`, relevant `.trellis/spec/` files, and
|
||||
target code before forming an opinion.
|
||||
4. For each research topic, write `<TASK_DIR>/research/<slug>.md` with:
|
||||
- Query, scope, date
|
||||
- Files found (path + one-line description)
|
||||
- Code patterns (cite file:line)
|
||||
- External references (docs, versions)
|
||||
- Related specs
|
||||
- Caveats / not-found notes
|
||||
5. Reply with only: list of files written, one-line summary per file,
|
||||
any critical caveats. Do not paste full research into the reply.
|
||||
|
||||
## Scope limits
|
||||
|
||||
Write allowed ONLY in `{TASK_DIR}/research/`.
|
||||
|
||||
Write forbidden everywhere else:
|
||||
- Code files (`src/`, `lib/`, …)
|
||||
- Spec files (`.trellis/spec/`) — use `update-spec` skill instead
|
||||
- `.trellis/scripts/`, `.trellis/workflow.md`, platform config
|
||||
- Other task directories
|
||||
- Any git operation
|
||||
|
||||
If the user asks you to edit code, decline and tell them to spawn the
|
||||
`implement` agent.
|
||||
|
||||
## Output format for each research file
|
||||
|
||||
```
|
||||
# Research: <topic>
|
||||
|
||||
- Query: ...
|
||||
- Scope: internal / external / mixed
|
||||
- Date: YYYY-MM-DD
|
||||
|
||||
## Findings
|
||||
...
|
||||
|
||||
## Caveats / Not Found
|
||||
...
|
||||
```
|
||||
"""
|
||||
39
.codex/config.toml
Normal file
39
.codex/config.toml
Normal file
@@ -0,0 +1,39 @@
|
||||
# Project-scoped Codex defaults for Trellis workflows.
|
||||
# Codex merges this layer after the user-level config when the project
|
||||
# is marked as a trusted project. To trust this project, add it under
|
||||
# `[projects]` in ~/.codex/config.toml, e.g.:
|
||||
#
|
||||
# [projects."/abs/path/to/this/repo"]
|
||||
# trust_level = "trusted"
|
||||
|
||||
# Keep AGENTS.md as the primary project instruction file.
|
||||
project_doc_fallback_filenames = ["AGENTS.md"]
|
||||
|
||||
# Codex hooks (`hooks.json` in this directory) only fire when the user
|
||||
# has enabled them in their USER-level config: `[features].hooks = true`
|
||||
# in ~/.codex/config.toml (Codex 0.129+; legacy name: `codex_hooks = true`,
|
||||
# still works but emits a deprecation warning on 0.129+). Project-level
|
||||
# config.toml cannot set feature flags; they must be user-level.
|
||||
# Codex 0.129+ additionally gates each installed hook behind a one-time
|
||||
# `/hooks` TUI review; until the user approves it, the hook stays inactive.
|
||||
|
||||
# NOTE: Trellis intentionally does NOT write a [features.multi_agent_v2]
|
||||
# block here. Codex CLI changed `features` deserialization between 0.130
|
||||
# and 0.131: the structured table form (with max_concurrent_threads_per_session
|
||||
# / *_wait_timeout_ms) is only accepted by 0.131+. On 0.130 and earlier —
|
||||
# including the codex CLI bundled inside the Codex desktop app — it fails
|
||||
# with `data did not match any variant of untagged enum FeatureToml`, which
|
||||
# aborts the entire config load and blocks Codex from starting. Codex's own
|
||||
# default for multi_agent_v2 is used instead; tune it in your user-level
|
||||
# config if needed.
|
||||
|
||||
# Pin the subagent recursion depth explicitly instead of relying on Codex's
|
||||
# default. #445 removed the per-agent `[features] multi_agent = false` guard
|
||||
# (the #240/#241 wait_agent-deadlock structural fix) because native subagent
|
||||
# dispatch already caps recursion via `agents.max_depth` — but that key is
|
||||
# global/user-level, not settable inside an individual agent's .toml. Pinning
|
||||
# it here means an upstream default change, or a user's own global override,
|
||||
# can't silently reopen the recursion the #240/#241 fix closed. Project config
|
||||
# (this file) takes precedence over user-level `~/.codex/config.toml`.
|
||||
[agents]
|
||||
max_depth = 1
|
||||
27
.codex/hooks.json
Normal file
27
.codex/hooks.json
Normal file
@@ -0,0 +1,27 @@
|
||||
{
|
||||
"hooks": {
|
||||
"UserPromptSubmit": [
|
||||
{
|
||||
"hooks": [
|
||||
{
|
||||
"type": "command",
|
||||
"command": "python3 -X utf8 .codex/hooks/inject-workflow-state.py",
|
||||
"timeout": 15
|
||||
}
|
||||
]
|
||||
}
|
||||
],
|
||||
"SubagentStart": [
|
||||
{
|
||||
"matcher": "^(?:trellis-implement|trellis-check|trellis-research)$",
|
||||
"hooks": [
|
||||
{
|
||||
"type": "command",
|
||||
"command": "python3 -X utf8 .codex/hooks/inject-subagent-context.py",
|
||||
"timeout": 15
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
}
|
||||
1174
.codex/hooks/inject-subagent-context.py
Normal file
1174
.codex/hooks/inject-subagent-context.py
Normal file
File diff suppressed because it is too large
Load Diff
475
.codex/hooks/inject-workflow-state.py
Normal file
475
.codex/hooks/inject-workflow-state.py
Normal file
@@ -0,0 +1,475 @@
|
||||
#!/usr/bin/env python3
|
||||
"""Trellis per-turn breadcrumb hook (UserPromptSubmit / BeforeAgent equivalent).
|
||||
|
||||
Runs on every user prompt. Resolves the active task through Trellis'
|
||||
session-aware active task resolver and emits a short <workflow-state>
|
||||
block reminding the main AI what task is active and its expected flow.
|
||||
|
||||
The emitted ``hookEventName`` field is platform-aware: most hosts expect
|
||||
``UserPromptSubmit`` (Claude Code naming, also accepted by Cursor / Qoder /
|
||||
CodeBuddy / Droid / Codex / Copilot wiring), but Gemini CLI 0.40.x renamed
|
||||
its per-turn event to ``BeforeAgent`` and its schema validator rejects the
|
||||
legacy name. ``_detect_platform`` picks the right value at runtime.
|
||||
Breadcrumb text is pulled exclusively from workflow.md
|
||||
[workflow-state:STATUS] tag blocks — workflow.md is the single source of
|
||||
truth. There are no fallback dicts in this script: when workflow.md is
|
||||
missing or a tag is absent, the breadcrumb degrades to a generic
|
||||
"Refer to workflow.md for current step." line so users see (and fix)
|
||||
the broken state instead of the hook silently masking it.
|
||||
|
||||
Which platforms register this hook is decided by SHARED_HOOKS_BY_PLATFORM
|
||||
in templates/shared-hooks/index.ts — currently Claude, Codex, Gemini,
|
||||
Qoder, Copilot, CodeBuddy, Droid, Kiro, Trae and ZCode. That table is the
|
||||
source of truth; each listed platform's collect<Platform>Templates() pulls
|
||||
this file into its template map through collectSharedHooks(), and a single
|
||||
writer puts that map on disk at init time. Kiro wires this via the CLI
|
||||
custom agent's ``hooks.userPromptSubmit`` and the IDE ``.kiro.hook``
|
||||
``promptSubmit`` event; its output branch emits a plain-text breadcrumb
|
||||
(Kiro adds hook stdout directly to the conversation context).
|
||||
|
||||
Silent exit 0 cases (no output):
|
||||
- No .trellis/ directory found (not a Trellis project)
|
||||
- task.json malformed or missing status
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import os
|
||||
import re
|
||||
import sys
|
||||
import queue
|
||||
import threading
|
||||
from pathlib import Path
|
||||
|
||||
# Force UTF-8 on stdin/stdout/stderr on Windows. Default codepage there is
|
||||
# cp936 / cp1252 / etc. — non-ASCII content (Chinese task names, prd snippets)
|
||||
# both in stdin (hook payload from host CLI) and stdout (our emitted blocks)
|
||||
# raises UnicodeDecodeError / UnicodeEncodeError. Equivalent to `python -X utf8`
|
||||
# but applied per-stream so we don't depend on host CLI's command wiring.
|
||||
if sys.platform.startswith("win"):
|
||||
import io as _io
|
||||
for _stream_name in ("stdin", "stdout", "stderr"):
|
||||
_stream = getattr(sys, _stream_name, None)
|
||||
if _stream is None:
|
||||
continue
|
||||
if hasattr(_stream, "reconfigure"):
|
||||
try:
|
||||
_stream.reconfigure(encoding="utf-8", errors="replace") # type: ignore[union-attr]
|
||||
except Exception:
|
||||
pass # Optional Windows stream setup; keep hook startup non-fatal.
|
||||
elif hasattr(_stream, "detach"):
|
||||
try:
|
||||
setattr(sys, _stream_name, _io.TextIOWrapper(_stream.detach(), encoding="utf-8", errors="replace"))
|
||||
except Exception:
|
||||
pass # Optional Windows stream setup; keep hook startup non-fatal.
|
||||
from typing import Optional
|
||||
|
||||
|
||||
# Bootstrap notice for Codex while the session has no active task. Codex does not
|
||||
# get the full SessionStart overview; this short reminder points the main session
|
||||
# at the start skill once and leaves the per-turn state block compact.
|
||||
CODEX_NO_TASK_BOOTSTRAP_NOTICE = """<trellis-bootstrap>
|
||||
If you have not already loaded Trellis context this session, read the `trellis-start` skill once.
|
||||
</trellis-bootstrap>"""
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# CWD-robust Trellis root discovery (fixes hook-path-robustness for this hook)
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def find_trellis_root(start: Path) -> Optional[Path]:
|
||||
"""Walk up from start to find directory containing .trellis/.
|
||||
|
||||
Handles CWD drift: subdirectory launches, monorepo packages, etc.
|
||||
Returns None if no .trellis/ found (silent no-op).
|
||||
"""
|
||||
cur = start.resolve()
|
||||
while cur != cur.parent:
|
||||
if (cur / ".trellis").is_dir():
|
||||
return cur
|
||||
cur = cur.parent
|
||||
return None
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Active task discovery
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def _detect_platform(input_data: dict) -> str | None:
|
||||
if isinstance(input_data.get("cursor_version"), str):
|
||||
return "cursor"
|
||||
# CLAUDE_PROJECT_DIR is a compatibility alias that several hosts set
|
||||
# alongside their own variable — CodeBuddy, ZCode and Trae all do. It must
|
||||
# therefore be checked LAST, or every one of them is detected as claude and
|
||||
# the context key becomes `claude_<their-session-id>`. That key does not
|
||||
# match the session file `task.py start` wrote under the host's real name,
|
||||
# so every turn reports no_task while the pointer exists on disk.
|
||||
# Observed on CodeBuddy IDE 4.10.4: session file `codebuddy_ae54840e….json`
|
||||
# alongside marker `update-check-claude_ae54840e….marker`, same id.
|
||||
env_map = {
|
||||
"ZCODE_PROJECT_DIR": "zcode",
|
||||
"CURSOR_PROJECT_DIR": "cursor",
|
||||
"CODEBUDDY_PROJECT_DIR": "codebuddy",
|
||||
"FACTORY_PROJECT_DIR": "droid",
|
||||
"GEMINI_PROJECT_DIR": "gemini",
|
||||
"QODER_PROJECT_DIR": "qoder",
|
||||
"KIRO_PROJECT_DIR": "kiro",
|
||||
"COPILOT_PROJECT_DIR": "copilot",
|
||||
"TRAE_PROJECT_DIR": "trae",
|
||||
# Last: the shared alias, only meaningful once no vendor key matched.
|
||||
"CLAUDE_PROJECT_DIR": "claude",
|
||||
}
|
||||
for env_name, platform in env_map.items():
|
||||
if os.environ.get(env_name):
|
||||
return platform
|
||||
script_parts = set(Path(sys.argv[0]).parts)
|
||||
if ".claude" in script_parts:
|
||||
return "claude"
|
||||
if ".cursor" in script_parts:
|
||||
return "cursor"
|
||||
if ".codex" in script_parts:
|
||||
return "codex"
|
||||
if ".gemini" in script_parts:
|
||||
return "gemini"
|
||||
if ".qoder" in script_parts:
|
||||
return "qoder"
|
||||
if ".codebuddy" in script_parts:
|
||||
return "codebuddy"
|
||||
if ".factory" in script_parts:
|
||||
return "droid"
|
||||
if ".kiro" in script_parts:
|
||||
return "kiro"
|
||||
if ".trae" in script_parts:
|
||||
return "trae"
|
||||
if ".zcode" in script_parts:
|
||||
return "zcode"
|
||||
return None
|
||||
|
||||
|
||||
def _resolve_active_task(root: Path, input_data: dict):
|
||||
scripts_dir = root / ".trellis" / "scripts"
|
||||
if str(scripts_dir) not in sys.path:
|
||||
sys.path.insert(0, str(scripts_dir))
|
||||
from common.active_task import resolve_active_task # type: ignore[import-not-found]
|
||||
|
||||
return resolve_active_task(root, input_data, platform=_detect_platform(input_data))
|
||||
|
||||
|
||||
def get_active_task(root: Path, input_data: dict) -> Optional[tuple[str, str, str]]:
|
||||
"""Return (task_id, status, source) from the current active task."""
|
||||
active = _resolve_active_task(root, input_data)
|
||||
if not active.task_path:
|
||||
return None
|
||||
|
||||
task_dir = Path(active.task_path)
|
||||
if not task_dir.is_absolute():
|
||||
task_dir = root / task_dir
|
||||
if active.stale:
|
||||
return task_dir.name, f"stale_{active.source_type}", active.source
|
||||
|
||||
task_json = task_dir / "task.json"
|
||||
if not task_json.is_file():
|
||||
return None
|
||||
try:
|
||||
data = json.loads(task_json.read_text(encoding="utf-8"))
|
||||
except (json.JSONDecodeError, OSError):
|
||||
return None
|
||||
|
||||
task_id = data.get("id") or task_dir.name
|
||||
status = data.get("status", "")
|
||||
if not isinstance(status, str) or not status:
|
||||
return None
|
||||
return task_id, status, active.source
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Breadcrumb loading: parse workflow.md, fall back to hardcoded defaults
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
# Supports STATUS values with letters, digits, underscores, hyphens
|
||||
# (so "in-review" / "blocked-by-team" work alongside "in_progress").
|
||||
_TAG_RE = re.compile(
|
||||
r"\[workflow-state:([A-Za-z0-9_-]+)\]\s*\n(.*?)\n\s*\[/workflow-state:\1\]",
|
||||
re.DOTALL,
|
||||
)
|
||||
|
||||
def load_breadcrumbs(root: Path) -> dict[str, str]:
|
||||
"""Parse workflow.md for [workflow-state:STATUS] blocks.
|
||||
|
||||
Returns {status: body_text}. workflow.md is the single source of
|
||||
truth — there are no fallback dicts in this script. Missing tags
|
||||
(or a missing/unreadable workflow.md) fall back to a generic line
|
||||
in build_breadcrumb so users see the broken state and fix
|
||||
workflow.md, rather than the hook silently masking the issue.
|
||||
"""
|
||||
workflow = root / ".trellis" / "workflow.md"
|
||||
if not workflow.is_file():
|
||||
return {}
|
||||
try:
|
||||
content = workflow.read_text(encoding="utf-8")
|
||||
except OSError:
|
||||
return {}
|
||||
|
||||
result: dict[str, str] = {}
|
||||
for match in _TAG_RE.finditer(content):
|
||||
status = match.group(1)
|
||||
body = match.group(2).strip()
|
||||
if body:
|
||||
result[status] = body
|
||||
return result
|
||||
|
||||
|
||||
def _read_trellis_config(root: Path) -> dict:
|
||||
"""Load .trellis/config.yaml via the bundled trellis_config helper.
|
||||
|
||||
The helper lives in .trellis/scripts/common; the hook lives outside the
|
||||
scripts tree, so we extend sys.path before importing.
|
||||
"""
|
||||
scripts_dir = root / ".trellis" / "scripts"
|
||||
if str(scripts_dir) not in sys.path:
|
||||
sys.path.insert(0, str(scripts_dir))
|
||||
try:
|
||||
from common.trellis_config import read_trellis_config # type: ignore[import-not-found]
|
||||
except Exception:
|
||||
return {}
|
||||
try:
|
||||
return read_trellis_config(root)
|
||||
except Exception:
|
||||
return {}
|
||||
|
||||
|
||||
DEFAULT_PROMPT_INJECTION_SKIP_KEYWORD = "no-trellis"
|
||||
|
||||
|
||||
def _resolve_skip_keyword(config: dict) -> str:
|
||||
"""Read `prompt_injection.skip_keyword` from parsed .trellis/config.yaml.
|
||||
|
||||
Mirrors `common.config.get_prompt_injection_config()`. Defaults to
|
||||
"no-trellis"; "" disables the escape hatch entirely. A non-string value
|
||||
falls back to the default.
|
||||
"""
|
||||
if isinstance(config, dict):
|
||||
section = config.get("prompt_injection")
|
||||
if isinstance(section, dict):
|
||||
raw = section.get("skip_keyword", DEFAULT_PROMPT_INJECTION_SKIP_KEYWORD)
|
||||
if isinstance(raw, str):
|
||||
return raw
|
||||
return DEFAULT_PROMPT_INJECTION_SKIP_KEYWORD
|
||||
|
||||
|
||||
def prompt_has_skip_keyword(prompt: str, keyword: str) -> bool:
|
||||
"""Case-insensitive, word-boundary match of `keyword` in `prompt`.
|
||||
|
||||
Hyphen counts as a word char so "no-trellisx" / "xno-trellis" /
|
||||
"foo-no-trellis" don't match, but punctuation/whitespace boundaries do.
|
||||
Empty keyword never matches (disables the escape hatch).
|
||||
"""
|
||||
if not keyword or not isinstance(prompt, str):
|
||||
return False
|
||||
pattern = r"(?<![\w-])" + re.escape(keyword) + r"(?![\w-])"
|
||||
return re.search(pattern, prompt, re.IGNORECASE) is not None
|
||||
|
||||
|
||||
def _resolve_codex_dispatch_mode(config: dict) -> str:
|
||||
"""Normalize `codex.dispatch_mode` from .trellis/config.yaml to "auto" or "inline".
|
||||
|
||||
Defaults to `auto`. The legacy `sub-agent` value is an alias for `auto`.
|
||||
Any other explicit value (including invalid ones) falls back to `inline`
|
||||
without per-turn warnings. Shared by `_codex_mode_banner` (the per-turn
|
||||
banner) and `resolve_breadcrumb_key` (the breadcrumb tag key) so the two
|
||||
stay in lockstep.
|
||||
"""
|
||||
mode = "auto"
|
||||
if isinstance(config, dict):
|
||||
codex_cfg = config.get("codex")
|
||||
if isinstance(codex_cfg, dict):
|
||||
cfg_mode = str(codex_cfg.get("dispatch_mode", mode)).strip().lower()
|
||||
if cfg_mode == "inline":
|
||||
mode = "inline"
|
||||
elif cfg_mode in ("auto", "sub-agent"):
|
||||
mode = "auto"
|
||||
else:
|
||||
mode = "inline"
|
||||
return mode
|
||||
|
||||
|
||||
def _codex_mode_banner(config: dict) -> str:
|
||||
"""Emit a `<codex-mode>` banner for the additionalContext payload.
|
||||
|
||||
Reads `codex.dispatch_mode` from .trellis/config.yaml; defaults to
|
||||
`auto`, which dispatches Trellis sub-agents using native Codex context
|
||||
injection with a child-side fallback. This does not rely on inherited
|
||||
parent transcripts: `fork_turns` remains caller-controlled, and
|
||||
fresh-history sub-agents still receive their explicit delegated task and
|
||||
inherited session configuration. `inline` is an explicit opt-out; the
|
||||
legacy `sub-agent` value is an alias for `auto`. Invalid explicit values
|
||||
fall back to `inline` without per-turn warnings. The banner makes the
|
||||
active mode explicit to Codex AI per turn, complementing the workflow-state
|
||||
body which is per-status. Mode tells AI which dispatch protocol to follow;
|
||||
workflow-state tells AI what step it's at.
|
||||
"""
|
||||
mode = _resolve_codex_dispatch_mode(config)
|
||||
if mode == "auto":
|
||||
meaning = (
|
||||
"auto: implement/check work defaults to Trellis sub-agents; native Codex "
|
||||
"context injection is preferred and child-side loading is the fallback. "
|
||||
"The main session still coordinates, clarifies, updates specs, commits, and finishes."
|
||||
)
|
||||
else:
|
||||
meaning = (
|
||||
"inline: the main session implements/checks directly; "
|
||||
"do not dispatch implement/check sub-agents."
|
||||
)
|
||||
return f"<codex-mode>{meaning}</codex-mode>"
|
||||
|
||||
|
||||
def resolve_breadcrumb_key(
|
||||
status: str, platform: str | None, config: dict
|
||||
) -> str:
|
||||
"""Pick the breadcrumb tag key based on Codex dispatch_mode.
|
||||
|
||||
Codex defaults to ``auto`` and therefore uses the ordinary ``<status>``
|
||||
breadcrumb for native SubagentStart dispatch with child-side fallback;
|
||||
it does not depend on an inherited parent transcript. ``inline`` selects
|
||||
the parallel ``<status>-inline`` tag; ``sub-agent`` remains an alias for
|
||||
``auto``. Invalid explicit values fall back to inline without per-turn
|
||||
warnings.
|
||||
|
||||
Non-codex platforms return the plain status unchanged.
|
||||
"""
|
||||
if platform == "codex":
|
||||
mode = _resolve_codex_dispatch_mode(config)
|
||||
return f"{status}-inline" if mode == "inline" else status
|
||||
return status
|
||||
|
||||
|
||||
def build_breadcrumb(
|
||||
task_id: Optional[str],
|
||||
status: str,
|
||||
templates: dict[str, str],
|
||||
source: str | None = None,
|
||||
breadcrumb_key: str | None = None,
|
||||
) -> str:
|
||||
"""Build the <workflow-state>...</workflow-state> block.
|
||||
|
||||
- Known status (tag present in workflow.md) → detailed template body
|
||||
- Unknown status (no tag, or workflow.md missing) → generic
|
||||
"Refer to workflow.md for current step." line
|
||||
- `no_task` pseudo-status (task_id is None) → header omits task info
|
||||
"""
|
||||
lookup_key = breadcrumb_key or status
|
||||
body = templates.get(lookup_key)
|
||||
if body is None and lookup_key != status:
|
||||
body = templates.get(status)
|
||||
if body is None:
|
||||
body = "Refer to workflow.md for current step."
|
||||
header = f"Status: {status}" if task_id is None else f"Task: {task_id} ({status})"
|
||||
return f"<workflow-state>\n{header}\n{body}\n</workflow-state>"
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Entry
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def _load_hook_input() -> dict:
|
||||
"""Read hook JSON without trusting host runners to close stdin.
|
||||
|
||||
Kiro IDE `runCommand` and similar hook runners can leave stdin open while
|
||||
sending no payload. A plain `json.load(sys.stdin)` then blocks forever.
|
||||
Normal hook runners write the complete JSON payload and close stdin, so the
|
||||
short daemon read preserves that path while failing closed to `{}` for
|
||||
non-piping hosts.
|
||||
"""
|
||||
result_queue: "queue.Queue[str | Exception]" = queue.Queue(maxsize=1)
|
||||
|
||||
def _read() -> None:
|
||||
try:
|
||||
result_queue.put(sys.stdin.read())
|
||||
except Exception as exc:
|
||||
result_queue.put(exc)
|
||||
|
||||
reader = threading.Thread(target=_read, daemon=True)
|
||||
reader.start()
|
||||
try:
|
||||
raw = result_queue.get(timeout=0.2)
|
||||
except queue.Empty:
|
||||
return {}
|
||||
|
||||
if isinstance(raw, Exception):
|
||||
return {}
|
||||
try:
|
||||
data = json.loads(raw) if raw.strip() else {}
|
||||
except (json.JSONDecodeError, ValueError):
|
||||
return {}
|
||||
return data if isinstance(data, dict) else {}
|
||||
|
||||
|
||||
def main() -> int:
|
||||
if os.environ.get("TRELLIS_HOOKS") == "0" or os.environ.get("TRELLIS_DISABLE_HOOKS") == "1":
|
||||
return 0
|
||||
|
||||
data = _load_hook_input()
|
||||
|
||||
cwd_str = data.get("cwd") or os.getcwd()
|
||||
cwd = Path(cwd_str)
|
||||
|
||||
root = find_trellis_root(cwd)
|
||||
if root is None:
|
||||
return 0 # not a Trellis project
|
||||
|
||||
config = _read_trellis_config(root)
|
||||
if prompt_has_skip_keyword(data.get("prompt", ""), _resolve_skip_keyword(config)):
|
||||
return 0 # user opted out of the per-turn breadcrumb for this turn
|
||||
|
||||
templates = load_breadcrumbs(root)
|
||||
platform = _detect_platform(data)
|
||||
task = get_active_task(root, data)
|
||||
if task is None:
|
||||
# No active task — still emit a breadcrumb nudging AI toward
|
||||
# trellis-brainstorm + task.py create when user describes real work.
|
||||
no_task_key = resolve_breadcrumb_key("no_task", platform, config)
|
||||
breadcrumb = build_breadcrumb(
|
||||
None, "no_task", templates, breadcrumb_key=no_task_key
|
||||
)
|
||||
else:
|
||||
task_id, status, source = task
|
||||
status_key = resolve_breadcrumb_key(status, platform, config)
|
||||
source_for_breadcrumb = None if platform == "codex" else source
|
||||
breadcrumb = build_breadcrumb(
|
||||
task_id, status, templates, source_for_breadcrumb, breadcrumb_key=status_key
|
||||
)
|
||||
if platform == "codex":
|
||||
parts: list[str] = []
|
||||
if task is None:
|
||||
parts.append(CODEX_NO_TASK_BOOTSTRAP_NOTICE)
|
||||
parts.append(_codex_mode_banner(config))
|
||||
parts.append(breadcrumb)
|
||||
breadcrumb = "\n\n".join(parts)
|
||||
|
||||
# Kiro (CLI userPromptSubmit / IDE promptSubmit) adds a hook's stdout
|
||||
# directly to the conversation context — no JSON envelope. Emit the bare
|
||||
# breadcrumb text. Conditionally isolated: all other platforms keep the
|
||||
# hookSpecificOutput JSON path below unchanged.
|
||||
if platform == "kiro":
|
||||
print(breadcrumb)
|
||||
return 0
|
||||
|
||||
# Gemini CLI 0.40.x rejects "UserPromptSubmit" — its per-turn event is
|
||||
# named "BeforeAgent". Other platforms (Claude/Cursor/Qoder/CodeBuddy/
|
||||
# Droid/Codex/Copilot) accept the original Claude-style name.
|
||||
hook_event_name = (
|
||||
"BeforeAgent" if platform == "gemini" else "UserPromptSubmit"
|
||||
)
|
||||
|
||||
output = {
|
||||
"hookSpecificOutput": {
|
||||
"hookEventName": hook_event_name,
|
||||
"additionalContext": breadcrumb,
|
||||
}
|
||||
}
|
||||
print(json.dumps(output))
|
||||
return 0
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
sys.exit(main())
|
||||
551
.codex/hooks/session-start.py
Normal file
551
.codex/hooks/session-start.py
Normal file
@@ -0,0 +1,551 @@
|
||||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
Codex Session Start Hook - Inject Trellis context into Codex sessions.
|
||||
|
||||
Output format follows Codex hook protocol:
|
||||
stdout JSON → { hookSpecificOutput: { hookEventName: "SessionStart", additionalContext: "..." } }
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import os
|
||||
import re
|
||||
import subprocess
|
||||
import sys
|
||||
import warnings
|
||||
from io import StringIO
|
||||
from pathlib import Path
|
||||
|
||||
# Force UTF-8 on stdin/stdout/stderr on Windows. Default codepage there is
|
||||
# cp936 / cp1252 / etc. — non-ASCII content (Chinese task names, prd snippets)
|
||||
# both in stdin (hook payload from host CLI) and stdout (our emitted blocks)
|
||||
# raises UnicodeDecodeError / UnicodeEncodeError. Equivalent to `python -X utf8`
|
||||
# but applied per-stream so we don't depend on host CLI's command wiring.
|
||||
if sys.platform.startswith("win"):
|
||||
import io as _io
|
||||
for _stream_name in ("stdin", "stdout", "stderr"):
|
||||
_stream = getattr(sys, _stream_name, None)
|
||||
if _stream is None:
|
||||
continue
|
||||
if hasattr(_stream, "reconfigure"):
|
||||
try:
|
||||
_stream.reconfigure(encoding="utf-8", errors="replace") # type: ignore[union-attr]
|
||||
except Exception:
|
||||
pass # Optional Windows stream setup; keep hook startup non-fatal.
|
||||
elif hasattr(_stream, "detach"):
|
||||
try:
|
||||
setattr(sys, _stream_name, _io.TextIOWrapper(_stream.detach(), encoding="utf-8", errors="replace"))
|
||||
except Exception:
|
||||
pass # Optional Windows stream setup; keep hook startup non-fatal.
|
||||
|
||||
|
||||
def _normalize_windows_shell_path(path_str: str) -> str:
|
||||
"""Normalize Unix-style shell paths to real Windows paths.
|
||||
|
||||
On Windows, shells like Git Bash / MSYS2 / Cygwin may report paths like
|
||||
`/d/Users/...` or `/cygdrive/d/Users/...`. `Path.resolve()` will misinterpret
|
||||
these as `D:/d/Users...` on drive D: (or similar), breaking repo root
|
||||
detection.
|
||||
|
||||
This function is intentionally conservative: it only rewrites patterns that
|
||||
unambiguously represent a drive letter mount.
|
||||
"""
|
||||
if not isinstance(path_str, str) or not path_str:
|
||||
return path_str
|
||||
|
||||
# Only relevant on Windows; keep other platforms untouched.
|
||||
if not sys.platform.startswith("win"):
|
||||
return path_str
|
||||
|
||||
p = path_str.strip()
|
||||
|
||||
# Already a Windows drive path (C:\... or C:/...)
|
||||
if re.match(r"^[A-Za-z]:[\/]", p):
|
||||
return p
|
||||
|
||||
# MSYS/Git-Bash style: /c/Users/... or /d/Work/...
|
||||
m = re.match(r"^/([A-Za-z])/(.*)", p)
|
||||
if m:
|
||||
drive, rest = m.group(1).upper(), m.group(2)
|
||||
rest = rest.replace('/', '\\')
|
||||
return f"{drive}:\\{rest}"
|
||||
|
||||
# Cygwin style: /cygdrive/c/Users/...
|
||||
m = re.match(r"^/cygdrive/([A-Za-z])/(.*)", p)
|
||||
if m:
|
||||
drive, rest = m.group(1).upper(), m.group(2)
|
||||
rest = rest.replace('/', '\\')
|
||||
return f"{drive}:\\{rest}"
|
||||
|
||||
# WSL mounted drive (sometimes leaked into env): /mnt/c/Users/...
|
||||
m = re.match(r"^/mnt/([A-Za-z])/(.*)", p)
|
||||
if m:
|
||||
drive, rest = m.group(1).upper(), m.group(2)
|
||||
rest = rest.replace('/', '\\')
|
||||
return f"{drive}:\\{rest}"
|
||||
|
||||
return path_str
|
||||
|
||||
|
||||
warnings.filterwarnings("ignore")
|
||||
|
||||
FIRST_REPLY_NOTICE = """<first-reply-notice>
|
||||
On the first visible assistant reply in this session, briefly acknowledge that Trellis SessionStart context loaded.
|
||||
Choose the acknowledgment language in this order:
|
||||
1. Use the language of the user's current request (the user message that triggered this reply).
|
||||
2. If that request has no clear natural language, use an explicitly established project communication language.
|
||||
3. If neither provides a language, output the language-neutral fallback exactly: `Trellis SessionStart ✓`.
|
||||
Continue directly with the user's request after the acknowledgment.
|
||||
The acknowledgment must not alter the language used for the remainder of the response.
|
||||
This notice is one-shot: do not repeat it after the first visible assistant reply in this session.
|
||||
</first-reply-notice>"""
|
||||
|
||||
|
||||
def should_skip_injection() -> bool:
|
||||
if os.environ.get("TRELLIS_HOOKS") == "0":
|
||||
return True
|
||||
if os.environ.get("TRELLIS_DISABLE_HOOKS") == "1":
|
||||
return True
|
||||
return os.environ.get("CODEX_NON_INTERACTIVE") == "1"
|
||||
|
||||
|
||||
def configure_project_encoding(project_dir: Path) -> None:
|
||||
"""Reuse Trellis' shared Windows stdio encoding helper before JSON output."""
|
||||
scripts_dir = project_dir / ".trellis" / "scripts"
|
||||
if str(scripts_dir) not in sys.path:
|
||||
sys.path.insert(0, str(scripts_dir))
|
||||
|
||||
try:
|
||||
from common import configure_encoding # type: ignore[import-not-found]
|
||||
|
||||
configure_encoding()
|
||||
except Exception:
|
||||
pass # Optional encoding helper; host defaults are still usable.
|
||||
|
||||
|
||||
def _has_curated_jsonl_entry(jsonl_path: Path) -> bool:
|
||||
"""Return True iff jsonl has at least one row with a ``file`` field.
|
||||
|
||||
A freshly seeded jsonl only contains a ``{"_example": ...}`` row (no
|
||||
``file`` key) — that is NOT "ready". Readiness requires at least one
|
||||
curated entry. Matches the contract used by ``inject-subagent-context.py``.
|
||||
"""
|
||||
try:
|
||||
for line in jsonl_path.read_text(encoding="utf-8").splitlines():
|
||||
line = line.strip()
|
||||
if not line:
|
||||
continue
|
||||
try:
|
||||
row = json.loads(line)
|
||||
except json.JSONDecodeError:
|
||||
continue
|
||||
if isinstance(row, dict) and row.get("file"):
|
||||
return True
|
||||
except (OSError, UnicodeDecodeError):
|
||||
return False
|
||||
return False
|
||||
|
||||
|
||||
def read_file(path: Path, fallback: str = "") -> str:
|
||||
try:
|
||||
return path.read_text(encoding="utf-8")
|
||||
except (FileNotFoundError, PermissionError):
|
||||
return fallback
|
||||
|
||||
|
||||
def _resolve_context_key(project_dir: Path, hook_input: dict) -> str | None:
|
||||
scripts_dir = project_dir / ".trellis" / "scripts"
|
||||
if str(scripts_dir) not in sys.path:
|
||||
sys.path.insert(0, str(scripts_dir))
|
||||
try:
|
||||
from common.active_task import resolve_context_key # type: ignore[import-not-found]
|
||||
except Exception:
|
||||
return None
|
||||
return resolve_context_key(hook_input, platform="codex")
|
||||
|
||||
|
||||
def _resolve_active_task(trellis_dir: Path, hook_input: dict):
|
||||
scripts_dir = trellis_dir / "scripts"
|
||||
if str(scripts_dir) not in sys.path:
|
||||
sys.path.insert(0, str(scripts_dir))
|
||||
from common.active_task import resolve_active_task # type: ignore[import-not-found]
|
||||
|
||||
return resolve_active_task(trellis_dir.parent, hook_input, platform="codex")
|
||||
|
||||
|
||||
def run_script(script_path: Path, context_key: str | None = None) -> str:
|
||||
try:
|
||||
env = os.environ.copy()
|
||||
env["PYTHONIOENCODING"] = "utf-8"
|
||||
if context_key:
|
||||
env["TRELLIS_CONTEXT_ID"] = context_key
|
||||
cmd = [sys.executable, "-W", "ignore", str(script_path)]
|
||||
result = subprocess.run(
|
||||
cmd,
|
||||
capture_output=True,
|
||||
text=True,
|
||||
encoding="utf-8",
|
||||
errors="replace",
|
||||
timeout=5,
|
||||
cwd=str(script_path.parent.parent.parent),
|
||||
env=env,
|
||||
)
|
||||
return result.stdout if result.returncode == 0 else "No context available"
|
||||
except (subprocess.TimeoutExpired, FileNotFoundError, PermissionError):
|
||||
return "No context available"
|
||||
|
||||
|
||||
def _normalize_task_ref(task_ref: str) -> str:
|
||||
normalized = task_ref.strip()
|
||||
if not normalized:
|
||||
return ""
|
||||
|
||||
path_obj = Path(normalized)
|
||||
if path_obj.is_absolute():
|
||||
return str(path_obj)
|
||||
|
||||
normalized = normalized.replace("\\", "/")
|
||||
while normalized.startswith("./"):
|
||||
normalized = normalized[2:]
|
||||
|
||||
if normalized.startswith("tasks/"):
|
||||
return f".trellis/{normalized}"
|
||||
|
||||
return normalized
|
||||
|
||||
|
||||
def _resolve_task_dir(trellis_dir: Path, task_ref: str) -> Path:
|
||||
normalized = _normalize_task_ref(task_ref)
|
||||
path_obj = Path(normalized)
|
||||
if path_obj.is_absolute():
|
||||
return path_obj
|
||||
if normalized.startswith(".trellis/"):
|
||||
return trellis_dir.parent / path_obj
|
||||
return trellis_dir / "tasks" / path_obj
|
||||
|
||||
|
||||
def _get_task_status(trellis_dir: Path, hook_input: dict) -> str:
|
||||
active = _resolve_active_task(trellis_dir, hook_input)
|
||||
if not active.task_path:
|
||||
return (
|
||||
"Status: NO ACTIVE TASK\n"
|
||||
"Next: Classify the current turn and ask for task-creation consent "
|
||||
"before creating any Trellis task."
|
||||
)
|
||||
|
||||
task_ref = active.task_path
|
||||
task_dir = _resolve_task_dir(trellis_dir, task_ref)
|
||||
if active.stale or not task_dir.is_dir():
|
||||
return (
|
||||
f"Status: STALE POINTER\nTask: {task_ref}\n"
|
||||
"Next: Task directory not found. Run: python3 ./.trellis/scripts/task.py finish"
|
||||
)
|
||||
|
||||
task_json_path = task_dir / "task.json"
|
||||
task_data: dict = {}
|
||||
if task_json_path.is_file():
|
||||
try:
|
||||
task_data = json.loads(task_json_path.read_text(encoding="utf-8"))
|
||||
except (json.JSONDecodeError, PermissionError):
|
||||
pass # Optional task metadata; fall back to generic status.
|
||||
|
||||
task_title = task_data.get("title", task_ref)
|
||||
task_status = task_data.get("status", "unknown")
|
||||
|
||||
if task_status == "completed":
|
||||
return (
|
||||
f"Status: COMPLETED\nTask: {task_title}\n"
|
||||
f"Next: Archive with `python3 ./.trellis/scripts/task.py archive {task_dir.name}` "
|
||||
"or start a new task."
|
||||
)
|
||||
|
||||
has_prd = (task_dir / "prd.md").is_file()
|
||||
has_design = (task_dir / "design.md").is_file()
|
||||
has_implement = (task_dir / "implement.md").is_file()
|
||||
present = [
|
||||
name
|
||||
for name in ("prd.md", "design.md", "implement.md", "implement.jsonl", "check.jsonl")
|
||||
if (task_dir / name).is_file()
|
||||
]
|
||||
present_line = ", ".join(present) if present else "none"
|
||||
|
||||
if not has_prd:
|
||||
return (
|
||||
f"Status: PLANNING\nTask: {task_title}\nPresent: {present_line}\n"
|
||||
"Next: Load trellis-brainstorm and write prd.md. Stay in planning."
|
||||
)
|
||||
|
||||
if task_status == "planning":
|
||||
if has_design and has_implement:
|
||||
next_action = "Review planning artifacts with the user before `task.py start`."
|
||||
else:
|
||||
next_action = (
|
||||
"Lightweight task can ask for start review with PRD-only; "
|
||||
"complex task must add design.md and implement.md before `task.py start`."
|
||||
)
|
||||
return (
|
||||
f"Status: PLANNING\nTask: {task_title}\nPresent: {present_line}\n"
|
||||
f"Next: {next_action}"
|
||||
)
|
||||
|
||||
return (
|
||||
f"Status: {task_status.upper()}\nTask: {task_title}\nPresent: {present_line}\n"
|
||||
"Next: Follow the matching per-turn workflow-state. Context order is jsonl entries, "
|
||||
"prd.md, design.md if present, implement.md if present."
|
||||
)
|
||||
|
||||
|
||||
def _run_git(repo_root: Path, args: list[str]) -> str:
|
||||
try:
|
||||
result = subprocess.run(
|
||||
["git", *args],
|
||||
capture_output=True,
|
||||
text=True,
|
||||
encoding="utf-8",
|
||||
errors="replace",
|
||||
timeout=3,
|
||||
cwd=str(repo_root),
|
||||
)
|
||||
except (subprocess.TimeoutExpired, FileNotFoundError, PermissionError):
|
||||
return ""
|
||||
if result.returncode != 0:
|
||||
return ""
|
||||
return result.stdout.strip()
|
||||
|
||||
|
||||
def _format_git_state(repo_root: Path) -> str:
|
||||
branch = _run_git(repo_root, ["branch", "--show-current"]) or "(detached)"
|
||||
dirty_lines = [
|
||||
line for line in _run_git(repo_root, ["status", "--porcelain"]).splitlines()
|
||||
if line.strip()
|
||||
]
|
||||
dirty_text = "clean" if not dirty_lines else f"dirty {len(dirty_lines)} paths"
|
||||
return f"Git: branch {branch}; {dirty_text}."
|
||||
|
||||
|
||||
def _repo_relative(repo_root: Path, path: Path) -> str:
|
||||
try:
|
||||
return path.relative_to(repo_root).as_posix()
|
||||
except ValueError:
|
||||
return str(path)
|
||||
|
||||
|
||||
def _collect_spec_index_paths(trellis_dir: Path) -> list[str]:
|
||||
paths: list[str] = []
|
||||
guides_index = trellis_dir / "spec" / "guides" / "index.md"
|
||||
if guides_index.is_file():
|
||||
paths.append(".trellis/spec/guides/index.md")
|
||||
|
||||
spec_dir = trellis_dir / "spec"
|
||||
if not spec_dir.is_dir():
|
||||
return paths
|
||||
|
||||
for sub in sorted(spec_dir.iterdir()):
|
||||
if not sub.is_dir() or sub.name.startswith(".") or sub.name == "guides":
|
||||
continue
|
||||
index_file = sub / "index.md"
|
||||
if index_file.is_file():
|
||||
paths.append(f".trellis/spec/{sub.name}/index.md")
|
||||
continue
|
||||
for nested in sorted(sub.iterdir()):
|
||||
if not nested.is_dir():
|
||||
continue
|
||||
nested_index = nested / "index.md"
|
||||
if nested_index.is_file():
|
||||
paths.append(f".trellis/spec/{sub.name}/{nested.name}/index.md")
|
||||
|
||||
return paths
|
||||
|
||||
|
||||
def _build_compact_current_state(
|
||||
trellis_dir: Path,
|
||||
hook_input: dict,
|
||||
spec_index_paths: list[str],
|
||||
) -> str:
|
||||
repo_root = trellis_dir.parent
|
||||
lines: list[str] = []
|
||||
|
||||
try:
|
||||
from common.paths import get_active_journal_file, get_developer, get_tasks_dir, count_lines # type: ignore[import-not-found]
|
||||
from common.tasks import iter_active_tasks # type: ignore[import-not-found]
|
||||
except Exception:
|
||||
get_active_journal_file = None # type: ignore[assignment]
|
||||
get_developer = None # type: ignore[assignment]
|
||||
get_tasks_dir = None # type: ignore[assignment]
|
||||
count_lines = None # type: ignore[assignment]
|
||||
iter_active_tasks = None # type: ignore[assignment]
|
||||
|
||||
developer = get_developer(repo_root) if get_developer else None
|
||||
lines.append(f"Developer: {developer or '(not initialized)'}")
|
||||
lines.append(_format_git_state(repo_root))
|
||||
|
||||
active = _resolve_active_task(trellis_dir, hook_input)
|
||||
if active.task_path:
|
||||
task_dir = _resolve_task_dir(trellis_dir, active.task_path)
|
||||
status = "unknown"
|
||||
task_json = task_dir / "task.json"
|
||||
if task_json.is_file():
|
||||
try:
|
||||
data = json.loads(task_json.read_text(encoding="utf-8"))
|
||||
if isinstance(data, dict):
|
||||
status = str(data.get("status") or "unknown")
|
||||
except (json.JSONDecodeError, OSError):
|
||||
pass # Optional task metadata; fall back to generic status.
|
||||
lines.append(f"Current task: {_repo_relative(repo_root, task_dir)}; status={status}.")
|
||||
else:
|
||||
lines.append("Current task: none.")
|
||||
|
||||
if get_tasks_dir and iter_active_tasks:
|
||||
try:
|
||||
task_count = sum(1 for _ in iter_active_tasks(get_tasks_dir(repo_root)))
|
||||
lines.append(
|
||||
f"Active tasks: {task_count} total. Use `python3 ./.trellis/scripts/task.py list --mine` only if needed."
|
||||
)
|
||||
except Exception:
|
||||
pass # Optional task summary; keep compact state available.
|
||||
|
||||
if get_active_journal_file and count_lines:
|
||||
journal = get_active_journal_file(repo_root)
|
||||
if journal:
|
||||
lines.append(
|
||||
f"Journal: {_repo_relative(repo_root, journal)}, {count_lines(journal)} / 2000 lines."
|
||||
)
|
||||
|
||||
if spec_index_paths:
|
||||
lines.append(f"Spec indexes: {len(spec_index_paths)} available.")
|
||||
|
||||
return "\n".join(lines)
|
||||
|
||||
|
||||
def _extract_range(content: str, start_header: str, end_header: str) -> str:
|
||||
"""Extract lines starting at `## start_header` up to (but excluding) `## end_header`."""
|
||||
lines = content.splitlines()
|
||||
start: "int | None" = None
|
||||
end: int = len(lines)
|
||||
start_match = f"## {start_header}"
|
||||
end_match = f"## {end_header}"
|
||||
for i, line in enumerate(lines):
|
||||
stripped = line.strip()
|
||||
if start is None and stripped == start_match:
|
||||
start = i
|
||||
continue
|
||||
if start is not None and stripped == end_match:
|
||||
end = i
|
||||
break
|
||||
if start is None:
|
||||
return ""
|
||||
return "\n".join(lines[start:end]).rstrip()
|
||||
|
||||
|
||||
_BREADCRUMB_TAG_RE = re.compile(
|
||||
r"\[workflow-state:([A-Za-z0-9_-]+)\]\s*\n.*?\n\s*\[/workflow-state:\1\]",
|
||||
re.DOTALL,
|
||||
)
|
||||
|
||||
|
||||
def _strip_breadcrumb_tag_blocks(content: str) -> str:
|
||||
stripped = _BREADCRUMB_TAG_RE.sub("", content)
|
||||
stripped = re.sub(r"<!--.*?-->", "", stripped, flags=re.DOTALL)
|
||||
stripped = re.sub(r"^\[(?!/?workflow-state:)/?[^\]\n]+\]\s*\n?", "", stripped, flags=re.MULTILINE)
|
||||
return re.sub(r"\n{3,}", "\n\n", stripped).strip()
|
||||
|
||||
|
||||
def _build_workflow_toc(workflow_path: Path) -> str:
|
||||
"""Inject only the compact Phase Index summary for SessionStart."""
|
||||
content = read_file(workflow_path)
|
||||
if not content:
|
||||
return "No workflow.md found"
|
||||
|
||||
out_lines = [
|
||||
"# Development Workflow - Session Summary",
|
||||
"Full guide: .trellis/workflow.md. Step detail: `python3 ./.trellis/scripts/get_context.py --mode phase --step <X.Y>`.",
|
||||
"",
|
||||
]
|
||||
|
||||
phases = _extract_range(content, "Phase Index", "Phase 1: Plan")
|
||||
if phases:
|
||||
out_lines.append(_strip_breadcrumb_tag_blocks(phases).rstrip())
|
||||
|
||||
return "\n".join(out_lines).rstrip()
|
||||
|
||||
|
||||
def main() -> None:
|
||||
if should_skip_injection():
|
||||
sys.exit(0)
|
||||
|
||||
# Read hook input from stdin
|
||||
try:
|
||||
hook_input = json.loads(sys.stdin.read())
|
||||
if not isinstance(hook_input, dict):
|
||||
hook_input = {}
|
||||
project_dir = Path(_normalize_windows_shell_path(hook_input.get("cwd", "."))).resolve()
|
||||
except (json.JSONDecodeError, KeyError):
|
||||
hook_input = {}
|
||||
project_dir = Path(".").resolve()
|
||||
|
||||
configure_project_encoding(project_dir)
|
||||
|
||||
trellis_dir = project_dir / ".trellis"
|
||||
spec_index_paths = _collect_spec_index_paths(trellis_dir)
|
||||
|
||||
output = StringIO()
|
||||
|
||||
output.write("""<session-context>
|
||||
Trellis compact SessionStart context. Use it to orient the session; load details on demand.
|
||||
</session-context>
|
||||
|
||||
""")
|
||||
output.write(FIRST_REPLY_NOTICE)
|
||||
output.write("\n\n")
|
||||
|
||||
output.write("<current-state>\n")
|
||||
output.write(_build_compact_current_state(trellis_dir, hook_input, spec_index_paths))
|
||||
output.write("\n</current-state>\n\n")
|
||||
|
||||
output.write("<trellis-workflow>\n")
|
||||
output.write(_build_workflow_toc(trellis_dir / "workflow.md"))
|
||||
output.write("\n</trellis-workflow>\n\n")
|
||||
|
||||
output.write("<guidelines>\n")
|
||||
output.write(
|
||||
"Task context order for implementation/check: jsonl entries -> `prd.md` -> "
|
||||
"`design.md if present` -> `implement.md if present`. Missing optional artifacts "
|
||||
"are skipped for lightweight tasks.\n\n"
|
||||
)
|
||||
|
||||
if spec_index_paths:
|
||||
output.write("## Available indexes (read on demand)\n")
|
||||
for p in spec_index_paths:
|
||||
output.write(f"- {p}\n")
|
||||
output.write("\n")
|
||||
|
||||
output.write(
|
||||
"Discover more via: "
|
||||
"`python3 ./.trellis/scripts/get_context.py --mode packages`\n"
|
||||
)
|
||||
output.write("</guidelines>\n\n")
|
||||
|
||||
task_status = _get_task_status(trellis_dir, hook_input)
|
||||
output.write(f"<task-status>\n{task_status}\n</task-status>\n\n")
|
||||
|
||||
output.write("""<ready>
|
||||
Context loaded. Follow <task-status>. Load workflow/spec/task details only when needed.
|
||||
</ready>""")
|
||||
|
||||
context = output.getvalue()
|
||||
result = {
|
||||
"suppressOutput": True,
|
||||
"systemMessage": f"Trellis context injected ({len(context)} chars)",
|
||||
"hookSpecificOutput": {
|
||||
"hookEventName": "SessionStart",
|
||||
"additionalContext": context,
|
||||
},
|
||||
}
|
||||
|
||||
print(json.dumps(result, ensure_ascii=False), flush=True)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user