From 78bb187dde0699999f5a710a15e5ae3338d44264 Mon Sep 17 00:00:00 2001
From: Dominic Gannaway <trueadm@users.noreply.github.com>
Date: Wed, 15 Jan 2025 21:44:25 +0000
Subject: [PATCH] another fix

---
 .../2-analyze/visitors/AwaitExpression.js     | 35 +++++++++++++++++++
 .../client/visitors/AwaitExpression.js        | 17 +++++++++
 2 files changed, 52 insertions(+)
 create mode 100644 packages/svelte/src/compiler/phases/2-analyze/visitors/AwaitExpression.js
 create mode 100644 packages/svelte/src/compiler/phases/3-transform/client/visitors/AwaitExpression.js

diff --git a/packages/svelte/src/compiler/phases/2-analyze/visitors/AwaitExpression.js b/packages/svelte/src/compiler/phases/2-analyze/visitors/AwaitExpression.js
new file mode 100644
index 0000000000..8fda993559
--- /dev/null
+++ b/packages/svelte/src/compiler/phases/2-analyze/visitors/AwaitExpression.js
@@ -0,0 +1,35 @@
+/** @import { AwaitExpression } from 'estree' */
+/** @import { Context } from '../types' */
+import { extract_identifiers } from '../../../utils/ast.js';
+import * as w from '../../../warnings.js';
+
+/**
+ * @param {AwaitExpression} node
+ * @param {Context} context
+ */
+export function AwaitExpression(node, context) {
+	const declarator = context.path.at(-1);
+	const declaration = context.path.at(-2);
+	const program = context.path.at(-3);
+
+	if (context.state.ast_type === 'instance') {
+		if (
+			declarator?.type !== 'VariableDeclarator' ||
+			context.state.function_depth !== 1 ||
+			declaration?.type !== 'VariableDeclaration' ||
+			program?.type !== 'Program'
+		) {
+			throw new Error('TODO: invalid usage of AwaitExpression in component');
+		}
+		for (const declarator of declaration.declarations) {
+			for (const id of extract_identifiers(declarator.id)) {
+				const binding = context.state.scope.get(id.name);
+				if (binding !== null) {
+					binding.kind = 'derived';
+				}
+			}
+		}
+	}
+
+	context.next();
+}
diff --git a/packages/svelte/src/compiler/phases/3-transform/client/visitors/AwaitExpression.js b/packages/svelte/src/compiler/phases/3-transform/client/visitors/AwaitExpression.js
new file mode 100644
index 0000000000..99096fa1a3
--- /dev/null
+++ b/packages/svelte/src/compiler/phases/3-transform/client/visitors/AwaitExpression.js
@@ -0,0 +1,17 @@
+/** @import { AwaitExpression, Expression } from 'estree' */
+/** @import { ComponentContext } from '../types' */
+
+import * as b from '../../../../utils/builders.js';
+
+/**
+ * @param {AwaitExpression} node
+ * @param {ComponentContext} context
+ */
+export function AwaitExpression(node, context) {
+	// Inside component
+	if (context.state.analysis.instance) {
+		return b.call('$.await_derived', b.thunk(/** @type {Expression} */ (context.visit(node.argument))));
+	}
+
+	context.next();
+}