forked from TrueCloudLab/lego
125 lines
3.4 KiB
Go
125 lines
3.4 KiB
Go
package linodego
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
)
|
|
|
|
// Invoice structs reflect an invoice for billable activity on the account.
|
|
type Invoice struct {
|
|
DateStr string `json:"date"`
|
|
|
|
ID int `json:"id"`
|
|
Label string `json:"label"`
|
|
Total float32 `json:"total"`
|
|
Date *time.Time `json:"-"`
|
|
}
|
|
|
|
// InvoiceItem structs reflect an single billable activity associate with an Invoice
|
|
type InvoiceItem struct {
|
|
FromStr string `json:"from"`
|
|
ToStr string `json:"to"`
|
|
|
|
Label string `json:"label"`
|
|
Type string `json:"type"`
|
|
UnitPrice int `json:"unitprice"`
|
|
Quantity int `json:"quantity"`
|
|
Amount float32 `json:"amount"`
|
|
From *time.Time `json:"-"`
|
|
To *time.Time `json:"-"`
|
|
}
|
|
|
|
// InvoicesPagedResponse represents a paginated Invoice API response
|
|
type InvoicesPagedResponse struct {
|
|
*PageOptions
|
|
Data []Invoice `json:"data"`
|
|
}
|
|
|
|
// endpoint gets the endpoint URL for Invoice
|
|
func (InvoicesPagedResponse) endpoint(c *Client) string {
|
|
endpoint, err := c.Invoices.Endpoint()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return endpoint
|
|
}
|
|
|
|
// appendData appends Invoices when processing paginated Invoice responses
|
|
func (resp *InvoicesPagedResponse) appendData(r *InvoicesPagedResponse) {
|
|
resp.Data = append(resp.Data, r.Data...)
|
|
}
|
|
|
|
// ListInvoices gets a paginated list of Invoices against the Account
|
|
func (c *Client) ListInvoices(ctx context.Context, opts *ListOptions) ([]Invoice, error) {
|
|
response := InvoicesPagedResponse{}
|
|
err := c.listHelper(ctx, &response, opts)
|
|
for i := range response.Data {
|
|
response.Data[i].fixDates()
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return response.Data, nil
|
|
}
|
|
|
|
// fixDates converts JSON timestamps to Go time.Time values
|
|
func (v *Invoice) fixDates() *Invoice {
|
|
v.Date, _ = parseDates(v.DateStr)
|
|
return v
|
|
}
|
|
|
|
// fixDates converts JSON timestamps to Go time.Time values
|
|
func (v *InvoiceItem) fixDates() *InvoiceItem {
|
|
v.From, _ = parseDates(v.FromStr)
|
|
v.To, _ = parseDates(v.ToStr)
|
|
return v
|
|
}
|
|
|
|
// GetInvoice gets the a single Invoice matching the provided ID
|
|
func (c *Client) GetInvoice(ctx context.Context, id int) (*Invoice, error) {
|
|
e, err := c.Invoices.Endpoint()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
e = fmt.Sprintf("%s/%d", e, id)
|
|
r, err := coupleAPIErrors(c.R(ctx).SetResult(&Invoice{}).Get(e))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return r.Result().(*Invoice).fixDates(), nil
|
|
}
|
|
|
|
// InvoiceItemsPagedResponse represents a paginated Invoice Item API response
|
|
type InvoiceItemsPagedResponse struct {
|
|
*PageOptions
|
|
Data []InvoiceItem `json:"data"`
|
|
}
|
|
|
|
// endpointWithID gets the endpoint URL for InvoiceItems associated with a specific Invoice
|
|
func (InvoiceItemsPagedResponse) endpointWithID(c *Client, id int) string {
|
|
endpoint, err := c.InvoiceItems.endpointWithID(id)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return endpoint
|
|
}
|
|
|
|
// appendData appends InvoiceItems when processing paginated Invoice Item responses
|
|
func (resp *InvoiceItemsPagedResponse) appendData(r *InvoiceItemsPagedResponse) {
|
|
resp.Data = append(resp.Data, r.Data...)
|
|
}
|
|
|
|
// ListInvoiceItems gets the invoice items associated with a specific Invoice
|
|
func (c *Client) ListInvoiceItems(ctx context.Context, id int, opts *ListOptions) ([]InvoiceItem, error) {
|
|
response := InvoiceItemsPagedResponse{}
|
|
err := c.listHelperWithID(ctx, &response, id, opts)
|
|
for i := range response.Data {
|
|
response.Data[i].fixDates()
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return response.Data, nil
|
|
}
|