76 lines
2.0 KiB
TypeScript
76 lines
2.0 KiB
TypeScript
|
import { KV, KvKey, KvQuery } from "$sb/types.ts";
|
||
|
|
import type { DataStore } from "../lib/datastore.ts";
|
||
|
|
import type { SyscallContext, SysCallMapping } from "../system.ts";
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Exposes the datastore API to plugs, but scoping everything to a prefix based on the plug's name
|
||
|
|
* @param ds the datastore to wrap
|
||
|
|
* @param prefix prefix to scope all keys to to which the plug name will be appended
|
||
|
|
*/
|
||
|
|
export function dataStoreSyscalls(
|
||
|
|
ds: DataStore,
|
||
|
|
prefix: KvKey = ["ds"],
|
||
|
|
): SysCallMapping {
|
||
|
|
return {
|
||
|
|
"datastore.delete": (ctx, key: KvKey) => {
|
||
|
|
return ds.delete(applyPrefix(ctx, key));
|
||
|
|
},
|
||
|
|
|
||
|
|
"datastore.set": (ctx, key: KvKey, value: any) => {
|
||
|
|
return ds.set(applyPrefix(ctx, key), value);
|
||
|
|
},
|
||
|
|
|
||
|
|
"datastore.batchSet": (ctx, kvs: KV[]) => {
|
||
|
|
return ds.batchSet(
|
||
|
|
kvs.map((kv) => ({ key: applyPrefix(ctx, kv.key), value: kv.value })),
|
||
|
|
);
|
||
|
|
},
|
||
|
|
|
||
|
|
"datastore.batchDelete": (ctx, keys: KvKey[]) => {
|
||
|
|
return ds.batchDelete(keys.map((k) => applyPrefix(ctx, k)));
|
||
|
|
},
|
||
|
|
|
||
|
|
"datastore.batchGet": (
|
||
|
|
ctx,
|
||
|
|
keys: KvKey[],
|
||
|
|
): Promise<(any | undefined)[]> => {
|
||
|
|
return ds.batchGet(keys.map((k) => applyPrefix(ctx, k)));
|
||
|
|
},
|
||
|
|
|
||
|
|
"datastore.get": (ctx, key: KvKey): Promise<any | null> => {
|
||
|
|
return ds.get(applyPrefix(ctx, key));
|
||
|
|
},
|
||
|
|
|
||
|
|
"datastore.query": async (
|
||
|
|
ctx,
|
||
|
|
query: KvQuery,
|
||
|
|
): Promise<KV[]> => {
|
||
|
|
return (await ds.query({
|
||
|
|
...query,
|
||
|
|
prefix: applyPrefix(ctx, query.prefix),
|
||
|
|
})).map((kv) => ({
|
||
|
|
key: stripPrefix(kv.key),
|
||
|
|
value: kv.value,
|
||
|
|
}));
|
||
|
|
},
|
||
|
|
|
||
|
|
"datastore.queryDelete": (
|
||
|
|
ctx,
|
||
|
|
query: KvQuery,
|
||
|
|
): Promise<void> => {
|
||
|
|
return ds.queryDelete({
|
||
|
|
...query,
|
||
|
|
prefix: applyPrefix(ctx, query.prefix),
|
||
|
|
});
|
||
|
|
},
|
||
|
|
};
|
||
|
|
|
||
|
|
function applyPrefix(ctx: SyscallContext, key?: KvKey): KvKey {
|
||
|
|
return [...prefix, ctx.plug.name!, ...(key ? key : [])];
|
||
|
|
}
|
||
|
|
|
||
|
|
function stripPrefix(key: KvKey): KvKey {
|
||
|
|
return key.slice(prefix.length + 1);
|
||
|
|
}
|
||
|
|
}
|