feat(config): allow generating models from information_schema tables

Author: sgtsquiggs

docs/reference/config.md

@@ -179,6 +179,8 @@ The `gen` mapping supports the following keys:
   - `camel` for camelCase, `pascal` for PascalCase, `snake` for snake_case or `none` to use the column name in the DB. Defaults to `none`.
 - `omit_unused_structs`:
   - If `true`, sqlc won't generate table and enum structs that aren't used in queries for a given package. Defaults to `false`.
+- `omit_catalog_schema`:
+  - If `true`, sqlc won't generate structs for tables and enums in the `pg_catalog` and `information_schema` catalog schemas. Set to `false` to generate models from these schemas. Defaults to `true`.
 - `output_batch_file_name`:
   - Customize the name of the batch file. Defaults to `batch.go`.
 - `output_db_file_name`:
@@ -396,6 +398,7 @@ packages:
     build_tags: "some_tag"
     json_tags_case_style: "camel"
     omit_unused_structs: false
+    omit_catalog_schema: true
     output_batch_file_name: "batch.go"
     output_db_file_name: "db.go"
     output_models_file_name: "models.go"
@@ -458,6 +461,8 @@ Each mapping in the `packages` collection has the following keys:
   - `camel` for camelCase, `pascal` for PascalCase, `snake` for snake_case or `none` to use the column name in the DB. Defaults to `none`.
 - `omit_unused_structs`:
   - If `true`, sqlc won't generate table and enum structs that aren't used in queries for a given package. Defaults to `false`.
+- `omit_catalog_schema`:
+  - If `true`, sqlc won't generate structs for tables and enums in the `pg_catalog` and `information_schema` catalog schemas. Set to `false` to generate models from these schemas. Defaults to `true`.
 - `output_batch_file_name`:
   - Customize the name of the batch file. Defaults to `batch.go`.
 - `output_db_file_name`:

internal/codegen/golang/opts/options.go

@@ -49,6 +49,7 @@ type Options struct {
 	QueryParameterLimit          *int32            `json:"query_parameter_limit,omitempty" yaml:"query_parameter_limit"`
 	OmitSqlcVersion              bool              `json:"omit_sqlc_version,omitempty" yaml:"omit_sqlc_version"`
 	OmitUnusedStructs            bool              `json:"omit_unused_structs,omitempty" yaml:"omit_unused_structs"`
+	OmitCatalogSchema            *bool             `json:"omit_catalog_schema,omitempty" yaml:"omit_catalog_schema"`
 	BuildTags                    string            `json:"build_tags,omitempty" yaml:"build_tags"`
 	Initialisms                  *[]string         `json:"initialisms,omitempty" yaml:"initialisms"`
 
@@ -185,6 +186,16 @@ func (o *Options) ModelsEmitEnabled() bool {
 	return *o.OutputModelsEmit
 }
 
+// OmitCatalogSchemaEnabled reports whether the pg_catalog and
+// information_schema catalog schemas should be excluded when generating
+// models. Defaults to true when the option is unset.
+func (o *Options) OmitCatalogSchemaEnabled() bool {
+	if o.OmitCatalogSchema == nil {
+		return true
+	}
+	return *o.OmitCatalogSchema
+}
+
 // ModelsImportAlias is the fixed Go import alias used for the models
 // package in query files. Using a constant alias keeps the type qualifier
 // consistent regardless of how the user names the actual package.

internal/codegen/golang/result.go

@@ -16,7 +16,7 @@ import (
 func buildEnums(req *plugin.GenerateRequest, options *opts.Options) []Enum {
 	var enums []Enum
 	for _, schema := range req.Catalog.Schemas {
-		if schema.Name == "pg_catalog" || schema.Name == "information_schema" {
+		if options.OmitCatalogSchemaEnabled() && (schema.Name == "pg_catalog" || schema.Name == "information_schema") {
 			continue
 		}
 		for _, enum := range schema.Enums {
@@ -63,7 +63,7 @@ func buildEnums(req *plugin.GenerateRequest, options *opts.Options) []Enum {
 func buildStructs(req *plugin.GenerateRequest, options *opts.Options) []Struct {
 	var structs []Struct
 	for _, schema := range req.Catalog.Schemas {
-		if schema.Name == "pg_catalog" || schema.Name == "information_schema" {
+		if options.OmitCatalogSchemaEnabled() && (schema.Name == "pg_catalog" || schema.Name == "information_schema") {
 			continue
 		}
 		for _, table := range schema.Tables {

internal/codegen/golang/result_test.go

@@ -3,10 +3,75 @@ package golang
 import (
 	"testing"
 
+	"github.com/sqlc-dev/sqlc/internal/codegen/golang/opts"
 	"github.com/sqlc-dev/sqlc/internal/metadata"
 	"github.com/sqlc-dev/sqlc/internal/plugin"
 )
 
+func boolPtr(b bool) *bool { return &b }
+
+func catalogSchemaRequest() *plugin.GenerateRequest {
+	col := func(name, typ string) *plugin.Column {
+		return &plugin.Column{Name: name, Type: &plugin.Identifier{Name: typ}}
+	}
+	table := func(schema, name string, cols ...*plugin.Column) *plugin.Schema {
+		return &plugin.Schema{
+			Name: schema,
+			Tables: []*plugin.Table{
+				{Rel: &plugin.Identifier{Schema: schema, Name: name}, Columns: cols},
+			},
+		}
+	}
+	return &plugin.GenerateRequest{
+		Settings: &plugin.Settings{Engine: "postgresql"},
+		Catalog: &plugin.Catalog{
+			DefaultSchema: "public",
+			Schemas: []*plugin.Schema{
+				table("pg_catalog", "pg_class", col("oid", "oid")),
+				table("information_schema", "tables", col("table_name", "text")),
+				table("public", "users", col("id", "int4")),
+			},
+		},
+	}
+}
+
+func TestBuildStructs_OmitCatalogSchema(t *testing.T) {
+	req := catalogSchemaRequest()
+
+	t.Run("unset (default) skips pg_catalog and information_schema", func(t *testing.T) {
+		structs := buildStructs(req, &opts.Options{})
+		for _, s := range structs {
+			if s.Table != nil && (s.Table.Schema == "pg_catalog" || s.Table.Schema == "information_schema") {
+				t.Errorf("unexpected catalog struct: %s.%s", s.Table.Schema, s.Table.Name)
+			}
+		}
+	})
+
+	t.Run("omit=true skips pg_catalog and information_schema", func(t *testing.T) {
+		structs := buildStructs(req, &opts.Options{OmitCatalogSchema: boolPtr(true)})
+		for _, s := range structs {
+			if s.Table != nil && (s.Table.Schema == "pg_catalog" || s.Table.Schema == "information_schema") {
+				t.Errorf("unexpected catalog struct: %s.%s", s.Table.Schema, s.Table.Name)
+			}
+		}
+	})
+
+	t.Run("omit=false includes pg_catalog and information_schema", func(t *testing.T) {
+		structs := buildStructs(req, &opts.Options{OmitCatalogSchema: boolPtr(false)})
+		schemas := make(map[string]bool)
+		for _, s := range structs {
+			if s.Table != nil {
+				schemas[s.Table.Schema] = true
+			}
+		}
+		for _, want := range []string{"pg_catalog", "information_schema", "public"} {
+			if !schemas[want] {
+				t.Errorf("expected structs for schema %q, got none", want)
+			}
+		}
+	})
+}
+
 func TestPutOutColumns_ForZeroColumns(t *testing.T) {
 	tests := []struct {
 		cmd  string

internal/config/v_one.go

@@ -62,6 +62,7 @@ type v1PackageSettings struct {
 	QueryParameterLimit          *int32            `json:"query_parameter_limit,omitempty" yaml:"query_parameter_limit"`
 	OmitSqlcVersion              bool              `json:"omit_sqlc_version,omitempty" yaml:"omit_sqlc_version"`
 	OmitUnusedStructs            bool              `json:"omit_unused_structs,omitempty" yaml:"omit_unused_structs"`
+	OmitCatalogSchema            *bool             `json:"omit_catalog_schema,omitempty" yaml:"omit_catalog_schema"`
 	Rules                        []string          `json:"rules" yaml:"rules"`
 	BuildTags                    string            `json:"build_tags,omitempty" yaml:"build_tags"`
 }
@@ -177,6 +178,7 @@ func (c *V1GenerateSettings) Translate() Config {
 					QueryParameterLimit:          pkg.QueryParameterLimit,
 					OmitSqlcVersion:              pkg.OmitSqlcVersion,
 					OmitUnusedStructs:            pkg.OmitUnusedStructs,
+					OmitCatalogSchema:            pkg.OmitCatalogSchema,
 					BuildTags:                    pkg.BuildTags,
 				},
 			},

internal/config/v_one.json

@@ -263,6 +263,9 @@
                     "omit_unused_structs": {
                         "type": "boolean"
                     },
+                    "omit_catalog_schema": {
+                        "type": "boolean"
+                    },
                     "rules": {
                         "type": "array",
                         "items": {

internal/config/v_two.json

@@ -280,6 +280,9 @@
                                 },
                                 "omit_unused_structs": {
                                     "type": "boolean"
+                                },
+                                "omit_catalog_schema": {
+                                    "type": "boolean"
                                 }
                             },
                             "json": {