107 lines
2.8 KiB
Go
107 lines
2.8 KiB
Go
|
// Test KVStore plugin for Navidrome plugin system integration tests.
|
||
|
|
// Build with: tinygo build -o ../test-kvstore.wasm -target wasip1 -buildmode=c-shared .
|
||
|
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/navidrome/navidrome/plugins/pdk/go/host"
|
||
|
|
"github.com/navidrome/navidrome/plugins/pdk/go/pdk"
|
||
|
|
)
|
||
|
|
|
||
|
|
// TestKVStoreInput is the input for nd_test_kvstore callback.
|
||
|
|
type TestKVStoreInput struct {
|
||
|
|
Operation string `json:"operation"` // "set", "get", "delete", "has", "list", "get_storage_used"
|
||
|
|
Key string `json:"key"` // Storage key
|
||
|
|
Value []byte `json:"value"` // For set operations
|
||
|
|
Prefix string `json:"prefix"` // For list operation
|
||
|
|
}
|
||
|
|
|
||
|
|
// TestKVStoreOutput is the output from nd_test_kvstore callback.
|
||
|
|
type TestKVStoreOutput struct {
|
||
|
|
Value []byte `json:"value,omitempty"`
|
||
|
|
Exists bool `json:"exists,omitempty"`
|
||
|
|
Keys []string `json:"keys,omitempty"`
|
||
|
|
StorageUsed int64 `json:"storage_used,omitempty"`
|
||
|
|
Error *string `json:"error,omitempty"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// nd_test_kvstore is the test callback that tests the kvstore host functions.
|
||
|
|
//
|
||
|
|
//go:wasmexport nd_test_kvstore
|
||
|
|
func ndTestKVStore() int32 {
|
||
|
|
var input TestKVStoreInput
|
||
|
|
if err := pdk.InputJSON(&input); err != nil {
|
||
|
|
errStr := err.Error()
|
||
|
|
pdk.OutputJSON(TestKVStoreOutput{Error: &errStr})
|
||
|
|
return 0
|
||
|
|
}
|
||
|
|
|
||
|
|
switch input.Operation {
|
||
|
|
case "set":
|
||
|
|
err := host.KVStoreSet(input.Key, input.Value)
|
||
|
|
if err != nil {
|
||
|
|
errStr := err.Error()
|
||
|
|
pdk.OutputJSON(TestKVStoreOutput{Error: &errStr})
|
||
|
|
return 0
|
||
|
|
}
|
||
|
|
pdk.OutputJSON(TestKVStoreOutput{})
|
||
|
|
return 0
|
||
|
|
|
||
|
|
case "get":
|
||
|
|
value, exists, err := host.KVStoreGet(input.Key)
|
||
|
|
if err != nil {
|
||
|
|
errStr := err.Error()
|
||
|
|
pdk.OutputJSON(TestKVStoreOutput{Error: &errStr})
|
||
|
|
return 0
|
||
|
|
}
|
||
|
|
pdk.OutputJSON(TestKVStoreOutput{Value: value, Exists: exists})
|
||
|
|
return 0
|
||
|
|
|
||
|
|
case "delete":
|
||
|
|
err := host.KVStoreDelete(input.Key)
|
||
|
|
if err != nil {
|
||
|
|
errStr := err.Error()
|
||
|
|
pdk.OutputJSON(TestKVStoreOutput{Error: &errStr})
|
||
|
|
return 0
|
||
|
|
}
|
||
|
|
pdk.OutputJSON(TestKVStoreOutput{})
|
||
|
|
return 0
|
||
|
|
|
||
|
|
case "has":
|
||
|
|
exists, err := host.KVStoreHas(input.Key)
|
||
|
|
if err != nil {
|
||
|
|
errStr := err.Error()
|
||
|
|
pdk.OutputJSON(TestKVStoreOutput{Error: &errStr})
|
||
|
|
return 0
|
||
|
|
}
|
||
|
|
pdk.OutputJSON(TestKVStoreOutput{Exists: exists})
|
||
|
|
return 0
|
||
|
|
|
||
|
|
case "list":
|
||
|
|
keys, err := host.KVStoreList(input.Prefix)
|
||
|
|
if err != nil {
|
||
|
|
errStr := err.Error()
|
||
|
|
pdk.OutputJSON(TestKVStoreOutput{Error: &errStr})
|
||
|
|
return 0
|
||
|
|
}
|
||
|
|
pdk.OutputJSON(TestKVStoreOutput{Keys: keys})
|
||
|
|
return 0
|
||
|
|
|
||
|
|
case "get_storage_used":
|
||
|
|
bytesUsed, err := host.KVStoreGetStorageUsed()
|
||
|
|
if err != nil {
|
||
|
|
errStr := err.Error()
|
||
|
|
pdk.OutputJSON(TestKVStoreOutput{Error: &errStr})
|
||
|
|
return 0
|
||
|
|
}
|
||
|
|
pdk.OutputJSON(TestKVStoreOutput{StorageUsed: bytesUsed})
|
||
|
|
return 0
|
||
|
|
|
||
|
|
default:
|
||
|
|
errStr := "unknown operation: " + input.Operation
|
||
|
|
pdk.OutputJSON(TestKVStoreOutput{Error: &errStr})
|
||
|
|
return 0
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func main() {}
|