From e2d130e68709bd4d5d295383b6470fa9ec979362 Mon Sep 17 00:00:00 2001
From: Yuxin Wang <yuxinwang.dev@gmail.com>
Date: Wed, 26 Sep 2018 23:16:33 -0400
Subject: [PATCH] Add list_peers command for Peer.

---
 p2pfs/core/peer.py   | 3 +++
 p2pfs/ui/terminal.py | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/p2pfs/core/peer.py b/p2pfs/core/peer.py
index 034e1e6..24004d3 100644
--- a/p2pfs/core/peer.py
+++ b/p2pfs/core/peer.py
@@ -67,6 +67,9 @@ class Peer(MessageServer):
             del self._publish_results[filename]
         return is_success, message
 
+    def peers(self):
+        return tuple(self._peers.values())
+
     def list_file(self):
         self._write_message(self._server_sock, {
             'type': MessageType.REQUEST_FILE_LIST,
diff --git a/p2pfs/ui/terminal.py b/p2pfs/ui/terminal.py
index c9a6d5c..37bfe7c 100644
--- a/p2pfs/ui/terminal.py
+++ b/p2pfs/ui/terminal.py
@@ -43,6 +43,9 @@ class PeerTerminal(cmd.Cmd):
         assert isinstance(peer, Peer)
         self._peer = peer
 
+    def do_list_peers(self, arg):
+        print(self._peer.peers())
+
     def do_publish(self, arg):
         arg = arg.split(' ')[0]
         _, message = self._peer.publish(arg)