40 lines
960 B
Go
40 lines
960 B
Go
package endpoint
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
pb "github.com/bestnite/go-igdb/proto"
|
|
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
type CollectionMemberships struct {
|
|
BaseEndpoint[pb.CollectionMembership]
|
|
}
|
|
|
|
func NewCollectionMemberships(request RequestFunc) *CollectionMemberships {
|
|
a := &CollectionMemberships{
|
|
BaseEndpoint[pb.CollectionMembership]{
|
|
endpointName: EPCollectionMemberships,
|
|
request: request,
|
|
},
|
|
}
|
|
a.queryFunc = a.Query
|
|
return a
|
|
}
|
|
|
|
func (a *CollectionMemberships) Query(ctx context.Context, query string) ([]*pb.CollectionMembership, 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.CollectionMembershipResult{}
|
|
if err = proto.Unmarshal(resp.Body(), &data); err != nil {
|
|
return nil, fmt.Errorf("failed to unmarshal: %w", err)
|
|
}
|
|
|
|
return data.Collectionmemberships, nil
|
|
}
|