[INTERNAL_BRANCH=status-selection-syntax]

Author: salazarm

Diff for: js_modules/dagster-ui/packages/ui-core/src/asset-graph/types.ts

@@ -0,0 +1,7 @@
+import {GraphQueryItem} from '../app/GraphQueryImpl';
+import {AssetNodeForGraphQueryFragment} from './types/useAssetGraphData.types';
+
+export type AssetNode = AssetNodeForGraphQueryFragment;
+export type AssetGraphQueryItem = GraphQueryItem & {
+  node: AssetNode;
+};

Diff for: js_modules/dagster-ui/packages/ui-core/src/asset-selection/AssetSelectionSupplementaryDataVisitor.tsx

@@ -0,0 +1,38 @@
+import {CharStreams, CommonTokenStream} from 'antlr4ts';
+import {AbstractParseTreeVisitor} from 'antlr4ts/tree/AbstractParseTreeVisitor';
+
+import {AssetSelectionLexer} from './generated/AssetSelectionLexer';
+import {AssetSelectionParser, StatusAttributeExprContext} from './generated/AssetSelectionParser';
+import {AssetSelectionVisitor} from './generated/AssetSelectionVisitor';
+import {getValue} from './util';
+
+export type Filter = {field: 'status'; value: string};
+
+export class AssetSelectionSupplementaryDataVisitor
+  extends AbstractParseTreeVisitor<void>
+  implements AssetSelectionVisitor<void>
+{
+  public filters: Filter[] = [];
+
+  constructor() {
+    super();
+  }
+
+  protected defaultResult() {}
+
+  visitStatusAttributeExpr(ctx: StatusAttributeExprContext) {
+    const value: string = getValue(ctx.value());
+    this.filters.push({field: 'status', value});
+  }
+}
+
+export const parseExpression = (expression: string) => {
+  const inputStream = CharStreams.fromString(expression);
+  const lexer = new AssetSelectionLexer(inputStream);
+  const tokenStream = new CommonTokenStream(lexer);
+  const parser = new AssetSelectionParser(tokenStream);
+  const tree = parser.start();
+  const visitor = new AssetSelectionSupplementaryDataVisitor();
+  tree.accept(visitor);
+  return visitor.filters;
+};

Diff for: js_modules/dagster-ui/packages/ui-core/src/asset-selection/types.ts

@@ -0,0 +1,3 @@
+import {AssetKey} from '../assets/types';
+
+export type SupplementaryInformation = Record<string, AssetKey[]> | null | undefined;

Diff for: js_modules/dagster-ui/packages/ui-core/src/scripts/findCircularDependencies.ts

@@ -0,0 +1,56 @@
+// findCircularDeps.ts
+import path from 'path';
+
+import {Project} from 'ts-morph';
+
+const project = new Project({
+  tsConfigFilePath: 'tsconfig.json',
+});
+
+const graph = new Map<string, Set<string>>();
+
+for (const sourceFile of project.getSourceFiles()) {
+  const filePath = sourceFile.getFilePath();
+  const imports = sourceFile
+    .getImportDeclarations()
+    .filter((decl) => !decl.isTypeOnly())
+    .map((decl) =>
+      path.resolve(sourceFile.getDirectoryPath(), decl.getModuleSpecifierValue() + '.ts'),
+    );
+  graph.set(filePath, new Set(imports.filter((f) => project.getSourceFile(f))));
+}
+
+const visited = new Set<string>();
+const stack = new Set<string>();
+
+function dfs(node: string, pathStack: string[]) {
+  if (stack.has(node)) {
+    const cycleStart = pathStack.indexOf(node);
+    const cycle = pathStack.slice(cycleStart).concat(node);
+    const isGenerated = cycle.every((p) => p.includes('/generated/'));
+    if (!isGenerated) {
+      console.log('\nCircular dependency detected:');
+      console.log(cycle.join(' -> '));
+    }
+    return;
+  }
+
+  if (visited.has(node)) {
+    return;
+  }
+
+  visited.add(node);
+  stack.add(node);
+  pathStack.push(node);
+
+  for (const neighbor of graph.get(node) || []) {
+    dfs(neighbor, pathStack);
+  }
+
+  stack.delete(node);
+  pathStack.pop();
+}
+
+for (const node of graph.keys()) {
+  dfs(node, []);
+}