first mvp

This commit is contained in:
2026-05-16 01:39:45 +08:00
parent b3a2ae4681
commit 69afb0a53e
13 changed files with 205 additions and 272 deletions
+19
View File
@@ -0,0 +1,19 @@
package tree
import (
"os"
"path/filepath"
ignore "github.com/sabhiram/go-gitignore"
)
// Load 加载 .gitignore
func Load(root string) (*ignore.GitIgnore, error) {
path := filepath.Join(root, ".gitignore")
if _, err := os.Stat(path); os.IsNotExist(err) {
return ignore.CompileIgnoreLines(), nil
}
return ignore.CompileIgnoreFile(path)
}
+4 -37
View File
@@ -5,73 +5,40 @@ import (
"os"
"path/filepath"
"strings"
"cliprepo/internal/gitignore"
"github.com/atotto/clipboard"
)
// Generate 从 root 目录开始遍历文件树,并生成类似 `tree` 命令的目录结构字符串
//
// 功能特点:
// 1. 递归遍历目录
// 2. 支持 .gitignore 规则过滤
// 3. 输出树形结构字符串
func Generate(root string) (string, error) {
// 加载 gitignore 规则(如果失败则忽略错误,默认不过滤)
ig, _ := gitignore.Load(root)
ig, _ := Load(root)
var builder strings.Builder
// 递归遍历目录结构
err := filepath.Walk(root, func(path string, info os.FileInfo, err error) error {
if err != nil {
return err
}
// 根目录特殊处理,统一显示为 "."
if path == root {
builder.WriteString(".\n")
return nil
}
// 转换为相对路径,便于匹配 gitignore 和计算层级
rel, _ := filepath.Rel(root, path)
// 如果路径匹配 gitignore 规则,则跳过
// 如果是目录,则直接跳过整个子树
if ig.MatchesPath(rel) {
// gitignore filter
if ig != nil && ig.MatchesPath(rel) {
if info.IsDir() {
return filepath.SkipDir
}
return nil
}
// 计算当前节点在目录树中的层级
// 通过路径分隔符数量判断深度
level := strings.Count(rel, string(os.PathSeparator))
// 根据层级生成缩进
prefix := strings.Repeat("│ ", level)
// 写入当前节点(文件或目录)
builder.WriteString(fmt.Sprintf("%s├── %s\n", prefix, info.Name()))
return nil
})
return builder.String(), err
}
// Copy 将生成的字符串写入系统剪贴板
//
// 常用于 CLI 工具:直接复制 tree 输出结果
func Copy(content string) error {
return clipboard.WriteAll(content)
}
// Write 将内容写入指定文件路径
//
// 如果文件存在则覆盖
// 权限默认设置为 0644rw-r--r--
func Write(path, content string) error {
return os.WriteFile(path, []byte(content), 0644)
}