Merge pull request #1408 from c9s/narumi/wise

This commit is contained in:
なるみ 2023-11-17 19:48:08 +08:00 committed by GitHub
commit 08a09c2fee
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
8 changed files with 414 additions and 4 deletions

4
go.sum
View File

@ -82,10 +82,6 @@ github.com/bytedance/sonic v1.9.1 h1:6iJ6NqdoxCDr6mbY8h18oSO+cShGSMRGCEo7F2h0x8s
github.com/bytedance/sonic v1.9.1/go.mod h1:i736AoUSYt75HyZLoJW9ERYxcy6eaN6h4BZXU064P/U= github.com/bytedance/sonic v1.9.1/go.mod h1:i736AoUSYt75HyZLoJW9ERYxcy6eaN6h4BZXU064P/U=
github.com/c-bata/goptuna v0.8.1 h1:25+n1MLv0yvCsD56xv4nqIus3oLHL9GuPAZDLIqmX1U= github.com/c-bata/goptuna v0.8.1 h1:25+n1MLv0yvCsD56xv4nqIus3oLHL9GuPAZDLIqmX1U=
github.com/c-bata/goptuna v0.8.1/go.mod h1:knmS8+Iyq5PPy1YUeIEq0pMFR4Y6x7z/CySc9HlZTCY= github.com/c-bata/goptuna v0.8.1/go.mod h1:knmS8+Iyq5PPy1YUeIEq0pMFR4Y6x7z/CySc9HlZTCY=
github.com/c9s/requestgen v1.3.4 h1:kK2rIO3OAt9JoY5gT0OSkSpq0dy/+JeuI22FwSKpUrY=
github.com/c9s/requestgen v1.3.4/go.mod h1:wp4saiPdh0zLF5AkopGCqPQfy9Q5xvRh+TQBOA1l1r4=
github.com/c9s/requestgen v1.3.5 h1:iGYAP0rWQW3JOo+Z3S0SoenSt581IQ9mupJxRFCrCJs=
github.com/c9s/requestgen v1.3.5/go.mod h1:QwkZudcv84kJ8g9+E0RDTj+13btFXbTvv2aI+zbuLbc=
github.com/c9s/requestgen v1.3.6 h1:ul7dZ2uwGYjNBjreooRfSY10WTXvQmQSjZsHebz6QfE= github.com/c9s/requestgen v1.3.6 h1:ul7dZ2uwGYjNBjreooRfSY10WTXvQmQSjZsHebz6QfE=
github.com/c9s/requestgen v1.3.6/go.mod h1:QwkZudcv84kJ8g9+E0RDTj+13btFXbTvv2aI+zbuLbc= github.com/c9s/requestgen v1.3.6/go.mod h1:QwkZudcv84kJ8g9+E0RDTj+13btFXbTvv2aI+zbuLbc=
github.com/c9s/rockhopper v1.2.2-0.20220617053729-ffdc87df194b h1:wT8c03PHLv7+nZUIGqxAzRvIfYHNxMCNVWwvdGkOXTs= github.com/c9s/rockhopper v1.2.2-0.20220617053729-ffdc87df194b h1:wT8c03PHLv7+nZUIGqxAzRvIfYHNxMCNVWwvdGkOXTs=

View File

@ -0,0 +1,25 @@
# Wise
[Wise API Docs](https://docs.wise.com/api-docs)
```go
c := wise.NewClient()
c.Auth(os.Getenv("WISE_TOKEN"))
ctx := context.Background()
rates, err := c.QueryRate(ctx, "USD", "TWD")
if err != nil {
panic(err)
}
fmt.Printf("%+v\n", rates)
// or
now := time.Now()
rates, err = c.QueryRateHistory(ctx, "USD", "TWD", now.Add(-time.Hour*24*7), now, types.Interval1h)
if err != nil {
panic(err)
}
for _, rate := range rates {
fmt.Printf("%+v\n", rate)
}
```

View File

@ -0,0 +1,80 @@
package wise
import (
"context"
"fmt"
"net/http"
"net/url"
"time"
"github.com/c9s/bbgo/pkg/types"
"github.com/c9s/requestgen"
)
const (
defaultHTTPTimeout = time.Second * 15
defaultBaseURL = "https://api.transferwise.com"
sandboxBaseURL = "https://api.sandbox.transferwise.tech"
)
type Client struct {
requestgen.BaseAPIClient
token string
}
func NewClient() *Client {
u, err := url.Parse(defaultBaseURL)
if err != nil {
panic(err)
}
return &Client{
BaseAPIClient: requestgen.BaseAPIClient{
BaseURL: u,
HttpClient: &http.Client{
Timeout: defaultHTTPTimeout,
},
},
}
}
func (c *Client) Auth(token string) {
c.token = token
}
func (c *Client) NewAuthenticatedRequest(ctx context.Context, method, refURL string, params url.Values, payload interface{}) (*http.Request, error) {
req, err := c.NewRequest(ctx, method, refURL, params, payload)
if err != nil {
return nil, err
}
req.Header.Add("Content-Type", "application/json")
if c.token != "" {
req.Header.Set("Authorization", "Bearer "+c.token)
}
return req, nil
}
func (c *Client) QueryRate(ctx context.Context, source string, target string) ([]Rate, error) {
req := c.NewRateRequest().Source(source).Target(target)
return req.Do(ctx)
}
func (c *Client) QueryRateHistory(ctx context.Context, source string, target string, from time.Time, to time.Time, interval types.Interval) ([]Rate, error) {
req := c.NewRateRequest().Source(source).Target(target).From(from).To(to)
switch interval {
case types.Interval1h:
req.Group(GroupHour)
case types.Interval1d:
req.Group(GroupDay)
case types.Interval1m:
req.Group(GroupMinute)
default:
return nil, fmt.Errorf("unsupported interval: %s", interval)
}
return req.Do(ctx)
}

View File

@ -0,0 +1,9 @@
package wise
type Group string
const (
GroupMinute = Group("minute")
GroupHour = Group("hour")
GroupDay = Group("day")
)

View File

@ -0,0 +1,10 @@
package wise
import "github.com/c9s/bbgo/pkg/fixedpoint"
type Rate struct {
Value fixedpoint.Value `json:"rate"`
Target string `json:"target"`
Source string `json:"source"`
Time Time `json:"time"`
}

View File

@ -0,0 +1,27 @@
package wise
import (
"time"
"github.com/c9s/requestgen"
)
// https://docs.wise.com/api-docs/api-reference/rate
//go:generate requestgen -method GET -url "/v1/rates" -type RateRequest -responseType []Rate
type RateRequest struct {
client requestgen.AuthenticatedAPIClient
source string `param:"source"`
target string `param:"target"`
time *time.Time `param:"time" timeFormat:"2006-01-02T15:04:05-0700"`
from *time.Time `param:"from" timeFormat:"2006-01-02T15:04:05-0700"`
to *time.Time `param:"to" timeFormat:"2006-01-02T15:04:05-0700"`
group *Group `param:"group"`
}
func (c *Client) NewRateRequest() *RateRequest {
return &RateRequest{
client: c,
}
}

View File

@ -0,0 +1,229 @@
// Code generated by "requestgen -method GET -url /v1/rates -type RateRequest -responseType []Rate"; DO NOT EDIT.
package wise
import (
"context"
"encoding/json"
"fmt"
"net/url"
"reflect"
"regexp"
"time"
)
func (r *RateRequest) Source(source string) *RateRequest {
r.source = source
return r
}
func (r *RateRequest) Target(target string) *RateRequest {
r.target = target
return r
}
func (r *RateRequest) Time(time time.Time) *RateRequest {
r.time = &time
return r
}
func (r *RateRequest) From(from time.Time) *RateRequest {
r.from = &from
return r
}
func (r *RateRequest) To(to time.Time) *RateRequest {
r.to = &to
return r
}
func (r *RateRequest) Group(group Group) *RateRequest {
r.group = &group
return r
}
// GetQueryParameters builds and checks the query parameters and returns url.Values
func (r *RateRequest) GetQueryParameters() (url.Values, error) {
var params = map[string]interface{}{}
query := url.Values{}
for _k, _v := range params {
query.Add(_k, fmt.Sprintf("%v", _v))
}
return query, nil
}
// GetParameters builds and checks the parameters and return the result in a map object
func (r *RateRequest) GetParameters() (map[string]interface{}, error) {
var params = map[string]interface{}{}
// check source field -> json key source
source := r.source
// assign parameter of source
params["source"] = source
// check target field -> json key target
target := r.target
// assign parameter of target
params["target"] = target
// check time field -> json key time
if r.time != nil {
time := *r.time
// assign parameter of time
params["time"] = time.Format("2006-01-02T15:04:05-0700")
} else {
}
// check from field -> json key from
if r.from != nil {
from := *r.from
// assign parameter of from
params["from"] = from.Format("2006-01-02T15:04:05-0700")
} else {
}
// check to field -> json key to
if r.to != nil {
to := *r.to
// assign parameter of to
params["to"] = to.Format("2006-01-02T15:04:05-0700")
} else {
}
// check group field -> json key group
if r.group != nil {
group := *r.group
// assign parameter of group
params["group"] = group
} else {
}
return params, nil
}
// GetParametersQuery converts the parameters from GetParameters into the url.Values format
func (r *RateRequest) GetParametersQuery() (url.Values, error) {
query := url.Values{}
params, err := r.GetParameters()
if err != nil {
return query, err
}
for _k, _v := range params {
if r.isVarSlice(_v) {
r.iterateSlice(_v, func(it interface{}) {
query.Add(_k+"[]", fmt.Sprintf("%v", it))
})
} else {
query.Add(_k, fmt.Sprintf("%v", _v))
}
}
return query, nil
}
// GetParametersJSON converts the parameters from GetParameters into the JSON format
func (r *RateRequest) GetParametersJSON() ([]byte, error) {
params, err := r.GetParameters()
if err != nil {
return nil, err
}
return json.Marshal(params)
}
// GetSlugParameters builds and checks the slug parameters and return the result in a map object
func (r *RateRequest) GetSlugParameters() (map[string]interface{}, error) {
var params = map[string]interface{}{}
return params, nil
}
func (r *RateRequest) applySlugsToUrl(url string, slugs map[string]string) string {
for _k, _v := range slugs {
needleRE := regexp.MustCompile(":" + _k + "\\b")
url = needleRE.ReplaceAllString(url, _v)
}
return url
}
func (r *RateRequest) iterateSlice(slice interface{}, _f func(it interface{})) {
sliceValue := reflect.ValueOf(slice)
for _i := 0; _i < sliceValue.Len(); _i++ {
it := sliceValue.Index(_i).Interface()
_f(it)
}
}
func (r *RateRequest) isVarSlice(_v interface{}) bool {
rt := reflect.TypeOf(_v)
switch rt.Kind() {
case reflect.Slice:
return true
}
return false
}
func (r *RateRequest) GetSlugsMap() (map[string]string, error) {
slugs := map[string]string{}
params, err := r.GetSlugParameters()
if err != nil {
return slugs, nil
}
for _k, _v := range params {
slugs[_k] = fmt.Sprintf("%v", _v)
}
return slugs, nil
}
// GetPath returns the request path of the API
func (r *RateRequest) GetPath() string {
return "/v1/rates"
}
// Do generates the request object and send the request object to the API endpoint
func (r *RateRequest) Do(ctx context.Context) ([]Rate, error) {
// empty params for GET operation
var params interface{}
query, err := r.GetParametersQuery()
if err != nil {
return nil, err
}
var apiURL string
apiURL = r.GetPath()
req, err := r.client.NewAuthenticatedRequest(ctx, "GET", apiURL, query, params)
if err != nil {
return nil, err
}
response, err := r.client.SendRequest(req)
if err != nil {
return nil, err
}
var apiResponse []Rate
if err := response.DecodeJSON(&apiResponse); err != nil {
return nil, err
}
type responseValidator interface {
Validate() error
}
validator, ok := interface{}(apiResponse).(responseValidator)
if ok {
if err := validator.Validate(); err != nil {
return nil, err
}
}
return apiResponse, nil
}

View File

@ -0,0 +1,34 @@
package wise
import (
"encoding/json"
"time"
)
const layout = "2006-01-02T15:04:05-0700"
type Time time.Time
func (t Time) Time() time.Time {
return time.Time(t)
}
func (t Time) String() string {
return time.Time(t).Format(layout)
}
func (t *Time) UnmarshalJSON(data []byte) error {
var s string
err := json.Unmarshal(data, &s)
if err != nil {
return err
}
parsed, err := time.Parse(layout, s)
if err != nil {
return err
}
*t = Time(parsed)
return nil
}