Add in Enum support and a Helper codegen to build enum type mapping for Npgsql.

Author: Rachel Clark

internal/gen.go

@@ -58,7 +58,7 @@ func Generate(ctx context.Context, req *plugin.Request) (*plugin.Response, error
 	}
 
 	log.Println("Beginning generation with config: ", conf)
-	//enums := core.BuildEnums(req)
+	enums := core.BuildEnums(req)
 	classes := core.BuildClasses(req)
 	queries, err := core.BuildQueries(req, conf, classes)
 	log.Println("queries built: ", queries)
@@ -72,6 +72,7 @@ func Generate(ctx context.Context, req *plugin.Request) (*plugin.Response, error
 		CsGenVersion: version,
 		Namespace:    conf.Namespace,
 		Classes:      classes,
+		Enums:        enums,
 	}
 
 	funcMap := template.FuncMap{
@@ -114,6 +115,10 @@ func Generate(ctx context.Context, req *plugin.Request) (*plugin.Response, error
 		return nil, err
 	}
 
+	if err := execute("DbHelper", "helpersFile"); err != nil {
+		return nil, err
+	}
+
 	files := map[string]struct{}{}
 	for _, gq := range queries {
 		files[gq.SourceName] = struct{}{}

internal/templates/npgsql/helpers.tmpl

@@ -0,0 +1,22 @@
+{{define "helpersFile" }}// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//     sqlc {{ .SqlcVersion }}
+//     sqlc-gen-cs {{ .CsGenVersion }}
+using Npgsql;
+
+namespace {{ .Namespace }}.helpers;
+
+public static class DbHelpers {
+    /// <summary>
+    /// RegisterEnumMappings is REQUIRED to be used for enums to work properly.
+    /// If you have any enums, remember to use this function when building your data source.
+    /// </summary>
+    public static NpgsqlDataSourceBuilder RegisterEnumMappings(this NpgsqlDataSourceBuilder dbBuilder) {
+        {{- range .Enums }}
+        dbBuilder.MapEnum<{{.Name}}>();
+        {{ end }}
+
+        return dbBuilder;
+    }  
+}
+{{ end }}

internal/templates/npgsql/models.tmpl

@@ -4,6 +4,14 @@
 //     sqlc-gen-cs {{ .CsGenVersion }}
 
 namespace {{ .Namespace }};
+{{ range .Enums }}
+{{ if .Comment }}{{ comment .Comment }}{{ end }}
+public enum {{.Name}} { {{- range .Members }}
+    {{ .Name }},
+    {{ end }}
+}
+{{ end }}
+
 
 {{range .Classes}}
 {{if .Comment}}{{comment.Comment}}{{end}}