tps-cli/internal/client/client.go

package client

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

	log "github.com/sirupsen/logrus"
	"golang.org/x/oauth2"
	"verint.com/tps/internal/properties"
)

type Config struct {
	Hostname string
	Username string
	Password string
	ClientID string
}

// Authenticate using OAuth2 password credentials
func GetAccessToken(c *Config) (string, error) {

	// Setup OAuth2 authentication.
	conf := oauth2.Config{
		ClientID:     c.ClientID,
		ClientSecret: "",
		Endpoint: oauth2.Endpoint{
			TokenURL: "https://" + c.Hostname + "/oidc-token-service/" + c.ClientID + "/token",
		},
		Scopes: []string{"oidc", "tags", "context_entitlements", "content_entitlements", "em_api_access"},
	}
	log.WithFields(log.Fields{"conf": fmt.Sprintf("%+v", conf)}).Debug()
	token, err := conf.PasswordCredentialsToken(context.TODO(), c.Username, c.Password)
	if err != nil {
		log.Error("Error retrieving AccessToken. Chech configuration, username and password")
		return "", err
	}
	log.WithFields(log.Fields{"token": fmt.Sprintf("%+v", token.AccessToken)}).Debug()
	return token.AccessToken, nil
}

func GetProperties(c *Config, q string) (string, error) {
	accessToken, err := GetAccessToken(c)

	if err != nil {
		log.Error(err)
		return "", err
	}

	url := "https://" + c.Hostname + "/tenant-properties-service/" + c.ClientID + "/properties?fields=name,value,lastModifiedDate,lastModifiedBy"

	if len(q) > 0 {
		url += "&q=" + q
	}

	req, err := http.NewRequest("GET", url, nil)
	if err != nil {
		log.Error(err)
		return "", err
	}
	req.Header.Add("Authorization", "OIDC_id_token "+accessToken)
	res, err := http.DefaultClient.Do(req)
	if err != nil {
		log.Error(err)
		return "", err
	}
	defer res.Body.Close()
	body, err := io.ReadAll(res.Body)
	if err != nil {
		log.Error(err)
		return "", err
	}

	return string(body), nil
}

func UpdateProperty(c *Config, p []properties.PropertyUpdateOrCreate) error {
	accessToken, err := GetAccessToken(c)

	if err != nil {
		return err
	}
	url := "https://" + c.Hostname + "/tenant-properties-service/" + c.ClientID + "/properties"

	body, err := json.Marshal(p)
	if err != nil {
		log.Error(err)
		return err
	}

	req, err := http.NewRequest("PATCH", url, bytes.NewBuffer(body))
	if err != nil {
		log.Error(err)
		return err
	}
	req.Header.Add("Authorization", "OIDC_id_token "+accessToken)
	req.Header.Add("Content-Type", "application/ld+json")

	res, err := http.DefaultClient.Do(req)
	if err != nil {
		log.Error(err)
		return err
	}
	defer res.Body.Close()
	response, err := io.ReadAll(res.Body)
	if err != nil {
		log.Error(err)
		return err
	}

	log.Debug(string(response))

	rp := []properties.PropertyUpdateOrCreate{}

	err = json.Unmarshal(response, &rp)
	if err != nil {
		log.Fatal(err)
		return err
	}
	if strings.HasPrefix("2", res.Status) {
		log.Error(rp[0].Status + rp[0].Description)
	}

	return nil
}

func DeleteProperty(c *Config, p []properties.PropertyDelete) error {
	accessToken, err := GetAccessToken(c)

	if err != nil {
		return err
	}
	url := "https://" + c.Hostname + "/tenant-properties-service/" + c.ClientID + "/properties"

	body, err := json.Marshal(p)
	if err != nil {
		log.Error(err)
		return err
	}

	req, err := http.NewRequest("PATCH", url, bytes.NewBuffer(body))
	if err != nil {
		log.Error(err)
		return err
	}
	req.Header.Add("Authorization", "OIDC_id_token "+accessToken)
	req.Header.Add("Content-Type", "application/ld+json")

	res, err := http.DefaultClient.Do(req)
	if err != nil {
		log.Error(err)
		return err
	}
	defer res.Body.Close()
	response, err := io.ReadAll(res.Body)
	if err != nil {
		log.Error(err)
		return err
	}

	log.Debug(string(response))

	rp := []properties.PropertyDelete{}

	err = json.Unmarshal(response, &rp)
	if err != nil {
		log.Error(err)
		return err
	}

	if strings.HasPrefix("2", res.Status) {
		log.Error(rp[0].Status + rp[0].Description)
	}

	return nil
}