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