@aws/lsp-codewhisperer 0.0.34 → 0.0.36

package/out/language-server/workspaceContext/workspaceFolderManager.d.ts

@@ -0,0 +1,113 @@
+import { WebSocketClient } from './client';
+import { WorkspaceFolder } from '@aws/language-server-runtimes/protocol';
+import { WorkspaceStatus } from '../../client/token/codewhispererbearertokenclient';
+import { CredentialsProvider, Logging } from '@aws/language-server-runtimes/server-interface';
+import { ArtifactManager, FileMetadata } from './artifactManager';
+import { DependencyDiscoverer } from './dependency/dependencyDiscoverer';
+import { AmazonQTokenServiceManager } from '../../shared/amazonQServiceManager/AmazonQTokenServiceManager';
+interface WorkspaceState {
+    remoteWorkspaceState: WorkspaceStatus;
+    webSocketClient?: WebSocketClient;
+    workspaceId?: string;
+    requiresS3Upload?: boolean;
+    messageQueue?: any[];
+}
+type WorkspaceRoot = string;
+export declare class WorkspaceFolderManager {
+    private serviceManager;
+    private logging;
+    private artifactManager;
+    private dependencyDiscoverer;
+    private workspaceMap;
+    private static instance;
+    private workspaceFolders;
+    private credentialsProvider;
+    private readonly INITIAL_CHECK_INTERVAL;
+    private readonly INITIAL_TIMEOUT;
+    private readonly CONTINUOUS_MONITOR_INTERVAL;
+    private monitorIntervals;
+    private isOptedOut;
+    static createInstance(serviceManager: AmazonQTokenServiceManager, logging: Logging, artifactManager: ArtifactManager, dependencyDiscoverer: DependencyDiscoverer, workspaceFolders: WorkspaceFolder[], credentialsProvider: CredentialsProvider): WorkspaceFolderManager;
+    static getInstance(): WorkspaceFolderManager | undefined;
+    private constructor();
+    /**
+     * The function is used to track the latest state of workspace folders.
+     * This state is updated irrespective of login/logout/optIn/optOut
+     * @param workspaceFolders
+     */
+    updateWorkspaceFolders(workspaceFolders: WorkspaceFolder[]): void;
+    getWorkspaces(): Map<WorkspaceRoot, WorkspaceState>;
+    getWorkspaceDetailsWithId(fileUri: string, workspaceFolders?: WorkspaceFolder[]): {
+        workspaceDetails: WorkspaceState;
+        workspaceRoot: WorkspaceFolder;
+    } | null;
+    getWorkspaceFolder(fileUri: string): WorkspaceFolder | undefined;
+    getWorkspaceId(workspaceFolder?: WorkspaceFolder): string | undefined;
+    getOptOutStatus(): boolean;
+    processNewWorkspaceFolders(folders: WorkspaceFolder[]): Promise<void>;
+    private snapshotWorkspace;
+    uploadToS3(fileMetadata: FileMetadata): Promise<string | undefined>;
+    clearAllWorkspaceResources(): Promise<void>;
+    /**
+     * The function sends a removed workspace folders notification to remote LSP, removes workspace entry
+     * from map and close the websocket connection
+     * @param workspaceFolder
+     */
+    processWorkspaceFoldersDeletion(workspaceFolders: WorkspaceFolder[]): Promise<void>;
+    processRemoteWorkspaceRefresh(workspaceFolders: WorkspaceFolder[]): void;
+    private updateWorkspaceEntry;
+    private removeWorkspaceEntry;
+    private handleDependencyChanges;
+    private notifyDependencyChange;
+    private createNewWorkspace;
+    private establishConnection;
+    private handleNewWorkspace;
+    private waitForInitialConnection;
+    private startContinuousMonitor;
+    private startOptOutMonitor;
+    /**
+     * Handles the workspace creation flow when a remote workspace is in CREATED state.
+     * Attempts to create the workspace and schedules a quick connection check on success.
+     * If the initial creation fails with a retryable error, attempts one retry before
+     * falling back to the regular polling cycle.
+     *
+     * The flow is:
+     * 1. Attempt initial workspace creation
+     * 2. On success: Schedule a quick check to establish connection
+     * 3. On retryable error: Attempt one immediate retry
+     * 4. On retry success: Schedule a quick check
+     * 5. On retry failure: Wait for next regular polling cycle
+     *
+     * @param workspace - The workspace to re-create
+     */
+    private handleWorkspaceCreatedState;
+    /**
+     * Schedules a one-time check after workspace creation to establish connection as soon as possible.
+     * This avoids waiting for the regular 5-minute polling interval when we know the workspace
+     * should be ready soon. Default check delay is 40 seconds after successful workspace creation.
+     *
+     * @param workspace - The workspace to check
+     * @param delayMs - Optional delay in milliseconds before the check (defaults to 40 seconds)
+     */
+    private scheduleQuickCheck;
+    private stopMonitoring;
+    private startWorkspaceStatusMonitor;
+    private processMessagesInQueue;
+    /**
+     * All the filesMetadata elements passed to the function belongs to the same workspace folder.
+     * @param filesMetadata
+     * @private
+     */
+    private uploadS3AndQueueEvents;
+    private uploadWithTimeout;
+    private deleteWorkspace;
+    /**
+     * The function fetches remote workspace metadata. There'll always be single entry for workspace
+     * metadata in the response, so intentionally picking the first index element.
+     * @param workspaceRoot
+     * @private
+     */
+    private listWorkspaceMetadata;
+    private createWorkspace;
+}
+export {};