diff --git a/packages/util/Repository.ts b/packages/util/Repository.ts index ad25a07..5647734 100644 --- a/packages/util/Repository.ts +++ b/packages/util/Repository.ts @@ -12,6 +12,7 @@ const getDay = (ts: number) => Math.floor(ts / DAY) export class Repository extends Emitter { eventsById = new Map() + eventsByWrap = new Map() eventsByAddress = new Map() eventsByTag = new Map() eventsByDay = new Map() @@ -40,6 +41,7 @@ export class Repository extends Emitter { clear = () => { this.eventsById.clear() + this.eventsByWrap.clear() this.eventsByAddress.clear() this.eventsByTag.clear() this.eventsByDay.clear() @@ -158,6 +160,10 @@ export class Repository extends Emitter { // Now add our new event this.eventsById.set(event.id, event) + if (event.wrap) { + this.eventsByWrap.set(event.wrap.id, event) + } + if (isReplaceable(event)) { this.eventsByAddress.set(address, event) }