deploy: current vibn theia state

Made-with: Cursor

packages/ai-ide/src/browser/context-file-validation-service-impl.ts

@@ -0,0 +1,120 @@
+// *****************************************************************************
+// Copyright (C) 2025 EclipseSource GmbH.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+
+import { injectable, inject } from '@theia/core/shared/inversify';
+import { URI } from '@theia/core';
+import { FileService } from '@theia/filesystem/lib/browser/file-service';
+import { WorkspaceService } from '@theia/workspace/lib/browser/workspace-service';
+import { ContextFileValidationService, FileValidationResult, FileValidationState } from '@theia/ai-chat/lib/browser/context-file-validation-service';
+import { WorkspaceFunctionScope } from './workspace-functions';
+
+@injectable()
+export class ContextFileValidationServiceImpl implements ContextFileValidationService {
+
+    @inject(FileService)
+    protected readonly fileService: FileService;
+
+    @inject(WorkspaceFunctionScope)
+    protected readonly workspaceScope: WorkspaceFunctionScope;
+
+    @inject(WorkspaceService)
+    protected readonly workspaceService: WorkspaceService;
+
+    async validateFile(pathOrUri: string | URI): Promise<FileValidationResult> {
+        try {
+            const resolvedUri = await this.workspaceScope.resolveToUri(pathOrUri);
+
+            if (!resolvedUri) {
+                return {
+                    state: FileValidationState.INVALID_NOT_FOUND,
+                    message: 'File does not exist'
+                };
+            }
+
+            const exists = await this.fileService.exists(resolvedUri);
+            if (!exists) {
+                const secondaryRootUri = await this.findInSecondaryWorkspaceRoots(pathOrUri);
+                if (secondaryRootUri) {
+                    return {
+                        state: FileValidationState.INVALID_SECONDARY,
+                        message: 'File is in a secondary workspace root. AI agents can only access files in the first workspace root.'
+                    };
+                }
+                return {
+                    state: FileValidationState.INVALID_NOT_FOUND,
+                    message: 'File does not exist'
+                };
+            }
+
+            if (this.workspaceScope.isInPrimaryWorkspace(resolvedUri)) {
+                return {
+                    state: FileValidationState.VALID
+                };
+            }
+
+            if (this.workspaceScope.isInWorkspace(resolvedUri)) {
+                return {
+                    state: FileValidationState.INVALID_SECONDARY,
+                    message: 'File is in a secondary workspace root. AI agents can only access files in the first workspace root.'
+                };
+            }
+
+            return {
+                state: FileValidationState.INVALID_NOT_FOUND,
+                message: 'File does not exist in the workspace'
+            };
+        } catch (error) {
+            return {
+                state: FileValidationState.INVALID_NOT_FOUND,
+                message: 'File does not exist'
+            };
+        }
+    }
+
+    protected async findInSecondaryWorkspaceRoots(pathOrUri: string | URI): Promise<URI | undefined> {
+        const roots = this.workspaceService.tryGetRoots();
+        if (roots.length <= 1) {
+            return undefined;
+        }
+
+        for (let i = 1; i < roots.length; i++) {
+            const root = roots[i];
+            let candidateUri: URI;
+
+            if (pathOrUri instanceof URI) {
+                candidateUri = pathOrUri;
+            } else if (pathOrUri.includes('://')) {
+                try {
+                    candidateUri = new URI(pathOrUri);
+                } catch {
+                    continue;
+                }
+            } else {
+                candidateUri = root.resource.resolve(pathOrUri);
+            }
+
+            try {
+                if (await this.fileService.exists(candidateUri)) {
+                    return candidateUri;
+                }
+            } catch {
+                continue;
+            }
+        }
+
+        return undefined;
+    }
+}