add_on.go

package lago

import (
	"context"
	"encoding/json"
	"fmt"
	"time"

	"github.com/google/uuid"
)

type AddOnRequest struct {
	client *Client
}

type AddOnParams struct {
	AddOn *AddOnInput `json:"add_on"`
}

type AddOnInput struct {
	Name        string `json:"name,omitempty"`
	Code        string `json:"code,omitempty"`
	Description string `json:"description,omitempty"`

	AmountCents    int      `json:"amount_cents,omitempty"`
	AmountCurrency Currency `json:"amount_currency,omitempty"`
}

type AddOnListInput struct {
	PerPage int `json:"per_page,omitempty,string"`
	Page    int `json:"page,omitempty,string"`
}

type AddOnResult struct {
	AddOn  *AddOn   `json:"add_on,omitempty"`
	AddOns []AddOn  `json:"add_ons,omitempty"`
	Meta   Metadata `json:"meta,omitempty"`
}

type AddOn struct {
	LagoID      uuid.UUID `json:"lago_id,omitempty"`
	Name        string    `json:"name,omitempty"`
	Code        string    `json:"code,omitempty"`
	Description string    `json:"description,omitempty"`

	AmountCents    int      `json:"amount_cents,omitempty"`
	AmountCurrency Currency `json:"amount_currency,omitempty"`

	CreatedAt time.Time `json:"created_at,omitempty"`
}

type AppliedAddOnResult struct {
	AppliedAddOn *AppliedAddOn `json:"applied_add_on,omitempty"`
}

type ApplyAddOnParams struct {
	AppliedAddOn *ApplyAddOnInput `json:"applied_add_on"`
}

type ApplyAddOnInput struct {
	ExternalCustomerID string   `json:"external_customer_id,omitempty"`
	AddOnCode          string   `json:"add_on_code,omitempty"`
	AmountCents        int      `json:"amount_cents,omitempty"`
	AmountCurrency     Currency `json:"amount_currency,omitempty"`
}

type AppliedAddOn struct {
	LagoID             uuid.UUID `json:"lago_id,omitempty"`
	LagoAddOnID        uuid.UUID `json:"lago_add_on_id,omitempty"`
	LagoCustomerID     uuid.UUID `json:"lago_customer_id,omitempty"`
	ExternalCustomerID string    `json:"external_customer_id,omitempty"`

	AddOnCode      string   `json:"add_on_code,omitempty"`
	AmountCents    int      `json:"amount_cents,omitempty"`
	AmountCurrency Currency `json:"amount_currency,omitempty"`

	CreatedAt time.Time `json:"created_at,omitempty"`
}

func (c *Client) AddOn() *AddOnRequest {
	return &AddOnRequest{
		client: c,
	}
}

func (adr *AddOnRequest) Get(ctx context.Context, addOnCode string) (*AddOn, *Error) {
	subPath := fmt.Sprintf("%s/%s", "add_ons", addOnCode)
	clientRequest := &ClientRequest{
		Path:   subPath,
		Result: &AddOnResult{},
	}

	result, err := adr.client.Get(ctx, clientRequest)
	if err != nil {
		return nil, err
	}

	addOnResult, ok := result.(*AddOnResult)
	if !ok {
		return nil, &ErrorTypeAssert
	}

	return addOnResult.AddOn, nil
}

func (adr *AddOnRequest) GetList(ctx context.Context, addOnListInput *AddOnListInput) (*AddOnResult, *Error) {
	jsonQueryparams, err := json.Marshal(addOnListInput)
	if err != nil {
		return nil, &Error{Err: err}
	}

	queryParams := make(map[string]string)
	if err = json.Unmarshal(jsonQueryparams, &queryParams); err != nil {
		return nil, &Error{Err: err}
	}

	clientRequest := &ClientRequest{
		Path:        "add_ons",
		QueryParams: queryParams,
		Result:      &AddOnResult{},
	}

	result, clientErr := adr.client.Get(ctx, clientRequest)
	if clientErr != nil {
		return nil, clientErr
	}

	addOnResult, ok := result.(*AddOnResult)
	if !ok {
		return nil, &ErrorTypeAssert
	}

	return addOnResult, nil
}

func (adr *AddOnRequest) Create(ctx context.Context, addOnInput *AddOnInput) (*AddOn, *Error) {
	addOnParams := &AddOnParams{
		AddOn: addOnInput,
	}

	clientRequest := &ClientRequest{
		Path:   "add_ons",
		Result: &AddOnResult{},
		Body:   addOnParams,
	}

	result, err := adr.client.Post(ctx, clientRequest)
	if err != nil {
		return nil, err
	}

	addOnResult, ok := result.(*AddOnResult)
	if !ok {
		return nil, &ErrorTypeAssert
	}

	return addOnResult.AddOn, nil
}

func (adr *AddOnRequest) Update(ctx context.Context, addOnInput *AddOnInput) (*AddOn, *Error) {
	subPath := fmt.Sprintf("%s/%s", "add_ons", addOnInput.Code)
	addOnParams := &AddOnParams{
		AddOn: addOnInput,
	}

	clientRequest := &ClientRequest{
		Path:   subPath,
		Result: &AddOnResult{},
		Body:   addOnParams,
	}

	result, err := adr.client.Put(ctx, clientRequest)
	if err != nil {
		return nil, err
	}

	addOnResult, ok := result.(*AddOnResult)
	if !ok {
		return nil, &ErrorTypeAssert
	}

	return addOnResult.AddOn, nil
}

func (adr *AddOnRequest) Delete(ctx context.Context, addOnCode string) (*AddOn, *Error) {
	subPath := fmt.Sprintf("%s/%s", "add_ons", addOnCode)

	clientRequest := &ClientRequest{
		Path:   subPath,
		Result: &AddOnResult{},
	}

	result, err := adr.client.Delete(ctx, clientRequest)
	if err != nil {
		return nil, err
	}

	addOnResult, ok := result.(*AddOnResult)
	if !ok {
		return nil, &ErrorTypeAssert
	}

	return addOnResult.AddOn, nil
}

func (adr *AddOnRequest) ApplyToCustomer(ctx context.Context, applyAddOnInput *ApplyAddOnInput) (*AppliedAddOn, *Error) {
	applyAddOnParams := &ApplyAddOnParams{
		AppliedAddOn: applyAddOnInput,
	}

	clientRequest := &ClientRequest{
		Path:   "applied_add_ons",
		Result: &AppliedAddOnResult{},
		Body:   applyAddOnParams,
	}

	result, err := adr.client.Post(ctx, clientRequest)
	if err != nil {
		return nil, err
	}

	appliedAddOnResult, ok := result.(*AppliedAddOnResult)
	if !ok {
		return nil, &ErrorTypeAssert
	}

	return appliedAddOnResult.AppliedAddOn, nil
}