mirror of
https://github.com/hanzoai/agency.git
synced 2026-11-02 19:13:28 +00:00
124 lines
3.5 KiB
Plaintext
124 lines
3.5 KiB
Plaintext
// src/mutationCache.ts
|
|
import { notifyManager } from "./notifyManager.js";
|
|
import { Mutation } from "./mutation.js";
|
|
import { matchMutation, noop } from "./utils.js";
|
|
import { Subscribable } from "./subscribable.js";
|
|
var MutationCache = class extends Subscribable {
|
|
constructor(config = {}) {
|
|
super();
|
|
this.config = config;
|
|
this.#mutations = /* @__PURE__ */ new Set();
|
|
this.#scopes = /* @__PURE__ */ new Map();
|
|
this.#mutationId = 0;
|
|
}
|
|
#mutations;
|
|
#scopes;
|
|
#mutationId;
|
|
build(client, options, state) {
|
|
const mutation = new Mutation({
|
|
client,
|
|
mutationCache: this,
|
|
mutationId: ++this.#mutationId,
|
|
options: client.defaultMutationOptions(options),
|
|
state
|
|
});
|
|
this.add(mutation);
|
|
return mutation;
|
|
}
|
|
add(mutation) {
|
|
this.#mutations.add(mutation);
|
|
const scope = scopeFor(mutation);
|
|
if (typeof scope === "string") {
|
|
const scopedMutations = this.#scopes.get(scope);
|
|
if (scopedMutations) {
|
|
scopedMutations.push(mutation);
|
|
} else {
|
|
this.#scopes.set(scope, [mutation]);
|
|
}
|
|
}
|
|
this.notify({ type: "added", mutation });
|
|
}
|
|
remove(mutation) {
|
|
if (this.#mutations.delete(mutation)) {
|
|
const scope = scopeFor(mutation);
|
|
if (typeof scope === "string") {
|
|
const scopedMutations = this.#scopes.get(scope);
|
|
if (scopedMutations) {
|
|
if (scopedMutations.length > 1) {
|
|
const index = scopedMutations.indexOf(mutation);
|
|
if (index !== -1) {
|
|
scopedMutations.splice(index, 1);
|
|
}
|
|
} else if (scopedMutations[0] === mutation) {
|
|
this.#scopes.delete(scope);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
this.notify({ type: "removed", mutation });
|
|
}
|
|
canRun(mutation) {
|
|
const scope = scopeFor(mutation);
|
|
if (typeof scope === "string") {
|
|
const mutationsWithSameScope = this.#scopes.get(scope);
|
|
const firstPendingMutation = mutationsWithSameScope?.find(
|
|
(m) => m.state.status === "pending"
|
|
);
|
|
return !firstPendingMutation || firstPendingMutation === mutation;
|
|
} else {
|
|
return true;
|
|
}
|
|
}
|
|
runNext(mutation) {
|
|
const scope = scopeFor(mutation);
|
|
if (typeof scope === "string") {
|
|
const foundMutation = this.#scopes.get(scope)?.find((m) => m !== mutation && m.state.isPaused);
|
|
return foundMutation?.continue() ?? Promise.resolve();
|
|
} else {
|
|
return Promise.resolve();
|
|
}
|
|
}
|
|
clear() {
|
|
notifyManager.batch(() => {
|
|
this.#mutations.forEach((mutation) => {
|
|
this.notify({ type: "removed", mutation });
|
|
});
|
|
this.#mutations.clear();
|
|
this.#scopes.clear();
|
|
});
|
|
}
|
|
getAll() {
|
|
return Array.from(this.#mutations);
|
|
}
|
|
find(filters) {
|
|
const defaultedFilters = { exact: true, ...filters };
|
|
return this.getAll().find(
|
|
(mutation) => matchMutation(defaultedFilters, mutation)
|
|
);
|
|
}
|
|
findAll(filters = {}) {
|
|
return this.getAll().filter((mutation) => matchMutation(filters, mutation));
|
|
}
|
|
notify(event) {
|
|
notifyManager.batch(() => {
|
|
this.listeners.forEach((listener) => {
|
|
listener(event);
|
|
});
|
|
});
|
|
}
|
|
resumePausedMutations() {
|
|
const pausedMutations = this.getAll().filter((x) => x.state.isPaused);
|
|
return notifyManager.batch(
|
|
() => Promise.all(
|
|
pausedMutations.map((mutation) => mutation.continue().catch(noop))
|
|
)
|
|
);
|
|
}
|
|
};
|
|
function scopeFor(mutation) {
|
|
return mutation.options.scope?.id;
|
|
}
|
|
export {
|
|
MutationCache
|
|
};
|
|
//# sourceMappingURL=mutationCache.js.map |