55 lines
1021 B
Go
55 lines
1021 B
Go
package arn
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/aerogo/aero"
|
|
"github.com/aerogo/api"
|
|
)
|
|
|
|
// IDCollection ...
|
|
type IDCollection interface {
|
|
Add(id ID) error
|
|
Remove(id ID) bool
|
|
Save()
|
|
}
|
|
|
|
// AddAction returns an API action that adds a new item to the IDCollection.
|
|
func AddAction() *api.Action {
|
|
return &api.Action{
|
|
Name: "add",
|
|
Route: "/add/:item-id",
|
|
Run: func(obj interface{}, ctx aero.Context) error {
|
|
list := obj.(IDCollection)
|
|
itemID := ctx.Get("item-id")
|
|
err := list.Add(itemID)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
list.Save()
|
|
return nil
|
|
},
|
|
}
|
|
}
|
|
|
|
// RemoveAction returns an API action that removes an item from the IDCollection.
|
|
func RemoveAction() *api.Action {
|
|
return &api.Action{
|
|
Name: "remove",
|
|
Route: "/remove/:item-id",
|
|
Run: func(obj interface{}, ctx aero.Context) error {
|
|
list := obj.(IDCollection)
|
|
itemID := ctx.Get("item-id")
|
|
|
|
if !list.Remove(itemID) {
|
|
return errors.New("This item does not exist in the list")
|
|
}
|
|
|
|
list.Save()
|
|
return nil
|
|
},
|
|
}
|
|
}
|