60 lines
1.5 KiB
Go
60 lines
1.5 KiB
Go
package wb
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"github.com/golang-jwt/jwt/v5"
|
|
"net/http"
|
|
"sipro-mps/internal/marketplace"
|
|
wbclient "sipro-mps/pkg/api/wb/client"
|
|
"time"
|
|
)
|
|
|
|
type WbAuthData struct {
|
|
Token string `json:"token"`
|
|
}
|
|
|
|
func NewWbAuthData(token string) WbAuthData {
|
|
return WbAuthData{
|
|
Token: token,
|
|
}
|
|
}
|
|
|
|
func DecodeWildberriesJwt(token []byte) (WbAuthData, jwt.MapClaims, error) {
|
|
var authData WbAuthData
|
|
err := json.Unmarshal(token, &authData)
|
|
if err != nil {
|
|
return authData, nil, fmt.Errorf("failed to unmarshal JWT: %w", err)
|
|
}
|
|
claims := jwt.MapClaims{}
|
|
_, _, err = jwt.NewParser().ParseUnverified(authData.Token, claims)
|
|
if err != nil {
|
|
return authData, nil, fmt.Errorf("invalid JWT: %w", err)
|
|
}
|
|
|
|
return authData, claims, nil
|
|
}
|
|
|
|
func GetClientFromMarketplace(mp *marketplace.Marketplace) (*wbclient.Client, error) {
|
|
|
|
authData, claims, err := DecodeWildberriesJwt(mp.AuthDataJson)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to decode Wildberries JWT")
|
|
}
|
|
exp := claims["exp"].(float64)
|
|
// chec if token is expired, for now unix date
|
|
now := float64(time.Now().Unix())
|
|
if exp < now {
|
|
return nil, fmt.Errorf("token is expired")
|
|
}
|
|
securityHandler := NewWildberriesSecurityHandler(authData.Token)
|
|
httpClient := &http.Client{
|
|
Transport: NewRateLimitTransport(),
|
|
}
|
|
client, err := wbclient.NewClient("https://content-api.wildberries.ru", securityHandler, wbclient.WithClient(httpClient))
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to create Wildberries client: %w", err)
|
|
}
|
|
return client, nil
|
|
}
|