]> Sergey Matveev's repositories - btrtrc.git/blob - spec.go
Add DisableInitialPieceCheck option (#677)
[btrtrc.git] / spec.go
1 package torrent
2
3 import (
4         "fmt"
5
6         "github.com/anacrolix/torrent/metainfo"
7         "github.com/anacrolix/torrent/storage"
8 )
9
10 // Specifies a new torrent for adding to a client. There are helpers for magnet URIs and torrent
11 // metainfo files.
12 type TorrentSpec struct {
13         // The tiered tracker URIs.
14         Trackers  [][]string
15         InfoHash  metainfo.Hash
16         InfoBytes []byte
17         // The name to use if the Name field from the Info isn't available.
18         DisplayName string
19         Webseeds    []string
20         DhtNodes    []string
21         PeerAddrs   []string
22         // The combination of the "xs" and "as" fields in magnet links, for now.
23         Sources []string
24
25         // The chunk size to use for outbound requests. Defaults to 16KiB if not set.
26         ChunkSize                int
27         DisableInitialPieceCheck bool
28         Storage                  storage.ClientImpl
29
30         // Whether to allow data download or upload
31         DisallowDataUpload   bool
32         DisallowDataDownload bool
33 }
34
35 func TorrentSpecFromMagnetUri(uri string) (spec *TorrentSpec, err error) {
36         m, err := metainfo.ParseMagnetUri(uri)
37         if err != nil {
38                 return
39         }
40         spec = &TorrentSpec{
41                 Trackers:    [][]string{m.Trackers},
42                 DisplayName: m.DisplayName,
43                 InfoHash:    m.InfoHash,
44                 Webseeds:    m.Params["ws"],
45                 Sources:     append(m.Params["xs"], m.Params["as"]...),
46                 PeerAddrs:   m.Params["x.pe"], // BEP 9
47                 // TODO: What's the parameter for DHT nodes?
48         }
49         return
50 }
51
52 func TorrentSpecFromMetaInfoErr(mi *metainfo.MetaInfo) (*TorrentSpec, error) {
53         info, err := mi.UnmarshalInfo()
54         if err != nil {
55                 return nil, fmt.Errorf("unmarshalling info: %w", err)
56         }
57         return &TorrentSpec{
58                 Trackers:    mi.UpvertedAnnounceList(),
59                 InfoHash:    mi.HashInfoBytes(),
60                 InfoBytes:   mi.InfoBytes,
61                 DisplayName: info.Name,
62                 Webseeds:    mi.UrlList,
63                 DhtNodes: func() (ret []string) {
64                         ret = make([]string, 0, len(mi.Nodes))
65                         for _, node := range mi.Nodes {
66                                 ret = append(ret, string(node))
67                         }
68                         return
69                 }(),
70         }, nil
71 }
72
73 func TorrentSpecFromMetaInfo(mi *metainfo.MetaInfo) *TorrentSpec {
74         ts, err := TorrentSpecFromMetaInfoErr(mi)
75         if err != nil {
76                 panic(err)
77         }
78         return ts
79 }