refactor(Cocoon/Service): Standardize service definitions with Effect-TS class-based Context

- Migrate service interfaces to class-based `Context.Tag` implementations across IPC, Cancellation, and core services
- Reorganize service layers into dedicated namespaces (e.g. `Cancellation`, `Client`) for improved architectural clarity
- Update import paths and references to align with new service organization
- Simplify layer composition in ExtensionHost through streamlined `Layer.mergeAll` usage

This refactoring completes the Effect-TS service pattern adoption in Cocoon's core, strengthening type safety and dependency management for the VS Code extension host implementation. The changes directly support the `Vine` gRPC contract stability by ensuring proper cancellation token handling and IPC client lifecycle management.

Author: NikolaRHristov

Source/Core/APIFactory/Service.ts

@@ -26,4 +26,7 @@ export interface Interface {
 /**
  * The `Context.Tag` for the `APIFactory` service.
  */
-export const Tag = Context.Tag<Interface>("Core/APIFactory");
+export class APIFactory extends Context.Tag("Core/APIFactory")<
+	APIFactory,
+	Interface
+>() {}

Source/Core/ExtensionHost.ts

@@ -18,11 +18,5 @@ import { Tag } from "./ExtensionHost/Service.js";
  * It depends on the APIFactory, logging, IPC, and initialization data services.
  */
 export const Live = Layer.effect(Tag, Definition).pipe(
-	Layer.provide(
-		Layer.mergeAll(
-			APIFactory.Live,
-			Log.Live,
-			IPC.Live,
-		),
-	),
+	Layer.provide(Layer.mergeAll(APIFactory.Live, Log.Live, IPC.Live)),
 );

Source/Service/Cancellation.ts

@@ -8,14 +8,19 @@
 import { Layer } from "effect";
 
 import { Definition } from "./Cancellation/Definition.js";
-import { Tag } from "./Cancellation/Service.js";
+import { Cancellation as CancellationTag } from "./Cancellation/Service.js";
 
 export type { InvalidTokenIDError } from "./Cancellation/Error.js";
-export { Tag, type Interface } from "./Cancellation/Service.js";
+
 export type { TokenAndScope } from "./Cancellation/Type.js";
 
-/**
- * The live implementation Layer for the CancellationTokenProvider service.
- * This is a self-contained layer with no external dependencies.
- */
-export const Live = Layer.effect(Tag, Definition);
+export namespace Cancellation {
+	export const Tag = CancellationTag;
+	export type Interface = CancellationTag;
+
+	/**
+	 * The live implementation Layer for the CancellationTokenProvider service.
+	 * This is a self-contained layer with no external dependencies.
+	 */
+	export const Live = Layer.effect(Tag, Definition);
+}

Source/Service/Cancellation/Service.ts

@@ -4,40 +4,37 @@
  * This service manages cancellation tokens for long-running RPC operations.
  */
 
-import { Context, type Effect } from "effect";
+import { Context, Scope, type Effect } from "effect";
 
 import type { InvalidTokenIDError } from "./Error.js";
 import type { TokenAndScope } from "./Type.js";
 
-/**
- * The service interface for the cancellation token provider.
- */
-export interface Interface {
-	/**
-	 * Acquires a CancellationToken for a given operation ID.
-	 * This returns a scoped Effect. When the scope is closed, the token
-	 * source is automatically disposed and cleaned up.
-	 * @param TokenID The numeric ID for the operation.
-	 * @returns A scoped `Effect` that resolves to a `TokenAndScope` object.
-	 */
-	readonly ObtainToken: (
-		TokenID: number,
-	) => Effect.Effect<TokenAndScope, InvalidTokenIDError, Effect.Scope>;
-
-	/**
-	 * Signals cancellation for a specific token ID.
-	 * @param TokenID The numeric ID of the operation to cancel.
-	 */
-	readonly CancelToken: (TokenID: number) => Effect.Effect<void, never>;
+export class Cancellation extends Context.Tag(
+	"Service/CancellationTokenProvider",
+)<
+	Cancellation,
+	{
+		/**
+		 * Acquires a CancellationToken for a given operation ID.
+		 * This returns a scoped Effect. When the scope is closed, the token
+		 * source is automatically disposed and cleaned up.
+		 * @param TokenID The numeric ID for the operation.
+		 * @returns A scoped `Effect` that resolves to a `TokenAndScope` object.
+		 */
+		readonly ObtainToken: (
+			TokenID: number,
+		) => Effect.Effect<TokenAndScope, InvalidTokenIDError, Scope.Scope>;
 
-	/**
-	 * Disposes of all currently managed cancellation token sources.
-	 * This is typically called during shutdown.
-	 */
-	readonly DisposeAll: () => Effect.Effect<void, never>;
-}
+		/**
+		 * Signals cancellation for a specific token ID.
+		 * @param TokenID The numeric ID of the operation to cancel.
+		 */
+		readonly CancelToken: (TokenID: number) => Effect.Effect<void, never>;
 
-/**
- * The Context.Tag for the CancellationTokenProvider service.
- */
-export const Tag = Context.Tag<Interface>("Service/CancellationTokenProvider");
+		/**
+		 * Disposes of all currently managed cancellation token sources.
+		 * This is typically called during shutdown.
+		 */
+		readonly DisposeAll: () => Effect.Effect<void, never>;
+	}
+>() {}

Source/Service/IPC/Client.ts

@@ -8,15 +8,20 @@
 import { Layer } from "effect";
 
 import { Acquire } from "./Client/Acquire.js";
-import { Tag, type Interface as ClientService } from "./Client/Service.js";
-import type { Configuration } from "./Configuration.js";
+import { Client as ClientTag } from "./Client/Service.js";
+import { Configuration } from "./Configuration.js";
 import type { gRPCConnectionError } from "./Error.js";
 
-/**
- * The live implementation `Layer` for the gRPC Client service.
- */
-export const Live: Layer.Layer<
-	ClientService,
-	gRPCConnectionError,
-	Configuration
-> = Layer.scoped(Tag, Acquire);
+export namespace Client {
+	export const Tag = ClientTag;
+	export type Interface = ClientTag;
+
+	/**
+	 * The live implementation `Layer` for the gRPC Client service.
+	 */
+	export const Live: Layer.Layer<
+		Interface,
+		gRPCConnectionError,
+		Configuration
+	> = Layer.scoped(Tag, Acquire);
+}

Source/Service/IPC/Client/Acquire.ts

@@ -16,8 +16,8 @@ import { Effect } from "effect";
 
 import { Configuration as ConfigurationService } from "../Configuration.js";
 import { gRPCConnectionError } from "../Error.js";
-import type { Interface as ClientService } from "./Service.js";
 import { Release } from "./Release.js";
+import type { Interface as ClientService } from "./Service.js";
 
 /**
  * An `Effect` that loads the gRPC `.proto` file definition from disk.

Source/Service/IPC/Client/Service.ts

@@ -10,17 +10,13 @@ import { Context } from "effect";
 // from the `vine.proto` file.
 import type { MountainService } from "../Generated.js";
 
-/**
- * The service interface for the raw, generated gRPC client.
- *
- * Higher-level IPC services will use this client to make calls to `Mountain`.
- */
-export type Interface = MountainService;
-
 /**
  * The `Context.Tag` for the gRPC client service.
  *
  * This tag is used by other services to declare their dependency on the raw
  * gRPC client.
  */
-export const Tag = Context.Tag<Interface>("IPC/Client");
+export class Client extends Context.Tag("IPC/Client")<
+	Client,
+	MountainService
+>() {}

Source/Service/IPC/Dispatcher.ts

@@ -6,7 +6,7 @@
 
 import { Layer } from "effect";
 
-import { Cancellation } from "../../Cancellation.js";
+import { Cancellation } from "../Cancellation.js";
 import { Definition } from "./Dispatcher/Definition.js";
 import { Dispatcher as DispatcherTag } from "./Dispatcher/Service.js";
 import { ProtocolAdapter } from "./ProtocolAdapter.js";

Source/Service/IPC/ProtocolAdapter.ts

@@ -7,20 +7,20 @@
 
 import { Layer } from "effect";
 
-import { Live as LiveClient } from "./Client.js";
+import { Client } from "./Client.js";
 import { Definition } from "./ProtocolAdapter/Definition.js";
-import { Tag, type Interface } from "./ProtocolAdapter/Service.js";
+import { ProtocolAdapter as ProtocolAdapterTag } from "./ProtocolAdapter/Service.js";
 
-// --- Public API Exports ---
-export { Tag, type Interface };
-
-// --- Live Implementation ---
-
-/**
- * The live implementation `Layer` for the `ProtocolAdapter` service.
- * It builds the adapter by composing its `Definition` with the
- * `LiveClient` layer it depends on for the underlying gRPC transport.
- */
-export const Live = Layer.effect(Tag, Definition).pipe(
-	Layer.provide(LiveClient),
-);
+export namespace ProtocolAdapter {
+	export const Tag = ProtocolAdapterTag;
+	export type Interface = ProtocolAdapterTag;
+	/**
+	 * The live implementation `Layer` for the `ProtocolAdapter` service.
+	 *
+	 * This layer builds the adapter by composing its `Definition` with the
+	 * `LiveClient` layer it depends on for the underlying gRPC transport.
+	 */
+	export const Live = Layer.effect(Tag, Definition).pipe(
+		Layer.provide(Client.Live),
+	);
+}

Source/Service/IPC/Server.ts

@@ -15,6 +15,7 @@ import { Server as ServerTag } from "./Server/Service.js";
 export namespace Server {
 	export const Tag = ServerTag;
 	export type Interface = ServerTag;
+
 	/**
 	 * The live implementation `Layer` for the gRPC Server service.
 	 *
@@ -25,7 +26,7 @@ export namespace Server {
 	 */
 	export const Live: Layer.Layer<
 		Interface,
-		gRPCConnectionError,
+		gRPCConnectionError | Error,
 		Configuration
 	> = Layer.scoped(Tag, Acquire).pipe(Layer.provide(Dispatcher.Live));
 }