fix: generate artifacts that correspond to transformed code

Author: skoshx

src/transform/artifacts/artifacts.ts

@@ -94,15 +94,9 @@ export function extractCurrentScope(
 		if (currentPath.node.type === 'BlockStatement') {
 			scopes.push('{}')
 		}
-		if (
-			['FunctionDeclaration', 'FunctionExpression', 'ArrowFunctionExpression'].includes(
-				currentPath.node.type
-			)
-		) {
+		if (['FunctionDeclaration', 'FunctionExpression'].includes(currentPath.node.type)) {
 			let scopeName: string
-			if (currentPath.node.type === 'ArrowFunctionExpression') {
-				scopeName = extractFunctionName(currentPath.parent as VariableDeclarator | ObjectProperty)
-			} else if (currentPath.node.type === 'FunctionExpression') {
+			if (currentPath.node.type === 'FunctionExpression') {
 				if (currentPath.node.id?.type === 'Identifier') {
 					scopeName = extractFunctionName(currentPath.node as FunctionExpression)
 				} else {

test/artifacts.test.ts

@@ -131,7 +131,7 @@ describe('extractCurrentScope', () => {
 				foo()
 			}
 			`,
-			scopes: ['arrowFuncScope', '{}']
+			scopes: ['{}']
 		},
 		{
 			name: 'function expression scope',

test/regression.test.ts

@@ -0,0 +1,34 @@
+import { flytrapTransformArtifacts } from '../src/transform/index'
+import { describe, it } from 'vitest'
+import { extractArtifacts } from '../src/transform/artifacts/artifacts'
+
+describe('Regression', () => {
+	it('regression #1 > generates correct artifacts that match the transformed code', async () => {
+		const fixture = `
+		export const useRequestAccess = create((set) => ({
+			setOpen: (open: boolean) => set((state) => ({ open }))
+		}))
+		`
+
+		const transformedCode = await flytrapTransformArtifacts(fixture, '/stores.ts').code
+		const artifacts = extractArtifacts(fixture, '/stores.ts')
+
+		const functionOrCallIds = artifacts.map((a) => a.functionOrCallId)
+		const missingIds = new Set<string>([])
+		for (let i = 0; i < functionOrCallIds.length; i++) {
+			if (!transformedCode.includes(functionOrCallIds[i])) {
+				missingIds.add(functionOrCallIds[i])
+			}
+		}
+
+		if (missingIds.size > 0) {
+			throw new Error(
+				`Transformed code is missing ${missingIds.size} IDs. Following are missing: ${new Array(
+					...missingIds
+				)
+					.map((m) => `"${m}"`)
+					.join(', ')}`
+			)
+		}
+	})
+})