adjustments.go

package rize

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

	"github.com/google/go-querystring/query"
)

// Handles all Adjustment operations
type adjustmentService service

// Adjustment data type
type Adjustment struct {
	UID                 string          `json:"uid,omitempty"`
	ExternalUID         string          `json:"external_uid,omitempty"`
	CustomerUID         string          `json:"customer_uid,omitempty"`
	USDAdjustmentAmount string          `json:"usd_adjustment_amount,omitempty"`
	AdjustmentType      *AdjustmentType `json:"adjustment_type,omitempty"`
	CreatedAt           time.Time       `json:"created_at,omitempty"`
	Status              string          `json:"status,omitempty"`
}

// AdjustmentType data type
type AdjustmentType struct {
	UID         string `json:"uid,omitempty"`
	Name        string `json:"name,omitempty"`
	Description string `json:"description,omitempty"`
	Fee         bool   `json:"fee,omitempty"`
	ProgramUID  string `json:"program_uid,omitempty"`
	Deprecated  bool   `json:"deprecated,omitempty"`
}

// AdjustmentListParams builds the query parameters used in querying Adjustments
type AdjustmentListParams struct {
	CustomerUID            string `url:"customer_uid,omitempty" json:"customer_uid,omitempty"`
	AdjustmentTypeUID      string `url:"adjustment_type_uid,omitempty" json:"adjustment_type_uid,omitempty"`
	ExternalUID            string `url:"external_uid,omitempty" json:"external_uid,omitempty"`
	USDAdjustmentAmountMax int    `url:"usd_adjustment_amount_max,omitempty" json:"usd_adjustment_amount_max,omitempty"`
	USDAdjustmentAmountMin int    `url:"usd_adjustment_amount_min,omitempty" json:"usd_adjustment_amount_min,omitempty"`
	Sort                   string `url:"sort,omitempty" json:"sort,omitempty"`
}

// AdjustmentCreateParams are the body params used when creating a new Adjustment
type AdjustmentCreateParams struct {
	ExternalUID         string `json:"external_uid,omitempty"`
	CustomerUID         string `json:"customer_uid"`
	USDAdjustmentAmount string `json:"usd_adjustment_amount"`
	AdjustmentTypeUID   string `json:"adjustment_type_uid"`
}

// AdjustmentTypeListParams builds the query parameters used in querying Adjustment Types
type AdjustmentTypeListParams struct {
	CustomerUID    string `url:"customer_uid,omitempty" json:"customer_uid,omitempty"`
	ProgramUID     string `url:"program_uid,omitempty" json:"program_uid,omitempty"`
	ShowDeprecated bool   `url:"show_deprecated,omitempty" json:"show_deprecated,omitempty"`
}

// AdjustmentListResponse is an API response containing a list of Adjustments
type AdjustmentListResponse struct {
	ListResponse
	Data []*Adjustment `json:"data"`
}

// AdjustmentTypeListResponse is an API response containing a list of Adjustments Types
type AdjustmentTypeListResponse struct {
	ListResponse
	Data []*AdjustmentType `json:"data"`
}

// List retrieves a list of Adjustments filtered by the given parameters
func (a *adjustmentService) List(ctx context.Context, params *AdjustmentListParams) (*AdjustmentListResponse, error) {
	// Build AdjustmentListParams into query string params
	v, err := query.Values(params)
	if err != nil {
		return nil, err
	}

	res, err := a.client.doRequest(ctx, http.MethodGet, "adjustments", v, nil)
	if err != nil {
		return nil, err
	}
	defer res.Body.Close()

	body, err := io.ReadAll(res.Body)
	if err != nil {
		return nil, err
	}

	response := &AdjustmentListResponse{}
	if err = json.Unmarshal(body, response); err != nil {
		return nil, err
	}

	return response, nil
}

// Create a new Adjustment with the provided specification
func (a *adjustmentService) Create(ctx context.Context, params *AdjustmentCreateParams) (*Adjustment, error) {
	if params.CustomerUID == "" ||
		params.USDAdjustmentAmount == "" ||
		params.AdjustmentTypeUID == "" {
		return nil, fmt.Errorf("CustomerUID, USDAdjustmentAmount and AdjustmentTypeUID are required")
	}

	bytesMessage, err := json.Marshal(params)
	if err != nil {
		return nil, err
	}

	res, err := a.client.doRequest(ctx, http.MethodPost, "adjustments", nil, bytes.NewBuffer(bytesMessage))
	if err != nil {
		return nil, err
	}
	defer res.Body.Close()

	body, err := io.ReadAll(res.Body)
	if err != nil {
		return nil, err
	}

	response := &Adjustment{}
	if err = json.Unmarshal(body, response); err != nil {
		return nil, err
	}

	return response, nil
}

// Get returns a single Adjustment
func (a *adjustmentService) Get(ctx context.Context, uid string) (*Adjustment, error) {
	if uid == "" {
		return nil, fmt.Errorf("UID is required")
	}

	res, err := a.client.doRequest(ctx, http.MethodGet, fmt.Sprintf("adjustments/%s", uid), nil, nil)
	if err != nil {
		return nil, err
	}
	defer res.Body.Close()

	body, err := io.ReadAll(res.Body)
	if err != nil {
		return nil, err
	}

	response := &Adjustment{}
	if err = json.Unmarshal(body, response); err != nil {
		return nil, err
	}

	return response, nil
}

// ListAdjustmentTypes retrieves a list of Adjustment Types filtered by the given parameters
func (a *adjustmentService) ListAdjustmentTypes(ctx context.Context, params *AdjustmentTypeListParams) (*AdjustmentTypeListResponse, error) {
	v, err := query.Values(params)
	if err != nil {
		return nil, err
	}

	res, err := a.client.doRequest(ctx, http.MethodGet, "adjustment_types", v, nil)
	if err != nil {
		return nil, err
	}
	defer res.Body.Close()

	body, err := io.ReadAll(res.Body)
	if err != nil {
		return nil, err
	}

	response := &AdjustmentTypeListResponse{}
	if err = json.Unmarshal(body, response); err != nil {
		return nil, err
	}

	return response, nil
}

// GetAdjustmentType returns a single Adjustment Type
func (a *adjustmentService) GetAdjustmentType(ctx context.Context, uid string) (*AdjustmentType, error) {
	if uid == "" {
		return nil, fmt.Errorf("UID is required")
	}

	res, err := a.client.doRequest(ctx, http.MethodGet, fmt.Sprintf("adjustment_types/%s", uid), nil, nil)
	if err != nil {
		return nil, err
	}
	defer res.Body.Close()

	body, err := io.ReadAll(res.Body)
	if err != nil {
		return nil, err
	}

	response := &AdjustmentType{}
	if err = json.Unmarshal(body, response); err != nil {
		return nil, err
	}

	return response, nil
}