nuri 0.5.1

data/modules/.gitignore

@@ -0,0 +1,4 @@
+*.tgz
+cell
+autonomy*
+idol*

data/modules/apache/apache.rb

@@ -0,0 +1,255 @@
+require 'augeas' # require package libaugeas-ruby (Ubuntu)
+
+require File.expand_path(File.dirname(__FILE__)) + '/../service/service.rb'
+
+class Sfp::Module::Apache < Sfp::Module::Service
+	include Sfp::Resource
+
+	ConfigFile = '/etc/apache2/sites-available/default'
+	SitesEnabledDir = '/etc/apache2/sites-enabled'
+	LoadBalancerConfigFile = '/etc/apache2/sites-enabled/load_balancer'
+	InstallingLockFile = '/tmp/sfp_apache_installing.lock'
+	NotRunningLockFile = '/tmp/sfp_apache_not_running.lock'
+
+	def update_state
+		# Call method 'update_state' of Sfp::Module::Service (superclass)
+		self.class.superclass.instance_method(:update_state).bind(self).call
+
+		@state['php_module'] = Sfp::Module::Package.installed?('libapache2-mod-php5')
+		@state['php_mysql_module'] = Sfp::Module::Package.installed?('php5-mysql')
+
+		if File.exist?(InstallingLockFile)
+			@state['installed'] = @state['running'] = false
+			@state['version'] = ''
+		else
+			@state['installed'] = Sfp::Module::Package.installed?('apache2')
+			@state['running'] = (File.exist?(NotRunningLockFile) ? false : Sfp::Module::Service.running?('apache2'))
+			@state['version'] = Sfp::Module::Package.version?('apache2').to_s
+		end
+
+		# port
+		data = (File.file?("/etc/apache2/ports.conf") ? `/bin/grep -e "^Listen " /etc/apache2/ports.conf` : "")
+		@state['port'] = (data.length > 0 ?	@state["port"] = data.split(' ')[1].to_i : 80)
+
+		# document root
+		data = (File.file?(ConfigFile) ? `/bin/grep -e "DocumentRoot " #{ConfigFile}` : "")
+		@state['document_root'] = (data.length > 0 ? data.strip.split(' ')[1] : '/var/www')
+
+		# server_name
+		data = (File.file?(ConfigFile) ? `/bin/grep -e "ServerName " #{ConfigFile}` : "")
+		@state['server_name'] = (data.length > 0 ? data.strip.split(' ')[1] : '')
+
+		# is_load_balancer
+		@state['is_load_balancer'] = File.exists?(LoadBalancerConfigFile)
+
+		# lb_members
+		members = []
+		if @state['is_load_balancer']
+			data =`grep "BalancerMember" #{LoadBalancerConfigFile} 2>/dev/null`.chop
+			agents = Sfp::Agent.get_agents
+			data.split("\n").each do |line|
+				member = line.strip.split(' ')
+				next if member[1] == nil
+				_, address = member[1].split('http://', 2)
+				agents.each { |k,v|
+					if v['sfpAddress'] == address
+						members << '$.' + k
+						break
+					end
+				}
+			end
+			members.sort!
+		end
+		@state['lb_members'] = members
+
+		# lb_method
+		if @state['is_load_balancer']
+			data = `grep 'ProxySet' #{LoadBalancerConfigFile} | grep 'lbmethod'`.strip.split("=")
+			@state['lb_method'] = data[1].to_s
+		else
+			@state['lb_method'] = ''
+		end
+	end
+
+	def install(p={})
+		begin
+			File.open(InstallingLockFile, 'w') { |f| f.write(' ') }
+			return (Sfp::Module::Package.install(@model['package_name']) and self.stop)
+		rescue Exception => e
+			Sfp::Agent.logger.error "#{e}\n#{e.backtrace.join("\n")}"
+		ensure
+			File.delete(InstallingLockFile) if File.exist?(InstallingLockFile)
+		end
+		false
+	end
+
+	def uninstall(p={})
+		begin
+			if Sfp::Module::Package.uninstall(@model['package_name'])
+				system('/bin/rm -rf /etc/apache2') if File.directory?('/etc/apache2')
+				return true
+			end
+		rescue Exception => e
+			Sfp::Agent.logger.error "#{e}\n#{e.backtrace.join("\n")}"
+		end
+		false
+	end
+
+	def set_port(p={})
+		return false if p['target'].nil?
+		port = p['target']
+		Augeas::open do |aug|
+			aug.set("/files/etc/apache2/ports.conf/*[self::directive='NameVirtualHost']/arg",
+				"*:#{port}")
+			aug.set("/files/etc/apache2/ports.conf/*[self::directive='Listen']/arg", port.to_s)
+			aug.set('/files/etc/apache2/sites-available/default/VirtualHost/arg', "*:#{port}")
+			return true if aug.save
+		end
+		false
+	end
+
+	def set_document_root(p={})
+		return false if not p.has_key?('target')
+		Augeas::open do |aug|
+			aug.set("/files/etc/apache2/sites-available/default/VirtualHost/*[self::directive='DocumentRoot']/arg", p['target'].to_s)
+			return true if aug.save
+		end
+		false
+	end
+
+	def set_server_name(p={})
+		return false if not p.has_key?('target')
+		server_name = p['target'].to_s
+		data = File.read(ConfigFile)
+		output = ""
+		data.split("\n").each do |line|
+			tuple = line.strip.split(' ')
+			if tuple[0] == 'ServerName'
+				# skip
+			elsif tuple[0] == 'DocumentRoot'
+				output += "#{line} \n"
+				output += "ServerName #{server_name}\n"
+			elsif line.strip != ''
+				output += "#{line} \n"
+			end
+		end
+		File.open(ConfigFile, 'w') { |f| f.write(output) }
+		true
+	end
+
+	def install_php_mysql_module(p={})
+		begin
+			File.open(NotRunningLockFile, 'w') { |f| f.write(' ') }
+			return self.stop if Sfp::Module::Package.install('php5-mysql')
+		rescue Exception => e
+			Sfp::Agent.logger.error e.to_s + "\n" + e.backtrace.join("\n")
+		ensure
+			File.delete(NotRunningLockFile) if File.exist?(NotRunningLockFile)
+		end
+		false
+	end
+
+	def uninstall_php_mysql_module(p={})
+		begin
+			File.open(NotRunningLockFile, 'w') { |f| f.write(' ') }
+			return self.stop if Sfp::Module::Package.uninstall('php5-mysql')
+		rescue Exception => e
+			Sfp::Agent.logger.error e.to_s + "\n" + e.backtrace.join("\n")
+		ensure
+			File.delete(NotRunningLockFile) if File.exist?(NotRunningLockFile)
+		end
+		false
+	end
+
+	def install_php_module(p={})
+		begin
+			File.open(NotRunningLockFile, 'w') { |f| f.write(' ') }
+			return self.stop if Sfp::Module::Package.install('libapache2-mod-php5')
+		rescue Exception => e
+			Sfp::Agent.logger.error e.to_s + "\n" + e.backtrace.join("\n")
+		ensure
+			File.delete(NotRunningLockFile) if File.exist?(NotRunningLockFile)
+		end
+		false
+	end
+
+	def uninstall_php_module(p={})
+		begin
+			File.open(NotRunningLockFile, 'w') { |f| f.write(' ') }
+			return self.stop if Sfp::Module::Package.uninstall('libapache2-mod-php5')
+		rescue Exception => e
+			Sfp::Agent.logger.error e.to_s + "\n" + e.backtrace.join("\n")
+		ensure
+			File.delete(NotRunningLockFile) if File.exist?(NotRunningLockFile)
+		end
+		false
+	end
+
+	# Load Balancer methods
+	#
+	def enable_load_balancer(p={})
+		begin
+			File.open(InstallingLockFile, 'w') { |f| f.write(' ') }
+			template_file = File.expand_path(File.dirname(__FILE__)) + "/load_balancer"
+			exec_seq 'a2enmod proxy',
+				'a2enmod proxy_balancer',
+				'a2enmod proxy_http',
+				'a2enmod status',
+				"rm -rf #{SitesEnabledDir}/*",
+				"cp -f #{template_file} #{LoadBalancerConfigFile}",
+				"sudo service #{@model['service_name']} stop"
+			return true
+		rescue Exception => e
+			Sfp::Agent.logger.error "#{e}\n#{e.backtrace.join("\n")}"
+		ensure
+			File.delete(InstallingLockFile) if File.exist?(InstallingLockFile)
+		end
+		false
+	end
+
+	def disable_load_balancer(p={})
+		begin
+			File.open(InstallingLockFile, 'w') { |f| f.write(' ') }
+			exec_seq  'a2dismod proxy_balancer',
+				'a2dismod proxy_http',
+				'a2dismod proxy',
+				'a2dismod status',
+				"rm -f #{LoadBalancerConfigFile}",
+				"ln -sf #{ConfigFile} #{SitesEnabledDir}/default",
+				"sudo service #{@model['service_name']} stop"
+			return true
+		rescue Exception => e
+			Sfp::Agent.logger.error "#{e}\n#{e.backtrace.join("\n")}"
+		ensure
+			File.delete(InstallingLockFile) if File.exist?(InstallingLockFile)
+		end
+		false
+	end
+
+	def set_lb_method(p={})
+		return false if ['byrequests', 'bytraffic', 'bybusyness'].index(p['target']).nil?
+		return !!system("sed -i -e 's/^\\s*ProxySet.*lbmethod=.*/ProxySet lbmethod=#{p['target']}/' #{LoadBalancerConfigFile}")
+	end
+
+	def set_lb_members(p={})
+		lbmembers = p['members'].map { |m| m.sub! /^\$\./, '' }
+		members = reverses = ''
+		lbmembers.each do |m|
+			address = resolve("$.#{m}.sfpAddress")
+Sfp::Agent.logger.info "lb_member => #{m}:#{address}"
+			return false if not address.is_a?(String) or address.strip == ''
+			members += "\n\tBalancerMember http://#{address}" #agents[m]['sfpAddress']}"
+			reverses += "\n\tProxyPassReverse / http://#{address}" #agents[m]['sfpAddress']}"
+		end
+		output = ''
+		File.read(LoadBalancerConfigFile).each_line do |line|
+			head, _ = line.strip.split(' ', 2)
+			next if head == 'BalancerMember' or head == 'ProxyPassReverse'
+			output += "#{line}"
+			output += "#{members}\n" if head == 'ProxySet'
+			output += "#{reverses}\n" if head == '</Location>'
+		end
+		File.open(LoadBalancerConfigFile, 'w') { |f| f.write(output); f.flush }
+		true
+	end
+end

data/modules/apache/apache.rb.old

@@ -0,0 +1,167 @@
+require 'augeas'
+
+class Sfp::Module::Apache
+	include Sfp::Resource
+
+	ConfigFile = '/etc/apache2/sites-available/default'
+	InstallingLockFile = '/tmp/nuri_apache_installing'
+	NotRunningLockFile = '/tmp/nuri_apache_not_running'
+
+	def initialize
+		@mutex = Mutex.new
+	end
+
+	# get state of this module
+	def update_state
+		to_model
+
+		# php module
+		@state['php_module'] = Nuri::Helper::Package.installed?('libapache2-mod-php5')
+				# php_mysql module
+				@state['php_mysql_module'] = Nuri::Helper::Package.installed?('php5-mysql')
+
+				# apache2 installed & running
+				if File.exist?(InstallingLockFile)
+					@state['installed'] = @state['running'] = false
+					@state['version'] = ''
+				else
+					@state['installed'] = Nuri::Helper::Package.installed?('apache2')
+					@state['version'] = Nuri::Helper::Package.version?('apache2')
+					@state['running'] = (File.exist?(NotRunningLockFile) ? false : Nuri::Helper::Service.running?('apache2'))
+				end
+
+				# port
+				data = (File.file?("/etc/apache2/ports.conf") ? `/bin/grep -e "^Listen " /etc/apache2/ports.conf` : "")
+				@state['port'] = (data.length > 0 ?	@state["port"] = data.split(' ')[1].to_i : 0)
+
+				# document root
+				data = (File.file?(ConfigFile) ? `/bin/grep -e "DocumentRoot " #{ConfigFile}` : "")
+				@state['document_root'] = (data.length > 0 ? data.strip.split(' ')[1] : '')
+
+				# ServerName
+				data = (File.file?(ConfigFile) ? `/bin/grep -e "ServerName " #{ConfigFile}` : "")
+				@state['server_name'] = (data.length > 0 ? data.strip.split(' ')[1] : '')
+
+				return @state
+			end
+
+			def install
+				begin
+					File.open(InstallingLockFile, 'w') { |f| f.write(' ') }
+					return (Nuri::Helper::Package.install('apache2') and
+							Nuri::Helper::Service.stop('apache2'))
+				rescue
+					return false
+				ensure
+					File.delete(InstallingLockFile) if File.exist?(InstallingLockFile)
+				end
+			end
+		
+			def uninstall
+				begin
+					Nuri::Helper::Package.uninstall('apache2')
+					system('/bin/rm -rf /etc/apache2') if File.directory?('/etc/apache2')
+					return true
+				rescue
+				end
+				return false
+			end
+		
+			def start
+				return Nuri::Helper::Service.start('apache2')
+			end
+		
+			def stop
+				return Nuri::Helper::Service.stop('apache2')
+			end
+		
+			def set_port(params=nil)
+				return false if params == nil
+				p = params['target']
+				Augeas::open do |aug|
+					aug.set("/files/etc/apache2/ports.conf/*[self::directive='NameVirtualHost']/arg",
+						"*:" + p.to_s)
+					aug.set("/files/etc/apache2/ports.conf/*[self::directive='Listen']/arg", p.to_s)
+					aug.set('/files/etc/apache2/sites-available/default/VirtualHost/arg', "*:" + p.to_s)
+					return true if aug.save
+				end
+				return false
+			end
+
+			def set_document_root(params={})
+				return false if not params.has_key?('target')
+				dir = params['target']
+				Augeas::open do |aug|
+					aug.set("/files/etc/apache2/sites-available/default/VirtualHost/*[self::directive='DocumentRoot']/arg", dir)
+					return true if aug.save
+				end
+				return false
+			end
+
+			def set_server_name(params={})
+				return false if not params.has_key?('target')
+				server_name = params['target']
+				data = File.read(ConfigFile)
+				output = ""
+				data.split("\n").each do |line|
+					tuple = line.strip.split(' ')
+					if tuple[0] == 'ServerName'
+						# skip
+					elsif tuple[0] == 'DocumentRoot'
+						output += "#{line} \n"
+						output += "ServerName #{server_name}\n"
+					elsif line.strip != ''
+						output += "#{line} \n"
+					end
+				end
+				File.open(ConfigFile, 'w') { |f| f.write(output) }
+
+				true
+			end
+
+			def install_php_mysql_module
+				begin
+					File.open(NotRunningLockFile, 'w') { |f| f.write(' ') }
+					return Nuri::Helper::Service.stop('apache2') if Nuri::Helper::Package.install('php5-mysql')
+					return false
+				rescue
+				ensure
+					File.delete(NotRunningLockFile) if File.exist?(NotRunningLockFile)
+				end
+			end
+
+			def uninstall_php_mysql_module
+				begin
+					File.open(NotRunningLockFile, 'w') { |f| f.write(' ') }
+					return Nuri::Helper::Service.stop('apache2') if Nuri::Helper::Package.uninstall('php5-mysql')
+					return false
+				rescue
+				ensure
+					File.delete(NotRunningLockFile) if File.exist?(NotRunningLockFile)
+				end
+			end
+
+			def install_php_module
+				begin
+					File.open(NotRunningLockFile, 'w') { |f| f.write(' ') }
+					return Nuri::Helper::Service.stop('apache2') if Nuri::Helper::Package.install('libapache2-mod-php5')
+					return false
+				rescue
+				ensure
+					File.delete(NotRunningLockFile) if File.exist?(NotRunningLockFile)
+				end
+			end
+
+			def uninstall_php_module
+				begin
+					File.open(NotRunningLockFile, 'w') { |f| f.write(' ') }
+					return Nuri::Helper::Service.stop('apache2') if Nuri::Helper::Package.uninstall('libapache2-mod-php5')
+					return false
+				rescue
+				ensure
+					File.delete(NotRunningLockFile) if File.exist?(NotRunningLockFile)
+				end
+			end
+		end
+	end
+end

data/modules/apache/apache.sfp

@@ -0,0 +1,146 @@
+include "../service/service.sfp"
+include "../node/node.sfp"
+
+schema Apache extends Service {
+	package_name = "apache2"
+	service_name = "apache2"
+	port : Int
+	document_root : String
+	installed : Bool
+	running : Bool
+	php_module : Bool
+	php_mysql_module : Bool
+	server_name : String
+
+	// load balancer config
+	is_load_balancer : Bool
+	lb_members isset Node
+	lb_method : String  // byrequests, bytraffic, bybusyness
+
+	sub set_port(target isref Integer) {
+		condition {
+			this.installed = true
+			this.running = false
+		}
+		effect {
+			this.port = target
+		}
+	}
+
+	synchronized sub uninstall {
+		condition {
+			this.installed = true
+			this.running = false
+			this.php_module = false
+			this.php_mysql_module = false
+		}
+		effect {
+			this.installed = false
+		}
+	}
+
+	sub set_document_root (target isref String) {
+		condition {
+			this.running = false
+			this.installed = true
+		}
+		effect {
+			this.document_root = target
+		}
+	}
+
+	sub set_server_name (target isref String) {
+		condition {
+			this.running = false
+			this.installed = true
+		}
+		effect {
+			this.server_name = target
+		}
+	}
+
+	synchronized sub install_php_module {
+		condition {
+			this.installed = true
+			this.running = false
+		}
+		effect {
+			this.php_module = true
+		}
+	}
+
+	synchronized sub uninstall_php_module {
+		condition {
+			this.running = false
+		}
+		effect {
+			this.php_module = false
+		}
+	}
+
+	synchronized sub install_php_mysql_module {
+		condition {
+			this.php_module = true
+			this.installed = true
+			this.running = false
+		}
+		effect {
+			this.php_mysql_module = true
+		}
+	}
+
+	synchronized sub uninstall_php_mysql_module {
+		condition {
+			this.running = false
+		}
+		effect {
+			this.php_mysql_module = false
+		}
+	}
+
+	sub enable_load_balancer {
+		condition {
+			this.installed = true
+			this.running = false
+		}
+		effect {
+			this.is_load_balancer = true
+		}
+	}
+
+	sub disable_load_balancer {
+		condition {
+			this.installed = true
+			this.running = false
+			this.is_load_balancer = true
+		}
+		effect {
+			this.is_load_balancer = false
+		}
+	}
+
+	sub set_lb_members (members isset Node) {
+		condition {
+			this.installed = true
+			this.running = false
+			this.is_load_balancer = true
+			foreach (members as node) {
+				node.created = true
+			}
+		}
+		effect {
+			this.lb_members = members
+		}
+	}
+
+	sub set_lb_method (target isref String) {
+		condition {
+			this.installed = true
+			this.running = false
+			this.is_load_balancer = true
+		}
+		effect {
+			this.lb_method = target
+		}
+	}
+}