-
Notifications
You must be signed in to change notification settings - Fork 1.2k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Refactor main so tests can execute the generator
- Loading branch information
Showing
10 changed files
with
288 additions
and
214 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,177 @@ | ||
| package codegen | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "go/build" | ||
| "io/ioutil" | ||
| "os" | ||
| "path/filepath" | ||
| "strings" | ||
|
|
||
| "github.com/pkg/errors" | ||
| "github.com/vektah/gqlgen/codegen/templates" | ||
| "github.com/vektah/gqlgen/neelance/schema" | ||
| "golang.org/x/tools/imports" | ||
| ) | ||
|
|
||
| type Config struct { | ||
| SchemaStr string | ||
| Typemap map[string]string | ||
|
|
||
| schema *schema.Schema | ||
|
|
||
| ExecFilename string | ||
| ExecPackageName string | ||
| execDir string | ||
| fullExecPackageName string | ||
|
|
||
| ModelFilename string | ||
| ModelPackageName string | ||
| modelDir string | ||
| fullModelPackageName string | ||
| } | ||
|
|
||
| func Generate(cfg Config) error { | ||
| if err := cfg.normalize(); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| modelsBuild, err := cfg.models() | ||
| if err != nil { | ||
| return errors.Wrap(err, "model plan failed") | ||
| } | ||
| if len(modelsBuild.Models) > 0 { | ||
| modelsBuild.PackageName = cfg.ModelPackageName | ||
|
|
||
| buf, err := templates.Run("models.gotpl", modelsBuild) | ||
| if err != nil { | ||
| return errors.Wrap(err, "model generation failed") | ||
| } | ||
|
|
||
| if err = write(cfg.ModelFilename, buf.Bytes()); err != nil { | ||
| return err | ||
| } | ||
| for _, model := range modelsBuild.Models { | ||
| cfg.Typemap[model.GQLType] = cfg.fullModelPackageName + "." + model.GoType | ||
| } | ||
|
|
||
| for _, enum := range modelsBuild.Enums { | ||
| cfg.Typemap[enum.GQLType] = cfg.fullModelPackageName + "." + enum.GoType | ||
| } | ||
| } | ||
|
|
||
| build, err := cfg.bind() | ||
| if err != nil { | ||
| return errors.Wrap(err, "exec plan failed") | ||
| } | ||
| build.SchemaRaw = cfg.SchemaStr | ||
| build.PackageName = cfg.ExecPackageName | ||
|
|
||
| buf, err := templates.Run("generated.gotpl", build) | ||
| if err != nil { | ||
| return errors.Wrap(err, "exec codegen failed") | ||
| } | ||
|
|
||
| if err = write(cfg.ExecFilename, buf.Bytes()); err != nil { | ||
| return err | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| func (cfg *Config) normalize() error { | ||
| if cfg.ModelFilename == "" { | ||
| return errors.New("ModelFilename is required") | ||
| } | ||
| cfg.ModelFilename = abs(cfg.ModelFilename) | ||
| cfg.modelDir = filepath.Dir(cfg.ModelFilename) | ||
| if cfg.ModelPackageName == "" { | ||
| cfg.ModelPackageName = filepath.Base(cfg.modelDir) | ||
| } | ||
| cfg.fullModelPackageName = fullPackageName(cfg.modelDir, cfg.ModelPackageName) | ||
|
|
||
| if cfg.ExecFilename == "" { | ||
| return errors.New("ModelFilename is required") | ||
| } | ||
| cfg.ExecFilename = abs(cfg.ExecFilename) | ||
| cfg.execDir = filepath.Dir(cfg.ExecFilename) | ||
| if cfg.ExecPackageName == "" { | ||
| cfg.ExecPackageName = filepath.Base(cfg.execDir) | ||
| } | ||
| cfg.fullExecPackageName = fullPackageName(cfg.execDir, cfg.ExecPackageName) | ||
|
|
||
| builtins := map[string]string{ | ||
| "__Directive": "github.com/vektah/gqlgen/neelance/introspection.Directive", | ||
| "__Type": "github.com/vektah/gqlgen/neelance/introspection.Type", | ||
| "__Field": "github.com/vektah/gqlgen/neelance/introspection.Field", | ||
| "__EnumValue": "github.com/vektah/gqlgen/neelance/introspection.EnumValue", | ||
| "__InputValue": "github.com/vektah/gqlgen/neelance/introspection.InputValue", | ||
| "__Schema": "github.com/vektah/gqlgen/neelance/introspection.Schema", | ||
| "Int": "github.com/vektah/gqlgen/graphql.Int", | ||
| "Float": "github.com/vektah/gqlgen/graphql.Float", | ||
| "String": "github.com/vektah/gqlgen/graphql.String", | ||
| "Boolean": "github.com/vektah/gqlgen/graphql.Boolean", | ||
| "ID": "github.com/vektah/gqlgen/graphql.ID", | ||
| "Time": "github.com/vektah/gqlgen/graphql.Time", | ||
| "Map": "github.com/vektah/gqlgen/graphql.Map", | ||
| } | ||
|
|
||
| if cfg.Typemap == nil { | ||
| cfg.Typemap = map[string]string{} | ||
| } | ||
| for k, v := range builtins { | ||
| if _, ok := cfg.Typemap[k]; !ok { | ||
| cfg.Typemap[k] = v | ||
| } | ||
| } | ||
|
|
||
| cfg.schema = schema.New() | ||
| return cfg.schema.Parse(cfg.SchemaStr) | ||
| } | ||
|
|
||
| func abs(path string) string { | ||
| absPath, err := filepath.Abs(path) | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
| return absPath | ||
| } | ||
|
|
||
| func fullPackageName(dir string, pkgName string) string { | ||
| fullPkgName := filepath.Join(filepath.Dir(dir), pkgName) | ||
|
|
||
| for _, gopath := range filepath.SplitList(build.Default.GOPATH) { | ||
| gopath = filepath.Join(gopath, "src") + string(os.PathSeparator) | ||
| if strings.HasPrefix(fullPkgName, gopath) { | ||
| fullPkgName = fullPkgName[len(gopath):] | ||
| } | ||
| } | ||
| return filepath.ToSlash(fullPkgName) | ||
| } | ||
|
|
||
| func gofmt(filename string, b []byte) ([]byte, error) { | ||
| out, err := imports.Process(filename, b, nil) | ||
| if err != nil { | ||
| return b, errors.Wrap(err, "unable to gofmt") | ||
| } | ||
| return out, nil | ||
| } | ||
|
|
||
| func write(filename string, b []byte) error { | ||
| err := os.MkdirAll(filepath.Dir(filename), 0755) | ||
| if err != nil { | ||
| return errors.Wrap(err, "failed to create directory") | ||
| } | ||
|
|
||
| formatted, err := gofmt(filename, b) | ||
| if err != nil { | ||
| fmt.Fprintf(os.Stderr, "gofmt failed: %s", err.Error()) | ||
| formatted = b | ||
| } | ||
|
|
||
| err = ioutil.WriteFile(filename, formatted, 0644) | ||
| if err != nil { | ||
| return errors.Wrapf(err, "failed to write %s", filename) | ||
| } | ||
|
|
||
| return nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.