cdn-go/server/LogPushCloudflareServer.go

package server

import (
	"cfTest/cloudflareApi/logPush"
	"context"
	"errors"
	"github.com/cloudflare/cloudflare-go"
	"log"
	"strings"
)

type LogPushCloudflareServer struct {
	logPush.UnimplementedLogPushCloudflareServer
}

func (s *LogPushCloudflareServer) ListLogPushJobsByZoneId(ctx context.Context,
	requestCloudflare *logPush.ListLogPushJobRequestByZoneId) (*logPush.ListLogPushJobReplay, error) {
	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[3:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		log.Printf("new cloudflare api error : %v", err)
		return nil, err
	}

	ctxAPI := context.Background()

	zoneIdentifier := cloudflare.ZoneIdentifier(requestCloudflare.GetZoneId())
	listLogpushJobsForDatasetParams := cloudflare.ListLogpushJobsForDatasetParams{
		Dataset: requestCloudflare.DataSet,
	}
	LogPushJobs, err := api.ListLogpushJobsForDataset(ctxAPI, zoneIdentifier, listLogpushJobsForDatasetParams)
	//LogPushJobs, err := api.ListZoneLogpushJobsForDataset(ctxAPI,
	//	requestCloudflare.GetZoneId(), requestCloudflare.GetDataSet())
	if err != nil {
		log.Printf("list logpush job error : %v", err)
		return nil, err
	}

	list := make([]*logPush.LogPushJobEntity, len(LogPushJobs))
	for i, job := range LogPushJobs {
		list[i] = Go2ProtoLogPushJob(job)
	}
	return &logPush.ListLogPushJobReplay{
		Result: list,
	}, err
}

func (s *LogPushCloudflareServer) CreateLogPushJobByZoneId(ctx context.Context,
	requestCloudflare *logPush.CreateLogPushJobRequestByZoneId) (*logPush.CreateLogPushJobReplay, error) {

	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[3:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		log.Printf("new cloudflare api error : %v", err)
		return nil, err
	}

	ctxAPI := context.Background()

	//job := cloudflare.LogpushJob{
	//	Dataset:         requestCloudflare.GetDataSet(),
	//	DestinationConf: requestCloudflare.GetDestinationConf(),
	//	Enabled:         true,
	//	Frequency:       "low",
	//	LogpullOptions: "fields=ClientIP,ClientCountry,ClientRequestMethod,ClientRequestProtocol," +
	//		"ClientRequestHost,ClientRequestPath,ClientRequestURI,ClientRequestBytes," +
	//		"EdgeResponseStatus,EdgeResponseBytes,ClientRequestUserAgent,CacheCacheStatus,CacheResponseBytes," +
	//		"EdgeStartTimestamp,EdgeEndTimestamp&timestamps=rfc3339",
	//	OwnershipChallenge: requestCloudflare.GetOwnershipChallenge(),
	//}
	zoneIdentifier := cloudflare.ZoneIdentifier(requestCloudflare.GetZoneId())
	createLogpushJobParams := cloudflare.CreateLogpushJobParams{
		Dataset:         requestCloudflare.GetDataSet(),
		DestinationConf: requestCloudflare.GetDestinationConf(),
		Enabled:         true,
		Frequency:       "low",
		LogpullOptions: "fields=ClientIP,ClientCountry,ClientRequestMethod,ClientRequestProtocol," +
			"ClientRequestHost,ClientRequestPath,ClientRequestURI,ClientRequestBytes," +
			"EdgeResponseStatus,EdgeResponseBytes,ClientRequestUserAgent,CacheCacheStatus,CacheResponseBytes," +
			"EdgeStartTimestamp,EdgeEndTimestamp&timestamps=rfc3339",
		OwnershipChallenge: requestCloudflare.GetOwnershipChallenge(),
	}
	result, err := api.CreateLogpushJob(ctxAPI, zoneIdentifier, createLogpushJobParams)
	//result, err := api.CreateZoneLogpushJob(ctxAPI,zoneIdentifier, job)
	if err != nil {
		log.Printf("create logpush job error : %v", err)
		return nil, err
	}

	return &logPush.CreateLogPushJobReplay{
		Result: Go2ProtoLogPushJob(*result),
	}, err
}

func (s *LogPushCloudflareServer) DeleteLogPushJobByZoneId(ctx context.Context,
	requestCloudflare *logPush.DeleteLogPushJobRequestByZoneId) (*logPush.DeleteLogPushJobReplay, error) {

	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[3:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		log.Printf("new cloudflare api error : %v", err)
		return nil, err
	}

	ctxAPI := context.Background()
	zoneIdentifier := cloudflare.ZoneIdentifier(requestCloudflare.GetZoneId())
	result := api.DeleteLogpushJob(ctxAPI, zoneIdentifier, int(requestCloudflare.GetJobId()))
	//result := api.DeleteZoneLogpushJob(ctxAPI, requestCloudflare.GetZoneId(), int(requestCloudflare.GetJobId()))

	if result != nil {
		log.Printf("delete zone LogpushJob error: %v", result.Error())
		return &logPush.DeleteLogPushJobReplay{
			Success: false,
			Message: result.Error(),
		}, err
	}
	return &logPush.DeleteLogPushJobReplay{
		Success: true,
		Message: "success",
	}, err
}

func (s *LogPushCloudflareServer) GetOwnershipChallengeByZoneId(ctx context.Context,
	requestCloudflare *logPush.GetOwnershipChallengeRequestByZoneId) (*logPush.GetOwnershipChallengeReply, error) {

	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[3:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		log.Printf("new cloudflare api error : %v", err)
		return nil, err
	}

	ctxAPI := context.Background()
	zoneIdentifier := cloudflare.ZoneIdentifier(requestCloudflare.GetZoneId())
	getLogpushOwnershipChallengeParams := cloudflare.GetLogpushOwnershipChallengeParams{
		DestinationConf: requestCloudflare.DestinationConf,
	}
	ownershipChallenge, err := api.GetLogpushOwnershipChallenge(ctxAPI, zoneIdentifier, getLogpushOwnershipChallengeParams)
	//ownershipChallenge, err := api.GetZoneLogpushOwnershipChallenge(ctxAPI,
	//	requestCloudflare.GetZoneId(), requestCloudflare.GetDestinationConf())
	if err != nil {
		log.Printf("get logpush job ownershipChallenge error : %v", err)
		return nil, err
	}

	if ownershipChallenge.Valid {
		log.Printf("ownershipChallenge success: %v", ownershipChallenge.Filename)
		return &logPush.GetOwnershipChallengeReply{
			Result: &logPush.OwnershipChallengeEntity{
				Filename: ownershipChallenge.Filename,
				Valid:    ownershipChallenge.Valid,
				Message:  ownershipChallenge.Message,
			},
		}, err
	}

	log.Printf("ownershipChallenge fail: %v", ownershipChallenge.Message)
	return nil, errors.New(ownershipChallenge.Message)
}

func (s *LogPushCloudflareServer) CheckDestinationExist(ctx context.Context,
	requestCloudflare *logPush.CheckDestinationExistRequest) (*logPush.CheckDestinationExistReply, error) {

	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[3:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		log.Printf("new cloudflare api error : %v", err)
		return nil, err
	}

	ctxAPI := context.Background()
	zoneIdentifier := cloudflare.ZoneIdentifier(requestCloudflare.GetZoneId())
	exists, err := api.CheckLogpushDestinationExists(ctxAPI, zoneIdentifier, requestCloudflare.GetDestinationConf())
	//exists, err := api.CheckZoneLogpushDestinationExists(ctxAPI, requestCloudflare.GetZoneId(), requestCloudflare.GetDestinationConf())

	if err != nil {
		log.Printf("check logpush job destination error : %v", err)
		return nil, err
	}

	return &logPush.CheckDestinationExistReply{
		Exists: exists,
	}, err
}

func (s *LogPushCloudflareServer) ValidateOwnershipChallenge(ctx context.Context,
	requestCloudflare *logPush.ValidateOwnershipChallengeRequest) (*logPush.ValidateOwnershipChallengeReply, error) {

	if strings.HasPrefix(requestCloudflare.GetApiKey(), TOKEN_PREFIX) {
		api, err = cloudflare.NewWithAPIToken(requestCloudflare.GetApiKey()[3:])
	} else {
		api, err = cloudflare.New(requestCloudflare.GetApiKey(), requestCloudflare.GetApiEmail())
	}
	if err != nil {
		log.Printf("new cloudflare api error : %v", err)
		return nil, err
	}

	ctxAPI := context.Background()
	zoneIdentifier := cloudflare.ZoneIdentifier(requestCloudflare.GetZoneId())
	validateLogpushOwnershipChallengeParams := cloudflare.ValidateLogpushOwnershipChallengeParams{
		DestinationConf:    requestCloudflare.DestinationConf,
		OwnershipChallenge: requestCloudflare.OwnershipChallenge,
	}
	valid, err := api.ValidateLogpushOwnershipChallenge(ctxAPI, zoneIdentifier, validateLogpushOwnershipChallengeParams)
	//valid, err := api.ValidateZoneLogpushOwnershipChallenge(ctxAPI, requestCloudflare.GetZoneId(),
	//	requestCloudflare.GetDestinationConf(), requestCloudflare.GetOwnershipChallenge())

	if err != nil {
		log.Printf("validate logpush job ownership challenge error : %v", err)
		return nil, err
	}

	return &logPush.ValidateOwnershipChallengeReply{
		Valid: valid,
	}, err
}