🌲 组合模式：让"文件"和"文件夹"一视同仁

package main

// Component 定义文件系统组件的统一接口
type Component interface {
    search(keyword string)
    getName() string
}

package main

import "fmt"

// File 叶节点：文件
type File struct {
    name string
}

func (f *File) search(keyword string) {
    fmt.Printf("📄 在文件 [%s] 中搜索关键词 '%s'\n", f.name, keyword)
}

func (f *File) getName() string {
    return f.name
}

package main

import "fmt"

// Folder 容器：文件夹
type Folder struct {
    components []Component
    name       string
}

func (f *Folder) search(keyword string) {
    fmt.Printf("📁 递归搜索文件夹 [%s] 中的 '%s'\n", f.name, keyword)
    for _, component := range f.components {
        component.search(keyword)  // 统一调用
    }
}

func (f *Folder) getName() string {
    return f.name
}

// add 添加子组件（可以是文件或文件夹）
func (f *Folder) add(c Component) {
    f.components = append(f.components, c)
}

// remove 移除子组件
func (f *Folder) remove(c Component) {
    for i, child := range f.components {
        if child.getName() == c.getName() {
            f.components = append(f.components[:i], f.components[i+1:]...)
            return
        }
    }
}

package main

import "fmt"

func main() {
    // 构建文件系统树
    //
    // 📁 project
    // ├── 📄 README.md
    // ├── 📄 main.go
    // └── 📁 src
    //     ├── 📄 handler.go
    //     └── 📁 utils
    //         └── 📄 logger.go

    // 创建文件
    file1 := &File{name: "README.md"}
    file2 := &File{name: "main.go"}
    file3 := &File{name: "handler.go"}
    file4 := &File{name: "logger.go"}

    // 创建 utils 文件夹
    utils := &Folder{name: "utils"}
    utils.add(file4)

    // 创建 src 文件夹
    src := &Folder{name: "src"}
    src.add(file3)
    src.add(utils)

    // 创建项目根目录
    project := &Folder{name: "project"}
    project.add(file1)
    project.add(file2)
    project.add(src)

    // 执行搜索 - 客户端不需要知道内部结构
    fmt.Println("=== 搜索整个项目 ===")
    project.search("TODO")

    fmt.Println("\n=== 只搜索 src 目录 ===")
    src.search("TODO")

    fmt.Println("\n=== 只搜索单个文件 ===")
    file1.search("TODO")
}