原型阶段

2026-06-09 23:44:31 +08:00
parent ea1d0fb3bd
commit 01b3375395
22 changed files with 1077 additions and 27 deletions
@@ -29,6 +29,8 @@ func registerPMTools(srv *mcpsdk.Server, deps ServerDeps) {
 	registerCloseRequirement(srv, deps)
 	registerReopenRequirement(srv, deps)
 	registerSetRequirementPhase(srv, deps)
+	registerAddRequirementPrototype(srv, deps)
+	registerListRequirementPrototypes(srv, deps)
 }

 // ===== list_projects =====
@@ -709,5 +711,98 @@ func registerSetRequirementPhase(srv *mcpsdk.Server, deps ServerDeps) {
 		})
 }

+// ===== add_requirement_prototype =====
+
+type addRequirementPrototypeIn struct {
+	ProjectSlug string `json:"project_slug" jsonschema:"required field"`
+	Number      int    `json:"number" jsonschema:"requirement number (>=1)"`
+	Content     string `json:"content" jsonschema:"prototype content (non-empty)"`
+	Note        string `json:"note,omitempty"`
+}
+type prototypeDetail struct {
+	Version   int    `json:"version"`
+	Content   string `json:"content"`
+	Note      string `json:"note,omitempty"`
+	CreatedAt string `json:"created_at"`
+}
+type prototypeCreateOut struct {
+	OK        bool            `json:"ok"`
+	Prototype prototypeDetail `json:"prototype"`
+}
+
+func registerAddRequirementPrototype(srv *mcpsdk.Server, deps ServerDeps) {
+	mcpsdk.AddTool(srv,
+		&mcpsdk.Tool{Name: "add_requirement_prototype", Description: "Add a new prototype version to a requirement."},
+		func(ctx context.Context, _ *mcpsdk.CallToolRequest, in addRequirementPrototypeIn) (*mcpsdk.CallToolResult, prototypeCreateOut, error) {
+			if err := CheckToolFromCtx(ctx, "add_requirement_prototype"); err != nil {
+				return nil, prototypeCreateOut{}, err
+			}
+			c, err := deps.Caller.Resolve(ctx)
+			if err != nil {
+				return nil, prototypeCreateOut{}, err
+			}
+			p, err := deps.Projects.Get(ctx, c, in.ProjectSlug)
+			if err != nil {
+				return nil, prototypeCreateOut{}, err
+			}
+			if err := CheckProjectFromCtx(ctx, p.ID); err != nil {
+				return nil, prototypeCreateOut{}, err
+			}
+			proto, err := deps.Prototypes.Create(ctx, c, in.ProjectSlug, in.Number, project.CreatePrototypeInput{
+				Content: in.Content, Note: in.Note,
+			})
+			if err != nil {
+				return nil, prototypeCreateOut{}, err
+			}
+			return nil, prototypeCreateOut{OK: true, Prototype: prototypeDetail{
+				Version: proto.Version, Content: proto.Content, Note: proto.Note,
+				CreatedAt: proto.CreatedAt.Format("2006-01-02T15:04:05Z07:00"),
+			}}, nil
+		})
+}
+
+// ===== list_requirement_prototypes =====
+
+type listRequirementPrototypesIn struct {
+	ProjectSlug string `json:"project_slug" jsonschema:"required field"`
+	Number      int    `json:"number" jsonschema:"requirement number (>=1)"`
+}
+type listRequirementPrototypesOut struct {
+	Prototypes []prototypeDetail `json:"prototypes"`
+}
+
+func registerListRequirementPrototypes(srv *mcpsdk.Server, deps ServerDeps) {
+	mcpsdk.AddTool(srv,
+		&mcpsdk.Tool{Name: "list_requirement_prototypes", Description: "List prototype versions of a requirement."},
+		func(ctx context.Context, _ *mcpsdk.CallToolRequest, in listRequirementPrototypesIn) (*mcpsdk.CallToolResult, listRequirementPrototypesOut, error) {
+			if err := CheckToolFromCtx(ctx, "list_requirement_prototypes"); err != nil {
+				return nil, listRequirementPrototypesOut{}, err
+			}
+			c, err := deps.Caller.Resolve(ctx)
+			if err != nil {
+				return nil, listRequirementPrototypesOut{}, err
+			}
+			p, err := deps.Projects.Get(ctx, c, in.ProjectSlug)
+			if err != nil {
+				return nil, listRequirementPrototypesOut{}, err
+			}
+			if err := CheckProjectFromCtx(ctx, p.ID); err != nil {
+				return nil, listRequirementPrototypesOut{}, err
+			}
+			protos, err := deps.Prototypes.List(ctx, c, in.ProjectSlug, in.Number)
+			if err != nil {
+				return nil, listRequirementPrototypesOut{}, err
+			}
+			out := listRequirementPrototypesOut{Prototypes: make([]prototypeDetail, 0, len(protos))}
+			for _, proto := range protos {
+				out.Prototypes = append(out.Prototypes, prototypeDetail{
+					Version: proto.Version, Content: proto.Content, Note: proto.Note,
+					CreatedAt: proto.CreatedAt.Format("2006-01-02T15:04:05Z07:00"),
+				})
+			}
+			return nil, out, nil
+		})
+}
+
 // 保证 fmt 引用（编译期检查）
 var _ = fmt.Sprintf