class Sfp::Module::WordpressDB
	include Sfp::Resource

	ConfigFile = '/tmp/sfp.module.wordpressdb.config'

	def update_state
		if ::File.exist?(ConfigFile)
			config = JSON[::File.read(ConfigFile)]
			root_password = resolve(config['mysql'] + '.root_password')
			data = `echo "show databases;" | mysql -u root --password=#{root_password} mysql`.strip.split("\n")
			@state['installed'] = !!data.index(config['db_name'])
			@state['mysql'] = (@state['installed'] ? config['mysql'] : @model['mysql'])
			@state['db_name'] = config['db_name']
			@state['db_user'] = config['db_user']
			@state['db_password'] = config['db_password']
		else
			@state['installed'] = false
			@state['mysql'] = @model['mysql']
			@state['db_name'] = @model['db_name']
			@state['db_user'] = @model['db_user']
			@state['db_password'] = @model['db_password']
		end
	end

	def install(p={})
		return false if @model['mysql'].nil?

		root_password = resolve(@model['mysql'] + '.root_password')

		# install database
		mysql_cmd = "mysql -u root --password=#{root_password}"
		sql1 = "CREATE USER '#{@model['db_user']}'@'%' IDENTIFIED BY '#{@model['db_password']}';"
		system("echo \"#{sql1}\" | #{mysql_cmd}")
		sql2 = "DROP DATABASE IF EXISTS #{@model['db_name']}; CREATE DATABASE #{@model['db_name']};"
		return false if not system("echo \"#{sql2}\" | #{mysql_cmd}")
		sql3 = "GRANT ALL PRIVILEGES ON #{@model['db_name']}.* TO '#{@model['db_user']}'@'%';"
		return false if not system("echo \"#{sql3}\" | #{mysql_cmd}")
		sql4 = "FLUSH PRIVILEGES;"
		return false if not system("echo \"#{sql4}\" | #{mysql_cmd}")

		config = { 'mysql' => @model['mysql'],
		           'db_name' => @model['db_name'],
		           'db_user' => @model['db_user'],
		           'db_password' => @model['db_password'] }
		::File.open(ConfigFile, 'w') { |f|
			f.write(JSON.generate(config))
			f.flush
		}

		true
	end

	def uninstall(p={})
		return false if not ::File.exist?(ConfigFile)

		config = JSON[::File.read(ConfigFile)]

		root_password = resolve(config['mysql'] + '.root_password')
		# uninstall database
		mysql_cmd = "mysql -u root --password=#{root_password}"
		sql1 = "DROP USER '#{config['db_user']}'@'%';"
		sql2 = "DROP DATABASE IF EXISTS #{config['db_name']};"
		sql3 = "REVOKE ALL PRIVILEGES ON *.* FROM '#{config['db_user']}'@'%';"
		sql4 = "FLUSH PRIVILEGES;"
		system("echo \"#{sql1} #{sql2} #{sql3} #{sql4}\" | #{mysql_cmd}")

		::File.delete(ConfigFile) if ::File.exist?(ConfigFile)

		true
	end
end

class Sfp::Module::WordpressWeb
	include Sfp::Resource

	ConfigFile = '/tmp/sfp.module.wordpressweb.config'

	def update_state
		@state['installed'] = false
		@state['installed'] = ::File.exist?(@model['path'])

		if ::File.exist?(ConfigFile)
			config = JSON[::File.read(ConfigFile)]
			@state['http'] = config['http']
			@state['database'] = config['database']
			@state['path'] = config['path']
		else
			@state['http'] = nil
			@state['database'] = nil
			@state['path'] = @model['path']
		end
		@state['source_url'] = @model['source_url']
	end

	def install(p={})
		return false if p['database'].nil? or p['http'].nil? 

		db_address = resolve(p['database'] + '.parent.sfpAddress')
		return false if db_address.to_s.strip == ''
		db = resolve(p['database'])

		system('apt-get update; apt-get install -y wget') if `which wget`.strip.length <= 0
		return false if `which wget`.strip.length <= 0

		system("wget --output-document='/tmp/wp.tgz' #{@model['source_url']}")
		return false if not ::File.exist?('/tmp/wp.tgz')
		system("cd /tmp; tar xvzf wp.tgz")

		home_path = @model['path']
		wp_config_path = "#{home_path}/wp-config.php"
		wp_config_sample_path = "#{home_path}/wp-config-sample.php"

		system("rm -rf #{home_path}") if ::File.file?(home_path)
		Dir.mkdir(home_path, 0755) if not ::File.directory?(home_path)
		system("cd /tmp; mv -f wordpress/* #{home_path}/; mv -f wordpress/.* #{home_path}/;rmdir wordpress;rm -f wp.tgz")
		return false if not ::File.exist?(wp_config_sample_path)

		system("cp -f #{wp_config_sample_path} #{wp_config_path}")

		system("sed -i 's/.*DB_NAME.*/define(\"DB_NAME\",\"#{db['db_name']}\");/' #{wp_config_path}")
		system("sed -i 's/.*DB_USER.*/define(\"DB_USER\",\"#{db['db_user']}\");/' #{wp_config_path}")
		system("sed -i 's/.*DB_PASSWORD.*/define(\"DB_PASSWORD\",\"#{db['db_password']}\");/' #{wp_config_path}")
		system("sed -i 's/.*DB_HOST.*/define(\"DB_HOST\",\"#{db_address}\");/' #{wp_config_path}")

		system("cd #{home_path}; mv -f index.html index.html.bak")

		::File.open(ConfigFile, 'w') { |f|
			config = { 'http' => p['http'], 'database' => p['database'], 'path' => @model['path'] }
			f.write(JSON.generate(config))
			f.flush
		}

		true
	end

	def uninstall(p={})
		home_path = @model['path']

		# delete all files
		if @model['path'] != '/'
			system("rm -rf #{home_path}")
		else
			system("cd #{home_path}; mv -f index.html.bak /tmp; rm -rf *; mv -f /tmp/index.html.bak index.html")
		end

		::File.delete(ConfigFile) if ::File.exist?(ConfigFile)

		true
	end
end