forked from atla/goirc-xdcc
-
Notifications
You must be signed in to change notification settings - Fork 0
/
bot.go
103 lines (74 loc) · 1.99 KB
/
bot.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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
package bot
import (
"fmt"
"math/rand"
"os"
"time"
"github.com/atla/goirc-xdcc/xdcc"
"github.com/docker/docker/pkg/namesgenerator"
irc "github.com/fluffle/goirc/client"
)
// XdccBot handles the download of a dcc package automatically
type XdccBot struct {
nick string
downloading bool
conn *irc.Conn
pack Package
}
// New creates a new XdccBot
func New() *XdccBot {
// create bot name
rand.Seed(time.Now().UnixNano())
nick := namesgenerator.GetRandomName(5)
return &XdccBot{
nick: nick,
}
}
func showDownloadUpdates(xdcc *xdcc.XDCC) {
}
// Get retrieves a file via xdcc with the given package information
func (bot *XdccBot) Get(pack Package, updates func(*xdcc.DownloadUpdate)) {
quit := make(chan bool)
bot.downloading = false
bot.conn = irc.SimpleClient(bot.nick)
bot.conn.Me().Ident = bot.nick
client := xdcc.New(bot.conn)
fmt.Printf("Starting bot %s\n", bot.nick)
if updates != nil {
go func() {
for update := range client.DownloadUpdates {
updates(update)
if update.Status == xdcc.DownloadStatusDone {
bot.conn.Quit("may the 4th...")
fmt.Printf("Quitting bot %s\n", bot.nick)
// quit bot
quit <- true
}
}
}()
}
bot.conn.HandleFunc(irc.CONNECTED,
func(conn *irc.Conn, line *irc.Line) {
if pack.CompanionChannel != "" {
conn.Join(pack.CompanionChannel)
}
conn.Join(pack.Channel)
})
bot.conn.HandleFunc(irc.JOIN,
func(conn *irc.Conn, line *irc.Line) {
if !bot.downloading && line.Args[0] == pack.Channel {
bot.downloading = true
downloadDirectory, exists := os.LookupEnv("DOWNLOAD_DIRECTORY")
if !exists {
downloadDirectory = "./downloads/"
}
client.GetXdcc(pack.Host, fmt.Sprintf("xdcc send #%d", pack.PackageID), downloadDirectory)
}
})
bot.conn.HandleFunc(irc.DISCONNECTED,
func(conn *irc.Conn, line *irc.Line) { quit <- true })
if err := bot.conn.ConnectTo(pack.Network); err != nil {
fmt.Printf("Connection error: %s\n", err.Error())
}
<-quit
}