pi-model-profiles 0.3.1 → 0.3.3

package/src/index.ts

@@ -1,158 +1,36 @@
-import type { ExtensionAPI, ExtensionCommandContext, ExtensionContext } from "@earendil-works/pi-coding-agent";
-
-import { COMMAND_NAME, INITIAL_PROFILE_NAME, PROFILE_NAME_SUFFIX, PROFILE_STORE_PATH } from "./constants.js";
-import { applySavedProfile, captureAgentSnapshots, detectActiveAgentName } from "./agent-writer.js";
-import { toErrorMessage } from "./errors.js";
-import { loadAndPrepareProfiles } from "./import-service.js";
-import { updateProfileAndReturn } from "./profile-update-service.js";
-import { appendProfile, createProfile, findProfileById, renameStoredProfile, resolveUniqueProfileName, saveProfilesFile } from "./profile-store.js";
-import { removeProfileAndUpdate } from "./profile-removal-service.js";
-import { openProfilesModal, type ProfileModalResult } from "./profile-modal.js";
-import { onResourcesDiscover, refreshModelRegistry } from "./pi-api-utils.js";
-import type { AppliedProfileOutcome, ProfilesFile } from "./types.js";
-
-function buildCurrentProfileName(activeAgentName: string | null, data: ProfilesFile): string {
-	const baseName = activeAgentName ? `${activeAgentName} ${PROFILE_NAME_SUFFIX}` : INITIAL_PROFILE_NAME;
-	return resolveUniqueProfileName(baseName, data.profiles);
-}
-
-function notifyWarnings(ctx: ExtensionCommandContext, warnings: readonly string[]): void {
-	if (warnings.length === 0) {
-		return;
-	}
-	ctx.ui.notify(warnings.join(" "), "warning");
-}
-
-function summarizeApplyOutcome(outcome: AppliedProfileOutcome): string {
-	const appliedCount = outcome.appliedAgents.length;
-	const missingCount = outcome.missingAgents.length;
-	const appliedLabel = `${appliedCount} agent file${appliedCount === 1 ? "" : "s"}`;
-	if (missingCount === 0) {
-		return appliedLabel;
-	}
-	return `${appliedLabel}; ${missingCount} missing`;
-}
-
-async function reloadAfterApply(ctx: ExtensionCommandContext, outcome: AppliedProfileOutcome): Promise<void> {
-	const summary = summarizeApplyOutcome(outcome);
-	ctx.ui.notify(`Profile '${outcome.profileName}' applied across ${summary}. Reloading…`, "info");
-
-	try {
-		await ctx.reload();
-	} catch (error) {
-		ctx.ui.notify(
-			`Profile '${outcome.profileName}' applied across ${summary}, but automatic reload failed: ${toErrorMessage(error)}. Run /reload.`,
-			"error",
-		);
-	}
-}
-
-async function handleModelProfilesCommand(ctx: ExtensionCommandContext): Promise<void> {
-	const agentOptions = { cwd: ctx.cwd, scope: "both" as const };
-	const prepared = loadAndPrepareProfiles(PROFILE_STORE_PATH, agentOptions);
-	notifyWarnings(ctx, prepared.warnings);
-
-	let data = prepared.data;
-	const activeAgentName = detectActiveAgentName(ctx.sessionManager, ctx.getSystemPrompt());
-
-	const result: ProfileModalResult = await openProfilesModal(ctx, data, activeAgentName, {
-		renameProfile: async (profileId, nextName) => {
-			data = renameStoredProfile(data, profileId, nextName);
-			saveProfilesFile(data, PROFILE_STORE_PATH);
-			return {
-				data,
-				message: `Renamed saved snapshot to '${nextName.trim()}'.`,
-				selectedProfileId: profileId,
-			};
-		},
-		addCurrentProfile: async () => {
-			const snapshot = captureAgentSnapshots(agentOptions);
-			notifyWarnings(ctx, snapshot.warnings);
-			const profile = createProfile(buildCurrentProfileName(activeAgentName, data), snapshot.agents);
-			data = appendProfile(data, profile);
-			saveProfilesFile(data, PROFILE_STORE_PATH);
-			return {
-				data,
-				message: `Saved current agents snapshot (${snapshot.agents.length} agents).`,
-				selectedProfileId: profile.id,
-			};
-		},
-		applyProfile: async (profileId) => {
-			const profile = findProfileById(data, profileId);
-			if (!profile) {
-				throw new Error(`Saved profile '${profileId}' was not found.`);
-			}
-			const applied = applySavedProfile(profile, agentOptions);
-			notifyWarnings(ctx, applied.warnings);
-			return {
-				...applied,
-				profileName: profile.name,
-			};
-		},
-		removeProfile: async (profileId) => {
-			const profile = findProfileById(data, profileId);
-			if (!profile) {
-				throw new Error(`Saved profile '${profileId}' was not found.`);
-			}
-			const removal = removeProfileAndUpdate(data, profileId);
-			data = removal.data;
-			saveProfilesFile(data, PROFILE_STORE_PATH);
-			return {
-				data,
-				message: `Removed saved snapshot '${removal.result.removedProfileName}' (${removal.result.remainingCount} snapshots remaining).`,
-				selectedProfileId: data.profiles[0]?.id,
-			};
-		},
-		updateProfile: async (profileId) => {
-			const profile = findProfileById(data, profileId);
-			if (!profile) {
-				throw new Error(`Saved profile '${profileId}' was not found.`);
-			}
-			const update = updateProfileAndReturn(data, profileId, agentOptions);
-			notifyWarnings(ctx, update.result.warnings);
-			data = update.data;
-			saveProfilesFile(data, PROFILE_STORE_PATH);
-			return {
-				data,
-				message: `Updated '${profile.name}' with current agent state (${update.result.updatedAgents} agents).`,
-				selectedProfileId: profileId,
-			};
-		},
-	});
-
-	if (result.type === "applied") {
-		await reloadAfterApply(ctx, result.outcome);
-		return;
-	}
-}
-
-export default function modelProfilesExtension(pi: ExtensionAPI): void {
-	// Register handler to refresh model registry on /reload
-	// This works around the issue where AgentSession.reload() doesn't call ModelRegistry.refresh()
-	onResourcesDiscover(pi, (event, ctx): void => {
-		if (event.reason === "reload") {
-			refreshModelRegistry(ctx);
-		}
-	});
-
-	pi.registerCommand(COMMAND_NAME, {
-		description: "Open saved whole-agent model profile snapshots",
-		handler: async (args: string, ctx: ExtensionCommandContext): Promise<void> => {
-			if (args.trim()) {
-				ctx.ui.notify(`Usage: /${COMMAND_NAME}`, "warning");
-				return;
-			}
-
-			if (!ctx.hasUI) {
-				ctx.ui.notify(`/${COMMAND_NAME} requires interactive TUI mode.`, "warning");
-				return;
-			}
-
-			try {
-				await handleModelProfilesCommand(ctx);
-			} catch (error) {
-				ctx.ui.notify(`/${COMMAND_NAME} failed: ${toErrorMessage(error)}`, "error");
-			}
-		},
-	});
-}
+import type { ExtensionAPI, ExtensionCommandContext } from "@earendil-works/pi-coding-agent";
+
+import { COMMAND_NAME, toErrorMessage } from "./errors.js";
+import { onResourcesDiscover, refreshModelRegistry } from "./pi-api-utils.js";
+
+export default function modelProfilesExtension(pi: ExtensionAPI): void {
+	// Register handler to refresh model registry on /reload
+	// This works around the issue where AgentSession.reload() doesn't call ModelRegistry.refresh()
+	onResourcesDiscover(pi, (event, ctx): void => {
+		if (event.reason === "reload") {
+			refreshModelRegistry(ctx);
+		}
+	});
+
+	pi.registerCommand(COMMAND_NAME, {
+		description: "Open saved whole-agent model profile snapshots",
+		handler: async (args: string, ctx: ExtensionCommandContext): Promise<void> => {
+			if (args.trim()) {
+				ctx.ui.notify(`Usage: /${COMMAND_NAME}`, "warning");
+				return;
+			}
+
+			if (!ctx.hasUI) {
+				ctx.ui.notify(`/${COMMAND_NAME} requires interactive TUI mode.`, "warning");
+				return;
+			}
+
+			try {
+				const { handleModelProfilesCommand } = await import("./command-handler.js");
+				await handleModelProfilesCommand(ctx);
+			} catch (error) {
+				ctx.ui.notify(`/${COMMAND_NAME} failed: ${toErrorMessage(error)}`, "error");
+			}
+		},
+	});
+}

package/src/profile-removal-service.ts

@@ -1,106 +1,106 @@
-import { multiProfilesDebugLogger } from "./debug-logger.js";
-import { ModelProfilesError } from "./errors.js";
-import type { ProfilesFile, ProfileRemovalResult, SavedProfile } from "./types.js";
-
-/**
- * Error code for profile not found during removal.
- */
-export const PROFILE_NOT_FOUND_CODE = "PROFILE_NOT_FOUND";
-
-export interface ProfileRemovalUpdateResult {
-	data: ProfilesFile;
-	result: ProfileRemovalResult;
-}
-
-interface ProfileRemovalPlan {
-	profile: SavedProfile;
-	remainingProfiles: SavedProfile[];
-	result: ProfileRemovalResult;
-}
-
-function createProfileNotFoundError(profileId: string): ModelProfilesError {
-	return new ModelProfilesError(
-		`Profile '${profileId}' not found. It may have been removed already.`,
-		PROFILE_NOT_FOUND_CODE,
-	);
-}
-
-function buildRemovalPlan(data: ProfilesFile, profileId: string): ProfileRemovalPlan {
-	const profileIndex = data.profiles.findIndex((profile) => profile.id === profileId);
-	const profile = data.profiles[profileIndex];
-
-	if (profileIndex === -1 || !profile) {
-		multiProfilesDebugLogger.log("profile-removal", {
-			event: "removal_failed",
-			profileId,
-			reason: "profile_not_found",
-			profileCount: data.profiles.length,
-		});
-		throw createProfileNotFoundError(profileId);
-	}
-
-	const remainingProfiles = [
-		...data.profiles.slice(0, profileIndex),
-		...data.profiles.slice(profileIndex + 1),
-	];
-
-	return {
-		profile,
-		remainingProfiles,
-		result: {
-			removedProfileId: profile.id,
-			removedProfileName: profile.name,
-			remainingCount: remainingProfiles.length,
-		},
-	};
-}
-
-function logProfileRemoved(plan: ProfileRemovalPlan): void {
-	multiProfilesDebugLogger.log("profile-removal", {
-		event: "profile_removed",
-		profileId: plan.profile.id,
-		profileName: plan.profile.name,
-		agentCount: plan.profile.agents.length,
-		remainingCount: plan.result.remainingCount,
-	});
-}
-
-/**
- * Remove a profile from the profiles file.
- *
- * Validates that the profile exists before removal.
- * Preserves other profiles unchanged (immutable pattern).
- * Logs removal event via debug logger.
- *
- * @param data - The profiles file data
- * @param profileId - The ID of the profile to remove
- * @returns ProfileRemovalResult with removed profile info and remaining count
- * @throws ModelProfilesError with code PROFILE_NOT_FOUND if profile doesn't exist
- */
-export function removeProfile(data: ProfilesFile, profileId: string): ProfileRemovalResult {
-	const plan = buildRemovalPlan(data, profileId);
-	logProfileRemoved(plan);
-	return plan.result;
-}
-
-/**
- * Remove a profile and return the updated profiles file with removal metadata.
- *
- * @param data - The profiles file data
- * @param profileId - The ID of the profile to remove
- * @returns New ProfilesFile with the profile removed and removal metadata
- * @throws ModelProfilesError with code PROFILE_NOT_FOUND if profile doesn't exist
- */
-export function removeProfileAndUpdate(data: ProfilesFile, profileId: string): ProfileRemovalUpdateResult {
-	const plan = buildRemovalPlan(data, profileId);
-	logProfileRemoved(plan);
-
-	return {
-		data: {
-			version: data.version,
-			importedAt: data.importedAt,
-			profiles: plan.remainingProfiles,
-		},
-		result: plan.result,
-	};
-}
+import { multiProfilesDebugLogger } from "./debug-logger.js";
+import { ModelProfilesError } from "./errors.js";
+import type { ProfilesFile, ProfileRemovalResult, SavedProfile } from "./types.js";
+
+/**
+ * Error code for profile not found during removal.
+ */
+export const PROFILE_NOT_FOUND_CODE = "PROFILE_NOT_FOUND";
+
+export interface ProfileRemovalUpdateResult {
+	data: ProfilesFile;
+	result: ProfileRemovalResult;
+}
+
+interface ProfileRemovalPlan {
+	profile: SavedProfile;
+	remainingProfiles: SavedProfile[];
+	result: ProfileRemovalResult;
+}
+
+function createProfileNotFoundError(profileId: string): ModelProfilesError {
+	return new ModelProfilesError(
+		`Profile '${profileId}' not found. It may have been removed already.`,
+		PROFILE_NOT_FOUND_CODE,
+	);
+}
+
+function buildRemovalPlan(data: ProfilesFile, profileId: string): ProfileRemovalPlan {
+	const profileIndex = data.profiles.findIndex((profile) => profile.id === profileId);
+	const profile = data.profiles[profileIndex];
+
+	if (profileIndex === -1 || !profile) {
+		multiProfilesDebugLogger.log("profile-removal", {
+			event: "removal_failed",
+			profileId,
+			reason: "profile_not_found",
+			profileCount: data.profiles.length,
+		});
+		throw createProfileNotFoundError(profileId);
+	}
+
+	const remainingProfiles = [
+		...data.profiles.slice(0, profileIndex),
+		...data.profiles.slice(profileIndex + 1),
+	];
+
+	return {
+		profile,
+		remainingProfiles,
+		result: {
+			removedProfileId: profile.id,
+			removedProfileName: profile.name,
+			remainingCount: remainingProfiles.length,
+		},
+	};
+}
+
+function logProfileRemoved(plan: ProfileRemovalPlan): void {
+	multiProfilesDebugLogger.log("profile-removal", {
+		event: "profile_removed",
+		profileId: plan.profile.id,
+		profileName: plan.profile.name,
+		agentCount: plan.profile.agents.length,
+		remainingCount: plan.result.remainingCount,
+	});
+}
+
+/**
+ * Remove a profile from the profiles file.
+ *
+ * Validates that the profile exists before removal.
+ * Preserves other profiles unchanged (immutable pattern).
+ * Logs removal event via debug logger.
+ *
+ * @param data - The profiles file data
+ * @param profileId - The ID of the profile to remove
+ * @returns ProfileRemovalResult with removed profile info and remaining count
+ * @throws ModelProfilesError with code PROFILE_NOT_FOUND if profile doesn't exist
+ */
+export function removeProfile(data: ProfilesFile, profileId: string): ProfileRemovalResult {
+	const plan = buildRemovalPlan(data, profileId);
+	logProfileRemoved(plan);
+	return plan.result;
+}
+
+/**
+ * Remove a profile and return the updated profiles file with removal metadata.
+ *
+ * @param data - The profiles file data
+ * @param profileId - The ID of the profile to remove
+ * @returns New ProfilesFile with the profile removed and removal metadata
+ * @throws ModelProfilesError with code PROFILE_NOT_FOUND if profile doesn't exist
+ */
+export function removeProfileAndUpdate(data: ProfilesFile, profileId: string): ProfileRemovalUpdateResult {
+	const plan = buildRemovalPlan(data, profileId);
+	logProfileRemoved(plan);
+
+	return {
+		data: {
+			version: data.version,
+			importedAt: data.importedAt,
+			profiles: plan.remainingProfiles,
+		},
+		result: plan.result,
+	};
+}

package/src/profile-sort-service.ts

@@ -1,105 +1,105 @@
-import { multiProfilesDebugLogger } from "./debug-logger.js";
-import { loadMultiProfilesConfig, saveMultiProfilesConfig } from "./config.js";
-import type { ProfileSortResult, ProfilesFile, SavedProfile } from "./types.js";
-import type { ProfileSortOrder } from "./types.js";
-
-/**
- * Compare two profile names using locale-aware string comparison.
- */
-function compareByName(left: SavedProfile, right: SavedProfile): number {
-	return left.name.localeCompare(right.name);
-}
-
-/**
- * Compare two profiles by their createdAt timestamp.
- */
-function compareByDate(left: SavedProfile, right: SavedProfile): number {
-	const leftDate = new Date(left.createdAt).getTime();
-	const rightDate = new Date(right.createdAt).getTime();
-	return leftDate - rightDate;
-}
-
-/**
- * Sort profiles based on the specified order.
- * Returns a new array (immutable pattern), preserving the original data.profiles.
- * 
- * @param data - The profiles file containing the profiles array
- * @param order - The sort order to apply
- * @returns ProfileSortResult with sorted profiles and applied sort order
- */
-export function sortProfiles(data: ProfilesFile, order: ProfileSortOrder): ProfileSortResult {
-	const comparator = getSortComparator(order);
-	const sortedProfiles = [...data.profiles].sort(comparator);
-
-	return {
-		sortedProfiles,
-		sortOrder: order,
-	};
-}
-
-/**
- * Get the comparator function for a given sort order.
- */
-function getSortComparator(order: ProfileSortOrder): (left: SavedProfile, right: SavedProfile) => number {
-	switch (order) {
-		case "name-asc":
-			return compareByName;
-		case "name-desc":
-			return (left, right) => compareByName(right, left);
-		case "date-asc":
-			return compareByDate;
-		case "date-desc":
-			return (left, right) => compareByDate(right, left);
-	}
-}
-
-/**
- * Get the current sort order from config.
- */
-export function getCurrentSortOrder(): ProfileSortOrder {
-	const result = loadMultiProfilesConfig();
-	return result.config.sorting.defaultSort;
-}
-
-/**
- * Persist the sort order to config.
- */
-export function persistSortOrder(order: ProfileSortOrder): void {
-	const result = loadMultiProfilesConfig();
-	const config = result.config;
-	config.sorting.defaultSort = order;
-	saveMultiProfilesConfig(config);
-
-	multiProfilesDebugLogger.log("config", {
-		event: "sort_order_persisted",
-		order,
-	});
-}
-
-/**
- * Get display label for a sort order.
- */
-export function getSortOrderLabel(order: ProfileSortOrder): string {
-	switch (order) {
-		case "name-asc":
-			return "Name (A-Z)";
-		case "name-desc":
-			return "Name (Z-A)";
-		case "date-asc":
-			return "Date (Oldest)";
-		case "date-desc":
-			return "Date (Newest)";
-	}
-}
-
-/**
- * Get all available sort orders with their labels.
- */
-export function getAvailableSortOrders(): Array<{ order: ProfileSortOrder; label: string }> {
-	return [
-		{ order: "name-asc", label: "Name (A-Z)" },
-		{ order: "name-desc", label: "Name (Z-A)" },
-		{ order: "date-asc", label: "Date (Oldest)" },
-		{ order: "date-desc", label: "Date (Newest)" },
-	];
-}
+import { multiProfilesDebugLogger } from "./debug-logger.js";
+import { loadMultiProfilesConfig, saveMultiProfilesConfig } from "./config.js";
+import type { ProfileSortResult, ProfilesFile, SavedProfile } from "./types.js";
+import type { ProfileSortOrder } from "./types.js";
+
+/**
+ * Compare two profile names using locale-aware string comparison.
+ */
+function compareByName(left: SavedProfile, right: SavedProfile): number {
+	return left.name.localeCompare(right.name);
+}
+
+/**
+ * Compare two profiles by their createdAt timestamp.
+ */
+function compareByDate(left: SavedProfile, right: SavedProfile): number {
+	const leftDate = new Date(left.createdAt).getTime();
+	const rightDate = new Date(right.createdAt).getTime();
+	return leftDate - rightDate;
+}
+
+/**
+ * Sort profiles based on the specified order.
+ * Returns a new array (immutable pattern), preserving the original data.profiles.
+ * 
+ * @param data - The profiles file containing the profiles array
+ * @param order - The sort order to apply
+ * @returns ProfileSortResult with sorted profiles and applied sort order
+ */
+export function sortProfiles(data: ProfilesFile, order: ProfileSortOrder): ProfileSortResult {
+	const comparator = getSortComparator(order);
+	const sortedProfiles = [...data.profiles].sort(comparator);
+
+	return {
+		sortedProfiles,
+		sortOrder: order,
+	};
+}
+
+/**
+ * Get the comparator function for a given sort order.
+ */
+function getSortComparator(order: ProfileSortOrder): (left: SavedProfile, right: SavedProfile) => number {
+	switch (order) {
+		case "name-asc":
+			return compareByName;
+		case "name-desc":
+			return (left, right) => compareByName(right, left);
+		case "date-asc":
+			return compareByDate;
+		case "date-desc":
+			return (left, right) => compareByDate(right, left);
+	}
+}
+
+/**
+ * Get the current sort order from config.
+ */
+export function getCurrentSortOrder(): ProfileSortOrder {
+	const result = loadMultiProfilesConfig();
+	return result.config.sorting.defaultSort;
+}
+
+/**
+ * Persist the sort order to config.
+ */
+export function persistSortOrder(order: ProfileSortOrder): void {
+	const result = loadMultiProfilesConfig();
+	const config = result.config;
+	config.sorting.defaultSort = order;
+	saveMultiProfilesConfig(config);
+
+	multiProfilesDebugLogger.log("config", {
+		event: "sort_order_persisted",
+		order,
+	});
+}
+
+/**
+ * Get display label for a sort order.
+ */
+export function getSortOrderLabel(order: ProfileSortOrder): string {
+	switch (order) {
+		case "name-asc":
+			return "Name (A-Z)";
+		case "name-desc":
+			return "Name (Z-A)";
+		case "date-asc":
+			return "Date (Oldest)";
+		case "date-desc":
+			return "Date (Newest)";
+	}
+}
+
+/**
+ * Get all available sort orders with their labels.
+ */
+export function getAvailableSortOrders(): Array<{ order: ProfileSortOrder; label: string }> {
+	return [
+		{ order: "name-asc", label: "Name (A-Z)" },
+		{ order: "name-desc", label: "Name (Z-A)" },
+		{ order: "date-asc", label: "Date (Oldest)" },
+		{ order: "date-desc", label: "Date (Newest)" },
+	];
+}