51 lines
1.5 KiB
Go
51 lines
1.5 KiB
Go
package issues
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"github.com/LazyBachelor/LazyPM/internal/models"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// Variables for completion options and functions.
|
|
var (
|
|
typeOptions = []string{"bug", "feature", "task", "chore"}
|
|
statusOptions = []string{"open", "closed", "in_progress", "ready_to_sprint"}
|
|
priorityRange = []string{"0", "1", "2", "3", "4"}
|
|
)
|
|
|
|
// completeIssues provides shell completion for issue IDs and titles.
|
|
func completeIssues(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
|
|
issues, _ := GetIssueCompletions(cmd.Context(), toComplete)
|
|
var ids []string
|
|
for _, issue := range issues {
|
|
ids = append(ids, issue.ID)
|
|
}
|
|
return ids, cobra.ShellCompDirectiveNoFileComp
|
|
}
|
|
|
|
// GetIssueCompletions fetches issues matching the toComplete string for shell completion.
|
|
func GetIssueCompletions(ctx context.Context, toComplete string) ([]*models.Issue, cobra.ShellCompDirective) {
|
|
app := AppFromContext(ctx)
|
|
if app == nil {
|
|
return nil, cobra.ShellCompDirectiveNoFileComp
|
|
}
|
|
|
|
issues, err := app.Issues.SearchIssues(ctx, "", models.IssueFilter{})
|
|
if err != nil {
|
|
return nil, cobra.ShellCompDirectiveNoFileComp
|
|
}
|
|
|
|
var completions []*models.Issue
|
|
for _, issue := range issues {
|
|
if strings.HasPrefix(issue.ID, toComplete) {
|
|
completions = append(completions, issue)
|
|
} else if strings.HasPrefix(issue.Title, toComplete) {
|
|
completions = append(completions, issue)
|
|
}
|
|
}
|
|
|
|
return completions, cobra.ShellCompDirectiveNoFileComp
|
|
}
|