import { describe, expect, it } from "vitest";

import { resolveSubagentCapabilities } from "./subagent-capabilities.js";

describe("resolveSubagentCapabilities", () => {

it("returns normalized depth for non-finite direct inputs", () => {

expect(

resolveSubagentCapabilities({

depth: Number.NaN,

maxSpawnDepth: Number.POSITIVE_INFINITY,

}),

).toEqual({

depth: 0,

role: "main",

controlScope: "children",

canSpawn: true,

canControlChildren: true,

});

});

it("floors finite depth and max depth consistently", () => {

expect(resolveSubagentCapabilities({ depth: 1.9, maxSpawnDepth: 1.2 })).toEqual({

depth: 1,

role: "leaf",

controlScope: "none",

canSpawn: false,

canControlChildren: false,

});

});

});

isSubagentSessionKey,

parseAgentSessionKey,

} from "../routing/session-key.js";

import {

resolveIntegerOption,

resolveNonNegativeIntegerOption,

} from "../shared/number-coercion.js";

import { normalizeOptionalLowercaseString } from "../shared/string-coerce.js";

import {

normalizeInheritedToolAllowlist,

depth: number;

maxSpawnDepth?: number;

}): SubagentSessionRole {

const depth = Number.isInteger(params.depth) ? Math.max(0, params.depth) : 0;

const maxSpawnDepth =

typeof params.maxSpawnDepth === "number" && Number.isFinite(params.maxSpawnDepth)

? Math.max(1, Math.floor(params.maxSpawnDepth))

: DEFAULT_SUBAGENT_MAX_SPAWN_DEPTH;

const depth = resolveNonNegativeIntegerOption(params.depth, 0);

const maxSpawnDepth = resolveIntegerOption(

params.maxSpawnDepth,

DEFAULT_SUBAGENT_MAX_SPAWN_DEPTH,

{ min: 1 },

);

if (depth <= 0) {

return "main";

}

}

export function resolveSubagentCapabilities(params: { depth: number; maxSpawnDepth?: number }) {

const depth = resolveNonNegativeIntegerOption(params.depth, 0);

const role = resolveSubagentRoleForDepth(params);

const controlScope = resolveSubagentControlScopeForRole(role);

return {

depth: Math.max(0, Math.floor(params.depth)),

depth,

role,

controlScope,

canSpawn: role === "main" || role === "orchestrator",