Adding menu items (#310)

Author: eleanorjboyd

src/common/pickers/managers.ts

@@ -131,14 +131,57 @@ export async function pickCreator(creators: PythonProjectCreator[]): Promise<Pyt
         return creators[0];
     }
 
-    const items: (QuickPickItem & { c: PythonProjectCreator })[] = creators.map((c) => ({
-        label: c.displayName ?? c.name,
-        description: c.description,
-        c: c,
-    }));
+    // First level menu
+    const autoFindCreator = creators.find((c) => c.name === 'autoProjects');
+    const existingProjectsCreator = creators.find((c) => c.name === 'existingProjects');
+    const otherCreators = creators.filter((c) => c.name !== 'autoProjects' && c.name !== 'existingProjects');
+
+    const items: QuickPickItem[] = [
+        {
+            label: 'Auto Find',
+            description: autoFindCreator?.description ?? 'Automatically find Python projects',
+        },
+        {
+            label: 'Select Existing',
+            description: existingProjectsCreator?.description ?? 'Select existing Python projects',
+        },
+        {
+            label: 'Create New',
+            description: 'Create a Python project from a template',
+        },
+    ];
+
     const selected = await showQuickPick(items, {
         placeHolder: Pickers.Managers.selectProjectCreator,
         ignoreFocusOut: true,
     });
-    return (selected as { c: PythonProjectCreator })?.c;
+
+    if (!selected) {
+        return undefined;
+    }
+
+    // Return appropriate creator based on selection
+    switch (selected.label) {
+        case 'Auto Find':
+            return autoFindCreator;
+        case 'Select Existing':
+            return existingProjectsCreator;
+        case 'Create New':
+            // Show second level menu for other creators
+            if (otherCreators.length === 0) {
+                return undefined;
+            }
+            const newItems: (QuickPickItem & { c: PythonProjectCreator })[] = otherCreators.map((c) => ({
+                label: c.displayName ?? c.name,
+                description: c.description,
+                c: c,
+            }));
+            const newSelected = await showQuickPick(newItems, {
+                placeHolder: 'Select project type for new project',
+                ignoreFocusOut: true,
+            });
+            return newSelected?.c;
+    }
+
+    return undefined;
 }

src/extension.ts

@@ -3,6 +3,8 @@ import { commands, ExtensionContext, LogOutputChannel, Terminal, Uri } from 'vsc
 import { PythonEnvironmentManagers } from './features/envManagers';
 import { registerLogger, traceInfo } from './common/logging';
 import { EnvManagerView } from './features/views/envManagersView';
+import { NewPackageProject } from './features/creators/newPackageProject';
+import { NewScriptProject } from './features/creators/newScriptProject';
 import {
     addPythonProject,
     createEnvironmentCommand,
@@ -90,6 +92,8 @@ export async function activate(context: ExtensionContext): Promise<PythonEnviron
         projectCreators,
         projectCreators.registerPythonProjectCreator(new ExistingProjects()),
         projectCreators.registerPythonProjectCreator(new AutoFindProjects(projectManager)),
+        projectCreators.registerPythonProjectCreator(new NewPackageProject()),
+        projectCreators.registerPythonProjectCreator(new NewScriptProject()),
     );
 
     setPythonApi(envManagers, projectManager, projectCreators, terminalManager, envVarManager);

src/features/creators/newPackageProject.ts

@@ -0,0 +1,18 @@
+import { MarkdownString, window } from 'vscode';
+import { PythonProject, PythonProjectCreator, PythonProjectCreatorOptions } from '../../api';
+// import { runInBackground } from '../execution/runInBackground';
+
+export class NewPackageProject implements PythonProjectCreator {
+    public readonly name = 'newPackage';
+    public readonly displayName = 'Package';
+    public readonly description = 'Create a new Python package';
+    public readonly tooltip = new MarkdownString('Create a new Python package');
+
+    constructor() {}
+
+    async create(_options?: PythonProjectCreatorOptions): Promise<PythonProject | undefined> {
+        // show notification that the package creation was selected than return undefined
+        window.showInformationMessage('Creating a new Python package...');
+        return undefined;
+    }
+}

src/features/creators/newScriptProject.ts

@@ -0,0 +1,17 @@
+import { MarkdownString, window } from 'vscode';
+import { PythonProject, PythonProjectCreator, PythonProjectCreatorOptions } from '../../api';
+
+export class NewScriptProject implements PythonProjectCreator {
+    public readonly name = 'newScript';
+    public readonly displayName = 'Project';
+    public readonly description = 'Create a new Python project';
+    public readonly tooltip = new MarkdownString('Create a new Python project');
+
+    constructor() {}
+
+    async create(_options?: PythonProjectCreatorOptions): Promise<PythonProject | undefined> {
+        // show notification that the script creation was selected than return undefined
+        window.showInformationMessage('Creating a new Python project...');
+        return undefined;
+    }
+}