seedgen 0.0.3

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 7c656032481489823a64542f0f864d64c8874d6156d1f17664f1770b2e945027
+  data.tar.gz: ff0b902ea7ef8a597732a957d214fa6f80d0a40b1993da7b7e4c994810336866
+SHA512:
+  metadata.gz: cb9c89738160a21995a42560e070c4ed7ee8a5f3ef9173096217a3cefca5dc006ccdf2048a8d19a2e6baea572af896e892ad81c82d448692826558cb414b57a3
+  data.tar.gz: 988d86cacea68dd32c98415268f1f271fb8c5f5c3e863d44dace229ae02f25fb962b19d01aae5984183bd2e9669ccc1f728fd095a51213601fcb2387371411cb

data/MIT-LICENSE

@@ -0,0 +1,20 @@
+Copyright Austin Wasson
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,31 @@
+# SeedGen
+Generate seed files based on your schema.
+
+## Installation
+Add this line to your application's Gemfile:
+
+```ruby
+gem "seedgen", "~> 0.0.3"
+```
+
+## Usage
+SeedGen hooks in to your models and their relationships to build you a seed file. After installation,
+run: 
+
+```bash
+SEED=1 rails db:schema:load
+```
+
+The above command generates `db/seedgen.rb` and seeds your database with the output of this new file based on your schema/models.
+
+Note: You may notice SeedGen missing some of your models. If this is the case, it is because
+your application needs to eager load its classes for SeedGen to see them. 
+In `config/application.rb`, add:
+
+```ruby
+config.after_initialize do
+  Rails.application.eager_load!
+end
+```
+
+and rerun `SEED=1 rails db:schema:load`.

data/Rakefile

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

data/lib/seedgen/database/database.rb

@@ -0,0 +1,15 @@
+module SeedGen
+  module Database 
+    def self.adapter
+      @adapter ||= ActiveRecord::Base.connection.adapter_name
+    end
+
+    def self.models
+      ApplicationRecord.descendants
+    end
+
+    module MySQL; end
+    module PostgreSQL; end
+    module SQLite; end
+  end
+end

data/lib/seedgen/faker.rb

@@ -0,0 +1,51 @@
+require 'faker'
+
+module SeedGen
+  module FakerData
+    class InvalidAttributeType < StandardError; end
+
+    def self.generate(model, column)
+      case column.sql_type_metadata.type
+      when :binary
+        Faker::Number.binary
+      when :boolean
+        # TODO
+        true
+      when :date
+        # TODO
+        Faker::Date.random
+      when :datetime
+        Faker::Date.random
+      when :decimal
+        Faker::Number.decimal
+      when :float
+        3.03
+      when :integer
+        # TODO
+        if enum?(model, column.name)
+          range = model.defined_enums[column.name].length - 1
+          return Faker::Number.between(from: 0, to: range)
+        end
+        Faker::Number.number(digits: 4)
+      when :bigint
+        Faker::Number.number(digits: 10)
+      when :string
+        Faker::Alphanumeric.alpha(number: 7)
+      when :text
+        Faker::Company.bs
+      when :time
+        # TODO
+        Faker::Date.time
+      when :timestamp
+        # TODO
+        Faker::Date.time
+      else
+        raise InvalidAttributeType
+      end
+    end
+
+    def self.enum?(model, column)
+      model.defined_enums.key?(column)
+    end
+  end
+end

data/lib/seedgen/railtie.rb

@@ -0,0 +1,7 @@
+module SeedGen
+  class Railtie < ::Rails::Railtie
+    rake_tasks do
+      load "tasks/seedgen_tasks.rake"
+    end
+  end
+end

data/lib/seedgen/seed_file.rb

@@ -0,0 +1,7 @@
+module SeedGen
+  module SeedFile
+    def self.build
+      # build
+    end
+  end
+end

data/lib/seedgen/version.rb

@@ -0,0 +1,3 @@
+module SeedGen
+  VERSION = "0.0.3"
+end

data/lib/seedgen.rb

@@ -0,0 +1,85 @@
+require "seedgen/database/database"
+require "seedgen/faker"
+require "seedgen/railtie"
+require "seedgen/seed_file"
+require "seedgen/version"
+
+module SeedGen
+  SKIP_ATTRS = %w[ id created_at updated_at ]
+
+  def self.run
+    @models = Database.models || []
+    @seeded_models = []
+
+    @models.each do |model|
+      seed(model)
+      @seeded_models << model
+    end
+
+    puts database_seeded?
+  end
+
+  def self.seed(model)
+    parents = parents(model)
+    if parents.empty?
+      create_record(model)
+    else
+      parents.each do |parent|
+        seed(parent)
+      end
+    end
+  end
+
+  # returns parents that have not been persisted
+  # TODO: refactor with filter
+  # Ignore this awful code 🤮
+  def self.parents(model)
+    parents = []
+    model.reflect_on_all_associations(:belongs_to) do |assoc|
+      parents << assoc.name.to_s.capitalize
+    end
+    nonpersisted_parents = []
+
+    parents.each do |parent|
+      next if parent.active_record.count > 0
+      nonpersisted_parents << parent.active_record
+    end
+    nonpersisted_parents
+  end
+
+  def self.create_record(model)
+    puts "===================="
+    puts "Creating: #{model}"
+    attributes = attrs(model)
+    parents = []
+    model.reflect_on_all_associations(:belongs_to).each do |assoc|
+      parents << assoc.name.to_s
+    end
+
+    unless parents.empty?
+      parents.each do |parent|
+        attributes.merge!({ parent.to_sym => Object.const_get(parent.capitalize).first })
+      end
+    end
+
+    model.create!(attributes)
+  end
+
+  def self.attrs(model)
+    data = {}
+
+    columns = model.content_columns
+    columns.each do |column|
+      unless SKIP_ATTRS.include?(column.name)
+        data[column.name.to_sym] = FakerData.generate(model, column)
+      end
+    end
+
+    data
+  end
+
+  # TODO: validate that all records have been created
+  def self.database_seeded?
+    @models == @seeded_models
+  end
+end

data/lib/tasks/seedgen_tasks.rake

@@ -0,0 +1,15 @@
+require "seedgen"
+
+namespace :seedgen do
+  desc "Seed database"
+  task :seed do
+    if ENV["SEED"] == "1"
+      # SeedGen.models
+      SeedGen.run
+    end
+  end
+
+  Rake.application["db:schema:load"].enhance do
+    Rake::Task["seedgen:seed"].invoke
+  end
+end

metadata

@@ -0,0 +1,83 @@
+--- !ruby/object:Gem::Specification
+name: seedgen
+version: !ruby/object:Gem::Version
+  version: 0.0.3
+platform: ruby
+authors:
+- Austin Wasson
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2024-01-17 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: faker
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rails
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '7.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '7.0'
+description:
+email:
+- austinpwasson@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- MIT-LICENSE
+- README.md
+- Rakefile
+- lib/seedgen.rb
+- lib/seedgen/database/database.rb
+- lib/seedgen/faker.rb
+- lib/seedgen/railtie.rb
+- lib/seedgen/seed_file.rb
+- lib/seedgen/version.rb
+- lib/tasks/seedgen_tasks.rake
+homepage: https://github.com/wassson/seedgen.git
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/wassson/seedgen.git
+  source_code_uri: https://github.com/wassson/seedgen.git
+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: []
+rubygems_version: 3.5.3
+signing_key:
+specification_version: 4
+summary: Generate seed files from your schema.
+test_files: []