Contents

require 'active_model'

module RailsDb

  class Table
    include Connection

    attr_reader :name, :data

    delegate :count,     to: :data
    delegate :paginate,  to: :data
    delegate :limit,     to: :data
    delegate :order,     to: :data
    delegate :asc,       to: :data
    delegate :desc,      to: :data

    def initialize(table_name)
      throw 'Access Denied' unless RailsDb::Database.accessible_tables.include?(table_name)
      @name = table_name
      @data = RailsDb::TableData.new(self)
      self
    end

    def to_csv
      CSV.generate do |csv|
        csv << column_names
        data.data.rows.each do |row|
          csv << row
        end
      end
    end

    def indexes
      RailsDb::Database.indexes(name)
    end

    def truncate
      RailsDb::Database.truncate(name)
    end

    def primary_key
      RailsDb::Database.primary_key(name)
    end

    def delete(id)
      RailsDb::Database.delete(name, primary_key, id)
    end

    def create_model(table_name, &block)
      begin
        klass = Class.new(ActiveRecord::Base) do
          def self.model_name
            ActiveModel::Name.new(self, nil, table_name)
          end
          self.table_name = table_name
          self.inheritance_column = nil
        end
        temp  = klass.count # verify that it works, if not load other
      rescue
        klass = ActiveRecord::Base.descendants.detect { |c| c.table_name == table_name }
      end

      klass.class_eval(&block) if block_given?

      if Rails::VERSION::MAJOR == 3
        klass.class_eval do
          klass.columns.each do |e|
            attr_accessor e.name
          end
        end
      end
      klass
    end

    def as_model
      @model ||= create_model(name)
    end

  end # module

end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
rails_db-1.6.0	lib/rails_db/table.rb
rails_db-1.5.0	lib/rails_db/table.rb