Admin-API/cmd/main.go

package main

import (
	"fmt"
	"github.com/99designs/gqlgen/api"
	"github.com/99designs/gqlgen/codegen/config"
	"github.com/spf13/pflag"
	"os"
	"strings"
)

var (
	cfg              = config.DefaultConfig()
	availableModules = []string{"customer", "product"}
	fModule          = pflag.StringP("module", "m", "", "module name")
)

func init() {
	pflag.Parse()

	if fModule == nil || *fModule == "" {
		_, _ = fmt.Fprintf(os.Stderr, "specify module name; available modules are: %v\n", availableModules)
		os.Exit(2)
	}
}

func main() {
	*fModule = strings.ToLower(strings.TrimSpace(*fModule))

	moduleSchema := fmt.Sprintf("subgraph/%s.tpl", *fModule)
	f, err := os.OpenFile(moduleSchema, os.O_RDONLY, 0644)
	if err != nil {
		_, _ = fmt.Fprintf(os.Stderr, "unknown module; %v\n", err.Error())
		os.Exit(2)
	}
	defer f.Close()

	if stat, err := f.Stat(); err != nil || stat.Size() < 1 {
		fmt.Printf("illegal module definition schema")
		_, _ = fmt.Fprintf(os.Stderr, "illegal module definition schema")
		os.Exit(2)
	}

	// Federation
	cfg.Federation = config.PackageConfig{
		Filename: fmt.Sprintf("graph/%s/generated/federation.go", *fModule),
		Version:  2,
	}

	// Exec
	cfg.Exec = config.ExecConfig{
		Filename: fmt.Sprintf("graph/%s/generated/generated.go", *fModule),
	}

	// Model
	cfg.Model = config.PackageConfig{
		Filename: fmt.Sprintf("graph/%s/generated/models.go", *fModule),
	}

	cfg.Resolver = config.ResolverConfig{
		Filename: fmt.Sprintf("graph/%s/resolvers.go", *fModule),
		Type:     "Resolver",
	}

	// SchemaFilename
	cfg.SchemaFilename = config.StringList{
		moduleSchema,
		"ext/*.graphql",
	}

	if err := config.CompleteConfig(cfg); err != nil {
		_, _ = fmt.Fprintln(os.Stderr, err.Error())
		os.Exit(3)
	}

	if err := api.Generate(cfg); err != nil {
		_, _ = fmt.Fprintln(os.Stderr, err.Error())
		os.Exit(3)
	}

	os.Exit(0)
}