Merge pull request 'viettel-ffm' (#19) from viettel-ffm into master

Reviewed-on: #19

const.go

@@ -0,0 +1,9 @@
+package tpl
+
+// ENV ...
+type ENV string
+
+const (
+	EnvStaging ENV = "STAGING"
+	EnvProd    ENV = "PROD"
+)

go.mod

@@ -1,10 +1,11 @@
 module git.selly.red/Selly-Modules/3pl
 
-go 1.17
+go 1.20
 
 require (
 	git.selly.red/Selly-Modules/logger v0.0.2-0.20221010053254-567df039afdb
 	git.selly.red/Selly-Modules/natsio v1.0.3-0.20231020090841-5edec97ee393
+	github.com/go-resty/resty/v2 v2.7.0
 	github.com/nats-io/nats.go v1.17.0
 	github.com/thoas/go-funk v0.9.2
 )
@@ -14,7 +15,6 @@ require (
 	github.com/elastic/go-licenser v0.4.1 // indirect
 	github.com/elastic/go-sysinfo v1.1.1 // indirect
 	github.com/elastic/go-windows v1.0.1 // indirect
-	github.com/go-resty/resty/v2 v2.7.0 // indirect
 	github.com/golang/snappy v0.0.3 // indirect
 	github.com/jcchavezs/porto v0.4.0 // indirect
 	github.com/joeshaw/multierror v0.0.0-20140124173710-69b34d4ec901 // indirect

go.sum

@@ -1,9 +1,5 @@
 git.selly.red/Selly-Modules/logger v0.0.2-0.20221010053254-567df039afdb h1:AmcYd88IcdSkH+NEvKyJLT7psidSkjJQT/nAg/KuzFk=
 git.selly.red/Selly-Modules/logger v0.0.2-0.20221010053254-567df039afdb/go.mod h1:Q1//Z6HRmfa7VyjH2J6YyT0YV2jT8+K6SIgwnYuS4II=
-git.selly.red/Selly-Modules/natsio v1.0.2-0.20221010041139-c11419a3ad33 h1:GvQjelaV4XZm++AOihYAKOD6k9510aMAr6B6MGnrXPs=
-git.selly.red/Selly-Modules/natsio v1.0.2-0.20221010041139-c11419a3ad33/go.mod h1:KNODhfeBqxRmHHQHHU+p3JfH42t8s5aNxfgr6X8fr6g=
-git.selly.red/Selly-Modules/natsio v1.0.3-0.20231006093940-b3bde5cd0960 h1:wL/BW1xGoB/EXeA2HtxT6Nr/cXpPJYVNfToV3aFtGls=
-git.selly.red/Selly-Modules/natsio v1.0.3-0.20231006093940-b3bde5cd0960/go.mod h1:KNODhfeBqxRmHHQHHU+p3JfH42t8s5aNxfgr6X8fr6g=
 git.selly.red/Selly-Modules/natsio v1.0.3-0.20231020090841-5edec97ee393 h1:43kE03FW3NONfE6hXlghafS1d233dfc7grlFqd+15SA=
 git.selly.red/Selly-Modules/natsio v1.0.3-0.20231020090841-5edec97ee393/go.mod h1:KNODhfeBqxRmHHQHHU+p3JfH42t8s5aNxfgr6X8fr6g=
 github.com/armon/go-radix v1.0.0 h1:F4z6KzEeeQIMeLFa97iZU6vupzoecKdU5TX24SNppXI=

partnerapi/viettelffm/model.go

@@ -0,0 +1,79 @@
+package viettelffm
+
+type AuthRes struct {
+	AccessToken      string `json:"access_token"`
+	ExpiresIn        int    `json:"expires_in"`
+	RefreshExpiresIn int    `json:"refresh_expires_in"`
+	RefreshToken     string `json:"refresh_token"`
+	TokenType        string `json:"token_type"`
+	NotBeforePolicy  int    `json:"not-before-policy"`
+	SessionState     string `json:"session_state"`
+	Scope            string `json:"scope"`
+}
+
+type UpdateLogisticInfoPayload struct {
+	PrintLabelLink string `json:"print_label_link"`
+	TrackingCode   string `json:"tracking_code"`
+	OrCode         string `json:"or_code,omitempty"`
+}
+
+type CancelORPayload struct {
+	OrID int `json:"or_id"`
+}
+
+type ORPayload struct {
+	OrProductLines []ORProductLine `json:"or_product_lines"`
+	AmountPaid     float64         `json:"amount_paid"`
+	CodAmount      float64         `json:"cod_amount"`
+	CodType        string          `json:"cod_type"`
+	Note           string          `json:"note"`
+	OrCode         string          `json:"or_code"`
+	OrType         string          `json:"or_type"`
+	ShippingType   string          `json:"shipping_type"`
+	CustomerName   string          `json:"customer_name"`
+	CustomerEmail  string          `json:"customer_email"`
+	PackType       string          `json:"pack_type"`
+	PriorityType   string          `json:"priority_type"`
+	WarehouseCode  string          `json:"warehouse_code"`
+}
+
+type ORProductLine struct {
+	Discount             float64              `json:"discount"`
+	Price                float64              `json:"price"`
+	Product              Product              `json:"product"`
+	Quantity             int                  `json:"quantity"`
+	SpecifiedProductLine SpecifiedProductLine `json:"specified_product_line"`
+	Unit                 string               `json:"unit"`
+}
+
+type SpecifiedProductLine struct {
+	ProductConditionType ProdCondType `json:"product_condition_type"`
+}
+
+type Product struct {
+	PartnerSku string `json:"partner_sku"`
+	SKU        string `json:"sku"`
+}
+
+type ProdCondType struct {
+	Name string `json:"name"`
+}
+
+type ORResult struct {
+	OrCode string `json:"or_code"`
+	OrId   int    `json:"or_id"`
+	Status string `json:"status"`
+}
+
+type Webhook struct {
+	Data WebhookData `json:"data"`
+}
+
+type WebhookData struct {
+	OrId          int     `json:"or_id"`
+	OrCode        string  `json:"or_code"`
+	PartnerOrCode string  `json:"partner_or_code"`
+	Status        string  `json:"status"`
+	TotalPrice    float64 `json:"total_price"`
+	ShippingFee   int     `json:"shipping_fee"`
+}

partnerapi/viettelffm/viettel_ffm.go

@@ -0,0 +1,233 @@
+package viettelffm
+
+import (
+	"errors"
+	"fmt"
+	"net/http"
+	"net/url"
+	"time"
+
+	"git.selly.red/Selly-Modules/natsio"
+	"git.selly.red/Selly-Modules/natsio/model"
+	"git.selly.red/Selly-Modules/natsio/subject"
+
+	tpl "git.selly.red/Selly-Modules/3pl"
+	"git.selly.red/Selly-Modules/3pl/util/httputil"
+	"git.selly.red/Selly-Modules/3pl/util/pjson"
+)
+
+const (
+	baseURLStag     = "https://stg-gw.viettelpost.vn"
+	baseURLAuthStag = "https://stg-keycloak.viettelpost.vn"
+
+	// TODO: update prod url
+	baseURLProd     = ""
+	baseURLAuthProd = ""
+
+	pathAuth                 = "/realms/wms/protocol/openid-connect/token"
+	pathUpdateORLogisticInfo = "/wms-core/api/v1/obms/outbound-request/outbound-request-partner/%s"
+	pathCreateOR             = "/wms-core/api/v1/obms/outbound-request/outbound-request-partner/hab"
+	pathCancelOR             = "/wms-core/api/v1/obms/outbound-request/cancel"
+
+	logTarget = "viettel-ffm"
+)
+
+var (
+	baseURLENVMapping = map[tpl.ENV]string{
+		tpl.EnvProd:    baseURLProd,
+		tpl.EnvStaging: baseURLStag,
+	}
+)
+
+type Client struct {
+	env      tpl.ENV
+	username string
+	password string
+
+	natsClient          natsio.Server
+	authInfo            *AuthRes
+	authTokenExpireTime time.Time
+}
+
+// NewClient generate OnPoint client
+func NewClient(env tpl.ENV, user, pwd string, nc natsio.Server) (*Client, error) {
+	if user == "" || pwd == "" {
+		return nil, errors.New("viettelffm: cannot init with empty api key")
+	}
+	return &Client{
+		env:        env,
+		username:   user,
+		password:   pwd,
+		natsClient: nc,
+	}, nil
+}
+
+func (c *Client) CreateOR(p ORPayload) (*ORResult, error) {
+	apiURL := c.getBaseURL() + pathCreateOR
+	token, err := c.getToken()
+	if err != nil {
+		return nil, err
+	}
+	natsPayload := model.CommunicationRequestHttp{
+		ResponseImmediately: true,
+		Payload: model.HttpRequest{
+			URL:    apiURL,
+			Method: http.MethodPost,
+			Data:   pjson.ToJSONString(p),
+			Header: map[string]string{
+				httputil.HeaderKeyAuthorization: fmt.Sprintf("Bearer %s", token),
+				httputil.HeaderKeyContentType:   httputil.HeaderValueApplicationJSON,
+			},
+		},
+		LogTarget: logTarget,
+	}
+	r, err := c.requestHttpViaNats(natsPayload)
+	if err != nil {
+		return nil, fmt.Errorf("viettelffm.Client.CreateOR - requestHttpViaNats %v", err)
+	}
+
+	res := r.Response
+	if res.StatusCode >= http.StatusBadRequest {
+		return nil, fmt.Errorf("viettelffm.Client.CreateOR - requestHttpViaNats %s", res.Body)
+	}
+	var data ORResult
+	if err = r.ParseResponseData(&data); err != nil {
+		return nil, fmt.Errorf("viettelffm.Client.CreateOR - requestHttpViaNats parse response %v, %s", err, res.Body)
+	}
+	return &data, nil
+}
+
+func (c *Client) UpdateORLogisticInfo(p UpdateLogisticInfoPayload) error {
+	apiURL := c.getBaseURL() + fmt.Sprintf(pathUpdateORLogisticInfo, p.OrCode)
+	p.OrCode = ""
+	token, err := c.getToken()
+	if err != nil {
+		return err
+	}
+	natsPayload := model.CommunicationRequestHttp{
+		ResponseImmediately: true,
+		Payload: model.HttpRequest{
+			URL:    apiURL,
+			Method: http.MethodPut,
+			Data:   pjson.ToJSONString(p),
+			Header: map[string]string{
+				httputil.HeaderKeyAuthorization: fmt.Sprintf("Bearer %s", token),
+				httputil.HeaderKeyContentType:   httputil.HeaderValueApplicationJSON,
+			},
+		},
+		LogTarget: logTarget,
+	}
+	r, err := c.requestHttpViaNats(natsPayload)
+	if err != nil {
+		return err
+	}
+	res := r.Response
+	if res.StatusCode >= http.StatusBadRequest {
+		return fmt.Errorf("viettelffm.Client.UpdateOutboundRequestLogisticInfo - requestHttpViaNats %s", res.Body)
+	}
+	return nil
+}
+
+func (c *Client) CancelOR(p CancelORPayload) error {
+	apiURL := c.getBaseURL() + pathCancelOR
+	token, err := c.getToken()
+	if err != nil {
+		return err
+	}
+	natsPayload := model.CommunicationRequestHttp{
+		ResponseImmediately: true,
+		Payload: model.HttpRequest{
+			URL:    apiURL,
+			Method: http.MethodPost,
+			Data:   pjson.ToJSONString(p),
+			Header: map[string]string{
+				httputil.HeaderKeyAuthorization: fmt.Sprintf("Bearer %s", token),
+				httputil.HeaderKeyContentType:   httputil.HeaderValueApplicationJSON,
+			},
+		},
+		LogTarget: logTarget,
+	}
+	r, err := c.requestHttpViaNats(natsPayload)
+	if err != nil {
+		return err
+	}
+	res := r.Response
+	if res.StatusCode >= http.StatusBadRequest {
+		return fmt.Errorf("viettelffm.Client.CancelOR - requestHttpViaNats %s", res.Body)
+	}
+	return nil
+}
+
+// Auth ...
+func (c *Client) Auth() (*AuthRes, error) {
+	v := url.Values{}
+	v.Set("username", c.username)
+	v.Set("password", c.password)
+	v.Set("grant_type", "password")
+	v.Set("client_id", "wms_account")
+	b := v.Encode()
+	header := map[string]string{
+		httputil.HeaderKeyContentType: httputil.HeaderValueApplicationURLEncoded,
+	}
+	bURL := baseURLAuthStag
+	if c.env == tpl.EnvProd {
+		bURL = baseURLAuthProd
+	}
+	api := bURL + pathAuth
+	natsPayload := model.CommunicationRequestHttp{
+		ResponseImmediately: true,
+		Payload: model.HttpRequest{
+			URL:    api,
+			Method: http.MethodPost,
+			Data:   b,
+			Header: header,
+		},
+		LogTarget: logTarget,
+	}
+	r, err := c.requestHttpViaNats(natsPayload)
+	if err != nil {
+		return nil, fmt.Errorf("viettelffm.Client.Auth - requestHttpViaNats: payload %+v, err %v\n", natsPayload, err)
+	}
+
+	res := r.Response
+	if res.StatusCode >= http.StatusBadRequest {
+		return nil, fmt.Errorf("viettelffm.Client.Auth - requestHttpViaNats %s", res.Body)
+	}
+	var data AuthRes
+	if err = r.ParseResponseData(&data); err != nil {
+		return nil, fmt.Errorf("viettelffm.Client.Auth - requestHttpViaNats parse response %v, %s", err, res.Body)
+	}
+	return &data, nil
+}
+
+func (c *Client) getToken() (string, error) {
+	if c.authInfo == nil || time.Now().After(c.authTokenExpireTime) {
+		auth, err := c.Auth()
+		if err != nil {
+			return "", err
+		}
+		c.authInfo = auth
+		c.authTokenExpireTime = time.Now().Add(time.Duration(auth.ExpiresIn) * time.Second).Add(-time.Minute)
+	}
+	return c.authInfo.AccessToken, nil
+}
+
+func (c *Client) getBaseURL() string {
+	return baseURLENVMapping[c.env]
+}
+
+func (c *Client) requestHttpViaNats(data model.CommunicationRequestHttp) (*model.CommunicationHttpResponse, error) {
+	b := pjson.ToBytes(data)
+	msg, err := c.natsClient.Request(subject.Communication.RequestHTTP, b)
+	if err != nil {
+		return nil, fmt.Errorf("viettelffm.Client.requestHttpViaNats err: %v, url %s", err, data.Payload.URL)
+	}
+	var r model.CommunicationHttpResponse
+	if err = pjson.Unmarshal(msg.Data, &r); err != nil {
+		return nil, fmt.Errorf("viettelffm.Client.requestHttpViaNats parse data err: %v, url %s, data %s", err, data.Payload.URL, string(msg.Data))
+	}
+	if r.Response == nil {
+		return nil, fmt.Errorf("viettelffm.Client.requestHttpViaNats empty reponse")
+	}
+	return &r, nil
+}

util/httputil/const.go

@@ -2,6 +2,7 @@ package httputil
 
 const (
 	HeaderKeyContentType   = "Content-Type"
+	HeaderKeyAuthorization = "Authorization"
 
 	HeaderValueApplicationJSON       = "application/json"
 	HeaderValueApplicationURLEncoded = "application/x-www-form-urlencoded"