95 lines
1.6 KiB
Go
95 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"io/fs"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"go.uber.org/zap"
|
|
"go.uber.org/zap/zapcore"
|
|
"golang.org/x/exp/slices"
|
|
)
|
|
|
|
const defaultDirPermissions = 0o755
|
|
|
|
var (
|
|
GoSourceFiles []string
|
|
GeneratedMockFiles []string
|
|
WorkingDir string
|
|
OutDir string
|
|
dirsToIgnore = []string{
|
|
".git",
|
|
"magefiles",
|
|
".concourse",
|
|
".run",
|
|
".task",
|
|
}
|
|
)
|
|
|
|
func init() {
|
|
if wd, err := os.Getwd(); err != nil {
|
|
panic(err)
|
|
} else {
|
|
WorkingDir = wd
|
|
}
|
|
|
|
OutDir = filepath.Join(WorkingDir, "out")
|
|
|
|
if err := os.MkdirAll(OutDir, defaultDirPermissions); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if err := initLogging(); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if err := initSourceFiles(); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
zap.L().Info("Completed initialization")
|
|
}
|
|
|
|
func initLogging() error {
|
|
cfg := zap.NewDevelopmentConfig()
|
|
cfg.Encoding = "console"
|
|
cfg.Level = zap.NewAtomicLevelAt(zapcore.InfoLevel)
|
|
|
|
if logger, err := cfg.Build(); err != nil {
|
|
return err
|
|
} else {
|
|
zap.ReplaceGlobals(logger)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func initSourceFiles() error {
|
|
return filepath.WalkDir(WorkingDir, func(path string, d fs.DirEntry, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if d.IsDir() {
|
|
if slices.Contains(dirsToIgnore, filepath.Base(path)) {
|
|
return fs.SkipDir
|
|
}
|
|
return nil
|
|
}
|
|
|
|
_, ext, found := strings.Cut(filepath.Base(path), ".")
|
|
if !found {
|
|
return nil
|
|
}
|
|
|
|
switch ext {
|
|
case "mock.go":
|
|
GeneratedMockFiles = append(GeneratedMockFiles, path)
|
|
case "go":
|
|
GoSourceFiles = append(GoSourceFiles, path)
|
|
}
|
|
|
|
return nil
|
|
})
|
|
}
|