noah 0.0.5 → 0.0.9

data/lib/noah/ephemeral_routes.rb

@@ -0,0 +1,19 @@
+class Noah::App
+  # Stubbing Ephemeral endpoints
+  get '/e/*' do
+    # Some logic to handle splats for ephemerals
+    # Eventually I'll move to root path
+  end
+
+  put '/e/*/watch' do
+    # Logic for adding watches to ephemerals
+  end
+
+  put '/e/*' do
+    # Some logic for creating ephemerals
+  end
+
+  delete '/e/*' do
+    # See previous two entries
+  end
+end

data/lib/noah/helpers.rb

@@ -1,56 +1,54 @@
-require File.join(File.dirname(__FILE__), 'models')
 module Noah
   module SinatraHelpers
-  extend(Ohm)
 
     def host(opts = {})
-     Host.find(opts).first
+      Noah::Host.find(opts).first
     end
 
     def hosts(opts = {})
-      Hosts.all(opts)
+      Noah::Hosts.all(opts)
     end
 
     def service(opts = {})
-      Service.find(options)
+      Noah::Service.find(options)
     end
 
     def services(opts = {})
-      Services.all(opts)
+      Noah::Services.all(opts)
     end 
 
     def host_service(hostname, servicename)
-      h = Host.find(:name => hostname).first
+      h = Noah::Host.find(:name => hostname).first
       if h.nil?
         nil
       else  
-        Service.find(:host_id => h.id, :name => servicename).first
+        Noah::Service.find(:host_id => h.id, :name => servicename).first
       end  
     end
 
     def host_services(hostname)
-      h = Host.find(:name => hostname).first
+      h = Noah::Host.find(:name => hostname).first
       if h.nil?
         nil
       else  
-        Services.all(:host_id => id)
+        Noah::Services.all(:host_id => id)
       end  
     end 
 
     def application(opts = {})
-      Application.find(opts).first
+      Noah::Application.find(opts).first
     end
 
     def applications(opts = {})
-      Applications.all(opts)
+      Noah::Applications.all(opts)
     end
 
     def configuration(opts = {})
-      Configuration.find(opts).first
+      Noah::Configuration.find(opts).first
     end
 
     def configurations(opts = {})
-      Configurations.all(opts)
+      Noah::Configurations.all(opts)
     end
   end
 

data/lib/noah/host_routes.rb

@@ -0,0 +1,69 @@
+class Noah::App
+# Host URIs
+
+  # GET named {Service} for named {Host}
+  get '/h/:hostname/:servicename/?' do |hostname, servicename|
+    h = host_service(hostname, servicename)
+    if h.nil?
+      halt 404
+    else
+      h.to_json
+    end
+  end
+
+  # GET named {Host}
+  # @param :hostname name of {Host}
+  # @return [JSON] representation of {Host}
+  get '/h/:hostname/?' do |hostname|
+    h = host(:name => hostname)
+    if h.nil?
+      halt 404
+    else
+      h.to_json
+    end
+  end
+
+  # GET all {Hosts}
+  get '/h/?' do
+    hosts.map {|h| h.to_hash}
+    if hosts.size == 0
+      halt 404
+    else
+      hosts.to_json
+    end
+  end
+
+  put '/h/:hostname/watch' do |hostname|
+    required_params = ["endpoint"]
+    data = JSON.parse(request.body.read)
+    (data.keys.sort == required_params.sort) ? (h = Noah::Host.find(:name => hostname).first) : (raise "Missing Parameters")
+    h.nil? ? (halt 404) : (w = h.watch!(:endpoint => data['endpoint']))
+    w.to_json
+  end
+
+  put '/h/:hostname/?' do |hostname|
+    required_params = ["name", "status"]
+    data = JSON.parse(request.body.read)
+    (data.keys.sort == required_params.sort && data['name'] == hostname) ? (host = Noah::Host.find_or_create(:name => data['name'], :status => data['status'])) : (raise "Missing Parameters")
+    if host.valid?
+      r = {"result" => "success","id" => "#{host.id}","status" => "#{host.status}", "name" => "#{host.name}", "new_record" => host.is_new?}
+      r.to_json
+    else
+      raise "#{host.errors}"
+    end
+  end
+
+  delete '/h/:hostname/?' do |hostname|
+    host = Noah::Host.find(:name => hostname).first
+    if host
+      services = []
+      Noah::Service.find(:host_id => host.id).sort.each {|x| services << x; x.delete} if host.services.size > 0
+      host.delete
+      r = {"result" => "success", "id" => "#{host.id}", "name" => "#{hostname}", "service_count" => "#{services.size}"}
+      r.to_json
+    else
+      halt 404
+    end
+  end
+
+end

data/lib/noah/models.rb

@@ -1,5 +1,86 @@
-require File.join(File.dirname(__FILE__),'hosts')
-require File.join(File.dirname(__FILE__),'services')
-require File.join(File.dirname(__FILE__),'applications')
-require File.join(File.dirname(__FILE__),'configurations')
-require File.join(File.dirname(__FILE__),'watchers')
+require 'ohm'
+require 'ohm/contrib'
+module Noah
+  class Model < Ohm::Model
+    def self.inherited(model)
+
+      model.send :include, Ohm::Timestamping
+      model.send :include, Ohm::Typecast
+      model.send :include, Ohm::Callbacks
+      model.send :include, Ohm::ExtraValidations
+
+      # removing this as it's simply redundant
+      # model.after :save, :notify_via_redis_save
+      model.after :create, :notify_via_redis_create
+      model.after :update, :notify_via_redis_update
+      model.before :delete, :stash_name
+      model.after :delete, :notify_via_redis_delete
+      model.send :include, ModelClassMethods
+    end
+  end
+
+  module ModelClassMethods
+
+    def is_new?
+      self.created_at == self.updated_at
+    end
+
+
+    def watch!(opts={:endpoint => nil, :pattern => nil})
+      base_pattern = "#{self.patternize_me}"
+      opts[:endpoint].nil? ? (raise ArgumentError, "Need an endpoint") : endpoint=opts[:endpoint]
+      opts[:pattern].nil? ? pattern=base_pattern : pattern=opts[:pattern]
+
+      begin
+        w = Watcher.new :pattern => pattern, :endpoint => endpoint
+        w.valid? ? w.save : (raise "#{w.errors}")
+        w.name
+      rescue Exception => e
+        e.message
+      end
+    end
+      
+    protected
+    def patternize_me
+      "//noah/#{self.class_to_lower}/#{name}"
+    end
+
+    def stash_name
+      @deleted_name = self.name
+    end
+
+    def class_to_lower
+      self.class.to_s.gsub(/(.*)::(\w)/,'\2').downcase
+    end
+    def dbnum
+      o = Ohm.options.first
+      return "0" if o.nil?
+      return "0" if (o[:db].nil? && o[:url].nil?)
+      o[:db].nil? ? "#{o[:url].split('/').last}" : "#{o[:db]}"
+    end
+
+    ["create", "update", "delete"].each do |meth|
+      class_eval do
+        define_method("notify_via_redis_#{meth}".to_sym) do
+          db = self.dbnum
+          self.name.nil? ? name=@deleted_name : name=self.name
+          # Pulling out dbnum for now. Need to rethink it
+          #pub_category = "#{db}:noah.#{self.class.to_s}[#{name}].#{meth}"
+          pub_category = "#{self.patternize_me}"
+          Ohm.redis.publish(pub_category, self.to_hash.merge({"action" => meth, "pubcategory" => pub_category}).to_json)
+
+          # The following provides a post post-action hook. It allows a class to provide it's own handling after the fact
+          # good example is in [Noah::Ephemeral] where it's used to check for/clean up expired ephemeral nodes entries
+          self.send("#{meth}_hook".to_sym) unless self.protected_methods.member?("#{meth}_hook".to_sym) == false
+        end
+      end
+    end
+
+  end
+end
+require File.join(File.dirname(__FILE__), 'models','hosts')
+require File.join(File.dirname(__FILE__), 'models','services')
+require File.join(File.dirname(__FILE__), 'models','applications')
+require File.join(File.dirname(__FILE__), 'models','configurations')
+require File.join(File.dirname(__FILE__), 'models','watchers')
+require File.join(File.dirname(__FILE__), 'models','ephemerals')

data/lib/noah/models/applications.rb

@@ -0,0 +1,41 @@
+require File.join(File.dirname(__FILE__), 'configurations')
+module Noah
+  class Application < Model
+    attribute :name
+    collection :configurations, Configuration
+
+    index :name
+
+    def validate
+      super
+      assert_present :name
+    end
+
+    def to_hash
+      arr = []
+      configurations.sort.each {|c| arr << c.to_hash}
+      super.merge(:name => name, :created_at => created_at, :updated_at => updated_at, :configurations => arr)
+    end
+
+    class << self
+    def find_or_create(opts = {})
+      begin
+        find(opts).first.nil? ? (app = create(opts)) : (app = find(opts).first)
+        if app.valid?
+          app.save
+        end
+        app
+      rescue Exception => e
+        e.message
+      end
+    end
+    end
+
+  end
+
+  class Applications
+    def self.all(options = {})
+      options.empty? ? Application.all.sort : Application.find(options).sort
+    end
+  end
+end

data/lib/noah/models/configurations.rb

@@ -0,0 +1,49 @@
+module Noah
+  class Configuration < Model
+
+    attribute :name
+    attribute :format
+    attribute :body
+    attribute :new_record
+    reference :application, Application
+
+    index :name
+    index :format
+    index :body
+
+    def validate
+      super
+      assert_present :name
+      assert_present :format
+      assert_present :body
+      assert_present :application_id
+      assert_unique [:name, :application_id]
+    end
+
+    def to_hash
+      Application[application_id].nil? ? app_name=nil : app_name=Application[application_id].name
+      super.merge(:name => name, :format => format, :body => body, :created_at => created_at, :updated_at => updated_at, :application => app_name)
+    end
+
+    class << self
+    def find_or_create(opts={})
+      begin
+        if find(opts).first.nil?
+          conf = create(opts)
+        else  
+          conf = find(opts).first
+        end  
+      rescue Exception => e
+        e.message
+      end
+    end
+    end
+
+  end
+
+  class Configurations
+    def self.all(options = {})
+      options.empty? ? Configuration.all.sort : Configuration.find(options).sort
+    end
+  end 
+end

data/lib/noah/models/ephemerals.rb

@@ -0,0 +1,33 @@
+require 'digest/sha1'
+module Noah
+  class Ephemeral < Model #NYI
+
+    attribute :path
+    attribute :data
+
+    index :path
+
+    def validate
+      super
+      assert_present :path
+      assert_unique :path
+    end
+
+    def name
+      @name = path
+    end
+
+    protected
+    def save_hook
+      # called after any create,update,delete
+      # logic needed to expire any orphaned ephemerals
+    end
+
+    private
+    def path_protected?(path_part)
+      # Check for protected paths in ephemeral nodes
+    end
+      
+  end
+
+end

data/lib/noah/models/hosts.rb

@@ -0,0 +1,56 @@
+require File.join(File.dirname(__FILE__), 'services')
+module Noah
+  class Host < Model
+    # Host model
+    # @return {Host} a {Host} object
+
+    attribute :name
+    attribute :status
+    collection :services, Service
+
+    index :name
+    index :status
+
+    def validate
+      super
+      assert_present :name
+      assert_present :status
+      assert_unique :name
+      assert_member :status, ["up","down","pending"]
+    end
+
+    # @return [Hash] A hash representation of a {Host}
+    def to_hash
+      arr = []
+      services.sort.each {|s| arr << s.to_hash}
+      h = {:name => name, :status => status, :created_at => created_at, :updated_at => updated_at, :services => arr}
+      super.merge(h)
+    end
+
+    class << self
+    def find_or_create(opts = {})
+      begin
+        # exclude requested status from lookup
+        h = find(opts.reject{|key,value| key == :status}).first
+        host = h.nil? ? create(opts) : h
+        host.status = opts[:status]
+        if host.valid?
+          host.save
+        end
+        host
+      rescue Exception => e
+        e.message
+      end
+    end
+    end
+
+  end
+
+  class Hosts
+    # @param [Hash] optional filters for results
+    # @return [Array] Array of {Host} objects
+    def self.all(options = {})
+      options.empty? ? Noah::Host.all.sort : Noah::Host.find(options).sort
+    end
+  end
+end

data/lib/noah/models/services.rb

@@ -0,0 +1,54 @@
+module Noah
+
+  class Service < Model
+
+    attribute :name
+    attribute :status
+    reference :host, Host
+
+    index :name
+    index :status
+
+    def validate
+      super
+      assert_present :name
+      assert_present :status
+      assert_present :host_id
+      assert_unique [:name, :host_id]
+      assert_member :status, ["up", "down", "pending"]
+    end
+
+    def to_hash
+      Host[host_id].nil? ? host_name=nil : host_name=Host[host_id].name
+      super.merge(:name => name, :status => status, :updated_at => updated_at, :host => host_name)
+    end
+
+    class << self
+      def find_or_create(opts = {})
+        begin
+          # convert passed host object to host_id if passed
+          if opts.has_key?(:host)
+            opts.merge!({:host_id => opts[:host].id})
+            opts.reject!{|key, value| key == :host}
+          end  
+          # exclude requested status from lookup
+          s = find(opts.reject{|key,value| key == :status}).first
+          service = s.nil? ? create(opts) : s
+          service.status = opts[:status]
+          if service.valid?
+            service.save
+          end
+          service
+        rescue Exception => e
+          e.message
+        end
+      end
+    end
+  end
+
+  class Services
+    def self.all(options = {})
+      options.empty? ? Service.all.sort : Service.find(options).sort
+    end
+  end
+end