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 ae97f748 2021-07-30 op "path"
12 5ad39add 2021-04-27 op "path/filepath"
13 7cc05214 2021-04-27 op "regexp"
14 88370dfe 2021-11-07 op "strings"
15 7cc05214 2021-04-27 op
16 deee939b 2022-01-08 op "github.com/andybalholm/cascadia"
17 7cc05214 2021-04-27 op irc "github.com/fluffle/goirc/client"
18 5659492e 2021-12-02 op "golang.org/x/net/html"
19 7cc05214 2021-04-27 op )
20 7cc05214 2021-04-27 op
21 7cc05214 2021-04-27 op var (
22 88370dfe 2021-11-07 op baseurl = flag.String("baseurl", "gemini://m2i.omarpolo.com", "base url")
23 7cc05214 2021-04-27 op matrixOutDir = flag.String("matrix-out", "", "matrix out directory")
24 7cc05214 2021-04-27 op
25 88370dfe 2021-11-07 op msgRe = regexp.MustCompile(`https://.*/[^\s]+\.(txt|png|jpg|jpeg|gif)`)
26 c6fd4def 2021-07-30 op channel = "#gemini-it"
27 88370dfe 2021-11-07 op
28 88370dfe 2021-11-07 op tooLongRe = regexp.MustCompile(`full message at (https://libera.ems.host/.*)[)]`)
29 5659492e 2021-12-02 op
30 4852f411 2021-12-02 op httplink = regexp.MustCompile(`https?://[^\s)]+`)
31 7cc05214 2021-04-27 op )
32 7cc05214 2021-04-27 op
33 7cc05214 2021-04-27 op func matrix2gemini(conn *irc.Conn, line *irc.Line) {
34 ae97f748 2021-07-30 op matches := msgRe.FindAllString(line.Text(), -1)
35 7cc05214 2021-04-27 op
36 7cc05214 2021-04-27 op // it's not a good idea to defer inside a loop, but we know
37 7cc05214 2021-04-27 op // len(matches) is small (usually just 1). Morover, I like
38 7cc05214 2021-04-27 op // living in danger!
39 7cc05214 2021-04-27 op
40 7cc05214 2021-04-27 op for _, link := range matches {
41 7cc05214 2021-04-27 op resp, err := http.Get(link)
42 7cc05214 2021-04-27 op if err != nil {
43 7cc05214 2021-04-27 op conn.Privmsg(
44 7cc05214 2021-04-27 op channel,
45 7cc05214 2021-04-27 op fmt.Sprintf("failed to download %q: %s", link, err),
46 7cc05214 2021-04-27 op )
47 7cc05214 2021-04-27 op continue
48 7cc05214 2021-04-27 op }
49 7cc05214 2021-04-27 op defer resp.Body.Close()
50 7cc05214 2021-04-27 op
51 ae97f748 2021-07-30 op ext := path.Ext(link)
52 88370dfe 2021-11-07 op tmpfile, err := ioutil.TempFile(*matrixOutDir, "message-*"+ext)
53 7cc05214 2021-04-27 op if err != nil {
54 7cc05214 2021-04-27 op conn.Privmsg(channel, fmt.Sprintf("failed to tmpfile: %s", err))
55 7cc05214 2021-04-27 op return
56 7cc05214 2021-04-27 op }
57 7cc05214 2021-04-27 op defer tmpfile.Close()
58 7cc05214 2021-04-27 op
59 7cc05214 2021-04-27 op io.Copy(tmpfile, resp.Body)
60 7cc05214 2021-04-27 op
61 7cc05214 2021-04-27 op conn.Privmsg(
62 7cc05214 2021-04-27 op channel,
63 5ad39add 2021-04-27 op fmt.Sprintf(
64 5ad39add 2021-04-27 op "better: %s/%s",
65 5ad39add 2021-04-27 op *baseurl,
66 5ad39add 2021-04-27 op filepath.Base(tmpfile.Name()),
67 5ad39add 2021-04-27 op ),
68 7cc05214 2021-04-27 op )
69 7cc05214 2021-04-27 op }
70 7cc05214 2021-04-27 op }
71 7cc05214 2021-04-27 op
72 88370dfe 2021-11-07 op func messageTooLong(conn *irc.Conn, line *irc.Line) {
73 88370dfe 2021-11-07 op matches := tooLongRe.FindStringSubmatch(line.Text())
74 88370dfe 2021-11-07 op if len(matches) != 2 {
75 88370dfe 2021-11-07 op return
76 88370dfe 2021-11-07 op }
77 88370dfe 2021-11-07 op
78 88370dfe 2021-11-07 op url := matches[1]
79 88370dfe 2021-11-07 op
80 88370dfe 2021-11-07 op resp, err := http.Get(url)
81 88370dfe 2021-11-07 op if err != nil {
82 88370dfe 2021-11-07 op conn.Privmsg(
83 88370dfe 2021-11-07 op channel,
84 88370dfe 2021-11-07 op fmt.Sprintf("failed to download %q: %s", url, err),
85 88370dfe 2021-11-07 op )
86 88370dfe 2021-11-07 op return
87 88370dfe 2021-11-07 op }
88 88370dfe 2021-11-07 op defer resp.Body.Close()
89 88370dfe 2021-11-07 op
90 88370dfe 2021-11-07 op sb := &strings.Builder{}
91 88370dfe 2021-11-07 op if _, err := io.Copy(sb, resp.Body); err != nil {
92 88370dfe 2021-11-07 op conn.Privmsg(
93 88370dfe 2021-11-07 op channel,
94 88370dfe 2021-11-07 op fmt.Sprintf("failed to read body of %q: %s", url, err),
95 88370dfe 2021-11-07 op )
96 88370dfe 2021-11-07 op return
97 88370dfe 2021-11-07 op }
98 88370dfe 2021-11-07 op
99 88370dfe 2021-11-07 op conn.Privmsg(channel, fmt.Sprintf("%s ha detto:", line.Nick))
100 88370dfe 2021-11-07 op for _, line := range strings.Split(sb.String(), "\n") {
101 88370dfe 2021-11-07 op conn.Privmsg(
102 88370dfe 2021-11-07 op channel,
103 88370dfe 2021-11-07 op line,
104 88370dfe 2021-11-07 op )
105 88370dfe 2021-11-07 op }
106 88370dfe 2021-11-07 op }
107 88370dfe 2021-11-07 op
108 3adf60f8 2021-12-02 op func stringifyNode(node *html.Node) string {
109 3adf60f8 2021-12-02 op s := ""
110 3adf60f8 2021-12-02 op
111 3adf60f8 2021-12-02 op if node.Type == html.TextNode {
112 3adf60f8 2021-12-02 op return node.Data
113 3adf60f8 2021-12-02 op }
114 3adf60f8 2021-12-02 op
115 3adf60f8 2021-12-02 op for child := node.FirstChild; child != nil; child = child.NextSibling {
116 3adf60f8 2021-12-02 op s += stringifyNode(child)
117 3adf60f8 2021-12-02 op }
118 3adf60f8 2021-12-02 op
119 3adf60f8 2021-12-02 op return s
120 3adf60f8 2021-12-02 op }
121 3adf60f8 2021-12-02 op
122 5659492e 2021-12-02 op func pagetitle(conn *irc.Conn, line *irc.Line) {
123 4852f411 2021-12-02 op log.Println("text is", line.Text())
124 5659492e 2021-12-02 op matches := httplink.FindAllString(line.Text(), -1)
125 5659492e 2021-12-02 op
126 5659492e 2021-12-02 op for _, link := range matches {
127 d26637f1 2021-12-02 op log.Println("fetching", link, "...")
128 d26637f1 2021-12-02 op
129 5659492e 2021-12-02 op resp, err := http.Get(link)
130 5659492e 2021-12-02 op if err != nil {
131 5659492e 2021-12-02 op continue
132 5659492e 2021-12-02 op }
133 5659492e 2021-12-02 op defer resp.Body.Close()
134 5659492e 2021-12-02 op
135 5659492e 2021-12-02 op doc, err := html.Parse(resp.Body)
136 5659492e 2021-12-02 op if err != nil {
137 5659492e 2021-12-02 op continue
138 5659492e 2021-12-02 op }
139 5659492e 2021-12-02 op
140 deee939b 2022-01-08 op sel := cascadia.MustCompile("head > title")
141 deee939b 2022-01-08 op n := cascadia.Query(doc, sel)
142 deee939b 2022-01-08 op
143 deee939b 2022-01-08 op if n == nil {
144 5659492e 2021-12-02 op continue
145 5659492e 2021-12-02 op }
146 deee939b 2022-01-08 op
147 0c3e9562 2022-01-08 op title := stringifyNode(n)
148 0c3e9562 2022-01-08 op if len(title) > 50 {
149 0c3e9562 2022-01-08 op title = title[:50]
150 0c3e9562 2022-01-08 op }
151 deee939b 2022-01-08 op conn.Privmsg(channel, stringifyNode(n))
152 5659492e 2021-12-02 op }
153 5659492e 2021-12-02 op }
154 5659492e 2021-12-02 op
155 7cc05214 2021-04-27 op func dostuff(conn *irc.Conn, line *irc.Line) {
156 7cc05214 2021-04-27 op matrix2gemini(conn, line)
157 88370dfe 2021-11-07 op messageTooLong(conn, line)
158 5659492e 2021-12-02 op pagetitle(conn, line)
159 7cc05214 2021-04-27 op // ...
160 7cc05214 2021-04-27 op }
161 7cc05214 2021-04-27 op
162 7cc05214 2021-04-27 op func main() {
163 7cc05214 2021-04-27 op flag.Parse()
164 7cc05214 2021-04-27 op
165 7cc05214 2021-04-27 op cfg := irc.NewConfig("gemitbot")
166 7cc05214 2021-04-27 op cfg.SSL = true
167 d0286003 2021-07-30 op cfg.SSLConfig = &tls.Config{ServerName: "irc.libera.chat"}
168 d0286003 2021-07-30 op cfg.Server = "irc.libera.chat:7000"
169 7cc05214 2021-04-27 op cfg.NewNick = func(n string) string { return n + "^" }
170 7cc05214 2021-04-27 op
171 7cc05214 2021-04-27 op c := irc.Client(cfg)
172 7cc05214 2021-04-27 op
173 7cc05214 2021-04-27 op c.HandleFunc(irc.CONNECTED, func(conn *irc.Conn, line *irc.Line) {
174 7cc05214 2021-04-27 op log.Println("connected, joining", channel)
175 7cc05214 2021-04-27 op conn.Join(channel)
176 7cc05214 2021-04-27 op })
177 7cc05214 2021-04-27 op
178 7cc05214 2021-04-27 op c.HandleFunc(irc.PRIVMSG, dostuff)
179 7cc05214 2021-04-27 op c.HandleFunc(irc.ACTION, dostuff)
180 7cc05214 2021-04-27 op
181 7cc05214 2021-04-27 op quit := make(chan bool)
182 7cc05214 2021-04-27 op
183 7cc05214 2021-04-27 op c.HandleFunc(irc.DISCONNECTED, func(conn *irc.Conn, line *irc.Line) {
184 7cc05214 2021-04-27 op quit <- true
185 7cc05214 2021-04-27 op })
186 7cc05214 2021-04-27 op
187 7cc05214 2021-04-27 op if err := c.Connect(); err != nil {
188 7cc05214 2021-04-27 op log.Fatalln("connection error:", err)
189 7cc05214 2021-04-27 op }
190 7cc05214 2021-04-27 op
191 7cc05214 2021-04-27 op <-quit
192 7cc05214 2021-04-27 op }