factory_bro 0.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: e69622ef3890178079a03e5a05f48f18031761d5
+  data.tar.gz: c2fe5d90948dd134d3bc7fa478301bee227ea217
+SHA512:
+  metadata.gz: 7bf9db34521a8ab5748c1f0f50c339abcd66965a29be392464d0b80371e3642e79dc25db8b305552668c2317813d6dad9837341e95c889e130680cfcda35942e
+  data.tar.gz: 6c97639c5948726a394e9a50fafec5ce2a169009ed71fd4601d9eb16be0d8d850c98ab744ec9a28dd8085a892c7548b63ed9880c8d06053fe375366d3cfa523c

data/lib/factory_bro.rb

@@ -0,0 +1,125 @@
+require 'pg'
+require 'faker'
+require 'json'
+
+class FactoryBro
+  def self.connect(db)
+    if db.include? 'postgres://'
+      dbData = parse_url(db)
+      conn = PG.connect(  user:     dbData[:user],
+                          password: dbData[:pw],
+                          host:     dbData[:host],
+                          port:     dbData[:port],
+                          dbname:   dbData[:dbname]
+                        )
+    else
+      # take arguments of user and pw
+      $conn = PG.connect( dbname: db)
+    end
+  end
+
+  def self.create_bases
+    tables = parse_tables
+    system 'mkdir', '-p', 'factories' unless File.directory?(Dir.pwd+"/factories")
+    tables.each do |table|
+      path = File.join(Dir.pwd, 'factories', "#{table}.rb")
+      unless File.exists?(path)
+
+        File.open(path, 'w') {|file| file.write(base_factory_content(table)) }
+        # puts "#{table} factory created at: #{path}"
+      end
+    end
+  end
+
+  def self.generate_data(table, data)
+    helperData = generate_helper(data[:factoryData])
+    res = $conn.exec("INSERT INTO #{table} (#{helperData[:columns]})
+    VALUES (#{helperData[:values]});")
+    data[:meta]
+  end
+
+  private
+
+  def self.parse_url(url)
+    { :user     => url.scan(/postgres:..[A-Za-z0-9]+/).first.split('//')[-1],
+      :pw       => url.scan(/:[A-Za-z0-9]+@/).first.gsub(/[^a-zA-Z0-9\-]/,""),
+      :host     => url.scan(/(?<=@)(.*)(?=:)/).first.first,
+      :port     => url.scan(/:[0-9]+./).first.split('/').first.gsub(/[^a-zA-Z0-9\-]/,""),
+      :dbname   => url.scan(/.[A-Za-z0-9]+$/).first.gsub(/[^a-zA-Z0-9\-]/,"")
+    }
+  end
+
+  def self.parse_tables
+    res = $conn.exec("SELECT TABLE_NAME
+                      FROM INFORMATION_SCHEMA.TABLES
+                      WHERE TABLE_TYPE = 'BASE TABLE' AND TABLE_SCHEMA='public'
+                      ORDER BY TABLE_NAME;")
+    res.values.flatten # as array of column names
+  end
+
+  def self.parse_columns(table)
+    # hash this
+    res = $conn.exec(" SELECT COLUMN_NAME, DATA_TYPE, CHARACTER_MAXIMUM_LENGTH
+                      FROM INFORMATION_SCHEMA.COLUMNS
+                      WHERE TABLE_NAME = '#{table}';")
+    res.values # as tuple
+  end
+
+  def self.base_factory_content(table)
+    "module Factory
+  class #{table.capitalize}
+    def self.base
+      # remember:
+      # to return any needed metadata (i.e. ID)
+      # run other required bases before running method below
+      FactoryBro.generate_data('#{table}' , #{base_method_helper(table)})
+    end
+  end
+end
+"
+  end
+
+  def self.base_method_helper(table)
+    data = {}
+    parse_columns(table).each do |column|
+      data.merge!(column.first => auto_faker_assignment(column)) # idk if auto faker assignment is helping or a burden?
+    end
+    JSON.pretty_generate({ "factoryData" => data, "meta"=> {}})
+  end
+
+  def self.auto_faker_assignment(column)
+    type = column[1] # hash this
+    name = column.first
+    if type == "character varying"
+      if name.include? 'sign_in_ip'
+        return "Faker::Internet.ip_v4_address"
+      elsif name == 'email'
+        return "Faker::Internet.safe_email"
+      else
+        return "Faker::Lorem.characters(32)"
+      end
+    elsif type == "text"
+      return "# password and password salt? don't know what to return"
+    elsif type == "integer"
+      return "rand(0...1000)"
+    elsif type == "boolean"
+      return "Faker::Boolean.boolean"
+    elsif type == "timestamp with time zone"
+      return "Faker::Time.between(DateTime.now - 1, DateTime.now)"
+    else
+      "# could not determine type: #{type}"
+    end
+  end
+
+  def self.generate_helper(data)
+    formatted = data.to_a
+    columns = "#{formatted.first.first.to_s}"
+    values = "'#{formatted.first.last}'"
+    formatted.shift
+    formatted.each do |column|
+      columns += ", #{column.first.to_s}"
+      values += ", '#{column.last}'"
+    end
+    { columns: columns, values: values}
+  end
+end

metadata

@@ -0,0 +1,85 @@
+--- !ruby/object:Gem::Specification
+name: factory_bro
+version: !ruby/object:Gem::Version
+  version: 0.0.0
+platform: ruby
+authors:
+- Nate Reynolds
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-09-12 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: pg
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.18.4
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.18.4
+- !ruby/object:Gem::Dependency
+  name: faker
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.6.6
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.6.6
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.10.4
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.10.4
+description: PSQL Parser and Data Generation similar to factory_girl
+email: nate@n8r.us
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/factory_bro.rb
+homepage: https://github.com/nvreynolds/factory_bro
+licenses: []
+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: Factory Bro
+test_files: []