package client

import (
	"encoding/json"
	"github.com/easysoft/zentaoatf/src/model"
	"github.com/easysoft/zentaoatf/src/utils/vari"
	"io/ioutil"
	"net/http"
	"strings"
)

func Get(url string, params map[string]string) (string, bool) {
	client := &http.Client{}

	req, err := http.NewRequest("GET", url, nil)
	if err != nil {
		return "", false
	}

	req.Header.Set("cookie", vari.SessionVar+"="+vari.SessionId)

	q := req.URL.Query()
	q.Add(vari.SessionVar, vari.SessionId)
	if params != nil {
		for pkey, pval := range params {
			q.Add(pkey, pval)
		}

	}
	req.URL.RawQuery = q.Encode()

	resp, err := client.Do(req)
	if err != nil {
		return "", false
	}

	bodyStr, _ := ioutil.ReadAll(resp.Body)
	println(string(bodyStr))

	var bodyJson model.ZentaoResponse
	json.Unmarshal(bodyStr, &bodyJson)

	defer resp.Body.Close()

	status := bodyJson.Status
	if status == "" { // 非嵌套结构
		return string(bodyStr), true
	} else { // 嵌套结构
		dataStr := bodyJson.Data
		return dataStr, status == "pass"
	}
}

func PostJson(url string, params map[string]interface{}) (string, bool) {
	client := &http.Client{}

	bytesData, _ := json.Marshal(params)

	req, err := http.NewRequest("POST", url, strings.NewReader(string(bytesData)))
	if err != nil {
		return "", false
	}

	//req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
	req.Header.Set("Content-Type", "application/json;charset=UTF-8")
	req.Header.Set("cookie", vari.SessionVar+"="+vari.SessionId)

	resp, err := client.Do(req)
	if err != nil {
		return "", false
	}

	bodyStr, _ := ioutil.ReadAll(resp.Body)
	println(string(bodyStr))

	var bodyJson model.ZentaoResponse
	json.Unmarshal(bodyStr, &bodyJson)

	defer resp.Body.Close()

	status := bodyJson.Status
	if status == "" { // 非嵌套结构
		return string(bodyStr), true
	} else { // 嵌套结构
		dataStr := bodyJson.Data
		return dataStr, status == "pass"
	}
}

func PostStr(url string, params map[string]string) (string, bool) {
	client := &http.Client{}

	paramStr := ""
	idx := 0
	for pkey, pval := range params {
		if idx > 0 {
			paramStr += "&"
		}
		paramStr = paramStr + pkey + "=" + pval
		idx++
	}

	req, err := http.NewRequest("POST", url, strings.NewReader(paramStr))
	if err != nil {
		return "", false
	}

	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
	req.Header.Set("cookie", vari.SessionVar+"="+vari.SessionId)

	resp, err := client.Do(req)
	if err != nil {
		return "", false
	}

	bodyStr, _ := ioutil.ReadAll(resp.Body)
	println(string(bodyStr))

	var bodyJson model.ZentaoResponse
	json.Unmarshal(bodyStr, &bodyJson)

	defer resp.Body.Close()

	status := bodyJson.Status
	if status == "" { // 非嵌套结构
		return string(bodyStr), true
	} else { // 嵌套结构
		dataStr := bodyJson.Data
		return dataStr, status == "pass"
	}
}