92 lines
2 KiB
Go
92 lines
2 KiB
Go
|
// +build codegen
|
||
|
|
|
||
|
|
package api
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Paginator keeps track of pagination configuration for an API operation.
|
||
|
|
type Paginator struct {
|
||
|
|
InputTokens interface{} `json:"input_token"`
|
||
|
|
OutputTokens interface{} `json:"output_token"`
|
||
|
|
LimitKey string `json:"limit_key"`
|
||
|
|
MoreResults string `json:"more_results"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// InputTokensString returns output tokens formatted as a list
|
||
|
|
func (p *Paginator) InputTokensString() string {
|
||
|
|
str := p.InputTokens.([]string)
|
||
|
|
return fmt.Sprintf("%#v", str)
|
||
|
|
}
|
||
|
|
|
||
|
|
// OutputTokensString returns output tokens formatted as a list
|
||
|
|
func (p *Paginator) OutputTokensString() string {
|
||
|
|
str := p.OutputTokens.([]string)
|
||
|
|
return fmt.Sprintf("%#v", str)
|
||
|
|
}
|
||
|
|
|
||
|
|
// used for unmarshaling from the paginators JSON file
|
||
|
|
type paginationDefinitions struct {
|
||
|
|
*API
|
||
|
|
Pagination map[string]Paginator
|
||
|
|
}
|
||
|
|
|
||
|
|
// AttachPaginators attaches pagination configuration from filename to the API.
|
||
|
|
func (a *API) AttachPaginators(filename string) {
|
||
|
|
p := paginationDefinitions{API: a}
|
||
|
|
|
||
|
|
f, err := os.Open(filename)
|
||
|
|
defer f.Close()
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
err = json.NewDecoder(f).Decode(&p)
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
p.setup()
|
||
|
|
}
|
||
|
|
|
||
|
|
// setup runs post-processing on the paginator configuration.
|
||
|
|
func (p *paginationDefinitions) setup() {
|
||
|
|
for n, e := range p.Pagination {
|
||
|
|
if e.InputTokens == nil || e.OutputTokens == nil {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
paginator := e
|
||
|
|
|
||
|
|
switch t := paginator.InputTokens.(type) {
|
||
|
|
case string:
|
||
|
|
paginator.InputTokens = []string{t}
|
||
|
|
case []interface{}:
|
||
|
|
toks := []string{}
|
||
|
|
for _, e := range t {
|
||
|
|
s := e.(string)
|
||
|
|
toks = append(toks, s)
|
||
|
|
}
|
||
|
|
paginator.InputTokens = toks
|
||
|
|
}
|
||
|
|
switch t := paginator.OutputTokens.(type) {
|
||
|
|
case string:
|
||
|
|
paginator.OutputTokens = []string{t}
|
||
|
|
case []interface{}:
|
||
|
|
toks := []string{}
|
||
|
|
for _, e := range t {
|
||
|
|
s := e.(string)
|
||
|
|
toks = append(toks, s)
|
||
|
|
}
|
||
|
|
paginator.OutputTokens = toks
|
||
|
|
}
|
||
|
|
|
||
|
|
if o, ok := p.Operations[n]; ok {
|
||
|
|
o.Paginator = &paginator
|
||
|
|
} else {
|
||
|
|
panic("unknown operation for paginator " + n)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|