torrent/magnet.go

package torrent

import (
	"encoding/base32"
	"encoding/hex"
	"fmt"
	"net/url"
	"strings"
)

type Magnet struct {
	InfoHash    [20]byte
	Trackers    []string
	DisplayName string
}

const xtPrefix = "urn:btih:"

func (m *Magnet) String() (ret string) {
	// net.URL likes to assume //, and encodes ':' on us, so we do most of
	// this manually.
	ret = "magnet:?xt="
	ret += xtPrefix + hex.EncodeToString(m.InfoHash[:])
	if m.DisplayName != "" {
		ret += "&dn=" + url.QueryEscape(m.DisplayName)
	}
	for _, tr := range m.Trackers {
		ret += "&tr=" + url.QueryEscape(tr)
	}
	return
}

func ParseMagnetURI(uri string) (m Magnet, err error) {
	u, err := url.Parse(uri)
	if err != nil {
		err = fmt.Errorf("error parsing uri: %s", err)
		return
	}
	if u.Scheme != "magnet" {
		err = fmt.Errorf("unexpected scheme: %q", u.Scheme)
		return
	}
	xt := u.Query().Get("xt")
	if !strings.HasPrefix(xt, xtPrefix) {
		err = fmt.Errorf("bad xt parameter")
		return
	}
	xt = xt[len(xtPrefix):]
	decode := func() func(dst, src []byte) (int, error) {
		switch len(xt) {
		case 40:
			return hex.Decode
		case 32:
			return base32.StdEncoding.Decode
		default:
			return nil
		}
	}()
	if decode == nil {
		err = fmt.Errorf("unhandled xt parameter encoding: encoded length %d", len(xt))
		return
	}
	n, err := decode(m.InfoHash[:], []byte(xt))
	if err != nil {
		err = fmt.Errorf("error decoding xt: %s", err)
		return
	}
	if n != 20 {
		panic(n)
	}
	m.DisplayName = u.Query().Get("dn")
	m.Trackers = u.Query()["tr"]
	return
}