go-fd/example_test.go at main · startvibecoding/go-fd · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
package gofd_test

import (
	"context"
	"fmt"
	"os"
	"path/filepath"
	"sort"

	gofd "github.com/startvibecoding/go-fd"
)

func ExampleFind() {
	dir, err := os.MkdirTemp("", "gofd-example-")
	if err != nil {
		panic(err)
	}
	defer os.RemoveAll(dir)

	for _, name := range []string{"main.go", "README.md", "internal/util.go"} {
		path := filepath.Join(dir, filepath.FromSlash(name))
		if err := os.MkdirAll(filepath.Dir(path), 0o755); err != nil {
			panic(err)
		}
		if err := os.WriteFile(path, []byte(""), 0o644); err != nil {
			panic(err)
		}
	}

	paths, err := gofd.Find(context.Background(), gofd.Options{
		Pattern: `\.go$`,
		Paths:   []string{dir},
	})
	if err != nil {
		panic(err)
	}

	var rels []string
	for _, path := range paths {
		rel, err := filepath.Rel(dir, path)
		if err != nil {
			panic(err)
		}
		rels = append(rels, filepath.ToSlash(rel))
	}
	sort.Strings(rels)
	for _, rel := range rels {
		fmt.Println(rel)
	}

	// Output:
	// internal/util.go
	// main.go
}

func ExampleStream() {
	dir, err := os.MkdirTemp("", "gofd-example-")
	if err != nil {
		panic(err)
	}
	defer os.RemoveAll(dir)

	for _, name := range []string{"main.go", "util.go"} {
		if err := os.WriteFile(filepath.Join(dir, name), []byte(""), 0o644); err != nil {
			panic(err)
		}
	}

	results, errs, err := gofd.Stream(context.Background(), gofd.Options{
		Pattern: `\.go$`,
		Paths:   []string{dir},
	})
	if err != nil {
		panic(err)
	}

	count := 0
	for range results {
		count++
	}
	for err := range errs {
		if err != nil {
			panic(err)
		}
	}
	fmt.Println(count)

	// Output:
	// 2
}