cdn-go/server/ZoneCloudflareServer.go

package server

import (
	"bytes"
	"cfTest/cloudflareApi/zone"
	"context"
	"encoding/json"
	"errors"
	"fmt"
	"github.com/cloudflare/cloudflare-go"
	"github.com/samber/lo"
	"io"
	"io/ioutil"
	"log"
	"net/http"
	"strings"
	"time"
)

type ZoneServer struct {
	zone.UnimplementedZoneCloudflareServer
}

type Subscription struct {
	ID       string `json:"id"`
	RatePlan struct {
		ID                string   `json:"id"`
		PublicName        string   `json:"public_name"`
		Currency          string   `json:"currency"`
		Scope             string   `json:"scope"`
		ExternallyManaged bool     `json:"externally_managed"`
		Sets              []string `json:"sets"`
		IsContract        bool     `json:"is_contract"`
	} `json:"rate_plan"`
	ComponentValues []struct {
		Name    string `json:"name"`
		Value   int    `json:"value"`
		Default int    `json:"default,omitempty"`
	} `json:"component_values"`
	Zone struct {
		ID   string `json:"id"`
		Name string `json:"name"`
	} `json:"zone"`
	Frequency string `json:"frequency"`
	Currency  string `json:"currency"`
	Price     int    `json:"price"`
	Product   struct {
		Name       string `json:"name"`
		Period     string `json:"period"`
		Billing    string `json:"billing"`
		PublicName string `json:"public_name"`
		Duration   int    `json:"duration"`
	} `json:"product"`
	CancelAtPeriodEnd bool      `json:"cancel_at_period_end"`
	CreatedDate       time.Time `json:"created_date"`
}

var ArgoSubscriptionId = "argo_zone_ent_tryout"

const TOKEN_PREFIX = "#pt"
const TOKEN_PREFIX_LENGTH = 3

var (
	api *cloudflare.API
	err error
)

func (s *ZoneServer) GetArgoSettingSubscription(ctx context.Context, request *zone.IdRequest) (*zone.ArgoSetting, error) {

	//设置HTTP请求
	client := &http.Client{}
	req, err := http.NewRequest("GET", "https://api.cloudflare.com/client/v4/zones/"+request.ZoneId+"/subscriptions", nil)
	if err != nil {
		return nil, err
	}
	if strings.HasPrefix(request.GetApiKey(), TOKEN_PREFIX) {
		req.Header.Add("Authorization", ApiToken2Authorization(request.GetApiKey()[TOKEN_PREFIX_LENGTH:]))
	} else {
		req.Header.Add("X-Auth-Key", request.GetApiKey())
		req.Header.Add("X-Auth-Email", request.GetApiEmail())
	}
	//执行HTTP请求
	resp, err := client.Do(req)
	log.Printf("Cloudflare GetArgoSettingSubscription API response: %v err:%s", resp, err)
	if err != nil {
		return nil, err
	}
	defer func(Body io.ReadCloser) {
		_ = Body.Close()
	}(resp.Body)
	body, _ := ioutil.ReadAll(resp.Body)

	type ResponseType struct {
		Errors     []interface{}  `json:"errors"`
		Messages   []interface{}  `json:"messages"`
		Result     []Subscription `json:"result"`
		ResultInfo struct {
			NextPage bool `json:"next_page"`
		} `json:"result_info"`
		Success bool `json:"success"`
	}
	var r ResponseType
	err = json.Unmarshal(body, &r)

	//处理返回逻辑
	if err != nil {
		return nil, fmt.Errorf("%s: %w", "error unmarshalling the JSON response", err)
	}
	if r.Success != true {
		return nil, fmt.Errorf("%s", r.Errors)
	}
	log.Printf("Response: %v", r)

	for _, v := range r.Result {

		if v.RatePlan.ID == ArgoSubscriptionId {
			return &zone.ArgoSetting{Enabled: true}, nil
		}
	}
	return &zone.ArgoSetting{Enabled: false}, nil
}

func (s *ZoneServer) CreateArgoSettingSubscription(ctx context.Context, request *zone.IdRequest) (*zone.ArgoSetting, error) {

	type RequestType struct {
		RatePlan struct {
			ID string `json:"id"`
		} `json:"rate_plan"`
		Zone struct {
			ID   string `json:"id"`
			Name string `json:"name"`
		} `json:"zone"`
	}
	type ResponseType struct {
		Errors   []interface{} `json:"errors"`
		Messages []interface{} `json:"messages"`
		Result   Subscription  `json:"result"`
		Success  bool          `json:"success"`
	}
	bytesData, err := json.Marshal(RequestType{
		RatePlan: struct {
			ID string `json:"id"`
		}{
			ID: ArgoSubscriptionId,
		},
		Zone: struct {
			ID   string `json:"id"`
			Name string `json:"name"`
		}{
			ID: request.ZoneId,
		},
	})
	if err != nil {
		return nil, err
	}
	//设置HTTP请求
	client := &http.Client{}
	req, err := http.NewRequest("POST", "https://api.cloudflare.com/client/v4/zones/"+request.ZoneId+"/subscription", bytes.NewReader(bytesData))
	if err != nil {
		return nil, err
	}
	if strings.HasPrefix(request.GetApiKey(), TOKEN_PREFIX) {
		req.Header.Add("Authorization", ApiToken2Authorization(request.GetApiKey()[TOKEN_PREFIX_LENGTH:]))
	} else {
		req.Header.Add("X-Auth-Key", request.GetApiKey())
		req.Header.Add("X-Auth-Email", request.GetApiEmail())
	}
	//执行HTTP请求
	resp, err := client.Do(req)
	log.Printf("Cloudflare CreateArgoSettingSubscription API response: %v err:%s", resp, err)
	if err != nil {
		return nil, err
	}
	defer func(Body io.ReadCloser) {
		_ = Body.Close()
	}(resp.Body)
	body, _ := ioutil.ReadAll(resp.Body)

	var r ResponseType
	err = json.Unmarshal(body, &r)

	//处理返回逻辑
	if err != nil {
		return nil, fmt.Errorf("%s: %w", "error unmarshalling the JSON response", err)
	}
	if r.Success != true {
		return nil, fmt.Errorf("%s", r.Errors)
	}
	log.Printf("Response: %v", r)

	if r.Result.RatePlan.ID == ArgoSubscriptionId {
		return &zone.ArgoSetting{Enabled: true}, nil
	}

	return &zone.ArgoSetting{Enabled: false}, nil
}

func (s *ZoneServer) UpdateArgoSetting(ctx context.Context, requestCloudflare *zone.UpdateArgoRequestCloudflare) (*zone.ArgoSetting, error) {
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		return nil, err
	}

	settingValue := Proto2GoArgoSetting(requestCloudflare.GetArgoSetting())

	setting, err := api.UpdateArgoSmartRouting(ctx, requestCloudflare.ZoneId, settingValue)
	log.Printf("Cloudflare UpdateArgoSmartRouting API response: %v err:%s", setting, err)
	if err != nil {
		return nil, err
	}

	return Go2ProtoArgoSetting(setting.Value), nil
}

func (s *ZoneServer) GetArgoSetting(ctx context.Context, requestCloudflare *zone.IdRequest) (*zone.ArgoSetting, error) {
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		return nil, err
	}

	setting, err := api.ArgoSmartRouting(ctx, requestCloudflare.ZoneId)
	log.Printf("Cloudflare ArgoSmartRouting API response: %v err:%s", setting, err)
	if err != nil {
		return nil, err
	}

	return Go2ProtoArgoSetting(setting.Value), nil
}

func (s *ZoneServer) ZoneIDByNameCloudflare(ctx context.Context, requestCloudflare *zone.ZoneIDByNameRequestCloudflare) (*zone.ZoneIDByNameReplyCloudflare, error) {
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		return nil, err
	}

	id, err := api.ZoneIDByName(requestCloudflare.ZoneName)
	log.Printf("Cloudflare ZoneIDByName API response: %v err:%s", id, err)
	if err != nil {
		return nil, err
	}

	return &zone.ZoneIDByNameReplyCloudflare{ZoneId: id}, err
}

func (s *ZoneServer) ZoneDetails(ctx context.Context, requestCloudflare *zone.ZoneDetailsRequestCloudflare) (*zone.ZoneDetailsReplyCloudflare, error) {
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		return nil, err
	}

	ctxAPI := context.Background()

	details, err := api.ZoneDetails(ctxAPI, requestCloudflare.GetZoneId())
	log.Printf("Cloudflare ZoneDetails API response: %v err:%s", details, err)
	if err != nil {
		return nil, err
	}

	return &zone.ZoneDetailsReplyCloudflare{
		Result: Go2ProtoZone(details),
	}, err
}

func (s *ZoneServer) GetZones(ctx context.Context, requestCloudflare *zone.GetZonesRequestCloudflare) (*zone.GetZonesReplyCloudflare, error) {
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		return nil, err
	}

	details, err := api.ListZones(ctx, "name", requestCloudflare.GetDomain())
	log.Printf("Cloudflare GetZones API response: %v err:%s", details, err)
	if err != nil {
		return nil, err
	}
	list := make([]*zone.ZoneCloudflareEntity, len(details))
	for i, detail := range details {
		list[i] = Go2ProtoZone(detail)
	}
	return &zone.GetZonesReplyCloudflare{
		Result:  list,
		Success: true,
	}, err
}

func (s *ZoneServer) ListZones(ctx context.Context, requestCloudflare *zone.ListZonesRequestCloudflare) (*zone.ZonesList, error) {
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		return nil, fmt.Errorf("failed to initialize Cloudflare API: %w", err)
	}

	ctxAPI := context.Background()

	details, err := api.ListZones(ctxAPI)
	log.Printf("Cloudflare ListZones API response: %v err:%s", details, err)
	if err != nil {
		return nil, err
	}
	list := make([]*zone.ZoneCloudflareEntity, len(details))
	for i, detail := range details {
		list[i] = Go2ProtoZone(detail)
	}
	return &zone.ZonesList{
		Result: list,
	}, err
}

func (s *ZoneServer) ListZonesByAccountId(ctx context.Context, requestCloudflare *zone.ListZonesByAccountIdRequestCloudflare) (*zone.ZonesList, error) {
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		return nil, err
	}

	ctxAPI := context.Background()
	opt := cloudflare.WithZoneFilters("", requestCloudflare.AccountId, "")
	details, err := api.ListZonesContext(ctxAPI, opt)
	log.Printf("Cloudflare ListZonesByAccountId API response: %v err:%s", details, err)
	if err != nil {
		return nil, err
	}
	list := make([]*zone.ZoneCloudflareEntity, len(details.Result))
	for i, detail := range details.Result {
		list[i] = Go2ProtoZone(detail)
	}
	return &zone.ZonesList{
		Result: list,
	}, err
}

func (s *ZoneServer) EditZone(ctx context.Context, requestCloudflare *zone.EditZoneRequestCloudflare) (*zone.EditZoneReplyCloudflare, error) {
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		return nil, err
	}

	ctxAPI := context.Background()

	opts := cloudflare.ZoneOptions{}

	opts.Paused = requestCloudflare.Paused

	if requestCloudflare.PlanId != nil {
		opts.Plan = &cloudflare.ZonePlan{
			ZonePlanCommon: cloudflare.ZonePlanCommon{
				ID: requestCloudflare.GetPlanId(),
			},
		}
	}

	if requestCloudflare.Type != nil {
		opts.Type = requestCloudflare.GetType()
	}

	editZone, err := api.EditZone(ctxAPI, requestCloudflare.GetZoneId(), opts)
	log.Printf("Cloudflare EditZone API response: %v err:%s", editZone, err)
	if err != nil {
		return nil, err
	}

	return &zone.EditZoneReplyCloudflare{
		Result:  Go2ProtoZone(editZone),
		Success: true,
	}, nil

}

func (s *ZoneServer) DeleteZone(ctx context.Context, requestCloudflare *zone.DeleteZoneRequestCloudflare) (*zone.DeleteZoneReplyCloudflare, error) {
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		return nil, err
	}

	ctxAPI := context.Background()

	deleteZone, err := api.DeleteZone(ctxAPI, requestCloudflare.GetZoneId())
	log.Printf("Cloudflare DeleteZone API response: %v err:%s", deleteZone, err)
	if err != nil {
		return nil, err
	}

	return &zone.DeleteZoneReplyCloudflare{
		Success: true,
		Id:      deleteZone.ID,
	}, nil
}

func (s *ZoneServer) CreateZone(ctx context.Context, requestCloudflare *zone.CreateZoneRequestCloudflare) (*zone.CreateZoneReplyCloudflare, error) {
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		return nil, err
	}

	ctxAPI := context.Background()
	account, _, err := api.Account(ctxAPI, requestCloudflare.GetAccountId())
	log.Printf("Cloudflare Account API response: %v err:%s", account, err)
	if err != nil {
		return nil, err
	}

	createZone, err := api.CreateZone(ctxAPI, requestCloudflare.GetDomain(), false, account, "full")
	log.Printf("Cloudflare CreateZone API response: %v err:%s", createZone, err)
	if err != nil {
		return nil, err
	}

	return &zone.CreateZoneReplyCloudflare{
		Success: true,
		Id:      createZone.ID,
	}, nil

}

func (s *ZoneServer) SetZonePlan(ctx context.Context, requestCloudflare *zone.SetZonePlanRequestCloudflare) (*zone.SetZonePlanReplyCloudflare, error) {
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		return nil, err
	}

	ctxAPI := context.Background()
	err = api.ZoneSetPlan(ctxAPI, requestCloudflare.GetZoneId(), requestCloudflare.GetZonePlan())
	log.Printf("Cloudflare SetZonePlan API  err:%s", err)
	if err != nil {
		return nil, err
	}

	return &zone.SetZonePlanReplyCloudflare{
		Success: true,
	}, nil
}

func (s *ZoneServer) ActivationCheck(ctx context.Context, requestCloudflare *zone.ActivationCheckRequestCloudflare) (*zone.ActivationCheckReplyCloudflare, error) {
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		return nil, err
	}

	ctxAPI := context.Background()

	activationCheck, err := api.ZoneActivationCheck(ctxAPI, requestCloudflare.GetZoneId())
	log.Printf("Cloudflare ZoneActivationCheck API response: %v err:%s", activationCheck, err)
	if err != nil {
		return nil, err
	}

	return &zone.ActivationCheckReplyCloudflare{
		Success: activationCheck.Success,
	}, nil
}

var phaseMap = map[zone.Phase]string{
	zone.Phase_http_request_sanitize:       "http_request_sanitize",
	zone.Phase_http_request_transform:      "http_request_transform",
	zone.Phase_http_request_origin:         "http_request_origin",
	zone.Phase_http_request_cache_settings: "http_request_cache_settings",
	zone.Phase_http_config_settings:        "http_config_settings",
	//zone.Phase_http_request_dynamic_redirect:   "http_request_dynamic_redirect",
	//zone.Phase_ddos_l7:                         "ddos_l7",
	//zone.Phase_http_request_firewall_custom:    "http_request_firewall_custom",
	//zone.Phase_http_ratelimit:                  "http_ratelimit",
	//zone.Phase_http_request_firewall_managed:   "http_request_firewall_managed",
	//zone.Phase_http_request_sbfm:               "http_request_sbfm",
	//zone.Phase_http_request_redirect:           "http_request_redirect",
	zone.Phase_http_request_late_transform: "http_request_late_transform",
	//zone.Phase_http_custom_errors:              "http_custom_errors",
	zone.Phase_http_response_headers_transform: "http_response_headers_transform",
	//zone.Phase_http_response_firewall_managed:  "http_response_firewall_managed",
	//zone.Phase_http_log_custom_fields:          "http_log_custom_fields",
}

func (s *ZoneServer) ListZoneRuleSets(ctx context.Context, requestCloudflare *zone.ListZoneRuleSetsRequestCloudflare) (*zone.RuleSetList, error) {
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		return nil, err
	}
	ctxAPI := context.Background()
	zoneIdentifier := cloudflare.ZoneIdentifier(requestCloudflare.GetZoneId())
	ListRulesetsParams := cloudflare.ListRulesetsParams{}
	goRuleSets, err := api.ListRulesets(ctxAPI, zoneIdentifier, ListRulesetsParams)
	log.Printf("Cloudflare ListZoneRuleSets API response: %v err:%s", goRuleSets, err)
	if err != nil {
		return nil, err
	}
	filterMap := lo.FilterMap[cloudflare.Ruleset, *zone.RuleSet](goRuleSets, func(goRuleSet cloudflare.Ruleset, _ int) (*zone.RuleSet, bool) {
		protoRuleSet := Go2ProtoRuleSet(&goRuleSet)
		return protoRuleSet, true
	})
	return &zone.RuleSetList{RuleSets: filterMap}, nil
}

func (s *ZoneServer) CreateZoneRuleSet(ctx context.Context, requestCloudflare *zone.CreateZoneRuleSetRequestCloudflare) (*zone.CreateZoneRuleSetReplyCloudflare, error) {
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		return nil, err
	}
	ctxAPI := context.Background()
	filterMap := lo.FilterMap[*zone.Rule, cloudflare.RulesetRule](requestCloudflare.Rules, func(rule *zone.Rule, index int) (cloudflare.RulesetRule, bool) {
		ruleResult, err := Proto2GoRulesetRuleByPhase(rule, requestCloudflare.RuleSet.Phase)
		if err != nil || ruleResult == nil {
			return cloudflare.RulesetRule{}, false
		}
		return *ruleResult, true
	})
	zoneIdentifier := cloudflare.ZoneIdentifier(requestCloudflare.GetZoneId())
	createRulesetParams := cloudflare.CreateRulesetParams{
		Name:  requestCloudflare.RuleSet.Name,
		Kind:  requestCloudflare.RuleSet.Kind,
		Phase: phaseMap[requestCloudflare.RuleSet.Phase],
		Rules: filterMap,
	}
	_, err = api.CreateRuleset(ctxAPI, zoneIdentifier, createRulesetParams)
	log.Printf("Cloudflare CreateRuleset API err:%s", err)
	if err != nil {
		return &zone.CreateZoneRuleSetReplyCloudflare{Success: false}, err
	}
	//protoRuleSet := Go2ProtoRuleSet(&goRuleSet)
	return &zone.CreateZoneRuleSetReplyCloudflare{Success: true}, nil
}

func (s *ZoneServer) DeleteZoneRuleSet(ctx context.Context, requestCloudflare *zone.DeleteZoneRuleSetRequestCloudflare) (*zone.DeleteZoneRuleSetReplyCloudflare, error) {
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		return nil, err
	}
	ctxAPI := context.Background()
	zoneIdentifier := cloudflare.ZoneIdentifier(requestCloudflare.GetZoneId())
	err = api.DeleteRuleset(ctxAPI, zoneIdentifier, requestCloudflare.GetRuleSetId())
	log.Printf("Cloudflare DeleteRuleset API err:%s", err)
	if err != nil {
		return &zone.DeleteZoneRuleSetReplyCloudflare{Success: false}, err
	}
	return &zone.DeleteZoneRuleSetReplyCloudflare{Success: true}, nil
}

func (s *ZoneServer) UpdateRuleSetByRuleSetId(ctx context.Context, requestCloudflare *zone.UpdateRuleSetByRuleSetIdRequestCloudflare) (*zone.RuleSet, error) {
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		return nil, err
	}
	ctxAPI := context.Background()
	filterMap := lo.FilterMap[*zone.Rule, cloudflare.RulesetRule](requestCloudflare.Rules, func(rule *zone.Rule, index int) (cloudflare.RulesetRule, bool) {
		ruleResult, err := Proto2GoRulesetRuleByPhase(rule, requestCloudflare.Phase)
		if err != nil || ruleResult == nil {
			return cloudflare.RulesetRule{}, false
		}
		return *ruleResult, true
	})
	zoneIdentifier := cloudflare.ZoneIdentifier(requestCloudflare.GetZoneId())
	updateRulesetParams := cloudflare.UpdateRulesetParams{
		ID:    requestCloudflare.RuleSetId,
		Rules: filterMap,
	}
	goRuleSet, err := api.UpdateRuleset(ctxAPI, zoneIdentifier, updateRulesetParams)
	log.Printf("Cloudflare UpdateRuleset API response: %v err:%s", goRuleSet, err)
	if err != nil {
		return nil, err
	}
	protoRuleSet := Go2ProtoRuleSet(&goRuleSet)
	return protoRuleSet, nil
}

func (s *ZoneServer) GetRuleSet(ctx context.Context, requestCloudflare *zone.GetRuleSetRequestCloudflare) (*zone.RuleList, error) {
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		return nil, err
	}
	ctxAPI := context.Background()
	switch phaseMap[requestCloudflare.Phase] {
	case "http_request_origin", "http_request_cache_settings", "http_config_settings":
	case "http_request_transform", "http_request_late_transform", "http_response_headers_transform":
	default:
		return nil, errors.New("未定义规则阶段")
	}
	zoneIdentifier := cloudflare.ZoneIdentifier(requestCloudflare.GetZoneId())
	ruleset, err := api.GetEntrypointRuleset(ctxAPI, zoneIdentifier, phaseMap[requestCloudflare.Phase])
	log.Printf("Cloudflare GetEntrypointRuleset API response: %v err:%s", ruleset, err)
	if err != nil {
		_, ok := err.(*cloudflare.NotFoundError)
		if ok {
			ruleset.Rules = make([]cloudflare.RulesetRule, 0)
			err = nil
		} else {
			return nil, err
		}
	}
	filterMap := lo.FilterMap[cloudflare.RulesetRule, *zone.Rule](ruleset.Rules, func(rule cloudflare.RulesetRule, _ int) (*zone.Rule, bool) {
		phase, err := Go2ProtoRuleRuleByPhase(&rule, ruleset.Phase)
		if err != nil {
			return nil, false
		}
		return phase, true
	})
	return &zone.RuleList{Rules: filterMap}, nil
}

func (s *ZoneServer) UpdateRuleSet(ctx context.Context, requestCloudflare *zone.UpdateRuleSetRequestCloudflare) (*zone.RuleList, error) {
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		return nil, err
	}
	ctxAPI := context.Background()

	phase := phaseMap[requestCloudflare.Phase]

	filterMap := lo.FilterMap[*zone.Rule, cloudflare.RulesetRule](requestCloudflare.Rules, func(rule *zone.Rule, index int) (cloudflare.RulesetRule, bool) {
		ruleResult, err := Proto2GoRulesetRuleByPhase(rule, requestCloudflare.Phase)
		if err != nil || ruleResult == nil {
			return cloudflare.RulesetRule{}, false
		}
		return *ruleResult, true
	})
	//for _, rule := range requestCloudflare.Rules {
	//	ruleResult, err := Proto2GoRulesetRuleByPhase(rule, requestCloudflare.Phase)
	//	if err != nil {
	//		return nil, err
	//	}
	//	if ruleResult != nil {
	//		ruleSet.Rules = append(ruleSet.Rules, *ruleResult)
	//	}
	//}
	zoneIdentifier := cloudflare.ZoneIdentifier(requestCloudflare.GetZoneId())
	updateEntrypointRulesetParams := cloudflare.UpdateEntrypointRulesetParams{
		Phase: phase,
		Rules: filterMap,
	}
	ruleset, err := api.UpdateEntrypointRuleset(ctxAPI, zoneIdentifier, updateEntrypointRulesetParams)
	log.Printf("Cloudflare UpdateEntrypointRuleset API response: %v err:%s", ruleset, err)
	if err != nil {
		return nil, err
	}

	rulesReturn := lo.FilterMap[cloudflare.RulesetRule, *zone.Rule](ruleset.Rules, func(rule cloudflare.RulesetRule, _ int) (*zone.Rule, bool) {
		ruleReturn, err := Go2ProtoRuleRuleByPhase(&rule, ruleset.Phase)
		if err != nil {
			return nil, false
		}
		return ruleReturn, true
	})
	//var rulesReturn []*zone.Rule
	//for _, rule := range rulesetResult.Rules {
	//	ruleReturn, err := Go2ProtoRuleRuleByPhase(&rule, rulesetResult.Phase)
	//	if err != nil {
	//		return nil, err
	//	}
	//	if ruleReturn != nil {
	//		rulesReturn = append(rulesReturn, ruleReturn)
	//	}
	//}

	return &zone.RuleList{Rules: rulesReturn}, nil
}

//func UpdateRuleSet(apiKey, apiEmail, zoneID, rulesetPhase string, ruleset cloudflare.Ruleset) (*cloudflare.Ruleset, error) {
//	api, err := cloudflare.New(apiKey, apiEmail)
//	if err != nil {
//		return nil, err
//	}
//	ctxAPI := context.Background()
//	phase, err := api.UpdateZoneRulesetPhase(ctxAPI, zoneID, rulesetPhase, ruleset)
//	if err != nil {
//		return nil, err
//	}
//	return &phase, nil
//}

//PageRule CURD

func (s *ZoneServer) ListPageRule(ctx context.Context, requestCloudflare *zone.ListPageRuleRequestCloudflare) (*zone.PageRuleList, error) {
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		return nil, err
	}
	ctxAPI := context.Background()
	rules, err := api.ListPageRules(ctxAPI, requestCloudflare.GetZoneId())
	log.Printf("Cloudflare ListPageRules API response: %v err:%s", rules, err)
	if err != nil {
		return nil, err
	}
	filterMap := lo.FilterMap[cloudflare.PageRule, *zone.PageRule](rules, func(rule cloudflare.PageRule, _ int) (*zone.PageRule, bool) {
		pageRule := Go2ProtoPageRule(&rule)
		return pageRule, true
	})
	return &zone.PageRuleList{PageRules: filterMap}, nil
}

func (s *ZoneServer) CreatePageRule(ctx context.Context, requestCloudflare *zone.CreatePageRuleRequestCloudflare) (*zone.PageRule, error) {
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		return nil, err
	}
	ctxAPI := context.Background()
	rule, err := Proto2GoPageRule(requestCloudflare.PageRule)
	if err != nil {
		return nil, err
	}
	pageRule, err := api.CreatePageRule(ctxAPI, requestCloudflare.GetZoneId(), *rule)
	log.Printf("Cloudflare CreatePageRule API response: %v err:%s", pageRule, err)
	if err != nil {
		return nil, err
	}
	return Go2ProtoPageRule(pageRule), nil
}

func (s *ZoneServer) UpdatePageRule(ctx context.Context, requestCloudflare *zone.UpdatePageRuleRequestCloudflare) (*zone.PageRule, error) {
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		return nil, err
	}
	ctxAPI := context.Background()
	rule, err := Proto2GoPageRule(requestCloudflare.PageRule)
	if err != nil {
		return nil, err
	}
	err = api.UpdatePageRule(ctxAPI, requestCloudflare.GetZoneId(), rule.ID, *rule)
	log.Printf("Cloudflare UpdatePageRule API err:%s", err)
	if err != nil {
		return nil, err
	}
	return requestCloudflare.PageRule, nil
}

func (s *ZoneServer) DeletePageRule(ctx context.Context, requestCloudflare *zone.DeletePageRuleRequestCloudflare) (*zone.PageRule, error) {
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		return nil, err
	}
	ctxAPI := context.Background()
	err = api.DeletePageRule(ctxAPI, requestCloudflare.ZoneId, requestCloudflare.PageRuleId)
	log.Printf("Cloudflare DeletePageRule API err:%s", err)
	if err != nil {
		return nil, err
	}
	return &zone.PageRule{}, nil
}

//func UpdatePageRule(apiKey, apiEmail, zoneID string, rule *zone.PageRule) (*zone.PageRule, error) {
//	api, err := cloudflare.New(apiKey, apiEmail)
//	if err != nil {
//		return nil, err
//	}
//
//	ctxAPI := context.Background()
//	goPageRule, err := Proto2GoPageRule(rule)
//	if err != nil {
//		return nil, err
//	}
//	pageRule, err := api.CreatePageRule(ctxAPI, zoneID, *goPageRule)
//	if err != nil {
//		return nil, err
//	}
//
//	protoPageRule := Go2ProtoPageRule(pageRule)
//
//	return protoPageRule, nil
//}

type PageRuleActionCacheKeyFieldsQueryString struct {
	Exclude interface{} `json:"exclude"`
	Include interface{} `json:"include"`
}
type PageRuleActionCacheKeyFields struct {
	QueryString PageRuleActionCacheKeyFieldsQueryString `json:"query_string"`
}
type UpdatePageRulesPrioritiesStruct struct {
	Id       string `json:"id"`
	Priority int    `json:"priority"`
}

func (s *ZoneServer) UpdatePageRulesPriorities(ctx context.Context, requestCloudflare *zone.UpdatePageRulesPrioritiesRequestCloudflare) (*zone.PageRuleList, error) {
	//TODO 参数检查

	//设置HTTP请求
	client := &http.Client{}
	bytesData, err := json.Marshal(requestCloudflare.Data)
	if err != nil {
		return nil, err
	}
	req, err := http.NewRequest("PUT", "https://api.cloudflare.com/client/v4/zones/"+requestCloudflare.ZoneId+"/pagerules/priorities", bytes.NewReader(bytesData))
	if err != nil {
		return nil, err
	}
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		req.Header.Add("Authorization", ApiToken2Authorization(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:]))
	} else {
		req.Header.Add("X-Auth-Key", requestCloudflare.GetApiKey())
		req.Header.Add("X-Auth-Email", requestCloudflare.GetApiEmail())
	}
	//执行HTTP请求
	resp, err := client.Do(req)
	log.Printf("Cloudflare UpdatePageRulesPriorities API response: %v err:%s", resp, err)
	if err != nil {
		return nil, err
	}
	defer func(Body io.ReadCloser) {
		_ = Body.Close()
	}(resp.Body)
	body, _ := ioutil.ReadAll(resp.Body)

	//解析HTTP返回数据
	var r cloudflare.PageRulesResponse
	err = json.Unmarshal(body, &r)

	//处理返回逻辑
	if err != nil {
		return nil, fmt.Errorf("%s: %w", "error unmarshalling the JSON response", err)
	}
	if r.Success != true {
		return nil, fmt.Errorf("%s", r.Errors)
	}
	rules := r.Result
	filterMap := lo.FilterMap[cloudflare.PageRule, *zone.PageRule](rules, func(rule cloudflare.PageRule, _ int) (*zone.PageRule, bool) {
		pageRule := Go2ProtoPageRule(&rule)
		return pageRule, true
	})
	return &zone.PageRuleList{PageRules: filterMap}, nil
}

// 默认开启Smarttiered cache
func (s *ZoneServer) SetSmartTieredCache(ctx context.Context, requestCloudflare *zone.SetSmartTieredCacheRequestCloudflare) (*zone.SetSmartTieredCacheReplyCloudflare, error) {

	type ResponseType struct {
		Errors   []interface{} `json:"errors"`
		Messages []interface{} `json:"messages"`
		Result   interface{}   `json:"result"`
		Success  bool          `json:"success"`
	}

	//设置HTTP请求
	client := &http.Client{}
	url := "https://api.cloudflare.com/client/v4/zones/" + requestCloudflare.ZoneId + "/cache/tiered_cache_smart_topology_enable"
	payload := strings.NewReader("{\n  \"value\": \"on\"\n}")
	req, err := http.NewRequest("PATCH", url, payload)

	if err != nil {
		return nil, err
	}
	req.Header.Add("Content-Type", "application/json")
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		req.Header.Add("Authorization", ApiToken2Authorization(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:]))
	} else {
		req.Header.Add("X-Auth-Key", requestCloudflare.GetApiKey())
		req.Header.Add("X-Auth-Email", requestCloudflare.GetApiEmail())
	}
	//执行HTTP请求
	resp, err := client.Do(req)
	log.Printf("Cloudflare SetSmartTieredCache API response: %v err:%s", resp, err)
	if err != nil {
		return nil, err
	}

	defer func(Body io.ReadCloser) {
		_ = Body.Close()
	}(resp.Body)
	body, _ := ioutil.ReadAll(resp.Body)
	print(body)
	//解析HTTP返回数据
	var r ResponseType
	err = json.Unmarshal(body, &r)
	if err != nil {
		return nil, fmt.Errorf("%s: %w", "error unmarshalling the JSON response", err)
	}
	if r.Success != true {
		print(fmt.Errorf("%s", r.Errors))
	}

	return &zone.SetSmartTieredCacheReplyCloudflare{Enabled: r.Success}, nil
}

// 开启强制使用https
func (s *ZoneServer) UpdateAlwaysUseHttps(ctx context.Context, requestCloudflare *zone.UpdateAlwaysUseHttpsRequestCloudflare) (*zone.UpdateAlwaysUseHttpsReplyCloudflare, error) {

	type ResponseType struct {
		Errors   []interface{} `json:"errors"`
		Messages []interface{} `json:"messages"`
		Result   interface{}   `json:"result"`
		Success  bool          `json:"success"`
	}

	//设置HTTP请求
	client := &http.Client{}
	url := "https://api.cloudflare.com/client/v4/zones/" + requestCloudflare.ZoneId + "/settings/always_use_https"
	var payload io.Reader
	if requestCloudflare.Enabled {
		payload = strings.NewReader("{\n  \"value\": \"on\"\n}")
	} else {
		payload = strings.NewReader("{\n  \"value\": \"off\"\n}")
	}
	req, err := http.NewRequest("PATCH", url, payload)

	if err != nil {
		return nil, err
	}
	req.Header.Add("Content-Type", "application/json")
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		req.Header.Add("Authorization", ApiToken2Authorization(requestCloudflare.GetApiKey()[TOKEN_PREFIX_LENGTH:]))
	} else {
		req.Header.Add("X-Auth-Key", requestCloudflare.GetApiKey())
		req.Header.Add("X-Auth-Email", requestCloudflare.GetApiEmail())
	}

	//执行HTTP请求
	resp, err := client.Do(req)
	log.Printf("Cloudflare UpdateAlwaysUseHttps API response: %v err:%s", resp, err)
	if err != nil {
		return nil, err
	}
	defer func(Body io.ReadCloser) {
		_ = Body.Close()
	}(resp.Body)
	body, _ := ioutil.ReadAll(resp.Body)
	print(body)
	//解析HTTP返回数据
	var r ResponseType
	err = json.Unmarshal(body, &r)
	if err != nil {
		return nil, fmt.Errorf("%s: %w", "error unmarshalling the JSON response", err)
	}
	if r.Success != true {
		print(fmt.Errorf("%s", r.Errors))
	}
	return &zone.UpdateAlwaysUseHttpsReplyCloudflare{Success: r.Success}, nil
}