Use watchfiles stop_event

plugins/contents/fps_contents/fileid.py

@@ -40,6 +40,8 @@ def __init__(self, db_path: str = "fileid.db"):
         self.initialized = asyncio.Event()
         self.watchers = {}
         self.watch_files_task = asyncio.create_task(self.watch_files())
+        self.stop_watching_files = asyncio.Event()
+        self.stopped_watching_files = asyncio.Event()
         self.lock = asyncio.Lock()
 
     async def get_id(self, path: str) -> Optional[str]:
@@ -68,11 +70,11 @@ async def index(self, path: str) -> Optional[str]:
                 if not await apath.exists():
                     return None
 
-            idx = uuid4().hex
-            mtime = (await apath.stat()).st_mtime
-            await db.execute("INSERT INTO fileids VALUES (?, ?, ?)", (idx, path, mtime))
-            await db.commit()
-            return idx
+                idx = uuid4().hex
+                mtime = (await apath.stat()).st_mtime
+                await db.execute("INSERT INTO fileids VALUES (?, ?, ?)", (idx, path, mtime))
+                await db.commit()
+                return idx
 
     async def watch_files(self):
         async with self.lock:
@@ -96,7 +98,7 @@ async def watch_files(self):
                 await db.commit()
                 self.initialized.set()
 
-        async for changes in awatch("."):
+        async for changes in awatch(".", stop_event=self.stop_watching_files):
             async with self.lock:
                 async with aiosqlite.connect(self.db_path) as db:
                     deleted_paths = []
@@ -156,6 +158,8 @@ async def watch_files(self):
                 for watcher in self.watchers.get(changed_path, []):
                     watcher.notify(change)
 
+        self.stopped_watching_files.set()
+
     def watch(self, path: str) -> Watcher:
         watcher = Watcher(path)
         self.watchers.setdefault(path, []).append(watcher)

plugins/yjs/fps_yjs/main.py

@@ -25,4 +25,5 @@ async def start(
 
         yield
 
-        contents.file_id_manager.watch_files_task.cancel()
+        contents.file_id_manager.stop_watching_files.set()
+        await contents.file_id_manager.stopped_watching_files.wait()