Simpler 'shape' function.

1 files changed, 1 insertions, 3 deletions

diff --git a/src/Network/BitTorrent/DHT/Routing.hs b/src/Network/BitTorrent/DHT/Routing.hs
index ba1f6799..8705a5a2 100644
--- a/src/Network/BitTorrent/DHT/Routing.hs
+++ b/src/Network/BitTorrent/DHT/Routing.hs
@@ -365,9 +365,7 @@ type NodeCount   = Int
 -- | Internally, routing table is similar to list of buckets or a
 -- /matrix/ of nodes. This function returns the shape of the matrix.
 shape :: Table ip -> [BucketSize]
-shape (Tip _ _ bucket) = [PSQ.size bucket]
-shape (Zero t  bucket) = PSQ.size bucket : shape t
-shape (One bucket t  ) = PSQ.size bucket : shape t
+shape = map PSQ.size . toBucketList
 
 -- | Get number of nodes in the table.
 size :: Table ip -> NodeCount