peczenyj/MooseX

require 'delegate'

module MooseX
  module Traits
    class Expires < SimpleDelegator
      def self.with(expires)
        expires2 = Class.new(MooseX::Traits::Expires)
        expires2.class_eval do
          define_method(:initialize) do |value|
            super([ value, expires ])
          end
        end
          
        expires2
      end
      
      def initialize(args)
        value, expires   = args[0], args[1]
        @value   = value
        @expires = ((expires >= 0)? Time.now + expires : nil)
        
        __setobj__(@value)
      end
      
      def valid?
        return true if @expires.nil?
        @expires > Time.now
      end
    end
    
    class Counter < SimpleDelegator
      def initialize(value)
        @value = value
        super(@value)
      end

      def inc(by=1)
        @value += by
        __setobj__(@value)
        @value
      end
      
      def dec(by=1)
        @value -= by
        __setobj__(@value)
        @value        
      end

      def reset(to=0)
        @value = to
        __setobj__(@value)
        @value 
      end
    end

    class Pair < SimpleDelegator
      attr_reader :first, :second

      def initialize(pair)
        @first, @second = pair[0], pair[1]
        super([@first, @second ])
      end

      def first=(first_value)
        @first = first_value
        __setobj__([@first, @second ])
      end
      def second=(second_value)
        @second = second_value
        __setobj__([@first, @second ])
      end      
    end

    class Bool < SimpleDelegator
      def initialize(value)
        @value = value
        super(value)
      end  

      def toggle!
        @value = self.not
        __setobj__(@value)
      end

      def set!
        @value = true
        __setobj__(@value)
      end

      def unset!
        @value = false
        __setobj__(@value)
      end

      def not
        ! @value
      end

      def value
        ! self.not
      end      
    end
  end  
end