feat: added kwargs and flags to pass to plugins

This commit is contained in:
David Allen 2025-09-01 18:59:12 -06:00
parent dc6a141ca1
commit 42c8fd7c1a
Signed by: towk
GPG key ID: 0430CDBE22619155
6 changed files with 95 additions and 7 deletions

View file

@ -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
}