# lib/tyranny.rb
 
require 'rubygems'
require 'rufus/tokyo/tyrant'
 
module Tyranny
  
  def self.configure(config)
    configuration.update(config) if config
  end
  
  def self.configuration
    @configuration ||= {}
  end
  
  def self.extractor
    @extractor ||= Extractor.new
  end
  
  def self.[](name)
    extractor[name]
  end
  
  class Extractor
  
    def [](name)
      tables[name.to_s] ||= table(name.to_s)
    end
  
    def tables
      @tables ||= {}
    end
    
    def close
      @tables.values.each do |table|
        table.close
      end
    end
    
    def forget
      close
      tables.clear
    end
  
    private
  
    def table(name)
      config = Tyranny.configuration[name]
      if config['filename'] && config['socket']
        local_table(config)
      else
        raise NotImplementedError, "Remote configuration not implemented"
      end
    end
  
    def local_table(config)
      local_server!(config)
      Rufus::Tokyo::TyrantTable.new(config['socket'])
    end
  
    def local_server!(config)
      return if File.socket?(config['socket'])
      system "ttserver -host '#{config['socket']}' -port 0 '#{config['filename']}' &"
    end
    
  end
      
end

# config/initializers/tyranny_init.rb
 
require File.dirname(__FILE__) << "/../../lib/tyranny"
 
configuration = YAML.load(File.read(File.join(Rails.root, 'config/tyranny.yml')))
 
tyranny_config = configuration[Rails.env]
if tyranny_config
  Tyranny.configure(tyranny_config)
end