Carlos Florêncio
44 lines
1.4 KiB
Go
44 lines
1.4 KiB
Go
package promptcontext
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/kandev/kandev/internal/sysprompt"
|
|
"github.com/kandev/kandev/internal/task/models"
|
|
)
|
|
|
|
// RepositoryNameResolver resolves a repository's display name for prompt context.
|
|
// An error leaves the repository ID as the stable fallback name.
|
|
type RepositoryNameResolver func(context.Context, string) (string, error)
|
|
|
|
// BuildPullRequestTargets converts task repository snapshots into the prompt
|
|
// context used by both task orchestration and task message handlers.
|
|
func BuildPullRequestTargets(
|
|
ctx context.Context,
|
|
links []*models.TaskRepository,
|
|
resolveRepositoryName RepositoryNameResolver,
|
|
) []sysprompt.PullRequestTarget {
|
|
targets := make([]sysprompt.PullRequestTarget, 0, len(links))
|
|
for _, link := range links {
|
|
if link == nil || link.BranchPolicyID == "" || link.BranchPolicyPullRequestTarget == "" {
|
|
continue
|
|
}
|
|
repositoryName := link.RepositoryID
|
|
if resolveRepositoryName != nil {
|
|
if name, err := resolveRepositoryName(ctx, link.RepositoryID); err == nil && name != "" {
|
|
repositoryName = name
|
|
}
|
|
}
|
|
workingBranch := link.CheckoutBranch
|
|
if workingBranch == "" {
|
|
workingBranch = link.BaseBranch
|
|
}
|
|
targets = append(targets, sysprompt.PullRequestTarget{
|
|
RepositoryName: repositoryName,
|
|
WorkingBranch: workingBranch,
|
|
TargetBranch: link.BranchPolicyPullRequestTarget,
|
|
})
|
|
}
|
|
return targets
|
|
}
|