feat: added kwargs and flags to pass to plugins
This commit is contained in:
parent
dc6a141ca1
commit
42c8fd7c1a
6 changed files with 95 additions and 7 deletions
|
|
@ -1,6 +1,10 @@
|
|||
package storage
|
||||
|
||||
import "fmt"
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"git.towk2.me/towk/makeshift/internal/kwargs"
|
||||
)
|
||||
|
||||
type MemoryStorage struct {
|
||||
Data map[string]any `json:"data"`
|
||||
|
|
@ -15,6 +19,15 @@ func (ms *MemoryStorage) Cleanup() error {
|
|||
return nil
|
||||
}
|
||||
|
||||
func (ms *MemoryStorage) SetKWArgs(kw *kwargs.KWArgs) error {
|
||||
return ms.Set(kwargs.RESERVED_KEY, kw)
|
||||
}
|
||||
|
||||
func (ms *MemoryStorage) GetKWArgs() (*kwargs.KWArgs, error) {
|
||||
kw, err := ms.Get(kwargs.RESERVED_KEY)
|
||||
return kw.(*kwargs.KWArgs), err
|
||||
}
|
||||
|
||||
func (ms *MemoryStorage) Get(k string) (any, error) {
|
||||
v, ok := ms.Data[k]
|
||||
if ok {
|
||||
|
|
@ -24,6 +37,9 @@ func (ms *MemoryStorage) Get(k string) (any, error) {
|
|||
}
|
||||
|
||||
func (ms *MemoryStorage) Set(k string, v any) error {
|
||||
if k == "kwargs" {
|
||||
return fmt.Errorf("cannot set reserved key '%s' (use SetKWArgs() instead)", k)
|
||||
}
|
||||
ms.Data[k] = v
|
||||
return nil
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue