31 lines
894 B
TypeScript
31 lines
894 B
TypeScript
|
import { Storage } from '@revenge-mod/storage'
|
||
|
|
import { useReRender } from '@revenge-mod/utils/react'
|
||
|
|
import { useEffect } from 'react'
|
||
|
|
import { instanceToSubscriptions } from '../../../../lib/storage/src/_internal'
|
||
|
|
import type { StorageSubscription } from '@revenge-mod/storage'
|
||
|
|
|
||
|
|
// Actual implementation of Storage#use
|
||
|
|
const SP: Storage<any> = Storage.prototype
|
||
|
|
SP.use = function (filter) {
|
||
|
|
if (!this.cache) this.get()
|
||
|
|
|
||
|
|
const reRender = useReRender()
|
||
|
|
const subscriptions = instanceToSubscriptions.get(this)!
|
||
|
|
|
||
|
|
useEffect(() => {
|
||
|
|
const sub: StorageSubscription = filter
|
||
|
|
? v => {
|
||
|
|
if (filter(v)) reRender()
|
||
|
|
}
|
||
|
|
: reRender
|
||
|
|
|
||
|
|
subscriptions.add(sub)
|
||
|
|
|
||
|
|
return () => {
|
||
|
|
subscriptions.delete(sub)
|
||
|
|
}
|
||
|
|
}, [filter, reRender, subscriptions.add, subscriptions.delete])
|
||
|
|
|
||
|
|
return this.cache
|
||
|
|
}
|