From e84e4ab2b14468874b9a42803da8af8886723f51 Mon Sep 17 00:00:00 2001
From: ChristianVisintin <christian.visintin1997@gmail.com>
Date: Sun, 29 Nov 2020 11:55:02 +0100
Subject: [PATCH] Draw method reviewed

---
 src/ui/activities/filetransfer_activity.rs | 20 ++++++++------------
 1 file changed, 8 insertions(+), 12 deletions(-)

diff --git a/src/ui/activities/filetransfer_activity.rs b/src/ui/activities/filetransfer_activity.rs
index 53584fa..4fc3f34 100644
--- a/src/ui/activities/filetransfer_activity.rs
+++ b/src/ui/activities/filetransfer_activity.rs
@@ -1489,8 +1489,12 @@ impl FileTransferActivity {
     /// ### draw
     ///
     /// Draw UI
-    fn draw(&mut self, frame: &mut Frame<CrosstermBackend<Stdout>>) {
-        // TODO: implement
+    fn draw(&mut self) {
+        let mut ctx: Context = self.context.take().unwrap();
+        let _ = ctx.terminal.draw(|f| {
+            // TODO: implement
+        });
+        self.context = Some(ctx);
     }
 
     /// ### draw_header
@@ -1660,11 +1664,7 @@ impl Activity for FileTransferActivity {
                 self.params.address, self.params.port
             )));
             // Force ui draw
-            let mut ctx: Context = self.context.take().unwrap();
-            let _ = ctx.terminal.draw(|f| {
-                self.draw(f);
-            });
-            self.context = Some(ctx);
+            self.draw();
             // Connect to remote
             self.connect();
         }
@@ -1676,11 +1676,7 @@ impl Activity for FileTransferActivity {
             }
         }
         // @! draw interface
-        let mut ctx: Context = self.context.take().unwrap();
-        let _ = ctx.terminal.draw(|f| {
-            self.draw(f);
-        });
-        self.context = Some(ctx);
+        self.draw();
     }
 
     /// ### on_destroy