feat: More extensive configuration

Author: kevinmichaelchen

.golangci.yml

@@ -0,0 +1,3 @@
+.PHONY: lint
+lint:
+	pkgx golangci-lint@latest run

Makefile

@@ -3,24 +3,29 @@ module github.com/kevinmichaelchen/graphql-schema-picker
 go 1.21.0
 
 require (
+	github.com/BurntSushi/toml v1.3.2
 	github.com/charmbracelet/log v0.3.0
 	github.com/dominikbraun/graph v0.23.0
 	github.com/graphql-go/graphql v0.8.1
 	github.com/spf13/cobra v1.7.0
+	github.com/stretchr/testify v1.8.4
 )
 
 require (
 	github.com/aymanbagabas/go-osc52/v2 v2.0.1 // indirect
 	github.com/charmbracelet/lipgloss v0.9.1 // indirect
+	github.com/davecgh/go-spew v1.1.1 // indirect
 	github.com/go-logfmt/logfmt v0.6.0 // indirect
 	github.com/inconshreveable/mousetrap v1.1.0 // indirect
 	github.com/lucasb-eyer/go-colorful v1.2.0 // indirect
 	github.com/mattn/go-isatty v0.0.18 // indirect
 	github.com/mattn/go-runewidth v0.0.15 // indirect
 	github.com/muesli/reflow v0.3.0 // indirect
 	github.com/muesli/termenv v0.15.2 // indirect
+	github.com/pmezard/go-difflib v1.0.0 // indirect
 	github.com/rivo/uniseg v0.2.0 // indirect
 	github.com/spf13/pflag v1.0.5 // indirect
 	golang.org/x/exp v0.0.0-20231006140011-7918f672742d // indirect
 	golang.org/x/sys v0.13.0 // indirect
+	gopkg.in/yaml.v3 v3.0.1 // indirect
 )

go.mod

@@ -1,3 +1,5 @@
+github.com/BurntSushi/toml v1.3.2 h1:o7IhLm0Msx3BaB+n3Ag7L8EVlByGnpq14C4YWiu/gL8=
+github.com/BurntSushi/toml v1.3.2/go.mod h1:CxXYINrC8qIiEnFrOxCa7Jy5BFHlXnUU2pbicEuybxQ=
 github.com/aymanbagabas/go-osc52/v2 v2.0.1 h1:HwpRHbFMcZLEVr42D4p7XBqjyuxQH5SMiErDT4WkJ2k=
 github.com/aymanbagabas/go-osc52/v2 v2.0.1/go.mod h1:uYgXzlJ7ZpABp8OJ+exZzJJhRNQ2ASbcXHWsFqH8hp8=
 github.com/charmbracelet/lipgloss v0.9.1 h1:PNyd3jvaJbg4jRHKWXnCj1akQm4rh8dbEzN1p/u1KWg=
@@ -43,6 +45,7 @@ golang.org/x/exp v0.0.0-20231006140011-7918f672742d/go.mod h1:ldy0pHrwJyGW56pPQz
 golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.13.0 h1:Af8nKPmuFypiUBjVoU9V20FiaFXOcuZI21p0ycVYYGE=
 golang.org/x/sys v0.13.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
 gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

go.sum

@@ -2,10 +2,11 @@ package cli
 
 import (
 	"bufio"
+	"os"
+
 	"github.com/charmbracelet/log"
 	"github.com/graphql-go/graphql/language/ast"
 	"github.com/graphql-go/graphql/language/printer"
-	"os"
 )
 
 func printSDL(doc *ast.Document) {

internal/cli/ast_print.go

@@ -1,9 +1,10 @@
 package cli
 
 import (
+	"os"
+
 	"github.com/charmbracelet/log"
 	"github.com/spf13/cobra"
-	"os"
 )
 
 var rootCmd = &cobra.Command{
@@ -16,6 +17,11 @@ var rootCmd = &cobra.Command{
 		if debug {
 			log.SetLevel(log.DebugLevel)
 		}
+
+		err := loadConfig()
+		if err != nil {
+			log.Fatal("error occurred while loading config", "err", err)
+		}
 	},
 }
 
@@ -26,6 +32,7 @@ var (
 	dryRun             bool
 	desiredDefinitions []string
 	output             string
+	configPath         string
 )
 
 // LDFlags contain fields that get linked and compiled into the final binary
@@ -46,6 +53,7 @@ func init() {
 
 	pick.Flags().StringSliceVarP(&desiredDefinitions, "definitions", "d", []string{}, "definitions from the SDL you want to pick/keep")
 	pick.Flags().StringVarP(&output, "output", "o", "", "where the resulting schema/SDL file is written")
+	pick.Flags().StringVarP(&configPath, "config", "c", "", "path to config file")
 }
 
 func Main(ldf LDFlags) {

internal/cli/cli.go

@@ -1,10 +1,11 @@
 package cli
 
 import (
+	"os"
+
 	"github.com/charmbracelet/log"
 	"github.com/dominikbraun/graph/draw"
 	"github.com/spf13/cobra"
-	"os"
 )
 
 var pick = &cobra.Command{

internal/cli/cli_pick.go

@@ -2,6 +2,7 @@ package cli
 
 import (
 	"fmt"
+
 	"github.com/spf13/cobra"
 )
 

internal/cli/cli_version.go

@@ -0,0 +1,30 @@
+package cli
+
+import (
+	"fmt"
+	"github.com/BurntSushi/toml"
+)
+
+var cfg config
+
+type config struct {
+	Types []Type `toml:"type"`
+}
+
+// Type - A GraphQL type in a GraphQL schema.
+type Type struct {
+	// Name - The name of the GraphQL type you want to keep/pick.
+	Name string `toml:"name"`
+
+	// DenyList - List of fields within that type you want to filter out.
+	DenyList []string `toml:"deny_list"`
+}
+
+func loadConfig() error {
+	_, err := toml.DecodeFile(configPath, &cfg)
+	if err != nil {
+		return fmt.Errorf("unable to decode config: %w", err)
+	}
+
+	return nil
+}

internal/cli/config.go

@@ -0,0 +1,28 @@
+package cli
+
+import (
+	"fmt"
+	"github.com/BurntSushi/toml"
+	"github.com/stretchr/testify/require"
+	"testing"
+)
+
+const blob = `
+[[type]]
+name = "Person"
+deny_list = ["foobar"]
+
+[[type]]
+name = "Stairway to Heaven"
+`
+
+func TestDecode(t *testing.T) {
+	var testCfg config
+
+	_, err := toml.Decode(blob, &testCfg)
+	require.NoError(t, err)
+
+	for _, s := range testCfg.Types {
+		fmt.Printf("%s (%v)\n", s.Name, s.DenyList)
+	}
+}