@workbench-kit/shell-react 0.0.2-prototype.0.2.18 → 0.0.2-prototype.0.2.19

package/src/extensions/management-model.ts

@@ -1,6 +1,4 @@
 import {
-  BUILTIN_WORKBENCH_EXTENSIONS,
-  SAMPLE_WORKBENCH_EXTENSIONS,
   createExtensionFeatureSpec,
   createExtensionInstallPlan,
   resolveBundledExtensionByManifestUrl,
@@ -20,12 +18,8 @@ import type {
   ExtensionManagementFeatureSummary,
 } from '@workbench-kit/react/workbench/management';
 
-export const BUNDLED_EXTENSIONS = [
-  ...BUILTIN_WORKBENCH_EXTENSIONS,
-  ...SAMPLE_WORKBENCH_EXTENSIONS,
-] as const;
-
 export interface CreateExtensionManagementEntriesInput {
+  readonly availableExtensions: readonly WorkbenchExtensionDescription[];
   readonly extensionRegistry: ExtensionRegistry;
   readonly installedRecords: readonly InstalledExtensionRecord[];
 }
@@ -42,38 +36,44 @@ export interface ExtensionInstallPlanningContext extends CreateExtensionManageme
 }
 
 export function createExtensionManagementEntries({
+  availableExtensions,
   extensionRegistry,
   installedRecords,
 }: CreateExtensionManagementEntriesInput): readonly ExtensionManagementEntry[] {
   const installedById = new Map(installedRecords.map((record) => [record.id, record]));
-  const extensionFeatures = createExtensionManagementFeatureMaps(extensionRegistry);
-  const bundledEntries = BUNDLED_EXTENSIONS.map((extension) => {
-    const installed = installedById.get(extension.manifest.id);
-    const isBuiltin = extension.manifest.id.startsWith('workbench-kit.builtin.');
-    const featureState = resolveExtensionManagementFeatureState(
-      extension.manifest.id,
-      extensionFeatures,
-    );
+  const extensionFeatures = createExtensionManagementFeatureMaps(
+    availableExtensions,
+    extensionRegistry,
+  );
+  const bundledEntries = availableExtensions
+    .map((extension) => {
+      const installed = installedById.get(extension.manifest.id);
+      const isBuiltin = extension.manifest.id.startsWith('workbench-kit.builtin.');
+      const featureState = resolveExtensionManagementFeatureState(
+        extension.manifest.id,
+        extensionFeatures,
+      );
 
-    return {
-      category: installed?.category ?? (isBuiltin ? 'builtin' : 'sample'),
-      description: extension.manifest.displayName,
-      diagnostics: featureState.diagnostics,
-      displayName: extension.manifest.displayName,
-      enabled: isBuiltin ? true : (installed?.enabled ?? false),
-      features: featureState.features,
-      id: extension.manifest.id,
-      installedAt: installed?.installedAt,
-      manifestUrl: installed?.manifestUrl,
-      source: isBuiltin ? ('bundled' as const) : ('installed' as const),
-    } satisfies ExtensionManagementEntry;
-  }).filter((entry) => entry.source === 'bundled' || installedById.has(entry.id));
+      return {
+        category: installed?.category ?? (isBuiltin ? 'builtin' : 'sample'),
+        description: extension.manifest.displayName,
+        diagnostics: featureState.diagnostics,
+        displayName: extension.manifest.displayName,
+        enabled: isBuiltin ? true : (installed?.enabled ?? false),
+        features: featureState.features,
+        id: extension.manifest.id,
+        installedAt: installed?.installedAt,
+        manifestUrl: installed?.manifestUrl,
+        source: isBuiltin ? ('bundled' as const) : ('installed' as const),
+      } satisfies ExtensionManagementEntry;
+    })
+    .filter((entry) => entry.source === 'bundled' || installedById.has(entry.id));
 
   const activeExtensions = extensionRegistry
     .getExtensions()
     .filter(
       (extension) =>
-        !BUNDLED_EXTENSIONS.some((bundled) => bundled.manifest.id === extension.manifest.id),
+        !availableExtensions.some((bundled) => bundled.manifest.id === extension.manifest.id),
     )
     .map((extension) => {
       const installed = installedById.get(extension.manifest.id);
@@ -102,12 +102,14 @@ export function createExtensionManagementEntries({
 }
 
 export function createExtensionCatalogBrowseEntries({
+  availableExtensions,
   catalogEntries,
   extensionRegistry,
   installedRecords,
 }: CreateExtensionCatalogBrowseEntriesInput): readonly ExtensionCatalogBrowseEntry[] {
   const installedIds = new Set(installedRecords.map((record) => record.id));
   const installContext = createExtensionInstallPlanningContext({
+    availableExtensions,
     catalogEntries,
     extensionRegistry,
     installedRecords,
@@ -135,23 +137,24 @@ export function createExtensionCatalogBrowseEntries({
 }
 
 export function createExtensionInstallPlanningContext({
+  availableExtensions,
   catalogEntries,
   extensionRegistry,
   installedRecords,
 }: CreateExtensionCatalogBrowseEntriesInput): ExtensionInstallPlanningContext {
-  const availableExtensions = mergeUniqueExtensionDescriptions([
-    ...BUNDLED_EXTENSIONS,
+  const installableExtensions = mergeUniqueExtensionDescriptions([
+    ...availableExtensions,
     ...extensionRegistry.getExtensions(),
   ]);
 
   return {
-    availableExtensions,
+    availableExtensions: installableExtensions,
     enabledExtensionIds: extensionRegistry
       .getExtensions()
       .map((extension) => extension.manifest.id),
     extensionRegistry,
     hostCapabilityIds: extensionRegistry.capabilityRegistry.listProviderIds(),
-    installSources: createExtensionInstallSources(catalogEntries, availableExtensions),
+    installSources: createExtensionInstallSources(catalogEntries, installableExtensions),
     installedRecords,
   };
 }
@@ -215,10 +218,13 @@ function mergeUniqueExtensionDescriptions(
   return [...byId.values()];
 }
 
-function createExtensionManagementFeatureMaps(extensionRegistry: ExtensionRegistry) {
+function createExtensionManagementFeatureMaps(
+  availableExtensions: readonly WorkbenchExtensionDescription[],
+  extensionRegistry: ExtensionRegistry,
+) {
   return {
     bundledFeaturesById: new Map(
-      BUNDLED_EXTENSIONS.map((extension) => [
+      availableExtensions.map((extension) => [
         extension.manifest.id,
         createExtensionFeatureSpec(extension),
       ]),

package/src/extensions/use-extension-management.ts

@@ -51,7 +51,7 @@ export function useExtensionManagementModel({
   installTrustStorageKey = DEFAULT_EXTENSION_INSTALL_TRUST_STORAGE_KEY,
 }: UseExtensionManagementModelOptions = {}) {
   const workbench = useWorkbench();
-  const { extensionRegistry } = workbench;
+  const { availableExtensions, extensionRegistry } = workbench;
   const resolvedInstalledExtensionsStorage =
     installedExtensionsStorage ?? workbench.installedExtensionsStorage;
   const resolvedInstalledExtensionsStorageKey =
@@ -138,20 +138,26 @@ export function useExtensionManagementModel({
   }, [catalogTrustPolicy, catalogUrl]);
 
   const installedEntries = useMemo<readonly ExtensionManagementEntry[]>(() => {
-    return createExtensionManagementEntries({ extensionRegistry, installedRecords });
-  }, [extensionRegistry, installedRecords]);
+    return createExtensionManagementEntries({
+      availableExtensions,
+      extensionRegistry,
+      installedRecords,
+    });
+  }, [availableExtensions, extensionRegistry, installedRecords]);
 
   const browseEntries = useMemo<readonly ExtensionCatalogBrowseEntry[]>(() => {
     return createExtensionCatalogBrowseEntries({
+      availableExtensions,
       catalogEntries,
       extensionRegistry,
       installedRecords,
     });
-  }, [catalogEntries, extensionRegistry, installedRecords]);
+  }, [availableExtensions, catalogEntries, extensionRegistry, installedRecords]);
 
   const installCatalogEntry = useCallback(
     (entry: ExtensionCatalogBrowseEntry, options?: ExtensionInstallOptions) => {
       const installContext = createExtensionInstallPlanningContext({
+        availableExtensions,
         catalogEntries,
         extensionRegistry,
         installedRecords,
@@ -186,6 +192,7 @@ export function useExtensionManagementModel({
       }
     },
     [
+      availableExtensions,
       catalogEntries,
       extensionRegistry,
       installedRecords,

package/src/index.ts

@@ -1,5 +1,7 @@
 export const WORKBENCH_KIT_WORKBENCH_REACT_VERSION = '0.0.0' as const;
 
+export { BUILTIN_WORKBENCH_EXTENSIONS } from './extensions/builtin-extensions.js';
+
 export {
   WorkbenchProvider,
   useWorkbench,

package/src/shell/provider.tsx

@@ -9,7 +9,6 @@ import {
   type ReactNode,
 } from 'react';
 import {
-  BUILTIN_WORKBENCH_EXTENSIONS,
   collectConfigurationDefaults,
   createEditorService,
   ExtensionRegistry,
@@ -21,7 +20,6 @@ import {
   registerHostWorkbenchThemes,
   resolveInstalledAvailableExtensions,
   resolveWorkbenchExtensions,
-  SAMPLE_WORKBENCH_EXTENSIONS,
   DEFAULT_INSTALLED_EXTENSIONS_STORAGE_KEY,
   WORKBENCH_EDITOR_SERVICE_CAPABILITY_ID,
   verifyWorkbenchExtensionsAgainstLock,
@@ -36,6 +34,7 @@ import {
   type WorkbenchLayoutStateInput,
   type ViewHostFactory,
 } from '@workbench-kit/workbench-core';
+import { BUILTIN_WORKBENCH_EXTENSIONS } from '../extensions/builtin-extensions.js';
 import {
   ContextKeyService,
   createWorkbenchPermissionContextKeys,
@@ -97,10 +96,7 @@ export interface WorkbenchWorkspaceHostPort extends WorkbenchEditorSavePort {
 
 export type { WorkbenchStorageAdapter };
 
-const DEFAULT_AVAILABLE_EXTENSIONS = [
-  ...BUILTIN_WORKBENCH_EXTENSIONS,
-  ...SAMPLE_WORKBENCH_EXTENSIONS,
-] as const;
+const DEFAULT_AVAILABLE_EXTENSIONS = BUILTIN_WORKBENCH_EXTENSIONS;
 
 const EMPTY_HOST_THEMES: readonly WorkbenchHostThemeRegistration[] = Object.freeze([]);
 const EMPTY_USER_COMMANDS: readonly WorkbenchUserCommandDefinition[] = Object.freeze([]);
@@ -151,6 +147,7 @@ export interface WorkbenchProviderProps {
 
 export interface WorkbenchContextValue {
   activateCommand(commandId: string): Promise<readonly { readonly extensionId: string }[]>;
+  availableExtensions: readonly WorkbenchExtensionDescription[];
   contextKeyService: ContextKeyService;
   editorDocumentViewProviders: EditorDocumentViewProviderRegistry;
   editorService: EditorService;
@@ -170,6 +167,7 @@ export interface WorkbenchContextValue {
 
 interface WorkbenchProviderServices {
   activateStartup(): void;
+  availableExtensions: readonly WorkbenchExtensionDescription[];
   dispose(): void;
   editorDocumentViewProviders: EditorDocumentViewProviderRegistry;
   editorService: EditorService;
@@ -429,6 +427,7 @@ export function WorkbenchProvider({
       activateStartup: () => {
         void ensureStartupActivation();
       },
+      availableExtensions: resolvedAvailableExtensions,
       dispose: () => {
         saveCommandDisposable?.dispose();
         userCommandDisposables.dispose();
@@ -565,6 +564,7 @@ export function WorkbenchProvider({
   const value = useMemo<WorkbenchContextValue>(
     () => ({
       activateCommand: (commandId) => services.extensionRegistry.activateCommand(commandId),
+      availableExtensions: services.availableExtensions,
       contextKeyService,
       editorDocumentViewProviders: services.editorDocumentViewProviders,
       editorService: services.editorService,