ts_schema 0.1.9 → 0.1.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 98fda5d599b1c67632f071da87640ec19167fbc90c0ef50f3090ef1deb19a394
-  data.tar.gz: 43145f323d86ce2acca7cf45687e0ac7b35e38e2fa0b5216f180fd920e7bd08f
+  metadata.gz: 8602c4a7ca10962ac35fd7c64a18d0669ed3f7d1ecdfdf7bc9c29a258bff754a
+  data.tar.gz: 322d779a986f2edf18d53829a6a8669326732be7aa4080567d2f4a1ecb1ebac0
 SHA512:
-  metadata.gz: 20d2257583b6fd2c219ba6740822d7e7b05df7a49e165b9d8dd7d370a326e33cdf21f4c5920afea1fd76ffd428b5f6796ed2d7dfd0e484a7ea6f37e9248298f5
-  data.tar.gz: e7640bf87f7bd2243a1bab5d3b5bc15188f5151f95236357571141ceef72f4ac95ed8e5fc2a79ac3aa90da4256b7d401ceedd48ebb89188650929031177da301
+  metadata.gz: 88fd048bbffe0eb93ac910bdcaf456a35e4ded1bf934dab9330d5ad1dba4d57ad3106c07a549a92555f312fe4d713d1b1b4b8c7d29d3909d85a7062df96e6426
+  data.tar.gz: 4c778296ba8de708bf9539bdcd163d79370c0477600d2bdbfd98a0e5c0355d3e02872728452be964922032cc880b3ac92ee270f40269e251734fb48ebfb74bcb

data/Rakefile

@@ -1,2 +1,3 @@
 require "bundler/setup"
 require "bundler/gem_tasks"
+require "ts_schema"

data/lib/generators/templates/ts_schema.rb

@@ -1,57 +1,89 @@
+# Default type mappings:
+#
+# string:                 string
+# text:                   string
+# integer:                number
+# enum:                   number
+# bigint:                 number
+# float:                  number
+# decimal:                number
+# json:                   Record<string, any>
+# jsonb:                  Record<string, any>
+# binary:                 string
+# boolean:                boolean
+# date:                   string
+# datetime:               string
+# timestamp:              string
+# datetime_with_timezone: string
+# inet:                   string
+# cidr:                   string
+# macaddr:                string
+
 TsSchema.setup do |config|
-  # Options: camel|snake|pascal
+  # Case options: camel|snake|pascal
+  #
   # config.case = :camel
 
+
   # Customize output path and file name
+  #
   # config.output = Rails.root.join('app', 'assets', 'javascripts', 'schema.d.ts')
 
-  # Whether to generate the schema file on migrations and rollbacks
+
+  # Whether to generate the schema file after running migrations
+  #
   # config.auto_generate = true
 
+
   # Add custom type mappings or overrides
-	#
-	# Default type mappings:
-	#
-	# string:                 string
-	# text:                   string
-	# integer:                number
-	# enum:                   number
-	# bigint:                 number
-	# float:                  number
-	# decimal:                number
-	# json:                   Record<string, any>
-	# jsonb:                  Record<string, any>
-	# binary:                 string
-	# boolean:                boolean
-	# date:                   string
-	# datetime:               string
-	# timestamp:              string
-	# datetime_with_timezone: string
-	# inet:                   string
-	# cidr:                   string
-	# macaddr:                string
-	#
+  #
   # config.custom_types = {
   #   
   # }
 
+
   # Default type for unrecognized types
+  #
   # config.default_type = :string
 
-  # Whether to generate types for associations
+
+  # Whether to generate types for associated models
+  #
   # config.include_associated = true
 
-	# Additional models to map which don't have a model file
-	# config.additional_models = [
-	#	
-	# ]
 
-  # Namespace
+  # Parent classes of models to generate (as strings or symbols)
+  # Only classes inheriting from those in this list will have types generated
+  #
+  # config.parent_classes = [
+  #	  "ApplicationRecord",
+  # ]
+
+
+  # Additional models to generate schema from, such as those added by other gems 
+  #  which don't have a model file. (as strings or symbols)
+  #
+  # config.additional_models = [
+  #	
+  # ]
+
+
+  # Namespace for generated types
+  #
   # config.namespace = :schema
 
-  # Options: tab|space
+
+  # Output schema as types or interfaces: type|interface
+  #
+  # config.schema_type = :interface
+
+
+  # Indentation options: tab|space
+  #
   # config.indent = :tab
 
+
   # If indent is spaces, specify how many
+  #
   # config.spaces = 2
 end

data/lib/tasks/ts_schema_tasks.rake

@@ -1,8 +1,9 @@
 require "rake"
+require "ts_schema"
 
 namespace :ts_schema do
   desc "Generates a schema file in the default javascripts location, or the location specified in the ts_config initializer options"
-  task :generate do
+  task generate: :environment do
     TsSchema.output_file
   end
 end
@@ -12,19 +13,11 @@ namespace :db do
     TsSchema.output_file if TsSchema.configuration.auto_generate
   end
 
-  task migrate: :environment do
-    auto_generate_and_save_file
+  task :migrate do
+    at_exit { auto_generate_and_save_file }
   end
 
-  task rollback: :environment do
-    auto_generate_and_save_file
-  end
-
-  task reset: :environment do
-    auto_generate_and_save_file
-  end
-
-  task setup: :environment do
-    auto_generate_and_save_file
+  task :setup do
+    at_exit { auto_generate_and_save_file }
   end
 end

data/lib/ts_schema/configuration.rb

@@ -11,8 +11,10 @@ module TsSchema
       default_type: :string,
       include_associated: true,
       namespace: :schema,
+			schema_type: :interface,
       indent: :tab,
       spaces: 2,
+			parent_classes: ["ApplicationRecord"],
 			additional_models: []
     }
 
@@ -37,6 +39,5 @@ module TsSchema
       end
       self
     end
-
   end
 end

data/lib/ts_schema/conversion_table.yml

@@ -2,7 +2,6 @@ string: string
 text: string
 integer: number
 enum: number
-bigint: number
 float: number
 decimal: number
 json: Record<string, any>
@@ -12,7 +11,6 @@ boolean: boolean
 date: string
 datetime: string
 timestamp: string
-datetime_with_timezone: string
 inet: string
 cidr: string
 macaddr: string

data/lib/ts_schema/railtie.rb

@@ -1,4 +1,7 @@
+require "rails"
+
 module TsSchema
   class Railtie < ::Rails::Railtie
+    load "tasks/ts_schema_tasks.rake"
   end
 end

data/lib/ts_schema/schema_generator.rb

@@ -2,56 +2,74 @@ require 'fileutils'
 
 module TsSchema
   class SchemaGenerator
-    def initialize
-      Rails.application.eager_load!
-      @models = ApplicationRecord.send(:subclasses)
-			@models.concat(TsSchema.configuration.additional_models) unless TsSchema.configuration.additional_models.empty?
-      @types = TsSchema.configuration.types.merge(TsSchema.configuration.custom_types || {})
-    end
-
-    def generate
-      generate_typescript
-    end
-
-    def output_file
-      path = TsSchema.configuration.output
-      FileUtils.mkdir_p(File.dirname(path))
+    attr_accessor :config
+    attr_accessor :models
 
-			content = generate
-      return if File.exist?(path) && File.read(path) == content
+    def initialize(config = nil)
+      @config = config || TsSchema::Configuration.new
+			@models = []
 
-      File.open(path, 'w') do |f|
-        f.write content
+      Rails.application.eager_load!
+			@config.parent_classes.each do |parent|
+				@models.concat(get_subclasses(parent.to_s.constantize))
+			end
+      unless @config.additional_models.empty?
+        @models.concat(@config.additional_models.map do |m| 
+          m.to_s.constantize
+        end)
       end
+      @types = @config.types.stringify_keys.merge(@config.custom_types.stringify_keys || {})
     end
 
-    private
+    def get_subclasses(model)
+      models = model.send(:subclasses)
+      unless nested_empty?(models)
+        models.concat(models.flat_map do |m|
+          get_subclasses(m)
+        end)
+      end
+      models
+    end
 
-    def generate_typescript
+    def generate
       type_template = ""
+			
       @models.each do |model|
         columns = map_column_types(model)
-        columns.concat(map_associations(model)) if TsSchema.configuration.include_associated
+        columns.concat(map_associations(model)) if @config.include_associated
         
         type_template += <<~TYPESCRIPT
-          interface #{model.model_name.param_key.camelize} {
+          #{@config.schema_type} #{model.model_name.param_key.camelize} #{@config.schema_type.to_sym == :type ? "= " : ""}{
           #{columns.map { |column| "#{indent_as_str}#{column_name_cased(column[:name])}: #{column[:ts_type]};" }.join("\n")}
           }\n
         TYPESCRIPT
       end
+
       type_template = <<~TPL
-        declare namespace #{TsSchema.configuration.namespace} {
+        declare namespace #{@config.namespace} {
         #{indent_wrapper(type_template)}
         }
       TPL
     end
 
+    def output_file
+      path = @config.output
+      FileUtils.mkdir_p(File.dirname(path))
+
+      content = generate
+      return if File.exist?(path) && File.read(path) == content
+
+      File.open(path, 'w') do |f|
+        f.write content
+      end
+    end
+
     def map_column_types(model)
       model.columns.map { |i|
-        type = @types[i.type.to_s] || TsSchema.configuration.default_type
+        type = @types[i.type.to_s] || @config.default_type
 
         if(enum = model.defined_enums[i.name])
-          type = enum.keys.map { |k| "'#{k}'" }.join(" | ")
+          type = enum.keys.map { |k| "'#{k}'" }.join("|")
         end
     
         {
@@ -78,8 +96,14 @@ module TsSchema
       end
     end
 
+    private
+
+    def nested_empty?(arr)
+      arr.is_a?(Array) && arr.flatten.empty?
+    end
+
     def column_name_cased(name)
-      case TsSchema.configuration.case.to_sym
+      case @config.case.to_sym
       when :camel
         name.camelize(:lower)
       when :pascal
@@ -90,18 +114,18 @@ module TsSchema
     end
 
     def indent_as_str
-      case TsSchema.configuration.indent.to_sym
+      case @config.indent.to_sym
       when :space || :spaces
-        "".rjust(TsSchema.configuration.spaces, " ")
+        "".rjust(@config.spaces, " ")
       else
         "\t"
       end
     end
 
     def indent_wrapper(str)
-      case TsSchema.configuration.indent.to_sym
+      case @config.indent.to_sym
       when :space || :spaces
-        str.indent(TsSchema.configuration.spaces)
+        str.indent(@config.spaces)
       else
         str.indent(1, "\t")
       end

data/lib/ts_schema/version.rb

@@ -1,3 +1,3 @@
 module TsSchema
-  VERSION = "0.1.9"
+  VERSION = "0.1.12"
 end

data/lib/ts_schema.rb

@@ -4,6 +4,7 @@ require "ts_schema/railtie"
 require "ts_schema/configuration"
 require "ts_schema/schema_generator"
 
+
 module TsSchema
   class << self
     def setup(&block)
@@ -15,11 +16,19 @@ module TsSchema
     end
 
     def generate
-      SchemaGenerator.new.generate
+      if ActiveRecord::Base.connection.migration_context.needs_migration?
+        puts "Aborting: There are pending migrations"
+      else 
+        SchemaGenerator.new(@configuration).generate
+      end
     end
 
 		def output_file
-			SchemaGenerator.new.output_file
+      if ActiveRecord::Base.connection.migration_context.needs_migration?
+        puts "Aborting: There are pending migrations"
+      else 
+        SchemaGenerator.new(@configuration).output_file
+      end
 		end
   end
 end

metadata

@@ -1,17 +1,17 @@
 --- !ruby/object:Gem::Specification
 name: ts_schema
 version: !ruby/object:Gem::Version
-  version: 0.1.9
+  version: 0.1.12
 platform: ruby
 authors:
 - Avram Walden
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-04-04 00:00:00.000000000 Z
+date: 2022-04-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: railties
+  name: rails
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -24,6 +24,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: 6.0.0
+- !ruby/object:Gem::Dependency
+  name: railties
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4'
 description: ''
 email:
 - aviemet@gmail.com
@@ -34,7 +48,6 @@ files:
 - MIT-LICENSE
 - README.md
 - Rakefile
-- lib/generators/generate_generator.rb
 - lib/generators/install_generator.rb
 - lib/generators/templates/ts_schema.rb
 - lib/tasks/ts_schema_tasks.rake

data/lib/generators/generate_generator.rb

@@ -1,17 +0,0 @@
-# frozen_string_literal: true
-
-require 'rails/generators'
-require 'rails/generators/base'
-
-module TsSchema
-  module Generators
-    class GenerateGenerator < Rails::Generators::Base
-
-      desc "Generates a schema.d.ts file"
-
-      def generate_schema
-        create_file TsSchema.configuration.output, TsSchema.generate
-      end
-    end
-  end
-end