From ee5c9708344db67dcd88a5b891aa86c287c13057 Mon Sep 17 00:00:00 2001
From: Abhinav Prakash <abhinav.prakash319@gmail.com>
Date: Wed, 10 Jul 2024 21:06:58 +0530
Subject: [PATCH] fix::> Changed the implementation Remove to broadcast the
 message

---
 server/server.go | 63 +++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 62 insertions(+), 1 deletion(-)

diff --git a/server/server.go b/server/server.go
index c473441..961441a 100644
--- a/server/server.go
+++ b/server/server.go
@@ -32,6 +32,11 @@ type MessageGetFile struct {
 	Key string
 }
 
+type MessageDeleteFile struct {
+	ID  string
+	Key string
+}
+
 type FileServerOpts struct {
 	ID                string
 	EncKey            []byte
@@ -116,6 +121,37 @@ func (s *FileServer) Get(key string) (io.Reader, error) {
 }
 
 func (s *FileServer) Remove(key string) error {
+	if !s.store.Has(s.ID, key) {
+		fmt.Printf("[%s] The file (%s) is not present in the disk.\n", s.Transport.ListenAddress(), key)
+		return fmt.Errorf("file is not present")
+	}
+
+	fmt.Printf("[%s] File (%s) found locally, deleting it...\n", s.Transport.ListenAddress(), key)
+
+	// Message to be broadcasted.
+	msg := Message{
+		Payload: MessageDeleteFile{
+			ID:  s.ID,
+			Key: cryptography.HashKey(key),
+		},
+	}
+
+	// Broadcasting the message
+	if err := s.broadcast(&msg); err != nil {
+		return err
+	}
+
+	time.Sleep(time.Millisecond * 500)
+
+	for _, peer := range s.peers {
+		fmt.Printf("[%s] receiving DELETE stream from peer: [%s]", s.Transport.ListenAddress(), peer.RemoteAddr())
+
+		if err := s.store.Delete(s.ID, key); err != nil {
+			return err
+		}
+	}
+
+	fmt.Printf("[%s] File (%s) DELETED.\n", s.Transport.ListenAddress(), key)
 	return s.store.Delete(s.ID, key)
 }
 
@@ -304,6 +340,10 @@ func (s *FileServer) handleMessage(from string, msg *Message) error {
 	case *MessageGetFile:
 		fmt.Println("Received MessageGetFile")
 		return s.handleMessageGetFile(from, t)
+
+	case *MessageDeleteFile:
+		fmt.Println("Received MessageDeleteFile")
+		return s.handleMessageDeleteFile(from, t)
 	}
 	return nil
 }
@@ -320,7 +360,7 @@ func (s *FileServer) handleMessageStoreFile(from string, msg *MessageStoreFile)
 		return err
 	}
 
-	log.Printf("[%s] Written (%d) bytes to disk\n", s.Transport.ListenAddress(), n)
+	fmt.Printf("[%s] Written (%d) bytes to disk\n", s.Transport.ListenAddress(), n)
 
 	peer.CloseStream() // Streaming is OVER!!
 
@@ -364,7 +404,28 @@ func (s *FileServer) handleMessageGetFile(from string, msg *MessageGetFile) erro
 	return nil
 }
 
+func (s *FileServer) handleMessageDeleteFile(from string, msg *MessageDeleteFile) error {
+	fmt.Printf("Received Message: %v\n", msg)
+
+	if !s.store.Has(msg.ID, msg.Key) {
+		return fmt.Errorf("[%s] file (%s) not found", s.Transport.ListenAddress(), msg.Key)
+	}
+
+	_, ok := s.peers[from]
+	if !ok {
+		return fmt.Errorf("peer {%s} not found", from)
+	}
+
+	if err := s.store.Delete(msg.ID, msg.Key); err != nil {
+		return err
+	}
+
+	fmt.Printf("[%s] Deleted file (%s) from disk\n", s.Transport.ListenAddress(), msg.Key)
+	return nil
+}
+
 func init() {
 	gob.Register(&MessageStoreFile{})
 	gob.Register(&MessageGetFile{})
+	gob.Register(&MessageDeleteFile{})
 }