Add an isSourceFile() type guard

openrewrite · Sep 17, 2024 · 4e35a2d · 4e35a2d
1 parent d197954
commit 4e35a2d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/openrewrite/src/core/tree.ts b/openrewrite/src/core/tree.ts
@@ -262,6 +262,11 @@ const SourceFileSymbol = Symbol('SourceFile');
 // This allows using `SourceFile` like a class in certain situations (e.g. `Cursor.firstEnclosing()`)
 export const SourceFile = SourceFileSymbol;
 
+export function isSourceFile(tree: any & Tree): tree is SourceFile {
+    // return 'sourcePath' in tree && 'printer' in tree;
+    return tree && tree[SourceFileSymbol] === true;
+}
+
 export interface SourceFile extends Tree {
     [SourceFileSymbol]: true;