update for existing project creation

Author: eleanorjboyd

src/features/creators/autoFindProjects.ts

@@ -1,11 +1,12 @@
 import * as path from 'path';
 import { Uri } from 'vscode';
-import { showQuickPickWithButtons } from '../../common/window.apis';
+import { showQuickPickWithButtons, showWarningMessage } from '../../common/window.apis';
 import { ProjectCreatorString } from '../../common/localize';
 import { PythonProject, PythonProjectCreator, PythonProjectCreatorOptions } from '../../api';
 import { PythonProjectManager } from '../../internal.api';
 import { showErrorMessage } from '../../common/errors/utils';
 import { findFiles } from '../../common/workspace.apis';
+import { traceInfo } from '../../common/logging';
 
 function getUniqueUri(uris: Uri[]): {
     label: string;
@@ -68,8 +69,9 @@ export class AutoFindProjects implements PythonProjectCreator {
         const filtered = files.filter((uri) => {
             const p = this.pm.get(uri);
             if (p) {
-                // If there ia already a project with the same path, skip it.
-                // If there is a project with the same parent path, skip it.
+                // Skip this project if:
+                // 1. There's already a project registered with exactly the same path
+                // 2. There's already a project registered with this project's parent directory path
                 const np = path.normalize(p.uri.fsPath);
                 const nf = path.normalize(uri.fsPath);
                 const nfp = path.dirname(nf);
@@ -79,11 +81,20 @@ export class AutoFindProjects implements PythonProjectCreator {
         });
 
         if (filtered.length === 0) {
+            // No new projects found that are not already in the project manager
+            traceInfo('All discovered projects are already registered in the project manager');
+            setImmediate(() => {
+                showWarningMessage('No new projects found');
+            });
             return;
         }
 
+        traceInfo(`Found ${filtered.length} new potential projects that aren't already registered`);
+
         const projects = await pickProjects(filtered);
         if (!projects || projects.length === 0) {
+            // User cancelled the selection.
+            traceInfo('User cancelled project selection.');
             return;
         }
 

src/features/creators/existingProjects.ts

@@ -1,12 +1,16 @@
 import * as path from 'path';
 import { PythonProject, PythonProjectCreator, PythonProjectCreatorOptions } from '../../api';
 import { ProjectCreatorString } from '../../common/localize';
-import { showOpenDialog } from '../../common/window.apis';
+import { showOpenDialog, showWarningMessage } from '../../common/window.apis';
+import { PythonProjectManager } from '../../internal.api';
+import { traceInfo } from '../../common/logging';
 
 export class ExistingProjects implements PythonProjectCreator {
     public readonly name = 'existingProjects';
     public readonly displayName = ProjectCreatorString.addExistingProjects;
 
+    constructor(private readonly pm: PythonProjectManager) {}
+
     async create(_options?: PythonProjectCreatorOptions): Promise<PythonProject | PythonProject[] | undefined> {
         const results = await showOpenDialog({
             canSelectFiles: true,
@@ -22,7 +26,30 @@ export class ExistingProjects implements PythonProjectCreator {
             return;
         }
 
-        return results.map((r) => ({
+        // do we have any limitations that need to be applied here?
+        // like selected folder not child of workspace folder?
+
+        const filtered = results.filter((uri) => {
+            const p = this.pm.get(uri);
+            if (p) {
+                // Skip this project if there's already a project registered with exactly the same path
+                const np = path.normalize(p.uri.fsPath);
+                const nf = path.normalize(uri.fsPath);
+                return np !== nf;
+            }
+            return true;
+        });
+
+        if (filtered.length === 0) {
+            // No new projects found that are not already in the project manager
+            traceInfo('All discovered projects are already registered in the project manager');
+            setImmediate(() => {
+                showWarningMessage('No new projects found');
+            });
+            return;
+        }
+
+        return filtered.map((r) => ({
             name: path.basename(r.fsPath),
             uri: r,
         }));

src/features/envCommands.ts

@@ -348,6 +348,14 @@ export async function addPythonProject(
         return;
     }
 
+    if (
+        resource instanceof ProjectPackageRootTreeItem ||
+        resource instanceof ProjectPackage ||
+        resource instanceof ProjectEnvironment
+    ) {
+        await addPythonProject(undefined, wm, em, pc);
+    }
+
     if (resource instanceof Uri) {
         const uri = resource as Uri;
         const envManagerId = getDefaultEnvManagerSetting(wm, uri);