feat: init mlkem project with Verilator test framework
- sync_rtl/common/: skid_buffer, pipeline_reg, defines (valid/ready) - sync_rtl/mod_add/: modular adder example with Verilator C++ TB - test_framework/: Python-driven Verilator compile/sim/compare pipeline - test_framework/modules/mod_add/: 50-vector test plan, full鏈路 PASS - .trellis/spec/: RTL and test_framework conventions documented
This commit is contained in:
497
.opencode/plugins/inject-subagent-context.js
Normal file
497
.opencode/plugins/inject-subagent-context.js
Normal file
@@ -0,0 +1,497 @@
|
||||
/* global process */
|
||||
/**
|
||||
* Trellis Context Injection Plugin
|
||||
*
|
||||
* Injects context when Task tool is called with supported subagent types.
|
||||
* Uses OpenCode's tool.execute.before hook.
|
||||
*/
|
||||
|
||||
import { existsSync, readdirSync } from "fs"
|
||||
import { join } from "path"
|
||||
import { TrellisContext, debugLog } from "../lib/trellis-context.js"
|
||||
|
||||
// Supported subagent types
|
||||
const AGENTS_ALL = ["implement", "check", "research"]
|
||||
const AGENTS_REQUIRE_TASK = ["implement", "check"]
|
||||
|
||||
// Match `Active task: <path>` on the first non-empty line of the dispatch
|
||||
// prompt. Mirrors the contract in workflow.md's [workflow-state:in_progress]
|
||||
// breadcrumb so multi-window users can disambiguate which task is targeted.
|
||||
const ACTIVE_TASK_HINT_RE = /^\s*Active task:\s*(\S+)\s*$/m
|
||||
|
||||
function extractActiveTaskHint(prompt) {
|
||||
if (typeof prompt !== "string" || !prompt) return null
|
||||
const match = prompt.match(ACTIVE_TASK_HINT_RE)
|
||||
return match ? match[1].trim() : null
|
||||
}
|
||||
|
||||
/**
|
||||
* Get context for implement agent. `taskDir` may be relative
|
||||
* (`.trellis/tasks/foo`) or absolute; both are resolved via
|
||||
* `ctx.resolveTaskDir`.
|
||||
*/
|
||||
function getImplementContext(ctx, taskDir) {
|
||||
const parts = []
|
||||
const taskDirFull = ctx.resolveTaskDir(taskDir)
|
||||
if (!taskDirFull) return ""
|
||||
|
||||
const jsonlPath = join(taskDirFull, "implement.jsonl")
|
||||
const entries = ctx.readJsonlWithFiles(jsonlPath)
|
||||
if (entries.length > 0) {
|
||||
parts.push(ctx.buildContextFromEntries(entries))
|
||||
}
|
||||
|
||||
const prd = ctx.readFile(join(taskDirFull, "prd.md"))
|
||||
if (prd) {
|
||||
parts.push(`=== ${taskDir}/prd.md (Requirements) ===\n${prd}`)
|
||||
}
|
||||
|
||||
const info = ctx.readFile(join(taskDirFull, "info.md"))
|
||||
if (info) {
|
||||
parts.push(`=== ${taskDir}/info.md (Technical Design) ===\n${info}`)
|
||||
}
|
||||
|
||||
return parts.join("\n\n")
|
||||
}
|
||||
|
||||
/**
|
||||
* Get context for check agent. `taskDir` may be relative or absolute.
|
||||
*/
|
||||
function getCheckContext(ctx, taskDir) {
|
||||
const parts = []
|
||||
const taskDirFull = ctx.resolveTaskDir(taskDir)
|
||||
if (!taskDirFull) return ""
|
||||
|
||||
const jsonlPath = join(taskDirFull, "check.jsonl")
|
||||
const entries = ctx.readJsonlWithFiles(jsonlPath)
|
||||
if (entries.length > 0) {
|
||||
parts.push(ctx.buildContextFromEntries(entries))
|
||||
}
|
||||
|
||||
const prd = ctx.readFile(join(taskDirFull, "prd.md"))
|
||||
if (prd) {
|
||||
parts.push(`=== ${taskDir}/prd.md (Requirements) ===\n${prd}`)
|
||||
}
|
||||
|
||||
return parts.join("\n\n")
|
||||
}
|
||||
|
||||
/**
|
||||
* Get context for finish phase (final check before PR)
|
||||
*/
|
||||
function getFinishContext(ctx, taskDir) {
|
||||
// Finish reuses check context (same JSONL source)
|
||||
return getCheckContext(ctx, taskDir)
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* Get context for research agent
|
||||
*/
|
||||
function getResearchContext(ctx) {
|
||||
const parts = []
|
||||
|
||||
// Dynamic project structure (scan actual spec directory)
|
||||
const specPath = ".trellis/spec"
|
||||
const specFull = join(ctx.directory, specPath)
|
||||
|
||||
const structureLines = [`## Project Spec Directory Structure\n\n\`\`\`\n${specPath}/`]
|
||||
if (existsSync(specFull)) {
|
||||
try {
|
||||
const entries = readdirSync(specFull, { withFileTypes: true })
|
||||
.filter(d => d.isDirectory() && !d.name.startsWith("."))
|
||||
.sort((a, b) => a.name.localeCompare(b.name))
|
||||
|
||||
for (const entry of entries) {
|
||||
const entryPath = join(specFull, entry.name)
|
||||
if (existsSync(join(entryPath, "index.md"))) {
|
||||
structureLines.push(`├── ${entry.name}/`)
|
||||
} else {
|
||||
try {
|
||||
const nested = readdirSync(entryPath, { withFileTypes: true })
|
||||
.filter(d => d.isDirectory() && existsSync(join(entryPath, d.name, "index.md")))
|
||||
.sort((a, b) => a.name.localeCompare(b.name))
|
||||
if (nested.length > 0) {
|
||||
structureLines.push(`├── ${entry.name}/`)
|
||||
for (const n of nested) {
|
||||
structureLines.push(`│ ├── ${n.name}/`)
|
||||
}
|
||||
}
|
||||
} catch {
|
||||
// Ignore nested read errors
|
||||
}
|
||||
}
|
||||
}
|
||||
} catch {
|
||||
// Ignore read errors
|
||||
}
|
||||
}
|
||||
structureLines.push("```")
|
||||
|
||||
parts.push(structureLines.join("\n") + `
|
||||
|
||||
## Search Tips
|
||||
|
||||
- Spec files: \`.trellis/spec/**/*.md\`
|
||||
- Known issues: \`.trellis/big-question/\`
|
||||
- Code search: Use Glob and Grep tools
|
||||
- Tech solutions: Use mcp__exa__web_search_exa or mcp__exa__get_code_context_exa`)
|
||||
|
||||
return parts.join("\n\n")
|
||||
}
|
||||
|
||||
/**
|
||||
* Build enhanced prompt with context
|
||||
*/
|
||||
function buildPrompt(agentType, originalPrompt, context, isFinish = false) {
|
||||
const templates = {
|
||||
implement: `<!-- trellis-hook-injected -->
|
||||
# Implement Agent Task
|
||||
|
||||
You are the Implement Agent in the Multi-Agent Pipeline.
|
||||
|
||||
## Your Context
|
||||
|
||||
${context}
|
||||
|
||||
---
|
||||
|
||||
## Your Task
|
||||
|
||||
${originalPrompt}
|
||||
|
||||
---
|
||||
|
||||
## Workflow
|
||||
|
||||
1. **Understand specs** - All dev specs are injected above
|
||||
2. **Understand requirements** - Read requirements and technical design
|
||||
3. **Implement feature** - Follow specs and design
|
||||
4. **Self-check** - Ensure code quality
|
||||
|
||||
## Important Constraints
|
||||
|
||||
- Do NOT execute git commit
|
||||
- Follow all dev specs injected above
|
||||
- Report list of modified/created files when done`,
|
||||
|
||||
check: isFinish ? `<!-- trellis-hook-injected -->
|
||||
# Finish Agent Task
|
||||
|
||||
You are performing the final check before creating a PR.
|
||||
|
||||
## Your Context
|
||||
|
||||
${context}
|
||||
|
||||
---
|
||||
|
||||
## Your Task
|
||||
|
||||
${originalPrompt}
|
||||
|
||||
---
|
||||
|
||||
## Workflow
|
||||
|
||||
1. **Review changes** - Run \`git diff --name-only\` to see all changed files
|
||||
2. **Verify requirements** - Check each requirement in prd.md is implemented
|
||||
3. **Spec sync** - Analyze whether changes introduce new patterns, contracts, or conventions
|
||||
- If new pattern/convention found: read target spec file → update it → update index.md if needed
|
||||
- If infra/cross-layer change: follow the 7-section mandatory template from update-spec.md
|
||||
- If pure code fix with no new patterns: skip this step
|
||||
4. **Run final checks** - Execute lint and typecheck
|
||||
5. **Confirm ready** - Ensure code is ready for PR
|
||||
|
||||
## Important Constraints
|
||||
|
||||
- You MAY update spec files when gaps are detected (use update-spec.md as guide)
|
||||
- MUST read the target spec file BEFORE editing (avoid duplicating existing content)
|
||||
- Do NOT update specs for trivial changes (typos, formatting, obvious fixes)
|
||||
- If critical CODE issues found, report them clearly (fix specs, not code)
|
||||
- Verify all acceptance criteria in prd.md are met` :
|
||||
`<!-- trellis-hook-injected -->
|
||||
# Check Agent Task
|
||||
|
||||
You are the Check Agent in the Multi-Agent Pipeline.
|
||||
|
||||
## Your Context
|
||||
|
||||
${context}
|
||||
|
||||
---
|
||||
|
||||
## Your Task
|
||||
|
||||
${originalPrompt}
|
||||
|
||||
---
|
||||
|
||||
## Workflow
|
||||
|
||||
1. **Get changes** - Run \`git diff --name-only\` and \`git diff\`
|
||||
2. **Check against specs** - Check item by item
|
||||
3. **Self-fix** - Fix issues directly, don't just report
|
||||
4. **Run verification** - Run lint and typecheck
|
||||
|
||||
## Important Constraints
|
||||
|
||||
- Fix issues yourself, don't just report
|
||||
- Must execute complete checklist`,
|
||||
|
||||
research: `<!-- trellis-hook-injected -->
|
||||
# Research Agent Task
|
||||
|
||||
You are the Research Agent in the Multi-Agent Pipeline.
|
||||
|
||||
## Core Principle
|
||||
|
||||
**You do one thing: find and explain information.**
|
||||
|
||||
## Project Info
|
||||
|
||||
${context}
|
||||
|
||||
---
|
||||
|
||||
## Your Task
|
||||
|
||||
${originalPrompt}
|
||||
|
||||
---
|
||||
|
||||
## Workflow
|
||||
|
||||
1. **Understand query** - Determine search type and scope
|
||||
2. **Plan search** - List search steps
|
||||
3. **Execute search** - Run multiple searches in parallel
|
||||
4. **Organize results** - Output structured report
|
||||
|
||||
## Strict Boundaries
|
||||
|
||||
**Only allowed**: Describe what exists, where it is, how it works
|
||||
|
||||
**Forbidden**: Suggest improvements, criticize implementation, modify files`
|
||||
}
|
||||
|
||||
return templates[agentType] || originalPrompt
|
||||
}
|
||||
|
||||
function shellQuote(value) {
|
||||
return `'${String(value).replace(/'/g, "'\\''")}'`
|
||||
}
|
||||
|
||||
function powershellQuote(value) {
|
||||
return `'${String(value).replace(/'/g, "''")}'`
|
||||
}
|
||||
|
||||
function envValue(env, key) {
|
||||
const value = env?.[key]
|
||||
return typeof value === "string" && value.trim() ? value.trim() : null
|
||||
}
|
||||
|
||||
function shellBasename(value) {
|
||||
return value.replace(/\\/g, "/").split("/").pop()?.toLowerCase() || ""
|
||||
}
|
||||
|
||||
function isWindowsPosixShell(env = process.env) {
|
||||
if (envValue(env, "MSYSTEM")) return true
|
||||
if (envValue(env, "MINGW_PREFIX")) return true
|
||||
if (envValue(env, "OPENCODE_GIT_BASH_PATH")) return true
|
||||
|
||||
const ostype = envValue(env, "OSTYPE")?.toLowerCase() || ""
|
||||
if (/(msys|mingw|cygwin)/.test(ostype)) return true
|
||||
|
||||
const shell = shellBasename(envValue(env, "SHELL") || "")
|
||||
return /^(bash|sh|zsh)(\.exe)?$/.test(shell)
|
||||
}
|
||||
|
||||
function buildTrellisContextPrefix(contextKey, hostPlatform = process.platform, env = process.env) {
|
||||
if (hostPlatform === "win32" && !isWindowsPosixShell(env)) {
|
||||
return `$env:TRELLIS_CONTEXT_ID = ${powershellQuote(contextKey)}; `
|
||||
}
|
||||
|
||||
return `export TRELLIS_CONTEXT_ID=${shellQuote(contextKey)}; `
|
||||
}
|
||||
|
||||
function getBashCommandKey(args) {
|
||||
if (!args || typeof args !== "object") return null
|
||||
if (typeof args.command === "string") return "command"
|
||||
if (typeof args.cmd === "string") return "cmd"
|
||||
return null
|
||||
}
|
||||
|
||||
function commandStartsWithTrellisContext(command) {
|
||||
const firstCommand = command.trimStart().split(/[;&|]/, 1)[0].trimStart()
|
||||
return (
|
||||
/^TRELLIS_CONTEXT_ID\s*=/.test(firstCommand) ||
|
||||
/^export\s+TRELLIS_CONTEXT_ID\s*=/.test(firstCommand) ||
|
||||
/^env\s+(?:(?:-\S+|[A-Za-z_][A-Za-z0-9_]*=\S*)\s+)*TRELLIS_CONTEXT_ID\s*=/.test(firstCommand) ||
|
||||
/^\$env:TRELLIS_CONTEXT_ID\s*=/i.test(firstCommand)
|
||||
)
|
||||
}
|
||||
|
||||
/**
|
||||
* OpenCode TUI may not expose OPENCODE_RUN_ID to Bash. The plugin hook still
|
||||
* receives session identity, so inject it into Bash commands before execution.
|
||||
*/
|
||||
function injectTrellisContextIntoBash(ctx, input, output, hostPlatform, env) {
|
||||
const args = output?.args
|
||||
const commandKey = getBashCommandKey(args)
|
||||
if (!commandKey) return false
|
||||
|
||||
const command = args[commandKey]
|
||||
if (!command.trim()) return false
|
||||
if (commandStartsWithTrellisContext(command)) return false
|
||||
|
||||
const contextKey = ctx.getContextKey(input)
|
||||
if (!contextKey) return false
|
||||
|
||||
args[commandKey] = `${buildTrellisContextPrefix(contextKey, hostPlatform, env)}${command}`
|
||||
return true
|
||||
}
|
||||
|
||||
// OpenCode plugin factory: `export default async (input) => hooks`.
|
||||
// OpenCode 1.2.x iterates every module export and invokes it as a function
|
||||
// (packages/opencode/src/plugin/index.ts — `for ([_, fn] of Object.entries(mod)) await fn(input)`);
|
||||
// the previous `{ id, server }` object shape failed with
|
||||
// `TypeError: fn is not a function` in 1.2.x.
|
||||
export default async ({ directory, platform: hostPlatform = process.platform, env = process.env }) => {
|
||||
const ctx = new TrellisContext(directory)
|
||||
debugLog("inject", "Plugin loaded, directory:", directory)
|
||||
|
||||
return {
|
||||
"tool.execute.before": async (input, output) => {
|
||||
try {
|
||||
if (process.env.TRELLIS_HOOKS === "0" || process.env.TRELLIS_DISABLE_HOOKS === "1") {
|
||||
return
|
||||
}
|
||||
debugLog("inject", "tool.execute.before called, tool:", input?.tool)
|
||||
|
||||
const toolName = input?.tool?.toLowerCase()
|
||||
if (toolName === "bash") {
|
||||
if (injectTrellisContextIntoBash(ctx, input, output, hostPlatform, env)) {
|
||||
debugLog("inject", "Injected TRELLIS_CONTEXT_ID into Bash command")
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
if (toolName !== "task") {
|
||||
return
|
||||
}
|
||||
|
||||
const args = output?.args
|
||||
if (!args) return
|
||||
|
||||
const rawSubagentType = args.subagent_type
|
||||
// Strip "trellis-" prefix added by v0.5.0-beta.5 agent rename migration
|
||||
const subagentType = (rawSubagentType || "").replace(/^trellis-/, "")
|
||||
const originalPrompt = args.prompt || ""
|
||||
|
||||
debugLog("inject", "Task tool called, subagent_type:", rawSubagentType)
|
||||
|
||||
if (!AGENTS_ALL.includes(subagentType)) {
|
||||
debugLog("inject", "Skipping - unsupported subagent_type")
|
||||
return
|
||||
}
|
||||
|
||||
// Resolve active task in this priority order (only later steps
|
||||
// run when earlier ones miss):
|
||||
// 1. Exact session runtime context lookup for input.sessionID
|
||||
// 2. `Active task: <path>` hint in the dispatch prompt
|
||||
// (explicit per-dispatch override — beats single-session
|
||||
// inference so multi-window users can disambiguate)
|
||||
// 3. Single-session fallback — only when exactly 1 session
|
||||
// runtime file exists locally
|
||||
let taskDir = null
|
||||
let taskSource = null
|
||||
|
||||
const contextKey = ctx.getContextKey(input)
|
||||
if (contextKey) {
|
||||
const context = ctx.readContext(contextKey)
|
||||
const exactRef = ctx.normalizeTaskRef(context?.current_task || "")
|
||||
if (exactRef) {
|
||||
taskDir = exactRef
|
||||
taskSource = `session:${contextKey}`
|
||||
}
|
||||
}
|
||||
|
||||
if (!taskDir) {
|
||||
const hintRef = extractActiveTaskHint(originalPrompt)
|
||||
if (hintRef) {
|
||||
const hintNormalized = ctx.normalizeTaskRef(hintRef)
|
||||
if (hintNormalized) {
|
||||
const hintDir = ctx.resolveTaskDir(hintNormalized)
|
||||
if (hintDir && existsSync(hintDir)) {
|
||||
taskDir = hintNormalized
|
||||
taskSource = "prompt-hint"
|
||||
debugLog("inject", "Resolved task from Active task: hint:", hintNormalized)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (!taskDir) {
|
||||
const fallback = ctx._resolveSingleSessionFallback()
|
||||
if (fallback?.taskPath) {
|
||||
const fallbackDir = ctx.resolveTaskDir(fallback.taskPath)
|
||||
if (fallbackDir && existsSync(fallbackDir)) {
|
||||
taskDir = fallback.taskPath
|
||||
taskSource = fallback.source
|
||||
debugLog("inject", "Resolved task via single-session fallback:", taskDir, "source:", taskSource)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Agents requiring task directory
|
||||
if (AGENTS_REQUIRE_TASK.includes(subagentType)) {
|
||||
// subagentType is already stripped of "trellis-" prefix above
|
||||
if (!taskDir) {
|
||||
debugLog("inject", "Skipping - no current task")
|
||||
return
|
||||
}
|
||||
const taskDirFull = ctx.resolveTaskDir(taskDir)
|
||||
if (!taskDirFull || !existsSync(taskDirFull)) {
|
||||
debugLog("inject", "Skipping - task directory not found")
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
// Check for [finish] marker
|
||||
const isFinish = originalPrompt.toLowerCase().includes("[finish]")
|
||||
|
||||
// Get context based on agent type
|
||||
let context = ""
|
||||
switch (subagentType) {
|
||||
case "implement":
|
||||
context = getImplementContext(ctx, taskDir)
|
||||
break
|
||||
case "check":
|
||||
context = isFinish
|
||||
? getFinishContext(ctx, taskDir)
|
||||
: getCheckContext(ctx, taskDir)
|
||||
break
|
||||
case "research":
|
||||
context = getResearchContext(ctx, taskDir)
|
||||
break
|
||||
}
|
||||
|
||||
if (!context) {
|
||||
debugLog("inject", "No context to inject")
|
||||
return
|
||||
}
|
||||
|
||||
const newPrompt = buildPrompt(subagentType, originalPrompt, context, isFinish)
|
||||
|
||||
// Mutate args in-place — whole-object replacement does NOT work for the task tool
|
||||
// because the runtime holds a local reference to the same args object.
|
||||
args.prompt = newPrompt
|
||||
|
||||
debugLog("inject", "Injected context for", subagentType, "prompt length:", newPrompt.length)
|
||||
|
||||
} catch (error) {
|
||||
debugLog("inject", "Error in tool.execute.before:", error.message, error.stack)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user