module Wukong
  module Store
    #
    # Barebones interface between a wukong class and a cassandra database
    #
    # Class must somehow provide a class-level cassandra_db accessor
    # that sets the @cassandra_db instance variable.
    #
    module CassandraModel
      #
      # Store model to the DB
      #
      def save
        self.class.insert key, self.to_db_hash
      end

      #
      # Flatten attributes for storage in the DB.
      #
      # * omits elements whose value is nil
      # * calls to_s on everything else
      # * This means that blank strings are preserved;
      # * and that false is saved as 'false'
      #
      # Override if you think something fancier than that should happen.
      #
      def to_db_hash
        db_hsh = {}
        to_hash.each{|k,v| db_hsh[k.to_s] = v.to_s unless v.nil? }
        db_hsh
      end

      module ClassMethods
        # Cassandra column family -- taken from the class name by default.
        def table_name
          class_basename
        end

        # Override to control how your class is instantiated from the DB hash
        def from_db_hash *args
          from_hash *args
        end

        # Insert into the cassandra database
        # uses object's #to_db_hash method
        def insert key, *args
          hsh = args.first
          cassandra_db.insert(table_name, key.to_s, hsh)
        end

        # Insert into the cassandra database
        # calls out to object's #from_db_hash method
        def load key
          hsh = cassandra_db.get(self.class_basename, key.to_s)
          from_db_hash(hsh) if hsh
        end

        # invalidates cassandra connection on errors where that makes sense.
        def handle_error action, e
          warn "#{action} failed: #{e} #{e.backtrace.join("\t")}" ;
          @cassandra_db = nil
          sleep 0.2
        end
      end
      # The standard 'inject class methods when module is included' trick
      def self.included base
        base.class_eval{ extend ClassMethods}
      end
    end

  end
end

Hash.class_eval do
  #
  # Flatten attributes for storage in the DB.
  #
  # * omits elements whose value is nil
  # * calls to_s on everything else
  # * This means that blank strings are preserved;
  # * and that false is saved as 'false'
  #
  # Override if you think something fancier than that should happen.
  #
  def to_db_hash
    db_hsh = {}
    to_hash.each{|k,v| db_hsh[k.to_s] = v.to_s unless v.nil? }
    db_hsh
  end
end