v1.12.2

client.go

@@ -4,6 +4,7 @@ import (
 	"bytes"
 	"context"
 	"crypto/tls"
+	"encoding/base64"
 	"encoding/json"
 	"fmt"
 	"io"
@@ -21,6 +22,7 @@ import (
 	"repository.basistech.ru/BASIS/decort-golang-sdk/internal/validators"
 	"repository.basistech.ru/BASIS/decort-golang-sdk/pkg/cloudapi"
 	"repository.basistech.ru/BASIS/decort-golang-sdk/pkg/cloudbroker"
+	"repository.basistech.ru/BASIS/decort-golang-sdk/pkg/sdn"
 )
 
 // DecortClient is HTTP-client for platform
@@ -38,12 +40,6 @@ func New(cfg config.Config) *DecortClient {
 		cfg.Retries = 5
 	}
 
-	var expiryTime time.Time
-
-	if cfg.Token != "" {
-		expiryTime = time.Now().AddDate(0, 0, 1)
-	}
-
 	return &DecortClient{
 		decortURL: cfg.DecortURL,
 		client: &http.Client{
@@ -54,9 +50,8 @@ func New(cfg config.Config) *DecortClient {
 				},
 			},
 		},
-		cfg:        trimConfig(&cfg),
-		expiryTime: expiryTime,
-		mutex:      &sync.Mutex{},
+		cfg:   trimConfig(&cfg),
+		mutex: &sync.Mutex{},
 	}
 }
 
@@ -70,6 +65,11 @@ func (dc *DecortClient) CloudBroker() *cloudbroker.CloudBroker {
 	return cloudbroker.New(dc)
 }
 
+// SDN builder
+func (dc *DecortClient) SDN() *sdn.SDN {
+	return sdn.New(dc)
+}
+
 // DecortApiCall method for sending requests to the platform
 func (dc *DecortClient) DecortApiCall(ctx context.Context, method, url string, params interface{}) ([]byte, error) {
 
@@ -107,6 +107,47 @@ func (dc *DecortClient) DecortApiCall(ctx context.Context, method, url string, p
 	return respBytes, err
 }
 
+// DecortApiCallCtype method for sending requests to the platform with content type
+func (dc *DecortClient) DecortApiCallCtype(ctx context.Context, method, url, ctype string, params interface{}) ([]byte, error) {
+
+	var body *bytes.Buffer
+
+	switch ctype {
+	case constants.MIMESTREAM:
+		body = bytes.NewBuffer(params.([]byte))
+	case constants.MIMEJSON:
+		jsonBody, err := json.Marshal(params)
+		if err != nil {
+			return nil, err
+		}
+		body = bytes.NewBuffer(jsonBody)
+	default:
+		ctype = constants.MIMEPOSTForm
+		values, err := query.Values(params)
+		if err != nil {
+			return nil, err
+		}
+		body = bytes.NewBufferString(values.Encode())
+	}
+
+	req, err := http.NewRequestWithContext(ctx, method, dc.decortURL+constants.RESTMACHINE+url, body)
+	if err != nil {
+		return nil, err
+	}
+
+	// get token
+	if err = dc.getToken(ctx); err != nil {
+		return nil, err
+	}
+	// perform request
+	respBytes, err := dc.do(req, ctype)
+	if err != nil {
+		return nil, err
+	}
+
+	return respBytes, err
+}
+
 // DecortApiCallMP method for sending requests to the platform
 func (dc *DecortClient) DecortApiCallMP(ctx context.Context, method, url string, params interface{}) ([]byte, error) {
 	body, ctype, err := multiPartReq(params)
@@ -170,8 +211,13 @@ func (dc *DecortClient) getToken(ctx context.Context) error {
 
 	// save token in config
 	token := string(tokenBytes)
+	expiryTime, err := getTokenExp(token)
+	if err != nil {
+		return fmt.Errorf("cannot get expiry time: %w", err)
+	}
+
 	dc.cfg.Token = token
-	dc.expiryTime = time.Now().AddDate(0, 0, 1)
+	dc.expiryTime = expiryTime
 
 	return nil
 }
@@ -375,3 +421,29 @@ func trimConfig(cfg *config.Config) config.Config {
 	cfg.DecortURL = strings.TrimSuffix(cfg.DecortURL, "/")
 	return *cfg
 }
+
+func getTokenExp(token string) (time.Time, error) {
+	parts := strings.Split(token, ".")
+	if len(parts) != 3 {
+		return time.Time{}, fmt.Errorf("invalid token format")
+	}
+
+	payload, err := base64.RawURLEncoding.DecodeString(parts[1])
+	if err != nil {
+		return time.Time{}, fmt.Errorf("error decode payload from token: %w", err)
+	}
+
+	var claims map[string]interface{}
+	if err := json.Unmarshal(payload, &claims); err != nil {
+		return time.Time{}, err
+	}
+
+	exp, ok := claims["exp"]
+	if !ok {
+		return time.Time{}, fmt.Errorf("exp time bot found")
+	}
+
+	expTime := time.Unix(int64(exp.(float64)), 0)
+
+	return expTime, nil
+}