
local ctx=tfe.context

function Sleep2(n)
   local t0 = os.clock()
   while os.clock() - t0 <= n do end
end

if(tfe.get_current_stage() == "http_stage_test")
then
	Sleep2(1000)
end

if(tfe.get_current_stage() == "http_req_header")
then
    local req_headers_tab = tfe.req.get_headers() 
	for k, v in pairs(req_headers_tab) do  
		print(k,v)
		if k == "User-Agent" then
			tfe.req.set_header("user-agent", "curl-v1.1") 
		end 
		if k == "Pragma" then 
			tfe.req.set_header("pragma", "nil") 
		end 
	end
	--tfe.exit(0)
	tfe.req.set_header("x-tg-construct-by", "tfe")
	local method = tfe.req.get_method()
	print(method)
end

if(tfe.get_current_stage() == "http_req_uri")
then
	local req_uri=tfe.req.get_uri()
	if req_uri then
		tfe.req.set_uri("team")
	end
end

if(tfe.get_current_stage() == "http_resp_header")
then
    local resp_headers_tab = tfe.resp.get_headers() 
	for k, v in pairs(resp_headers_tab) do  
		print(k,v)
		if k == "Content-Type" then 
			tfe.resp.set_header("Content-Type", "utf8") 
		end 
		if k == "Expires" then 
			tfe.resp.set_header("Expires", "nil") 
		end 
	end
	local status_code = tfe.resp.get_status_code()
	print(status_code)
end

if(tfe.get_current_stage() == "http_req_body")
then
	local req_body = tfe.req.get_body_data()
	if req_body then
		tfe.req.set_body_data(req_body, " set req body")
	end
	len=#req_body
	return len
end

if(tfe.get_current_stage() == "http_resp_body")
then
	local resp_body = tfe.resp.get_body_data()
	if resp_body then
		tfe.resp.set_body_data(resp_body, " set resp body")
	end
end

