mirror of
https://github.com/HeyPuter/puter.git
synced 2025-12-30 09:40:00 -06:00
54b8990c70
The basic `.get()` and `.set()` calls should be the same between persistent key-value store (puter-kvstore) and in-memory cache (kv.js). This commit wraps these methods to support the "shorthand" usage.
33 lines
915 B
JavaScript
33 lines
915 B
JavaScript
//@extension priority -10000
|
|
|
|
const { DB_WRITE } = extension.import('core').database;
|
|
const svc_database = extension.import('service:database');
|
|
const svc_kvstore = extension.import('service:puter-kvstore');
|
|
|
|
// Methods on the object from `.as()` come from TraitsFeature.js,
|
|
// and they are already bound to their respective instance.
|
|
const simplified_kv = { ...svc_kvstore.as('puter-kvstore') };
|
|
|
|
const original_get = simplified_kv.get;
|
|
const original_set = simplified_kv.set;
|
|
|
|
simplified_kv.get = (...a) => {
|
|
if ( typeof a[0] === 'string' ) {
|
|
return original_get({ key: a[0] });
|
|
}
|
|
return original_get(...a);
|
|
};
|
|
|
|
simplified_kv.set = (...a) => {
|
|
if ( typeof a[0] === 'string' ) {
|
|
return original_set({ key: a[0], value: a[1] });
|
|
}
|
|
return original_set(...a);
|
|
};
|
|
|
|
extension.exports = {
|
|
db: svc_database.get(DB_WRITE, 'extensions'),
|
|
kv: simplified_kv,
|
|
cache: kv,
|
|
};
|