rails_extend 1.0.0 → 1.0.1

data/lib/rails_extend/core/numeric.rb

@@ -0,0 +1,11 @@
+class Numeric
+
+  def half_round
+    (self * 2).round / 2.0
+  end
+
+  def half_floor
+    (self * 2).floor / 2.0
+  end
+
+end

data/lib/rails_extend/core/pathname.rb

@@ -0,0 +1,15 @@
+class Pathname
+
+  # like present? and presence
+  # if exist? return self
+  def existence
+    self if exist?
+  end
+
+  # xx.html.erb => xx
+  def without_extname
+    suffix = basename.to_s.split('.')[0]
+    dirname.join(suffix)
+  end
+
+end

data/lib/rails_extend/core/string.rb

@@ -0,0 +1,15 @@
+class String
+
+  def integer?
+    Integer(self).is_a?(Integer)
+  rescue ArgumentError, TypeError
+    false
+  end
+
+  def numeric?
+    Float(self).is_a?(Numeric)
+  rescue ArgumentError, TypeError
+    false
+  end
+
+end

data/lib/rails_extend/core/time_format.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+Time::DATE_FORMATS[:datetime] = '%Y-%m-%d %H:%M'
+Time::DATE_FORMATS[:local] = '%Y-%m-%dT%H:%M:%S'
+Time::DATE_FORMATS[:week] = ->(time) {
+  I18n.t('date.day_names')[time.wday]
+}
+Time::DATE_FORMATS[:date] = ->(time) {
+  t = Time.zone.at time
+  t.strftime '%Y-%m-%d'
+}
+Time::DATE_FORMATS[:month] = ->(time) {
+  I18n.t('date.month_names')[time.month]
+}
+Date::DATE_FORMATS[:week] = ->(date) {
+  I18n.t('date.day_names')[date.wday]
+}
+Date::DATE_FORMATS[:month] = ->(date) {
+  I18n.t('date.month_names')[date.month]
+}

data/lib/rails_extend/core.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+# Ruby core extension
+require 'rails_extend/core/hash'
+require 'rails_extend/core/nil'
+require 'rails_extend/core/array'
+require 'rails_extend/core/date'
+require 'rails_extend/core/numeric'
+require 'rails_extend/core/string'
+require 'rails_extend/core/time_format'
+require 'rails_extend/core/module'
+
+# Ruby Strand Library extensions
+require 'rails_extend/core/pathname'
+
+# Rails
+require 'rails_extend/core/duration'

data/lib/rails_extend/engine.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module RailsExtend
+  class Engine < ::Rails::Engine #:nodoc:
+
+    initializer 'rails_extend.quiet_logs' do |app|
+      if RailsExtend.config.quiet_logs.present?
+        app.middleware.insert_before ::Rails::Rack::Logger, ::RailsExtend::QuietLogs
+      end
+    end
+
+  end
+end

data/lib/rails_extend/env.rb

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+module RailsExtend::Env
+  extend self
+
+  def environments
+    Dir.children(Rails.root.join('config', 'environments')).map! { |rb| rb.delete_suffix!('.rb') }
+  end
+
+end

data/lib/rails_extend/generators/named_base.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+require 'rails/generators'
+module RailsExtend::Generators
+  module NamedBase
+
+    def class_name
+      file_name.classify
+    end
+
+    def singular_route_name
+      [class_path[-1], singular_name].join('_')
+    end
+
+    def plural_route_name
+      [class_path[-1], plural_name].join('_')
+    end
+
+  end
+end
+
+Rails::Generators::NamedBase.prepend RailsExtend::Generators::NamedBase

data/lib/rails_extend/generators.rb

@@ -0,0 +1,3 @@
+# frozen_string_literal: true
+
+require 'rails_extend/generators/named_base'

data/lib/rails_extend/models.rb

@@ -0,0 +1,120 @@
+module RailsExtend::Models
+  extend self
+
+  def models_hash(root = ActiveRecord::Base)
+    return @models_hash if defined? @models_hash
+    Zeitwerk::Loader.eager_load_all
+    @models_hash = root.subclasses_tree
+  end
+
+  def models
+    return @models if defined? @models
+    Zeitwerk::Loader.eager_load_all
+    @models = ActiveRecord::Base.descendants
+    @models.reject!(&:abstract_class?)
+    @models
+  end
+
+  def db_tables_hash
+    result = {}
+
+    models.group_by(&->(i) { i.connection.migrations_paths }).each do |migrations_paths, record_classes|
+      result[migrations_paths] = migrate_tables_hash
+    end
+
+    result
+  end
+
+  def tables_hash(root = ActiveRecord::Base, records_hash = models_hash)
+    @tables ||= {}
+
+    records_hash[root].each_key do |node|
+      next if RailsExtend.config.ignore_models.include?(node.to_s)
+
+      unless node.abstract_class?
+        @tables[node.table_name] ||= {}
+        r = @tables[node.table_name]
+        r[:models] ||= []
+        r[:models] << node
+
+        r[:table_exists] = r[:table_exists] || node.table_exists?
+
+        r[:model_attributes] ||= {}
+        r[:model_attributes].reverse_merge! node.migrate_attributes_by_model
+
+        r[:model_references] ||= {}
+        r[:model_references].reverse_merge! node.references_by_model
+
+        r[:indexes] ||= []
+        r[:indexes] |= node.indexes_by_model
+
+        r[:model_defaults] ||= []
+        r[:model_defaults] |= node.attributes_by_default
+
+        r[:belongs_attributes] ||= {}
+        r[:belongs_attributes].reverse_merge! node.attributes_by_belongs
+      end
+
+      tables_hash(node, records_hash[root])
+    end
+
+    @tables
+  end
+
+  def migrate_tables_hash
+    tables = {}
+
+    tables_hash.each do |table_name, cols|
+      db = cols[:models][0].migrate_attributes_by_db
+
+      r = cols.slice(:indexes, :table_exists)
+      r[:add_attributes] = cols[:model_attributes].except *db.keys
+      r[:add_references] = cols[:model_references].except *db.keys
+      r[:timestamps] = ['created_at', 'updated_at'] & r[:add_attributes].keys
+      r[:remove_attributes] = db.except(*cols[:model_attributes].keys, *cols[:belongs_attributes].keys, *cols[:model_defaults])
+
+      tables[table_name] = r unless r[:add_attributes].blank? && r[:add_references].blank? && r[:remove_attributes].blank?
+    end
+
+    tables
+  end
+
+  def migrate_modules_hash
+    @modules = {}
+
+    models.group_by(&:module_parent).each do |module_name, record_classes|
+      new_prefix = (module_name.respond_to?(:table_name_prefix) && module_name.table_name_prefix) || ''
+      old_prefix = (module_name.respond_to?(:old_table_name_prefix) && module_name.old_table_name_prefix) || ''
+
+      record_classes.each do |record_class|
+        unless record_class.table_exists?
+          possible = record_class.table_name.sub(/^#{new_prefix}/, old_prefix)
+          @modules.merge! record_class.table_name => possible if tables.any?(possible)
+        end
+      end
+    end
+
+    arr = @modules.values
+    result = arr.find_all { |e| arr.rindex(e) != arr.index(e) }
+    warn "Please check #{result}"
+
+    @modules
+  end
+
+  def unbound_tables
+    tables - models.map(&:table_name) - ['schema_migrations', 'ar_internal_metadata']
+  end
+
+  def ignore_models
+    models.group_by(&->(i){ i.attributes_to_define_after_schema_loads.size }).transform_values!(&->(i) { i.map(&:to_s) })
+  end
+
+  def tables
+    ActiveRecord::Base.connection.tables
+  end
+
+  def model_names
+    models.map(&:to_s)
+  end
+
+end

data/lib/rails_extend/quiet_logs.rb

@@ -0,0 +1,22 @@
+module RailsExtend
+  class QuietLogs
+
+    def initialize(app)
+      @app = app
+      @assets_regex = %r(\A/{0,2}(#{RailsExtend.config.quiet_logs.join('|')}))
+    end
+
+    def call(env)
+      if env['PATH_INFO'] =~ @assets_regex
+        Rails.logger.debug "Silenced: #{env['PATH_INFO']}"
+        Rails.logger.silence { @app.call(env) }
+      else
+        unless Rails.env.development?
+          Rails.logger.debug "\e[33m #{'- ~ ' * 40}- \e[0m"
+        end
+        @app.call(env)
+      end
+    end
+
+  end
+end

data/lib/rails_extend/routes.rb

@@ -0,0 +1,81 @@
+# frozen_string_literal: true
+
+module RailsExtend::Routes
+  extend self
+
+  def verbs(controller, action)
+    routes_wrapper.select { |i| i[:controller] == controller.to_s && i[:action] == action.to_s }.map { |i| i[:verb] }.uniq
+  end
+
+  def actions(cached = true)
+    return @actions if cached && defined? @actions
+
+    @actions = routes_wrapper(cached).group_by(&->(i){ i[:business] }).transform_values! do |businesses|
+      businesses.group_by(&->(i){ i[:namespace] }).transform_values! do |namespaces|
+        namespaces.group_by(&->(i){ i[:controller] }).transform_values! do |controllers|
+          controllers.each_with_object({}) { |i, h| h.merge! i[:action] => i }
+        end
+      end
+    end
+  end
+
+  def controllers(cached = true)
+    return @controllers if cached && defined? @controllers
+
+    @controllers = routes_wrapper(cached).group_by(&->(i){ i[:controller] }).transform_values! do |controllers|
+      controllers.each_with_object({}) { |i, h| h.merge! i[:action] => i }
+    end
+  end
+
+  def controller_paths(cached = true)
+    return @controller_paths if cached && defined? @controller_paths
+
+    @controller_paths = routes_wrapper(cached).group_by(&->(i){ i[:controller] }).transform_values! do |controllers|
+      controllers[0].slice(:business, :namespace)
+    end
+  end
+
+  def businesses(cached = true)
+    return @businesses if cached && defined? @businesses
+
+    @businesses = routes_wrapper(cached).group_by(&->(i){ i[:business] })
+  end
+
+  def namespaces(cached = true)
+    return @namespaces if cached && defined? @namespaces
+
+    @namespaces = routes_wrapper(cached).group_by(&->(i){ i[:namespace] })
+  end
+
+  def routes_wrapper(cached = true)
+    return @routes_wrapper if cached && defined?(@routes_wrapper)
+
+    @routes_wrapper = Rails.application.routes.routes.map do |route|
+      next if (route.defaults[:controller].blank? || route.defaults[:action].blank?)
+      detail(route)
+    end.compact
+  end
+
+  def blank_routes_wrapper(cached = true)
+    return @blank_routes_wrapper if cached && defined?(@blank_routes_wrapper)
+
+    @blank_routes_wrapper = Rails.application.routes.routes.map do |route|
+      next unless (route.defaults[:controller].blank? || route.defaults[:action].blank?)
+      detail(route)
+    end.compact
+  end
+
+  private
+  def detail(route)
+    {
+      verb: route.verb,
+      path: route.path.spec.to_s,
+      namespace: route.defaults[:namespace].to_s,
+      business: route.defaults[:business].to_s,
+      controller: route.defaults[:controller],
+      action: route.defaults[:action],
+      required_parts: route.required_parts
+    }
+  end
+
+end

data/lib/rails_extend/type/i18n.rb

@@ -0,0 +1,23 @@
+module ActiveRecord::Type
+  class I18n < Json
+
+    def type
+      :i18n
+    end
+
+    def deserialize(value)
+      if super && super.respond_to?(:[])
+        super[::I18n.locale.to_s]
+      else
+        value
+      end
+    end
+
+    def migrate_type
+      :json
+    end
+
+  end
+end
+
+ActiveRecord::Type.register(:i18n, ActiveRecord::Type::I18n)

data/lib/rails_extend/type/taxon.rb

@@ -0,0 +1,15 @@
+module ActiveRecord::Type
+  class Taxon < Json
+
+    def input_type
+      :taxon
+    end
+
+    def migrate_type
+      :json
+    end
+
+  end
+end
+
+ActiveRecord::Type.register(:taxon, ActiveRecord::Type::Taxon)

data/lib/rails_extend/type.rb

@@ -0,0 +1,4 @@
+# frozen_string_literal: true
+
+require 'rails_extend/type/i18n'
+require 'rails_extend/type/taxon'

data/lib/rails_extend.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+require 'rails_extend/config'
+require 'rails_extend/engine'
+
+require 'rails_extend/core'
+
+require 'rails_extend/active_model'
+require 'rails_extend/active_record'
+require 'rails_extend/active_storage'
+require 'rails_extend/type'  # 支持的 attribute type 扩展
+require 'rails_extend/generators'
+
+# Meta for Rails
+require 'rails_extend/routes'
+require 'rails_extend/models'
+require 'rails_extend/env'
+
+require 'rails_extend/quiet_logs'

data/lib/templates/test_unit/model/fixtures.yml.tt

@@ -0,0 +1,15 @@
+<% ['one', 'two'].each do |name| -%>
+<%= name %>:
+<% attributes.each do |attribute| -%>
+  <%- if attribute.password_digest? -%>
+  password_digest: <%%= BCrypt::Password.create('secret') %>
+  <%- elsif attribute.reference? -%>
+  <%= yaml_key_value(attribute.column_name.delete_suffix("_id"), attribute.default || name) %>
+  <%- elsif !attribute.virtual? -%>
+  <%= yaml_key_value(attribute.column_name, attribute.default) %>
+  <%- end -%>
+  <%- if attribute.polymorphic? -%>
+  <%= yaml_key_value("#{attribute.name}_type", attribute.human_name) %>
+  <%- end -%>
+<% end -%>
+<% end -%>

data/lib/templates/test_unit/model/unit_test.rb.tt

@@ -0,0 +1,7 @@
+require 'test_helper'
+
+<% module_namespacing do -%>
+class <%= class_name %>Test < ActiveSupport::TestCase
+
+end
+<% end -%>

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rails_extend
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.0.1
 platform: ruby
 authors:
 - qinmingyuan
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-09-06 00:00:00.000000000 Z
+date: 2021-10-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -16,14 +16,14 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '6.1'
+        version: '5.0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '6.1'
+        version: '5.0'
 description: Common utils for Rails Application
 email:
 - mingyuan0715@foxmail.com
@@ -31,7 +31,58 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- lib/active_model/type_value.rb
+- LICENSE
+- README.md
+- config/locales/en.datetime.yml
+- config/locales/zh.datetime.yml
+- config/locales/zh.support.yml
+- lib/generators/rails_extend/ignore_generator.rb
+- lib/generators/rails_extend/migrations_generator.rb
+- lib/generators/rails_extend/remove_table_generator.rb
+- lib/generators/rails_extend/rename_module_generator.rb
+- lib/generators/rails_extend/templates/README.md
+- lib/generators/rails_extend/templates/initializer.rb.tt
+- lib/generators/rails_extend/templates/migration.rb.tt
+- lib/generators/rails_extend/templates/remove_table.rb.tt
+- lib/generators/rails_extend/templates/rename_module.rb.tt
+- lib/rails_extend.rb
+- lib/rails_extend/active_model.rb
+- lib/rails_extend/active_model/type_value.rb
+- lib/rails_extend/active_record.rb
+- lib/rails_extend/active_record/enum.rb
+- lib/rails_extend/active_record/extend.rb
+- lib/rails_extend/active_record/i18n.rb
+- lib/rails_extend/active_record/include.rb
+- lib/rails_extend/active_record/taxon.rb
+- lib/rails_extend/active_record/translation.rb
+- lib/rails_extend/active_storage.rb
+- lib/rails_extend/active_storage/attachment.rb
+- lib/rails_extend/active_storage/blob_prepend.rb
+- lib/rails_extend/active_storage/variant.rb
+- lib/rails_extend/config.rb
+- lib/rails_extend/core.rb
+- lib/rails_extend/core/array.rb
+- lib/rails_extend/core/date.rb
+- lib/rails_extend/core/duration.rb
+- lib/rails_extend/core/hash.rb
+- lib/rails_extend/core/module.rb
+- lib/rails_extend/core/nil.rb
+- lib/rails_extend/core/numeric.rb
+- lib/rails_extend/core/pathname.rb
+- lib/rails_extend/core/string.rb
+- lib/rails_extend/core/time_format.rb
+- lib/rails_extend/engine.rb
+- lib/rails_extend/env.rb
+- lib/rails_extend/generators.rb
+- lib/rails_extend/generators/named_base.rb
+- lib/rails_extend/models.rb
+- lib/rails_extend/quiet_logs.rb
+- lib/rails_extend/routes.rb
+- lib/rails_extend/type.rb
+- lib/rails_extend/type/i18n.rb
+- lib/rails_extend/type/taxon.rb
+- lib/templates/test_unit/model/fixtures.yml.tt
+- lib/templates/test_unit/model/unit_test.rb.tt
 homepage: https://github.com/work-design/rails_extend
 licenses:
 - MIT

data/lib/active_model/type_value.rb

@@ -1,20 +0,0 @@
-require 'active_model/type/value'
-module RailsCom
-  module ActiveModel
-    module TypeValue
-      attr_reader :options
-
-      def initialize(precision: nil, limit: nil, scale: nil, **options)
-        @options = options
-        super(precision: precision, limit: scale, scale: limit)
-      end
-
-      def input_type
-        type
-      end
-
-    end
-  end
-end
-
-ActiveModel::Type::Value.prepend RailsCom::ActiveModel::TypeValue