ardb 0.26.0 → 0.27.0

data/lib/ardb/clirb.rb

@@ -0,0 +1,58 @@
+module Ardb
+
+  class CLIRB  # Version 1.0.0, https://github.com/redding/cli.rb
+    Error    = Class.new(RuntimeError);
+    HelpExit = Class.new(RuntimeError); VersionExit = Class.new(RuntimeError)
+    attr_reader :argv, :args, :opts, :data
+
+    def initialize(&block)
+      @options = []; instance_eval(&block) if block
+      require 'optparse'
+      @data, @args, @opts = [], [], {}; @parser = OptionParser.new do |p|
+        p.banner = ''; @options.each do |o|
+          @opts[o.name] = o.value; p.on(*o.parser_args){ |v| @opts[o.name] = v }
+        end
+        p.on_tail('--version', ''){ |v| raise VersionExit, v.to_s }
+        p.on_tail('--help',    ''){ |v| raise HelpExit,    v.to_s }
+      end
+    end
+
+    def option(*args); @options << Option.new(*args); end
+    def parse!(argv)
+      @args = (argv || []).dup.tap do |args_list|
+        begin; @parser.parse!(args_list)
+        rescue OptionParser::ParseError => err; raise Error, err.message; end
+      end; @data = @args + [@opts]
+    end
+    def to_s; @parser.to_s; end
+    def inspect
+      "#<#{self.class}:#{'0x0%x' % (object_id << 1)} @data=#{@data.inspect}>"
+    end
+
+    class Option
+      attr_reader :name, :opt_name, :desc, :abbrev, :value, :klass, :parser_args
+
+      def initialize(name, *args)
+        settings, @desc = args.last.kind_of?(::Hash) ? args.pop : {}, args.pop || ''
+        @name, @opt_name, @abbrev = parse_name_values(name, settings[:abbrev])
+        @value, @klass = gvalinfo(settings[:value])
+        @parser_args = if [TrueClass, FalseClass, NilClass].include?(@klass)
+          ["-#{@abbrev}", "--[no-]#{@opt_name}", @desc]
+        else
+          ["-#{@abbrev}", "--#{@opt_name} #{@opt_name.upcase}", @klass, @desc]
+        end
+      end
+
+      private
+
+      def parse_name_values(name, custom_abbrev)
+        [ (processed_name = name.to_s.strip.downcase), processed_name.gsub('_', '-'),
+          custom_abbrev || processed_name.gsub(/[^a-z]/, '').chars.first || 'a'
+        ]
+      end
+      def gvalinfo(v); v.kind_of?(Class) ? [nil,gklass(v)] : [v,gklass(v.class)]; end
+      def gklass(k); k == Fixnum ? Integer : k; end
+    end
+  end
+
+end

data/lib/ardb/default_order_by.rb

@@ -0,0 +1,59 @@
+require 'much-plugin'
+
+module Ardb
+
+  module DefaultOrderBy
+    include MuchPlugin
+
+    DEFAULT_ATTRIBUTE  = :order_by
+    DEFAULT_SCOPE_PROC = proc{ self.class.scoped }
+
+    plugin_included do
+      extend ClassMethods
+      include InstanceMethods
+
+      @ardb_default_order_by_config = {}
+
+    end
+
+    module ClassMethods
+
+      def default_order_by(options = nil)
+        options ||= {}
+
+        @ardb_default_order_by_config.merge!({
+          :attribute  => options[:attribute] || DEFAULT_ATTRIBUTE,
+          :scope_proc => options[:scope]     || DEFAULT_SCOPE_PROC
+        })
+
+        before_validation :ardb_default_order_by, :on => :create
+      end
+
+      def ardb_default_order_by_config
+        @ardb_default_order_by_config
+      end
+
+    end
+
+    module InstanceMethods
+
+      private
+
+      def reset_order_by
+        attr_name  = self.class.ardb_default_order_by_config[:attribute]
+        scope_proc = self.class.ardb_default_order_by_config[:scope_proc]
+
+        current_max = self.instance_eval(&scope_proc).maximum(attr_name) || 0
+        self.send("#{attr_name}=", current_max + 1)
+      end
+
+      def ardb_default_order_by
+        attr_name = self.class.ardb_default_order_by_config[:attribute]
+        reset_order_by if self.send(attr_name).to_s.empty?
+        true
+      end
+
+    end
+
+  end
+end

data/lib/ardb/has_slug.rb

@@ -1,19 +1,20 @@
+require 'much-plugin'
+
 module Ardb
 
   module HasSlug
+    include MuchPlugin
 
     DEFAULT_ATTRIBUTE    = :slug
     DEFAULT_PREPROCESSOR = :downcase
     DEFAULT_SEPARATOR    = '-'.freeze
 
-    def self.included(klass)
-      klass.class_eval do
-        extend ClassMethods
-        include InstanceMethods
+    plugin_included do
+      extend ClassMethods
+      include InstanceMethods
 
-        @ardb_has_slug_config = {}
+      @ardb_has_slug_config = {}
 
-      end
     end
 
     module ClassMethods

data/lib/ardb/migration.rb

@@ -0,0 +1,42 @@
+require 'fileutils'
+
+module Ardb
+
+  class Migration
+
+    attr_reader :identifier, :class_name, :file_name, :file_path
+    attr_reader :source
+
+    def initialize(identifier)
+      raise NoIdentifierError if identifier.to_s.empty?
+
+      @identifier = identifier
+      @class_name = @identifier.classify.pluralize
+      @file_name  = get_file_name(@identifier)
+      @file_path  = File.join(Ardb.config.migrations_path, "#{@file_name}.rb")
+
+      @source = "require 'ardb/migration_helpers'\n\n" \
+                "class #{@class_name} < ActiveRecord::Migration\n" \
+                "  include Ardb::MigrationHelpers\n\n" \
+                "  def change\n" \
+                "  end\n\n" \
+                "end\n"
+    end
+
+    def save!
+      FileUtils.mkdir_p Ardb.config.migrations_path
+      File.open(self.file_path, 'w'){ |f| f.write(self.source) }
+      self
+    end
+
+    private
+
+    def get_file_name(identifier)
+      "#{Time.now.strftime("%Y%m%d%H%M%S")}_#{identifier.underscore}"
+    end
+
+    NoIdentifierError = Class.new(ArgumentError)
+
+  end
+
+end

data/lib/ardb/migration_helpers.rb

@@ -1,72 +1,75 @@
 require 'ardb'
 
-module Ardb; end
-module Ardb::MigrationHelpers
-  module_function
+module Ardb
 
-  def foreign_key(from_table, from_column, to_table, options={})
-    fk = ForeignKey.new(from_table, from_column, to_table, options)
-    execute(fk.add_sql)
-  end
-
-  def drop_foreign_key(*args)
-    from_table, from_column = args[0..1]
-    options = args.last.kind_of?(Hash) ? args.last : {}
-    fk = ForeignKey.new(from_table, from_column, nil, options)
-    execute(fk.drop_sql)
-  end
-
-  def remove_column_with_fk(table, column)
-    drop_foreign_key(table, column)
-    remove_column(table, column)
-  end
+  module MigrationHelpers
+    module_function
 
-  class ForeignKey
-    attr_reader :from_table, :from_column, :to_table, :to_column, :name, :adapter
-
-    def initialize(from_table, from_column, to_table, options=nil)
-      options ||= {}
-      @from_table  = from_table.to_s
-      @from_column = from_column.to_s
-      @to_table    = to_table.to_s
-      @to_column   = (options[:to_column] || 'id').to_s
-      @name        = (options[:name] || "fk_#{@from_table}_#{@from_column}").to_s
-      @adapter     = Ardb::Adapter.send(Ardb.config.db.adapter)
+    def foreign_key(from_table, from_column, to_table, options={})
+      fk = ForeignKey.new(from_table, from_column, to_table, options)
+      execute(fk.add_sql)
     end
 
-    def add_sql
-      apply_data(@adapter.foreign_key_add_sql)
+    def drop_foreign_key(*args)
+      from_table, from_column = args[0..1]
+      options = args.last.kind_of?(Hash) ? args.last : {}
+      fk = ForeignKey.new(from_table, from_column, nil, options)
+      execute(fk.drop_sql)
     end
 
-    def drop_sql
-      apply_data(@adapter.foreign_key_drop_sql)
+    def remove_column_with_fk(table, column)
+      drop_foreign_key(table, column)
+      remove_column(table, column)
     end
 
-    private
-
-    def apply_data(template_sql)
-      template_sql.
-        gsub(':from_table',  @from_table).
-        gsub(':from_column', @from_column).
-        gsub(':to_table',    @to_table).
-        gsub(':to_column',   @to_column).
-        gsub(':name',        @name)
+    class ForeignKey
+      attr_reader :from_table, :from_column, :to_table, :to_column, :name, :adapter
+
+      def initialize(from_table, from_column, to_table, options=nil)
+        options ||= {}
+        @from_table  = from_table.to_s
+        @from_column = from_column.to_s
+        @to_table    = to_table.to_s
+        @to_column   = (options[:to_column] || 'id').to_s
+        @name        = (options[:name] || "fk_#{@from_table}_#{@from_column}").to_s
+        @adapter     = Ardb::Adapter.send(Ardb.config.db.adapter)
+      end
+
+      def add_sql
+        apply_data(@adapter.foreign_key_add_sql)
+      end
+
+      def drop_sql
+        apply_data(@adapter.foreign_key_drop_sql)
+      end
+
+      private
+
+      def apply_data(template_sql)
+        template_sql.
+          gsub(':from_table',  @from_table).
+          gsub(':from_column', @from_column).
+          gsub(':to_table',    @to_table).
+          gsub(':to_column',   @to_column).
+          gsub(':name',        @name)
+      end
     end
-  end
 
-  # This file will setup the AR migration command recorder for being able to change our
-  # stuff, require it in an initializer
+    # This file will setup the AR migration command recorder for being able to
+    # change our stuff, require it in an initializer
 
-  module RecorderMixin
+    module RecorderMixin
 
-    def foreign_key(*args)
-      record(:foreign_key, args)
-    end
+      def foreign_key(*args)
+        record(:foreign_key, args)
+      end
+
+      protected
 
-    protected
+      def invert_foreign_key(args)
+        [ :drop_foreign_key, args ]
+      end
 
-    def invert_foreign_key(args)
-      [ :drop_foreign_key, args ]
     end
 
   end

data/lib/ardb/record_spy.rb

@@ -1,9 +1,11 @@
 require 'arel'
+require 'much-plugin'
 require 'ardb/relation_spy'
 
 module Ardb
 
   module RecordSpy
+    include MuchPlugin
 
     def self.new(&block)
       block ||= proc{ }
@@ -12,11 +14,9 @@ module Ardb
       record_spy
     end
 
-    def self.included(klass)
-      klass.class_eval do
-        extend ClassMethods
-        include InstanceMethods
-      end
+    plugin_included do
+      extend ClassMethods
+      include InstanceMethods
     end
 
     module ClassMethods

data/lib/ardb/relation_spy.rb

@@ -4,13 +4,16 @@ module Ardb
 
     attr_reader :applied
     attr_accessor :limit_value, :offset_value
-    attr_accessor :pluck_values
+    attr_accessor :pluck_values, :maximum_values, :minimum_values
     attr_accessor :results
 
     def initialize
       @applied, @results = [], []
       @offset_value, @limit_value = nil, nil
-      @pluck_values = {}
+
+      @pluck_values   = {}
+      @maximum_values = {}
+      @minimum_values = {}
     end
 
     def initialize_copy(copied_from)
@@ -134,6 +137,14 @@ module Ardb
       [@pluck_values[column_name]] * @results.size
     end
 
+    def maximum(column_name)
+      @maximum_values[column_name]
+    end
+
+    def minimum(column_name)
+      @minimum_values[column_name]
+    end
+
     class AppliedExpression < Struct.new(:type, :args)
       def to_sql
         "#{self.type}: #{self.args.inspect}"

data/lib/ardb/root_path.rb

@@ -2,11 +2,14 @@
 # to the full qualifed root path.  The goal here is to specify path options
 # with root-relative path strings.
 
-module Ardb; end
-class Ardb::RootPath < String
+module Ardb
+
+  class RootPath < String
+
+    def initialize(path_string)
+      super(Ardb.config.root_path.join(path_string).to_s)
+    end
 
-  def initialize(path_string)
-    super(Ardb.config.root_path.join(path_string).to_s)
   end
 
 end

data/lib/ardb/test_helpers.rb

@@ -4,62 +4,76 @@ require 'ardb'
 # Use theses helpers in your test suite.  They all generally assume Ardb has
 # already been initialized by calling `Ardb.init`.
 
-module Ardb; end
-module Ardb::TestHelpers
-  module_function
+module Ardb
 
-  def drop_tables
-    Ardb.adapter.drop_tables
-  end
+  module TestHelpers
+    module_function
 
-  def load_schema
-    Ardb.adapter.load_schema
-  end
+    def drop_tables
+      Ardb.adapter.drop_tables
+    end
 
-  def create_db!
-    Ardb.adapter.create_db
-  end
+    def load_schema
+      Ardb.adapter.load_schema
+    end
 
-  def create_db
-    @create_db ||= begin
-      self.create_db!
-      true
+    def create_db!
+      Ardb.adapter.create_db
     end
-  end
 
-  def drop_db!
-    Ardb.adapter.drop_db
-  end
+    def create_db
+      @create_db ||= begin
+        self.create_db!
+        true
+      end
+    end
 
-  def drop_db
-    @drop_db ||= begin
-      self.drop_db!
-      true
+    def drop_db!
+      Ardb.adapter.drop_db
     end
-  end
 
-  def migrate_db!
-    Ardb.adapter.migrate_db
-  end
+    def drop_db
+      @drop_db ||= begin
+        self.drop_db!
+        true
+      end
+    end
 
-  def migrate_db
-    @migrate_db ||= begin
-      self.migrate_db!
-      true
+    def connect_db!
+      Ardb.adapter.connect_db
     end
-  end
 
-  def reset_db!
-    self.drop_db!
-    self.create_db!
-    self.load_schema
-  end
+    def connect_db
+      @connect_db ||= begin
+        self.connect_db!
+        true
+      end
+    end
+
+    def migrate_db!
+      Ardb.adapter.migrate_db
+    end
+
+    def migrate_db
+      @migrate_db ||= begin
+        self.migrate_db!
+        true
+      end
+    end
+
+    def reset_db!
+      self.drop_db!
+      self.create_db!
+      self.load_schema
+    end
 
-  def reset_db
-    @reset_db ||= begin
-      self.reset_db!
-      true
+    def reset_db
+      @reset_db ||= begin
+        self.reset_db!
+        true
+      end
     end
+
   end
 
 end