From daf93c8232fdf9062669613cc4ebefbeab9b39b7 Mon Sep 17 00:00:00 2001
From: Matt Joiner <anacrolix@gmail.com>
Date: Fri, 27 Feb 2015 01:41:35 +1100
Subject: [PATCH] Add various helpers to Info

---
 metainfo/metainfo.go | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)

diff --git a/metainfo/metainfo.go b/metainfo/metainfo.go
index 0b605645..52ba9bf6 100644
--- a/metainfo/metainfo.go
+++ b/metainfo/metainfo.go
@@ -46,6 +46,42 @@ type Info struct {
 	Files       []FileInfo `bencode:"files,omitempty"`
 }
 
+func (me *Info) TotalLength() (ret int64) {
+	if me.IsDir() {
+		for _, fi := range me.Files {
+			ret += fi.Length
+		}
+	} else {
+		ret = me.Length
+	}
+	return
+}
+
+func (me *Info) NumPieces() int {
+	return len(me.Pieces) / 20
+}
+
+type Piece interface {
+	Hash() []byte
+	Length() int64
+}
+
+type piece struct {
+	Info *Info
+	i    int
+}
+
+func (me piece) Length() int64 {
+	if me.i == me.Info.NumPieces()-1 {
+		return me.Info.TotalLength() - int64(me.i)*me.Info.PieceLength
+	}
+	return me.Info.PieceLength
+}
+
+func (me *Info) Piece(i int) piece {
+	return piece{me, i}
+}
+
 func (i *Info) IsDir() bool {
 	return len(i.Files) != 0
 }
-- 
2.48.1