lib/ar_condition.rb

   1 # redMine - project management software
   2 # Copyright (C) 2006-2008  Jean-Philippe Lang
   3 #
   4 # This program is free software; you can redistribute it and/or
   5 # modify it under the terms of the GNU General Public License
   6 # as published by the Free Software Foundation; either version 2
   7 # of the License, or (at your option) any later version.
   8 #
   9 # This program is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 # GNU General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU General Public License
  15 # along with this program; if not, write to the Free Software
  16 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  17
  18 class ARCondition
  19   attr_reader :conditions
  20
  21   def initialize(condition=nil)
  22     @conditions = ['1=1']
  23     add(condition) if condition
  24   end
  25
  26   def add(condition)
  27     if condition.is_a?(Array)
  28       @conditions.first << " AND (#{condition.first})"
  29       @conditions += condition[1..-1]
  30     elsif condition.is_a?(String)
  31       @conditions.first << " AND (#{condition})"
  32     else
  33       raise "Unsupported #{condition.class} condition: #{condition}"
  34     end
  35     self
  36   end
  37
  38   def <<(condition)
  39     add(condition)
  40   end
  41 end