44 lines
937 B
Go
44 lines
937 B
Go
package endpoint
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
pb "github.com/bestnite/go-igdb/proto"
|
|
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
type Collections struct {
|
|
BaseEndpoint[pb.Collection]
|
|
}
|
|
|
|
func NewCollections(request RequestFunc) *Collections {
|
|
a := &Collections{
|
|
BaseEndpoint[pb.Collection]{
|
|
endpointName: EPCollections,
|
|
request: request,
|
|
},
|
|
}
|
|
a.queryFunc = a.Query
|
|
return a
|
|
}
|
|
|
|
func (a *Collections) Query(ctx context.Context, query string) ([]*pb.Collection, error) {
|
|
resp, err := a.request(ctx, "POST", fmt.Sprintf("https://api.igdb.com/v4/%s.pb", a.endpointName), query)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to request: %w", err)
|
|
}
|
|
|
|
data := pb.CollectionResult{}
|
|
if err = proto.Unmarshal(resp.Body(), &data); err != nil {
|
|
return nil, fmt.Errorf("failed to unmarshal: %w", err)
|
|
}
|
|
|
|
if len(data.Collections) == 0 {
|
|
return nil, fmt.Errorf("no results: %s", query)
|
|
}
|
|
|
|
return data.Collections, nil
|
|
}
|