85 lines
1.5 KiB
Go
85 lines
1.5 KiB
Go
package arn
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/aerogo/aero"
|
|
"github.com/aerogo/api"
|
|
)
|
|
|
|
// Lockable ...
|
|
type Lockable interface {
|
|
Lock(userID UserID)
|
|
Unlock(userID UserID)
|
|
IsLocked() bool
|
|
Save()
|
|
}
|
|
|
|
// LockEventReceiver ...
|
|
type LockEventReceiver interface {
|
|
OnLock(user *User)
|
|
OnUnlock(user *User)
|
|
}
|
|
|
|
// LockAction ...
|
|
func LockAction() *api.Action {
|
|
return &api.Action{
|
|
Name: "lock",
|
|
Route: "/lock",
|
|
Run: func(obj interface{}, ctx aero.Context) error {
|
|
lockable := obj.(Lockable)
|
|
user := GetUserFromContext(ctx)
|
|
|
|
if user == nil {
|
|
return errors.New("Not logged in")
|
|
}
|
|
|
|
lockable.Lock(user.ID)
|
|
|
|
// Call OnLock if the object implements it
|
|
receiver, ok := lockable.(LockEventReceiver)
|
|
|
|
if ok {
|
|
receiver.OnLock(user)
|
|
}
|
|
|
|
lockable.Save()
|
|
return nil
|
|
},
|
|
}
|
|
}
|
|
|
|
// UnlockAction ...
|
|
func UnlockAction() *api.Action {
|
|
return &api.Action{
|
|
Name: "unlock",
|
|
Route: "/unlock",
|
|
Run: func(obj interface{}, ctx aero.Context) error {
|
|
lockable := obj.(Lockable)
|
|
user := GetUserFromContext(ctx)
|
|
|
|
if user == nil {
|
|
return errors.New("Not logged in")
|
|
}
|
|
|
|
lockable.Unlock(user.ID)
|
|
|
|
// Call OnUnlock if the object implements it
|
|
receiver, ok := lockable.(LockEventReceiver)
|
|
|
|
if ok {
|
|
receiver.OnUnlock(user)
|
|
}
|
|
|
|
lockable.Save()
|
|
return nil
|
|
},
|
|
}
|
|
}
|
|
|
|
// IsLocked returns true if the given object is locked.
|
|
func IsLocked(obj interface{}) bool {
|
|
lockable, isLockable := obj.(Lockable)
|
|
return isLockable && lockable.IsLocked()
|
|
}
|