146 lines
2.8 KiB
Go
146 lines
2.8 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"sort"
|
|
"strings"
|
|
|
|
"cliprepo/internal/config"
|
|
"cliprepo/internal/resolver"
|
|
"cliprepo/internal/tree"
|
|
"cliprepo/internal/utils"
|
|
|
|
"github.com/atotto/clipboard"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var delineate string
|
|
var useClipboard bool
|
|
|
|
var runCmd = &cobra.Command{
|
|
Use: "run",
|
|
Short: "Generate LLM context from repo",
|
|
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
|
|
cfg, err := config.Load()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 1. tree
|
|
treeStr, err := tree.Generate(".")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var builder strings.Builder
|
|
builder.WriteString(treeStr)
|
|
builder.WriteString("\n\n")
|
|
|
|
// 2. resolver (GLOBAL FILE REGISTRY)
|
|
r := resolver.New(".")
|
|
|
|
raw, ok := cfg.Preset[delineate]
|
|
if !ok {
|
|
return fmt.Errorf("preset not found: %s", delineate)
|
|
}
|
|
|
|
patterns := []string(raw)
|
|
|
|
var files []string
|
|
|
|
for _, pattern := range patterns {
|
|
|
|
// =========================
|
|
// FULL REPO SCAN
|
|
// =========================
|
|
if pattern == "*" {
|
|
all, err := r.WalkAll()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
files = append(files, all...)
|
|
continue
|
|
}
|
|
|
|
// =========================
|
|
// recursive glob support
|
|
// =========================
|
|
if strings.Contains(pattern, "**") {
|
|
matches, err := utils.ExpandGlob([]string{pattern})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, m := range matches {
|
|
r.AddFile(m, &files)
|
|
}
|
|
continue
|
|
}
|
|
|
|
// =========================
|
|
// normal glob
|
|
// =========================
|
|
matches, err := utils.ExpandGlob([]string{pattern})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, m := range matches {
|
|
r.AddFile(m, &files)
|
|
}
|
|
}
|
|
|
|
// 3. dedup + stable order
|
|
files = r.Unique(files)
|
|
sort.Strings(files)
|
|
|
|
// 4. template
|
|
template := cfg.Prompts.Preset
|
|
if template == "" {
|
|
template = "The following code is from file {{path}}:\n---\n{{content}}"
|
|
}
|
|
|
|
// 5. render
|
|
for _, file := range files {
|
|
|
|
content, err := os.ReadFile(file)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
|
|
// 先初始化 filePath 变量，避免作用域及同名包冲突
|
|
filePath := file
|
|
if cfg.Config.PathMode == "absolute" {
|
|
abs, _ := filepath.Abs(file)
|
|
filePath = abs
|
|
}
|
|
|
|
block := strings.ReplaceAll(template, "{{path}}", filePath)
|
|
block = strings.ReplaceAll(block, "{{content}}", string(content))
|
|
|
|
builder.WriteString(block)
|
|
builder.WriteString("\n\n")
|
|
}
|
|
|
|
result := builder.String()
|
|
|
|
// 6. output
|
|
if useClipboard {
|
|
return clipboard.WriteAll(result)
|
|
}
|
|
|
|
fmt.Println(result)
|
|
return nil
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(runCmd)
|
|
|
|
runCmd.Flags().StringVar(&delineate, "delineate", "", "preset key")
|
|
runCmd.Flags().BoolVar(&useClipboard, "clipboard", false, "copy output to clipboard")
|
|
} |