mirror of
https://github.com/github/spec-kit.git
synced 2026-07-04 04:45:43 +08:00
Extension command registration now resolves the active skills directory before writing command artifacts. This lets initialized skills-backed agents recover a missing active skills directory while preserving the existing preset registration behavior. Add regression coverage for missing active skills directories, shared skills directories, and symlinked parent guards. Fixes #2769. Co-authored-by: OpenAI Codex <codex@openai.com>
1083 lines
42 KiB
Python
1083 lines
42 KiB
Python
"""
|
|
Agent Command Registrar for Spec Kit
|
|
|
|
Shared infrastructure for registering commands with AI agents.
|
|
Used by both the extension system and the preset system to write
|
|
command files into agent-specific directories in the correct format.
|
|
"""
|
|
|
|
import os
|
|
import platform
|
|
import re
|
|
from copy import deepcopy
|
|
from pathlib import Path
|
|
from typing import Any, Dict, List, Optional
|
|
|
|
import yaml
|
|
|
|
from ._init_options import is_ai_skills_enabled, load_init_options
|
|
|
|
|
|
def _build_agent_configs() -> dict[str, Any]:
|
|
"""Derive CommandRegistrar.AGENT_CONFIGS from INTEGRATION_REGISTRY."""
|
|
from specify_cli.integrations import INTEGRATION_REGISTRY
|
|
|
|
configs: dict[str, dict[str, Any]] = {}
|
|
for key, integration in INTEGRATION_REGISTRY.items():
|
|
if key == "generic":
|
|
continue
|
|
if integration.registrar_config:
|
|
config = dict(integration.registrar_config)
|
|
# Propagate invoke_separator from the integration class when the
|
|
# registrar_config dict doesn't already declare it explicitly.
|
|
# SkillsIntegration subclasses (claude, codex, …) set
|
|
# invoke_separator="-" as a class attribute but omit it from
|
|
# registrar_config, so without this they would fall back to "."
|
|
# when register_commands() resolves __SPECKIT_COMMAND_*__ tokens.
|
|
if "invoke_separator" not in config:
|
|
config["invoke_separator"] = integration.invoke_separator
|
|
configs[key] = config
|
|
return configs
|
|
|
|
|
|
class CommandRegistrar:
|
|
"""Handles registration of commands with AI agents.
|
|
|
|
Supports writing command files in Markdown or TOML format to the
|
|
appropriate agent directory, with correct argument placeholders
|
|
and companion files (e.g. Copilot .prompt.md).
|
|
"""
|
|
|
|
# Derived from INTEGRATION_REGISTRY — single source of truth.
|
|
# Populated lazily via _ensure_configs() on first use.
|
|
AGENT_CONFIGS: dict[str, dict[str, Any]] = {}
|
|
_configs_loaded: bool = False
|
|
|
|
def __init__(self) -> None:
|
|
self._ensure_configs()
|
|
|
|
def __init_subclass__(cls, **kwargs: Any) -> None:
|
|
super().__init_subclass__(**kwargs)
|
|
cls._ensure_configs()
|
|
|
|
@classmethod
|
|
def _ensure_configs(cls) -> None:
|
|
if not cls._configs_loaded:
|
|
try:
|
|
cls.AGENT_CONFIGS = _build_agent_configs()
|
|
cls._configs_loaded = True
|
|
except ImportError:
|
|
pass # Circular import during module init; retry on next access
|
|
|
|
@staticmethod
|
|
def _hyphenate_frontmatter_refs(val: Any) -> Any:
|
|
"""Recursively find any dotted references starting with speckit. and hyphenate them."""
|
|
if isinstance(val, dict):
|
|
return {
|
|
k: CommandRegistrar._hyphenate_frontmatter_refs(v)
|
|
for k, v in val.items()
|
|
}
|
|
elif isinstance(val, list):
|
|
return [CommandRegistrar._hyphenate_frontmatter_refs(x) for x in val]
|
|
elif isinstance(val, str):
|
|
return re.sub(
|
|
r"\bspeckit\.[A-Za-z0-9-_]+(?:\.[A-Za-z0-9-_]+)*\b",
|
|
lambda m: m.group(0).replace(".", "-"),
|
|
val,
|
|
)
|
|
return val
|
|
|
|
@staticmethod
|
|
def _hyphenate_body_refs(body: str) -> str:
|
|
"""Hyphenate dotted speckit references in command body text."""
|
|
return re.sub(
|
|
r"\bspeckit\.[A-Za-z0-9-_]+(?:\.[A-Za-z0-9-_]+)*\b",
|
|
lambda m: m.group(0).replace(".", "-"),
|
|
body,
|
|
)
|
|
|
|
@staticmethod
|
|
def parse_frontmatter(content: str) -> tuple[dict, str]:
|
|
"""Parse YAML frontmatter from Markdown content.
|
|
|
|
Args:
|
|
content: Markdown content with YAML frontmatter
|
|
|
|
Returns:
|
|
Tuple of (frontmatter_dict, body_content)
|
|
"""
|
|
if not content.startswith("---"):
|
|
return {}, content
|
|
|
|
# Find second ---
|
|
end_marker = content.find("---", 3)
|
|
if end_marker == -1:
|
|
return {}, content
|
|
|
|
frontmatter_str = content[3:end_marker].strip()
|
|
body = content[end_marker + 3 :].strip()
|
|
|
|
try:
|
|
frontmatter = yaml.safe_load(frontmatter_str) or {}
|
|
except yaml.YAMLError:
|
|
frontmatter = {}
|
|
|
|
if not isinstance(frontmatter, dict):
|
|
frontmatter = {}
|
|
|
|
return frontmatter, body
|
|
|
|
@staticmethod
|
|
def render_frontmatter(fm: dict) -> str:
|
|
"""Render frontmatter dictionary as YAML.
|
|
|
|
Args:
|
|
fm: Frontmatter dictionary
|
|
|
|
Returns:
|
|
YAML-formatted frontmatter with delimiters
|
|
"""
|
|
if not fm:
|
|
return ""
|
|
|
|
yaml_str = yaml.dump(
|
|
fm, default_flow_style=False, sort_keys=False, allow_unicode=True
|
|
)
|
|
return f"---\n{yaml_str}---\n"
|
|
|
|
def _adjust_script_paths(self, frontmatter: dict) -> dict:
|
|
"""Normalize script paths in frontmatter to generated project locations.
|
|
|
|
Rewrites known repo-relative and top-level script paths under the
|
|
``scripts`` key (for example ``../../scripts/``,
|
|
``../../templates/``, ``../../memory/``, ``scripts/``, ``templates/``, and
|
|
``memory/``) to the ``.specify/...`` paths used in generated projects.
|
|
|
|
Args:
|
|
frontmatter: Frontmatter dictionary
|
|
|
|
Returns:
|
|
Modified frontmatter with normalized project paths
|
|
"""
|
|
frontmatter = deepcopy(frontmatter)
|
|
|
|
scripts = frontmatter.get("scripts")
|
|
if isinstance(scripts, dict):
|
|
for key, script_path in scripts.items():
|
|
if isinstance(script_path, str):
|
|
scripts[key] = self.rewrite_project_relative_paths(script_path)
|
|
return frontmatter
|
|
|
|
@staticmethod
|
|
def rewrite_project_relative_paths(text: str) -> str:
|
|
"""Rewrite repo-relative paths to their generated project locations."""
|
|
if not isinstance(text, str) or not text:
|
|
return text
|
|
|
|
for old, new in (
|
|
("../../memory/", ".specify/memory/"),
|
|
("../../scripts/", ".specify/scripts/"),
|
|
("../../templates/", ".specify/templates/"),
|
|
):
|
|
text = text.replace(old, new)
|
|
|
|
# Only rewrite top-level style references so extension-local paths like
|
|
# ".specify/extensions/<ext>/scripts/..." remain intact.
|
|
text = re.sub(r'(^|[\s`"\'(])(?:\.?/)?memory/', r"\1.specify/memory/", text)
|
|
text = re.sub(r'(^|[\s`"\'(])(?:\.?/)?scripts/', r"\1.specify/scripts/", text)
|
|
text = re.sub(
|
|
r'(^|[\s`"\'(])(?:\.?/)?templates/', r"\1.specify/templates/", text
|
|
)
|
|
|
|
return text.replace(".specify/.specify/", ".specify/").replace(
|
|
".specify.specify/", ".specify/"
|
|
)
|
|
|
|
def render_markdown_command(
|
|
self, frontmatter: dict, body: str, source_id: str, context_note: str = None
|
|
) -> str:
|
|
"""Render command in Markdown format.
|
|
|
|
Args:
|
|
frontmatter: Command frontmatter
|
|
body: Command body content
|
|
source_id: Source identifier (extension or preset ID)
|
|
context_note: Custom context comment (default: <!-- Source: {source_id} -->)
|
|
|
|
Returns:
|
|
Formatted Markdown command file content
|
|
"""
|
|
if context_note is None:
|
|
context_note = f"\n<!-- Source: {source_id} -->\n"
|
|
return self.render_frontmatter(frontmatter) + "\n" + context_note + body
|
|
|
|
def render_toml_command(self, frontmatter: dict, body: str, source_id: str) -> str:
|
|
"""Render command in TOML format.
|
|
|
|
Args:
|
|
frontmatter: Command frontmatter
|
|
body: Command body content
|
|
source_id: Source identifier (extension or preset ID)
|
|
|
|
Returns:
|
|
Formatted TOML command file content
|
|
"""
|
|
toml_lines = []
|
|
|
|
if "description" in frontmatter:
|
|
toml_lines.append(
|
|
f"description = {self._render_basic_toml_string(frontmatter['description'])}"
|
|
)
|
|
toml_lines.append("")
|
|
|
|
toml_lines.append(f"# Source: {source_id}")
|
|
toml_lines.append("")
|
|
|
|
# Keep TOML output valid even when body contains triple-quote delimiters.
|
|
# Prefer multiline forms, then fall back to escaped basic string.
|
|
if '"""' not in body:
|
|
toml_lines.append('prompt = """')
|
|
toml_lines.append(body)
|
|
toml_lines.append('"""')
|
|
elif "'''" not in body:
|
|
toml_lines.append("prompt = '''")
|
|
toml_lines.append(body)
|
|
toml_lines.append("'''")
|
|
else:
|
|
toml_lines.append(f"prompt = {self._render_basic_toml_string(body)}")
|
|
|
|
return "\n".join(toml_lines)
|
|
|
|
@staticmethod
|
|
def _render_basic_toml_string(value: str) -> str:
|
|
"""Render *value* as a TOML basic string literal."""
|
|
escaped = (
|
|
value.replace("\\", "\\\\")
|
|
.replace('"', '\\"')
|
|
.replace("\n", "\\n")
|
|
.replace("\r", "\\r")
|
|
.replace("\t", "\\t")
|
|
)
|
|
return f'"{escaped}"'
|
|
|
|
def render_yaml_command(
|
|
self,
|
|
frontmatter: dict,
|
|
body: str,
|
|
source_id: str,
|
|
cmd_name: str = "",
|
|
) -> str:
|
|
"""Render command in YAML recipe format for Goose.
|
|
|
|
Args:
|
|
frontmatter: Command frontmatter
|
|
body: Command body content
|
|
source_id: Source identifier (extension or preset ID)
|
|
cmd_name: Command name used as title fallback
|
|
|
|
Returns:
|
|
Formatted YAML recipe file content
|
|
"""
|
|
from specify_cli.integrations.base import YamlIntegration
|
|
|
|
title = frontmatter.get("title", "") or frontmatter.get("name", "")
|
|
if not isinstance(title, str):
|
|
title = str(title) if title is not None else ""
|
|
if not title and cmd_name:
|
|
title = YamlIntegration._human_title(cmd_name)
|
|
if not title and source_id:
|
|
title = YamlIntegration._human_title(Path(str(source_id)).stem)
|
|
if not title:
|
|
title = "Command"
|
|
|
|
description = frontmatter.get("description", "")
|
|
if not isinstance(description, str):
|
|
description = str(description) if description is not None else ""
|
|
return YamlIntegration._render_yaml(title, description, body, source_id)
|
|
|
|
def render_skill_command(
|
|
self,
|
|
agent_name: str,
|
|
skill_name: str,
|
|
frontmatter: dict,
|
|
body: str,
|
|
source_id: str,
|
|
source_file: str,
|
|
project_root: Path,
|
|
) -> str:
|
|
"""Render a command override as a SKILL.md file.
|
|
|
|
SKILL-target agents should receive the same skills-oriented
|
|
frontmatter shape used elsewhere in the project instead of the
|
|
original command frontmatter.
|
|
|
|
Technical debt note:
|
|
Spec-kit currently has multiple SKILL.md generators (template packaging,
|
|
init-time conversion, and extension/preset overrides). Keep the skill
|
|
frontmatter keys aligned (name/description/compatibility/metadata, with
|
|
metadata.author and metadata.source subkeys) to avoid drift across agents.
|
|
"""
|
|
if not isinstance(frontmatter, dict):
|
|
frontmatter = {}
|
|
|
|
agent_config = self.AGENT_CONFIGS.get(agent_name, {})
|
|
if agent_config.get("extension") == "/SKILL.md":
|
|
body = self.resolve_skill_placeholders(
|
|
agent_name, frontmatter, body, project_root
|
|
)
|
|
|
|
description = frontmatter.get(
|
|
"description", f"Spec-kit workflow command: {skill_name}"
|
|
)
|
|
skill_frontmatter = self.build_skill_frontmatter(
|
|
agent_name,
|
|
skill_name,
|
|
description,
|
|
f"{source_id}:{source_file}",
|
|
)
|
|
return self.render_frontmatter(skill_frontmatter) + "\n" + body
|
|
|
|
@staticmethod
|
|
def build_skill_frontmatter(
|
|
agent_name: str,
|
|
skill_name: str,
|
|
description: str,
|
|
source: str,
|
|
) -> dict:
|
|
"""Build consistent SKILL.md frontmatter across all skill generators."""
|
|
skill_frontmatter = {
|
|
"name": skill_name,
|
|
"description": description,
|
|
"compatibility": "Requires spec-kit project structure with .specify/ directory",
|
|
"metadata": {
|
|
"author": "github-spec-kit",
|
|
"source": source,
|
|
},
|
|
}
|
|
return skill_frontmatter
|
|
|
|
@staticmethod
|
|
def resolve_skill_placeholders(
|
|
agent_name: str, frontmatter: dict, body: str, project_root: Path
|
|
) -> str:
|
|
"""Resolve script placeholders for skills-backed agents."""
|
|
if not isinstance(frontmatter, dict):
|
|
frontmatter = {}
|
|
|
|
scripts = frontmatter.get("scripts", {}) or {}
|
|
if not isinstance(scripts, dict):
|
|
scripts = {}
|
|
|
|
init_opts = load_init_options(project_root)
|
|
if not isinstance(init_opts, dict):
|
|
init_opts = {}
|
|
|
|
script_variant = init_opts.get("script")
|
|
if script_variant not in {"sh", "ps"}:
|
|
fallback_order = []
|
|
default_variant = (
|
|
"ps" if platform.system().lower().startswith("win") else "sh"
|
|
)
|
|
secondary_variant = "sh" if default_variant == "ps" else "ps"
|
|
|
|
if default_variant in scripts:
|
|
fallback_order.append(default_variant)
|
|
if secondary_variant in scripts:
|
|
fallback_order.append(secondary_variant)
|
|
|
|
for key in scripts:
|
|
if key not in fallback_order:
|
|
fallback_order.append(key)
|
|
|
|
script_variant = fallback_order[0] if fallback_order else None
|
|
|
|
script_command = scripts.get(script_variant) if script_variant else None
|
|
if script_command:
|
|
script_command = script_command.replace("{ARGS}", "$ARGUMENTS")
|
|
body = body.replace("{SCRIPT}", script_command)
|
|
|
|
body = body.replace("{ARGS}", "$ARGUMENTS").replace("__AGENT__", agent_name)
|
|
|
|
# Resolve __CONTEXT_FILE__ from the agent-context extension config.
|
|
# Fall back to init-options.json for projects that haven't migrated.
|
|
# Local import: _load_agent_context_config lives in __init__.py which
|
|
# imports agents.py, so a top-level import would be circular.
|
|
from . import _load_agent_context_config
|
|
ac_cfg = _load_agent_context_config(project_root)
|
|
context_file = ac_cfg.get("context_file") or ""
|
|
if not context_file:
|
|
context_file = init_opts.get("context_file") or ""
|
|
body = body.replace("__CONTEXT_FILE__", context_file)
|
|
|
|
return CommandRegistrar.rewrite_project_relative_paths(body)
|
|
|
|
def _convert_argument_placeholder(
|
|
self, content: str, from_placeholder: str, to_placeholder: str
|
|
) -> str:
|
|
"""Convert argument placeholder format.
|
|
|
|
Args:
|
|
content: Command content
|
|
from_placeholder: Source placeholder (e.g., "$ARGUMENTS")
|
|
to_placeholder: Target placeholder (e.g., "{{args}}")
|
|
|
|
Returns:
|
|
Content with converted placeholders
|
|
"""
|
|
return content.replace(from_placeholder, to_placeholder)
|
|
|
|
@staticmethod
|
|
def _compute_output_name(
|
|
agent_name: str, cmd_name: str, agent_config: Dict[str, Any]
|
|
) -> str:
|
|
"""Compute the on-disk command or skill name for an agent."""
|
|
if agent_config["extension"] != "/SKILL.md":
|
|
format_name = agent_config.get("format_name")
|
|
if format_name:
|
|
return format_name(cmd_name)
|
|
return cmd_name
|
|
|
|
short_name = cmd_name
|
|
if short_name.startswith("speckit."):
|
|
short_name = short_name[len("speckit.") :]
|
|
short_name = short_name.replace(".", "-")
|
|
|
|
return f"speckit-{short_name}"
|
|
|
|
@staticmethod
|
|
def _ensure_inside(candidate: Path, base: Path) -> None:
|
|
"""Validate that a write target stays within the expected base directory.
|
|
|
|
Uses lexical normalization so traversal via ``..`` or absolute paths is
|
|
rejected while intentionally symlinked sub-directories remain
|
|
supported.
|
|
|
|
Args:
|
|
candidate: Path that will be written.
|
|
base: Directory the write must remain within.
|
|
|
|
Raises:
|
|
ValueError: If the normalized candidate path escapes ``base``.
|
|
"""
|
|
normalized = Path(os.path.normpath(candidate))
|
|
base_normalized = Path(os.path.normpath(base))
|
|
if not normalized.is_relative_to(base_normalized):
|
|
raise ValueError(f"Output path {candidate!r} escapes directory {base!r}")
|
|
|
|
@staticmethod
|
|
def _is_safe_command_name(name: str) -> bool:
|
|
"""Reject names that could escape the commands directory via path traversal."""
|
|
if os.path.sep in name or "/" in name or "\\" in name:
|
|
return False
|
|
return os.path.normpath(name) == name
|
|
|
|
@staticmethod
|
|
def _same_lexical_path(left: Path, right: Path) -> bool:
|
|
"""Compare paths after lexical normalization without resolving symlinks."""
|
|
return os.path.normcase(os.path.normpath(os.fspath(left))) == os.path.normcase(
|
|
os.path.normpath(os.fspath(right))
|
|
)
|
|
|
|
@staticmethod
|
|
def _active_skills_agent(project_root: Path) -> Optional[str]:
|
|
"""Return the initialized skills-backed agent, if skills mode is active."""
|
|
opts = load_init_options(project_root)
|
|
if not isinstance(opts, dict):
|
|
return None
|
|
|
|
agent = opts.get("ai")
|
|
if not isinstance(agent, str) or not agent:
|
|
return None
|
|
# Kimi is a native skills integration; when ai_skills is not boolean
|
|
# True, Kimi still uses its existing SKILL.md layout.
|
|
if not is_ai_skills_enabled(opts) and agent != "kimi":
|
|
return None
|
|
return agent
|
|
|
|
def register_commands(
|
|
self,
|
|
agent_name: str,
|
|
commands: List[Dict[str, Any]],
|
|
source_id: str,
|
|
source_dir: Path,
|
|
project_root: Path,
|
|
context_note: str = None,
|
|
_resolved_dir: Path = None,
|
|
link_outputs: bool = False,
|
|
) -> List[str]:
|
|
"""Register commands for a specific agent.
|
|
|
|
Args:
|
|
agent_name: Agent name (claude, gemini, copilot, etc.)
|
|
commands: List of command info dicts with 'name', 'file', and optional 'aliases'
|
|
source_id: Identifier of the source (extension or preset ID)
|
|
source_dir: Directory containing command source files
|
|
project_root: Path to project root
|
|
context_note: Custom context comment for markdown output
|
|
_resolved_dir: Pre-resolved command directory (internal use
|
|
only — avoids a second ``_resolve_agent_dir`` call and
|
|
duplicate deprecation warnings when invoked from
|
|
``register_commands_for_all_agents``).
|
|
link_outputs: If True, write rendered output to a source-local
|
|
dev cache and symlink the agent command file to it. Falls back
|
|
to a normal file write when symlinks are unavailable.
|
|
|
|
Returns:
|
|
List of registered command names
|
|
|
|
Raises:
|
|
ValueError: If agent is not supported
|
|
"""
|
|
self._ensure_configs()
|
|
if agent_name not in self.AGENT_CONFIGS:
|
|
raise ValueError(f"Unsupported agent: {agent_name}")
|
|
|
|
agent_config = self.AGENT_CONFIGS[agent_name]
|
|
commands_dir = _resolved_dir or self._resolve_agent_dir(
|
|
agent_name, agent_config, project_root,
|
|
)
|
|
commands_dir.mkdir(parents=True, exist_ok=True)
|
|
|
|
registered = []
|
|
is_cline_ext = agent_name == "cline" and source_id != "core"
|
|
|
|
for cmd_info in commands:
|
|
cmd_name = cmd_info["name"]
|
|
aliases = cmd_info.get("aliases", [])
|
|
cmd_file = cmd_info["file"]
|
|
|
|
source_file = source_dir / cmd_file
|
|
if not source_file.exists():
|
|
continue
|
|
|
|
content = source_file.read_text(encoding="utf-8")
|
|
frontmatter, body = self.parse_frontmatter(content)
|
|
|
|
if frontmatter.get("strategy") == "wrap":
|
|
from .presets import _substitute_core_template
|
|
|
|
body, core_frontmatter = _substitute_core_template(
|
|
body, cmd_name, project_root, self
|
|
)
|
|
frontmatter = dict(frontmatter)
|
|
for key in ("scripts", "agent_scripts"):
|
|
if key not in frontmatter and key in core_frontmatter:
|
|
frontmatter[key] = core_frontmatter[key]
|
|
frontmatter.pop("strategy", None)
|
|
|
|
frontmatter = self._adjust_script_paths(frontmatter)
|
|
|
|
for key in agent_config.get("strip_frontmatter_keys", []):
|
|
frontmatter.pop(key, None)
|
|
|
|
if agent_config.get("inject_name") and not frontmatter.get("name"):
|
|
# Use custom name formatter if provided (e.g., Forge's hyphenated format)
|
|
format_name = agent_config.get("format_name")
|
|
frontmatter["name"] = format_name(cmd_name) if format_name else cmd_name
|
|
|
|
if is_cline_ext:
|
|
frontmatter = self._hyphenate_frontmatter_refs(frontmatter)
|
|
body = self._hyphenate_body_refs(body)
|
|
|
|
body = self._convert_argument_placeholder(
|
|
body, "$ARGUMENTS", agent_config["args"]
|
|
)
|
|
|
|
# Resolve __SPECKIT_COMMAND_*__ tokens using the agent's invoke separator.
|
|
# The separator is sourced from agent_config (populated by _build_agent_configs,
|
|
# which propagates each integration's invoke_separator class attribute).
|
|
# Deferred import of IntegrationBase avoids a circular import at module load
|
|
# (base.py itself imports CommandRegistrar lazily).
|
|
from specify_cli.integrations.base import IntegrationBase # noqa: PLC0415
|
|
|
|
_sep = agent_config.get("invoke_separator", ".")
|
|
body = IntegrationBase.resolve_command_refs(body, _sep)
|
|
|
|
output_name = self._compute_output_name(agent_name, cmd_name, agent_config)
|
|
|
|
if agent_config["extension"] == "/SKILL.md":
|
|
output = self.render_skill_command(
|
|
agent_name,
|
|
output_name,
|
|
frontmatter,
|
|
body,
|
|
source_id,
|
|
cmd_file,
|
|
project_root,
|
|
)
|
|
elif agent_config["format"] == "markdown":
|
|
body = self.resolve_skill_placeholders(
|
|
agent_name, frontmatter, body, project_root
|
|
)
|
|
body = self._convert_argument_placeholder(
|
|
body, "$ARGUMENTS", agent_config["args"]
|
|
)
|
|
output = self.render_markdown_command(
|
|
frontmatter, body, source_id, context_note
|
|
)
|
|
elif agent_config["format"] == "toml":
|
|
body = self.resolve_skill_placeholders(
|
|
agent_name, frontmatter, body, project_root
|
|
)
|
|
body = self._convert_argument_placeholder(
|
|
body, "$ARGUMENTS", agent_config["args"]
|
|
)
|
|
output = self.render_toml_command(frontmatter, body, source_id)
|
|
elif agent_config["format"] == "yaml":
|
|
output = self.render_yaml_command(
|
|
frontmatter, body, source_id, cmd_name
|
|
)
|
|
else:
|
|
raise ValueError(f"Unsupported format: {agent_config['format']}")
|
|
|
|
dest_file = commands_dir / f"{output_name}{agent_config['extension']}"
|
|
self._ensure_inside(dest_file, commands_dir)
|
|
dest_file.parent.mkdir(parents=True, exist_ok=True)
|
|
self._write_registered_output(
|
|
dest_file,
|
|
output,
|
|
source_dir,
|
|
agent_name,
|
|
output_name,
|
|
agent_config["extension"],
|
|
link_outputs,
|
|
)
|
|
|
|
if agent_name == "copilot":
|
|
self.write_copilot_prompt(project_root, cmd_name)
|
|
|
|
registered.append(cmd_name)
|
|
|
|
for alias in aliases:
|
|
alias_output_name = self._compute_output_name(
|
|
agent_name, alias, agent_config
|
|
)
|
|
|
|
# For agents with inject_name, render with alias-specific frontmatter
|
|
if agent_config.get("inject_name"):
|
|
alias_frontmatter = deepcopy(frontmatter)
|
|
# Use custom name formatter if provided (e.g., Forge's hyphenated format)
|
|
format_name = agent_config.get("format_name")
|
|
alias_frontmatter["name"] = (
|
|
format_name(alias) if format_name else alias
|
|
)
|
|
|
|
if agent_config["extension"] == "/SKILL.md":
|
|
alias_output = self.render_skill_command(
|
|
agent_name,
|
|
alias_output_name,
|
|
alias_frontmatter,
|
|
body,
|
|
source_id,
|
|
cmd_file,
|
|
project_root,
|
|
)
|
|
elif agent_config["format"] == "markdown":
|
|
alias_output = self.render_markdown_command(
|
|
alias_frontmatter, body, source_id, context_note
|
|
)
|
|
elif agent_config["format"] == "toml":
|
|
alias_output = self.render_toml_command(
|
|
alias_frontmatter, body, source_id
|
|
)
|
|
elif agent_config["format"] == "yaml":
|
|
alias_output = self.render_yaml_command(
|
|
alias_frontmatter, body, source_id, alias
|
|
)
|
|
else:
|
|
raise ValueError(
|
|
f"Unsupported format: {agent_config['format']}"
|
|
)
|
|
else:
|
|
# For other agents, reuse the primary output
|
|
alias_output = output
|
|
if agent_config["extension"] == "/SKILL.md":
|
|
alias_output = self.render_skill_command(
|
|
agent_name,
|
|
alias_output_name,
|
|
frontmatter,
|
|
body,
|
|
source_id,
|
|
cmd_file,
|
|
project_root,
|
|
)
|
|
|
|
alias_file = (
|
|
commands_dir / f"{alias_output_name}{agent_config['extension']}"
|
|
)
|
|
self._ensure_inside(alias_file, commands_dir)
|
|
alias_file.parent.mkdir(parents=True, exist_ok=True)
|
|
self._write_registered_output(
|
|
alias_file,
|
|
alias_output,
|
|
source_dir,
|
|
agent_name,
|
|
alias_output_name,
|
|
agent_config["extension"],
|
|
link_outputs,
|
|
)
|
|
if agent_name == "copilot":
|
|
self.write_copilot_prompt(project_root, alias)
|
|
registered.append(alias)
|
|
|
|
return registered
|
|
|
|
@staticmethod
|
|
def _write_registered_output(
|
|
dest_file: Path,
|
|
content: str,
|
|
source_dir: Path,
|
|
agent_name: str,
|
|
output_name: str,
|
|
extension: str,
|
|
link_outputs: bool,
|
|
) -> None:
|
|
"""Write a rendered agent artifact, optionally as a dev-mode symlink."""
|
|
if not link_outputs:
|
|
dest_file.write_text(content, encoding="utf-8")
|
|
return
|
|
|
|
rel_output = Path(f"{output_name}{extension}")
|
|
cache_root = source_dir / ".specify-dev" / "agent-commands" / agent_name
|
|
cache_file = cache_root / rel_output
|
|
CommandRegistrar._ensure_inside(cache_file, cache_root)
|
|
|
|
try:
|
|
cache_file.parent.mkdir(parents=True, exist_ok=True)
|
|
cache_file.write_text(content, encoding="utf-8")
|
|
if dest_file.exists() or dest_file.is_symlink():
|
|
dest_file.unlink()
|
|
target = os.path.relpath(cache_file, dest_file.parent)
|
|
os.symlink(target, dest_file)
|
|
except (OSError, ValueError):
|
|
# Windows often requires Developer Mode or admin privileges for
|
|
# symlinks, and relpath can fail across drives. Keep dev installs
|
|
# functional by falling back to a copy.
|
|
if dest_file.is_symlink():
|
|
dest_file.unlink()
|
|
dest_file.write_text(content, encoding="utf-8")
|
|
|
|
@staticmethod
|
|
def write_copilot_prompt(project_root: Path, cmd_name: str) -> None:
|
|
"""Generate a companion .prompt.md file for a Copilot agent command.
|
|
|
|
Args:
|
|
project_root: Path to project root
|
|
cmd_name: Command name (e.g. 'speckit.my-ext.example')
|
|
"""
|
|
prompts_dir = project_root / ".github" / "prompts"
|
|
prompts_dir.mkdir(parents=True, exist_ok=True)
|
|
prompt_file = prompts_dir / f"{cmd_name}.prompt.md"
|
|
CommandRegistrar._ensure_inside(prompt_file, prompts_dir)
|
|
prompt_file.write_text(f"---\nagent: {cmd_name}\n---\n", encoding="utf-8")
|
|
|
|
@staticmethod
|
|
def _resolve_agent_dir(
|
|
agent_name: str,
|
|
agent_config: dict[str, Any],
|
|
project_root: Path,
|
|
) -> Path:
|
|
"""Return the agent command directory, falling back to legacy_dir.
|
|
|
|
Supports project-relative paths (e.g. ``.claude/skills/``),
|
|
home-relative paths (e.g. ``~/.hermes/skills``), and absolute
|
|
paths — the ``agent_config["dir"]`` value is resolved verbatim
|
|
when absolute or starting with ``~/``, or joined with
|
|
``project_root`` when relative.
|
|
|
|
When the canonical directory does not exist but a ``legacy_dir``
|
|
is configured and present on disk, returns the legacy path and
|
|
emits a deprecation warning advising the user to upgrade.
|
|
|
|
Integrations that do not declare ``legacy_dir`` get the canonical
|
|
path unconditionally — no fallback, no warning.
|
|
"""
|
|
dir_str = agent_config["dir"]
|
|
if dir_str.startswith("~"):
|
|
# Use Path.home() + remainder instead of expanduser() so tests
|
|
# that monkeypatch Path.home() can properly isolate the home dir.
|
|
# expanduser() uses OS env/user lookup and ignores monkeypatches.
|
|
agent_dir = Path.home() / dir_str[1:].lstrip("/")
|
|
else:
|
|
p = Path(dir_str)
|
|
agent_dir = p if p.is_absolute() else project_root / p
|
|
if not agent_dir.exists():
|
|
legacy = agent_config.get("legacy_dir")
|
|
if legacy:
|
|
legacy_dir = project_root / legacy
|
|
if legacy_dir.exists():
|
|
import warnings
|
|
|
|
warnings.warn(
|
|
f"Found legacy '{legacy}' directory for "
|
|
f"{agent_name}. Run 'specify integration "
|
|
f"upgrade {agent_name}' to migrate to "
|
|
f"'{agent_config['dir']}'.",
|
|
stacklevel=3,
|
|
)
|
|
return legacy_dir
|
|
return agent_dir
|
|
|
|
def register_commands_for_all_agents(
|
|
self,
|
|
commands: List[Dict[str, Any]],
|
|
source_id: str,
|
|
source_dir: Path,
|
|
project_root: Path,
|
|
context_note: str = None,
|
|
link_outputs: bool = False,
|
|
create_missing_active_skills_dir: bool = False,
|
|
) -> Dict[str, List[str]]:
|
|
"""Register commands for all detected agents in the project.
|
|
|
|
Args:
|
|
commands: List of command info dicts
|
|
source_id: Identifier of the source (extension or preset ID)
|
|
source_dir: Directory containing command source files
|
|
project_root: Path to project root
|
|
context_note: Custom context comment for markdown output
|
|
link_outputs: If True, create dev-mode symlinks for rendered
|
|
command files when supported by the OS.
|
|
create_missing_active_skills_dir: If True, attempt missing-dir
|
|
recovery only for the active initialized skills-backed agent.
|
|
Recovery requires active skills mode (or Kimi's existing native
|
|
skills directory) and is skipped when safe resolution or
|
|
creation fails.
|
|
|
|
Returns:
|
|
Dictionary mapping agent names to list of registered commands
|
|
"""
|
|
results = {}
|
|
|
|
self._ensure_configs()
|
|
active_skills_agent = (
|
|
self._active_skills_agent(project_root)
|
|
if create_missing_active_skills_dir else None
|
|
)
|
|
active_created_skills_dir: Optional[Path] = None
|
|
for agent_name, agent_config in self.AGENT_CONFIGS.items():
|
|
active_skills_output = (
|
|
agent_name == active_skills_agent
|
|
and agent_config.get("extension") == "/SKILL.md"
|
|
)
|
|
recovered_active_skills_dir: Optional[Path] = None
|
|
# Check detect_dir first (project-local marker) if configured,
|
|
# falling back to the resolved dir for output. This prevents
|
|
# global dirs (e.g. ~/.hermes/skills) from causing false
|
|
# detection in every project.
|
|
detect_dir_str = agent_config.get("detect_dir")
|
|
if detect_dir_str:
|
|
detect_path = project_root / detect_dir_str
|
|
if not detect_path.is_dir():
|
|
if not active_skills_output:
|
|
continue
|
|
try:
|
|
from . import resolve_active_skills_dir
|
|
|
|
recovered_active_skills_dir = (
|
|
resolve_active_skills_dir(project_root)
|
|
)
|
|
except (ValueError, OSError):
|
|
continue
|
|
if recovered_active_skills_dir is None or not detect_path.is_dir():
|
|
continue
|
|
active_created_skills_dir = recovered_active_skills_dir
|
|
agent_dir = self._resolve_agent_dir(
|
|
agent_name, agent_config, project_root,
|
|
)
|
|
|
|
agent_dir_existed = agent_dir.is_dir()
|
|
register_missing_active_skills_agent = (
|
|
not agent_dir_existed
|
|
and active_skills_output
|
|
)
|
|
if register_missing_active_skills_agent:
|
|
if recovered_active_skills_dir is None:
|
|
try:
|
|
from . import resolve_active_skills_dir
|
|
|
|
recovered_active_skills_dir = (
|
|
resolve_active_skills_dir(project_root)
|
|
)
|
|
except (ValueError, OSError):
|
|
continue
|
|
if recovered_active_skills_dir is None:
|
|
continue
|
|
active_created_skills_dir = recovered_active_skills_dir
|
|
# Shared skill dirs such as .agents/skills should not make
|
|
# later integrations look detected when the active agent just
|
|
# recreated the directory during this registration pass.
|
|
created_by_active_agent = (
|
|
active_created_skills_dir is not None
|
|
and self._same_lexical_path(agent_dir, active_created_skills_dir)
|
|
and agent_name != active_skills_agent
|
|
)
|
|
should_register = (
|
|
agent_dir_existed and not created_by_active_agent
|
|
) or register_missing_active_skills_agent
|
|
|
|
if should_register:
|
|
try:
|
|
registered = self.register_commands(
|
|
agent_name,
|
|
commands,
|
|
source_id,
|
|
source_dir,
|
|
project_root,
|
|
context_note=context_note,
|
|
_resolved_dir=agent_dir,
|
|
link_outputs=link_outputs,
|
|
)
|
|
if registered:
|
|
results[agent_name] = registered
|
|
if register_missing_active_skills_agent:
|
|
active_created_skills_dir = (
|
|
recovered_active_skills_dir or agent_dir
|
|
)
|
|
except ValueError:
|
|
continue
|
|
except OSError:
|
|
if register_missing_active_skills_agent:
|
|
continue
|
|
raise
|
|
|
|
return results
|
|
|
|
def register_commands_for_non_skill_agents(
|
|
self,
|
|
commands: List[Dict[str, Any]],
|
|
source_id: str,
|
|
source_dir: Path,
|
|
project_root: Path,
|
|
context_note: Optional[str] = None,
|
|
link_outputs: bool = False,
|
|
) -> Dict[str, List[str]]:
|
|
"""Register commands for all non-skill agents in the project.
|
|
|
|
Like register_commands_for_all_agents but skips skill-based agents
|
|
(those with extension '/SKILL.md'). Used by reconciliation to avoid
|
|
overwriting properly formatted SKILL.md files.
|
|
|
|
Args:
|
|
commands: List of command info dicts
|
|
source_id: Identifier of the source
|
|
source_dir: Directory containing command source files
|
|
project_root: Path to project root
|
|
context_note: Custom context comment for markdown output
|
|
link_outputs: If True, create dev-mode symlinks for rendered
|
|
command files when supported by the OS.
|
|
|
|
Returns:
|
|
Dictionary mapping agent names to list of registered commands
|
|
"""
|
|
results = {}
|
|
self._ensure_configs()
|
|
for agent_name, agent_config in self.AGENT_CONFIGS.items():
|
|
if agent_config.get("extension") == "/SKILL.md":
|
|
continue
|
|
detect_dir_str = agent_config.get("detect_dir")
|
|
if detect_dir_str:
|
|
detect_path = project_root / detect_dir_str
|
|
if not detect_path.is_dir():
|
|
continue
|
|
agent_dir = self._resolve_agent_dir(
|
|
agent_name, agent_config, project_root,
|
|
)
|
|
if agent_dir.is_dir():
|
|
try:
|
|
registered = self.register_commands(
|
|
agent_name,
|
|
commands,
|
|
source_id,
|
|
source_dir,
|
|
project_root,
|
|
context_note=context_note,
|
|
_resolved_dir=agent_dir,
|
|
link_outputs=link_outputs,
|
|
)
|
|
if registered:
|
|
results[agent_name] = registered
|
|
except ValueError:
|
|
continue
|
|
return results
|
|
|
|
def unregister_commands(
|
|
self, registered_commands: Dict[str, List[str]], project_root: Path
|
|
) -> None:
|
|
"""Remove previously registered command files from agent directories.
|
|
|
|
When a ``legacy_dir`` is configured, files are removed from
|
|
*both* the canonical and the legacy directory so that orphaned
|
|
commands left behind after an ``integration upgrade`` are
|
|
cleaned up as well.
|
|
|
|
Args:
|
|
registered_commands: Dict mapping agent names to command name lists
|
|
project_root: Path to project root
|
|
"""
|
|
self._ensure_configs()
|
|
for agent_name, cmd_names in registered_commands.items():
|
|
if agent_name not in self.AGENT_CONFIGS:
|
|
continue
|
|
|
|
agent_config = self.AGENT_CONFIGS[agent_name]
|
|
commands_dir = self._resolve_agent_dir(
|
|
agent_name, agent_config, project_root,
|
|
)
|
|
|
|
# Collect all directories to clean: canonical (or resolved
|
|
# legacy) plus the legacy dir if it exists separately.
|
|
dirs_to_clean = [commands_dir]
|
|
legacy = agent_config.get("legacy_dir")
|
|
if legacy:
|
|
legacy_dir = project_root / legacy
|
|
if legacy_dir.exists() and legacy_dir != commands_dir:
|
|
dirs_to_clean.append(legacy_dir)
|
|
|
|
for cmd_name in cmd_names:
|
|
output_name = self._compute_output_name(
|
|
agent_name, cmd_name, agent_config
|
|
)
|
|
|
|
names_to_clean = [output_name]
|
|
if output_name != cmd_name and self._is_safe_command_name(cmd_name):
|
|
names_to_clean.append(cmd_name)
|
|
|
|
for target_dir in dirs_to_clean:
|
|
for name in names_to_clean:
|
|
cmd_file = (
|
|
target_dir / f"{name}{agent_config['extension']}"
|
|
)
|
|
try:
|
|
self._ensure_inside(cmd_file, target_dir)
|
|
except ValueError:
|
|
continue
|
|
if cmd_file.exists() or cmd_file.is_symlink():
|
|
cmd_file.unlink()
|
|
# For SKILL.md agents each command lives in its own
|
|
# subdirectory (e.g. .agents/skills/speckit-ext-cmd/
|
|
# SKILL.md). Remove the parent dir when it becomes
|
|
# empty to avoid orphaned directories.
|
|
parent = cmd_file.parent
|
|
if parent != target_dir and parent.exists():
|
|
try:
|
|
parent.rmdir()
|
|
except OSError:
|
|
pass
|
|
|
|
if agent_name == "copilot":
|
|
prompt_file = (
|
|
project_root / ".github" / "prompts" / f"{cmd_name}.prompt.md"
|
|
)
|
|
if prompt_file.exists():
|
|
prompt_file.unlink()
|
|
|
|
|
|
# Populate AGENT_CONFIGS after class definition.
|
|
# Catches ImportError from circular imports during module loading;
|
|
# _configs_loaded stays False so the next explicit access retries.
|
|
try:
|
|
CommandRegistrar._ensure_configs()
|
|
except ImportError:
|
|
pass
|