Custom mongodb scalar and generate commands

CHANGELOG.md

@@ -2,6 +2,11 @@
 
 All notable changes will be docmented here.
 
+## 2.0.2
+
+* Added custom scalar in GraphQL configuration for MongoDB ObjectID
+* Added go generate action in main.go for API Application and JS Application
+
 ## 2.0.1
 
 * Fixed readme issue

VERSION

@@ -1 +1 @@
-2.0.1
+2.0.2

internal/generators/ApiAppGenerator.go

@@ -37,6 +37,11 @@ func ApiAppGenerator(context *answercontext.Context, localFS filesystem.FileSyst
 		{TemplateName: "schema.resolvers.go.tmpl", OutputName: filepath.Join("cmd", context.AppName, "graph", "schema.resolvers.go")},
 	}
 
+	if context.WantDatabase && context.WhatTypeOfDatabase == "MongoDB" {
+		mapping = append(mapping, mappings.MappingType{TemplateName: "bson.go.tmpl", OutputName: filepath.Join("cmd", context.AppName, "internal", "hooks", "bson.go")})
+		mapping = append(mapping, mappings.MappingType{TemplateName: "ObjectID.go.tmpl", OutputName: filepath.Join("cmd", context.AppName, "internal", "scalars", "ObjectID.go")})
+	}
+
 	dir.MakeDirs(localFS, dirs)
 	templates.Execute(localFS, templateFS, "templates/apiapp/*.tmpl", mapping, context)
 }

internal/generators/JSAppGenerator.go

@@ -47,6 +47,11 @@ func JSAppGenerator(context *answercontext.Context, localFS filesystem.FileSyste
 		{TemplateName: "About.js.tmpl", OutputName: filepath.Join("cmd", context.AppName, "app", "static", "views", "About.js")},
 	}
 
+	if context.WantDatabase && context.WhatTypeOfDatabase == "MongoDB" {
+		mapping = append(mapping, mappings.MappingType{TemplateName: "bson.go.tmpl", OutputName: filepath.Join("cmd", context.AppName, "internal", "hooks", "bson.go")})
+		mapping = append(mapping, mappings.MappingType{TemplateName: "ObjectID.go.tmpl", OutputName: filepath.Join("cmd", context.AppName, "internal", "scalars", "ObjectID.go")})
+	}
+
 	dir.MakeDirs(localFS, dirs)
 	templates.Execute(localFS, templateFS, "templates/jsapp/*.tmpl", mapping, context)
 }

templates/apiapp/ObjectID.go.tmpl

@@ -0,0 +1,26 @@
+package scalars
+
+import (
+	"fmt"
+	"io"
+
+	"github.com/99designs/gqlgen/graphql"
+	"github.com/globalsign/mgo/bson"
+)
+
+func MarshalObjectIdScalar(o bson.ObjectId) graphql.Marshaler {
+	return graphql.WriterFunc(func(w io.Writer) {
+		w.Write([]byte(fmt.Sprintf(`"%s"`, o.Hex())))
+	})
+}
+
+func UnmarshalObjectIdScalar(v interface{}) (bson.ObjectId, error) {
+	switch v := v.(type) {
+	case string:
+		return bson.ObjectId(v), nil
+
+	default:
+		return bson.ObjectId(""), fmt.Errorf("%T is not a bson.ObjectId", v)
+	}
+}
+

templates/apiapp/bson.go.tmpl

@@ -0,0 +1,45 @@
+package main
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/99designs/gqlgen/api"
+	"github.com/99designs/gqlgen/codegen/config"
+	"github.com/99designs/gqlgen/plugin/modelgen"
+)
+
+func mutateHook(b *modelgen.ModelBuild) *modelgen.ModelBuild {
+	for _, model := range b.Models {
+		for _, field := range model.Fields {
+			name := field.Name
+			if name == "id" {
+				name = "_id"
+			}
+			field.Tag += ` bson:"` + name + `"`
+		}
+	}
+	return b
+}
+
+func main() {
+	cfg, err := config.LoadConfigFromDefaultLocations()
+	if err != nil {
+		fmt.Fprintln(os.Stderr, "failed to load config", err.Error())
+		os.Exit(2)
+	}
+
+	p := modelgen.Plugin{
+		MutateHook: mutateHook,
+	}
+
+	err = api.Generate(cfg,
+		api.NoPlugins(),
+		api.AddPlugin(&p),
+	)
+	if err != nil {
+		fmt.Fprintln(os.Stderr, err.Error())
+		os.Exit(3)
+	}
+}
+

templates/apiapp/gqlgen.yml.tmpl

@@ -61,4 +61,8 @@ models:
       - github.com/99designs/gqlgen/graphql.Int
       - github.com/99designs/gqlgen/graphql.Int64
       - github.com/99designs/gqlgen/graphql.Int32
+{{- if and .WantDatabase (eq .WhatTypeOfDatabase "MongoDB")}}
+  ObjectID:
+    model: {{.GithubPath}}/cmd/{{.AppName}}/internal/scalars.ObjectIdScalar
+{{- end}}
 

templates/apiapp/main.go.tmpl

@@ -1,3 +1,8 @@
+//go:generate go run github.com/99designs/gqlgen generate
+{{- if and .WantDatabase (eq .WhatTypeOfDatabase "MongoDB")}}
+//go:generate go run internal/hooks/bson.go
+{{- end}}
+
 /*
  * Copyright © {{.Year}}. App Nerds LLC All Rights Reserved
  */

templates/jsapp/ObjectID.go.tmpl

@@ -0,0 +1,26 @@
+package scalars
+
+import (
+	"fmt"
+	"io"
+
+	"github.com/99designs/gqlgen/graphql"
+	"github.com/globalsign/mgo/bson"
+)
+
+func MarshalObjectIdScalar(o bson.ObjectId) graphql.Marshaler {
+	return graphql.WriterFunc(func(w io.Writer) {
+		w.Write([]byte(fmt.Sprintf(`"%s"`, o.Hex())))
+	})
+}
+
+func UnmarshalObjectIdScalar(v interface{}) (bson.ObjectId, error) {
+	switch v := v.(type) {
+	case string:
+		return bson.ObjectId(v), nil
+
+	default:
+		return bson.ObjectId(""), fmt.Errorf("%T is not a bson.ObjectId", v)
+	}
+}
+

templates/jsapp/bson.go.tmpl

@@ -0,0 +1,45 @@
+package main
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/99designs/gqlgen/api"
+	"github.com/99designs/gqlgen/codegen/config"
+	"github.com/99designs/gqlgen/plugin/modelgen"
+)
+
+func mutateHook(b *modelgen.ModelBuild) *modelgen.ModelBuild {
+	for _, model := range b.Models {
+		for _, field := range model.Fields {
+			name := field.Name
+			if name == "id" {
+				name = "_id"
+			}
+			field.Tag += ` bson:"` + name + `"`
+		}
+	}
+	return b
+}
+
+func main() {
+	cfg, err := config.LoadConfigFromDefaultLocations()
+	if err != nil {
+		fmt.Fprintln(os.Stderr, "failed to load config", err.Error())
+		os.Exit(2)
+	}
+
+	p := modelgen.Plugin{
+		MutateHook: mutateHook,
+	}
+
+	err = api.Generate(cfg,
+		api.NoPlugins(),
+		api.AddPlugin(&p),
+	)
+	if err != nil {
+		fmt.Fprintln(os.Stderr, err.Error())
+		os.Exit(3)
+	}
+}
+

templates/jsapp/gqlgen.yml.tmpl

@@ -61,4 +61,8 @@ models:
       - github.com/99designs/gqlgen/graphql.Int
       - github.com/99designs/gqlgen/graphql.Int64
       - github.com/99designs/gqlgen/graphql.Int32
+{{- if and .WantDatabase (eq .WhatTypeOfDatabase "MongoDB")}}
+  ObjectID:
+    model: {{.GithubPath}}/cmd/{{.AppName}}/internal/scalars.ObjectIdScalar
+{{- end}}
 

templates/jsapp/main.go.tmpl

@@ -1,3 +1,8 @@
+//go:generate go run github.com/99designs/gqlgen generate
+{{- if and .WantDatabase (eq .WhatTypeOfDatabase "MongoDB")}}
+//go:generate go run internal/hooks/bson.go
+{{- end}}
+
 /*
  * Copyright © {{.Year}}. App Nerds LLC All Rights Reserved
  */