Set shared document file_id

plugins/yjs/fps_yjs/routes.py

@@ -178,6 +178,7 @@ async def serve(self, websocket: YWebsocket, permissions) -> None:
                 file_path = await self.contents.file_id_manager.get_path(file_id)
                 logger.info(f"Opening collaboration room: {websocket.path} ({file_path})")
                 document = YDOCS.get(file_type, YFILE)(room.ydoc)
+                document.file_id = file_id
                 self.documents[websocket.path] = document
                 async with self.lock:
                     model = await self.contents.read_content(file_path, True, file_format)

plugins/yjs/fps_yjs/ydocs/ybasedoc.py

@@ -47,10 +47,18 @@ def dirty(self, value: bool) -> None:
     def path(self) -> Optional[str]:
         return self._ystate.get("path")
 
+    @property
+    def file_id(self) -> Optional[str]:
+        return self._ystate.get("file_id")
+
     @path.setter
     def path(self, value: str) -> None:
         self._ystate["path"] = value
 
+    @file_id.setter
+    def file_id(self, value: str) -> None:
+        self._ystate["file_id"] = value
+
     @abstractmethod
     def get(self) -> Any:
         ...

plugins/yjs/fps_yjs/ydocs/ynotebook.py

@@ -123,7 +123,7 @@ def set(self, value: Dict) -> None:
             # clear document
             self._ymeta.clear()
             self._ycells.clear()
-            for key in [k for k in self._ystate.keys() if k not in ("dirty", "path")]:
+            for key in [k for k in self._ystate.keys() if k not in ("dirty", "path", "file_id")]:
                 del self._ystate[key]
 
             # initialize document