fix: TSNonNullExpression node support (#2264)

Author: cieplypolar

packages/tinyest-for-wgsl/src/parsers.ts

@@ -24,6 +24,10 @@ function isDeclared(ctx: Context, name: string) {
   return ctx.stack.some((scope) => scope.declaredNames.includes(name));
 }
 
+const tsFallthrough = (ctx: Context, node: { expression: babel.Expression }): tinyest.AnyNode => {
+  return transpile(ctx, node.expression);
+};
+
 const Transpilers: Partial<{
   [Type in JsNode['type']]: (
     ctx: Context,
@@ -292,13 +296,9 @@ const Transpilers: Partial<{
     return [NODE.break];
   },
 
-  TSAsExpression(ctx, node) {
-    return transpile(ctx, node.expression);
-  },
-
-  TSSatisfiesExpression(ctx, node) {
-    return transpile(ctx, node.expression);
-  },
+  TSAsExpression: tsFallthrough,
+  TSSatisfiesExpression: tsFallthrough,
+  TSNonNullExpression: tsFallthrough,
 };
 
 function transpile(ctx: Context, node: JsNode): tinyest.AnyNode {

packages/tinyest-for-wgsl/tests/parsers.test.ts

@@ -6,7 +6,7 @@ import { transpileFn } from '../src/parsers.ts';
 
 const parseRollup = (code: string) => acorn.parse(code, { ecmaVersion: 'latest' });
 const parseBabel = (code: string) =>
-  babel.parse(code, { sourceType: 'module' }).program.body[0] as Node;
+  babel.parse(code, { sourceType: 'module', plugins: ['typescript'] }).program.body[0] as Node;
 
 function dualTest(test: (p: (code: string) => Node | acorn.AnyNode) => void) {
   return () => {
@@ -189,4 +189,10 @@ describe('transpileFn', () => {
       expect(externalNames).toStrictEqual([]);
     }),
   );
+
+  it('handles TSNonNullExpression', () => {
+    const { body } = transpileFn(parseBabel('() => x!.y'));
+
+    expect(JSON.stringify(body)).toMatchInlineSnapshot(`"[0,[[10,[7,"x","y"]]]]"`);
+  });
 });