diff --git a/webui/src/main/java/com/muwire/webui/FileManager.java b/webui/src/main/java/com/muwire/webui/FileManager.java index d5a26208..6d62efb7 100644 --- a/webui/src/main/java/com/muwire/webui/FileManager.java +++ b/webui/src/main/java/com/muwire/webui/FileManager.java @@ -61,6 +61,13 @@ public class FileManager { } } + public void onFileUnsharedEvent(FileUnsharedEvent e) { + if (!e.getDeleted()) + return; + fileTree.remove(e.getUnsharedFile().getFile()); + revision++; + } + void list(File parent, FileListCallback callback) { fileTree.list(parent, callback); } diff --git a/webui/src/main/java/com/muwire/webui/MuWireClient.java b/webui/src/main/java/com/muwire/webui/MuWireClient.java index 4fc32ac7..dd34b93e 100644 --- a/webui/src/main/java/com/muwire/webui/MuWireClient.java +++ b/webui/src/main/java/com/muwire/webui/MuWireClient.java @@ -34,6 +34,7 @@ import com.muwire.core.files.FileHashedEvent; import com.muwire.core.files.FileHashingEvent; import com.muwire.core.files.FileLoadedEvent; import com.muwire.core.files.FileSharedEvent; +import com.muwire.core.files.FileUnsharedEvent; import com.muwire.core.search.BrowseStatusEvent; import com.muwire.core.search.UIResultBatchEvent; import com.muwire.core.search.UIResultEvent; @@ -150,6 +151,7 @@ public class MuWireClient { core.getEventBus().register(FileHashedEvent.class, fileManager); core.getEventBus().register(FileDownloadedEvent.class, fileManager); core.getEventBus().register(FileHashingEvent.class, fileManager); + core.getEventBus().register(FileUnsharedEvent.class, fileManager); BrowseManager browseManager = new BrowseManager(core); core.getEventBus().register(BrowseStatusEvent.class, browseManager);