github.com/zyedidia/knit@v1.1.2-0.20230901152954-f7d4e39a0e24/config.go

package knit

import (
	"errors"
	"os"
	"path/filepath"

	"github.com/adrg/xdg"
	"github.com/pelletier/go-toml/v2"
)

var ErrBuildFileNotFound = errors.New("build file not found")

const defaultFile = "knitfile.def"

// Returns true if 'path' exists.
func exists(path string) bool {
	_, err := os.Stat(path)
	return err == nil
}

// Returns true if 'path' exists and is not a directory.
func existsFile(path string) bool {
	info, err := os.Stat(path)
	return err == nil && !info.IsDir()
}

var configDirs = []string{
	filepath.Join(xdg.ConfigHome, "knit"),
}

func init() {
	for _, dir := range xdg.DataDirs {
		configDirs = append(configDirs, filepath.Join(dir, "knit"))
	}
}
func DefaultBuildFile() (string, bool) {
	for _, dir := range configDirs {
		if existsFile(filepath.Join(dir, defaultFile)) {
			return filepath.Join(defaultFile), true
		} else if existsFile(filepath.Join(dir, title(defaultFile))) {
			return filepath.Join(dir, title(defaultFile)), true
		}
	}
	return "", false
}

func FindBuildFile(name string) (string, string, error) {
	if filepath.IsAbs(name) {
		if existsFile(name) {
			return name, filepath.Dir(name), nil
		} else {
			return "", "", errors.New("does not exist")
		}
	}

	wd, err := os.Getwd()
	if err != nil {
		return "", "", err
	}
	path := wd
	for filepath.Dir(path) != path {
		if existsFile(filepath.Join(path, name)) {
			p, e := filepath.Rel(wd, path)
			return name, p, e
		}
		if existsFile(filepath.Join(path, title(name))) {
			p, e := filepath.Rel(wd, path)
			return title(name), p, e
		}
		path = filepath.Dir(path)
	}
	return "", "", nil
}

const configFile = ".knit.toml"

func UserDefaults() (UserFlags, error) {
	var flags UserFlags

	loadFile := func(dir string) error {
		configPath := filepath.Join(dir, configFile)
		if exists(configPath) {
			data, err := os.ReadFile(configPath)
			if err != nil {
				return err
			}
			err = toml.Unmarshal(data, &flags)
			if err != nil {
				return err
			}
		}
		return nil
	}

	wd, err := os.Getwd()
	if err != nil {
		return UserFlags{}, err
	}
	dirs := []string{wd}
	path := wd
	for filepath.Dir(path) != path {
		path = filepath.Dir(path)
		dirs = append(dirs, path)
	}
	dirs = append(dirs, configDirs...)
	for i := len(dirs) - 1; i >= 0; i-- {
		dir := dirs[i]
		err := loadFile(dir)
		if err != nil {
			return UserFlags{}, err
		}
	}
	return flags, nil
}