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