mermaid-ast 0.4.0 → 0.6.0

package/dist/class-diagram.d.ts

@@ -0,0 +1,129 @@
+/**
+ * Class Diagram Wrapper Class
+ *
+ * A unified API for building, mutating, and querying class diagrams.
+ * Provides a fluent interface that wraps the ClassDiagramAST.
+ */
+import { DiagramWrapper } from './diagram-wrapper.js';
+import type { ClassDefinition, ClassDiagramAST, ClassDirection, ClassMember, ClassRelation, LineType, RelationType } from './types/class.js';
+import type { RenderOptions } from './types/render-options.js';
+/** Options for adding a class */
+export interface AddClassOptions {
+    label?: string;
+    annotation?: string;
+}
+/** Options for adding a member */
+export interface AddMemberOptions {
+    visibility?: '+' | '-' | '#' | '~';
+    type?: 'method' | 'attribute';
+}
+/** Options for adding a relation */
+export interface AddRelationOptions {
+    label?: string;
+    labelFrom?: string;
+    labelTo?: string;
+    lineType?: LineType;
+}
+/**
+ * A fluent wrapper for ClassDiagramAST that supports building, mutating, and querying.
+ */
+export declare class ClassDiagram extends DiagramWrapper<ClassDiagramAST> {
+    private constructor();
+    /** Create a new empty class diagram */
+    static create(direction?: ClassDirection): ClassDiagram;
+    /** Create from an existing AST */
+    static from(ast: ClassDiagramAST): ClassDiagram;
+    /** Parse Mermaid syntax into a ClassDiagram */
+    static parse(input: string): ClassDiagram;
+    render(options?: RenderOptions): string;
+    clone(): ClassDiagram;
+    get direction(): ClassDirection;
+    get classCount(): number;
+    get relationCount(): number;
+    get classes(): Map<string, {
+        id: string;
+        label?: string;
+        memberCount: number;
+    }>;
+    setDirection(direction: ClassDirection): this;
+    /** Add a class */
+    addClass(id: string, options?: AddClassOptions): this;
+    /** Remove a class and optionally its relations */
+    removeClass(id: string, options?: {
+        removeRelations?: boolean;
+    }): this;
+    /** Rename a class */
+    renameClass(id: string, newId: string): this;
+    /** Get a class by ID */
+    getClass(id: string): ClassDefinition | undefined;
+    /** Check if class exists */
+    hasClass(id: string): boolean;
+    /** Set class label */
+    setClassLabel(id: string, label: string): this;
+    /** Add annotation to class */
+    addAnnotation(id: string, annotation: string): this;
+    /** Remove annotation from class */
+    removeAnnotation(id: string, annotation: string): this;
+    /** Add a member (attribute or method) to a class */
+    addMember(classId: string, text: string, options?: AddMemberOptions): this;
+    /** Add an attribute to a class */
+    addAttribute(classId: string, text: string, visibility?: '+' | '-' | '#' | '~'): this;
+    /** Add a method to a class */
+    addMethod(classId: string, text: string, visibility?: '+' | '-' | '#' | '~'): this;
+    /** Get members of a class */
+    getMembers(classId: string): ClassMember[];
+    /** Remove a member from a class */
+    removeMember(classId: string, text: string): this;
+    /** Add a relation between classes */
+    addRelation(from: string, to: string, relationType: RelationType, options?: AddRelationOptions): this;
+    /** Add inheritance relation (A extends B) */
+    addInheritance(child: string, parent: string, options?: AddRelationOptions): this;
+    /** Add composition relation */
+    addComposition(whole: string, part: string, options?: AddRelationOptions): this;
+    /** Add aggregation relation */
+    addAggregation(whole: string, part: string, options?: AddRelationOptions): this;
+    /** Add dependency relation */
+    addDependency(from: string, to: string, options?: AddRelationOptions): this;
+    /** Add association (simple line) */
+    addAssociation(from: string, to: string, options?: AddRelationOptions): this;
+    /** Get all relations */
+    getRelations(): ClassRelation[];
+    /** Get relations for a class */
+    getRelationsFor(classId: string): ClassRelation[];
+    /** Remove a relation */
+    removeRelation(from: string, to: string): this;
+    /** Add a namespace */
+    addNamespace(name: string, classIds?: string[]): this;
+    /** Add class to namespace */
+    addToNamespace(namespace: string, classId: string): this;
+    /** Remove class from namespace */
+    removeFromNamespace(namespace: string, classId: string): this;
+    /** Get namespace for a class */
+    getNamespaceFor(classId: string): string | undefined;
+    /** Add a note */
+    addNote(text: string, forClass?: string): this;
+    /** Get all notes */
+    getNotes(): Array<{
+        text: string;
+        forClass?: string;
+    }>;
+    /** Define a class style */
+    defineStyle(name: string, styles: string[]): this;
+    /** Apply style to a class */
+    applyStyle(classId: string, styleName: string): this;
+    /** Find classes by criteria */
+    findClasses(query: {
+        hasAnnotation?: string;
+        hasStyle?: string;
+        inNamespace?: string;
+    }): ClassDefinition[];
+    /** Get all subclasses of a class (direct inheritance) */
+    getSubclasses(classId: string): string[];
+    /** Get parent class (direct inheritance) */
+    getParentClass(classId: string): string | undefined;
+    /** Get all ancestors (transitive inheritance) */
+    getAncestors(classId: string): string[];
+    /** Get all descendants (transitive inheritance) */
+    getDescendants(classId: string): string[];
+}
+//# sourceMappingURL=class-diagram.d.ts.map

package/dist/class-diagram.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"class-diagram.d.ts","sourceRoot":"","sources":["../src/class-diagram.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAGtD,OAAO,KAAK,EACV,eAAe,EACf,eAAe,EACf,cAAc,EACd,WAAW,EACX,aAAa,EACb,QAAQ,EACR,YAAY,EACb,MAAM,kBAAkB,CAAC;AAE1B,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAE/D,iCAAiC;AACjC,MAAM,WAAW,eAAe;IAC9B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,kCAAkC;AAClC,MAAM,WAAW,gBAAgB;IAC/B,UAAU,CAAC,EAAE,GAAG,GAAG,GAAG,GAAG,GAAG,GAAG,GAAG,CAAC;IACnC,IAAI,CAAC,EAAE,QAAQ,GAAG,WAAW,CAAC;CAC/B;AAED,oCAAoC;AACpC,MAAM,WAAW,kBAAkB;IACjC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE,QAAQ,CAAC;CACrB;AAED;;GAEG;AACH,qBAAa,YAAa,SAAQ,cAAc,CAAC,eAAe,CAAC;IAC/D,OAAO;IAMP,uCAAuC;IACvC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,cAAc,GAAG,YAAY;IAMvD,kCAAkC;IAClC,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,eAAe,GAAG,YAAY;IAI/C,+CAA+C;IAC/C,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,GAAG,YAAY;IAMzC,MAAM,CAAC,OAAO,CAAC,EAAE,aAAa,GAAG,MAAM;IAIvC,KAAK,IAAI,YAAY;IAMrB,IAAI,SAAS,IAAI,cAAc,CAE9B;IAED,IAAI,UAAU,IAAI,MAAM,CAEvB;IAED,IAAI,aAAa,IAAI,MAAM,CAE1B;IAED,IAAI,OAAO,IAAI,GAAG,CAAC,MAAM,EAAE;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAE,CAAC,CAO9E;IAID,YAAY,CAAC,SAAS,EAAE,cAAc,GAAG,IAAI;IAO7C,kBAAkB;IAClB,QAAQ,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,eAAe,GAAG,IAAI;IAarD,kDAAkD;IAClD,WAAW,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE;QAAE,eAAe,CAAC,EAAE,OAAO,CAAA;KAAE,GAAG,IAAI;IAQtE,qBAAqB;IACrB,WAAW,CAAC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAoB5C,wBAAwB;IACxB,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,eAAe,GAAG,SAAS;IAIjD,4BAA4B;IAC5B,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO;IAI7B,sBAAsB;IACtB,aAAa,CAAC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAM9C,8BAA8B;IAC9B,aAAa,CAAC,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,IAAI;IAQnD,mCAAmC;IACnC,gBAAgB,CAAC,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,IAAI;IAUtD,oDAAoD;IACpD,SAAS,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,gBAAgB,GAAG,IAAI;IAe1E,kCAAkC;IAClC,YAAY,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,GAAG,GAAG,GAAG,GAAG,GAAG,GAAG,GAAG,GAAG,IAAI;IAIrF,8BAA8B;IAC9B,SAAS,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,GAAG,GAAG,GAAG,GAAG,GAAG,GAAG,GAAG,GAAG,IAAI;IAIlF,6BAA6B;IAC7B,UAAU,CAAC,OAAO,EAAE,MAAM,GAAG,WAAW,EAAE;IAI1C,mCAAmC;IACnC,YAAY,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI;IAUjD,qCAAqC;IACrC,WAAW,CACT,IAAI,EAAE,MAAM,EACZ,EAAE,EAAE,MAAM,EACV,YAAY,EAAE,YAAY,EAC1B,OAAO,CAAC,EAAE,kBAAkB,GAC3B,IAAI;IAqBP,6CAA6C;IAC7C,cAAc,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,kBAAkB,GAAG,IAAI;IAIjF,+BAA+B;IAC/B,cAAc,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,kBAAkB,GAAG,IAAI;IAI/E,+BAA+B;IAC/B,cAAc,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,kBAAkB,GAAG,IAAI;IAI/E,8BAA8B;IAC9B,aAAa,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,kBAAkB,GAAG,IAAI;IAI3E,oCAAoC;IACpC,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,kBAAkB,GAAG,IAAI;IAI5E,wBAAwB;IACxB,YAAY,IAAI,aAAa,EAAE;IAI/B,gCAAgC;IAChC,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,aAAa,EAAE;IAIjD,wBAAwB;IACxB,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,GAAG,IAAI;IAS9C,sBAAsB;IACtB,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,IAAI;IAQrD,6BAA6B;IAC7B,cAAc,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI;IAYxD,kCAAkC;IAClC,mBAAmB,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI;IAQ7D,gCAAgC;IAChC,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;IASpD,iBAAiB;IACjB,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI;IAK9C,oBAAoB;IACpB,QAAQ,IAAI,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IAMtD,2BAA2B;IAC3B,WAAW,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI;IAKjD,6BAA6B;IAC7B,UAAU,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,IAAI;IAUpD,+BAA+B;IAC/B,WAAW,CAAC,KAAK,EAAE;QACjB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,GAAG,eAAe,EAAE;IAcrB,yDAAyD;IACzD,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE;IAMxC,4CAA4C;IAC5C,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;IAOnD,iDAAiD;IACjD,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE;IAUvC,mDAAmD;IACnD,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE;CAU1C"}

package/dist/class-diagram.js

@@ -0,0 +1,341 @@
+/**
+ * Class Diagram Wrapper Class
+ *
+ * A unified API for building, mutating, and querying class diagrams.
+ * Provides a fluent interface that wraps the ClassDiagramAST.
+ */
+import { DiagramWrapper } from './diagram-wrapper.js';
+import { parseClassDiagram } from './parser/class-parser.js';
+import { renderClassDiagram } from './renderer/class-renderer.js';
+import { createClassDiagramAST } from './types/class.js';
+/**
+ * A fluent wrapper for ClassDiagramAST that supports building, mutating, and querying.
+ */
+export class ClassDiagram extends DiagramWrapper {
+    constructor(ast) {
+        super(ast);
+    }
+    // ============ Factory Methods ============
+    /** Create a new empty class diagram */
+    static create(direction) {
+        const ast = createClassDiagramAST();
+        if (direction)
+            ast.direction = direction;
+        return new ClassDiagram(ast);
+    }
+    /** Create from an existing AST */
+    static from(ast) {
+        return new ClassDiagram(structuredClone(ast));
+    }
+    /** Parse Mermaid syntax into a ClassDiagram */
+    static parse(input) {
+        return new ClassDiagram(parseClassDiagram(input));
+    }
+    // ============ Core Methods ============
+    render(options) {
+        return renderClassDiagram(this.ast, options);
+    }
+    clone() {
+        return new ClassDiagram(structuredClone(this.ast));
+    }
+    // ============ Properties ============
+    get direction() {
+        return this.ast.direction;
+    }
+    get classCount() {
+        return this.ast.classes.size;
+    }
+    get relationCount() {
+        return this.ast.relations.length;
+    }
+    get classes() {
+        return new Map(Array.from(this.ast.classes.entries()).map(([id, cls]) => [
+            id,
+            { id: cls.id, label: cls.label, memberCount: cls.members.length },
+        ]));
+    }
+    // ============ Direction Operations ============
+    setDirection(direction) {
+        this.ast.direction = direction;
+        return this;
+    }
+    // ============ Class Operations ============
+    /** Add a class */
+    addClass(id, options) {
+        const cls = {
+            id,
+            label: options?.label,
+            members: [],
+            annotations: options?.annotation ? [options.annotation] : [],
+            cssClasses: [],
+            styles: [],
+        };
+        this.ast.classes.set(id, cls);
+        return this;
+    }
+    /** Remove a class and optionally its relations */
+    removeClass(id, options) {
+        this.ast.classes.delete(id);
+        if (options?.removeRelations) {
+            this.ast.relations = this.ast.relations.filter((r) => r.id1 !== id && r.id2 !== id);
+        }
+        return this;
+    }
+    /** Rename a class */
+    renameClass(id, newId) {
+        const cls = this.ast.classes.get(id);
+        if (cls) {
+            cls.id = newId;
+            this.ast.classes.delete(id);
+            this.ast.classes.set(newId, cls);
+            // Update relations
+            for (const rel of this.ast.relations) {
+                if (rel.id1 === id)
+                    rel.id1 = newId;
+                if (rel.id2 === id)
+                    rel.id2 = newId;
+            }
+            // Update namespaces
+            for (const ns of this.ast.namespaces.values()) {
+                const idx = ns.classes.indexOf(id);
+                if (idx >= 0)
+                    ns.classes[idx] = newId;
+            }
+        }
+        return this;
+    }
+    /** Get a class by ID */
+    getClass(id) {
+        return this.ast.classes.get(id);
+    }
+    /** Check if class exists */
+    hasClass(id) {
+        return this.ast.classes.has(id);
+    }
+    /** Set class label */
+    setClassLabel(id, label) {
+        const cls = this.ast.classes.get(id);
+        if (cls)
+            cls.label = label;
+        return this;
+    }
+    /** Add annotation to class */
+    addAnnotation(id, annotation) {
+        const cls = this.ast.classes.get(id);
+        if (cls && !cls.annotations.includes(annotation)) {
+            cls.annotations.push(annotation);
+        }
+        return this;
+    }
+    /** Remove annotation from class */
+    removeAnnotation(id, annotation) {
+        const cls = this.ast.classes.get(id);
+        if (cls) {
+            cls.annotations = cls.annotations.filter((a) => a !== annotation);
+        }
+        return this;
+    }
+    // ============ Member Operations ============
+    /** Add a member (attribute or method) to a class */
+    addMember(classId, text, options) {
+        const cls = this.ast.classes.get(classId);
+        if (!cls) {
+            this.addClass(classId);
+            return this.addMember(classId, text, options);
+        }
+        const member = {
+            text,
+            visibility: options?.visibility,
+            type: options?.type,
+        };
+        cls.members.push(member);
+        return this;
+    }
+    /** Add an attribute to a class */
+    addAttribute(classId, text, visibility) {
+        return this.addMember(classId, text, { visibility, type: 'attribute' });
+    }
+    /** Add a method to a class */
+    addMethod(classId, text, visibility) {
+        return this.addMember(classId, text, { visibility, type: 'method' });
+    }
+    /** Get members of a class */
+    getMembers(classId) {
+        return this.ast.classes.get(classId)?.members ?? [];
+    }
+    /** Remove a member from a class */
+    removeMember(classId, text) {
+        const cls = this.ast.classes.get(classId);
+        if (cls) {
+            cls.members = cls.members.filter((m) => m.text !== text);
+        }
+        return this;
+    }
+    // ============ Relation Operations ============
+    /** Add a relation between classes */
+    addRelation(from, to, relationType, options) {
+        // Auto-create classes if they don't exist
+        if (!this.ast.classes.has(from))
+            this.addClass(from);
+        if (!this.ast.classes.has(to))
+            this.addClass(to);
+        const relation = {
+            id1: from,
+            id2: to,
+            relation: {
+                type1: relationType,
+                type2: 'none',
+                lineType: options?.lineType ?? 'solid',
+            },
+            title: options?.label,
+            relationTitle1: options?.labelFrom,
+            relationTitle2: options?.labelTo,
+        };
+        this.ast.relations.push(relation);
+        return this;
+    }
+    /** Add inheritance relation (A extends B) */
+    addInheritance(child, parent, options) {
+        return this.addRelation(parent, child, 'extension', options);
+    }
+    /** Add composition relation */
+    addComposition(whole, part, options) {
+        return this.addRelation(whole, part, 'composition', options);
+    }
+    /** Add aggregation relation */
+    addAggregation(whole, part, options) {
+        return this.addRelation(whole, part, 'aggregation', options);
+    }
+    /** Add dependency relation */
+    addDependency(from, to, options) {
+        return this.addRelation(from, to, 'dependency', options);
+    }
+    /** Add association (simple line) */
+    addAssociation(from, to, options) {
+        return this.addRelation(from, to, 'none', options);
+    }
+    /** Get all relations */
+    getRelations() {
+        return this.ast.relations;
+    }
+    /** Get relations for a class */
+    getRelationsFor(classId) {
+        return this.ast.relations.filter((r) => r.id1 === classId || r.id2 === classId);
+    }
+    /** Remove a relation */
+    removeRelation(from, to) {
+        this.ast.relations = this.ast.relations.filter((r) => !(r.id1 === from && r.id2 === to) && !(r.id1 === to && r.id2 === from));
+        return this;
+    }
+    // ============ Namespace Operations ============
+    /** Add a namespace */
+    addNamespace(name, classIds) {
+        this.ast.namespaces.set(name, {
+            name,
+            classes: classIds ?? [],
+        });
+        return this;
+    }
+    /** Add class to namespace */
+    addToNamespace(namespace, classId) {
+        let ns = this.ast.namespaces.get(namespace);
+        if (!ns) {
+            this.addNamespace(namespace);
+            ns = this.ast.namespaces.get(namespace);
+        }
+        if (!ns.classes.includes(classId)) {
+            ns.classes.push(classId);
+        }
+        return this;
+    }
+    /** Remove class from namespace */
+    removeFromNamespace(namespace, classId) {
+        const ns = this.ast.namespaces.get(namespace);
+        if (ns) {
+            ns.classes = ns.classes.filter((c) => c !== classId);
+        }
+        return this;
+    }
+    /** Get namespace for a class */
+    getNamespaceFor(classId) {
+        for (const [name, ns] of this.ast.namespaces) {
+            if (ns.classes.includes(classId))
+                return name;
+        }
+        return undefined;
+    }
+    // ============ Note Operations ============
+    /** Add a note */
+    addNote(text, forClass) {
+        this.ast.notes.push({ text, forClass });
+        return this;
+    }
+    /** Get all notes */
+    getNotes() {
+        return this.ast.notes;
+    }
+    // ============ Style Operations ============
+    /** Define a class style */
+    defineStyle(name, styles) {
+        this.ast.classDefs.set(name, { name, styles });
+        return this;
+    }
+    /** Apply style to a class */
+    applyStyle(classId, styleName) {
+        const cls = this.ast.classes.get(classId);
+        if (cls && !cls.cssClasses.includes(styleName)) {
+            cls.cssClasses.push(styleName);
+        }
+        return this;
+    }
+    // ============ Query Operations ============
+    /** Find classes by criteria */
+    findClasses(query) {
+        const results = [];
+        for (const cls of this.ast.classes.values()) {
+            if (query.hasAnnotation && !cls.annotations.includes(query.hasAnnotation))
+                continue;
+            if (query.hasStyle && !cls.cssClasses.includes(query.hasStyle))
+                continue;
+            if (query.inNamespace) {
+                const ns = this.ast.namespaces.get(query.inNamespace);
+                if (!ns || !ns.classes.includes(cls.id))
+                    continue;
+            }
+            results.push(cls);
+        }
+        return results;
+    }
+    /** Get all subclasses of a class (direct inheritance) */
+    getSubclasses(classId) {
+        return this.ast.relations
+            .filter((r) => r.id1 === classId && r.relation.type1 === 'extension')
+            .map((r) => r.id2);
+    }
+    /** Get parent class (direct inheritance) */
+    getParentClass(classId) {
+        const rel = this.ast.relations.find((r) => r.id2 === classId && r.relation.type1 === 'extension');
+        return rel?.id1;
+    }
+    /** Get all ancestors (transitive inheritance) */
+    getAncestors(classId) {
+        const ancestors = [];
+        let current = this.getParentClass(classId);
+        while (current) {
+            ancestors.push(current);
+            current = this.getParentClass(current);
+        }
+        return ancestors;
+    }
+    /** Get all descendants (transitive inheritance) */
+    getDescendants(classId) {
+        const descendants = [];
+        const queue = this.getSubclasses(classId);
+        while (queue.length > 0) {
+            const child = queue.shift();
+            descendants.push(child);
+            queue.push(...this.getSubclasses(child));
+        }
+        return descendants;
+    }
+}

package/dist/diagram-wrapper.d.ts

@@ -0,0 +1,66 @@
+/**
+ * DiagramWrapper Base Class
+ *
+ * Abstract base class for diagram wrapper classes.
+ * Provides common functionality for building, mutating, and querying diagrams.
+ *
+ * Each diagram type (Flowchart, Sequence, Class, etc.) extends this base class
+ * and implements diagram-specific operations.
+ */
+import type { RenderOptions } from './types/render-options.js';
+/**
+ * Base interface for all diagram ASTs
+ */
+export interface BaseDiagramAST {
+    type: string;
+    title?: string;
+    accDescription?: string;
+}
+/**
+ * Abstract base class for diagram wrappers.
+ *
+ * @typeParam T - The specific AST type this wrapper manages
+ */
+export declare abstract class DiagramWrapper<T extends BaseDiagramAST> {
+    protected ast: T;
+    protected constructor(ast: T);
+    /**
+     * Render the diagram to Mermaid syntax
+     * @param options - Render options
+     */
+    abstract render(options?: RenderOptions): string;
+    /**
+     * Create a deep clone of this diagram.
+     * Note: Returns the concrete type, not `this`, to avoid TypeScript polymorphic this issues.
+     */
+    abstract clone(): DiagramWrapper<T>;
+    /**
+     * Get the underlying AST
+     */
+    toAST(): T;
+    /**
+     * Get the diagram type
+     */
+    get type(): string;
+    /**
+     * Get the diagram title
+     */
+    get title(): string | undefined;
+    /**
+     * Set the diagram title
+     */
+    set title(value: string | undefined);
+    /**
+     * Get the accessibility description
+     */
+    get accDescription(): string | undefined;
+    /**
+     * Set the accessibility description
+     */
+    set accDescription(value: string | undefined);
+}
+/**
+ * Helper type for extracting the AST type from a DiagramWrapper subclass
+ */
+export type ASTOf<W extends DiagramWrapper<any>> = W extends DiagramWrapper<infer T> ? T : never;
+//# sourceMappingURL=diagram-wrapper.d.ts.map

package/dist/diagram-wrapper.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"diagram-wrapper.d.ts","sourceRoot":"","sources":["../src/diagram-wrapper.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAE/D;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAED;;;;GAIG;AACH,8BAAsB,cAAc,CAAC,CAAC,SAAS,cAAc;IAC3D,SAAS,CAAC,GAAG,EAAE,CAAC,CAAC;IAEjB,SAAS,aAAa,GAAG,EAAE,CAAC;IAQ5B;;;OAGG;IACH,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,aAAa,GAAG,MAAM;IAEhD;;;OAGG;IACH,QAAQ,CAAC,KAAK,IAAI,cAAc,CAAC,CAAC,CAAC;IAMnC;;OAEG;IACH,KAAK,IAAI,CAAC;IAIV;;OAEG;IACH,IAAI,IAAI,IAAI,MAAM,CAEjB;IAED;;OAEG;IACH,IAAI,KAAK,IAAI,MAAM,GAAG,SAAS,CAE9B;IAED;;OAEG;IACH,IAAI,KAAK,CAAC,KAAK,EAAE,MAAM,GAAG,SAAS,EAElC;IAED;;OAEG;IACH,IAAI,cAAc,IAAI,MAAM,GAAG,SAAS,CAEvC;IAED;;OAEG;IACH,IAAI,cAAc,CAAC,KAAK,EAAE,MAAM,GAAG,SAAS,EAE3C;CACF;AAED;;GAEG;AACH,MAAM,MAAM,KAAK,CAAC,CAAC,SAAS,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,cAAc,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC"}

package/dist/diagram-wrapper.js

@@ -0,0 +1,59 @@
+/**
+ * DiagramWrapper Base Class
+ *
+ * Abstract base class for diagram wrapper classes.
+ * Provides common functionality for building, mutating, and querying diagrams.
+ *
+ * Each diagram type (Flowchart, Sequence, Class, etc.) extends this base class
+ * and implements diagram-specific operations.
+ */
+/**
+ * Abstract base class for diagram wrappers.
+ *
+ * @typeParam T - The specific AST type this wrapper manages
+ */
+export class DiagramWrapper {
+    ast;
+    constructor(ast) {
+        this.ast = ast;
+    }
+    // ============================================
+    // Common Methods
+    // ============================================
+    /**
+     * Get the underlying AST
+     */
+    toAST() {
+        return this.ast;
+    }
+    /**
+     * Get the diagram type
+     */
+    get type() {
+        return this.ast.type;
+    }
+    /**
+     * Get the diagram title
+     */
+    get title() {
+        return this.ast.title;
+    }
+    /**
+     * Set the diagram title
+     */
+    set title(value) {
+        this.ast.title = value;
+    }
+    /**
+     * Get the accessibility description
+     */
+    get accDescription() {
+        return this.ast.accDescription;
+    }
+    /**
+     * Set the accessibility description
+     */
+    set accDescription(value) {
+        this.ast.accDescription = value;
+    }
+}