announcement-test/temp_file/app/controllers/admin/playground_controller.rb

require "open3"
class Admin::PlaygroundController < OrbitAdminController
	before_filter :check_for_testers, :only => "index"
	layout "structure"
	include Admin::PlaygroundHelper
	before_action :clean_multithread_model, :only => :command
	
	def index
		modules = ["announcement", "faq", "ad_banner", "archive", "gallery", "web_resource"]
		@default_modules = modules.collect do |mod| 
			ma = ModuleApp.where(:key => mod).first
			[ma.title,mod]
		end
	end

	def clean_multithread_model
		mul = Multithread.where(key: 'playground').first
		mul.destroy if !mul.nil?
	end

	def generatefakedata
		ma = ModuleApp.where(:key => params[:module]).first rescue nil
		if !ma.nil?
			case ma.key 
			when "announcement"
				make_announcement_fake_data(ma, params[:number].to_i)
			when "faq"
				make_faq_fake_data(ma, params[:number].to_i)
			when "ad_banner"
				make_ad_banner_fake_data(ma, params[:number].to_i)
			when "archive"
				make_archive_fake_data(ma, params[:number].to_i)
			when "gallery"
				make_gallery_fake_data(ma, params[:number].to_i)
			when "web_resource"
				make_web_resource_fake_data(ma, params[:number].to_i)
			end
		end 
		render :json => {"success" => true}.to_json
	end

	def deletefakedata
		ma = params[:module]
		fd = FakeData.where(:module => ma).first rescue nil
		if !fd.nil?
			case ma
			when "announcement"
				delete_announcement_fake_data(ma)
			when "faq"
				delete_faq_fake_data(ma)
			when "ad_banner"
				delete_ad_banner_fake_data(ma)
			when "archive"
				delete_archive_fake_data(ma)
			when "gallery"
				delete_gallery_fake_data(ma)
			when "web_resource"
				delete_web_resource_fake_data(ma)
			end
		else
			render :json => {"success" => false}.to_json and return
		end
		render :json => {"success" => true}.to_json
	end

	def command
		response = {}
		Multithread.create(key: 'playground',status: {output: [],error: false,alive:true})
		case params[:command]
		when "clean"
			clean_assets
			response["success"] = true
		when "precompile"
			precompile
			response["success"] = true
		when "bundle"
			bundle_update
			response["success"] = true
		when "restart_server"
			restart_server
			response["success"] = true
		when "restart_unicorn"
			restart_unicorn(Rails.env)
			response["success"] = true
		when "switch_to_production"
			restart_unicorn("production")
			response["success"] = true
		when "switch_to_development"
			restart_unicorn("development")
			response["success"] = true
		else
			exec_other_command(params[:command])
			response["success"] = true
		end

		render :json => response.to_json
	end

	def console_output
		if params[:type] == "restart"
			render :json => {"success" => true}.to_json
			file = File.join(Rails.root,"tmp","restart_unicorn.sh")
			File.delete(file) if File.exists?(file)
		else
			mul = Multithread.where(key: 'playground').first
		  if !mul.nil?
        render :json => {"alive" => mul.status['alive'], "response" => mul.status['output'][params[:count].to_i..-1], "error" => mul.status['error']}.to_json
			  if !mul.status['alive']
			    mul.status['output'] = []
				  mul.status['error'] = false
				  mul.save
			  end
			else
        render :json => {"alive" => true, "response" => "finish", "error" => false}.to_json
      end
    end
	end

	private

	def precompile
		cmd = "RAILS_ENV=production bundle exec rake assets:precompile"
		exec_other_command(cmd)
	end

	def exec_other_command(command)
  	mul = Multithread.where(key: 'playground').first
  	Thread.new do
  		begin
  			Bundler.with_clean_env do
  				IO.popen(command,:err=>[:child, :out]) do |stdout|
  					stdout.each do |line|
	  		    		l = line.chomp
	  		    		mul.status['output'] << l
	  		    		if l == "rake aborted!"
	  	      				mul.status['error'] = true
	  	      			end
	  	      			mul.save
  		    		end
  				end
	  			mul.status['alive'] = false
	  		    mul.save
				if command == 'bundle update'
					system('sleep 2')
				    Bundler.with_clean_env{system("UNICORN_PID=\"`fuser tmp/pids/unicorn.sock tmp/sockets/unicorn.sock tmp/unicorn.sock` `cat tmp/pids/unicorn.pid `\" && kill -s USR2 $UNICORN_PID ; n=20; while (kill -0 $UNICORN_PID > /dev/null 2>&1) && test $n -ge 0; do printf '.' && sleep 1 && n=$(( $n - 1 )); done ; if test $n -lt 0; then kill -s TERM $UNICORN_PID; sleep 3; bundle exec unicorn_rails -c config/unicorn.rb -D -E #{Rails.env}; else kill -s QUIT $UNICORN_PID; fi")}
				end
  			end
  		rescue => e
  			mul.status['alive'] = false
  			mul.status['output'] << e.inspect
  			mul.status['error'] = true
  			mul.save
  		end
  	end
	end

	def bundle_update
		cmd = "bundle update"
		exec_other_command(cmd)
	end

	def restart_server
		cmd ="UNICORN_PID=\"`fuser tmp/pids/unicorn.sock tmp/sockets/unicorn.sock tmp/unicorn.sock` `cat tmp/pids/unicorn.pid `\" && kill -s USR2 $UNICORN_PID ; n=20; while (kill -0 $UNICORN_PID > /dev/null 2>&1) && test $n -ge 0; do printf '.' && sleep 1 && n=$(( $n - 1 )); done ; if test $n -lt 0; then kill -s TERM $UNICORN_PID; sleep 3; bundle exec unicorn_rails -c config/unicorn.rb -D -E #{Rails.env}; else kill -s QUIT $UNICORN_PID; fi"
		exec_other_command(cmd)
	end

	def clean_assets
		cmd = "rm -r public/assets"
		exec_other_command(cmd)
	end

	def restart_unicorn(mode)
		mode = Rails.env if mode.nil?
		unicorn_rails = %x[which unicorn_rails].sub("\n",'')
		content = "UNICORN_PID=\"`fuser tmp/pids/unicorn.sock tmp/sockets/unicorn.sock tmp/unicorn.sock` `cat tmp/pids/unicorn.pid `\" && kill -s TERM $UNICORN_PID ; while (kill -0 $UNICORN_PID > /dev/null 2>&1) ; do printf '.' && sleep 1 ; done ; unset UNICORN_FD; bundle exec unicorn_rails -c config/unicorn.rb -D -E #{mode}"
		Thread.new do
			Bundler.with_clean_env{system(content)}
		end
	end

	def check_for_testers
		render_401 if !current_user.beta_tester && current_user.user_name != "rulingcom"
	end
end