124 lines
3.3 KiB
Go
124 lines
3.3 KiB
Go
package tdd
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
iexec "github.com/mathiasbq/supervisor/internal/exec"
|
|
)
|
|
|
|
func (s *Skill) Handle(ctx context.Context, tool string, args json.RawMessage) (json.RawMessage, error) {
|
|
switch tool {
|
|
case "tdd_red":
|
|
return s.handleRed(ctx, args)
|
|
case "tdd_green":
|
|
return s.handleGreen(ctx, args)
|
|
case "tdd_refactor":
|
|
return s.handleRefactor(ctx, args)
|
|
default:
|
|
return nil, fmt.Errorf("unknown tool: %s", tool)
|
|
}
|
|
}
|
|
|
|
type redArgs struct {
|
|
ProjectRoot string `json:"project_root"`
|
|
Spec string `json:"spec"`
|
|
Model string `json:"model"`
|
|
TestCmd string `json:"test_cmd"`
|
|
}
|
|
|
|
func (s *Skill) handleRed(ctx context.Context, raw json.RawMessage) (json.RawMessage, error) {
|
|
var args redArgs
|
|
if err := json.Unmarshal(raw, &args); err != nil {
|
|
return nil, fmt.Errorf("parse args: %w", err)
|
|
}
|
|
if args.ProjectRoot == "" {
|
|
return nil, fmt.Errorf("project_root is required")
|
|
}
|
|
if args.Spec == "" {
|
|
return nil, fmt.Errorf("spec is required")
|
|
}
|
|
task := fmt.Sprintf(
|
|
"phase: red\nproject_root: %s\nspec: %s\nmodel: %s\ntest_cmd: %s",
|
|
args.ProjectRoot, args.Spec, s.resolveModel(args.Model), args.TestCmd,
|
|
)
|
|
return s.execute(ctx, task)
|
|
}
|
|
|
|
type greenArgs struct {
|
|
ProjectRoot string `json:"project_root"`
|
|
TestPath string `json:"test_path"`
|
|
Model string `json:"model"`
|
|
TestCmd string `json:"test_cmd"`
|
|
}
|
|
|
|
func (s *Skill) handleGreen(ctx context.Context, raw json.RawMessage) (json.RawMessage, error) {
|
|
var args greenArgs
|
|
if err := json.Unmarshal(raw, &args); err != nil {
|
|
return nil, fmt.Errorf("parse args: %w", err)
|
|
}
|
|
if args.ProjectRoot == "" {
|
|
return nil, fmt.Errorf("project_root is required")
|
|
}
|
|
if args.TestPath == "" {
|
|
return nil, fmt.Errorf("test_path is required")
|
|
}
|
|
task := fmt.Sprintf(
|
|
"phase: green\nproject_root: %s\ntest_path: %s\nmodel: %s\ntest_cmd: %s",
|
|
args.ProjectRoot, args.TestPath, s.resolveModel(args.Model), args.TestCmd,
|
|
)
|
|
return s.execute(ctx, task)
|
|
}
|
|
|
|
type refactorArgs struct {
|
|
ProjectRoot string `json:"project_root"`
|
|
TestPath string `json:"test_path"`
|
|
ImplPath string `json:"impl_path"`
|
|
Model string `json:"model"`
|
|
TestCmd string `json:"test_cmd"`
|
|
}
|
|
|
|
func (s *Skill) handleRefactor(ctx context.Context, raw json.RawMessage) (json.RawMessage, error) {
|
|
var args refactorArgs
|
|
if err := json.Unmarshal(raw, &args); err != nil {
|
|
return nil, fmt.Errorf("parse args: %w", err)
|
|
}
|
|
if args.ProjectRoot == "" {
|
|
return nil, fmt.Errorf("project_root is required")
|
|
}
|
|
if args.TestPath == "" {
|
|
return nil, fmt.Errorf("test_path is required")
|
|
}
|
|
if args.ImplPath == "" {
|
|
return nil, fmt.Errorf("impl_path is required")
|
|
}
|
|
task := fmt.Sprintf(
|
|
"phase: refactor\nproject_root: %s\ntest_path: %s\nimpl_path: %s\nmodel: %s\ntest_cmd: %s",
|
|
args.ProjectRoot, args.TestPath, args.ImplPath, s.resolveModel(args.Model), args.TestCmd,
|
|
)
|
|
return s.execute(ctx, task)
|
|
}
|
|
|
|
func (s *Skill) resolveModel(override string) string {
|
|
if override != "" {
|
|
return override
|
|
}
|
|
return s.cfg.DefaultModel
|
|
}
|
|
|
|
func (s *Skill) execute(ctx context.Context, task string) (json.RawMessage, error) {
|
|
if s.cfg.ExecutorFn == nil {
|
|
return nil, fmt.Errorf("no executor configured")
|
|
}
|
|
req := iexec.Request{
|
|
SkillPrompt: s.cfg.SkillPrompt,
|
|
TaskPrompt: task,
|
|
}
|
|
result, err := s.cfg.ExecutorFn(ctx, req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return json.Marshal(result)
|
|
}
|