chore: try to type context as cell

Author: Gozala

deno.lock

@@ -1,7 +1,7 @@
 import { css, html, LitElement, PropertyValues } from "lit";
 import { createRef, Ref, ref } from "lit/directives/ref.js";
 import * as IPC from "./ipc.ts";
-import { getIframeContextHandler } from "./context.ts";
+import { type Cell, getIframeContextHandler } from "./context.ts";
 import OuterFrame from "./outer-frame.ts";
 import {
   HealthCheck,
@@ -43,7 +43,7 @@ export class CommonIframeSandboxElement extends LitElement {
     crashed: { type: Boolean, attribute: false, state: true },
   };
   declare src: string;
-  declare context?: object;
+  declare context?: Cell<object>;
   declare crashed: boolean;
 
   constructor() {
@@ -354,7 +354,7 @@ export class CommonIframeSandboxElement extends LitElement {
     const IframeHandler = getIframeContextHandler();
     if (IframeHandler != null) {
       for (const [_, receipt] of this.subscriptions) {
-        IframeHandler.unsubscribe(this, this.context, receipt);
+        IframeHandler.unsubscribe(this, this.context!, receipt);
       }
       this.subscriptions.clear();
     }

iframe-sandbox/src/common-iframe-sandbox.ts

@@ -1,36 +1,39 @@
 import { TaskPerform } from "./ipc.ts";
 import { CommonIframeSandboxElement } from "./common-iframe-sandbox.ts";
+import type { Cell } from "@commontools/runner";
+
+export type { Cell };
 
 // An `IframeContextHandler` is used by consumers to
 // register how read/writing values from frames are handled.
-export interface IframeContextHandler {
+export interface IframeContextHandler<Doc extends object = object> {
   read(element: CommonIframeSandboxElement, context: any, key: string): any;
   write(
     element: CommonIframeSandboxElement,
-    context: any,
+    context: Cell<Doc>,
     key: string,
     value: any,
   ): void;
   subscribe(
     element: CommonIframeSandboxElement,
-    context: any,
+    context: Cell<Doc>,
     key: string,
     callback: (key: string, value: any) => void,
     doNotSendMyDataBack: boolean,
   ): any;
   unsubscribe(
     element: CommonIframeSandboxElement,
-    context: any,
+    context: Cell<Doc>,
     receipt: any,
   ): void;
   onLLMRequest(
     element: CommonIframeSandboxElement,
-    context: any,
+    context: Cell<Doc>,
     payload: string,
   ): Promise<object>;
   onReadWebpageRequest(
     element: CommonIframeSandboxElement,
-    context: any,
+    context: Cell<Doc>,
     payload: string,
   ): Promise<object>;