io.go

// Copyright 2023 Volvo Car Corporation
// SPDX-License-Identifier: Apache-2.0

package kubeutil

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

	"golang.org/x/exp/slices"
)

// ListGoFiles returns a list of all go files in the root directory and
// its children directories
func ListGoFiles(root string) ([]string, error) {
	return listFiles(root, []string{".go"})
}

// ListYAMLFiles returns a list of all yaml files in the root directory and
// its children directories
func ListYAMLFiles(root string) ([]string, error) {
	return listFiles(root, []string{".yaml", ".yml"})
}

// ListJSONFiles returns a list of all json files in the root directory and
// its children directories
func ListJSONFiles(root string) ([]string, error) {
	return listFiles(root, []string{".json"})
}

func listFiles(root string, extensions []string) ([]string, error) {
	var files []string

	fi, err := os.Stat(root)
	if err != nil {
		return nil, err
	}
	if !fi.IsDir() {
		return nil, errors.New("root is not a directory")
	}
	err = filepath.Walk(
		root, func(path string, info os.FileInfo, err error) error {
			if err != nil {
				return fmt.Errorf("walk %q %q, %w", path, info.Name(), err)
			}

			if !info.IsDir() && slices.Contains(
				extensions,
				filepath.Ext(filepath.Base(path)),
			) {
				files = append(files, path)
			}

			return nil
		},
	)
	if err != nil {
		return nil, fmt.Errorf("walk: %w", err)
	}
	return files, nil
}

func FileExists(filename string) bool {
	info, err := os.Stat(filename)
	if os.IsNotExist(err) {
		return false
	}
	return !info.IsDir()
}