Resolve FIXMEs related to estree

src/compiler/transform/define-meta/index.ts

@@ -1,3 +1,4 @@
+import type { Program } from 'estree';
 import { toJs } from 'estree-util-to-js';
 import type MagicString from 'magic-string';
 
@@ -34,16 +35,11 @@ export function transformDefineMeta(params: Params): void {
 
   const { compiled } = nodes;
   const { defineMetaVariableDeclaration } = compiled;
-  // @ts-expect-error FIXME: These keys exists at runtime, perhaps I missed some type extension from `svelte/compiler`?
   const { start, end } = defineMetaVariableDeclaration;
 
   code.update(
-    start,
-    end,
-    toJs({
-      type: 'Program',
-      sourceType: 'module',
-      body: [defineMetaVariableDeclaration],
-    }).value
+    start as number,
+    end as number,
+    toJs(defineMetaVariableDeclaration as unknown as Program).value
   );
 }

src/compiler/transform/remove-export-default.ts

@@ -18,13 +18,12 @@ export function removeExportDefault(params: Params) {
   const { exportDefault, storiesFunctionDeclaration } = nodes;
 
   if (exportDefault.declaration.type === 'FunctionDeclaration') {
-    // @ts-expect-error FIXME: I couldn't find the right type (extension?) in the `estree`, but these exists at runtime
     const { start, end } = exportDefault;
 
     // NOTE: It updates code by removing `export default` from the stories function declaration.
     code.update(
-      start,
-      end,
+      start as number,
+      end as number,
       toJs({
         type: 'Program',
         sourceType: 'module',
@@ -34,7 +33,6 @@ export function removeExportDefault(params: Params) {
   }
 
   if (exportDefault.declaration.type === 'Identifier') {
-    // @ts-expect-error FIXME: I couldn't find the right type (extension?) in the `estree`, but these exists at runtime
-    code.remove(exportDefault.start, exportDefault.end);
+    code.remove(exportDefault.start as number, exportDefault.end as number);
   }
 }

src/compiler/transform/story/index.ts

@@ -40,8 +40,7 @@ export function transformStory(params: Params): void {
   });
 
   const { compiled } = component;
-  // @ts-expect-error FIXME: These keys exists at runtime, perhaps I missed some type extension from `svelte/compiler`?
   const { start, end } = compiled;
 
-  code.update(start, end, toJs(compiled as unknown as Program).value);
+  code.update(start as number, end as number, toJs(compiled as unknown as Program).value);
 }

src/estree.d.ts

@@ -0,0 +1,15 @@
+// Copied & modified from: https://github.com/sveltejs/language-tools/blob/master/packages%2Fsvelte2tsx%2Fsrc%2Festree.d.ts
+import 'estree';
+
+// estree does not have start/end in their public Node interface,
+// but the AST returned by svelte/compiler does.
+// We add the those properties here and add Node as an interface
+// to estree.
+
+declare module 'estree' {
+  export interface BaseNode {
+    start?: number;
+    end?: number;
+    [propName: string]: any;
+  }
+}

src/parser/extract/compiled/nodes.ts

@@ -9,6 +9,7 @@ import type {
   Program,
   VariableDeclaration,
 } from 'estree';
+import type { ProgramNode } from 'rollup';
 import type { Visitors } from 'zimmerframe';
 
 import {
@@ -58,7 +59,7 @@ const AST_NODES_NAMES = {
 } as const;
 
 interface Params {
-  ast: Program;
+  ast: Program | ProgramNode;
   filename?: string;
 }
 
@@ -147,7 +148,7 @@ export async function extractCompiledASTNodes(params: Params): Promise<CompiledA
     },
   };
 
-  walk(ast, state, visitors);
+  walk(ast as Program, state, visitors);
 
   const {
     defineMetaImport,