insights_export 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 830f36f902942abbc20985d74ef05fc45c27cf34
+  data.tar.gz: 162aa9da745f357847d55f22478f626c999a68f0
+SHA512:
+  metadata.gz: 5500276d0aab589700a5d0dcecf8f17bd5735b9a90b58a0e22d1b53cd0a6478f77c2e562cb9cf5ad750f8dfddbee0cb3e3dc159f2b16927d0c9031c1f689513e
+  data.tar.gz: 5180a572f9c374fc1217401b7843a2f81d6d391ef336599275bf705763d5d762c58fad0cb5a170c6f44b9f6b95f4a65575c36baab2f4f5b6fccd9c1819e2a902

data/Rakefile

@@ -0,0 +1,2 @@
+require 'bundler/gem_tasks'
+require 'insights_export'

data/lib/insights_export/export_models.rb

@@ -0,0 +1,137 @@
+module InsightsExport
+  class ExportModels
+    def self.config_file
+      "#{Rails.root}/config/insights.yml"
+    end
+
+    def self.load
+      structure = YAML::load_file(config_file) rescue get_structure
+
+      structure.select { |k, v| v['enabled'] }.map do |k, v|
+        [k, v.merge({ 'columns' => v['columns'].select { |_, vv| vv.present? } })]
+      end.to_h
+    end
+
+    def self.export
+      input = YAML::load_file(config_file) rescue nil
+      structure = get_structure.deep_stringify_keys
+      output = {}
+
+      if input.present?
+        output = input.dup.deep_stringify_keys
+        structure.each do |model_name, model_structure|
+          # we already had this model in the output
+          if output[model_name].present?
+            model_structure.each do |key, value|
+              if key == 'custom'
+                next
+              elsif key == 'columns' || key == 'aggregate'
+                output[model_name][key] ||= {}
+                value.each do |value_key, value_value|
+                  existing = output[model_name][key][value_key]
+                  if existing != false
+                    output[model_name][key][value_key] = (existing || {}).merge(value_value)
+                  end
+                end
+              elsif key != 'enabled'
+                output[model_name][key] = value
+              end
+            end
+          else
+            output[model_name] = model_structure
+          end
+        end
+      else
+        output = structure
+      end
+
+      File.open(config_file, 'w') {|f| f.write output.deep_stringify_keys.to_yaml }
+    end
+
+    def self.get_structure
+      Rails.application.eager_load! if Rails.env.development?
+
+      models = ApplicationRecord.descendants
+
+      models.map do |model|
+        begin
+          columns_hash = model.columns_hash
+        rescue
+          next
+        end
+
+        model_structure = {
+          enabled: true,
+          model: model.to_s,
+          table_name: model.table_name,
+          primary_key: model.primary_key,
+          columns: columns_hash.map do |key, column|
+            obj = if column.type.in? %i(datetime date)
+                    { type: :time }
+                  elsif column.type.in? %i(integer decimal float)
+                    { type: :number }
+                  elsif column.type.in? %i(string text)
+                    { type: :string }
+                  elsif column.type.in? %i(boolean)
+                    { type: :boolean }
+                  elsif column.type.in? %i(json)
+                    { type: :payload }
+                  elsif column.type.in? %i(geography)
+                    { type: :geo }
+                  else
+                    puts "Warning! Unknown column type: :#{column.type} for #{model.to_s}, column #{key}"
+                    { unknown: column.type }
+                  end
+
+            if key == model.primary_key
+              obj[:index] = :primary_key
+            end
+
+            [key.to_sym, obj]
+          end.to_h,
+          custom: {},
+          aggregate: {
+            count: {
+              sql: "count($$.#{model.primary_key})"
+            }
+          },
+          links: {
+            incoming: {},
+            outgoing: {}
+          }
+        }
+
+        model.reflections.each do |association_name, reflection|
+          if reflection.macro == :belongs_to
+            # reflection.class_name # User
+            # reflection.foreign_key # user_id
+            # reflection.association_primary_key # id
+
+            model_structure[:columns].delete(reflection.foreign_key.to_sym)
+            model_structure[:links][:outgoing][association_name] = {
+              model: reflection.class_name,
+              model_key: reflection.association_primary_key,
+              my_key: reflection.foreign_key
+            }
+          elsif reflection.macro.in? %i(has_one has_many)
+            # skip has_many :through associations
+            if reflection.options.try(:[], :through).present?
+              next
+            end
+
+            model_structure[:links][:incoming][association_name] = {
+              model: reflection.class_name,
+              model_key: reflection.foreign_key,
+              my_key: reflection.association_primary_key
+            }
+          else
+            puts "Warning! Unknown reflection :#{reflection.macro} for association #{association_name} on model #{model.to_s}"
+          end
+        end
+
+        [model.to_s, model_structure]
+      end.select(&:present?).sort_by { |k, v| k }.to_h.deep_stringify_keys
+    end
+  end
+end
+

data/lib/insights_export/railtie.rb

@@ -0,0 +1,7 @@
+module InsightsExport
+  class Railtie < Rails::Railtie
+    rake_tasks do
+      require 'immigrant/task'
+    end
+  end
+end

data/lib/insights_export/task.rb

@@ -0,0 +1,7 @@
+namespace :insights do
+  desc 'Export database structure to config/insights.yml'
+
+  task :export do
+    InsightsExport::ExportModels.export
+  end
+end

data/lib/insights_export/version.rb

@@ -0,0 +1,3 @@
+module InsightsExport
+  VERSION = '0.0.1'
+end

data/lib/insights_export.rb

@@ -0,0 +1,3 @@
+require 'insights_export/export_models'
+
+require 'insights_export/railtie' if defined?(Rails)

metadata

@@ -0,0 +1,63 @@
+--- !ruby/object:Gem::Specification
+name: insights_export
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Marius Andra
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-05-16 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: for use with insights.
+email: marius.andra@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- Rakefile
+- lib/insights_export.rb
+- lib/insights_export/export_models.rb
+- lib/insights_export/railtie.rb
+- lib/insights_export/task.rb
+- lib/insights_export/version.rb
+homepage: https://github.com/mariusandra/insights_export
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.1
+signing_key: 
+specification_version: 4
+summary: Export your database structure into config/insights.yml!
+test_files: []