/
torrent.go
75 lines (62 loc) · 1.34 KB
/
torrent.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
package main
import (
"io"
"log"
"os"
"path"
"github.com/anacrolix/torrent"
"github.com/anacrolix/torrent/metainfo"
)
var Client *torrent.Client
func InitClient() error {
dataDir := path.Join(os.TempDir(), "omckle")
var err error
Client, err = torrent.NewClient(&torrent.Config{
DataDir: dataDir,
Debug: true,
})
if err != nil {
log.Println("Error creating torrent client: ", err)
return err
}
log.Println("Torrent data: ", len(Client.Torrents()))
log.Println("Torrent data dir: ", dataDir)
return nil
}
func AddMagnetLink(magnet string) error {
t, err := Client.AddMagnet(magnet)
if err != nil {
log.Println("Cannot load torrent from magnet file: ", err)
return err
}
return initTorrent(t)
}
func AddFromReader(r io.Reader) error {
mi, err := metainfo.Load(r)
if err != nil {
log.Println("Error reading meta info from Reader: ", err)
return err
}
t, err := Client.AddTorrent(mi)
if err != nil {
log.Println("Cannot load torrent from reader: ", err)
return err
}
return initTorrent(t)
}
func initTorrent(t *torrent.Torrent) error {
<-t.GotInfo()
log.Printf("Torrent added: %s\n", t.Name())
return Play(t)
}
func getMovieFile(t *torrent.Torrent) *torrent.File {
var f torrent.File
var s int64
for _, file := range t.Files() {
if s < file.Length() {
s = file.Length()
f = file
}
}
return &f
}