Contents

# frozen_string_literal: true

require 'rails/generators'
require 'rails/generators/migration'

class ActsAsGraphDiagramGenerator < Rails::Generators::Base # :nodoc:
  include Rails::Generators::Migration

  def self.source_root
    @source_root ||= File.join(File.dirname(__FILE__), 'templates')
  end

  # Implement the required interface for Rails::Generators::Migration.
  # taken from https://github.com/rails/rails/blob/master/activerecord/lib/rails/generators/active_record.rb
  def self.next_migration_number(dirname)
    if ActiveRecord::Base.timestamped_migrations
      Time.now.utc.strftime('%Y%m%d%H%M%S')
    else
      format('%.3d', (current_migration_number(dirname) + 1))
    end
  end

  def create_migration_file
    migration_template 'migration.rb', 'db/migrate/acts_as_graph_diagram_migration.rb'
  end

  def create_model
    template 'model.rb', File.join('app/models', 'edge.rb')
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
acts_as_graph_diagram-0.1.1	lib/generators/acts_as_graph_diagram_generator.rb
acts_as_graph_diagram-0.1.0	lib/generators/acts_as_graph_diagram_generator.rb