Changes

WebSSH/AppHandler.py

@@ -50,16 +50,17 @@ class AppHandler(tornado.web.RequestHandler):
                 with open(private_key_path, "r") as file:
                     private_key = file.read()
             else:
-                self.set_status(404)
-                self.write("SSH key not found for existing user.")
-                return
+                #self.set_status(404)
+                #self.write("SSH key not found for existing user.")
+                return {response: 404, message: "SSH key not found for existing user."}
         else:
-            self.set_status(404)
-            self.write("User not found.")
-            return
+            #self.set_status(404)
+            #self.write("User not found.")
+            return {response: 404, message:"User not found."}
 
         # Return the private key to the user
         self.set_header('Content-Type', 'application/octet-stream')
         self.set_header('Content-Disposition', f'attachment; filename=id_rsa')
+        self.set_header('response', f'ok')
         self.write(private_key)
     

WebSSH/templates/index.html

@@ -87,6 +87,7 @@
                 body: JSON.stringify({ username: username, password: password })
             })
             .then(response => {
+                console.log(response)
                 if (!response.ok) {
                     throw new Error('Network response was not ok');
                 }