From 7a1c344964abdc9826364b2ae37c5c4fd31a6ed4 Mon Sep 17 00:00:00 2001
From: Justin Hawkins <justin@hawkins.id.au>
Date: Sat, 15 Jan 2022 16:47:06 +1030
Subject: [PATCH] Make it possible to retrieve a particular file (not
 implemented on client yet)

---
 client.go |  9 ++++----
 main.go   |  1 -
 server.go | 67 +++++++++++++++++++++++++++++++++++++++++++++++++------
 3 files changed, 64 insertions(+), 13 deletions(-)

diff --git a/client.go b/client.go
index 81f72c9..f3fcb02 100644
--- a/client.go
+++ b/client.go
@@ -83,7 +83,7 @@ func (c *Client) Connect() error {
 		}
 
 		req := secure.PacketReceiveDataStartRequest{
-			Id: 234,
+			Id: 0, // 0 means last? Change to do a fetch?
 		}
 		err = enc.Encode(req)
 		if err != nil {
@@ -96,7 +96,6 @@ func (c *Client) Connect() error {
 			panic(err)
 		}
 
-		log.Printf("server said %v", res)
 		if res.Status == secure.ReceiveDataStartResponseOK {
 			for {
 				res := secure.PacketReceiveDataNext{}
@@ -104,16 +103,16 @@ func (c *Client) Connect() error {
 				if err != nil {
 					panic(err)
 				}
-				// log.Printf("got %d bytes, last is %t", res.Size, res.Last)
-				// print(res.Data)
 				os.Stdout.Write(res.Data[:res.Size])
 				if res.Last {
 					break
 				}
 			}
 			log.Printf("finished")
+		} else if res.Status == secure.ReceiveDataStartResponseNotFound {
+			log.Printf("ngf not found")
 		} else {
-			panic("don't handle this yet")
+			panic("unknown status")
 		}
 
 		conn.Close()
diff --git a/main.go b/main.go
index 658afe9..f23135e 100644
--- a/main.go
+++ b/main.go
@@ -21,7 +21,6 @@ func main() {
 		s := Server{port: *port}
 		s.Run()
 	} else {
-		log.Printf("Client running on %d\n", *port)
 		c := Client{port: *port, address: *addr, list: *isList, receive: *isReceive}
 		err := c.Connect()
 		if err != nil {
diff --git a/server.go b/server.go
index feb027e..3c455ad 100644
--- a/server.go
+++ b/server.go
@@ -8,6 +8,7 @@ import (
 	"log"
 	"net"
 	"os"
+	"os/signal"
 	"sync/atomic"
 	"time"
 
@@ -27,6 +28,7 @@ type NGF struct {
 	Filename  string // could be empty string if we were not supplied with one
 	Kind      string //
 	Size      uint64 // file size
+	Timestamp time.Time
 }
 
 var ngfs []NGF
@@ -44,6 +46,23 @@ func (s *Server) Run() {
 
 	ngfs = make([]NGF, 0)
 
+	go func() {
+		sigchan := make(chan os.Signal)
+		signal.Notify(sigchan, os.Interrupt)
+		<-sigchan
+
+		for _, ngf := range ngfs {
+			log.Printf("removing file: %s", ngf.StorePath)
+			err := os.Remove(ngf.StorePath)
+			if err != nil {
+				log.Printf("could not remove %s: %v", ngf.StorePath, err)
+			}
+		}
+		os.Exit(0)
+	}()
+
+	// start main program tasks
+
 	for {
 		conn, err := listener.AcceptTCP()
 
@@ -106,6 +125,7 @@ func handleConnection(conn *net.TCPConn) {
 			Kind:      "",
 			Size:      0,
 			Id:        atomic.AddUint32(&globalId, 1),
+			Timestamp: time.Now(),
 		}
 
 		if err != nil {
@@ -158,25 +178,58 @@ func handleConnection(conn *net.TCPConn) {
 		}
 
 		log.Printf("The asked for %v", req)
+
+		// do we have this ngf by id?
+		var requestedNGF *NGF
+
+		if len(ngfs) > 0 {
+			if req.Id == 0 {
+				// they want the most recent one
+				requestedNGF = &ngfs[len(ngfs)-1]
+			} else {
+				for _, ngf := range ngfs {
+					if ngf.Id == req.Id {
+						requestedNGF = &ngf
+					}
+				}
+			}
+		}
+
+		if requestedNGF == nil {
+			// not found
+			log.Printf("user requested %d, not found", req.Id)
+			res := secure.PacketReceiveDataStartResponse{
+				Status: secure.ReceiveDataStartResponseNotFound,
+			}
+			err = enc.Encode(res)
+			if err != nil {
+				log.Printf("could not send NotFound: %v", err)
+			}
+
+			return
+		}
+
 		res := secure.PacketReceiveDataStartResponse{
 			Status:    secure.ReceiveDataStartResponseOK,
-			Filename:  "abcdef",
-			Kind:      "",
-			TotalSize: 12340,
+			Filename:  requestedNGF.Filename,
+			Kind:      requestedNGF.Kind,
+			TotalSize: uint32(requestedNGF.Size),
 		}
 		err = enc.Encode(res)
 		if err != nil {
 			log.Printf("error sending PacketReceiveDataStartResponse: %v", err)
 			return
 		}
-
-		log.Printf("%#v", ngfs)
-
 		// now just start sending the file in batches
 		buf := make([]byte, 2048)
-		filename := ngfs[0].StorePath
+		filename := requestedNGF.StorePath
 		log.Printf("opening %s", filename)
 		f, err := os.Open(filename)
+		if err != nil {
+			log.Printf("could not find file %s: %v", filename, err)
+			return
+		}
+
 		for {
 			n, err := f.Read(buf)
 			eof := false