client_management/lib/tasks/change_site_server_name.rake

51 lines
2.1 KiB
Ruby

require 'net/ssh'
require 'pathname'
require 'json'
namespace :create_site do
desc "Change Site Server Name"
task :change_site_server_name,[:id,:server_name,:port] => :environment do |task,args|
begin
@site_construct = SiteConstruct.find(args.id)
site_server = @site_construct.site_server
if !site_server.nil?
@password = site_server.password
update_infos("Starting change domain name and ports.")
Net::SSH.start(site_server.ip , site_server.account , password: site_server.password) do |ssh|
@site_construct.update(:domain_name=>args.server_name)
if args.port.present?
all_ports = args.port.split('////')
@site_construct.update(:port=> all_ports )
end
change_construct_nginx(ssh,@site_construct)
@site_construct.update(:status=>'finish')
end
end
rescue => e
update_infos(e.to_s)
@site_construct.update(:status=>'error')
puts [e,e.backtrace]
end
end
def change_construct_nginx(ssh,site_construct,update_multithread=false)
@is_multithread = update_multithread
auto_update_infos("Reading setting file...")
nginx_file_content = exec_command_by_user(ssh,"cat #{site_construct.nginx_file}")
nginx_file_content = site_construct.generate_nginx_text(nginx_file_content)
auto_update_infos("Writing...")
cmd = "x='#{nginx_file_content.gsub("\\$","$").gsub(/'{1,3}/,"\"\'\"")}'; echo '#{@password}' | sudo -S sh -c \"echo '$x' > #{site_construct.nginx_file}\""
exec_command_by_user(ssh,cmd)
exec_ssh_command_by_sudo(ssh,"service nginx restart")
auto_update_infos("Finish!")
end
def exec_command_by_user(session,command)
output = session.exec!(command)
return output[0...-1].gsub(/^\n[\n]+/,'')
end
def exec_ssh_command_by_sudo(session,command)
output = session.exec!("echo '#{@password}' | sudo -S #{command}")
if output.include?("sudo:") && output.include?("command not found")
output = session.exec!(command)
end
return output.encode!("UTF-8", :invalid => :replace, :undef => :replace, :replace => '')
end
end