Make env types optional

Author: Kartik Raj

src/client/apiTypes.ts

@@ -5,12 +5,14 @@
 
 import { Disposable, Event, Uri } from 'vscode';
 import { Resource } from './common/types';
-import { Architecture } from './common/utils/platform';
 import { IDataViewerDataProvider, IJupyterUriProvider } from './jupyter/types';
 import { EnvPathType } from './pythonEnvironments/base/info';
 import {
+    EnvironmentDetails,
+    EnvironmentDetailsOptions,
+    EnvironmentProviderMetadata,
     GetRefreshEnvironmentsOptions,
-    IPythonEnvsIterator,
+    IEnvironmentProvider,
     ProgressNotificationEvent,
 } from './pythonEnvironments/base/locator';
 
@@ -96,152 +98,6 @@ export interface IExtensionApi {
     };
 }
 
-export interface EnvironmentDetailsOptions {
-    useCache: boolean;
-    /**
-     * Return details provided by the specific provider, throws error if provider not found.
-     */
-    providerId?: ProviderID;
-}
-
-type VersionInfo = {
-    major: number;
-    minor: number;
-    micro: number;
-    releaselevel: 'alpha' | 'beta' | 'candidate' | 'final';
-    serial: number;
-};
-
-export interface EnvironmentDetails {
-    executable: {
-        path: string;
-        bitness?: Architecture;
-        sysPrefix: string;
-    };
-    environment?: {
-        type: EnvType;
-        name?: string;
-        path: string;
-        project?: string; // Any specific project environment is created for.
-        source: EnvSource[];
-    };
-    version: VersionInfo & {
-        sysVersion?: string;
-    };
-    implementation?: {
-        // `sys.implementation`
-        name: string;
-        version: VersionInfo & {
-            serial: number;
-        };
-    };
-}
-
-/**
- * Provider is only required to provide the `executable` key, rest are optional. So construct a type using
- * `EnvironmentDetails` where `executable` is the only required key.
- */
-type EnvironmentDetailsByProvider = Partial<EnvironmentDetails> & Pick<EnvironmentDetails, 'executable'>;
-
-interface IEnvironmentProvider extends ILocatorFactoryAPI, IResolverAPI {}
-
-interface ILocatorFactoryAPI {
-    /**
-     * Factory function calling which create the locator.
-     */
-    createLocator: ILocatorFactory;
-}
-
-interface IResolverAPI {
-    /**
-     * Returns true if provided environment is recognized by the provider.
-     */
-    canIdentifyEnvironment: (env: BaseEnvInfo) => Promise<boolean>;
-    /**
-     * This is only called if this provider can identify the environment.
-     * Returns details or `undefined` if it was found if env is invalid.
-     */
-    getEnvironmentDetails: (env: BaseEnvInfo) => Promise<EnvironmentDetailsByProvider | undefined>;
-}
-
-export type ILocatorFactory = IWorkspaceLocatorFactory | INonWorkspaceLocatorFactory;
-export type INonWorkspaceLocatorFactory = () => ILocatorAPI;
-export type IWorkspaceLocatorFactory = (root: string) => ILocatorAPI;
-
-export interface ILocatorAPI {
-    iterEnvs?(): IPythonEnvsIterator<EnvInfo>;
-    readonly onChanged?: Event<LocatorEnvsChangedEvent>;
-}
-
-export type EnvInfo = BaseEnvInfo & {
-    envSources: EnvSource[];
-};
-
-export type BaseEnvInfo = {
-    executablePath: string;
-    envPath?: string;
-};
-
-type ProviderID = string;
-
-/**
- * These can be used when querying for a particular env.
- */
-interface EnvironmentProviderMetadata {
-    /**
-     * Details about the environments the locator provides.
-     * Useful when querying for a particular env.
-     */
-    readonly environments?: EnvironmentMetaData;
-    /**
-     * If locator requires a workspace root to search envs within.
-     */
-    readonly isWorkspaceBasedLocator: boolean;
-    /**
-     * An Identifier for the provider.
-     */
-    readonly providerId: ProviderID;
-}
-
-interface EnvironmentMetaData {
-    readonly envType: EnvType;
-    readonly envSources: EnvSource[];
-}
-
-export interface LocatorEnvsChangedEvent {
-    /**
-     * Any details known about the environment which can be used for query.
-     */
-    env?: EnvironmentMetaData;
-    /**
-     * Details about how the environment was modified.
-     * */
-    type: EnvChangeType;
-}
-
-export type EnvChangeType = 'add' | 'remove' | 'update';
-
-export type EnvType = KnownEnvTypes | string;
-
-export enum KnownEnvTypes {
-    VirtualEnv = 'VirtualEnv',
-    Conda = 'Conda',
-    Unknown = 'Unknown',
-    Global = 'Global',
-}
-
-export type EnvSource = KnownEnvSourceTypes | string;
-
-export enum KnownEnvSourceTypes {
-    Conda = 'Conda',
-    Pipenv = 'PipEnv',
-    Poetry = 'Poetry',
-    VirtualEnv = 'VirtualEnv',
-    Venv = 'Venv',
-    VirtualEnvWrapper = 'VirtualEnvWrapper',
-    Pyenv = 'Pyenv',
-}
-
 export interface EnvironmentsChangedParams {
     /**
      * Path to environment folder or path to interpreter that uniquely identifies an environment.

src/client/proposedApi.ts

@@ -4,8 +4,6 @@
 import { ConfigurationTarget, EventEmitter } from 'vscode';
 import {
     ActiveEnvironmentChangedParams,
-    EnvironmentDetails,
-    EnvironmentDetailsOptions,
     EnvironmentsChangedParams,
     IProposedExtensionAPI,
     RefreshEnvironmentsOptions,

src/client/pythonEnvironments/base/locator.ts

@@ -126,7 +126,6 @@ export interface EnvironmentProviderMetadata {
 }
 
 interface InternalEnvironmentMetaData {
-    readonly envType: EnvType;
     readonly envKinds: PythonEnvKind[];
 }
 
@@ -147,7 +146,7 @@ export interface InternalEnvironmentProviderMetadata {
 
 interface EnvironmentMetaData {
     readonly envType: EnvType;
-    readonly envSources: EnvSource[];
+    readonly envSources?: EnvSource[];
 }
 
 export interface LocatorEnvsChangedEvent {

src/client/pythonEnvironments/base/locators/wrappers.ts

@@ -3,7 +3,6 @@
 
 // eslint-disable-next-line max-classes-per-file
 import { Uri } from 'vscode';
-import { ILocatorFactory } from '../../../apiTypes';
 import { IDisposable } from '../../../common/types';
 import { iterEmpty } from '../../../common/utils/async';
 import { getURIFilter } from '../../../common/utils/misc';
@@ -18,6 +17,7 @@ import {
     PythonLocatorQuery,
     IWorkspaceLocatorFactory,
     ILocatorRegister,
+    ILocatorFactory,
 } from '../locator';
 import { combineIterators, Locators } from '../locators';
 import { LazyResourceBasedLocator } from './common/resourceBasedLocator';

src/client/pythonEnvironments/converter.ts

@@ -2,7 +2,6 @@
 // Licensed under the MIT License.
 
 import { EventEmitter, Event } from 'vscode';
-import { EnvChangeType, ILocatorAPI, LocatorEnvsChangedEvent, EnvInfo } from '../apiTypes';
 import { FileChangeType } from '../common/platform/fileSystemWatcher';
 import { traceVerbose } from '../logging';
 import { PythonEnvInfo, PythonEnvKind } from './base/info';
@@ -23,6 +22,10 @@ import {
     IEnvironmentProvider,
     EnvironmentProviderMetadata,
     InternalEnvironmentProviderMetadata,
+    EnvChangeType,
+    EnvInfo,
+    ILocatorAPI,
+    LocatorEnvsChangedEvent,
 } from './base/locator';
 import { PythonEnvsChangedEvent } from './base/watcher';
 
@@ -37,8 +40,7 @@ export function convertProviderMetaData(proposed: EnvironmentProviderMetadata):
     return {
         providerId: proposed.providerId,
         environments: {
-            envKinds: proposed.environments.envSources.map((e) => convertKind(e)),
-            envType: proposed.environments.envType,
+            envKinds: proposed.environments.envSources?.map((e) => convertKind(e)) ?? [PythonEnvKind.Unknown],
         },
     };
 }