-
Notifications
You must be signed in to change notification settings - Fork 6
/
invoice.go
132 lines (113 loc) · 3.99 KB
/
invoice.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
// Copyright (c) 2021 AccelByte Inc. All Rights Reserved.
// This is licensed software from AccelByte Inc, for limitations
// and restrictions contact your company contract manager.
// Code generated. DO NOT EDIT.
package platform
import (
"io"
"github.com/AccelByte/accelbyte-go-sdk/platform-sdk/pkg/platformclient"
"github.com/AccelByte/accelbyte-go-sdk/platform-sdk/pkg/platformclient/invoice"
"github.com/AccelByte/accelbyte-go-sdk/platform-sdk/pkg/platformclientmodels"
"github.com/AccelByte/accelbyte-go-sdk/services-api/pkg/repository"
"github.com/AccelByte/accelbyte-go-sdk/services-api/pkg/utils"
"github.com/AccelByte/accelbyte-go-sdk/services-api/pkg/utils/auth"
"github.com/go-openapi/runtime/client"
)
type InvoiceService struct {
Client *platformclient.JusticePlatformService
ConfigRepository repository.ConfigRepository
TokenRepository repository.TokenRepository
FlightIdRepository *utils.FlightIdContainer
}
var tempFlightIdInvoice *string
func (aaa *InvoiceService) UpdateFlightId(flightId string) {
tempFlightIdInvoice = &flightId
}
func (aaa *InvoiceService) GetAuthSession() auth.Session {
return auth.Session{
aaa.TokenRepository,
aaa.ConfigRepository,
nil,
}
}
// Deprecated: 2022-01-10 - please use DownloadInvoiceDetailsShort instead.
func (aaa *InvoiceService) DownloadInvoiceDetails(input *invoice.DownloadInvoiceDetailsParams, writer io.Writer) (io.Writer, error) {
token, err := aaa.TokenRepository.GetToken()
if err != nil {
return nil, err
}
ok, err := aaa.Client.Invoice.DownloadInvoiceDetails(input, client.BearerToken(*token.AccessToken), writer)
if err != nil {
return nil, err
}
return ok.GetPayload(), nil
}
// Deprecated: 2022-01-10 - please use GenerateInvoiceSummaryShort instead.
func (aaa *InvoiceService) GenerateInvoiceSummary(input *invoice.GenerateInvoiceSummaryParams) (*platformclientmodels.InvoiceSummary, error) {
token, err := aaa.TokenRepository.GetToken()
if err != nil {
return nil, err
}
ok, unprocessableEntity, err := aaa.Client.Invoice.GenerateInvoiceSummary(input, client.BearerToken(*token.AccessToken))
if unprocessableEntity != nil {
return nil, unprocessableEntity
}
if err != nil {
return nil, err
}
return ok.GetPayload(), nil
}
func (aaa *InvoiceService) DownloadInvoiceDetailsShort(input *invoice.DownloadInvoiceDetailsParams, writer io.Writer) (io.Writer, error) {
authInfoWriter := input.AuthInfoWriter
if authInfoWriter == nil {
security := [][]string{
{"bearer"},
}
authInfoWriter = auth.AuthInfoWriter(aaa.GetAuthSession(), security, "")
}
if input.RetryPolicy == nil {
input.RetryPolicy = &utils.Retry{
MaxTries: utils.MaxTries,
Backoff: utils.NewConstantBackoff(0),
Transport: aaa.Client.Runtime.Transport,
RetryCodes: utils.RetryCodes,
}
}
if tempFlightIdInvoice != nil {
input.XFlightId = tempFlightIdInvoice
} else if aaa.FlightIdRepository != nil {
utils.GetDefaultFlightID().SetFlightID(aaa.FlightIdRepository.Value)
}
ok, err := aaa.Client.Invoice.DownloadInvoiceDetailsShort(input, authInfoWriter, writer)
if err != nil {
return nil, err
}
return ok.GetPayload(), nil
}
func (aaa *InvoiceService) GenerateInvoiceSummaryShort(input *invoice.GenerateInvoiceSummaryParams) (*platformclientmodels.InvoiceSummary, error) {
authInfoWriter := input.AuthInfoWriter
if authInfoWriter == nil {
security := [][]string{
{"bearer"},
}
authInfoWriter = auth.AuthInfoWriter(aaa.GetAuthSession(), security, "")
}
if input.RetryPolicy == nil {
input.RetryPolicy = &utils.Retry{
MaxTries: utils.MaxTries,
Backoff: utils.NewConstantBackoff(0),
Transport: aaa.Client.Runtime.Transport,
RetryCodes: utils.RetryCodes,
}
}
if tempFlightIdInvoice != nil {
input.XFlightId = tempFlightIdInvoice
} else if aaa.FlightIdRepository != nil {
utils.GetDefaultFlightID().SetFlightID(aaa.FlightIdRepository.Value)
}
ok, err := aaa.Client.Invoice.GenerateInvoiceSummaryShort(input, authInfoWriter)
if err != nil {
return nil, err
}
return ok.GetPayload(), nil
}