Function constructFunctionDefinitionChunk

core/indexing/chunk/code.ts:125–171

(
  node: SyntaxNode,
  code: string,
  maxChunkSize: number,
)

}

async function constructFunctionDefinitionChunk(
  node: SyntaxNode,
  code: string,
  maxChunkSize: number,
): Promise<string> {
  const bodyNode = node.children[node.children.length - 1];
  const collapsedBody = collapsedReplacement(bodyNode);
  const signature = code.slice(node.startIndex, bodyNode.startIndex);
  const funcText = signature + collapsedBody;

  const isInClass =
    node.parent &&
    ["block", "declaration_list"].includes(node.parent.type) &&
    node.parent.parent &&
    ["class_definition", "impl_item"].includes(node.parent.parent.type);

  if (isInClass) {
    const classNode = node.parent!.parent!;
    const classBlock = node.parent!;
    const classHeader = code.slice(classNode.startIndex, classBlock.startIndex);
    const indent = " ".repeat(node.startPosition.column);
    const combined = `${classHeader}...\n\n${indent}${funcText}`;

    if ((await countTokensAsync(combined)) <= maxChunkSize) {
      return combined;
    }
    if ((await countTokensAsync(funcText)) <= maxChunkSize) {
      return funcText;
    }
    const firstLine = signature.split("\n")[0] ?? "";
    const minimal = `${firstLine} ${collapsedBody}`;
    if ((await countTokensAsync(minimal)) <= maxChunkSize) {
      return minimal;
    }
    return collapsedBody;
  }

  if ((await countTokensAsync(funcText)) <= maxChunkSize) {
    return funcText;
  }
  const firstLine = signature.split("\n")[0] ?? "";
  const minimal = `${firstLine} ${collapsedBody}`;
  if ((await countTokensAsync(minimal)) <= maxChunkSize) {
    return minimal;
  }
  return collapsedBody;
}

const collapsedNodeConstructors: {
  [key: string]: (