Carlos Florêncio
52 lines
1.4 KiB
Go
52 lines
1.4 KiB
Go
package controller
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/kandev/kandev/internal/prompts/dto"
|
|
"github.com/kandev/kandev/internal/prompts/service"
|
|
)
|
|
|
|
type Controller struct {
|
|
service *service.Service
|
|
}
|
|
|
|
func NewController(svc *service.Service) *Controller {
|
|
return &Controller{service: svc}
|
|
}
|
|
|
|
func (c *Controller) ListPrompts(ctx context.Context) (dto.PromptsResponse, error) {
|
|
prompts, err := c.service.ListPrompts(ctx)
|
|
if err != nil {
|
|
return dto.PromptsResponse{}, err
|
|
}
|
|
result := make([]dto.PromptDTO, 0, len(prompts))
|
|
for _, prompt := range prompts {
|
|
if prompt == nil {
|
|
continue
|
|
}
|
|
result = append(result, dto.FromPrompt(prompt))
|
|
}
|
|
return dto.PromptsResponse{Prompts: result}, nil
|
|
}
|
|
|
|
func (c *Controller) CreatePrompt(ctx context.Context, req dto.CreatePromptRequest) (dto.PromptDTO, error) {
|
|
prompt, err := c.service.CreatePrompt(ctx, req.Name, req.Content)
|
|
if err != nil {
|
|
return dto.PromptDTO{}, err
|
|
}
|
|
return dto.FromPrompt(prompt), nil
|
|
}
|
|
|
|
func (c *Controller) UpdatePrompt(ctx context.Context, promptID string, req dto.UpdatePromptRequest) (dto.PromptDTO, error) {
|
|
prompt, err := c.service.UpdatePrompt(ctx, promptID, req.Name, req.Content)
|
|
if err != nil {
|
|
return dto.PromptDTO{}, err
|
|
}
|
|
return dto.FromPrompt(prompt), nil
|
|
}
|
|
|
|
func (c *Controller) DeletePrompt(ctx context.Context, promptID string) error {
|
|
return c.service.DeletePrompt(ctx, promptID)
|
|
}
|