@ts-graphviz/ast 0.0.0-pr956-20240225073457 → 0.0.0-pr956-20240225160253

package/src/model-shim/to-model/plugins/DotPlugin.ts

@@ -1,25 +0,0 @@
-import { DotASTNode } from '../../../types.js';
-import { ConvertToModelPlugin } from '../types.js';
-import { CommentHolder } from './utils/comment-holder.js';
-
-export const DotPlugin: ConvertToModelPlugin<DotASTNode> = {
-  match(ast) {
-    return ast.type === 'Dot';
-  },
-  convert(context, ast) {
-    const commentHolder = new CommentHolder();
-    for (const stmt of ast.children) {
-      switch (stmt.type) {
-        case 'Comment':
-          commentHolder.set(stmt);
-          break;
-        case 'Graph': {
-          const graph = context.convert(stmt);
-          commentHolder.apply(graph, stmt.location);
-          return graph;
-        }
-      }
-    }
-    throw Error();
-  },
-};

package/src/model-shim/to-model/plugins/EdgePlugin.ts

@@ -1,26 +0,0 @@
-import { AttributeASTNode, EdgeASTNode } from '../../../types.js';
-import { ConvertToModelPlugin } from '../types.js';
-import { convertToEdgeTargetTuple } from './utils/convert-to-edge-target-tuple.js';
-
-export const EdgePlugin: ConvertToModelPlugin<EdgeASTNode> = {
-  match(ast) {
-    return ast.type === 'Edge';
-  },
-  convert(context, ast) {
-    const edge = new context.models.Edge(
-      convertToEdgeTargetTuple(ast),
-      ast.children
-        .filter<AttributeASTNode>(
-          (v): v is AttributeASTNode => v.type === 'Attribute',
-        )
-        .reduce(
-          (acc, curr) => {
-            acc[curr.key.value] = curr.value.value;
-            return acc;
-          },
-          {} as { [key: string]: string },
-        ),
-    );
-    return edge;
-  },
-};

package/src/model-shim/to-model/plugins/GraphPlugin.ts

@@ -1,15 +0,0 @@
-import { GraphASTNode } from '../../../types.js';
-import { ConvertToModelPlugin } from '../types.js';
-import { applyStatements } from './utils/apply-statments.js';
-
-export const GraphPlugin: ConvertToModelPlugin<GraphASTNode> = {
-  match(ast) {
-    return ast.type === 'Graph';
-  },
-  convert(context, ast) {
-    const G = ast.directed ? context.models.Digraph : context.models.Graph;
-    const graph = new G(ast.id?.value, ast.strict);
-    applyStatements(graph, ast.children);
-    return graph;
-  },
-};

package/src/model-shim/to-model/plugins/NodePlugin.ts

@@ -1,25 +0,0 @@
-import { AttributeASTNode, NodeASTNode } from '../../../types.js';
-import { ConvertToModelPlugin } from '../types.js';
-
-export const NodePlugin: ConvertToModelPlugin<NodeASTNode> = {
-  match(ast) {
-    return ast.type === 'Node';
-  },
-  convert(context, ast) {
-    const node = new context.models.Node(
-      ast.id.value,
-      ast.children
-        .filter<AttributeASTNode>(
-          (v): v is AttributeASTNode => v.type === 'Attribute',
-        )
-        .reduce(
-          (acc, curr) => {
-            acc[curr.key.value] = curr.value.value;
-            return acc;
-          },
-          {} as { [key: string]: string },
-        ),
-    );
-    return node;
-  },
-};

package/src/model-shim/to-model/plugins/SubgraphPlugin.ts

@@ -1,14 +0,0 @@
-import { SubgraphASTNode } from '../../../types.js';
-import { ConvertToModelPlugin } from '../types.js';
-import { applyStatements } from './utils/apply-statments.js';
-
-export const SubgraphPlugin: ConvertToModelPlugin<SubgraphASTNode> = {
-  match(ast) {
-    return ast.type === 'Subgraph';
-  },
-  convert(context, ast) {
-    const subgraph = new context.models.Subgraph(ast.id?.value);
-    applyStatements(subgraph, ast.children);
-    return subgraph;
-  },
-};

package/src/model-shim/to-model/plugins/index.ts

@@ -1,13 +0,0 @@
-import { DotPlugin } from './DotPlugin.js';
-import { EdgePlugin } from './EdgePlugin.js';
-import { GraphPlugin } from './GraphPlugin.js';
-import { NodePlugin } from './NodePlugin.js';
-import { SubgraphPlugin } from './SubgraphPlugin.js';
-
-export const defaultPlugins = [
-  NodePlugin,
-  EdgePlugin,
-  SubgraphPlugin,
-  GraphPlugin,
-  DotPlugin,
-];

package/src/model-shim/to-model/plugins/utils/apply-statments.ts

@@ -1,96 +0,0 @@
-import { GraphBaseModel } from '@ts-graphviz/common';
-import {
-  AttributeASTNode,
-  ClusterStatementASTNode,
-} from '../../../../types.js';
-import { CommentHolder } from './comment-holder.js';
-import { convertToEdgeTargetTuple } from './convert-to-edge-target-tuple.js';
-
-export function applyStatements(
-  graph: GraphBaseModel,
-  statements: ClusterStatementASTNode[],
-): void {
-  const commentHolder = new CommentHolder();
-  for (const stmt of statements) {
-    switch (stmt.type) {
-      case 'Subgraph': {
-        const subgraph = stmt.id
-          ? graph.subgraph(stmt.id.value)
-          : graph.subgraph();
-        applyStatements(subgraph, stmt.children);
-        commentHolder.apply(subgraph, stmt.location);
-        break;
-      }
-      case 'Attribute':
-        graph.set(stmt.key.value, stmt.value.value);
-        commentHolder.reset();
-        break;
-      case 'Node':
-        commentHolder.apply(
-          graph.node(
-            stmt.id.value,
-            stmt.children
-              .filter<AttributeASTNode>(
-                (v): v is AttributeASTNode => v.type === 'Attribute',
-              )
-              .reduce(
-                (acc, curr) => {
-                  acc[curr.key.value] = curr.value.value;
-                  return acc;
-                },
-                {} as { [key: string]: string },
-              ),
-          ),
-          stmt.location,
-        );
-        break;
-      case 'Edge':
-        commentHolder.apply(
-          graph.edge(
-            convertToEdgeTargetTuple(stmt),
-            stmt.children
-              .filter<AttributeASTNode>(
-                (v): v is AttributeASTNode => v.type === 'Attribute',
-              )
-              .reduce(
-                (acc, curr) => {
-                  acc[curr.key.value] = curr.value.value;
-                  return acc;
-                },
-                {} as { [key: string]: string },
-              ),
-          ),
-          stmt.location,
-        );
-        break;
-      case 'AttributeList': {
-        const attrs = stmt.children
-          .filter<AttributeASTNode>(
-            (v): v is AttributeASTNode => v.type === 'Attribute',
-          )
-          .reduce(
-            (acc, curr) => {
-              acc[curr.key.value] = curr.value.value;
-              return acc;
-            },
-            {} as { [key: string]: string },
-          );
-        switch (stmt.kind) {
-          case 'Edge':
-            graph.edge(attrs);
-            break;
-          case 'Node':
-            graph.node(attrs);
-            break;
-          case 'Graph':
-            graph.graph(attrs);
-            break;
-        }
-        commentHolder.reset();
-        break;
-      }
-      case 'Comment':
-        commentHolder.set(stmt);
-    }
-  }
-}

package/src/model-shim/to-model/plugins/utils/comment-holder.ts

@@ -1,31 +0,0 @@
-import type { HasComment } from '@ts-graphviz/common';
-import { CommentASTNode, FileRange } from '../../../../types.js';
-
-export class CommentHolder {
-  public comment: CommentASTNode | null = null;
-
-  public set(comment: CommentASTNode): void {
-    this.comment = comment;
-  }
-
-  public reset(): void {
-    this.comment = null;
-  }
-
-  public apply(model: HasComment, location?: FileRange): void {
-    if (location && this.comment?.location) {
-      if (this.comment?.kind === 'Block') {
-        if (this.comment.location.end.line === location.start.line - 1) {
-          model.comment = this.comment.value;
-        }
-      } else {
-        if (this.comment.location.end.line === location.start.line) {
-          model.comment = this.comment.value;
-        }
-      }
-    } else {
-      model.comment = this.comment?.value;
-    }
-    this.reset();
-  }
-}

package/src/model-shim/to-model/plugins/utils/convert-to-edge-target-tuple.ts

@@ -1,21 +0,0 @@
-import { EdgeTarget, EdgeTargetTuple } from '@ts-graphviz/common';
-import { EdgeASTNode } from '../../../../types.js';
-
-export function convertToEdgeTargetTuple(edge: EdgeASTNode): EdgeTargetTuple {
-  return edge.targets.map((t): EdgeTarget => {
-    switch (t.type) {
-      case 'NodeRef':
-        return {
-          id: t.id.value,
-          port: t.port?.value,
-          compass: t.compass?.value,
-        };
-      case 'NodeRefGroup':
-        return t.children.map((t) => ({
-          id: t.id.value,
-          port: t.port?.value,
-          compass: t.compass?.value,
-        }));
-    }
-  }) as EdgeTargetTuple;
-}

package/src/model-shim/to-model/to-model.test.ts

@@ -1,34 +0,0 @@
-import '@ts-graphviz/common';
-import '@ts-graphviz/core';
-import { expect, test } from 'vitest';
-import { createElement } from '../../builder/create-element.js';
-import { toModel } from './to-model.js';
-
-test('brank ast throw error', () => {
-  const ast = createElement('Dot', {}, []);
-
-  expect(() => toModel(ast)).toThrow();
-});
-
-test('comment', () => {
-  const ast = createElement('Dot', {}, [
-    createElement(
-      'Comment',
-      {
-        kind: 'Macro',
-        value: 'This is comment',
-      },
-      [],
-    ),
-    createElement(
-      'Graph',
-      {
-        strict: false,
-        directed: true,
-      },
-      [],
-    ),
-  ]);
-  const model = toModel(ast);
-  expect(model.comment).toStrictEqual('This is comment');
-});

package/src/model-shim/to-model/to-model.ts

@@ -1,16 +0,0 @@
-import { ToModelConverter } from './converter.js';
-import {
-  ASTToModel,
-  ConvertToModelOptions,
-  ToModelConvertableASTNode,
-} from './types.js';
-
-/**
- * @group Convert AST to Model
- */
-export function toModel<T extends ToModelConvertableASTNode>(
-  ast: T,
-  options?: ConvertToModelOptions,
-): ASTToModel<T> {
-  return new ToModelConverter(options).convert(ast);
-}

package/src/model-shim/to-model/types.ts

@@ -1,71 +0,0 @@
-import {
-  EdgeModel,
-  ModelsContext,
-  NodeModel,
-  RootGraphModel,
-  SubgraphModel,
-} from '@ts-graphviz/common';
-import {
-  DotASTNode,
-  EdgeASTNode,
-  GraphASTNode,
-  NodeASTNode,
-  SubgraphASTNode,
-} from '../../types.js';
-
-/**
- *  ModelOf is a type that determines the type of model to use depending on the value of T.
- * @group AST
- */
-export type ModelOf<T> = T extends 'Dot' | 'Graph'
-  ? RootGraphModel
-  : T extends 'Edge'
-    ? EdgeModel
-    : T extends 'Node'
-      ? NodeModel
-      : T extends 'Subgraph'
-        ? SubgraphModel
-        : never;
-
-/**
- * ASTToModel is a type that determines a model type from an AST.
- *
- * @group AST
- */
-export type ASTToModel<T> = T extends { type: infer U } ? ModelOf<U> : never;
-
-/**
- * This type is used to define what AST nodes can be converted to a model.
- * @group Convert AST to Model
- */
-export type ToModelConvertableASTNode =
-  | DotASTNode
-  | GraphASTNode
-  | SubgraphASTNode
-  | NodeASTNode
-  | EdgeASTNode;
-
-/**
- * @group Convert AST to Model
- */
-export interface ConvertToModelOptions {
-  models?: Partial<ModelsContext>;
-}
-
-/**
- * @group Convert AST to Model
- */
-export interface ConvertToModelContext {
-  models: ModelsContext;
-  convert<T extends ToModelConvertableASTNode>(ast: T): ASTToModel<T>;
-}
-
-/**
- * @group Convert AST to Model
- */
-export interface ConvertToModelPlugin<
-  T extends ToModelConvertableASTNode = ToModelConvertableASTNode,
-> {
-  match(ast: T): boolean;
-  convert(context: ConvertToModelContext, ast: T): ASTToModel<T>;
-}