package fwh_pro

import (
	"encoding/json"
	"errors"
	"fmt"
	"fwh-toolbox-server/config"
	"fwh-toolbox-server/dto"
	"net/url"
	"strconv"
	"strings"
)

type SiteBaseInfoListPayload struct {
	Search        *string `json:"search,omitempty"`  // 只能搜索单条SN
	Serials       *string `json:"serials,omitempty"` // 可搜索多条SN，SN之间以‘,’隔开
	CountryId     *int    `json:"countryId,omitempty"`
	ProvinceId    *int    `json:"provinceId,omitempty"`
	CityId        *int    `json:"cityId,omitempty"`
	NetworkStatus *int    `json:"networkStatus,omitempty"`
	DeviceStatus  *int    `json:"deviceStatus,omitempty"`
	OnOffStatus   *int    `json:"onOffStatus,omitempty"`
	PageSize      uint    `json:"pageSize"`
	PageNum       uint    `json:"pageNum"`
	Lang          string  `json:"lang"`
}

type SiteInfoListPayload struct {
	Search     *string `json:"search,omitempty"` // 只能搜索单条SN
	CountryId  *int    `json:"countryId,omitempty"`
	ProvinceId *int    `json:"provinceId,omitempty"`
	CityId     *int    `json:"cityId,omitempty"`
	PageSize   uint    `json:"pageSize"`
	PageNum    uint    `json:"pageNum"`
	Lang       string  `json:"lang"`
}

type SiteModeInfoPayload struct {
	GatewayId string `json:"gatewayId"`
	ShowType  int    `json:"showType"`
	Lang      string `json:"lang"`
}

// 查询站点基本信息
func querySiteBaseInfoList(sn []string, pageSize uint, pageNum uint) ([]byte, error) {
	snList := strings.Join(sn, ",")

	// 要发送的 JSON
	payload := SiteBaseInfoListPayload{
		Serials:  &snList,
		PageSize: pageSize,
		PageNum:  pageNum,
		Lang:     "ZH_CN",
	}

	// 要发送的表单数据
	form := url.Values{}
	form.Set("serials", *payload.Serials) // *string 转 string
	form.Set("pageSize", fmt.Sprintf("%d", payload.PageSize))
	form.Set("pageNum", fmt.Sprintf("%d", payload.PageNum))
	form.Set("lang", payload.Lang)

	url := fmt.Sprintf("https://%s/hes-gateway/manage/selectIotFirmwareGatewayList", config.FWH_ENERGY)
	response, err := FwhHttpPost(url, form)
	if err != nil {
		return nil, err
	}
	return response.Body, nil
}

// 从fwh_pro获取站点基本信息表
func QuerySiteBaseInfoList(para dto.SiteBaseInfoPara) (dto.SiteBaseInfoResponse, error) {
	resByte, err := querySiteBaseInfoList(para.SiteList, para.PageSize, para.PageNum)
	if err != nil {
		return dto.SiteBaseInfoResponse{}, err
	}
	var siteInfoRes dto.SiteBaseInfoResponse
	// 反序列化（unmarshal）
	jsonErr := json.Unmarshal(resByte, &siteInfoRes)
	if jsonErr != nil {
		return dto.SiteBaseInfoResponse{}, jsonErr
	}

	if siteInfoRes.Code == 401 {
		// TODO 尝试重新登录
		return dto.SiteBaseInfoResponse{}, errors.New("certification expired")
	}

	return siteInfoRes, nil
}

// 查询站点信息
func querySiteInfoList(sn *string, pageSize uint, pageNum uint) ([]byte, error) {
	// 构造 query 参数
	params := url.Values{}
	if sn != nil {
		params.Add("search", *sn)
	}
	params.Add("searchFlag", fmt.Sprintf("%d", 0)) // 0-全部 1-aGate名称 11-aGate序列号 2-aPower序列号 7-EMS_MAIN版本 8-EMS_DSP版本 10-aPower子版本号 13-协议版本 14-硬件版本 16-备注
	params.Add("pageSize", fmt.Sprintf("%d", pageSize))
	params.Add("pageNum", fmt.Sprintf("%d", pageNum))
	params.Add("lang", "ZH_CN")

	reqUrl := fmt.Sprintf("https://%s/hes-gateway/manage/selectIotGatewayList", config.FWH_ENERGY)
	response, err := FwhHttpGet(reqUrl, params)
	if err != nil {
		return nil, err
	}
	return response.Body, nil
}

// 从fwh_pro获取站点基本信息表
func QuerySiteInfoList(para dto.SiteInfoPara) (dto.SiteInfoResponse, error) {
	resByte, err := querySiteInfoList(para.SN, para.PageSize, para.PageNum)
	if err != nil {
		return dto.SiteInfoResponse{}, err
	}
	var siteInfoRes dto.SiteInfoResponse
	// 反序列化（unmarshal）
	jsonErr := json.Unmarshal(resByte, &siteInfoRes)
	if jsonErr != nil {
		return dto.SiteInfoResponse{}, jsonErr
	}

	if siteInfoRes.Code == 401 {
		// TODO 尝试重新登录
		return dto.SiteInfoResponse{}, errors.New("certification expired")
	}

	return siteInfoRes, nil
}

// 查询站点列表(通用)
func querySiteList(
	search string,
	searchFlag int,
	pageSize uint,
	pageNum uint,
) ([]byte, error) {
	// 构造 query 参数
	params := url.Values{}
	params.Add("search", search)
	params.Add("searchFlag", fmt.Sprintf("%d", searchFlag)) // 0-全部 1-aGate名称 11-aGate序列号 2-aPower序列号 7-EMS_MAIN版本 8-EMS_DSP版本 10-aPower子版本号 13-协议版本 14-硬件版本 16-备注
	params.Add("pageSize", fmt.Sprintf("%d", pageSize))
	params.Add("pageNum", fmt.Sprintf("%d", pageNum))
	params.Add("lang", "ZH_CN")

	reqUrl := fmt.Sprintf("https://%s/hes-gateway/manage/selectIotGatewayList", config.FWH_ENERGY)
	response, err := FwhHttpGet(reqUrl, params)
	if err != nil {
		return nil, err
	}
	return response.Body, nil
}

// 从fwh_pro获取站点列表
func QuerySiteList(
	search string,
	searchFlag int,
	pageSize uint,
	pageNum uint,
) (dto.SiteInfoResponse, error) {
	resByte, err := querySiteList(search, searchFlag, pageSize, pageNum)
	if err != nil {
		return dto.SiteInfoResponse{}, err
	}
	var siteInfoRes dto.SiteInfoResponse
	// 反序列化（unmarshal）
	jsonErr := json.Unmarshal(resByte, &siteInfoRes)
	if jsonErr != nil {
		return dto.SiteInfoResponse{}, jsonErr
	}

	if siteInfoRes.Code == 401 {
		// TODO 尝试重新登录
		return dto.SiteInfoResponse{}, errors.New("certification expired")
	}

	return siteInfoRes, nil
}

// 查询站点工作模式信息
func querySiteModeInfo(sn string) ([]byte, error) {
	// 要发送的 JSON
	payload := SiteModeInfoPayload{
		GatewayId: sn,
		ShowType:  1,
		Lang:      "ZH_CN",
	}

	// 要发送的表单数据
	form := url.Values{}
	form.Set("gatewayId", payload.GatewayId)
	form.Set("showType", strconv.Itoa(payload.ShowType))
	form.Set("lang", payload.Lang)

	url := fmt.Sprintf("https://%s/hes-gateway/manage/tou/getGatewayTouListV2", config.FWH_ENERGY)
	response, err := FwhHttpPost(url, form)
	if err != nil {
		return nil, err
	}
	return response.Body, nil
}

// 从fwh_pro获取站点工作模式信息
func QuerySiteModeInfo(site string) (dto.SiteModeInfoResponse, error) {
	resByte, err := querySiteModeInfo(site)
	if err != nil {
		return dto.SiteModeInfoResponse{}, err
	}
	var siteModeInfoRes dto.SiteModeInfoResponse
	// 反序列化（unmarshal）
	jsonErr := json.Unmarshal(resByte, &siteModeInfoRes)
	if jsonErr != nil {
		return dto.SiteModeInfoResponse{}, jsonErr
	}

	if siteModeInfoRes.Code == 401 {
		// TODO 尝试重新登录
		return dto.SiteModeInfoResponse{}, errors.New("certification expired")
	}

	return siteModeInfoRes, nil
}