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
|
package solver
import (
"github.com/kong/deck/crud"
"github.com/kong/deck/diff"
"github.com/kong/deck/state"
"github.com/kong/deck/utils"
"github.com/kong/go-kong/kong"
)
// keyAuthCRUD implements crud.Actions interface.
type keyAuthCRUD struct {
client *kong.Client
}
func keyAuthFromStuct(arg diff.Event) *state.KeyAuth {
keyAuth, ok := arg.Obj.(*state.KeyAuth)
if !ok {
panic("unexpected type, expected *state.Route")
}
return keyAuth
}
// Create creates a Route in Kong.
// The arg should be of type diff.Event, containing the keyAuth to be created,
// else the function will panic.
// It returns a the created *state.Route.
func (s *keyAuthCRUD) Create(arg ...crud.Arg) (crud.Arg, error) {
event := eventFromArg(arg[0])
keyAuth := keyAuthFromStuct(event)
createdKeyAuth, err := s.client.KeyAuths.Create(nil, keyAuth.Consumer.ID,
&keyAuth.KeyAuth)
if err != nil {
return nil, err
}
return &state.KeyAuth{KeyAuth: *createdKeyAuth}, nil
}
// Delete deletes a Route in Kong.
// The arg should be of type diff.Event, containing the keyAuth to be deleted,
// else the function will panic.
// It returns a the deleted *state.Route.
func (s *keyAuthCRUD) Delete(arg ...crud.Arg) (crud.Arg, error) {
event := eventFromArg(arg[0])
keyAuth := keyAuthFromStuct(event)
cid := ""
if !utils.Empty(keyAuth.Consumer.Username) {
cid = *keyAuth.Consumer.Username
}
if !utils.Empty(keyAuth.Consumer.ID) {
cid = *keyAuth.Consumer.ID
}
err := s.client.KeyAuths.Delete(nil, &cid, keyAuth.ID)
if err != nil {
return nil, err
}
return keyAuth, nil
}
// Update updates a Route in Kong.
// The arg should be of type diff.Event, containing the keyAuth to be updated,
// else the function will panic.
// It returns a the updated *state.Route.
func (s *keyAuthCRUD) Update(arg ...crud.Arg) (crud.Arg, error) {
event := eventFromArg(arg[0])
keyAuth := keyAuthFromStuct(event)
updatedKeyAuth, err := s.client.KeyAuths.Create(nil, keyAuth.Consumer.ID,
&keyAuth.KeyAuth)
if err != nil {
return nil, err
}
return &state.KeyAuth{KeyAuth: *updatedKeyAuth}, nil
}
|