From ccc95f15ef1c1facf977fb5de4f022f29b5e79ae Mon Sep 17 00:00:00 2001
From: Joren <joren.schipman@student.howest.be>
Date: Thu, 2 May 2024 15:12:11 +0200
Subject: [PATCH] add END_OF_DATA

---
 app/src/main/java/com/ti/m/GoodSoftware.kt | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/app/src/main/java/com/ti/m/GoodSoftware.kt b/app/src/main/java/com/ti/m/GoodSoftware.kt
index fff5978..4eda3d3 100644
--- a/app/src/main/java/com/ti/m/GoodSoftware.kt
+++ b/app/src/main/java/com/ti/m/GoodSoftware.kt
@@ -320,7 +320,7 @@ class GoodSoftware (private val activity: MainActivity) {
                 reader.readLine()
             }
 
-            writer.println("END_OF_COMMUNICATION")
+            writer.println("END_OF_DATA")
             reader.readLine()
 
             println("Client: Ready for next operation")
@@ -337,6 +337,19 @@ class GoodSoftware (private val activity: MainActivity) {
         }
     }
 
+    fun disconnect(connectionResult: ConnectionResult) {
+        val (socket, reader, writer, _, _, _) = connectionResult
+        try {
+            writer.println("END_OF_COMMUNICATION")
+            reader.readLine()
+            writer.close()
+            reader.close()
+            socket.close()
+        } catch (e: Exception) {
+            e.printStackTrace()
+        }
+    }
+
     suspend fun takeBeautifulPicture(context: Context, lifecycleOwner: LifecycleOwner) {
         // Ensure that cameraProvider is initialized
         cameraProvider = ProcessCameraProvider.getInstance(context).await()