client.go

package main

import (
	"bytes"
	"context"
	"flag_sender/internal/models"
	"flag_sender/pkg/plugins"
	"fmt"
	"io"
	"net/http"
	"strings"

	"github.com/bytedance/sonic"
)

type Client struct {
	url   string
	token string
}

type FlagInfo struct {
	Flag string `json:"flag"`
	Msg  string `json:"msg"`
}

var NewClient plugins.NewClientFunc = func(url, token string) plugins.IClient {
	return &Client{
		url:   url,
		token: token,
	}
}

func (c *Client) SendFlags(ctx context.Context, flags []string) (map[string]*plugins.FlagResult, error) {
	data, err := sonic.Marshal(flags)
	if err != nil {
		return nil, err
	}

	req, err := http.NewRequestWithContext(ctx, "PUT", c.url, bytes.NewBuffer(data))
	if err != nil {
		return nil, err
	}
	req.Header.Set("X-Team-Token", c.token)

	client := &http.Client{}
	res, err := client.Do(req)
	if err != nil {
		return nil, err
	}
	defer res.Body.Close()

	body, err := io.ReadAll(res.Body)
	if err != nil {
		return nil, err
	}

	if res.StatusCode != 200 {
		return nil, fmt.Errorf("incorrect status code: %d, body: %s", res.StatusCode, body)
	}

	var flagInfos []*FlagInfo
	if err := sonic.Unmarshal(body, &flagInfos); err != nil {
		return nil, err
	}
	flagMap := make(map[string]*plugins.FlagResult)
	for _, flagInfo := range flagInfos {
		flagStatus := models.FlagStatusReject
		if strings.Contains(flagInfo.Msg, "accepted") {
			flagStatus = models.FlagStatusSuccess
		}
		if strings.Contains(flagInfo.Msg, "old") {
			flagStatus = models.FlagStatusOld
		}
		flagMap[flagInfo.Flag] = &plugins.FlagResult{
			Status: flagStatus,
			Msg:    flagInfo.Msg,
		}
	}

	return flagMap, nil
}