First commit

8 files changed, 297 insertions, 0 deletions

diff --git a/.envrc b/.envrc
new file mode 100644
index 0000000..1d953f4
--- /dev/null
+++ b/.envrc
@@ -0,0 +1 @@
+use nix
diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..d23f327
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,3 @@
+.direnv
+*.toml
+/noticeme
diff --git a/cmd/noticeme/main.go b/cmd/noticeme/main.go
new file mode 100644
index 0000000..f85ad5b
--- /dev/null
+++ b/cmd/noticeme/main.go
@@ -0,0 +1,54 @@
+package main
+
+import (
+        "flag"
+        "io/ioutil"
+        "log"
+        "os/exec"
+
+        "git.sixfoisneuf.fr/noticeme"
+        "github.com/pelletier/go-toml"
+)
+
+type Config struct {
+        Server   string
+        Username string
+        Password string
+
+        Highlights []string
+        Command    string
+}
+
+func monitorMessage(hl []string, cmd string, c chan noticeme.Message) {
+        for msg := range c {
+                if msg.IsHighlight(hl) {
+                        c := exec.Command(cmd, msg.String())
+                        if err := c.Run(); err != nil {
+                                log.Printf("Error running command: %v", err)
+                        }
+                }
+        }
+}
+
+func main() {
+        configPath := flag.String("config", "config.toml", "Path to the configuration file")
+        flag.Parse()
+
+        var cfg Config
+        data, err := ioutil.ReadFile(*configPath)
+        if err != nil {
+                log.Fatalf("error reading '%s': %v", *configPath, err)
+        }
+
+        if err := toml.Unmarshal(data, &cfg); err != nil {
+                log.Fatalf("error parsing '%s': %v", *configPath, err)
+        }
+
+        bouncer := noticeme.NewBouncer(cfg.Server, cfg.Username, cfg.Password)
+
+        go monitorMessage(cfg.Highlights, cfg.Command, bouncer.Messages)
+
+        if err := bouncer.Connect(); err != nil {
+                panic(err)
+        }
+}
diff --git a/go.mod b/go.mod
new file mode 100644
index 0000000..117ee4d
--- /dev/null
+++ b/go.mod
@@ -0,0 +1,9 @@
+module git.sixfoisneuf.fr/noticeme
+
+go 1.20
+
+require (
+        git.sr.ht/~taiite/senpai v0.2.0 // indirect
+        github.com/pelletier/go-toml v1.9.5 // indirect
+        golang.org/x/time v0.0.0-20210611083556-38a9dc6acbc6 // indirect
+)
diff --git a/go.sum b/go.sum
new file mode 100644
index 0000000..e148884
--- /dev/null
+++ b/go.sum
@@ -0,0 +1,49 @@
+git.sr.ht/~emersion/go-scfg v0.0.0-20201019143924-142a8aa629fc/go.mod h1:t+Ww6SR24yYnXzEWiNlOY0AFo5E9B73X++10lrSpp4U=
+git.sr.ht/~taiite/senpai v0.2.0 h1:a9ZzJj6gKFJ5gz+s6sVcq94DSCPrzFFR/hIh6yONS+U=
+git.sr.ht/~taiite/senpai v0.2.0/go.mod h1:oSJWECrxhhe0IFO8mJLn5YtP5Ze/r70W6fuCHmhwqCE=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/delthas/go-libnp v0.0.0-20221222161248-0e45ece1f878/go.mod h1:aGVXnhWpDlt5U4SphG97o1gszctZKvBTXy320E8Buw4=
+github.com/delthas/go-localeinfo v0.0.0-20221116001557-686a1e185118/go.mod h1:sG54BxlyQgIskYURLrg7mvhoGBe0Qq12DNtYRALwNa4=
+github.com/gdamore/encoding v1.0.0/go.mod h1:alR0ol34c49FCSBLjhosxzcPHQbf2trDkoo5dl+VrEg=
+github.com/gdamore/tcell/v2 v2.5.4-0.20221017224006-ede1dd5ee680/go.mod h1:XmCynGHvvGG7UFI6az9zzoEHBvZB1PGf5APwOJMFUyE=
+github.com/godbus/dbus/v5 v5.1.0/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA=
+github.com/google/shlex v0.0.0-20191202100458-e7afc7fbc510/go.mod h1:pupxD2MaaD3pAXIBCelhxNneeOaAeabZDe5s4K6zSpQ=
+github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
+github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
+github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
+github.com/lucasb-eyer/go-colorful v1.2.0/go.mod h1:R4dSotOR9KMtayYi1e77YzuveK+i7ruzyGqttikkLy0=
+github.com/mattn/go-runewidth v0.0.14/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w=
+github.com/pelletier/go-toml v1.9.5 h1:4yBQzkHv+7BHq2PQUZF3Mx0IYxG7LsP222s7Agd3ve8=
+github.com/pelletier/go-toml v1.9.5/go.mod h1:u1nR/EPcESfeI/szUZKdtJ0xRNbUoANCkoOuaOx1Y+c=
+github.com/pkg/diff v0.0.0-20210226163009-20ebb0f2a09e/go.mod h1:pJLUxLENpZxwdsKMEsNbx1VGcRFpLqf3715MtcvvzbA=
+github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc=
+github.com/rogpeppe/go-internal v1.8.0/go.mod h1:WmiCO8CzOY8rg0OYDC4/i/2WRWAB6poM+XZ2dLUbcbE=
+github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
+golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
+golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
+golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
+golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
+golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
+golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
+golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
+golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
+golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
+golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
+golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
+golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
+golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
+golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
+golang.org/x/text v0.4.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
+golang.org/x/time v0.0.0-20210611083556-38a9dc6acbc6 h1:Vv0JUPWTyeqUq42B2WJ1FeIDjjvGKoA2Ss+Ts0lAVbs=
+golang.org/x/time v0.0.0-20210611083556-38a9dc6acbc6/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ=
+golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
+golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
+golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
+golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
+gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/errgo.v2 v2.1.0/go.mod h1:hNsd1EY+bozCKY1Ytp96fpM3vjJbqLJn88ws8XvfDNI=
+mvdan.cc/xurls/v2 v2.3.0/go.mod h1:AjuTy7gEiUArFMjgBBDU4SMxlfUYsRokpJQgNWOt3e4=
diff --git a/irc.go b/irc.go
new file mode 100644
index 0000000..ecf3d72
--- /dev/null
+++ b/irc.go
@@ -0,0 +1,107 @@
+package noticeme
+
+import (
+        "fmt"
+        "log"
+        "net"
+
+        "git.sr.ht/~taiite/senpai/irc"
+)
+
+type network struct {
+        NetID string
+
+        in      <-chan irc.Message
+        out     chan<- irc.Message
+        session *irc.Session
+}
+
+type Bouncer struct {
+        networks []*network
+
+        Server             string
+        Username, Password string
+
+        Messages chan Message
+}
+
+func NewBouncer(server, username, password string) *Bouncer {
+        b := Bouncer{
+                Server:   server,
+                Username: username,
+                Password: password,
+        }
+        b.Messages = make(chan Message, 100)
+
+        return &b
+}
+
+func (b *Bouncer) Connect() error {
+        return b.newNetwork("", false)
+}
+
+func (b *Bouncer) newNetwork(netID string, background bool) error {
+        auth := irc.SASLPlain{
+                Username: b.Username,
+                Password: b.Password,
+        }
+
+        params := irc.SessionParams{
+                Nickname: "NoticeMe",
+                Username: b.Username,
+                RealName: b.Username,
+                NetID:    netID,
+                Auth:     &auth,
+        }
+
+        conn, err := net.Dial("tcp", b.Server)
+        if err != nil {
+                return fmt.Errorf("net.Dial: %w", err)
+        }
+
+        n := &network{
+                NetID: netID,
+        }
+
+        n.in, n.out = irc.ChanInOut(conn)
+        n.session = irc.NewSession(n.out, params)
+
+        b.networks = append(b.networks, n)
+
+        if background {
+                go b.monitorNetwork(n)
+        } else {
+                b.monitorNetwork(n)
+        }
+
+        return nil
+}
+
+func (b *Bouncer) hasNetID(netID string) bool {
+        for _, net := range b.networks {
+                if net.NetID == netID {
+                        return true
+                }
+        }
+        return false
+}
+
+func (b *Bouncer) monitorNetwork(n *network) {
+        log.Printf("Monitoring NetID<%s>", n.NetID)
+
+        for msg := range n.in {
+                switch msg.Command {
+                case "PRIVMSG":
+                        b.Messages <- fromIrcMessage(msg, n.session)
+                case "BOUNCER":
+                        if msg.Params[0] != "NETWORK" {
+                                log.Printf("[%s] Invalid BOUNCER message, stopping. msg: %+v", n.NetID, msg)
+                                break
+                        }
+
+                        if !b.hasNetID(msg.Params[1]) {
+                                b.newNetwork(msg.Params[1], true)
+                        }
+                }
+        }
+}
diff --git a/message.go b/message.go
new file mode 100644
index 0000000..19c4e30
--- /dev/null
+++ b/message.go
@@ -0,0 +1,64 @@
+package noticeme
+
+import (
+        "fmt"
+        "strings"
+
+        "git.sr.ht/~taiite/senpai/irc"
+)
+
+type Message struct {
+        Channel string // a #channel or a user
+        Author  string
+
+        Text string
+
+        isMe bool
+}
+
+func fromIrcMessage(msg irc.Message, sess *irc.Session) Message {
+        user := msg.Prefix.Name
+        channel := msg.Params[0]
+        txt := msg.Params[1]
+
+        return Message{
+                Channel: channel,
+                Author:  user,
+                Text:    txt,
+                isMe:    sess.IsMe(user),
+        }
+}
+
+func (m Message) IsPrivate() bool {
+        return m.Channel[0] != '#'
+}
+
+func (m Message) IsHighlight(matches []string) bool {
+        // special case: if the message is a PM and is not
+        // sent by us, highlight
+        if m.IsPrivate() && !m.isMe {
+                return true
+        }
+
+        lower := make([]string, len(matches))
+
+        for i, m := range matches {
+                lower[i] = strings.ToLower(m)
+        }
+
+        for _, l := range lower {
+                if strings.Contains(strings.ToLower(m.Text), l) {
+                        return true
+                }
+        }
+
+        return false
+}
+func (m Message) String() string {
+        c := m.Channel
+
+        if m.IsPrivate() {
+                c = "private"
+        }
+        return fmt.Sprintf("[%s] <%s> %s", c, m.Author, m.Text)
+}
diff --git a/shell.nix b/shell.nix
new file mode 100644
index 0000000..867d0d3
--- /dev/null
+++ b/shell.nix
@@ -0,0 +1,10 @@
+let
+  pkgs = import <nixpkgs> {};
+in
+pkgs.mkShell {
+  packages = [
+    pkgs.go
+    pkgs.gopls
+    pkgs.gotools
+  ];
+}