feat: allow to override startupKind on lifecycle service

Author: Loïc Mangeonjean

src/service-override/lifecycle.ts

@@ -1,13 +1,48 @@
 import { type IEditorOverrideServices } from 'vs/editor/standalone/browser/standaloneServices'
 import { SyncDescriptor } from 'vs/platform/instantiation/common/descriptors'
+import { ILogService } from 'vs/platform/log/common/log.service'
+import { IStorageService } from 'vs/platform/storage/common/storage.service'
 import { BrowserLifecycleService } from 'vs/workbench/services/lifecycle/browser/lifecycleService'
+import { StartupKind } from 'vs/workbench/services/lifecycle/common/lifecycle'
 import { ILifecycleService } from 'vs/workbench/services/lifecycle/common/lifecycle.service'
 import { TimerService } from 'vs/workbench/services/timer/browser/timerService'
 import { ITimerService } from 'vs/workbench/services/timer/browser/timerService.service'
 
-export default function getServiceOverride(): IEditorOverrideServices {
+interface StartupKindResolver {
+  (resolveDefault: () => StartupKind | undefined): StartupKind | undefined
+}
+export interface LifecycleServiceParams {
+  /**
+   * Allows to force the startup kind
+   */
+  resolveStartupKind?: StartupKindResolver
+}
+
+const DEFAULT_RESOLVE_STARTUP_KIND: StartupKindResolver = (resolveDefault) => resolveDefault()
+
+class BrowserLifecycleServiceOverride extends BrowserLifecycleService {
+  constructor(
+    private params: LifecycleServiceParams,
+    @ILogService logService: ILogService,
+    @IStorageService storageService: IStorageService
+  ) {
+    super(logService, storageService)
+  }
+
+  protected override doResolveStartupKind(): StartupKind | undefined {
+    return (this.params.resolveStartupKind ?? DEFAULT_RESOLVE_STARTUP_KIND)(() =>
+      super.doResolveStartupKind()
+    )
+  }
+}
+
+export default function getServiceOverride(
+  options: LifecycleServiceParams = {}
+): IEditorOverrideServices {
   return {
-    [ILifecycleService.toString()]: new SyncDescriptor(BrowserLifecycleService),
+    [ILifecycleService.toString()]: new SyncDescriptor(BrowserLifecycleServiceOverride, [options]),
     [ITimerService.toString()]: new SyncDescriptor(TimerService)
   }
 }
+
+export { StartupKind }