bringing over couch_changes from leap_ca including tests

5 files changed, 157 insertions, 0 deletions

diff --git a/lib/tapicero.rb b/lib/tapicero.rb
new file mode 100644
index 0000000..ee52db4
--- /dev/null
+++ b/lib/tapicero.rb
@@ -0,0 +1,19 @@
+unless defined? BASE_DIR
+  BASE_DIR = File.expand_path('../..', __FILE__)
+end
+unless defined? LEAP_CA_CONFIG
+  LEAP_CA_CONFIG = '/etc/leap/tapicero.yaml'
+end
+
+#
+# Load Config
+# this must come first, because CouchRest needs the connection defined before the models are defined.
+#
+require 'tapicero/config'
+Tapicero::Config.load(BASE_DIR, 'config/default.yaml', LEAP_CA_CONFIG, ARGV.grep(/\.ya?ml$/).first)
+
+#
+# Load Tapicero
+#
+require 'tapicero/couch_stream'
+require 'tapicero/couch_changes'
diff --git a/lib/tapicero/config.rb b/lib/tapicero/config.rb
new file mode 100644
index 0000000..cb22792
--- /dev/null
+++ b/lib/tapicero/config.rb
@@ -0,0 +1,69 @@
+require 'yaml'
+
+module Tapicero
+  module Config
+    extend self
+
+    attr_accessor :users_db_name
+    attr_accessor :db_prefix
+    attr_accessor :couch_connection
+
+    def self.load(base_dir, *configs)
+      configs.each do |file_path|
+        load_config find_file(base_dir, file_path)
+      end
+    end
+
+    # TODO: enable username and password
+    def couch_host
+      couch_connection[:protocol] + '://' +
+        couch_connection[:host] + ':' +
+        couch_connection[:port] + '/'
+    end
+
+    private
+
+    def load_config(file_path)
+      return unless file_path
+      puts " * Loading configuration #{file_path}"
+      load_settings YAML.load(File.read(file_path))
+    rescue NoMethodError => exc
+      STDERR.puts "ERROR in file #{file_path}"
+      exit(1)
+    end
+
+    def load_settings(hash)
+      return unless hash
+      hash.each do |key, value|
+        apply_setting(key, value)
+      end
+    end
+
+    def apply_setting(key, value)
+      if value.is_a? Hash
+        value = symbolize_keys(value)
+      end
+      self.send("#{key}=", value)
+    rescue NoMethodError => exc
+      STDERR.puts "'#{key}' is not a valid option"
+      raise exc
+    end
+
+    def self.symbolize_keys(hsh)
+      newhsh = {}
+      hsh.keys.each do |key|
+        newhsh[key.to_sym] = hsh[key]
+      end
+      newhsh
+    end
+
+    def self.find_file(base_dir, file_path)
+      return nil unless file_path
+      if defined? CWD
+        return File.expand_path(file_path, CWD)  if File.exists?(File.expand_path(file_path, CWD))
+      end
+      return File.expand_path(file_path, base_dir) if File.exists?(File.expand_path(file_path, base_dir))
+      return nil
+    end
+  end
+end
diff --git a/lib/tapicero/couch_changes.rb b/lib/tapicero/couch_changes.rb
new file mode 100644
index 0000000..2e668dc
--- /dev/null
+++ b/lib/tapicero/couch_changes.rb
@@ -0,0 +1,19 @@
+module Tapicero
+  class CouchChanges
+    def initialize(stream)
+      @stream = stream
+    end
+
+    def last_seq
+      @stream.get "_changes", :limit => 1, :descending => true do |hash|
+        return hash[:last_seq]
+      end
+    end
+
+    def follow
+      @stream.get "_changes", :feed => :continuous, :since => last_seq do |hash|
+        yield(hash)
+      end
+    end
+  end
+end
diff --git a/lib/tapicero/couch_stream.rb b/lib/tapicero/couch_stream.rb
new file mode 100644
index 0000000..96c752a
--- /dev/null
+++ b/lib/tapicero/couch_stream.rb
@@ -0,0 +1,25 @@
+require 'yajl/http_stream'
+
+module Tapicero
+  class CouchStream
+    def initialize(database_url)
+      @database_url = database_url
+    end
+
+    def get(path, options)
+      url = url_for(path, options)
+      # puts url
+      Yajl::HttpStream.get(url, :symbolize_keys => true) do |hash|
+        yield(hash)
+      end
+    end
+
+    protected
+
+    def url_for(path, options = {})
+      url = [@database_url, path].join('/')
+      url += '?' if options.any?
+      url += options.map {|k,v| "#{k}=#{v}"}.join('&')
+    end
+  end
+end
diff --git a/lib/tapicero_daemon.rb b/lib/tapicero_daemon.rb
new file mode 100644
index 0000000..a5347c7
--- /dev/null
+++ b/lib/tapicero_daemon.rb
@@ -0,0 +1,25 @@
+#
+# This file should not be required directly. Use the wrapper in /bin
+# instead or run this using daemons:
+#
+#  Daemons.run('tapicero_daemon.rb')
+#
+
+require 'tapicero'
+
+module Tapicero
+  puts    " * Tracking #{Config.users_db_name}"
+  couch   = CouchStream.new(Config.couch_host + Config.users_db_name)
+  changes = CouchChanges.new(couch)
+
+  # fill the pool
+  # pool.fill
+
+  # watch for deletions, fill the pool whenever it gets low
+  changes.follow do |hash|
+    if hash[:created]
+      puts " Created #{hash.inspect}"
+      # pool.fill
+    end
+  end
+end