queirozfcom/rachinations

require 'set'
require_relative '../../domain/resources/token'
require_relative '../../domain/nodes/node'
require_relative '../../domain/modules/common/refiners/proc_convenience_methods'

using ProcConvenienceMethods

# @abstract Subclass and override {#trigger!}, {#resource_count},
#   {#pull_any!}, {#pull_all!}, {#push_any!}, {#push_all!}, {#take_resource!}
#   and {#put_resource!} to implement nodes that can store resources
class ResourcefulNode < Node

  include Invariant

  def initialize(hsh=nil)
    @resources_added=Hash.new(0)
    @resources_removed=Hash.new(0)
  end

  def initialize_copy(orig)
    super

    #need to clone the resource bag as well...
    @resources = @resources.clone()

    #don't need this. takes too much space
    @diagram = nil

  end

  def supports?(klass)
    if klass.eql?(Token)
      untyped?
    else
      # untyped nodes support everything.
      if untyped?
        true
      else
        typed? && types.include?(klass)
      end
    end
  end

  alias_method :support?, :supports?


  def resources_added(klass=nil)
    if klass.nil?
      @resources_added.values.reduce(0) { |acc, elem| acc += elem }
    else
      @resources_added[klass]
    end
  end

  def resources_removed(klass=nil)
    if klass.nil?
      @resources_removed.values.reduce(0) { |acc, elem| acc += elem }
    else
      @resources_removed[klass]
    end
  end

  def commit!
    super
  end

  def unlock_resources!
    @resources.each_where { |r|
      if r.locked?
        r.unlock!
      end
    }
  end

  def trigger!
    raise NotImplementedError, "Please update class #{self.class} to respond to: :#{__callee__}"
  end

  def resource_count(type=nil)
    raise NotImplementedError, "Please update class #{self.class} to respond to: :#{__callee__}"
  end

  def push_any
    raise NotImplementedError, "Please update class #{self.class} to respond to: :#{__callee__}"
  end

  def push_all
    raise NotImplementedError, "Please update class #{self.class} to respond to: :#{__callee__}"
  end

  def pull_any
    raise NotImplementedError, "Please update class #{self.class} to respond to: :#{__callee__}"
  end

  def pull_all
    raise NotImplementedError, "Please update class #{self.class} to respond to: :#{__callee__}"
  end

end