#
# file::    sqlitestore.rb
# author::  Jon A. Lambert
# version:: 2.9.0
# date::    03/16/2006
#
# This source code copyright (C) 2005, 2006 by Jon A. Lambert
# All rights reserved.
#
# Released under the terms of the TeensyMUD Public License
# See LICENSE file for additional information.
#
$:.unshift "lib" if !$:.include? "lib"
$:.unshift "vendor" if !$:.include? "vendor"

require 'sqlite'
require 'storage/sqlitehash'
require 'utility/log'
require 'storage/store'
require 'storage/cache'


# The SqliteStore class manages access to all object storage.
#
# [+db+] is a handle to the database.
# [+dbtop+] stores the highest id used in the database.
# [+cache+] is a handle to the cache
class SqliteStore < Store
  logger 'INFO'

  def initialize(dbfile)
    super()
    @dbfile = "#{dbfile}.sqlite"

    # check if database exists and build it if not
    build_database
    log.info "Loading world..."

    # open database and sets @dbtop to highest object id
    @db = SQLite::Database.open(@dbfile)
    @db.cache_size = 1000;     # default is 2000 1.5K pages
    @db.synchronous = 0;       # dangerous if OS crash, but fast
    @db.type_translation = true;
    @db.execute("select id from tmud;") do |i|
      @dbtop = i.first if i.first > @dbtop
    end

    @cache = CacheManager.new(@db)
    log.info "Database '#{@dbfile}' loaded...highest id = #{@dbtop}."
#    log.debug @db.inspect
  rescue
    log.fatal $!
    raise
  end

  # Save the world
  # [+return+] Undefined.
  def save
    @cache.sync
  end

  # Close the database
  # [+return+] Undefined.
  def close
    @db.close
  end

  # inspect the store cache (only for caches)
  # [+return+] string
  def inspect
    @cache.inspect
  end

  # Adds a new object to the database.
  # [+obj+] is a reference to object to be added
  # [+return+] Undefined.
  def put(obj)
    @cache.put(obj)
    obj # return really should not be checked
  end

  # Deletes an object from the database.
  # [+oid+] is the id to to be deleted.
  # [+return+] Undefined.
  def delete(oid)
    @cache.delete(oid)
  end

  # Finds an object in the database by its id.
  # [+oid+] is the id to use in the search.
  # [+return+] Handle to the object or nil.
  def get(oid)
    @cache.get(oid)
  end

  # Check if an object is in the database by its id.
  # [+oid+] is the id to use in the search.
  # [+return+] true or false
  def check(oid)
    @db.has_key? oid.to_s
  end

  # Marks an object dirty
  # [+oid+] is the id to use in the search.
  # [+return+] undefined
  def mark(oid)
    @cache.mark(oid)
  end

  # Marks an object nonswappable
  # [+oid+] is the object id
  # [+return+] undefined
  def makenoswap(oid)
    @cache.makenoswap(oid)
  end

  # Marks an object swappable
  # [+oid+] is the object id
  # [+return+] undefined
  def makeswap(oid)
    @cache.makeswap(oid)
  end

  # Iterate through all objects
  # This needs to have all the possible keys first.
  # So we need to fetch them from the cache and from the database and
  # and produce a list of unique keys.
  # [+yield+] Each object in database to block of caller.
  def each
    kys = @cache.keys
    @db.execute("select id from tmud;") do |k|
      kys << k.to_i
    end
    kys.uniq!
    kys.each {|k| yield @cache.get(k)}
  end

  # produces a statistical report of the database
  # [+return+] a string containing the report
  def stats
    stats = super
    stats << @cache.stats
  end

private

  # Checks that the database exists and builds one if not
  # Will raise an exception if something goes wrong.
  def build_database
    if !test(?e, @dbfile)
      log.info "Building minimal world database..."
      db = SQLite::Database.open(@dbfile)
      db.type_translation = true;
      db.execute("create table tmud (id integer primary key, data text);")
      YAML::load(MINIMAL_DB).each do |o|
        db[o.id] = Utility.encode(o)
        log.debug "insert #{o.id}:#{db[o.id]}"
      end
      db.close
    end
  rescue
    log.fatal "Unable to find or build database - '#{@dbfile}'"
    log.fatal $!
    raise
  end

end