models

src/personal_mcp/registry/models/prompt.py

@@ -1,18 +1,23 @@
 import re
 from collections.abc import Mapping
 from pathlib import PurePosixPath
+from typing import TYPE_CHECKING
 from typing import Any
 from typing import Literal
 
 import yaml
 from pydantic import Field
 from pydantic import field_validator
+from pydantic import model_validator
 
 from .common import SEMVER_RE
 from .common import SKILL_ID_RE
 from .common import StrictFrozenModel
 from .common import frozen_mapping
 
+if TYPE_CHECKING:
+    from personal_mcp.registry.ingest.prompt import PromptFilesBundle
+
 type PromptArgumentType = Literal[
     "string",
     "number",
@@ -101,6 +106,18 @@ class PromptFrontmatter(StrictFrozenModel):
             raise ValueError("name must not contain reserved words anthropic or claude")
         return value
 
+    @classmethod
+    def from_raw_yaml(cls, raw: str | None) -> "PromptFrontmatter":
+        if raw is None:
+            raise ValueError("missing YAML frontmatter")
+        try:
+            data = yaml.safe_load(raw)
+        except yaml.YAMLError as e:
+            raise ValueError(f"invalid YAML in frontmatter: {e}") from e
+        if not isinstance(data, dict):
+            raise TypeError("frontmatter must parse to an object")
+        return cls.model_validate(data)
+
 
 class StoredPrompt(StrictFrozenModel):
     """Normalized prompt document content with path and frontmatter for storage in the registry."""
@@ -108,15 +125,43 @@ class StoredPrompt(StrictFrozenModel):
     prompt_id: str
     relpath: PurePosixPath
     content: str
-    frontmatter: PromptFrontmatter | None = None
+    frontmatter: PromptFrontmatter
 
     @field_validator("frontmatter", mode="before")
     @classmethod
-    def parse_frontmatter_yaml(cls, value: str | None):
-        if value is None:
-            return None
-        try:
-            data = yaml.safe_load(value)
-        except yaml.YAMLError as e:
-            raise ValueError(f"invalid YAML in frontmatter: {e}") from e
-        return PromptFrontmatter.model_validate(data)
+    def parse_frontmatter_yaml(cls, value: PromptFrontmatter | str | None) -> PromptFrontmatter:
+        if isinstance(value, PromptFrontmatter):
+            return value
+        return PromptFrontmatter.from_raw_yaml(value)
+
+    @model_validator(mode="after")
+    def validate_contract(self) -> "StoredPrompt":
+        parts = self.relpath.parts
+        if len(parts) < 3 or parts[0] != "prompts":
+            raise ValueError("prompt relpath must be under prompts/<slug>/")
+
+        prompt_dir_name = parts[1]
+        if self.frontmatter.name != prompt_dir_name:
+            raise ValueError("frontmatter name must exactly match prompt directory name")
+        if self.frontmatter.x_personal_mcp.id != self.frontmatter.name:
+            raise ValueError("x-personal-mcp.id must exactly match name")
+
+        expected_capability = f"resource://prompts/{self.frontmatter.name}/document"
+        if expected_capability not in self.frontmatter.x_personal_mcp.capabilities:
+            raise ValueError(f"capabilities must include {expected_capability}")
+
+        if self.prompt_id != self.frontmatter.x_personal_mcp.id:
+            raise ValueError("prompt_id must exactly match x-personal-mcp.id")
+        return self
+
+    @classmethod
+    def from_bundle(cls, bundle: "PromptFilesBundle") -> "StoredPrompt":
+        frontmatter = PromptFrontmatter.from_raw_yaml(bundle.prompt.frontmatter)
+        return cls.model_validate(
+            {
+                "prompt_id": frontmatter.x_personal_mcp.id,
+                "relpath": bundle.prompt.relpath,
+                "content": bundle.prompt.content,
+                "frontmatter": frontmatter,
+            }
+        )