@soda-gql/formatter 0.2.0 → 0.4.0

package/dist/index.cjs

@@ -0,0 +1,221 @@
+let __swc_core = require("@swc/core");
+let neverthrow = require("neverthrow");
+
+//#region packages/formatter/src/detection.ts
+/**
+* Check if an expression is a reference to gql
+* Handles: gql, namespace.gql
+*/
+const isGqlReference = (node, gqlIdentifiers) => {
+	if (node.type === "Identifier") return gqlIdentifiers.has(node.value);
+	if (node.type === "MemberExpression" && node.property.type === "Identifier" && node.property.value === "gql") return true;
+	return false;
+};
+/**
+* Check if a call expression is a gql definition call
+* Handles: gql.default(...), gql.model(...), gql.schemaName(...) (multi-schema)
+*/
+const isGqlDefinitionCall = (node, gqlIdentifiers) => {
+	if (node.callee.type !== "MemberExpression") return false;
+	const { object } = node.callee;
+	if (!isGqlReference(object, gqlIdentifiers)) return false;
+	const firstArg = node.arguments[0];
+	if (!firstArg || firstArg.expression.type !== "ArrowFunctionExpression") return false;
+	return true;
+};
+/**
+* Check if an object expression is a field selection object
+* Field selection objects are returned from arrow functions with ({ f }) or ({ f, $ }) parameter
+*/
+const isFieldSelectionObject = (object, parent) => {
+	let bodyObject = null;
+	if (parent.body.type === "ObjectExpression") bodyObject = parent.body;
+	else if (parent.body.type === "ParenthesisExpression") {
+		const inner = parent.body.expression;
+		if (inner.type === "ObjectExpression") bodyObject = inner;
+	}
+	if (!bodyObject) return false;
+	if (bodyObject.span.start !== object.span.start) return false;
+	const param = parent.params[0];
+	if (!param || param.type !== "ObjectPattern") return false;
+	return param.properties.some((p) => {
+		if (p.type === "KeyValuePatternProperty" && p.key.type === "Identifier") return p.key.value === "f";
+		if (p.type === "AssignmentPatternProperty" && p.key.type === "Identifier") return p.key.value === "f";
+		return false;
+	});
+};
+/**
+* Collect gql identifiers from import declarations
+*/
+const collectGqlIdentifiers = (module$1) => {
+	const gqlIdentifiers = /* @__PURE__ */ new Set();
+	for (const item of module$1.body) {
+		if (item.type !== "ImportDeclaration") continue;
+		for (const specifier of item.specifiers) {
+			if (specifier.type === "ImportSpecifier") {
+				if ((specifier.imported?.value ?? specifier.local.value) === "gql") gqlIdentifiers.add(specifier.local.value);
+			}
+			if (specifier.type === "ImportDefaultSpecifier") {
+				if (specifier.local.value === "gql") gqlIdentifiers.add(specifier.local.value);
+			}
+			if (specifier.type === "ImportNamespaceSpecifier") {}
+		}
+	}
+	return gqlIdentifiers;
+};
+
+//#endregion
+//#region packages/formatter/src/insertion.ts
+/**
+* The newline string to insert after object opening brace
+*/
+const NEWLINE_INSERTION = "\n";
+/**
+* Check if there's already a newline after the opening brace.
+* Uses string inspection rather than AST.
+*
+* @param source - The source code string
+* @param objectStartPos - The position of the `{` character in the source
+*/
+const hasExistingNewline = (source, objectStartPos) => {
+	const nextChar = source[objectStartPos + 1];
+	return nextChar === "\n" || nextChar === "\r";
+};
+
+//#endregion
+//#region packages/formatter/src/format.ts
+/**
+* Simple recursive AST traversal
+*/
+const traverseNode = (node, context, gqlIdentifiers, onObjectExpression) => {
+	if (node.type === "CallExpression" && isGqlDefinitionCall(node, gqlIdentifiers)) context = {
+		...context,
+		insideGqlDefinition: true
+	};
+	if (node.type === "ObjectExpression" && context.insideGqlDefinition && context.currentArrowFunction && isFieldSelectionObject(node, context.currentArrowFunction)) onObjectExpression(node, context.currentArrowFunction);
+	if (node.type === "CallExpression") {
+		const call = node;
+		traverseNode(call.callee, context, gqlIdentifiers, onObjectExpression);
+		for (const arg of call.arguments) traverseNode(arg.expression, context, gqlIdentifiers, onObjectExpression);
+	} else if (node.type === "ArrowFunctionExpression") {
+		const arrow = node;
+		const childContext = {
+			...context,
+			currentArrowFunction: arrow
+		};
+		if (arrow.body.type !== "BlockStatement") traverseNode(arrow.body, childContext, gqlIdentifiers, onObjectExpression);
+	} else if (node.type === "ParenthesisExpression") {
+		const paren = node;
+		if (paren.expression) traverseNode(paren.expression, context, gqlIdentifiers, onObjectExpression);
+	} else if (node.type === "MemberExpression") traverseNode(node.object, context, gqlIdentifiers, onObjectExpression);
+	else if (node.type === "ObjectExpression") {
+		const obj = node;
+		for (const prop of obj.properties) if (prop.type === "SpreadElement") traverseNode(prop.arguments, context, gqlIdentifiers, onObjectExpression);
+		else if (prop.type === "KeyValueProperty") traverseNode(prop.value, context, gqlIdentifiers, onObjectExpression);
+	} else for (const value of Object.values(node)) if (Array.isArray(value)) {
+		for (const child of value) if (child && typeof child === "object" && "type" in child) traverseNode(child, context, gqlIdentifiers, onObjectExpression);
+	} else if (value && typeof value === "object" && "type" in value) traverseNode(value, context, gqlIdentifiers, onObjectExpression);
+};
+const traverse = (module$1, gqlIdentifiers, onObjectExpression) => {
+	for (const statement of module$1.body) traverseNode(statement, {
+		insideGqlDefinition: false,
+		currentArrowFunction: null
+	}, gqlIdentifiers, onObjectExpression);
+};
+/**
+* Format soda-gql field selection objects by inserting newlines.
+* This preserves multi-line formatting when using Biome/Prettier.
+*/
+const format = (options) => {
+	const { sourceCode, filePath } = options;
+	let module$1;
+	try {
+		const program = (0, __swc_core.parseSync)(sourceCode, {
+			syntax: "typescript",
+			tsx: filePath?.endsWith(".tsx") ?? true,
+			target: "es2022",
+			decorators: false,
+			dynamicImport: true
+		});
+		if (program.type !== "Module") return (0, neverthrow.err)({
+			type: "FormatError",
+			code: "PARSE_ERROR",
+			message: `Not a module${filePath ? ` (${filePath})` : ""}`
+		});
+		module$1 = program;
+	} catch (cause) {
+		return (0, neverthrow.err)({
+			type: "FormatError",
+			code: "PARSE_ERROR",
+			message: `Failed to parse source code${filePath ? ` (${filePath})` : ""}`,
+			cause
+		});
+	}
+	const spanOffset = module$1.span.end - sourceCode.length + 1;
+	const gqlIdentifiers = collectGqlIdentifiers(module$1);
+	if (gqlIdentifiers.size === 0) return (0, neverthrow.ok)({
+		modified: false,
+		sourceCode
+	});
+	const insertionPoints = [];
+	traverse(module$1, gqlIdentifiers, (object, _parent) => {
+		const objectStart = object.span.start - spanOffset;
+		if (hasExistingNewline(sourceCode, objectStart)) return;
+		insertionPoints.push(objectStart + 1);
+	});
+	if (insertionPoints.length === 0) return (0, neverthrow.ok)({
+		modified: false,
+		sourceCode
+	});
+	const sortedPoints = [...insertionPoints].sort((a, b) => b - a);
+	let result = sourceCode;
+	for (const pos of sortedPoints) result = result.slice(0, pos) + NEWLINE_INSERTION + result.slice(pos);
+	return (0, neverthrow.ok)({
+		modified: true,
+		sourceCode: result
+	});
+};
+/**
+* Check if a file needs formatting (has unformatted field selections).
+* Useful for pre-commit hooks or CI checks.
+*/
+const needsFormat = (options) => {
+	const { sourceCode, filePath } = options;
+	let module$1;
+	try {
+		const program = (0, __swc_core.parseSync)(sourceCode, {
+			syntax: "typescript",
+			tsx: filePath?.endsWith(".tsx") ?? true,
+			target: "es2022",
+			decorators: false,
+			dynamicImport: true
+		});
+		if (program.type !== "Module") return (0, neverthrow.err)({
+			type: "FormatError",
+			code: "PARSE_ERROR",
+			message: `Not a module${filePath ? ` (${filePath})` : ""}`
+		});
+		module$1 = program;
+	} catch (cause) {
+		return (0, neverthrow.err)({
+			type: "FormatError",
+			code: "PARSE_ERROR",
+			message: `Failed to parse source code${filePath ? ` (${filePath})` : ""}`,
+			cause
+		});
+	}
+	const spanOffset = module$1.span.end - sourceCode.length + 1;
+	const gqlIdentifiers = collectGqlIdentifiers(module$1);
+	if (gqlIdentifiers.size === 0) return (0, neverthrow.ok)(false);
+	let needsFormatting = false;
+	traverse(module$1, gqlIdentifiers, (object, _parent) => {
+		if (needsFormatting) return;
+		if (!hasExistingNewline(sourceCode, object.span.start - spanOffset)) needsFormatting = true;
+	});
+	return (0, neverthrow.ok)(needsFormatting);
+};
+
+//#endregion
+exports.format = format;
+exports.needsFormat = needsFormat;
+//# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.cjs","names":["bodyObject: ObjectExpression | null","module","module","module: Module","insertionPoints: number[]"],"sources":["../src/detection.ts","../src/insertion.ts","../src/format.ts"],"sourcesContent":["import type {\n  ArrowFunctionExpression,\n  CallExpression,\n  Expression,\n  Module,\n  ObjectExpression,\n  ObjectPatternProperty,\n} from \"@swc/types\";\n\n/**\n * Check if an expression is a reference to gql\n * Handles: gql, namespace.gql\n */\nexport const isGqlReference = (node: Expression, gqlIdentifiers: ReadonlySet<string>): boolean => {\n  if (node.type === \"Identifier\") {\n    return gqlIdentifiers.has(node.value);\n  }\n  if (node.type === \"MemberExpression\" && node.property.type === \"Identifier\" && node.property.value === \"gql\") {\n    return true;\n  }\n  return false;\n};\n\n/**\n * Check if a call expression is a gql definition call\n * Handles: gql.default(...), gql.model(...), gql.schemaName(...) (multi-schema)\n */\nexport const isGqlDefinitionCall = (node: CallExpression, gqlIdentifiers: ReadonlySet<string>): boolean => {\n  if (node.callee.type !== \"MemberExpression\") return false;\n  const { object } = node.callee;\n\n  // Check object is a gql reference first\n  if (!isGqlReference(object, gqlIdentifiers)) return false;\n\n  // Verify first argument is an arrow function (factory pattern)\n  // SWC's arguments are ExprOrSpread[], so access via .expression\n  const firstArg = node.arguments[0];\n  if (!firstArg || firstArg.expression.type !== \"ArrowFunctionExpression\") return false;\n\n  return true;\n};\n\n/**\n * Check if an object expression is a field selection object\n * Field selection objects are returned from arrow functions with ({ f }) or ({ f, $ }) parameter\n */\nexport const isFieldSelectionObject = (object: ObjectExpression, parent: ArrowFunctionExpression): boolean => {\n  // The object must be the body of the arrow function\n  // Handle both direct ObjectExpression and parenthesized ObjectExpression: `({ ... })`\n  let bodyObject: ObjectExpression | null = null;\n\n  if (parent.body.type === \"ObjectExpression\") {\n    bodyObject = parent.body;\n  } else if (parent.body.type === \"ParenthesisExpression\") {\n    // Handle `({ f }) => ({ ...f.id() })` pattern where body is parenthesized\n    const inner = parent.body.expression;\n    if (inner.type === \"ObjectExpression\") {\n      bodyObject = inner;\n    }\n  }\n\n  if (!bodyObject) return false;\n  if (bodyObject.span.start !== object.span.start) return false;\n\n  // Check if first parameter has 'f' destructured\n  const param = parent.params[0];\n  if (!param || param.type !== \"ObjectPattern\") return false;\n\n  return param.properties.some((p: ObjectPatternProperty) => {\n    if (p.type === \"KeyValuePatternProperty\" && p.key.type === \"Identifier\") {\n      return p.key.value === \"f\";\n    }\n    if (p.type === \"AssignmentPatternProperty\" && p.key.type === \"Identifier\") {\n      return p.key.value === \"f\";\n    }\n    return false;\n  });\n};\n\n/**\n * Collect gql identifiers from import declarations\n */\nexport const collectGqlIdentifiers = (module: Module): Set<string> => {\n  const gqlIdentifiers = new Set<string>();\n\n  for (const item of module.body) {\n    if (item.type !== \"ImportDeclaration\") continue;\n\n    for (const specifier of item.specifiers) {\n      if (specifier.type === \"ImportSpecifier\") {\n        const imported = specifier.imported?.value ?? specifier.local.value;\n        if (imported === \"gql\") {\n          gqlIdentifiers.add(specifier.local.value);\n        }\n      }\n      if (specifier.type === \"ImportDefaultSpecifier\") {\n        // Check if default import might be gql\n        if (specifier.local.value === \"gql\") {\n          gqlIdentifiers.add(specifier.local.value);\n        }\n      }\n      if (specifier.type === \"ImportNamespaceSpecifier\") {\n        // namespace import: import * as ns from \"...\"\n        // Would need ns.gql pattern - handled by isGqlReference\n      }\n    }\n  }\n\n  return gqlIdentifiers;\n};\n","/**\n * The newline string to insert after object opening brace\n */\nexport const NEWLINE_INSERTION = \"\\n\";\n\n/**\n * Check if there's already a newline after the opening brace.\n * Uses string inspection rather than AST.\n *\n * @param source - The source code string\n * @param objectStartPos - The position of the `{` character in the source\n */\nexport const hasExistingNewline = (source: string, objectStartPos: number): boolean => {\n  // Skip the `{` character\n  const pos = objectStartPos + 1;\n\n  // Check if next character is a newline\n  const nextChar = source[pos];\n  return nextChar === \"\\n\" || nextChar === \"\\r\";\n};\n","import { parseSync } from \"@swc/core\";\nimport type { ArrowFunctionExpression, CallExpression, Module, Node, ObjectExpression } from \"@swc/types\";\nimport { err, ok, type Result } from \"neverthrow\";\nimport { collectGqlIdentifiers, isFieldSelectionObject, isGqlDefinitionCall } from \"./detection\";\nimport { hasExistingNewline, NEWLINE_INSERTION } from \"./insertion\";\nimport type { FormatError, FormatOptions, FormatResult } from \"./types\";\n\ntype TraversalContext = {\n  insideGqlDefinition: boolean;\n  currentArrowFunction: ArrowFunctionExpression | null;\n};\n\n/**\n * Simple recursive AST traversal\n */\nconst traverseNode = (\n  node: Node,\n  context: TraversalContext,\n  gqlIdentifiers: ReadonlySet<string>,\n  onObjectExpression: (object: ObjectExpression, parent: ArrowFunctionExpression) => void,\n): void => {\n  // Check for gql definition call entry\n  if (node.type === \"CallExpression\" && isGqlDefinitionCall(node as CallExpression, gqlIdentifiers)) {\n    context = { ...context, insideGqlDefinition: true };\n  }\n\n  // Handle object expressions - check if it's the body of the current arrow function\n  if (\n    node.type === \"ObjectExpression\" &&\n    context.insideGqlDefinition &&\n    context.currentArrowFunction &&\n    isFieldSelectionObject(node as ObjectExpression, context.currentArrowFunction)\n  ) {\n    onObjectExpression(node as ObjectExpression, context.currentArrowFunction);\n  }\n\n  // Recursively visit children\n  if (node.type === \"CallExpression\") {\n    const call = node as CallExpression;\n    traverseNode(call.callee as Node, context, gqlIdentifiers, onObjectExpression);\n    for (const arg of call.arguments) {\n      traverseNode(arg.expression as Node, context, gqlIdentifiers, onObjectExpression);\n    }\n  } else if (node.type === \"ArrowFunctionExpression\") {\n    const arrow = node as ArrowFunctionExpression;\n    // Update context with the new arrow function\n    const childContext = { ...context, currentArrowFunction: arrow };\n    if (arrow.body.type !== \"BlockStatement\") {\n      traverseNode(arrow.body as Node, childContext, gqlIdentifiers, onObjectExpression);\n    }\n  } else if (node.type === \"ParenthesisExpression\") {\n    // Handle parenthesized expressions like `({ ...f.id() })`\n    // biome-ignore lint/suspicious/noExplicitAny: SWC types\n    const paren = node as any;\n    if (paren.expression) {\n      traverseNode(paren.expression as Node, context, gqlIdentifiers, onObjectExpression);\n    }\n  } else if (node.type === \"MemberExpression\") {\n    // biome-ignore lint/suspicious/noExplicitAny: SWC types\n    const member = node as any;\n    traverseNode(member.object as Node, context, gqlIdentifiers, onObjectExpression);\n  } else if (node.type === \"ObjectExpression\") {\n    const obj = node as ObjectExpression;\n    for (const prop of obj.properties) {\n      if (prop.type === \"SpreadElement\") {\n        traverseNode(prop.arguments as Node, context, gqlIdentifiers, onObjectExpression);\n      } else if (prop.type === \"KeyValueProperty\") {\n        traverseNode(prop.value as Node, context, gqlIdentifiers, onObjectExpression);\n      }\n    }\n  } else {\n    // Generic traversal for other node types\n    for (const value of Object.values(node)) {\n      if (Array.isArray(value)) {\n        for (const child of value) {\n          if (child && typeof child === \"object\" && \"type\" in child) {\n            traverseNode(child as Node, context, gqlIdentifiers, onObjectExpression);\n          }\n        }\n      } else if (value && typeof value === \"object\" && \"type\" in value) {\n        traverseNode(value as Node, context, gqlIdentifiers, onObjectExpression);\n      }\n    }\n  }\n};\n\nconst traverse = (\n  module: Module,\n  gqlIdentifiers: ReadonlySet<string>,\n  onObjectExpression: (object: ObjectExpression, parent: ArrowFunctionExpression) => void,\n): void => {\n  for (const statement of module.body) {\n    traverseNode(statement, { insideGqlDefinition: false, currentArrowFunction: null }, gqlIdentifiers, onObjectExpression);\n  }\n};\n\n/**\n * Format soda-gql field selection objects by inserting newlines.\n * This preserves multi-line formatting when using Biome/Prettier.\n */\nexport const format = (options: FormatOptions): Result<FormatResult, FormatError> => {\n  const { sourceCode, filePath } = options;\n\n  // Parse source code with SWC\n  let module: Module;\n  try {\n    const program = parseSync(sourceCode, {\n      syntax: \"typescript\",\n      tsx: filePath?.endsWith(\".tsx\") ?? true,\n      target: \"es2022\",\n      decorators: false,\n      dynamicImport: true,\n    });\n\n    if (program.type !== \"Module\") {\n      return err({\n        type: \"FormatError\",\n        code: \"PARSE_ERROR\",\n        message: `Not a module${filePath ? ` (${filePath})` : \"\"}`,\n      });\n    }\n    module = program;\n  } catch (cause) {\n    return err({\n      type: \"FormatError\",\n      code: \"PARSE_ERROR\",\n      message: `Failed to parse source code${filePath ? ` (${filePath})` : \"\"}`,\n      cause,\n    });\n  }\n\n  // Calculate span offset for position normalization\n  // SWC's BytePos counter accumulates across parseSync calls within the same process\n  const spanOffset = module.span.end - sourceCode.length + 1;\n\n  // Collect gql identifiers from imports\n  const gqlIdentifiers = collectGqlIdentifiers(module);\n  if (gqlIdentifiers.size === 0) {\n    return ok({ modified: false, sourceCode });\n  }\n\n  // Collect insertion points\n  const insertionPoints: number[] = [];\n\n  traverse(module, gqlIdentifiers, (object, _parent) => {\n    // Calculate actual position in source\n    const objectStart = object.span.start - spanOffset;\n\n    // Check if already has newline\n    if (hasExistingNewline(sourceCode, objectStart)) return;\n\n    // Record insertion point (position after `{`)\n    insertionPoints.push(objectStart + 1);\n  });\n\n  // Apply insertions\n  if (insertionPoints.length === 0) {\n    return ok({ modified: false, sourceCode });\n  }\n\n  // Sort in descending order to insert from end to beginning\n  // This preserves earlier positions while modifying later parts\n  const sortedPoints = [...insertionPoints].sort((a, b) => b - a);\n\n  let result = sourceCode;\n  for (const pos of sortedPoints) {\n    result = result.slice(0, pos) + NEWLINE_INSERTION + result.slice(pos);\n  }\n\n  return ok({ modified: true, sourceCode: result });\n};\n\n/**\n * Check if a file needs formatting (has unformatted field selections).\n * Useful for pre-commit hooks or CI checks.\n */\nexport const needsFormat = (options: FormatOptions): Result<boolean, FormatError> => {\n  const { sourceCode, filePath } = options;\n\n  // Parse source code with SWC\n  let module: Module;\n  try {\n    const program = parseSync(sourceCode, {\n      syntax: \"typescript\",\n      tsx: filePath?.endsWith(\".tsx\") ?? true,\n      target: \"es2022\",\n      decorators: false,\n      dynamicImport: true,\n    });\n\n    if (program.type !== \"Module\") {\n      return err({\n        type: \"FormatError\",\n        code: \"PARSE_ERROR\",\n        message: `Not a module${filePath ? ` (${filePath})` : \"\"}`,\n      });\n    }\n    module = program;\n  } catch (cause) {\n    return err({\n      type: \"FormatError\",\n      code: \"PARSE_ERROR\",\n      message: `Failed to parse source code${filePath ? ` (${filePath})` : \"\"}`,\n      cause,\n    });\n  }\n\n  const spanOffset = module.span.end - sourceCode.length + 1;\n  const gqlIdentifiers = collectGqlIdentifiers(module);\n\n  if (gqlIdentifiers.size === 0) {\n    return ok(false);\n  }\n\n  let needsFormatting = false;\n\n  traverse(module, gqlIdentifiers, (object, _parent) => {\n    if (needsFormatting) return; // Early exit\n\n    const objectStart = object.span.start - spanOffset;\n    if (!hasExistingNewline(sourceCode, objectStart)) {\n      needsFormatting = true;\n    }\n  });\n\n  return ok(needsFormatting);\n};\n"],"mappings":";;;;;;;;AAaA,MAAa,kBAAkB,MAAkB,mBAAiD;AAChG,KAAI,KAAK,SAAS,aAChB,QAAO,eAAe,IAAI,KAAK,MAAM;AAEvC,KAAI,KAAK,SAAS,sBAAsB,KAAK,SAAS,SAAS,gBAAgB,KAAK,SAAS,UAAU,MACrG,QAAO;AAET,QAAO;;;;;;AAOT,MAAa,uBAAuB,MAAsB,mBAAiD;AACzG,KAAI,KAAK,OAAO,SAAS,mBAAoB,QAAO;CACpD,MAAM,EAAE,WAAW,KAAK;AAGxB,KAAI,CAAC,eAAe,QAAQ,eAAe,CAAE,QAAO;CAIpD,MAAM,WAAW,KAAK,UAAU;AAChC,KAAI,CAAC,YAAY,SAAS,WAAW,SAAS,0BAA2B,QAAO;AAEhF,QAAO;;;;;;AAOT,MAAa,0BAA0B,QAA0B,WAA6C;CAG5G,IAAIA,aAAsC;AAE1C,KAAI,OAAO,KAAK,SAAS,mBACvB,cAAa,OAAO;UACX,OAAO,KAAK,SAAS,yBAAyB;EAEvD,MAAM,QAAQ,OAAO,KAAK;AAC1B,MAAI,MAAM,SAAS,mBACjB,cAAa;;AAIjB,KAAI,CAAC,WAAY,QAAO;AACxB,KAAI,WAAW,KAAK,UAAU,OAAO,KAAK,MAAO,QAAO;CAGxD,MAAM,QAAQ,OAAO,OAAO;AAC5B,KAAI,CAAC,SAAS,MAAM,SAAS,gBAAiB,QAAO;AAErD,QAAO,MAAM,WAAW,MAAM,MAA6B;AACzD,MAAI,EAAE,SAAS,6BAA6B,EAAE,IAAI,SAAS,aACzD,QAAO,EAAE,IAAI,UAAU;AAEzB,MAAI,EAAE,SAAS,+BAA+B,EAAE,IAAI,SAAS,aAC3D,QAAO,EAAE,IAAI,UAAU;AAEzB,SAAO;GACP;;;;;AAMJ,MAAa,yBAAyB,aAAgC;CACpE,MAAM,iCAAiB,IAAI,KAAa;AAExC,MAAK,MAAM,QAAQC,SAAO,MAAM;AAC9B,MAAI,KAAK,SAAS,oBAAqB;AAEvC,OAAK,MAAM,aAAa,KAAK,YAAY;AACvC,OAAI,UAAU,SAAS,mBAErB;SADiB,UAAU,UAAU,SAAS,UAAU,MAAM,WAC7C,MACf,gBAAe,IAAI,UAAU,MAAM,MAAM;;AAG7C,OAAI,UAAU,SAAS,0BAErB;QAAI,UAAU,MAAM,UAAU,MAC5B,gBAAe,IAAI,UAAU,MAAM,MAAM;;AAG7C,OAAI,UAAU,SAAS,4BAA4B;;;AAOvD,QAAO;;;;;;;;ACzGT,MAAa,oBAAoB;;;;;;;;AASjC,MAAa,sBAAsB,QAAgB,mBAAoC;CAKrF,MAAM,WAAW,OAHL,iBAAiB;AAI7B,QAAO,aAAa,QAAQ,aAAa;;;;;;;;ACH3C,MAAM,gBACJ,MACA,SACA,gBACA,uBACS;AAET,KAAI,KAAK,SAAS,oBAAoB,oBAAoB,MAAwB,eAAe,CAC/F,WAAU;EAAE,GAAG;EAAS,qBAAqB;EAAM;AAIrD,KACE,KAAK,SAAS,sBACd,QAAQ,uBACR,QAAQ,wBACR,uBAAuB,MAA0B,QAAQ,qBAAqB,CAE9E,oBAAmB,MAA0B,QAAQ,qBAAqB;AAI5E,KAAI,KAAK,SAAS,kBAAkB;EAClC,MAAM,OAAO;AACb,eAAa,KAAK,QAAgB,SAAS,gBAAgB,mBAAmB;AAC9E,OAAK,MAAM,OAAO,KAAK,UACrB,cAAa,IAAI,YAAoB,SAAS,gBAAgB,mBAAmB;YAE1E,KAAK,SAAS,2BAA2B;EAClD,MAAM,QAAQ;EAEd,MAAM,eAAe;GAAE,GAAG;GAAS,sBAAsB;GAAO;AAChE,MAAI,MAAM,KAAK,SAAS,iBACtB,cAAa,MAAM,MAAc,cAAc,gBAAgB,mBAAmB;YAE3E,KAAK,SAAS,yBAAyB;EAGhD,MAAM,QAAQ;AACd,MAAI,MAAM,WACR,cAAa,MAAM,YAAoB,SAAS,gBAAgB,mBAAmB;YAE5E,KAAK,SAAS,mBAGvB,cADe,KACK,QAAgB,SAAS,gBAAgB,mBAAmB;UACvE,KAAK,SAAS,oBAAoB;EAC3C,MAAM,MAAM;AACZ,OAAK,MAAM,QAAQ,IAAI,WACrB,KAAI,KAAK,SAAS,gBAChB,cAAa,KAAK,WAAmB,SAAS,gBAAgB,mBAAmB;WACxE,KAAK,SAAS,mBACvB,cAAa,KAAK,OAAe,SAAS,gBAAgB,mBAAmB;OAKjF,MAAK,MAAM,SAAS,OAAO,OAAO,KAAK,CACrC,KAAI,MAAM,QAAQ,MAAM,EACtB;OAAK,MAAM,SAAS,MAClB,KAAI,SAAS,OAAO,UAAU,YAAY,UAAU,MAClD,cAAa,OAAe,SAAS,gBAAgB,mBAAmB;YAGnE,SAAS,OAAO,UAAU,YAAY,UAAU,MACzD,cAAa,OAAe,SAAS,gBAAgB,mBAAmB;;AAMhF,MAAM,YACJ,UACA,gBACA,uBACS;AACT,MAAK,MAAM,aAAaC,SAAO,KAC7B,cAAa,WAAW;EAAE,qBAAqB;EAAO,sBAAsB;EAAM,EAAE,gBAAgB,mBAAmB;;;;;;AAQ3H,MAAa,UAAU,YAA8D;CACnF,MAAM,EAAE,YAAY,aAAa;CAGjC,IAAIC;AACJ,KAAI;EACF,MAAM,oCAAoB,YAAY;GACpC,QAAQ;GACR,KAAK,UAAU,SAAS,OAAO,IAAI;GACnC,QAAQ;GACR,YAAY;GACZ,eAAe;GAChB,CAAC;AAEF,MAAI,QAAQ,SAAS,SACnB,4BAAW;GACT,MAAM;GACN,MAAM;GACN,SAAS,eAAe,WAAW,KAAK,SAAS,KAAK;GACvD,CAAC;AAEJ,aAAS;UACF,OAAO;AACd,6BAAW;GACT,MAAM;GACN,MAAM;GACN,SAAS,8BAA8B,WAAW,KAAK,SAAS,KAAK;GACrE;GACD,CAAC;;CAKJ,MAAM,aAAaD,SAAO,KAAK,MAAM,WAAW,SAAS;CAGzD,MAAM,iBAAiB,sBAAsBA,SAAO;AACpD,KAAI,eAAe,SAAS,EAC1B,2BAAU;EAAE,UAAU;EAAO;EAAY,CAAC;CAI5C,MAAME,kBAA4B,EAAE;AAEpC,UAASF,UAAQ,iBAAiB,QAAQ,YAAY;EAEpD,MAAM,cAAc,OAAO,KAAK,QAAQ;AAGxC,MAAI,mBAAmB,YAAY,YAAY,CAAE;AAGjD,kBAAgB,KAAK,cAAc,EAAE;GACrC;AAGF,KAAI,gBAAgB,WAAW,EAC7B,2BAAU;EAAE,UAAU;EAAO;EAAY,CAAC;CAK5C,MAAM,eAAe,CAAC,GAAG,gBAAgB,CAAC,MAAM,GAAG,MAAM,IAAI,EAAE;CAE/D,IAAI,SAAS;AACb,MAAK,MAAM,OAAO,aAChB,UAAS,OAAO,MAAM,GAAG,IAAI,GAAG,oBAAoB,OAAO,MAAM,IAAI;AAGvE,2BAAU;EAAE,UAAU;EAAM,YAAY;EAAQ,CAAC;;;;;;AAOnD,MAAa,eAAe,YAAyD;CACnF,MAAM,EAAE,YAAY,aAAa;CAGjC,IAAIC;AACJ,KAAI;EACF,MAAM,oCAAoB,YAAY;GACpC,QAAQ;GACR,KAAK,UAAU,SAAS,OAAO,IAAI;GACnC,QAAQ;GACR,YAAY;GACZ,eAAe;GAChB,CAAC;AAEF,MAAI,QAAQ,SAAS,SACnB,4BAAW;GACT,MAAM;GACN,MAAM;GACN,SAAS,eAAe,WAAW,KAAK,SAAS,KAAK;GACvD,CAAC;AAEJ,aAAS;UACF,OAAO;AACd,6BAAW;GACT,MAAM;GACN,MAAM;GACN,SAAS,8BAA8B,WAAW,KAAK,SAAS,KAAK;GACrE;GACD,CAAC;;CAGJ,MAAM,aAAaD,SAAO,KAAK,MAAM,WAAW,SAAS;CACzD,MAAM,iBAAiB,sBAAsBA,SAAO;AAEpD,KAAI,eAAe,SAAS,EAC1B,2BAAU,MAAM;CAGlB,IAAI,kBAAkB;AAEtB,UAASA,UAAQ,iBAAiB,QAAQ,YAAY;AACpD,MAAI,gBAAiB;AAGrB,MAAI,CAAC,mBAAmB,YADJ,OAAO,KAAK,QAAQ,WACQ,CAC9C,mBAAkB;GAEpB;AAEF,2BAAU,gBAAgB"}

package/dist/index.d.cts

@@ -0,0 +1,32 @@
+import { Result } from "neverthrow";
+
+//#region packages/formatter/src/types.d.ts
+type FormatOptions = {
+  readonly sourceCode: string;
+  readonly filePath?: string;
+};
+type FormatResult = {
+  readonly modified: boolean;
+  readonly sourceCode: string;
+};
+type FormatError = {
+  readonly type: "FormatError";
+  readonly code: "PARSE_ERROR" | "TRANSFORM_ERROR";
+  readonly message: string;
+  readonly cause?: unknown;
+};
+//#endregion
+//#region packages/formatter/src/format.d.ts
+/**
+ * Format soda-gql field selection objects by inserting newlines.
+ * This preserves multi-line formatting when using Biome/Prettier.
+ */
+declare const format: (options: FormatOptions) => Result<FormatResult, FormatError>;
+/**
+ * Check if a file needs formatting (has unformatted field selections).
+ * Useful for pre-commit hooks or CI checks.
+ */
+declare const needsFormat: (options: FormatOptions) => Result<boolean, FormatError>;
+//#endregion
+export { type FormatError, type FormatOptions, type FormatResult, format, needsFormat };
+//# sourceMappingURL=index.d.cts.map

package/dist/index.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.cts","names":[],"sources":["../src/types.ts","../src/format.ts"],"sourcesContent":[],"mappings":";;;KAAY,aAAA;;;AAAZ,CAAA;AAKY,KAAA,YAAA,GAAY;EAKZ,SAAA,QAAW,EAAA,OAAA;;;KAAX,WAAA;EC0FC,SAsEZ,IAAA,EAAA,aAAA;EAtE+B,SAAA,IAAA,EAAA,aAAA,GAAA,iBAAA;EAAuB,SAAA,OAAA,EAAA,MAAA;EAAc,SAAA,KAAA,CAAA,EAAA,OAAA;CAArB;;;;ADpGhD;AAKA;AAKA;cC0Fa,kBAAmB,kBAAgB,OAAO,cAAc;;;AAArE;;AAAuD,cA4E1C,WA5E0C,EAAA,CAAA,OAAA,EA4ElB,aA5EkB,EAAA,GA4EF,MA5EE,CAAA,OAAA,EA4Ec,WA5Ed,CAAA"}

package/dist/index.d.mts

@@ -0,0 +1,32 @@
+import { Result } from "neverthrow";
+
+//#region packages/formatter/src/types.d.ts
+type FormatOptions = {
+  readonly sourceCode: string;
+  readonly filePath?: string;
+};
+type FormatResult = {
+  readonly modified: boolean;
+  readonly sourceCode: string;
+};
+type FormatError = {
+  readonly type: "FormatError";
+  readonly code: "PARSE_ERROR" | "TRANSFORM_ERROR";
+  readonly message: string;
+  readonly cause?: unknown;
+};
+//#endregion
+//#region packages/formatter/src/format.d.ts
+/**
+ * Format soda-gql field selection objects by inserting newlines.
+ * This preserves multi-line formatting when using Biome/Prettier.
+ */
+declare const format: (options: FormatOptions) => Result<FormatResult, FormatError>;
+/**
+ * Check if a file needs formatting (has unformatted field selections).
+ * Useful for pre-commit hooks or CI checks.
+ */
+declare const needsFormat: (options: FormatOptions) => Result<boolean, FormatError>;
+//#endregion
+export { type FormatError, type FormatOptions, type FormatResult, format, needsFormat };
+//# sourceMappingURL=index.d.mts.map

package/dist/index.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.mts","names":[],"sources":["../src/types.ts","../src/format.ts"],"sourcesContent":[],"mappings":";;;KAAY,aAAA;;;AAAZ,CAAA;AAKY,KAAA,YAAA,GAAY;EAKZ,SAAA,QAAW,EAAA,OAAA;;;KAAX,WAAA;EC0FC,SAsEZ,IAAA,EAAA,aAAA;EAtE+B,SAAA,IAAA,EAAA,aAAA,GAAA,iBAAA;EAAuB,SAAA,OAAA,EAAA,MAAA;EAAc,SAAA,KAAA,CAAA,EAAA,OAAA;CAArB;;;;ADpGhD;AAKA;AAKA;cC0Fa,kBAAmB,kBAAgB,OAAO,cAAc;;;AAArE;;AAAuD,cA4E1C,WA5E0C,EAAA,CAAA,OAAA,EA4ElB,aA5EkB,EAAA,GA4EF,MA5EE,CAAA,OAAA,EA4Ec,WA5Ed,CAAA"}

package/dist/index.mjs

@@ -0,0 +1,220 @@
+import { parseSync } from "@swc/core";
+import { err, ok } from "neverthrow";
+
+//#region packages/formatter/src/detection.ts
+/**
+* Check if an expression is a reference to gql
+* Handles: gql, namespace.gql
+*/
+const isGqlReference = (node, gqlIdentifiers) => {
+	if (node.type === "Identifier") return gqlIdentifiers.has(node.value);
+	if (node.type === "MemberExpression" && node.property.type === "Identifier" && node.property.value === "gql") return true;
+	return false;
+};
+/**
+* Check if a call expression is a gql definition call
+* Handles: gql.default(...), gql.model(...), gql.schemaName(...) (multi-schema)
+*/
+const isGqlDefinitionCall = (node, gqlIdentifiers) => {
+	if (node.callee.type !== "MemberExpression") return false;
+	const { object } = node.callee;
+	if (!isGqlReference(object, gqlIdentifiers)) return false;
+	const firstArg = node.arguments[0];
+	if (!firstArg || firstArg.expression.type !== "ArrowFunctionExpression") return false;
+	return true;
+};
+/**
+* Check if an object expression is a field selection object
+* Field selection objects are returned from arrow functions with ({ f }) or ({ f, $ }) parameter
+*/
+const isFieldSelectionObject = (object, parent) => {
+	let bodyObject = null;
+	if (parent.body.type === "ObjectExpression") bodyObject = parent.body;
+	else if (parent.body.type === "ParenthesisExpression") {
+		const inner = parent.body.expression;
+		if (inner.type === "ObjectExpression") bodyObject = inner;
+	}
+	if (!bodyObject) return false;
+	if (bodyObject.span.start !== object.span.start) return false;
+	const param = parent.params[0];
+	if (!param || param.type !== "ObjectPattern") return false;
+	return param.properties.some((p) => {
+		if (p.type === "KeyValuePatternProperty" && p.key.type === "Identifier") return p.key.value === "f";
+		if (p.type === "AssignmentPatternProperty" && p.key.type === "Identifier") return p.key.value === "f";
+		return false;
+	});
+};
+/**
+* Collect gql identifiers from import declarations
+*/
+const collectGqlIdentifiers = (module) => {
+	const gqlIdentifiers = /* @__PURE__ */ new Set();
+	for (const item of module.body) {
+		if (item.type !== "ImportDeclaration") continue;
+		for (const specifier of item.specifiers) {
+			if (specifier.type === "ImportSpecifier") {
+				if ((specifier.imported?.value ?? specifier.local.value) === "gql") gqlIdentifiers.add(specifier.local.value);
+			}
+			if (specifier.type === "ImportDefaultSpecifier") {
+				if (specifier.local.value === "gql") gqlIdentifiers.add(specifier.local.value);
+			}
+			if (specifier.type === "ImportNamespaceSpecifier") {}
+		}
+	}
+	return gqlIdentifiers;
+};
+
+//#endregion
+//#region packages/formatter/src/insertion.ts
+/**
+* The newline string to insert after object opening brace
+*/
+const NEWLINE_INSERTION = "\n";
+/**
+* Check if there's already a newline after the opening brace.
+* Uses string inspection rather than AST.
+*
+* @param source - The source code string
+* @param objectStartPos - The position of the `{` character in the source
+*/
+const hasExistingNewline = (source, objectStartPos) => {
+	const nextChar = source[objectStartPos + 1];
+	return nextChar === "\n" || nextChar === "\r";
+};
+
+//#endregion
+//#region packages/formatter/src/format.ts
+/**
+* Simple recursive AST traversal
+*/
+const traverseNode = (node, context, gqlIdentifiers, onObjectExpression) => {
+	if (node.type === "CallExpression" && isGqlDefinitionCall(node, gqlIdentifiers)) context = {
+		...context,
+		insideGqlDefinition: true
+	};
+	if (node.type === "ObjectExpression" && context.insideGqlDefinition && context.currentArrowFunction && isFieldSelectionObject(node, context.currentArrowFunction)) onObjectExpression(node, context.currentArrowFunction);
+	if (node.type === "CallExpression") {
+		const call = node;
+		traverseNode(call.callee, context, gqlIdentifiers, onObjectExpression);
+		for (const arg of call.arguments) traverseNode(arg.expression, context, gqlIdentifiers, onObjectExpression);
+	} else if (node.type === "ArrowFunctionExpression") {
+		const arrow = node;
+		const childContext = {
+			...context,
+			currentArrowFunction: arrow
+		};
+		if (arrow.body.type !== "BlockStatement") traverseNode(arrow.body, childContext, gqlIdentifiers, onObjectExpression);
+	} else if (node.type === "ParenthesisExpression") {
+		const paren = node;
+		if (paren.expression) traverseNode(paren.expression, context, gqlIdentifiers, onObjectExpression);
+	} else if (node.type === "MemberExpression") traverseNode(node.object, context, gqlIdentifiers, onObjectExpression);
+	else if (node.type === "ObjectExpression") {
+		const obj = node;
+		for (const prop of obj.properties) if (prop.type === "SpreadElement") traverseNode(prop.arguments, context, gqlIdentifiers, onObjectExpression);
+		else if (prop.type === "KeyValueProperty") traverseNode(prop.value, context, gqlIdentifiers, onObjectExpression);
+	} else for (const value of Object.values(node)) if (Array.isArray(value)) {
+		for (const child of value) if (child && typeof child === "object" && "type" in child) traverseNode(child, context, gqlIdentifiers, onObjectExpression);
+	} else if (value && typeof value === "object" && "type" in value) traverseNode(value, context, gqlIdentifiers, onObjectExpression);
+};
+const traverse = (module, gqlIdentifiers, onObjectExpression) => {
+	for (const statement of module.body) traverseNode(statement, {
+		insideGqlDefinition: false,
+		currentArrowFunction: null
+	}, gqlIdentifiers, onObjectExpression);
+};
+/**
+* Format soda-gql field selection objects by inserting newlines.
+* This preserves multi-line formatting when using Biome/Prettier.
+*/
+const format = (options) => {
+	const { sourceCode, filePath } = options;
+	let module;
+	try {
+		const program = parseSync(sourceCode, {
+			syntax: "typescript",
+			tsx: filePath?.endsWith(".tsx") ?? true,
+			target: "es2022",
+			decorators: false,
+			dynamicImport: true
+		});
+		if (program.type !== "Module") return err({
+			type: "FormatError",
+			code: "PARSE_ERROR",
+			message: `Not a module${filePath ? ` (${filePath})` : ""}`
+		});
+		module = program;
+	} catch (cause) {
+		return err({
+			type: "FormatError",
+			code: "PARSE_ERROR",
+			message: `Failed to parse source code${filePath ? ` (${filePath})` : ""}`,
+			cause
+		});
+	}
+	const spanOffset = module.span.end - sourceCode.length + 1;
+	const gqlIdentifiers = collectGqlIdentifiers(module);
+	if (gqlIdentifiers.size === 0) return ok({
+		modified: false,
+		sourceCode
+	});
+	const insertionPoints = [];
+	traverse(module, gqlIdentifiers, (object, _parent) => {
+		const objectStart = object.span.start - spanOffset;
+		if (hasExistingNewline(sourceCode, objectStart)) return;
+		insertionPoints.push(objectStart + 1);
+	});
+	if (insertionPoints.length === 0) return ok({
+		modified: false,
+		sourceCode
+	});
+	const sortedPoints = [...insertionPoints].sort((a, b) => b - a);
+	let result = sourceCode;
+	for (const pos of sortedPoints) result = result.slice(0, pos) + NEWLINE_INSERTION + result.slice(pos);
+	return ok({
+		modified: true,
+		sourceCode: result
+	});
+};
+/**
+* Check if a file needs formatting (has unformatted field selections).
+* Useful for pre-commit hooks or CI checks.
+*/
+const needsFormat = (options) => {
+	const { sourceCode, filePath } = options;
+	let module;
+	try {
+		const program = parseSync(sourceCode, {
+			syntax: "typescript",
+			tsx: filePath?.endsWith(".tsx") ?? true,
+			target: "es2022",
+			decorators: false,
+			dynamicImport: true
+		});
+		if (program.type !== "Module") return err({
+			type: "FormatError",
+			code: "PARSE_ERROR",
+			message: `Not a module${filePath ? ` (${filePath})` : ""}`
+		});
+		module = program;
+	} catch (cause) {
+		return err({
+			type: "FormatError",
+			code: "PARSE_ERROR",
+			message: `Failed to parse source code${filePath ? ` (${filePath})` : ""}`,
+			cause
+		});
+	}
+	const spanOffset = module.span.end - sourceCode.length + 1;
+	const gqlIdentifiers = collectGqlIdentifiers(module);
+	if (gqlIdentifiers.size === 0) return ok(false);
+	let needsFormatting = false;
+	traverse(module, gqlIdentifiers, (object, _parent) => {
+		if (needsFormatting) return;
+		if (!hasExistingNewline(sourceCode, object.span.start - spanOffset)) needsFormatting = true;
+	});
+	return ok(needsFormatting);
+};
+
+//#endregion
+export { format, needsFormat };
+//# sourceMappingURL=index.mjs.map

package/dist/index.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.mjs","names":["bodyObject: ObjectExpression | null","module: Module","insertionPoints: number[]"],"sources":["../src/detection.ts","../src/insertion.ts","../src/format.ts"],"sourcesContent":["import type {\n  ArrowFunctionExpression,\n  CallExpression,\n  Expression,\n  Module,\n  ObjectExpression,\n  ObjectPatternProperty,\n} from \"@swc/types\";\n\n/**\n * Check if an expression is a reference to gql\n * Handles: gql, namespace.gql\n */\nexport const isGqlReference = (node: Expression, gqlIdentifiers: ReadonlySet<string>): boolean => {\n  if (node.type === \"Identifier\") {\n    return gqlIdentifiers.has(node.value);\n  }\n  if (node.type === \"MemberExpression\" && node.property.type === \"Identifier\" && node.property.value === \"gql\") {\n    return true;\n  }\n  return false;\n};\n\n/**\n * Check if a call expression is a gql definition call\n * Handles: gql.default(...), gql.model(...), gql.schemaName(...) (multi-schema)\n */\nexport const isGqlDefinitionCall = (node: CallExpression, gqlIdentifiers: ReadonlySet<string>): boolean => {\n  if (node.callee.type !== \"MemberExpression\") return false;\n  const { object } = node.callee;\n\n  // Check object is a gql reference first\n  if (!isGqlReference(object, gqlIdentifiers)) return false;\n\n  // Verify first argument is an arrow function (factory pattern)\n  // SWC's arguments are ExprOrSpread[], so access via .expression\n  const firstArg = node.arguments[0];\n  if (!firstArg || firstArg.expression.type !== \"ArrowFunctionExpression\") return false;\n\n  return true;\n};\n\n/**\n * Check if an object expression is a field selection object\n * Field selection objects are returned from arrow functions with ({ f }) or ({ f, $ }) parameter\n */\nexport const isFieldSelectionObject = (object: ObjectExpression, parent: ArrowFunctionExpression): boolean => {\n  // The object must be the body of the arrow function\n  // Handle both direct ObjectExpression and parenthesized ObjectExpression: `({ ... })`\n  let bodyObject: ObjectExpression | null = null;\n\n  if (parent.body.type === \"ObjectExpression\") {\n    bodyObject = parent.body;\n  } else if (parent.body.type === \"ParenthesisExpression\") {\n    // Handle `({ f }) => ({ ...f.id() })` pattern where body is parenthesized\n    const inner = parent.body.expression;\n    if (inner.type === \"ObjectExpression\") {\n      bodyObject = inner;\n    }\n  }\n\n  if (!bodyObject) return false;\n  if (bodyObject.span.start !== object.span.start) return false;\n\n  // Check if first parameter has 'f' destructured\n  const param = parent.params[0];\n  if (!param || param.type !== \"ObjectPattern\") return false;\n\n  return param.properties.some((p: ObjectPatternProperty) => {\n    if (p.type === \"KeyValuePatternProperty\" && p.key.type === \"Identifier\") {\n      return p.key.value === \"f\";\n    }\n    if (p.type === \"AssignmentPatternProperty\" && p.key.type === \"Identifier\") {\n      return p.key.value === \"f\";\n    }\n    return false;\n  });\n};\n\n/**\n * Collect gql identifiers from import declarations\n */\nexport const collectGqlIdentifiers = (module: Module): Set<string> => {\n  const gqlIdentifiers = new Set<string>();\n\n  for (const item of module.body) {\n    if (item.type !== \"ImportDeclaration\") continue;\n\n    for (const specifier of item.specifiers) {\n      if (specifier.type === \"ImportSpecifier\") {\n        const imported = specifier.imported?.value ?? specifier.local.value;\n        if (imported === \"gql\") {\n          gqlIdentifiers.add(specifier.local.value);\n        }\n      }\n      if (specifier.type === \"ImportDefaultSpecifier\") {\n        // Check if default import might be gql\n        if (specifier.local.value === \"gql\") {\n          gqlIdentifiers.add(specifier.local.value);\n        }\n      }\n      if (specifier.type === \"ImportNamespaceSpecifier\") {\n        // namespace import: import * as ns from \"...\"\n        // Would need ns.gql pattern - handled by isGqlReference\n      }\n    }\n  }\n\n  return gqlIdentifiers;\n};\n","/**\n * The newline string to insert after object opening brace\n */\nexport const NEWLINE_INSERTION = \"\\n\";\n\n/**\n * Check if there's already a newline after the opening brace.\n * Uses string inspection rather than AST.\n *\n * @param source - The source code string\n * @param objectStartPos - The position of the `{` character in the source\n */\nexport const hasExistingNewline = (source: string, objectStartPos: number): boolean => {\n  // Skip the `{` character\n  const pos = objectStartPos + 1;\n\n  // Check if next character is a newline\n  const nextChar = source[pos];\n  return nextChar === \"\\n\" || nextChar === \"\\r\";\n};\n","import { parseSync } from \"@swc/core\";\nimport type { ArrowFunctionExpression, CallExpression, Module, Node, ObjectExpression } from \"@swc/types\";\nimport { err, ok, type Result } from \"neverthrow\";\nimport { collectGqlIdentifiers, isFieldSelectionObject, isGqlDefinitionCall } from \"./detection\";\nimport { hasExistingNewline, NEWLINE_INSERTION } from \"./insertion\";\nimport type { FormatError, FormatOptions, FormatResult } from \"./types\";\n\ntype TraversalContext = {\n  insideGqlDefinition: boolean;\n  currentArrowFunction: ArrowFunctionExpression | null;\n};\n\n/**\n * Simple recursive AST traversal\n */\nconst traverseNode = (\n  node: Node,\n  context: TraversalContext,\n  gqlIdentifiers: ReadonlySet<string>,\n  onObjectExpression: (object: ObjectExpression, parent: ArrowFunctionExpression) => void,\n): void => {\n  // Check for gql definition call entry\n  if (node.type === \"CallExpression\" && isGqlDefinitionCall(node as CallExpression, gqlIdentifiers)) {\n    context = { ...context, insideGqlDefinition: true };\n  }\n\n  // Handle object expressions - check if it's the body of the current arrow function\n  if (\n    node.type === \"ObjectExpression\" &&\n    context.insideGqlDefinition &&\n    context.currentArrowFunction &&\n    isFieldSelectionObject(node as ObjectExpression, context.currentArrowFunction)\n  ) {\n    onObjectExpression(node as ObjectExpression, context.currentArrowFunction);\n  }\n\n  // Recursively visit children\n  if (node.type === \"CallExpression\") {\n    const call = node as CallExpression;\n    traverseNode(call.callee as Node, context, gqlIdentifiers, onObjectExpression);\n    for (const arg of call.arguments) {\n      traverseNode(arg.expression as Node, context, gqlIdentifiers, onObjectExpression);\n    }\n  } else if (node.type === \"ArrowFunctionExpression\") {\n    const arrow = node as ArrowFunctionExpression;\n    // Update context with the new arrow function\n    const childContext = { ...context, currentArrowFunction: arrow };\n    if (arrow.body.type !== \"BlockStatement\") {\n      traverseNode(arrow.body as Node, childContext, gqlIdentifiers, onObjectExpression);\n    }\n  } else if (node.type === \"ParenthesisExpression\") {\n    // Handle parenthesized expressions like `({ ...f.id() })`\n    // biome-ignore lint/suspicious/noExplicitAny: SWC types\n    const paren = node as any;\n    if (paren.expression) {\n      traverseNode(paren.expression as Node, context, gqlIdentifiers, onObjectExpression);\n    }\n  } else if (node.type === \"MemberExpression\") {\n    // biome-ignore lint/suspicious/noExplicitAny: SWC types\n    const member = node as any;\n    traverseNode(member.object as Node, context, gqlIdentifiers, onObjectExpression);\n  } else if (node.type === \"ObjectExpression\") {\n    const obj = node as ObjectExpression;\n    for (const prop of obj.properties) {\n      if (prop.type === \"SpreadElement\") {\n        traverseNode(prop.arguments as Node, context, gqlIdentifiers, onObjectExpression);\n      } else if (prop.type === \"KeyValueProperty\") {\n        traverseNode(prop.value as Node, context, gqlIdentifiers, onObjectExpression);\n      }\n    }\n  } else {\n    // Generic traversal for other node types\n    for (const value of Object.values(node)) {\n      if (Array.isArray(value)) {\n        for (const child of value) {\n          if (child && typeof child === \"object\" && \"type\" in child) {\n            traverseNode(child as Node, context, gqlIdentifiers, onObjectExpression);\n          }\n        }\n      } else if (value && typeof value === \"object\" && \"type\" in value) {\n        traverseNode(value as Node, context, gqlIdentifiers, onObjectExpression);\n      }\n    }\n  }\n};\n\nconst traverse = (\n  module: Module,\n  gqlIdentifiers: ReadonlySet<string>,\n  onObjectExpression: (object: ObjectExpression, parent: ArrowFunctionExpression) => void,\n): void => {\n  for (const statement of module.body) {\n    traverseNode(statement, { insideGqlDefinition: false, currentArrowFunction: null }, gqlIdentifiers, onObjectExpression);\n  }\n};\n\n/**\n * Format soda-gql field selection objects by inserting newlines.\n * This preserves multi-line formatting when using Biome/Prettier.\n */\nexport const format = (options: FormatOptions): Result<FormatResult, FormatError> => {\n  const { sourceCode, filePath } = options;\n\n  // Parse source code with SWC\n  let module: Module;\n  try {\n    const program = parseSync(sourceCode, {\n      syntax: \"typescript\",\n      tsx: filePath?.endsWith(\".tsx\") ?? true,\n      target: \"es2022\",\n      decorators: false,\n      dynamicImport: true,\n    });\n\n    if (program.type !== \"Module\") {\n      return err({\n        type: \"FormatError\",\n        code: \"PARSE_ERROR\",\n        message: `Not a module${filePath ? ` (${filePath})` : \"\"}`,\n      });\n    }\n    module = program;\n  } catch (cause) {\n    return err({\n      type: \"FormatError\",\n      code: \"PARSE_ERROR\",\n      message: `Failed to parse source code${filePath ? ` (${filePath})` : \"\"}`,\n      cause,\n    });\n  }\n\n  // Calculate span offset for position normalization\n  // SWC's BytePos counter accumulates across parseSync calls within the same process\n  const spanOffset = module.span.end - sourceCode.length + 1;\n\n  // Collect gql identifiers from imports\n  const gqlIdentifiers = collectGqlIdentifiers(module);\n  if (gqlIdentifiers.size === 0) {\n    return ok({ modified: false, sourceCode });\n  }\n\n  // Collect insertion points\n  const insertionPoints: number[] = [];\n\n  traverse(module, gqlIdentifiers, (object, _parent) => {\n    // Calculate actual position in source\n    const objectStart = object.span.start - spanOffset;\n\n    // Check if already has newline\n    if (hasExistingNewline(sourceCode, objectStart)) return;\n\n    // Record insertion point (position after `{`)\n    insertionPoints.push(objectStart + 1);\n  });\n\n  // Apply insertions\n  if (insertionPoints.length === 0) {\n    return ok({ modified: false, sourceCode });\n  }\n\n  // Sort in descending order to insert from end to beginning\n  // This preserves earlier positions while modifying later parts\n  const sortedPoints = [...insertionPoints].sort((a, b) => b - a);\n\n  let result = sourceCode;\n  for (const pos of sortedPoints) {\n    result = result.slice(0, pos) + NEWLINE_INSERTION + result.slice(pos);\n  }\n\n  return ok({ modified: true, sourceCode: result });\n};\n\n/**\n * Check if a file needs formatting (has unformatted field selections).\n * Useful for pre-commit hooks or CI checks.\n */\nexport const needsFormat = (options: FormatOptions): Result<boolean, FormatError> => {\n  const { sourceCode, filePath } = options;\n\n  // Parse source code with SWC\n  let module: Module;\n  try {\n    const program = parseSync(sourceCode, {\n      syntax: \"typescript\",\n      tsx: filePath?.endsWith(\".tsx\") ?? true,\n      target: \"es2022\",\n      decorators: false,\n      dynamicImport: true,\n    });\n\n    if (program.type !== \"Module\") {\n      return err({\n        type: \"FormatError\",\n        code: \"PARSE_ERROR\",\n        message: `Not a module${filePath ? ` (${filePath})` : \"\"}`,\n      });\n    }\n    module = program;\n  } catch (cause) {\n    return err({\n      type: \"FormatError\",\n      code: \"PARSE_ERROR\",\n      message: `Failed to parse source code${filePath ? ` (${filePath})` : \"\"}`,\n      cause,\n    });\n  }\n\n  const spanOffset = module.span.end - sourceCode.length + 1;\n  const gqlIdentifiers = collectGqlIdentifiers(module);\n\n  if (gqlIdentifiers.size === 0) {\n    return ok(false);\n  }\n\n  let needsFormatting = false;\n\n  traverse(module, gqlIdentifiers, (object, _parent) => {\n    if (needsFormatting) return; // Early exit\n\n    const objectStart = object.span.start - spanOffset;\n    if (!hasExistingNewline(sourceCode, objectStart)) {\n      needsFormatting = true;\n    }\n  });\n\n  return ok(needsFormatting);\n};\n"],"mappings":";;;;;;;;AAaA,MAAa,kBAAkB,MAAkB,mBAAiD;AAChG,KAAI,KAAK,SAAS,aAChB,QAAO,eAAe,IAAI,KAAK,MAAM;AAEvC,KAAI,KAAK,SAAS,sBAAsB,KAAK,SAAS,SAAS,gBAAgB,KAAK,SAAS,UAAU,MACrG,QAAO;AAET,QAAO;;;;;;AAOT,MAAa,uBAAuB,MAAsB,mBAAiD;AACzG,KAAI,KAAK,OAAO,SAAS,mBAAoB,QAAO;CACpD,MAAM,EAAE,WAAW,KAAK;AAGxB,KAAI,CAAC,eAAe,QAAQ,eAAe,CAAE,QAAO;CAIpD,MAAM,WAAW,KAAK,UAAU;AAChC,KAAI,CAAC,YAAY,SAAS,WAAW,SAAS,0BAA2B,QAAO;AAEhF,QAAO;;;;;;AAOT,MAAa,0BAA0B,QAA0B,WAA6C;CAG5G,IAAIA,aAAsC;AAE1C,KAAI,OAAO,KAAK,SAAS,mBACvB,cAAa,OAAO;UACX,OAAO,KAAK,SAAS,yBAAyB;EAEvD,MAAM,QAAQ,OAAO,KAAK;AAC1B,MAAI,MAAM,SAAS,mBACjB,cAAa;;AAIjB,KAAI,CAAC,WAAY,QAAO;AACxB,KAAI,WAAW,KAAK,UAAU,OAAO,KAAK,MAAO,QAAO;CAGxD,MAAM,QAAQ,OAAO,OAAO;AAC5B,KAAI,CAAC,SAAS,MAAM,SAAS,gBAAiB,QAAO;AAErD,QAAO,MAAM,WAAW,MAAM,MAA6B;AACzD,MAAI,EAAE,SAAS,6BAA6B,EAAE,IAAI,SAAS,aACzD,QAAO,EAAE,IAAI,UAAU;AAEzB,MAAI,EAAE,SAAS,+BAA+B,EAAE,IAAI,SAAS,aAC3D,QAAO,EAAE,IAAI,UAAU;AAEzB,SAAO;GACP;;;;;AAMJ,MAAa,yBAAyB,WAAgC;CACpE,MAAM,iCAAiB,IAAI,KAAa;AAExC,MAAK,MAAM,QAAQ,OAAO,MAAM;AAC9B,MAAI,KAAK,SAAS,oBAAqB;AAEvC,OAAK,MAAM,aAAa,KAAK,YAAY;AACvC,OAAI,UAAU,SAAS,mBAErB;SADiB,UAAU,UAAU,SAAS,UAAU,MAAM,WAC7C,MACf,gBAAe,IAAI,UAAU,MAAM,MAAM;;AAG7C,OAAI,UAAU,SAAS,0BAErB;QAAI,UAAU,MAAM,UAAU,MAC5B,gBAAe,IAAI,UAAU,MAAM,MAAM;;AAG7C,OAAI,UAAU,SAAS,4BAA4B;;;AAOvD,QAAO;;;;;;;;ACzGT,MAAa,oBAAoB;;;;;;;;AASjC,MAAa,sBAAsB,QAAgB,mBAAoC;CAKrF,MAAM,WAAW,OAHL,iBAAiB;AAI7B,QAAO,aAAa,QAAQ,aAAa;;;;;;;;ACH3C,MAAM,gBACJ,MACA,SACA,gBACA,uBACS;AAET,KAAI,KAAK,SAAS,oBAAoB,oBAAoB,MAAwB,eAAe,CAC/F,WAAU;EAAE,GAAG;EAAS,qBAAqB;EAAM;AAIrD,KACE,KAAK,SAAS,sBACd,QAAQ,uBACR,QAAQ,wBACR,uBAAuB,MAA0B,QAAQ,qBAAqB,CAE9E,oBAAmB,MAA0B,QAAQ,qBAAqB;AAI5E,KAAI,KAAK,SAAS,kBAAkB;EAClC,MAAM,OAAO;AACb,eAAa,KAAK,QAAgB,SAAS,gBAAgB,mBAAmB;AAC9E,OAAK,MAAM,OAAO,KAAK,UACrB,cAAa,IAAI,YAAoB,SAAS,gBAAgB,mBAAmB;YAE1E,KAAK,SAAS,2BAA2B;EAClD,MAAM,QAAQ;EAEd,MAAM,eAAe;GAAE,GAAG;GAAS,sBAAsB;GAAO;AAChE,MAAI,MAAM,KAAK,SAAS,iBACtB,cAAa,MAAM,MAAc,cAAc,gBAAgB,mBAAmB;YAE3E,KAAK,SAAS,yBAAyB;EAGhD,MAAM,QAAQ;AACd,MAAI,MAAM,WACR,cAAa,MAAM,YAAoB,SAAS,gBAAgB,mBAAmB;YAE5E,KAAK,SAAS,mBAGvB,cADe,KACK,QAAgB,SAAS,gBAAgB,mBAAmB;UACvE,KAAK,SAAS,oBAAoB;EAC3C,MAAM,MAAM;AACZ,OAAK,MAAM,QAAQ,IAAI,WACrB,KAAI,KAAK,SAAS,gBAChB,cAAa,KAAK,WAAmB,SAAS,gBAAgB,mBAAmB;WACxE,KAAK,SAAS,mBACvB,cAAa,KAAK,OAAe,SAAS,gBAAgB,mBAAmB;OAKjF,MAAK,MAAM,SAAS,OAAO,OAAO,KAAK,CACrC,KAAI,MAAM,QAAQ,MAAM,EACtB;OAAK,MAAM,SAAS,MAClB,KAAI,SAAS,OAAO,UAAU,YAAY,UAAU,MAClD,cAAa,OAAe,SAAS,gBAAgB,mBAAmB;YAGnE,SAAS,OAAO,UAAU,YAAY,UAAU,MACzD,cAAa,OAAe,SAAS,gBAAgB,mBAAmB;;AAMhF,MAAM,YACJ,QACA,gBACA,uBACS;AACT,MAAK,MAAM,aAAa,OAAO,KAC7B,cAAa,WAAW;EAAE,qBAAqB;EAAO,sBAAsB;EAAM,EAAE,gBAAgB,mBAAmB;;;;;;AAQ3H,MAAa,UAAU,YAA8D;CACnF,MAAM,EAAE,YAAY,aAAa;CAGjC,IAAIC;AACJ,KAAI;EACF,MAAM,UAAU,UAAU,YAAY;GACpC,QAAQ;GACR,KAAK,UAAU,SAAS,OAAO,IAAI;GACnC,QAAQ;GACR,YAAY;GACZ,eAAe;GAChB,CAAC;AAEF,MAAI,QAAQ,SAAS,SACnB,QAAO,IAAI;GACT,MAAM;GACN,MAAM;GACN,SAAS,eAAe,WAAW,KAAK,SAAS,KAAK;GACvD,CAAC;AAEJ,WAAS;UACF,OAAO;AACd,SAAO,IAAI;GACT,MAAM;GACN,MAAM;GACN,SAAS,8BAA8B,WAAW,KAAK,SAAS,KAAK;GACrE;GACD,CAAC;;CAKJ,MAAM,aAAa,OAAO,KAAK,MAAM,WAAW,SAAS;CAGzD,MAAM,iBAAiB,sBAAsB,OAAO;AACpD,KAAI,eAAe,SAAS,EAC1B,QAAO,GAAG;EAAE,UAAU;EAAO;EAAY,CAAC;CAI5C,MAAMC,kBAA4B,EAAE;AAEpC,UAAS,QAAQ,iBAAiB,QAAQ,YAAY;EAEpD,MAAM,cAAc,OAAO,KAAK,QAAQ;AAGxC,MAAI,mBAAmB,YAAY,YAAY,CAAE;AAGjD,kBAAgB,KAAK,cAAc,EAAE;GACrC;AAGF,KAAI,gBAAgB,WAAW,EAC7B,QAAO,GAAG;EAAE,UAAU;EAAO;EAAY,CAAC;CAK5C,MAAM,eAAe,CAAC,GAAG,gBAAgB,CAAC,MAAM,GAAG,MAAM,IAAI,EAAE;CAE/D,IAAI,SAAS;AACb,MAAK,MAAM,OAAO,aAChB,UAAS,OAAO,MAAM,GAAG,IAAI,GAAG,oBAAoB,OAAO,MAAM,IAAI;AAGvE,QAAO,GAAG;EAAE,UAAU;EAAM,YAAY;EAAQ,CAAC;;;;;;AAOnD,MAAa,eAAe,YAAyD;CACnF,MAAM,EAAE,YAAY,aAAa;CAGjC,IAAID;AACJ,KAAI;EACF,MAAM,UAAU,UAAU,YAAY;GACpC,QAAQ;GACR,KAAK,UAAU,SAAS,OAAO,IAAI;GACnC,QAAQ;GACR,YAAY;GACZ,eAAe;GAChB,CAAC;AAEF,MAAI,QAAQ,SAAS,SACnB,QAAO,IAAI;GACT,MAAM;GACN,MAAM;GACN,SAAS,eAAe,WAAW,KAAK,SAAS,KAAK;GACvD,CAAC;AAEJ,WAAS;UACF,OAAO;AACd,SAAO,IAAI;GACT,MAAM;GACN,MAAM;GACN,SAAS,8BAA8B,WAAW,KAAK,SAAS,KAAK;GACrE;GACD,CAAC;;CAGJ,MAAM,aAAa,OAAO,KAAK,MAAM,WAAW,SAAS;CACzD,MAAM,iBAAiB,sBAAsB,OAAO;AAEpD,KAAI,eAAe,SAAS,EAC1B,QAAO,GAAG,MAAM;CAGlB,IAAI,kBAAkB;AAEtB,UAAS,QAAQ,iBAAiB,QAAQ,YAAY;AACpD,MAAI,gBAAiB;AAGrB,MAAI,CAAC,mBAAmB,YADJ,OAAO,KAAK,QAAQ,WACQ,CAC9C,mBAAkB;GAEpB;AAEF,QAAO,GAAG,gBAAgB"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@soda-gql/formatter",
-  "version": "0.2.0",
+  "version": "0.4.0",
   "description": "GraphQL document formatter for soda-gql",
   "type": "module",
   "private": false,