finder.go

package main

import (
	"os"
	"path/filepath"
	"slices"
	"strings"
)

// baseName is like filepath.Base but returns empty string in the following cases:
// if name is just line-separator or ends with line-separator, or it's a dot.
// if name is an empty string
func baseName(name string) string {
	base := filepath.Base(name)
	if len(name) == 0 || base == "." || base == "/" || name[len(name)-1] == filepath.Separator {
		return ""
	}
	return base
}

func findFiles(root string, search string, exclude []string, searchResultPrepend string) ([]string, error) {
	Verbose("find files ",
		"root: ", root,
		"search: ", search,
		"exclude: ", exclude,
		"search_result_prepend: ", searchResultPrepend,
	)

	root = strings.TrimSuffix(root, string(os.PathSeparator)) + string(os.PathSeparator)
	var result []string
	walkFn := func(path string, info os.FileInfo, err error) error {
		path = strings.TrimPrefix(path, root)
		if err != nil {
			return err
		}

		if slices.Contains(exclude, path) {
			if info.IsDir() {
				return filepath.SkipDir
			}
			return nil
		}

		if path == "" { // the root itself should be ignored
			return nil
		}

		if strings.Contains(strings.ToLower(path), strings.ToLower(search)) || search == "" {
			res := filepath.Join(searchResultPrepend, strings.TrimSuffix(path, ".md")) // Remove .md from end of markdown files.
			if info.IsDir() {
				res = res + "/" // change style of directories (colorize and append / to them)
			}

			result = append(result, res)
			if info.IsDir() {
				return filepath.SkipDir
			}
		}

		return nil
	}

	err := filepath.Walk(root, walkFn)
	if err != nil {
		return nil, err
	}

	return result, nil
}