snapshot of https://github.com/toeverything/AFFiNE.git (canary, 2026-09-16; blobs > 131072B omitted)

blocksuite/framework/store/src/adapter/context.ts

@@ -0,0 +1,119 @@
+type Keyof<T> = T extends unknown ? keyof T : never;
+
+export class ASTWalkerContext<TNode extends object> {
+  private _defaultProp: Keyof<TNode> = 'children' as unknown as Keyof<TNode>;
+
+  private _globalContext: Record<string, unknown> = Object.create(null);
+
+  private readonly _stack: {
+    node: TNode;
+    prop: Keyof<TNode>;
+    context: Record<string, unknown>;
+  }[] = [];
+
+  _skip = false;
+
+  _skipChildrenNum = 0;
+
+  setDefaultProp = (parentProp: Keyof<TNode>) => {
+    this._defaultProp = parentProp;
+  };
+
+  get stack() {
+    return this._stack;
+  }
+
+  private current() {
+    return this._stack[this._stack.length - 1];
+  }
+
+  cleanGlobalContextStack(key: string) {
+    if (this._globalContext[key] instanceof Array) {
+      this._globalContext[key] = [];
+    }
+  }
+
+  closeNode() {
+    const ele = this._stack.pop();
+    if (!ele) return this;
+    const parent = this._stack.pop();
+    if (!parent) {
+      this._stack.push(ele);
+      return this;
+    }
+    if (parent.node[ele.prop] instanceof Array) {
+      (parent.node[ele.prop] as Array<object>).push(ele.node);
+    }
+    this._stack.push(parent);
+    return this;
+  }
+
+  currentNode() {
+    return this.current()?.node;
+  }
+
+  getGlobalContext(key: string) {
+    return this._globalContext[key];
+  }
+
+  getGlobalContextStack<StackElement>(key: string) {
+    const stack = this._globalContext[key];
+    if (stack instanceof Array) {
+      return stack as StackElement[];
+    } else {
+      return [] as StackElement[];
+    }
+  }
+
+  getNodeContext(key: string) {
+    return this.current().context[key];
+  }
+
+  getPreviousNodeContext(key: string) {
+    return this._stack[this._stack.length - 2]?.context[key];
+  }
+
+  openNode(node: TNode, parentProp?: Keyof<TNode>) {
+    this._stack.push({
+      node,
+      prop: parentProp ?? this._defaultProp,
+      context: Object.create(null),
+    });
+    return this;
+  }
+
+  previousNode() {
+    return this._stack[this._stack.length - 2]?.node;
+  }
+
+  pushGlobalContextStack<StackElement>(key: string, value: StackElement) {
+    const stack = this._globalContext[key];
+    if (stack instanceof Array) {
+      stack.push(value);
+    } else {
+      this._globalContext[key] = [value];
+    }
+  }
+
+  setGlobalContext(key: string, value: unknown) {
+    this._globalContext[key] = value;
+    return this;
+  }
+
+  setGlobalContextStack<StackElement>(key: string, value: StackElement[]) {
+    this._globalContext[key] = value;
+  }
+
+  setNodeContext(key: string, value: unknown) {
+    this._stack[this._stack.length - 1].context[key] = value;
+    return this;
+  }
+
+  skipAllChildren() {
+    this._skip = true;
+  }
+
+  skipChildren(num = 1) {
+    this._skipChildrenNum = num;
+  }
+}