@statelyai/sdk 0.29.1 → 0.31.0

package/dist/relay.d.mts

@@ -1,4 +1,4 @@
-import { ProtocolMessage } from "./protocol.mjs";
+import { dt as ProtocolMessage } from "./protocol-DOt1Vz-5.mjs";
 
 //#region src/relay.d.ts
 type InspectionRelayRole = 'producer' | 'viewer';

package/dist/statelyPragma-D46lczJR.mjs

@@ -0,0 +1,102 @@
+import ts from "typescript";
+
+//#region src/statelyPragma.ts
+function findStatelyPragmaAttachments(sourceText, fileName = "machine.ts") {
+	const sourceFile = ts.createSourceFile(fileName, sourceText, ts.ScriptTarget.Latest, true, getScriptKind(fileName));
+	const attachments = [];
+	const visit = (node) => {
+		if (ts.isCallExpression(node) && isCreateMachineExpression(node.expression)) {
+			const attachNode = findAttachNode(node);
+			if (attachNode) attachments.push({
+				machineStart: getMachineExpressionStart(node.expression, sourceFile),
+				attachStart: attachNode.getStart(sourceFile),
+				pragma: findAttachedStatelyPragma(sourceText, attachNode)
+			});
+		}
+		ts.forEachChild(node, visit);
+	};
+	visit(sourceFile);
+	return attachments;
+}
+function getStatelyPragma(sourceText, fileName = "machine.ts", machineIndex = 0) {
+	return findStatelyPragmaAttachments(sourceText, fileName)[machineIndex]?.pragma;
+}
+function upsertStatelyPragma(sourceText, id, options = {}) {
+	const attachment = findStatelyPragmaAttachments(sourceText, options.fileName ?? "machine.ts")[options.machineIndex ?? 0];
+	if (!attachment) return sourceText;
+	const lineEnding = options.lineEnding ?? detectLineEnding(sourceText);
+	const canonicalComment = `// @statelyai id=${id}`;
+	if (attachment.pragma) return sourceText.slice(0, attachment.pragma.start) + canonicalComment + sourceText.slice(attachment.pragma.end);
+	const insertText = `${getIndentationAtOffset(sourceText, attachment.attachStart)}${canonicalComment}${lineEnding}`;
+	return sourceText.slice(0, attachment.attachStart) + insertText + sourceText.slice(attachment.attachStart);
+}
+function findAttachedStatelyPragma(sourceText, node) {
+	const comments = ts.getLeadingCommentRanges(sourceText, node.getFullStart()) ?? [];
+	for (let index = comments.length - 1; index >= 0; index -= 1) {
+		const comment = comments[index];
+		const parsed = parseStatelyPragma(sourceText, comment.pos, comment.end);
+		if (parsed) return parsed;
+	}
+}
+function parseStatelyPragma(sourceText, start, end) {
+	const match = normalizeCommentContent(sourceText.slice(start, end)).match(/^@statelyai(?:\s+(.*))?$/s);
+	if (!match) return;
+	const fields = parseFields(match[1] ?? "");
+	return {
+		tag: "@statelyai",
+		id: fields.id,
+		fields,
+		start,
+		end
+	};
+}
+function normalizeCommentContent(rawComment) {
+	if (rawComment.startsWith("//")) return rawComment.slice(2).trim();
+	if (rawComment.startsWith("/*")) return rawComment.slice(2, rawComment.endsWith("*/") ? -2 : void 0).split(/\r?\n/).map((line) => line.replace(/^\s*\*\s?/, "")).join(" ").trim();
+	return rawComment.trim();
+}
+function parseFields(input) {
+	const fields = {};
+	for (const token of input.split(/\s+/)) {
+		if (!token) continue;
+		const equalsIndex = token.indexOf("=");
+		if (equalsIndex <= 0) continue;
+		const key = token.slice(0, equalsIndex);
+		const value = token.slice(equalsIndex + 1);
+		if (!key || !value) continue;
+		fields[key] = value;
+	}
+	return fields;
+}
+function findAttachNode(node) {
+	let current = node;
+	while (current?.parent) {
+		if (ts.isVariableStatement(current.parent) || ts.isExpressionStatement(current.parent) || ts.isExportAssignment(current.parent)) return current.parent;
+		current = current.parent;
+	}
+	return current;
+}
+function isCreateMachineExpression(expression) {
+	const isMachineFactoryName = (name) => name === "createMachine" || name === "createMachineFromConfig";
+	return ts.isIdentifier(expression) && isMachineFactoryName(expression.text) || ts.isPropertyAccessExpression(expression) && isMachineFactoryName(expression.name.text);
+}
+function getMachineExpressionStart(expression, sourceFile) {
+	if (ts.isPropertyAccessExpression(expression)) return expression.name.getStart(sourceFile);
+	return expression.getStart(sourceFile);
+}
+function getIndentationAtOffset(sourceText, offset) {
+	const lineStart = sourceText.lastIndexOf("\n", Math.max(0, offset - 1)) + 1;
+	return sourceText.slice(lineStart, offset).match(/^[ \t]*/)?.[0] ?? "";
+}
+function detectLineEnding(sourceText) {
+	return sourceText.includes("\r\n") ? "\r\n" : "\n";
+}
+function getScriptKind(fileName) {
+	if (fileName.endsWith(".tsx")) return ts.ScriptKind.TSX;
+	if (fileName.endsWith(".jsx")) return ts.ScriptKind.JSX;
+	if (fileName.endsWith(".js")) return ts.ScriptKind.JS;
+	return ts.ScriptKind.TS;
+}
+
+//#endregion
+export { getStatelyPragma as n, upsertStatelyPragma as r, findStatelyPragmaAttachments as t };

package/dist/{studio-D-yqFd0u.d.mts → studio-GZ6E_xns.d.mts}

@@ -1,5 +1,5 @@
-import { A as ZodObject, D as ZodLiteral, E as ZodEnum, F as ZodUnknown, I as output, L as $loose, M as ZodRecord, N as ZodString, O as ZodNullable, P as ZodUnion, R as $strip, T as ZodBoolean, j as ZodOptional, k as ZodNumber, w as ZodArray } from "./index-D4pdeDpT.mjs";
-import { n as StatelyCredential, t as StatelyAuthMode } from "./auth-BPyCCHYo.mjs";
+import { A as ZodObject, D as ZodLiteral, E as ZodEnum, F as ZodUnknown, I as output, L as $loose, M as ZodRecord, N as ZodString, O as ZodNullable, P as ZodUnion, R as $strip, T as ZodBoolean, j as ZodOptional, k as ZodNumber, w as ZodArray } from "./index-ZlKF1x_z.mjs";
+import { n as StatelyCredential, t as StatelyAuthMode } from "./auth-Dg7prjeL.mjs";
 
 //#region ../api-contract/dist/errors.d.mts
 /**

package/dist/studio.d.mts

@@ -1,2 +1,2 @@
-import { A as Machine, C as StudioMachineRecord, D as VerifyApiKeyResponse, E as UpdateProjectInput, F as ProjectVisibility, I as VerifyResult, L as XStateVersion, M as Project, N as ProjectMachineSummary, O as createStatelyClient, P as ProjectSummary, R as ResourceErrorCode, S as StudioClientOptions, T as UpdateMachineInput, _ as ProjectMachine, a as CreateMachineTemplate, b as StudioApiErrorOptions, c as EnsureProjectInput, d as ExtractedMachine, f as GetMachineOptions, g as ProjectData, h as LegacyUpdateMachineInput, i as CreateMachineInput, j as MachineVersion, k as ExtractedMachines, l as ExtractMachinesInput, m as LegacyProjectVisibility, n as CreateMachineFromDefinitionInput, o as CreateProjectInput, p as LegacyExtractMachinesOptions, r as CreateMachineFromTemplateInput, s as DEFAULT_STATELY_BASE_URL, t as ConnectedRepo, u as ExtractMachinesResponse, v as RepoType, w as StudioRequestOptions, x as StudioClient, y as StudioApiError } from "./studio-D-yqFd0u.mjs";
+import { A as Machine, C as StudioMachineRecord, D as VerifyApiKeyResponse, E as UpdateProjectInput, F as ProjectVisibility, I as VerifyResult, L as XStateVersion, M as Project, N as ProjectMachineSummary, O as createStatelyClient, P as ProjectSummary, R as ResourceErrorCode, S as StudioClientOptions, T as UpdateMachineInput, _ as ProjectMachine, a as CreateMachineTemplate, b as StudioApiErrorOptions, c as EnsureProjectInput, d as ExtractedMachine, f as GetMachineOptions, g as ProjectData, h as LegacyUpdateMachineInput, i as CreateMachineInput, j as MachineVersion, k as ExtractedMachines, l as ExtractMachinesInput, m as LegacyProjectVisibility, n as CreateMachineFromDefinitionInput, o as CreateProjectInput, p as LegacyExtractMachinesOptions, r as CreateMachineFromTemplateInput, s as DEFAULT_STATELY_BASE_URL, t as ConnectedRepo, u as ExtractMachinesResponse, v as RepoType, w as StudioRequestOptions, x as StudioClient, y as StudioApiError } from "./studio-GZ6E_xns.mjs";
 export { ConnectedRepo, CreateMachineFromDefinitionInput, CreateMachineFromTemplateInput, CreateMachineInput, CreateMachineTemplate, CreateProjectInput, DEFAULT_STATELY_BASE_URL, EnsureProjectInput, ExtractMachinesInput, ExtractMachinesResponse, ExtractedMachine, ExtractedMachines, GetMachineOptions, LegacyExtractMachinesOptions, LegacyProjectVisibility, LegacyUpdateMachineInput, Machine, MachineVersion, Project, ProjectData, ProjectMachine, ProjectMachineSummary, ProjectSummary, ProjectVisibility, RepoType, ResourceErrorCode, StudioApiError, StudioApiErrorOptions, StudioClient, StudioClientOptions, StudioMachineRecord, StudioRequestOptions, UpdateMachineInput, UpdateProjectInput, VerifyApiKeyResponse, VerifyResult, XStateVersion, createStatelyClient };

package/dist/sync.d.mts

@@ -1,5 +1,5 @@
-import { b as DigraphConfig, i as StatelyGraph } from "./graph-C8E1Exad.mjs";
-import { C as StudioMachineRecord, L as XStateVersion, g as ProjectData, o as CreateProjectInput, x as StudioClient } from "./studio-D-yqFd0u.mjs";
+import { b as DigraphConfig, i as StatelyGraph } from "./graph-CehpcL7P.mjs";
+import { C as StudioMachineRecord, L as XStateVersion, g as ProjectData, o as CreateProjectInput, x as StudioClient } from "./studio-GZ6E_xns.mjs";
 import { GraphDiff } from "@statelyai/graph";
 
 //#region src/sync.d.ts

package/dist/sync.mjs

@@ -1,6 +1,7 @@
 import "./openapi-BTDjYWXd.mjs";
 import { r as createStatelyClient } from "./studio-rSI54Ese.mjs";
-import { b as serializeJS, c as detectXStateSourceProfile, f as graphChangeToPatchesResult, h as graphToXStateTS, i as INLINE_EXPRESSION_DIRECTIVE_FIELDS, l as diffGraphsSemantically, n as getStatelyPragma, o as alignCompareGraphIds, p as graphToMachineConfig, r as upsertStatelyPragma, t as findStatelyPragmaAttachments, v as normalizeMachineProfile, x as xstateSourceToConfig } from "./statelyPragma-p1vu8PmH.mjs";
+import { a as detectXStateSourceProfile, c as extractXStateMachineConfigEntries, d as graphToMachineConfig, g as normalizeMachineProfile, o as diffGraphsSemantically, p as graphToXStateTS, r as alignCompareGraphIds, t as INLINE_EXPRESSION_DIRECTIVE_FIELDS, u as graphChangeToPatchesResult, v as serializeJS } from "./dist-BQEaIgBI.mjs";
+import { n as getStatelyPragma, r as upsertStatelyPragma, t as findStatelyPragmaAttachments } from "./statelyPragma-D46lczJR.mjs";
 import { fromStudioMachine, toStudioMachine } from "./graph.mjs";
 import * as ts$1 from "typescript";
 import ts from "typescript";
@@ -820,11 +821,15 @@ function parseSourceToMachine(text, document, options = {}) {
 		...statesObjectReplacements,
 		...importReplacements
 	].filter(removeOverlappingRanges).sort(compareSourceRangesDescending));
-	if (sourceLocations.profile === "xstate-v6-alpha") {
+	if (sourceLocations.profile === "xstate-v6-alpha" || options.machineIndex !== void 0) {
 		const importedTransitionSources = collectImportedXStateTransitionSources(text, document, options);
+		const parseText = importedTransitionSources.length ? `${hydratedText}\n${importedTransitionSources.join("\n")}` : hydratedText;
+		const selectedMachineIndex = options.machineIndex ?? 0;
+		const config = extractXStateMachineConfigEntries(parseText, { profile: sourceLocations.profile }).find((entry) => entry.machineIndex === selectedMachineIndex)?.config;
+		if (!config) throw new Error(`No machine at index ${selectedMachineIndex} was found in ${document.fileName}.`);
 		return `import { createMachine } from "xstate";
 
-export const machine = createMachine(${serializeJS(xstateSourceToConfig(importedTransitionSources.length ? `${hydratedText}\n${importedTransitionSources.join("\n")}` : hydratedText, { profile: "xstate-v6-alpha" }), 0)});
+export const machine = createMachine(${serializeJS(config, 0)});
 `;
 	}
 	return hydratedText;
@@ -1092,12 +1097,13 @@ function offsetAtRangePosition$1(sourceText, line, character) {
 function applyMachineToSource(machineConfig, document, options = {}) {
 	if (!isMachineConfig(machineConfig)) return [];
 	const sourceFile = createSourceFile(document, document.getText());
-	const machineConfigNode = findMachineConfigObjectLiteral(sourceFile, options.machineIndex);
+	const machineConfigNode = getMappedRootObjectLiteral(document, sourceFile, options.sourceLocations) ?? findMachineConfigObjectLiteral(sourceFile, options.machineIndex);
 	if (!machineConfigNode) return [];
+	if (options.sourceUpdateStrategy === "rewrite") return addCodeImportReplacements(document, sourceFile, machineConfig, [createRewriteReplacement(document, sourceFile, machineConfigNode, machineConfig)]).sort(compareReplacementRangesDescending);
 	const parsedPatches = parseGraphPatches(options.patches);
 	const prevGraph = parseGraph(options.prevGraph);
 	const nextGraph = parseGraph(options.nextGraph);
-	const rootNode = getMappedRootObjectLiteral(document, sourceFile, options.sourceLocations) ?? machineConfigNode;
+	const rootNode = machineConfigNode;
 	if (!parsedPatches || !prevGraph || !nextGraph || parsedPatches.length === 0) {
 		if (wouldDeModularizeRootReplacement(options.sourceLocations)) return [];
 		return [createReplacement(document, sourceFile, rootNode, machineConfig)];
@@ -1171,6 +1177,7 @@ async function applyMachineToSourceDocuments(machineConfig, document, options =
 	const sourceFile = createSourceFile(document, document.getText());
 	const machineConfigNode = getMappedRootObjectLiteral(document, sourceFile, options.sourceLocations) ?? findMachineConfigObjectLiteral(sourceFile, options.machineIndex);
 	if (!machineConfigNode) return [];
+	if (options.sourceUpdateStrategy === "rewrite") return addCodeImportReplacements(document, sourceFile, machineConfig, [createRewriteReplacement(document, sourceFile, machineConfigNode, machineConfig)]).sort(compareReplacementRangesDescending);
 	const parsedPatches = parseGraphPatches(options.patches);
 	const prevGraph = parseGraph(options.prevGraph);
 	const nextGraph = parseGraph(options.nextGraph);
@@ -1319,8 +1326,8 @@ async function createTransitionReplacementForDocuments(document, sourceFile, mac
 	if (transitionSource && transitionTarget.previousEventType === transitionTarget.eventType) return [createPropertyValueReplacement(targetDocument, targetSourceFile, transitionSource.eventProperty, nextTransitionValue)];
 	if (transitionSource) return [createPropertyAssignmentReplacement(targetDocument, targetSourceFile, transitionSource.eventProperty, transitionTarget.eventType, nextTransitionValue)];
 	const onObject = getObjectLiteralProperty(stateObject, "on");
-	if (onObject) return [createObjectLiteralPropertyInsertion(targetDocument, targetSourceFile, onObject, transitionTarget.eventType, nextTransitionValue)];
-	return [createObjectLiteralPropertyInsertion(targetDocument, targetSourceFile, stateObject, "on", { [transitionTarget.eventType]: nextTransitionValue })];
+	if (onObject) return createObjectLiteralPropertyInsertion(targetDocument, targetSourceFile, onObject, transitionTarget.eventType, nextTransitionValue);
+	return createObjectLiteralPropertyInsertion(targetDocument, targetSourceFile, stateObject, "on", { [transitionTarget.eventType]: nextTransitionValue });
 }
 async function createStatePropertiesReplacementForDocuments(document, sourceFile, machineConfigNode, machineConfig, path, propertyNames, sourceLocations, openTextDocument) {
 	const nextStateConfig = getStateConfigByPath(machineConfig, path);
@@ -1485,8 +1492,30 @@ function createPropertyAssignmentReplacement(document, sourceFile, property, key
 	return createOffsetReplacement(document, property.getStart(sourceFile), property.getEnd(), text);
 }
 function createObjectLiteralPropertyInsertion(document, sourceFile, objectLiteral, key, nextValue) {
-	const text = objectPropertyTextForInsertion(objectLiteral, printObjectLiteralProperty(key, nextValue, sourceFile, getFormattingContext(document, sourceFile, objectLiteral)));
-	return createOffsetReplacement(document, objectLiteral.properties.end, objectLiteral.properties.end, text);
+	const propertyText = objectPropertyTextForInsertion(objectLiteral, printObjectLiteralProperty(key, nextValue, sourceFile, getFormattingContext(document, sourceFile, objectLiteral)));
+	return createObjectPropertyInsertionReplacements(document, objectLiteral, getObjectLiteralPropertyInsertionOffset(sourceFile, objectLiteral), propertyText);
+}
+function getObjectLiteralPropertyInsertionOffset(sourceFile, objectLiteral) {
+	const propertiesEnd = objectLiteral.properties.end;
+	const objectEnd = objectLiteral.getEnd() - 1;
+	const trailingTrivia = sourceFile.text.slice(propertiesEnd, objectEnd);
+	const commentMatch = /^[ \t]*(\/\/|\/\*)/.exec(trailingTrivia);
+	if (!commentMatch) return propertiesEnd;
+	const commentStart = propertiesEnd + commentMatch.index + commentMatch[0].length - 2;
+	if (commentMatch[1] === "//") {
+		const lineFeed = sourceFile.text.indexOf("\n", commentStart);
+		const lineEnd = lineFeed === -1 ? objectEnd : Math.min(lineFeed, objectEnd);
+		return lineEnd > commentStart && sourceFile.text.charCodeAt(lineEnd - 1) === 13 ? lineEnd - 1 : lineEnd;
+	}
+	const commentEnd = sourceFile.text.indexOf("*/", commentStart + 2);
+	return commentEnd === -1 ? propertiesEnd : Math.min(commentEnd + 2, objectEnd);
+}
+function createObjectPropertyInsertionReplacements(document, objectLiteral, insertionOffset, propertyText) {
+	const propertiesEnd = objectLiteral.properties.end;
+	const needsComma = objectLiteral.properties.length > 0 && !objectLiteral.properties.hasTrailingComma;
+	const replacements = [createOffsetReplacement(document, insertionOffset, insertionOffset, `${needsComma && insertionOffset === propertiesEnd ? "," : ""}${propertyText}`)];
+	if (needsComma && insertionOffset !== propertiesEnd) replacements.push(createOffsetReplacement(document, propertiesEnd, propertiesEnd, ","));
+	return replacements;
 }
 function getArrayElementDeletionRange(sourceFile, arrayLiteral, element) {
 	const elements = [...arrayLiteral.elements];
@@ -1583,14 +1612,16 @@ function createImplementationCreateReplacements(document, sourceFile, implementa
 	if (!implementation.id || !implementationCode) return null;
 	const categoryObject = getObjectLiteralProperty(implementationRoot, categoryName);
 	if (!categoryObject) {
-		const text = objectPropertyTextForInsertion(implementationRoot, `${propertyNameTextForSource(categoryName)}: {\n${getIndentForNewObjectProperty(sourceFile, implementationRoot)}${propertyNameTextForSource(implementation.id)}: ${implementationCode},\n${getIndentForObjectClose(sourceFile, implementationRoot)}}`);
-		return [createOffsetReplacement(document, implementationRoot.properties.end, implementationRoot.properties.end, text)];
+		const propertyText = objectPropertyTextForInsertion(implementationRoot, `${propertyNameTextForSource(categoryName)}: {\n${getIndentForNewObjectProperty(sourceFile, implementationRoot)}${propertyNameTextForSource(implementation.id)}: ${implementationCode},\n${getIndentForObjectClose(sourceFile, implementationRoot)}}`);
+		return createObjectPropertyInsertionReplacements(document, implementationRoot, getObjectLiteralPropertyInsertionOffset(sourceFile, implementationRoot), propertyText);
 	}
-	const text = objectPropertyTextForInsertion(categoryObject, `${propertyNameTextForSource(implementation.id)}: ${implementationCode}`);
-	return [createOffsetReplacement(document, categoryObject.properties.end, categoryObject.properties.end, text)];
+	const propertyText = objectPropertyTextForInsertion(categoryObject, `${propertyNameTextForSource(implementation.id)}: ${implementationCode}`);
+	return createObjectPropertyInsertionReplacements(document, categoryObject, getObjectLiteralPropertyInsertionOffset(sourceFile, categoryObject), propertyText);
 }
 function objectPropertyTextForInsertion(objectLiteral, propertyText) {
-	return `${objectLiteral.properties.length > 0 ? "," : ""}${getLineBreak(objectLiteral.getSourceFile())}${getIndentForNewObjectProperty(objectLiteral.getSourceFile(), objectLiteral)}${propertyText}`;
+	const lineBreak = getLineBreak(objectLiteral.getSourceFile());
+	const indent = getIndentForNewObjectProperty(objectLiteral.getSourceFile(), objectLiteral);
+	return `${lineBreak}${indent}${propertyText.startsWith(indent) ? propertyText.slice(indent.length) : propertyText}`;
 }
 function getLineBreak(sourceFile) {
 	return sourceFile.text.includes("\r\n") ? "\r\n" : "\n";
@@ -1886,7 +1917,7 @@ function getObjectLiteralPropertyDeletionRange(sourceFile, objectLiteral, proper
 	const propertyEnd = property.getEnd();
 	if (properties.length === 1) return {
 		start: propertyStart,
-		end: propertyEnd
+		end: objectLiteral.properties.hasTrailingComma ? objectLiteral.properties.end : propertyEnd
 	};
 	if (propertyIndex === properties.length - 1) {
 		const previousProperty = properties[propertyIndex - 1];
@@ -2254,6 +2285,21 @@ function createReplacement(document, sourceFile, node, value) {
 		newText: printExpression(valueToExpression(value, node), sourceFile, formatting)
 	};
 }
+function createRewriteReplacement(document, sourceFile, node, value) {
+	const start = document.positionAt(node.getStart(sourceFile));
+	const end = document.positionAt(node.getEnd());
+	const formatting = getFormattingContext(document, sourceFile, node);
+	return {
+		uri: getDocumentUri(document),
+		range: {
+			startLine: start.line,
+			startChar: start.character,
+			endLine: end.line,
+			endChar: end.character
+		},
+		newText: printExpression(valueToExpression(value), sourceFile, formatting)
+	};
+}
 function addCodeImportReplacements(document, sourceFile, machineConfig, replacements) {
 	const importReplacements = createCodeImportReplacements(document, sourceFile, collectCodeImports(machineConfig));
 	return [...replacements, ...importReplacements];
@@ -3165,11 +3211,12 @@ const xstateAdapter = {
 		return collectXStateSourceLocations(text, document, options);
 	},
 	async applyMachineToDocument(machineConfig, document, options) {
-		const rootSourceLocation = collectXStateSourceLocations(document.getText(), document, { machineIndex: options?.machineIndex })?.root ?? options?.sourceLocations?.root;
+		const detectedSourceLocations = collectXStateSourceLocations(document.getText(), document, { machineIndex: options?.machineIndex });
+		const rootSourceLocation = options?.sourceLocations?.root ?? detectedSourceLocations?.root;
 		const sanitizedMachineConfig = options?.preserveStatelyMetadata === false ? stripStatelyMetadata(machineConfig) : machineConfig;
 		if (!options?.patches?.length) return [];
 		const replacements = await applyMachineToSourceDocuments(sanitizedMachineConfig, document, options);
-		if (replacements.some(containsXStateExpressionPlaceholder) || replacements.some((replacement) => isImplicitRootReplacement(replacement, rootSourceLocation))) return [];
+		if (replacements.some(containsXStateExpressionPlaceholder) || options?.sourceUpdateStrategy !== "rewrite" && replacements.some((replacement) => isImplicitRootReplacement(replacement, rootSourceLocation))) return [];
 		return replacements;
 	}
 };
@@ -3503,7 +3550,7 @@ async function extractMachinesFromLocalSource(sourcePath, contents, machineIndex
 		return { machines: extractMachineConfigsFromAst(sourcePath, parseSourceToMachine(contents, createTextDocument(sourcePath, contents), {
 			machineIndex,
 			resolver: createFileSourceDocumentResolver()
-		}), machineIndex) };
+		}), machineIndex === void 0 ? void 0 : 0) };
 	} catch (error) {
 		return {
 			machines: [],

package/dist/{version-DC3yW9Dk.mjs → version-DbnY8nLV.mjs}

@@ -365,7 +365,7 @@ function createWebSocketTransport(options) {
 
 //#endregion
 //#region src/version.ts
-const SDK_VERSION = "0.29.1";
+const SDK_VERSION = "0.31.0";
 function inspectionRuntime() {
 	if (typeof window !== "undefined") return "browser";
 	if (typeof process !== "undefined" && process.versions?.node) return "node";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@statelyai/sdk",
-  "version": "0.29.1",
+  "version": "0.31.0",
   "license": "MIT",
   "bin": {
     "stately-sdk": "./dist/bin.mjs"
@@ -93,8 +93,8 @@
     "jsdom": "^27.4.0",
     "tsdown": "0.21.0-beta.2",
     "vitest": "^3.2.4",
-    "@statelyai/graph-tools": "0.7.2",
-    "@statelyai/api-contract": "0.4.3"
+    "@statelyai/api-contract": "0.4.5",
+    "@statelyai/graph-tools": "0.8.1"
   },
   "engines": {
     "node": ">=22.4"

package/schemas/statelyai.schema.json

@@ -34,6 +34,12 @@
       "default": 5,
       "description": "Default XState major version for sources that do not override it. Only XState 5+ is supported."
     },
+    "sourceUpdateStrategy": {
+      "type": "string",
+      "enum": ["preserve", "rewrite"],
+      "default": "preserve",
+      "description": "Controls whether visual edits preserve existing source formatting or rewrite the selected machine."
+    },
     "include": {
       "type": "array",
       "items": {

package/dist/embedPanel-Bpbq4sfJ.d.mts

@@ -1,13 +0,0 @@
-//#region src/embedPanel.d.ts
-type EmbedPanelPosition = 'left' | 'right' | 'bottom';
-interface EmbedPanelConfig {
-  id: string;
-  /** Restrict the panel to one side. Omit to allow any side. */
-  position?: EmbedPanelPosition;
-  /** Keep a one-click tab visible on its positioned or preferred side. */
-  tab?: boolean;
-  /** Open the panel when the embed is initialized. */
-  initiallyOpen?: boolean;
-}
-//#endregion
-export { EmbedPanelPosition as n, EmbedPanelConfig as t };