main.go

package main

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

	"go-simpler.org/errorsx"
)

func main() {
	if err := run(); err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
}

func run() (runErr error) {
	cfg, err := readFlags(os.Args[1:])
	if err != nil {
		if errors.Is(err, flag.ErrHelp) {
			return nil
		}
		return err
	}

	path, dir := cfg.path, cfg.dir

	if path != "" {
		cfgFile, err := os.Open(path)
		if err != nil {
			return err
		}
		defer errorsx.Close(cfgFile, &runErr)

		cfg, err = readConfig(cfgFile)
		if err != nil {
			return err
		}
	}

	if dir != "" {
		if err := os.Chdir(dir); err != nil {
			return err
		}
	}

	if err := os.Mkdir(cfg.Pkg, 0o755); err != nil && !errors.Is(err, os.ErrExist) {
		return err
	}

	genFile, err := os.Create(filepath.Join(cfg.Pkg, cfg.Pkg+".go"))
	if err != nil {
		return err
	}
	defer errorsx.Close(genFile, &runErr)

	return writeCode(genFile, cfg)
}