Ping Pong and prefix list code

skp-github · skp-github · commit 3721aaed9d15 · 2023-05-26T12:19:00.000+02:00
diff --git a/app/src/main/java/com/googleresearch/capturesync/MainActivity.java b/app/src/main/java/com/googleresearch/capturesync/MainActivity.java
@@ -91,7 +91,19 @@
 import java.util.concurrent.FutureTask;
 import java.util.stream.Collectors;
 
+import org.apache.hc.client5.http.classic.methods.HttpPost;
+import org.apache.hc.client5.http.entity.UrlEncodedFormEntity;
+import org.apache.hc.client5.http.entity.mime.FileBody;
+import org.apache.hc.client5.http.entity.mime.HttpMultipartMode;
+import org.apache.hc.client5.http.entity.mime.MultipartEntityBuilder;
+import org.apache.hc.client5.http.impl.classic.CloseableHttpClient;
+import org.apache.hc.client5.http.impl.classic.HttpClients;
+import org.apache.hc.core5.http.HttpEntity;
+import org.apache.hc.core5.http.HttpResponse;
+import org.apache.hc.core5.http.NameValuePair;
+import org.apache.hc.core5.http.message.BasicNameValuePair;
 import org.jetbrains.annotations.NotNull;
+import org.json.JSONArray;
 import org.json.JSONException;
 import org.json.JSONObject;
 import io.javalin.Javalin;
@@ -423,6 +435,19 @@ private void handleWebSocketMsg(@NotNull WsMessageContext wsMessageContext){
 
                 break;
 
+            case "PREFIX_LIST" :
+                Log.i(TAG,"handling the message in PREFIX_LIST" );
+                sendPrefixList();
+                ((SoftwareSyncLeader) softwareSyncController.softwareSync)
+                        .broadcastRpc(
+                                SoftwareSyncController.METHOD_PREFIX_LIST,
+                                "0");
+
+                break;
+            case "PING" :
+                Log.i(TAG,"handling the message in PING" );
+                wsMessageContext.send("PONG");
+                break;
         }
     }
 
@@ -682,6 +707,40 @@ private void startSoftwareSync() {
         }
     }
 
+    public void sendPrefixList(){
+        HttpResponse httpResponse;
+        int statusCode;
+        JSONArray fileNamesArray=new JSONArray();
+        File sdcard = Environment.getExternalStorageDirectory();
+        String videoFilePath = sdcard.getAbsolutePath()+ "/RecSync/VID/";
+        String clientID = Settings.Secure.getString(this.getContentResolver(), Settings.Secure.ANDROID_ID);
+        File path = new File(videoFilePath);
+        File fileList[] = path.listFiles();
+        List<NameValuePair> nameValuePairs = new ArrayList<NameValuePair>(2);
+        String filename;
+        for(int i=0; i< fileList.length; i++){
+            filename = fileList[i].getName();
+            try{
+                fileNamesArray.put(filename);
+            }catch (Exception e){
+                e.printStackTrace();
+            }
+
+        }
+        nameValuePairs.add(new BasicNameValuePair("file_list", fileNamesArray.toString()));
+        nameValuePairs.add(new BasicNameValuePair("client_id",clientID ));
+        try (final CloseableHttpClient httpClient = HttpClients.createDefault()) {
+            String endpoint = "http://192.168.5.1:5000/namelist";
+            final HttpPost httpPost = new HttpPost(endpoint);
+            httpPost.setEntity(new UrlEncodedFormEntity(nameValuePairs));
+            httpResponse = httpClient.execute(httpPost);
+            statusCode = httpResponse.getCode();
+            System.out.println("Response Status:" + statusCode);
+
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+    }
     private PhaseConfig loadPhaseConfigFile() throws JSONException {
         // Load phase config file and pass to phase aligner.
 
diff --git a/app/src/main/java/com/googleresearch/capturesync/SoftwareSyncController.java b/app/src/main/java/com/googleresearch/capturesync/SoftwareSyncController.java
@@ -64,6 +64,7 @@ public class SoftwareSyncController implements Closeable {
     public static final int METHOD_STOP_RECORDING = 200_004;
     public static final int METHOD_UPLOAD_RECORDED_FILES = 200_005;
     public static final int METHOD_EMPTY_DEVICE = 200_006;
+    public static final int METHOD_PREFIX_LIST = 200_007;
 
     private long upcomingTriggerTimeNs;
 
@@ -219,6 +220,14 @@ private void setupSoftwareSync() {
                                 () -> context.sendFilesToServer(payload)
                         );
                     });
+            clientRpcs.put(
+                    METHOD_PREFIX_LIST,
+                    payload -> {
+                        Log.v(TAG, "Sending Files List");
+                        context.runOnUiThread(
+                                () -> context.sendPrefixList()
+                        );
+                    });
 
             clientRpcs.put(
                     METHOD_EMPTY_DEVICE,
diff --git a/remote_control/remote_control.py b/remote_control/remote_control.py
@@ -12,7 +12,8 @@
 from PyQt5 import QtCore, QtGui, QtWidgets
 from PyQt5.QtWidgets import QMessageBox
 import websocket
-
+import rel
+import threading
 
 class RemoteController(object):
 
@@ -79,6 +80,15 @@ def delete_all_btn(self):
     def clearStatusBtn(self):
         self.status_label.setPlainText("")
 
+    def prefixList(self):
+        try:
+            self.ws.send("PREFIX_LIST")
+        except Exception as e:
+            self.show_popup()
+            with open('last_prefix.txt','w+') as file:
+                file.writelines(self.download_prefix_text.toPlainText())
+            sys.exit()
+
     def downloadBtn(self):
         endpoint = self.api_input.toPlainText()
         download_prefix = self.download_prefix_text.toPlainText()
@@ -93,12 +103,26 @@ def isPrefix(self, prefix_text):
            return False
         return True
 
+    def asyncTask(self, f_stop):
+        self.ws.send("PING")
+        if not f_stop.is_set():
+                # call f() again in 60 seconds
+                threading.Timer(5, self.asyncTask, [f_stop]).start()
+
     def setupUi(self, MainWindow):
         # Setup the WEB SOCKET
+        #self.ws = websocket.WebSocketApp("ws://192.168.5.2:7867/remotecon")
         self.ws = websocket.WebSocket()
-
-        self.ws.connect("ws://192.168.5.2:7867/remotecon", ping_interval=1)
-
+        self.ws.connect("ws://192.168.5.2:7867/remotecon")
+        f_stop = threading.Event()
+        self.asyncTask(f_stop)
+#         self.ws = websocket.WebSocketApp("ws://192.168.5.2:7867/remotecon")
+#
+#         self.ws.run_forever(ping_interval=1)
+#         self.ws.run_forever(dispatcher=rel, reconnect=5)
+#         rel.signal(2, rel.abort)
+#         rel.dispatch()
+#         await ws.send('2')
         # Setup the GUI
         MainWindow.setObjectName("MainWindow")
         MainWindow.resize(800, 800)
@@ -139,8 +163,15 @@ def setupUi(self, MainWindow):
                 self.download_prefix_text.setText(data[0])
         except:
             pass
+
+        self.prefix_list_btn = QtWidgets.QPushButton(self.centralwidget)
+        self.prefix_list_btn.setGeometry(QtCore.QRect(120, 380, 241, 61))
+        self.prefix_list_btn.setFont(font)
+        self.prefix_list_btn.setObjectName("prefix_list_button")
+        self.prefix_list_btn.clicked.connect(self.prefixList)
+
         self.download_btn = QtWidgets.QPushButton(self.centralwidget)
-        self.download_btn.setGeometry(QtCore.QRect(280, 380, 161, 61))
+        self.download_btn.setGeometry(QtCore.QRect(380, 380, 241, 61))
         self.download_btn.setFont(font)
         self.download_btn.setObjectName("pushButton_4")
         self.download_btn.clicked.connect(self.downloadBtn)
@@ -186,6 +217,7 @@ def retranslateUi(self, MainWindow):
         self.api_input.setPlaceholderText(_translate("MainWindow", "Please enter the api endpoint where you want the files to be uploaded."))
         self.download_prefix_text.setPlaceholderText(_translate("MainWindow", " Enter Session Prefix"))
         self.download_btn.setText(_translate("MainWindow", "Download"))
+        self.prefix_list_btn.setText(_translate("MainWindow", "Prefix List"))
         self.status_label.setPlaceholderText(_translate("MainWindow", "No status "))
 
 
