Blame


1 7cc05214 2021-04-27 op package main
2 7cc05214 2021-04-27 op
3 7cc05214 2021-04-27 op import (
4 7cc05214 2021-04-27 op "crypto/tls"
5 7cc05214 2021-04-27 op "flag"
6 7cc05214 2021-04-27 op "fmt"
7 7cc05214 2021-04-27 op "io"
8 7cc05214 2021-04-27 op "io/ioutil"
9 7cc05214 2021-04-27 op "log"
10 7cc05214 2021-04-27 op "net/http"
11 5ad39add 2021-04-27 op "path/filepath"
12 7cc05214 2021-04-27 op "regexp"
13 7cc05214 2021-04-27 op
14 7cc05214 2021-04-27 op irc "github.com/fluffle/goirc/client"
15 7cc05214 2021-04-27 op )
16 7cc05214 2021-04-27 op
17 7cc05214 2021-04-27 op var (
18 7cc05214 2021-04-27 op baseurl = flag.String("baseurl", "gemini://m2i.omarpolo.com", "base url")
19 7cc05214 2021-04-27 op matrixOutDir = flag.String("matrix-out", "", "matrix out directory")
20 7cc05214 2021-04-27 op
21 7cc05214 2021-04-27 op matrixRe = regexp.MustCompile(`https://matrix.org/[^\s]+\.txt`)
22 d1fc3257 2021-04-27 op channel = "#gemini-italia"
23 7cc05214 2021-04-27 op )
24 7cc05214 2021-04-27 op
25 7cc05214 2021-04-27 op func matrix2gemini(conn *irc.Conn, line *irc.Line) {
26 7cc05214 2021-04-27 op matches := matrixRe.FindAllString(line.Text(), -1)
27 7cc05214 2021-04-27 op
28 7cc05214 2021-04-27 op // it's not a good idea to defer inside a loop, but we know
29 7cc05214 2021-04-27 op // len(matches) is small (usually just 1). Morover, I like
30 7cc05214 2021-04-27 op // living in danger!
31 7cc05214 2021-04-27 op
32 7cc05214 2021-04-27 op for _, link := range matches {
33 7cc05214 2021-04-27 op resp, err := http.Get(link)
34 7cc05214 2021-04-27 op if err != nil {
35 7cc05214 2021-04-27 op conn.Privmsg(
36 7cc05214 2021-04-27 op channel,
37 7cc05214 2021-04-27 op fmt.Sprintf("failed to download %q: %s", link, err),
38 7cc05214 2021-04-27 op )
39 7cc05214 2021-04-27 op continue
40 7cc05214 2021-04-27 op }
41 7cc05214 2021-04-27 op defer resp.Body.Close()
42 7cc05214 2021-04-27 op
43 7cc05214 2021-04-27 op tmpfile, err := ioutil.TempFile(*matrixOutDir, "message-")
44 7cc05214 2021-04-27 op if err != nil {
45 7cc05214 2021-04-27 op conn.Privmsg(channel, fmt.Sprintf("failed to tmpfile: %s", err))
46 7cc05214 2021-04-27 op return
47 7cc05214 2021-04-27 op }
48 7cc05214 2021-04-27 op defer tmpfile.Close()
49 7cc05214 2021-04-27 op
50 7cc05214 2021-04-27 op io.Copy(tmpfile, resp.Body)
51 7cc05214 2021-04-27 op
52 7cc05214 2021-04-27 op conn.Privmsg(
53 7cc05214 2021-04-27 op channel,
54 5ad39add 2021-04-27 op fmt.Sprintf(
55 5ad39add 2021-04-27 op "better: %s/%s",
56 5ad39add 2021-04-27 op *baseurl,
57 5ad39add 2021-04-27 op filepath.Base(tmpfile.Name()),
58 5ad39add 2021-04-27 op ),
59 7cc05214 2021-04-27 op )
60 7cc05214 2021-04-27 op }
61 7cc05214 2021-04-27 op }
62 7cc05214 2021-04-27 op
63 7cc05214 2021-04-27 op func dostuff(conn *irc.Conn, line *irc.Line) {
64 7cc05214 2021-04-27 op matrix2gemini(conn, line)
65 7cc05214 2021-04-27 op // ...
66 7cc05214 2021-04-27 op }
67 7cc05214 2021-04-27 op
68 7cc05214 2021-04-27 op func main() {
69 7cc05214 2021-04-27 op flag.Parse()
70 7cc05214 2021-04-27 op
71 7cc05214 2021-04-27 op cfg := irc.NewConfig("gemitbot")
72 7cc05214 2021-04-27 op cfg.SSL = true
73 7cc05214 2021-04-27 op cfg.SSLConfig = &tls.Config{ServerName: "irc.freenode.net"}
74 7cc05214 2021-04-27 op cfg.Server = "irc.freenode.net:7000"
75 7cc05214 2021-04-27 op cfg.NewNick = func(n string) string { return n + "^" }
76 7cc05214 2021-04-27 op
77 7cc05214 2021-04-27 op c := irc.Client(cfg)
78 7cc05214 2021-04-27 op
79 7cc05214 2021-04-27 op c.HandleFunc(irc.CONNECTED, func(conn *irc.Conn, line *irc.Line) {
80 7cc05214 2021-04-27 op log.Println("connected, joining", channel)
81 7cc05214 2021-04-27 op conn.Join(channel)
82 7cc05214 2021-04-27 op })
83 7cc05214 2021-04-27 op
84 7cc05214 2021-04-27 op c.HandleFunc(irc.PRIVMSG, dostuff)
85 7cc05214 2021-04-27 op c.HandleFunc(irc.ACTION, dostuff)
86 7cc05214 2021-04-27 op
87 7cc05214 2021-04-27 op quit := make(chan bool)
88 7cc05214 2021-04-27 op
89 7cc05214 2021-04-27 op c.HandleFunc(irc.DISCONNECTED, func(conn *irc.Conn, line *irc.Line) {
90 7cc05214 2021-04-27 op quit <- true
91 7cc05214 2021-04-27 op })
92 7cc05214 2021-04-27 op
93 7cc05214 2021-04-27 op if err := c.Connect(); err != nil {
94 7cc05214 2021-04-27 op log.Fatalln("connection error:", err)
95 7cc05214 2021-04-27 op }
96 7cc05214 2021-04-27 op
97 7cc05214 2021-04-27 op <-quit
98 7cc05214 2021-04-27 op }