23 lines
635 B
TypeScript
23 lines
635 B
TypeScript
import { Storage } from '@revenge-mod/storage'
|
|
import { useReRender } from '@revenge-mod/utils/react'
|
|
import { useEffect } from 'react'
|
|
import type { StorageSubscription } from '@revenge-mod/storage'
|
|
|
|
// Actual implementation of Storage#use
|
|
;(Storage.prototype as Storage<any>).use = function (filter) {
|
|
if (!this.cache) this.get()
|
|
|
|
const reRender = useReRender()
|
|
|
|
useEffect(() => {
|
|
const sub: StorageSubscription = filter
|
|
? v => {
|
|
if (filter(v)) reRender()
|
|
}
|
|
: reRender
|
|
|
|
return this.subscribe(sub)
|
|
}, [filter, reRender])
|
|
|
|
return this.cache
|
|
}
|