ts_schema 0.1.14 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e970a98bd2aea544c74c48dac67799e5b713ef56ab8eaa023662f2901973983a
-  data.tar.gz: f259f07ed9fd61a06bf8b0c777df0a50696472534ba48f43737351a8982b3fb5
+  metadata.gz: cf11b26c1cf7b979dc749c5ed2a7cd5c6f580a1d93d715dc684a3dd11cdf0f1d
+  data.tar.gz: 671e245c29ff9511be5db1f83815bfe6553bccea06021eaabe86414e9f034fff
 SHA512:
-  metadata.gz: 30fcd65e0da2df3d5741fe12ed25f84139d3d94b1bc840661a9404105fe5a15e5a0f7a328817283c23592de13ab196d1d31df4398281c6a186d3afcef0d23eb4
-  data.tar.gz: 1a44cc70e7acfb79209f03d2311eb368c1d128832c4e8f586c5ca7f7fa209d5f80641fd8ee57848f6e7b7f9573b1c97e399d372fcb8407b772dd4096b5d1be57
+  metadata.gz: 39fd032dd6b861491e69439e46ca7d0150455795f6c3c85e192d14720f2b9299471ec9c333f8dddeb6e41f3eeae2759e3b6beed2c989c4f51d805340277e14e5
+  data.tar.gz: 938c2fd5b9bc1320e56a5db9444a644473909f876de0c9561d04af0245c2b91e2e9ec5f3feb4d1cd619d490e77317631334c3d458e76ec036080ebd6a736be36

data/lib/generators/templates/ts_schema.rb

@@ -80,6 +80,14 @@ TsSchema.setup do |config|
   #   password: :optional,
   # }
 
+  # Override types for fields by field name. This is globally applied.
+  # Useful for polymorphic associations which would otherwise be of type string, e.g.:
+  #   commentable_type: "'Product'|'Review'" 
+
+  # config.field_type_overrides: {
+
+  # }
+
 
   # Namespace for generated types
   #

data/lib/ts_schema/configuration.rb

@@ -16,6 +16,7 @@ module TsSchema
         "encrypted_password" => :password,
         "password" => :optional,
       },
+      field_type_overrides: {},
       namespace: :schema,
       schema_type: :interface,
       indent: :tab,

data/lib/ts_schema/schema_generator.rb

@@ -8,6 +8,7 @@ module TsSchema
     def initialize(config = nil)
       @config = config || TsSchema::Configuration.new
       @config.field_overrides = @config.field_overrides.stringify_keys
+      @config.field_type_overrides = @config.field_type_overrides.stringify_keys
 			@models = []
 
       Rails.application.eager_load!
@@ -68,12 +69,15 @@ module TsSchema
     end
 
     def map_column_types(model)
-      model.columns.map { |i|
-        next if @config.field_overrides[i.name.to_s] == :omit
-
-        type = @types[i.type.to_s] || @config.default_type
-        name = map_name(i.name)
-        null = i.null
+      model.columns.map { |column|
+        column_name = column.name.to_s
+        next if @config.field_overrides[column_name] == :omit
+
+        type_override = config.field_type_overrides[column_name]
+                 
+        type = type_override ? type_override : @types[column.type.to_s] || @config.default_type
+        name = map_name(column.name)
+        null = column.null
         null = true if @config.field_overrides[name]&.to_s == "optional"
 
         if(enum = model.defined_enums[name])

data/lib/ts_schema/version.rb

@@ -1,3 +1,3 @@
 module TsSchema
-  VERSION = "0.1.14"
+  VERSION = "0.2.1"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ts_schema
 version: !ruby/object:Gem::Version
-  version: 0.1.14
+  version: 0.2.1
 platform: ruby
 authors:
 - Avram Walden
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-07-09 00:00:00.000000000 Z
+date: 2022-07-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails