github.com/please-build/puku@v1.7.3-0.20240516143641-f7d7f4941f57/cmd/puku/puku.go

package main

import (
	"os"
	"path/filepath"

	"github.com/peterebden/go-cli-init/v5/flags"
	clilogging "github.com/peterebden/go-cli-init/v5/logging"

	"github.com/please-build/puku/config"
	"github.com/please-build/puku/generate"
	"github.com/please-build/puku/graph"
	"github.com/please-build/puku/licences"
	"github.com/please-build/puku/logging"
	"github.com/please-build/puku/migrate"
	"github.com/please-build/puku/please"
	"github.com/please-build/puku/proxy"
	"github.com/please-build/puku/sync"
	"github.com/please-build/puku/watch"
	"github.com/please-build/puku/work"
)

var opts = struct {
	Usage     string
	Verbosity clilogging.Verbosity `short:"v" long:"verbosity" description:"Verbosity of output (error, warning, notice, info, debug)" default:"info"`

	Fmt struct {
		Args struct {
			Paths []string `positional-arg-name:"packages" description:"The packages to process"`
		} `positional-args:"true"`
	} `command:"fmt" description:"Format build files in the provided paths"`
	Sync struct {
		Format string `short:"f" long:"format" choice:"json" choice:"text" default:"text" description:"output format when outputting to stdout"` //nolint
		Write  bool   `short:"w" long:"write" description:"Whether to write the files back or just print them to stdout"`
	} `command:"sync" description:"Synchronises the go.mod to the third party build file"`
	Lint struct {
		Format string `short:"f" long:"format" choice:"json" choice:"text" default:"text" description:"output format when outputting to stdout"` //nolint
		Args   struct {
			Paths []string `positional-arg-name:"packages" description:"The packages to process"`
		} `positional-args:"true"`
	} `command:"lint" description:"Lint build files in the provided paths"`
	Watch struct {
		Args struct {
			Paths []string `positional-arg-name:"packages" description:"The packages to process"`
		} `positional-args:"true"`
	} `command:"watch" description:"Watch build files in the provided paths and update them when needed"`
	Migrate struct {
		Write          bool     `short:"w" long:"write" description:"Whether to write the files back or just print them to stdout"`
		Format         string   `short:"f" long:"format" choice:"json" choice:"text" default:"text" description:"output format when outputting to stdout"` //nolint
		ThirdPartyDirs []string `long:"third_party_dir" description:"Directories to find go_module rules to migrate"`
		UpdateGoMod    bool     `short:"g" long:"update_go_mod" description:"Update the go mod with the module(s) being migrated"`
		Args           struct {
			Modules []string `positional-arg-name:"modules" description:"The modules to migrate to go_repo"`
		} `positional-args:"true"`
	} `command:"migrate" description:"Migrates from go_module to go_repo"`
	Licenses struct {
		Update struct {
			Format string `short:"f" long:"format" choice:"json" choice:"text" default:"text" description:"output format when outputting to stdout"` //nolint
			Write  bool   `short:"w" long:"write" description:"Whether to write the files back or just print them to stdout"`
			Args   struct {
				Paths []string `positional-arg-name:"packages" description:"The packages to process"`
			} `positional-args:"true"`
		} `command:"update" description:"Updates licences in the given paths"`
	} `command:"licences" description:"Commands relating to licences"`
}{
	Usage: `
puku is a tool used to generate and update Go targets in build files
`,
}

var log = logging.GetLogger()

var funcs = map[string]func(conf *config.Config, plzConf *please.Config, orignalWD string) int{
	"fmt": func(_ *config.Config, plzConf *please.Config, orignalWD string) int {
		paths := work.MustExpandPaths(orignalWD, opts.Fmt.Args.Paths)
		if err := generate.Update(plzConf, paths...); err != nil {
			log.Fatalf("%v", err)
		}
		return 0
	},
	"sync": func(_ *config.Config, plzConf *please.Config, _ string) int {
		g := graph.New(plzConf.BuildFileNames())
		if opts.Sync.Write {
			if err := sync.Sync(plzConf, g); err != nil {
				log.Fatalf("%v", err)
			}
		} else {
			if err := sync.SyncToStdout(opts.Sync.Format, plzConf, g); err != nil {
				log.Fatalf("%v", err)
			}
		}
		return 0
	},
	"lint": func(_ *config.Config, plzConf *please.Config, orignalWD string) int {

		paths := work.MustExpandPaths(orignalWD, opts.Lint.Args.Paths)
		if err := generate.UpdateToStdout(opts.Lint.Format, plzConf, paths...); err != nil {
			log.Fatalf("%v", err)
		}
		return 0
	},
	"watch": func(_ *config.Config, plzConf *please.Config, orignalWD string) int {
		paths := work.MustExpandPaths(orignalWD, opts.Watch.Args.Paths)
		if err := generate.Update(plzConf, paths...); err != nil {
			log.Fatalf("%v", err)
		}

		if err := watch.Watch(plzConf, paths...); err != nil {
			log.Fatalf("%v", err)
		}
		return 0
	},
	"migrate": func(conf *config.Config, plzConf *please.Config, orignalWD string) int {
		paths := opts.Migrate.ThirdPartyDirs
		if len(paths) == 0 {
			paths = []string{conf.GetThirdPartyDir()}
		}
		paths = work.MustExpandPaths(orignalWD, paths)
		if opts.Migrate.Write {
			if err := migrate.Migrate(conf, plzConf, opts.Migrate.UpdateGoMod, opts.Migrate.Args.Modules, paths); err != nil {
				log.Fatalf("%v", err)
			}
		} else {
			if err := migrate.MigrateToStdout(opts.Migrate.Format, conf, plzConf, opts.Migrate.UpdateGoMod, opts.Migrate.Args.Modules, paths); err != nil {
				log.Fatalf("%v", err)
			}
		}
		return 0
	},
	"update": func(_ *config.Config, plzConf *please.Config, orignalWD string) int {
		paths := work.MustExpandPaths(orignalWD, opts.Licenses.Update.Args.Paths)
		l := licences.New(proxy.New(proxy.DefaultURL), graph.New(plzConf.BuildFileNames()))
		if opts.Licenses.Update.Write {
			if err := l.Update(paths); err != nil {
				log.Fatalf("%v", err)
			}
		} else {
			if err := l.UpdateToStdout(opts.Licenses.Update.Format, paths); err != nil {
				log.Fatalf("%v", err)
			}
		}
		return 0
	},
}

func main() {
	cmd := flags.ParseFlagsOrDie("puku", &opts, nil)
	logging.InitLogging(opts.Verbosity)

	wd, err := os.Getwd()
	if err != nil {
		log.Fatalf("failed to get wd: %v", err)
	}

	root, err := work.FindRoot()
	if err != nil {
		log.Fatalf("%v", err)
	}

	wd, err = filepath.Rel(root, wd)
	if err != nil {
		log.Fatalf("failed to get wd: %v", err)
	}

	if err := os.Chdir(root); err != nil {
		log.Fatalf("failed to set working dir to repo root: %v", err)
	}

	conf, err := config.ReadConfig(".")
	if err != nil {
		log.Fatalf("failed to read config: %v", err)
	}

	plzConf, err := please.QueryConfig(conf.GetPlzPath())
	if err != nil {
		log.Fatalf("failed to query config: %w", err)
	}
	os.Exit(funcs[cmd](conf, plzConf, wd))
}