signoz/ee/query-service/integrations/signozio/signozio.go

package signozio

import (
	"bytes"
	"context"
	"encoding/json"
	"fmt"
	"io"
	"net/http"
	"time"

	"github.com/SigNoz/signoz/ee/query-service/constants"
	"github.com/SigNoz/signoz/ee/query-service/model"
	"github.com/pkg/errors"
)

var C *Client

const (
	POST             = "POST"
	APPLICATION_JSON = "application/json"
)

type Client struct {
	Prefix     string
	GatewayUrl string
}

func New() *Client {
	return &Client{
		Prefix:     constants.LicenseSignozIo,
		GatewayUrl: constants.ZeusURL,
	}
}

func init() {
	C = New()
}

func ValidateLicenseV3(licenseKey string) (*model.LicenseV3, *model.ApiError) {

	// Creating an HTTP client with a timeout for better control
	client := &http.Client{
		Timeout: 10 * time.Second,
	}

	req, err := http.NewRequest("GET", C.GatewayUrl+"/v2/licenses/me", nil)
	if err != nil {
		return nil, model.BadRequest(errors.Wrap(err, "failed to create request"))
	}

	// Setting the custom header
	req.Header.Set("X-Signoz-Cloud-Api-Key", licenseKey)

	response, err := client.Do(req)
	if err != nil {
		return nil, model.BadRequest(errors.Wrap(err, "failed to make post request"))
	}

	body, err := io.ReadAll(response.Body)
	if err != nil {
		return nil, model.BadRequest(errors.Wrap(err, fmt.Sprintf("failed to read validation response from %v", C.GatewayUrl)))
	}

	defer response.Body.Close()

	switch response.StatusCode {
	case 200:
		a := ValidateLicenseResponse{}
		err = json.Unmarshal(body, &a)
		if err != nil {
			return nil, model.BadRequest(errors.Wrap(err, "failed to marshal license validation response"))
		}

		license, err := model.NewLicenseV3(a.Data)
		if err != nil {
			return nil, model.BadRequest(errors.Wrap(err, "failed to generate new license v3"))
		}

		return license, nil
	case 400:
		return nil, model.BadRequest(errors.Wrap(fmt.Errorf(string(body)),
			fmt.Sprintf("bad request error received from %v", C.GatewayUrl)))
	case 401:
		return nil, model.Unauthorized(errors.Wrap(fmt.Errorf(string(body)),
			fmt.Sprintf("unauthorized request error received from %v", C.GatewayUrl)))
	default:
		return nil, model.InternalError(errors.Wrap(fmt.Errorf(string(body)),
			fmt.Sprintf("internal request error received from %v", C.GatewayUrl)))
	}

}

func NewPostRequestWithCtx(ctx context.Context, url string, contentType string, body io.Reader) (*http.Request, error) {
	req, err := http.NewRequestWithContext(ctx, POST, url, body)
	if err != nil {
		return nil, err
	}
	req.Header.Add("Content-Type", contentType)
	return req, err

}

// SendUsage reports the usage of signoz to license server
func SendUsage(ctx context.Context, usage model.UsagePayload) *model.ApiError {
	reqString, _ := json.Marshal(usage)
	req, err := NewPostRequestWithCtx(ctx, C.Prefix+"/usage", APPLICATION_JSON, bytes.NewBuffer(reqString))
	if err != nil {
		return model.BadRequest(errors.Wrap(err, "unable to create http request"))
	}

	res, err := http.DefaultClient.Do(req)
	if err != nil {
		return model.BadRequest(errors.Wrap(err, "unable to connect with license.signoz.io, please check your network connection"))
	}

	body, err := io.ReadAll(res.Body)
	if err != nil {
		return model.BadRequest(errors.Wrap(err, "failed to read usage response from license.signoz.io"))
	}

	defer res.Body.Close()

	switch res.StatusCode {
	case 200, 201:
		return nil
	case 400, 401:
		return model.BadRequest(errors.Wrap(errors.New(string(body)),
			"bad request error received from license.signoz.io"))
	default:
		return model.InternalError(errors.Wrap(errors.New(string(body)),
			"internal error received from license.signoz.io"))
	}
}

func CheckoutSession(ctx context.Context, checkoutRequest *model.CheckoutRequest, licenseKey string) (string, *model.ApiError) {
	hClient := &http.Client{}

	reqString, err := json.Marshal(checkoutRequest)
	if err != nil {
		return "", model.BadRequest(err)
	}

	req, err := http.NewRequestWithContext(ctx, "POST", C.GatewayUrl+"/v2/subscriptions/me/sessions/checkout", bytes.NewBuffer(reqString))
	if err != nil {
		return "", model.BadRequest(err)
	}
	req.Header.Set("X-Signoz-Cloud-Api-Key", licenseKey)

	response, err := hClient.Do(req)
	if err != nil {
		return "", model.BadRequest(err)
	}
	body, err := io.ReadAll(response.Body)
	if err != nil {
		return "", model.BadRequest(errors.Wrap(err, fmt.Sprintf("failed to read checkout response from %v", C.GatewayUrl)))
	}
	defer response.Body.Close()

	switch response.StatusCode {
	case 201:
		a := CheckoutResponse{}
		err = json.Unmarshal(body, &a)
		if err != nil {
			return "", model.BadRequest(errors.Wrap(err, "failed to unmarshal zeus checkout response"))
		}
		return a.Data.RedirectURL, nil
	case 400:
		return "", model.BadRequest(errors.Wrap(errors.New(string(body)),
			fmt.Sprintf("bad request error received from %v", C.GatewayUrl)))
	case 401:
		return "", model.Unauthorized(errors.Wrap(errors.New(string(body)),
			fmt.Sprintf("unauthorized request error received from %v", C.GatewayUrl)))
	default:
		return "", model.InternalError(errors.Wrap(errors.New(string(body)),
			fmt.Sprintf("internal request error received from %v", C.GatewayUrl)))
	}
}

func PortalSession(ctx context.Context, checkoutRequest *model.PortalRequest, licenseKey string) (string, *model.ApiError) {
	hClient := &http.Client{}

	reqString, err := json.Marshal(checkoutRequest)
	if err != nil {
		return "", model.BadRequest(err)
	}

	req, err := http.NewRequestWithContext(ctx, "POST", C.GatewayUrl+"/v2/subscriptions/me/sessions/portal", bytes.NewBuffer(reqString))
	if err != nil {
		return "", model.BadRequest(err)
	}
	req.Header.Set("X-Signoz-Cloud-Api-Key", licenseKey)

	response, err := hClient.Do(req)
	if err != nil {
		return "", model.BadRequest(err)
	}
	body, err := io.ReadAll(response.Body)
	if err != nil {
		return "", model.BadRequest(errors.Wrap(err, fmt.Sprintf("failed to read portal response from %v", C.GatewayUrl)))
	}
	defer response.Body.Close()

	switch response.StatusCode {
	case 201:
		a := CheckoutResponse{}
		err = json.Unmarshal(body, &a)
		if err != nil {
			return "", model.BadRequest(errors.Wrap(err, "failed to unmarshal zeus portal response"))
		}
		return a.Data.RedirectURL, nil
	case 400:
		return "", model.BadRequest(errors.Wrap(errors.New(string(body)),
			fmt.Sprintf("bad request error received from %v", C.GatewayUrl)))
	case 401:
		return "", model.Unauthorized(errors.Wrap(errors.New(string(body)),
			fmt.Sprintf("unauthorized request error received from %v", C.GatewayUrl)))
	default:
		return "", model.InternalError(errors.Wrap(errors.New(string(body)),
			fmt.Sprintf("internal request error received from %v", C.GatewayUrl)))
	}
}