pi-model-profiles 0.2.0

package/src/profile-update-service.ts

@@ -0,0 +1,134 @@
+import { multiProfilesDebugLogger } from "./debug-logger.js";
+import { ModelProfilesError } from "./errors.js";
+import { captureAgentSnapshots, type AgentSelectionOptions } from "./agent-writer.js";
+import type { ProfilesFile, ProfileUpdateResult, SavedProfile } from "./types.js";
+
+/**
+ * Error code for profile not found during update.
+ */
+export const PROFILE_NOT_FOUND_CODE = "PROFILE_NOT_FOUND";
+
+export interface ProfileUpdateDataResult {
+	data: ProfilesFile;
+	result: ProfileUpdateResult;
+}
+
+interface ProfileUpdatePlan {
+	updatedData: ProfilesFile;
+	updatedProfile: SavedProfile;
+	previousProfile: SavedProfile;
+	result: ProfileUpdateResult;
+}
+
+function createProfileNotFoundError(profileId: string): ModelProfilesError {
+	return new ModelProfilesError(
+		`Profile '${profileId}' not found. It may have been removed already.`,
+		PROFILE_NOT_FOUND_CODE,
+	);
+}
+
+function buildUpdatePlan(
+	data: ProfilesFile,
+	profileId: string,
+	agentOptions: AgentSelectionOptions,
+): ProfileUpdatePlan {
+	const profileIndex = data.profiles.findIndex((profile) => profile.id === profileId);
+	const profile = data.profiles[profileIndex];
+
+	if (profileIndex === -1 || !profile) {
+		multiProfilesDebugLogger.log("profile-update", {
+			event: "update_failed",
+			profileId,
+			reason: "profile_not_found",
+			profileCount: data.profiles.length,
+		});
+		throw createProfileNotFoundError(profileId);
+	}
+
+	const snapshot = captureAgentSnapshots(agentOptions);
+	const updatedProfile: SavedProfile = {
+		id: profile.id,
+		name: profile.name,
+		agents: snapshot.agents,
+		createdAt: profile.createdAt,
+		updatedAt: new Date().toISOString(),
+	};
+	const updatedProfiles = [
+		...data.profiles.slice(0, profileIndex),
+		updatedProfile,
+		...data.profiles.slice(profileIndex + 1),
+	];
+
+	return {
+		updatedData: {
+			version: data.version,
+			importedAt: data.importedAt,
+			profiles: updatedProfiles,
+		},
+		updatedProfile,
+		previousProfile: profile,
+		result: {
+			updatedProfileId: profile.id,
+			updatedProfileName: profile.name,
+			updatedAgents: snapshot.agents.length,
+			warnings: snapshot.warnings,
+		},
+	};
+}
+
+function logProfileUpdated(plan: ProfileUpdatePlan): void {
+	multiProfilesDebugLogger.log("profile-update", {
+		event: "profile_updated",
+		profileId: plan.updatedProfile.id,
+		profileName: plan.updatedProfile.name,
+		beforeAgentCount: plan.previousProfile.agents.length,
+		afterAgentCount: plan.updatedProfile.agents.length,
+		warnings: plan.result.warnings,
+	});
+}
+
+/**
+ * Update a profile with the current agent state.
+ *
+ * Captures the current agent snapshots and replaces the profile's agents array.
+ * Preserves profile.id, profile.name, and profile.createdAt.
+ * Updates profile.updatedAt timestamp to current time.
+ * Logs update event with before/after agent counts via debug logger.
+ *
+ * @param data - The profiles file data
+ * @param profileId - The ID of the profile to update
+ * @param agentOptions - Options for capturing agent snapshots
+ * @returns ProfileUpdateResult with updated profile info and agent counts
+ * @throws ModelProfilesError with code PROFILE_NOT_FOUND if profile doesn't exist
+ */
+export function updateProfile(
+	data: ProfilesFile,
+	profileId: string,
+	agentOptions: AgentSelectionOptions,
+): ProfileUpdateResult {
+	const plan = buildUpdatePlan(data, profileId, agentOptions);
+	logProfileUpdated(plan);
+	return plan.result;
+}
+
+/**
+ * Update a profile and return the updated profiles file with update metadata.
+ *
+ * @param data - The profiles file data
+ * @param profileId - The ID of the profile to update
+ * @param agentOptions - Options for capturing agent snapshots
+ * @returns New ProfilesFile with the profile updated and update metadata
+ * @throws ModelProfilesError with code PROFILE_NOT_FOUND if profile doesn't exist
+ */
+export function updateProfileAndReturn(
+	data: ProfilesFile,
+	profileId: string,
+	agentOptions: AgentSelectionOptions,
+): ProfileUpdateDataResult {
+	const plan = buildUpdatePlan(data, profileId, agentOptions);
+	logProfileUpdated(plan);
+	return {
+		data: plan.updatedData,
+		result: plan.result,
+	};
+}

package/src/types-shims.d.ts

@@ -0,0 +1,121 @@
+declare namespace NodeJS {
+  interface Process {
+    pid: number;
+    stdout: { columns?: number; rows?: number };
+  }
+}
+
+declare const process: NodeJS.Process;
+
+declare module "node:assert/strict" {
+  const assert: any;
+  export default assert;
+}
+
+declare module "node:crypto" {
+  export function randomUUID(): string;
+}
+
+declare module "node:fs" {
+  export function existsSync(path: string): boolean;
+  export function mkdirSync(path: string, options?: unknown): void;
+  export function mkdtempSync(path: string): string;
+  export function readFileSync(path: string, encoding: string): string;
+  export function readdirSync(path: string, options?: unknown): unknown[];
+  export function renameSync(oldPath: string, newPath: string): void;
+  export function rmSync(path: string, options?: unknown): void;
+  export function unlinkSync(path: string): void;
+  export function writeFileSync(path: string, content: string, encoding: string): void;
+}
+
+declare module "node:os" {
+  export function homedir(): string;
+  export function tmpdir(): string;
+}
+
+declare module "node:path" {
+  export function basename(path: string): string;
+  export function dirname(path: string): string;
+  export function join(...parts: string[]): string;
+}
+
+declare module "node:test" {
+  const test: any;
+  export default test;
+}
+
+declare module "@mariozechner/pi-coding-agent" {
+  export interface Theme {
+    [key: string]: any;
+  }
+
+  export interface ExtensionContext {
+    hasUI: boolean;
+    cwd: string;
+    sessionManager: {
+      getEntries(): unknown[];
+    };
+    ui: {
+      notify(message: string, level?: "info" | "warning" | "error"): void;
+      custom<T>(...args: any[]): Promise<T>;
+    };
+    getSystemPrompt(): string;
+  }
+
+  export interface ExtensionCommandContext extends ExtensionContext {
+    reload(): Promise<void>;
+  }
+
+  export interface ExtensionAPI {
+    registerCommand(name: string, options: { description: string; handler: (...args: any[]) => Promise<void> | void }): void;
+    sendUserMessage(content: string, options?: { deliverAs?: "steer" | "followUp" }): void;
+  }
+
+  export function getAgentDir(): string;
+  export function getSettingsListTheme(theme?: Theme): any;
+}
+
+declare module "@mariozechner/pi-tui" {
+  export type SettingItem = any;
+
+  export class Box {
+    [key: string]: any;
+    constructor(...args: any[]);
+  }
+
+  export class Container {
+    [key: string]: any;
+    constructor(...args: any[]);
+  }
+
+  export class Input {
+    focused: boolean;
+    onSubmit?: (value: string) => void;
+    onEscape?: () => void;
+    constructor(...args: any[]);
+    setValue(value: string): void;
+    render(width: number): string[];
+    handleInput(data: string): void;
+  }
+
+  export class SettingsList {
+    [key: string]: any;
+    constructor(...args: any[]);
+    handleInput(data: string): void;
+    updateValue(id: string, value: string): void;
+  }
+
+  export class Spacer {
+    [key: string]: any;
+    constructor(...args: any[]);
+  }
+
+  export class Text {
+    [key: string]: any;
+    constructor(...args: any[]);
+  }
+
+  export function matchesKey(data: string, key: string): boolean;
+  export function truncateToWidth(text: string, width: number, ellipsis?: string, trimWhitespace?: boolean): string;
+  export function visibleWidth(text: string): number;
+}

package/src/types.ts

@@ -0,0 +1,104 @@
+export const PROFILE_FIELD_KEYS = ["model", "temperature", "reasoningEffort"] as const;
+
+export type ProfileFieldKey = (typeof PROFILE_FIELD_KEYS)[number];
+
+/** Sort order for profile listing */
+export type ProfileSortOrder = "name-asc" | "name-desc" | "date-asc" | "date-desc";
+
+export interface ProfileFields {
+	model?: string;
+	temperature?: number;
+	reasoningEffort?: string;
+}
+
+export interface SavedProfileAgent {
+	fileName: string;
+	agentName: string;
+	fields: ProfileFields;
+}
+
+export interface SavedProfile {
+	id: string;
+	name: string;
+	agents: SavedProfileAgent[];
+	createdAt: string;
+	updatedAt: string;
+}
+
+export interface ProfilesFile {
+	version: 2;
+	importedAt?: string;
+	profiles: SavedProfile[];
+}
+
+export interface ProfileStoreLoadResult {
+	data: ProfilesFile;
+	warning?: string;
+	needsSave: boolean;
+}
+
+export interface AgentFileRecord {
+	path: string;
+	fileName: string;
+	agentName: string;
+	fields: ProfileFields;
+}
+
+export interface AgentScanResult {
+	agents: AgentFileRecord[];
+	warnings: string[];
+}
+
+export interface AgentSnapshotResult {
+	agents: SavedProfileAgent[];
+	warnings: string[];
+}
+
+export interface AppliedAgentUpdate {
+	updatedPath: string;
+	fileName: string;
+	agentName: string;
+	appliedKeys: ProfileFieldKey[];
+	removedKeys: ProfileFieldKey[];
+}
+
+export interface ApplyProfileResult {
+	appliedAgents: AppliedAgentUpdate[];
+	missingAgents: string[];
+	warnings: string[];
+}
+
+export interface ImportProfilesResult {
+	data: ProfilesFile;
+	imported: boolean;
+	importedCount: number;
+	warnings: string[];
+}
+
+export interface AppliedProfileOutcome extends ApplyProfileResult {
+	profileName: string;
+}
+
+/** Result of profile removal operation */
+export interface ProfileRemovalResult {
+	removedProfileId: string;
+	removedProfileName: string;
+	remainingCount: number;
+}
+
+/** Result of profile update operation */
+export interface ProfileUpdateResult {
+	updatedProfileId: string;
+	updatedProfileName: string;
+	updatedAgents: number;
+	warnings: string[];
+}
+
+/** Result of profile sort operation */
+export interface ProfileSortResult {
+	sortedProfiles: SavedProfile[];
+	sortOrder: ProfileSortOrder;
+}
+
+// Re-export config types for type access
+export type { MultiProfilesConfig } from "./config.js";