dynamic-active-model 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0f4acad4070f218fe60381f8dd22b442ca152fa3528fc309d35f72e6a755f238
-  data.tar.gz: 4c15e0e3f7046514a16053d757497be8a16617c96c4e6e5f3b473a4b0a642dc2
+  metadata.gz: 67d4e71c097cafd08ed7e9c3ed2ffd4d8b22b5787fd2f6ab498a709b75de5215
+  data.tar.gz: a5cad0b2293d861f93bdf9d6c8fbc983ca9c4af1ab9bb60c22216c074e2b01c2
 SHA512:
-  metadata.gz: edf688fbec02059b722d397f47fb4a2872972db6e5855e4a62641d91f57d4f07fc0d58240ca938f9e38c0277a84498e2d7b4ae2857de5c9ee81856b253ca9b00
-  data.tar.gz: 766b08e10f4a11c25329a21d646f887a296c9b04553db92c7a3bffcb703c837034cfea3b53ae0780d472b00471aa6400e37d1b19c1514ee1f850eaf14a382801
+  metadata.gz: 5939469d103fb9c6a6d71ea987e6b0f8fe9c117d1014ca5d752c25307c1d5ff586d64cebae3571cc1eb276e9cfd760e577bae2066c9812223055c1f309b0cb50
+  data.tar.gz: 422a33565554689d2055122cc45234e7fd4c7f1f766f018b4fd080df6c86051cc79d950fec706702f489bd225793e84b3dc1d4cc7b66ae2dd2667fe507e0f5f1

data/bin/dynamic-db-explorer

@@ -0,0 +1,102 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require 'dynamic-active-model'
+require 'optparse'
+require 'yaml'
+
+DEFAULT_CONFIG_FILE = "#{ENV['HOME']}/.dynamic-db-explorer.yml"
+
+options = {
+  base_module: Object,
+  connection_options: {},
+  skip_tables: [],
+  relationships: {},
+  config_file: nil,
+  config_section: nil,
+  class_files_dir: nil
+}
+
+# rubocop:disable Metrics/BlockLength
+OptionParser.new do |opts|
+  opts.banner = 'Usage: dynamic-db-explorer'
+
+  opts.on('-h', '--help', 'Prints this help') do
+    puts opts
+    exit
+  end
+
+  opts.on('--username USERMNAME', 'DB username') do |v|
+    options[:connection_options][:username] = v
+  end
+
+  opts.on('--host HOST', 'DB host') do |v|
+    options[:connection_options][:host] = v
+  end
+
+  opts.on('--port PORT', 'DB port') do |v|
+    options[:connection_options][:port] = v
+  end
+
+  opts.on('--password PASSWORD', 'DB password') do |v|
+    options[:connection_options][:password] = v
+  end
+
+  opts.on('--adapter ADAPTER', 'DB adapter') do |v|
+    options[:connection_options][:adapter] = v
+  end
+
+  opts.on('--database DATABASE', 'DB name') do |v|
+    options[:connection_options][:database] = v
+  end
+
+  opts.on('--module MODULE_NAME', 'module name') do |v|
+    Object.const_set(v, Module.new)
+    options[:base_module] = Object.const_get(v)
+  end
+
+  opts.on('-c', '--config FILE', "config file, default #{DEFAULT_CONFIG_FILE}") do |v|
+    options[:config_file] = v
+  end
+
+  opts.on('-s', '--section SECTION', 'config file section') do |v|
+    options[:config_file] = DEFAULT_CONFIG_FILE if !options[:config_file] && File.exist?(DEFAULT_CONFIG_FILE)
+    options[:config_section] = v
+  end
+
+  opts.on('--create-class-files DIR', 'Create ActiveRecord class files') do |v|
+    raise("#{v} not found") unless File.exist?(v)
+
+    options[:class_files_dir] = v
+  end
+end.parse!
+# rubocop:enable Metrics/BlockLength
+
+if options[:config_file]
+  raise('must specify a section') unless options[:config_section]
+
+  config = YAML.load_file(options[:config_file])[options[:config_section]]
+  options[:skip_tables] = config.delete('skip_tables') || []
+  options[:relationships] = config.delete('relationships') || {}
+  if (module_name = config.delete('module'))
+    Object.const_set(module_name, Module.new)
+    options[:base_module] = Object.const_get(module_name)
+  end
+  options[:connection_options] = config.each_with_object({}) { |(name, value), hsh| hsh[name.to_sym] = value }
+end
+
+database = DynamicActiveModel::Explorer.explore(
+  options[:base_module],
+  options[:connection_options],
+  options[:skip_tables],
+  options[:relationships]
+)
+
+if options[:class_files_dir]
+  database.models.each do |model|
+    DynamicActiveModel::TemplateClassFile.new(model).create_template!(options[:class_files_dir])
+  end
+else
+  require 'irb'
+  IRB.start(__FILE__)
+end

data/lib/dynamic-active-model.rb

@@ -3,7 +3,9 @@
 # DynamicActiveModel module for create ActiveRecord models
 module DynamicActiveModel
   autoload :Database, 'dynamic-active-model/database'
+  autoload :Explorer, 'dynamic-active-model/explorer'
   autoload :Factory, 'dynamic-active-model/factory'
   autoload :ForeignKey, 'dynamic-active-model/foreign_key'
   autoload :Relations, 'dynamic-active-model/relations'
+  autoload :TemplateClassFile, 'dynamic-active-model/template_class_file'
 end

data/lib/dynamic-active-model/explorer.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module DynamicActiveModel
+  # DynamicActiveModel::EWplorer creates models and relationships
+  module Explorer
+    def self.explore(base_module, connection_options, skip_tables = [], relationships = {})
+      database = create_models!(base_module, connection_options, skip_tables)
+      build_relationships!(database, relationships)
+      database
+    end
+
+    def self.create_models!(base_module, connection_options, skip_tables)
+      database = Database.new(base_module, connection_options)
+      skip_tables.each do |table_name|
+        table_name = Regexp.new("^#{table_name}") if table_name.include?('*')
+        database.skip_table(table_name)
+      end
+      database.create_models!
+      database
+    end
+
+    def self.build_relationships!(database, relationships)
+      relations = Relations.new(database)
+      relationships.each do |table_name, foreign_keys|
+        foreign_keys.each do |foreign_key, relationship_name|
+          relations.add_foreign_key(table_name, foreign_key, relationship_name)
+        end
+      end
+      relations.build!
+    end
+  end
+end

data/lib/dynamic-active-model/foreign_key.rb

@@ -29,7 +29,7 @@ module DynamicActiveModel
     end
 
     def generate_foreign_key(table_name)
-      table_name.singularize + self.class.id_suffix
+      table_name.underscore.singularize + self.class.id_suffix
     end
   end
 end

data/lib/dynamic-active-model/template_class_file.rb

@@ -0,0 +1,78 @@
+# frozen_string_literal: true
+
+module DynamicActiveModel
+  # DynamicActiveModel::TemplateClassFile creates ActiveRecord file for model
+  class TemplateClassFile
+    def initialize(model)
+      @model = model
+    end
+
+    def create_template!(dir)
+      file = dir + '/' + @model.name.underscore + '.rb'
+      File.open(file, 'wb') { |f| f.write(to_s) }
+    end
+
+    def to_s
+      str = "class #{@model.name} < ActiveRecord::Base\n".dup
+      str << "  self.table_name = #{@model.table_name.to_sym.inspect}\n" unless @model.name.underscore.pluralize == @model.table_name
+      all_has_many_relationships.each do |assoc|
+        append_association!(str, assoc)
+      end
+      all_belongs_to_relationships.each do |assoc|
+        append_association!(str, assoc)
+      end
+      str << "end\n"
+      str
+    end
+
+    private
+
+    def all_has_many_relationships
+      @model.reflect_on_all_associations.select do |assoc|
+        assoc.is_a?(ActiveRecord::Reflection::HasManyReflection)
+      end
+    end
+
+    def all_belongs_to_relationships
+      @model.reflect_on_all_associations.select do |assoc|
+        assoc.is_a?(ActiveRecord::Reflection::BelongsToReflection)
+      end
+    end
+
+    def append_association!(str, assoc)
+      assoc_type = assoc.is_a?(ActiveRecord::Reflection::HasManyReflection) ? 'has_many' : 'belongs_to'
+      association_options = assoc_type == 'has_many' ? has_many_association_options(assoc) : belongs_to_association_options(assoc)
+      str << "  #{assoc_type} #{assoc.name.inspect}"
+      unless association_options.empty?
+        association_options.each do |name, value|
+          str << ", #{name}: '#{value}'"
+        end
+      end
+      str << "\n"
+    end
+
+    def has_many_association_options(assoc)
+      options = {}
+      options[:class_name] = assoc.options[:class_name] unless assoc.options[:class_name].underscore.pluralize == assoc.name.to_s
+      options[:foreign_key] = assoc.options[:foreign_key] unless assoc.options[:foreign_key] == default_foreign_key_name
+      options[:primary_key] = assoc.options[:primary_key] unless assoc.options[:primary_key] == 'id'
+      options
+    end
+
+    def belongs_to_association_options(assoc)
+      options = {}
+      options[:class_name] = assoc.options[:class_name] unless assoc.options[:class_name] == assoc.name.to_s.classify
+      options[:foreign_key] = assoc.options[:foreign_key] unless assoc.options[:foreign_key] == (assoc.options[:class_name].underscore + '_id')
+      options[:primary_key] = assoc.options[:primary_key] unless assoc.options[:primary_key] == 'id'
+      options
+    end
+
+    def default_foreign_key_name
+      @model.table_name.underscore.singularize + '_id'
+    end
+
+    def const_get(class_name)
+      class_name.split('::').inject(Object) { |mod, name| mod.const_get(name) }
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: dynamic-active-model
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Doug Youch
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-07-12 00:00:00.000000000 Z
+date: 2019-07-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -26,15 +26,19 @@ dependencies:
         version: '0'
 description: Dynamically create ActiveRecord models for tables
 email: dougyouch@gmail.com
-executables: []
+executables:
+- dynamic-db-explorer
 extensions: []
 extra_rdoc_files: []
 files:
+- bin/dynamic-db-explorer
 - lib/dynamic-active-model.rb
 - lib/dynamic-active-model/database.rb
+- lib/dynamic-active-model/explorer.rb
 - lib/dynamic-active-model/factory.rb
 - lib/dynamic-active-model/foreign_key.rb
 - lib/dynamic-active-model/relations.rb
+- lib/dynamic-active-model/template_class_file.rb
 homepage: https://github.com/dougyouch/dynamic-active-model
 licenses: []
 metadata: {}