From 8e3a433afb888a80f7224a989ea5efcd9db2f01f Mon Sep 17 00:00:00 2001
From: Zlatin Balevsky <zlatinb@gmail.com>
Date: Sun, 8 Mar 2020 16:06:28 +0000
Subject: [PATCH] persist shared file on publish/unpublish

---
 core/src/main/groovy/com/muwire/core/Core.groovy      |  4 ++++
 .../core/filefeeds/UIFileUnpublishedEvent.groovy      |  3 ++-
 .../muwire/core/files/PersisterFolderService.groovy   | 11 +++++++++++
 3 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/core/src/main/groovy/com/muwire/core/Core.groovy b/core/src/main/groovy/com/muwire/core/Core.groovy
index e60a26cd..9409725b 100644
--- a/core/src/main/groovy/com/muwire/core/Core.groovy
+++ b/core/src/main/groovy/com/muwire/core/Core.groovy
@@ -32,6 +32,8 @@ import com.muwire.core.filecert.CertificateManager
 import com.muwire.core.filecert.UICreateCertificateEvent
 import com.muwire.core.filecert.UIFetchCertificatesEvent
 import com.muwire.core.filecert.UIImportCertificateEvent
+import com.muwire.core.filefeeds.UIFIlePublishedEvent
+import com.muwire.core.filefeeds.UIFileUnpublishedEvent
 import com.muwire.core.files.FileDownloadedEvent
 import com.muwire.core.files.FileHashedEvent
 import com.muwire.core.files.FileHasher
@@ -269,6 +271,8 @@ public class Core {
         eventBus.register(FileHashedEvent.class, persisterFolderService)
         eventBus.register(FileUnsharedEvent.class, persisterFolderService)
         eventBus.register(UICommentEvent.class, persisterFolderService)
+        eventBus.register(UIFIlePublishedEvent.class, persisterFolderService)
+        eventBus.register(UIFileUnpublishedEvent.class, persisterFolderService)
 
         log.info("initializing host cache")
         File hostStorage = new File(home, "hosts.json")
diff --git a/core/src/main/groovy/com/muwire/core/filefeeds/UIFileUnpublishedEvent.groovy b/core/src/main/groovy/com/muwire/core/filefeeds/UIFileUnpublishedEvent.groovy
index b6fc47e2..ee6f5ce0 100644
--- a/core/src/main/groovy/com/muwire/core/filefeeds/UIFileUnpublishedEvent.groovy
+++ b/core/src/main/groovy/com/muwire/core/filefeeds/UIFileUnpublishedEvent.groovy
@@ -1,7 +1,8 @@
 package com.muwire.core.filefeeds
 
+import com.muwire.core.Event
 import com.muwire.core.SharedFile
 
-class UIFileUnpublishedEvent {
+class UIFileUnpublishedEvent extends Event {
     SharedFile sf
 }
diff --git a/core/src/main/groovy/com/muwire/core/files/PersisterFolderService.groovy b/core/src/main/groovy/com/muwire/core/files/PersisterFolderService.groovy
index b2899f05..0592f9f9 100644
--- a/core/src/main/groovy/com/muwire/core/files/PersisterFolderService.groovy
+++ b/core/src/main/groovy/com/muwire/core/files/PersisterFolderService.groovy
@@ -1,6 +1,9 @@
 package com.muwire.core.files
 
 import com.muwire.core.*
+import com.muwire.core.filefeeds.UIFIlePublishedEvent
+import com.muwire.core.filefeeds.UIFileUnpublishedEvent
+
 import groovy.json.JsonOutput
 import groovy.json.JsonSlurper
 import groovy.util.logging.Log
@@ -92,6 +95,14 @@ class PersisterFolderService extends BasePersisterService {
     void onUICommentEvent(UICommentEvent e) {
         persistFile(e.sharedFile,null)
     }
+    
+    void onUIFilePublishedEvent(UIFIlePublishedEvent e) {
+        persistFile(e.sf, null)
+    }
+    
+    void onUIFileUnpublishedEvent(UIFileUnpublishedEvent e) {
+        persistFile(e.sf, null)
+    }
 
     void load() {
         log.fine("Loading...")