processwanker 0.0.7

data/lib/config/config_client_host.rb

@@ -0,0 +1,152 @@
+############################################################################
+#
+# config_client_host.rb
+#
+# host("name") {
+# 	auth {
+# 		...
+# 	}
+# 	hostname		"hostname"
+# 	port				port
+# 	tag					"tag"
+# }
+# 
+#
+############################################################################
+
+require 'config_auth'
+require 'net_util'
+require 'config'
+
+module ProcessWanker
+	
+	############################################################################
+	#
+	#
+	#
+	############################################################################
+
+	class ConfigClientHost < ConfigNode
+		
+		attr_accessor					:name
+		attr_accessor					:hostname
+		attr_accessor					:port
+		attr_accessor					:tags
+		attr_accessor					:auth
+		
+		def initialize(name)
+			@name=name
+			@hostname=name
+			@tags={}
+			@port=NetUtil::DEFAULT_PORT
+		end
+		
+	  ############################################################################
+	  #
+	  #
+	  #
+	  ############################################################################
+	  
+		def matches_spec(spec)
+									
+			# ensure it's in array form
+			spec=spec.split(",")
+			
+			# check for inversion on first item
+			if(spec.first[0..0]=="~")
+				# insert implicit "all" at front
+				spec=["all"] + spec
+			end
+			
+			matches=false
+			spec.each do |p|
+				matches = matches_single(p,matches)
+			end
+			matches
+						
+		end
+	
+	  ############################################################################
+	  #
+	  #
+	  #
+	  ############################################################################
+	
+	  def matches_single(p,prev)
+		
+			if(p == "all")
+				return(true)
+			elsif(p[0..0] == "/")
+				r=Regexp.new(p.split("/")[1])
+				return(true) if(hostname =~ r)
+				return(true) if(name =~ r)
+			elsif(p[0..0] == "~")
+				return(prev && !matches_single(p[1..-1],false))
+			elsif(p[0..3] == "tag:")
+				tag=p[4..-1]
+	 			return(true) if(tags.include?(tag))
+			elsif(p == name)
+				return(true)
+			elsif(p == hostname)
+				return(true)
+			end
+			
+			prev
+		end
+	
+	end	
+	
+	############################################################################
+	#
+	#
+	#
+	############################################################################
+
+	class ConfigClientHostBuilder < Builder
+		
+		def build(container,args,block)
+			super(container,args,block)
+			@config.tags=@config.tags.keys.sort
+			@config
+		end
+		
+		def auth(&block)
+			@config.auth=Deferred.new(@config,[],block,ConfigAuthBuilder)
+		end
+
+		def hostname(v)
+			@config.hostname=v
+		end
+
+		def port(v)
+			@config.port=v.to_i
+		end
+		
+		def tags(*t)
+			t.each do |v|
+				@config.tags[v.to_s]=true
+			end
+		end
+	
+	end
+
+	############################################################################
+	#
+	#
+	#
+	############################################################################
+
+	############################################################################
+	#
+	#
+	#
+	############################################################################
+
+	############################################################################
+	#
+	#
+	#
+	############################################################################
+
+
+end

data/lib/config/config_daemon.rb

@@ -0,0 +1,94 @@
+############################################################################
+#
+# config_daemon.rb
+#
+# daemon {
+# 	listen_hostname			"0.0.0.0"
+# 	listen_port					45231
+#   log_file						"filename"
+# 	auth {
+# 		...
+# 	}
+#   services("group-name") {
+# 		...
+#   }
+# }
+#
+############################################################################
+
+require 'config_auth'
+require 'config_smtp'
+require 'config_daemon_services'
+require 'config'
+require 'config_hook'
+
+module ProcessWanker
+
+	############################################################################
+	#
+	#
+	#
+	############################################################################
+
+	class ConfigDaemon < ConfigNode
+		
+		DEFAULT_LISTEN_HOSTNAME="127.0.0.1"
+		
+		attr_accessor					:services	
+		attr_accessor					:auth
+		attr_accessor					:smtp
+		attr_accessor					:hooks
+		attr_accessor					:listen_hostname
+		attr_accessor					:listen_port
+		attr_accessor					:log_file
+		
+		def initialize()
+			@hooks=[]
+			@services={}
+			@listen_hostname=DEFAULT_LISTEN_HOSTNAME
+			@listen_port=NetUtil::DEFAULT_PORT
+		end
+		
+		
+	end
+	
+	############################################################################
+	#
+	#
+	#
+	############################################################################
+
+	class ConfigDaemonBuilder < Builder
+		
+		def auth(&block)
+			@config.auth=Deferred.new(@config,[],block,ConfigAuthBuilder)
+		end
+		
+		def hook(pattern,&block)
+			@config.hooks << ConfigHook.new(@config,pattern,block)
+		end
+		
+		def smtp(&block)
+			@config.smtp=Deferred.new(@config,[],block,ConfigSMTPBuilder)
+		end
+		
+		def listen_hostname(v)
+			@config.listen_hostname=v
+		end
+		
+		def listen_port(v)
+			@config.listen_port=v.to_i
+		end
+		
+		def log_file(v)
+			@config.log_file=v
+		end
+		
+		def services(name="default",&block)
+			@config.services[name]=Deferred.new(@config,[name],block,ConfigDaemonServicesBuilder)
+		end
+
+	end
+
+	
+end

data/lib/config/config_daemon_service.rb

@@ -0,0 +1,84 @@
+############################################################################
+#
+# config_daemon_service.rb
+#
+# *_service {
+# 	...
+# }
+#
+############################################################################
+
+require 'net_api'
+require 'config_daemon_service_dependency'
+require 'config'
+require 'config_hook'
+
+module ProcessWanker
+	
+	############################################################################
+	#
+	#
+	#
+	############################################################################
+
+	class ConfigDaemonService < ConfigNode
+		
+		attr_accessor					:klass
+		attr_accessor					:params
+		attr_accessor					:name
+		attr_accessor					:hooks
+
+		def initialize(name,klass)
+			@hooks=[]
+			@name=name
+			@klass=klass
+			@params={}
+		end
+		
+	end
+
+	############################################################################
+	#
+	#
+	#
+	############################################################################
+
+	class ConfigDaemonServiceBuilder < Builder
+
+		def hook(pattern,&block)
+			@config.hooks << ConfigHook.new(@config,pattern,block)
+		end
+		
+		def tags(*t)
+			@config.params[:tags] ||= []
+			@config.params[:tags] += t
+		end
+		
+		def depends(&block)
+			@config.params[:dependencies] ||= []
+			@config.params[:dependencies] << ConfigDaemonServiceDependencyBuilder.new.build(@config,[],block)
+		end
+		
+		def method_missing(method,*args,&block)
+			if(block)
+				@config.params[method] = block
+			elsif(args.length == 0)
+				@config.params[method] = true
+			elsif(args.length == 1)
+				@config.params[method] = args[0]
+			else
+				@config.params[method] = args
+			end
+		end
+	
+	
+	end
+
+	############################################################################
+	#
+	#
+	#
+	############################################################################
+
+
+end

data/lib/config/config_daemon_service_dependency.rb

@@ -0,0 +1,59 @@
+############################################################################
+#
+# config_daemon_service_dependency.rb
+#
+# depends {
+#   service		"spec"
+#   up_for		5
+# }
+#
+############################################################################
+
+require 'net_api'
+require 'config'
+
+module ProcessWanker
+	
+	############################################################################
+	#
+	#
+	#
+	############################################################################
+
+	class ConfigDaemonServiceDependency < ConfigNode
+		
+		attr_accessor					:service
+		attr_accessor					:up_for
+
+		def initialize()
+			@up_for=3
+		end
+		
+	end
+
+	############################################################################
+	#
+	#
+	#
+	############################################################################
+
+	class ConfigDaemonServiceDependencyBuilder < Builder
+
+		def service(v)
+			@config.service=v
+		end
+		
+		def up_for(v)
+			@config.up_for=v.to_i
+		end
+	
+	end
+
+	############################################################################
+	#
+	#
+	#
+	############################################################################
+
+
+end

data/lib/config/config_daemon_services.rb

@@ -0,0 +1,89 @@
+#!/usr/bin/env ruby
+############################################################################
+#
+# config_client_services.rb
+#
+# services("name") {
+#   *_service {
+#    ...
+# 	}
+# }
+#
+#
+#
+############################################################################
+
+require 'config_auth'
+require 'config_daemon_service'
+require 'config'
+require 'config_hook'
+
+module ProcessWanker
+
+	############################################################################
+	#
+	#
+	#
+	############################################################################
+
+	class ConfigDaemonServices < ConfigNode
+		
+		attr_accessor			:services
+		attr_accessor			:group_name
+		attr_accessor			:hooks
+		
+		def initialize(group_name)
+			@hooks=[]
+			@group_name=group_name
+			@services={}
+		end
+		
+	end
+	
+	############################################################################
+	#
+	#
+	#
+	############################################################################
+
+	class ConfigDaemonServicesBuilder < Builder
+
+		def build(container,args,block)
+			
+			# create construction methods for various service classes
+			ServiceMgr.instance.service_classes.each do |k,v|
+				str="def #{k}(name,&block) ; "
+				str << "@config.services[name] = ConfigDaemonServiceBuilder.new.build(@config,[name,#{v.name}],block) ; "
+				str << "end"
+				instance_eval(str)
+			end
+			
+			super(container,args,block)
+		end
+
+		def hook(pattern,&block)
+			@config.hooks << ConfigHook.new(@config,pattern,block)
+		end
+		
+
+#		def build(args,block)
+#			
+#			# create construction methods for various service classes
+#			ServiceMgr.instance.service_classes.each do |k,v|
+#				str="def #{k}(name,&block) ; "
+#				str << "@config.services << ConfigDaemonServiceBuilder.new.build(@config,[name],@default_group_name,#{v.name},block) ; "
+#				str << "end"
+#				instance_eval(str)
+#			end
+			
+#			@config=ConfigDaemonServices.new(*args)
+#			instance_eval(&block)
+#			@config
+#		end
+
+		
+	end
+
+	
+end
+

data/lib/config/config_hook.rb

@@ -0,0 +1,40 @@
+############################################################################
+#
+# config_hook.rb
+#
+# custom hooks
+#
+# hook("pattern") {
+# 	...
+# }
+#
+############################################################################
+
+require 'ipaddr'
+require 'openssl'
+require 'config'
+
+module ProcessWanker
+	
+	############################################################################
+	#
+	#
+	#
+	############################################################################
+
+	class ConfigHook < ConfigNode
+		
+		attr_accessor					:pattern
+		attr_accessor					:block
+		
+		def initialize(container,pattern,block)
+			@container=container
+			@pattern=pattern
+			@block=block
+		end
+		
+	end
+
+
+end
+

data/lib/config/config_node.rb

@@ -0,0 +1,160 @@
+#!/usr/bin/env ruby
+############################################################################
+#
+# config_node.rb
+#
+# config helper stuff
+#
+############################################################################
+
+module ProcessWanker
+
+	############################################################################
+	#
+	#
+	#
+	############################################################################
+
+	class ConfigNode
+		attr_accessor					:container
+
+		############################################################################
+		#
+		# find_attributes
+		#
+		# walk from the current object up to the root, looking for the named attribute
+		#
+		############################################################################
+		
+		def find_attributes(name)
+			r=[]
+			p=self
+			while(p)
+				if(p.respond_to?(name))
+					v=p.send(name)
+					r << v if(v)
+				end
+				break if(!p.respond_to?("container"))
+				p=p.container
+			end
+			r
+		end
+
+		# find the innermost auth block
+		def get_auth()
+			find_attributes("auth").first
+		end
+		
+	end
+
+	############################################################################
+	#
+	# Deferred
+	#
+	# there are situations where we don't wish to evaluate all of the
+	# configuration unless necessary - for instance, we don't want the daemon
+	# to evaluate the list of clusters in the client configuration (particularly
+	# if it involves querying the cloud).
+	#
+	# so, we wrap blocks in Deferred objects which are transparently evaluated
+	# on-demand.
+	#
+	############################################################################
+
+	class Deferred
+		
+		def initialize(deferred_container,deferred_args,deferred_block,deferred_builder)
+			@deferred_container=deferred_container
+			@deferred_args=deferred_args
+			@deferred_block=deferred_block
+			@deferred_builder=deferred_builder
+			@deferred_object=nil
+		end
+		
+		def method_missing(name,*args)
+			if(!@deferred_object)
+				@deferred_object=@deferred_builder.new.build(
+					@deferred_container,
+					@deferred_args,
+					@deferred_block)
+			end
+			@deferred_object.send(name,*args)
+		end
+		
+	end
+	
+	############################################################################
+	#
+	#
+	#
+	############################################################################
+
+	class Builder
+
+		def build(container,args,block)
+			@config=klass.new(*args)
+			@config.container=container
+			instance_eval(&block)
+			@config
+		end
+		
+		def klass
+			cn=self.class.name
+			cn.gsub!("Builder","")
+			cn=cn.split("::")[-1]
+			ProcessWanker::const_get(cn)
+		end
+		
+	end
+	
+	############################################################################
+	#
+	#
+	#
+	############################################################################
+
+	module Config
+
+    require 'erb'
+
+		@@config_path=ENV["PW_CFG"] || "/etc/pw/pw.cfg"
+	
+		def get_config_path()
+			@@config_path
+		end
+		module_function			:get_config_path
+	
+		def set_config_path(path)
+			@@config_path=path
+		end
+		module_function			:set_config_path
+		
+		def load_config(path)
+			ProcessWanker::loaded_config=nil
+			# load(path)
+			config = ERB.new(File.read(path)).result
+      eval(config)
+			ProcessWanker::loaded_config
+		end
+		module_function			:load_config
+		
+	end
+	
+
+	############################################################################
+	#
+	#
+	#
+	############################################################################
+
+	attr_accessor					:loaded_config
+	module_function				:loaded_config
+	module_function				:loaded_config=
+
+	def config(&block)
+		@loaded_config=ConfigurationBuilder.new.build(nil,[],block)
+	end
+	module_function			:config
+	
+	
+end