summaryrefslogtreecommitdiff
path: root/rfc.go
blob: ceeef45bdf62b915bb13b74da6de61484d1537ae (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
package main

import (
	"fmt"
	"log"
	"strings"

	"dim13.org/rfc"
	irc "github.com/fluffle/goirc/client"
)

type RFC struct{}

var (
	rfcMap = make(map[int]rfc.Entry)
	rfcKW  = make(map[string][]string)
)

func (RFC) Handle(conn *irc.Conn, line *irc.Line) {
	if f := strings.Fields(line.Text()); len(f) > 1 {
		kw := strings.ToLower(f[1])
		if e, ok := rfcKW[kw]; ok {
			if line.Public() && len(e) > 10 {
				n := fmt.Sprint("... ", len(e), " total")
				e = append(e[:10], n)
			}
			s := strings.Join(e, ", ")
			conn.Notice(line.Target(), s)
		} else {
			conn.Notice(line.Target(), "no matches")
		}
	}
}

func (RFC) Help() string {
	return "search rfc by keyword (msg privat to see all matches)"
}

func ExtractRFC(conn *irc.Conn, line *irc.Line) {
	for _, v := range strings.Fields(line.Text()) {
		var id int
		w := strings.ToUpper(v)
		if _, err := fmt.Sscanf(w, "RFC%d", &id); err == nil {
			if entry, ok := rfcMap[id]; ok {
				s := fmt.Sprint(entry)
				conn.Notice(line.Target(), s)
			}
		}
	}
}

func init() {
	fd, err := rfc.Open()
	if err != nil {
		log.Fatal(err)
	}
	entries, err := rfc.Decode(fd)
	if err != nil {
		log.Fatal(err)
	}
	for _, e := range entries {
		id := e.ID()
		rfcMap[id] = e
		for _, w := range e.Keywords {
			for _, f := range strings.Fields(w) {
				kw := strings.ToLower(f)
				rfcKW[kw] = append(rfcKW[kw], e.DocID)
			}
		}
	}
	Register("rfc", &RFC{})
}