mirror of https://github.com/velour/catbase.git
slackApp: stop racing to serve on the http port
This commit is contained in:
parent
4925069ac9
commit
a1e170aa40
|
@ -81,8 +81,6 @@ func New(config *config.Config, connector Connector) Bot {
|
||||||
bot.migrateDB()
|
bot.migrateDB()
|
||||||
|
|
||||||
http.HandleFunc("/", bot.serveRoot)
|
http.HandleFunc("/", bot.serveRoot)
|
||||||
addr := config.Get("HttpAddr", "127.0.0.1:1337")
|
|
||||||
go http.ListenAndServe(addr, nil)
|
|
||||||
|
|
||||||
connector.RegisterEvent(bot.Receive)
|
connector.RegisterEvent(bot.Receive)
|
||||||
|
|
||||||
|
|
|
@ -101,7 +101,6 @@ func (s *SlackApp) Serve() error {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
log.Fatal(http.ListenAndServe("0.0.0.0:1337", nil))
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
295
irc/irc.go
295
irc/irc.go
|
@ -1,295 +0,0 @@
|
||||||
// © 2016 the CatBase Authors under the WTFPL license. See AUTHORS for the list of authors.
|
|
||||||
|
|
||||||
package irc
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
"io"
|
|
||||||
"log"
|
|
||||||
"os"
|
|
||||||
"strings"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/velour/catbase/bot"
|
|
||||||
"github.com/velour/catbase/bot/msg"
|
|
||||||
"github.com/velour/catbase/bot/user"
|
|
||||||
"github.com/velour/catbase/config"
|
|
||||||
"github.com/velour/velour/irc"
|
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
|
||||||
// DefaultPort is the port used to connect to
|
|
||||||
// the server if one is not specified.
|
|
||||||
defaultPort = "6667"
|
|
||||||
|
|
||||||
// InitialTimeout is the initial amount of time
|
|
||||||
// to delay before reconnecting. Each failed
|
|
||||||
// reconnection doubles the timout until
|
|
||||||
// a connection is made successfully.
|
|
||||||
initialTimeout = 2 * time.Second
|
|
||||||
|
|
||||||
// PingTime is the amount of inactive time
|
|
||||||
// to wait before sending a ping to the server.
|
|
||||||
pingTime = 120 * time.Second
|
|
||||||
|
|
||||||
actionPrefix = "\x01ACTION"
|
|
||||||
)
|
|
||||||
|
|
||||||
var throttle <-chan time.Time
|
|
||||||
|
|
||||||
type Irc struct {
|
|
||||||
Client *irc.Client
|
|
||||||
config *config.Config
|
|
||||||
quit chan bool
|
|
||||||
|
|
||||||
event func(bot.Kind, msg.Message, ...interface{})
|
|
||||||
}
|
|
||||||
|
|
||||||
func New(c *config.Config) *Irc {
|
|
||||||
i := Irc{}
|
|
||||||
i.config = c
|
|
||||||
|
|
||||||
return &i
|
|
||||||
}
|
|
||||||
|
|
||||||
func (i *Irc) RegisterEvent(f func(bot.Kind, msg.Message, ...interface{})) {
|
|
||||||
i.event = f
|
|
||||||
}
|
|
||||||
|
|
||||||
func (i *Irc) Send(kind bot.Kind, args ...interface{}) (string, error) {
|
|
||||||
switch kind {
|
|
||||||
case bot.Reply:
|
|
||||||
case bot.Message:
|
|
||||||
return i.sendMessage(args[0].(string), args[1].(string))
|
|
||||||
case bot.Action:
|
|
||||||
return i.sendAction(args[0].(string), args[1].(string))
|
|
||||||
default:
|
|
||||||
}
|
|
||||||
return "", nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (i *Irc) JoinChannel(channel string) {
|
|
||||||
log.Printf("Joining channel: %s", channel)
|
|
||||||
i.Client.Out <- irc.Msg{Cmd: irc.JOIN, Args: []string{channel}}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (i *Irc) sendMessage(channel, message string) (string, error) {
|
|
||||||
for len(message) > 0 {
|
|
||||||
m := irc.Msg{
|
|
||||||
Cmd: "PRIVMSG",
|
|
||||||
Args: []string{channel, message},
|
|
||||||
}
|
|
||||||
_, err := m.RawString()
|
|
||||||
if err != nil {
|
|
||||||
mtl := err.(irc.MsgTooLong)
|
|
||||||
m.Args[1] = message[:mtl.NTrunc]
|
|
||||||
message = message[mtl.NTrunc:]
|
|
||||||
} else {
|
|
||||||
message = ""
|
|
||||||
}
|
|
||||||
|
|
||||||
if throttle == nil {
|
|
||||||
ratePerSec := i.config.GetInt("RatePerSec", 5)
|
|
||||||
throttle = time.Tick(time.Second / time.Duration(ratePerSec))
|
|
||||||
}
|
|
||||||
|
|
||||||
<-throttle
|
|
||||||
|
|
||||||
i.Client.Out <- m
|
|
||||||
}
|
|
||||||
return "NO_IRC_IDENTIFIERS", nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Sends action to channel
|
|
||||||
func (i *Irc) sendAction(channel, message string) (string, error) {
|
|
||||||
message = actionPrefix + " " + message + "\x01"
|
|
||||||
|
|
||||||
return i.sendMessage(channel, message)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (i *Irc) GetEmojiList() map[string]string {
|
|
||||||
//we're not going to do anything because it's IRC
|
|
||||||
return make(map[string]string)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (i *Irc) Serve() error {
|
|
||||||
if i.event == nil {
|
|
||||||
return fmt.Errorf("Missing an event handler")
|
|
||||||
}
|
|
||||||
|
|
||||||
var err error
|
|
||||||
i.Client, err = irc.DialSSL(
|
|
||||||
i.config.Get("Irc.Server", "localhost"),
|
|
||||||
i.config.Get("Nick", "bot"),
|
|
||||||
i.config.Get("FullName", "bot"),
|
|
||||||
i.config.Get("Irc.Pass", ""),
|
|
||||||
true,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("%s", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, c := range i.config.GetArray("channels", []string{}) {
|
|
||||||
i.JoinChannel(c)
|
|
||||||
}
|
|
||||||
|
|
||||||
i.quit = make(chan bool)
|
|
||||||
go i.handleConnection()
|
|
||||||
<-i.quit
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (i *Irc) handleConnection() {
|
|
||||||
t := time.NewTimer(pingTime)
|
|
||||||
|
|
||||||
defer func() {
|
|
||||||
t.Stop()
|
|
||||||
close(i.Client.Out)
|
|
||||||
for err := range i.Client.Errors {
|
|
||||||
if err != io.EOF {
|
|
||||||
log.Println(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
for {
|
|
||||||
select {
|
|
||||||
case msg, ok := <-i.Client.In:
|
|
||||||
if !ok { // disconnect
|
|
||||||
i.quit <- true
|
|
||||||
return
|
|
||||||
}
|
|
||||||
t.Stop()
|
|
||||||
t = time.NewTimer(pingTime)
|
|
||||||
i.handleMsg(msg)
|
|
||||||
|
|
||||||
case <-t.C:
|
|
||||||
i.Client.Out <- irc.Msg{Cmd: irc.PING, Args: []string{i.Client.Server}}
|
|
||||||
t = time.NewTimer(pingTime)
|
|
||||||
|
|
||||||
case err, ok := <-i.Client.Errors:
|
|
||||||
if ok && err != io.EOF {
|
|
||||||
log.Println(err)
|
|
||||||
i.quit <- true
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// HandleMsg handles IRC messages from the server.
|
|
||||||
func (i *Irc) handleMsg(msg irc.Msg) {
|
|
||||||
botMsg := i.buildMessage(msg)
|
|
||||||
|
|
||||||
switch msg.Cmd {
|
|
||||||
case irc.ERROR:
|
|
||||||
log.Println(1, "Received error: "+msg.Raw)
|
|
||||||
|
|
||||||
case irc.PING:
|
|
||||||
i.Client.Out <- irc.Msg{Cmd: irc.PONG}
|
|
||||||
|
|
||||||
case irc.PONG:
|
|
||||||
// OK, ignore
|
|
||||||
|
|
||||||
case irc.ERR_NOSUCHNICK:
|
|
||||||
fallthrough
|
|
||||||
|
|
||||||
case irc.ERR_NOSUCHCHANNEL:
|
|
||||||
fallthrough
|
|
||||||
|
|
||||||
case irc.RPL_MOTD:
|
|
||||||
fallthrough
|
|
||||||
|
|
||||||
case irc.RPL_NAMREPLY:
|
|
||||||
fallthrough
|
|
||||||
|
|
||||||
case irc.RPL_TOPIC:
|
|
||||||
fallthrough
|
|
||||||
|
|
||||||
case irc.KICK:
|
|
||||||
fallthrough
|
|
||||||
|
|
||||||
case irc.TOPIC:
|
|
||||||
fallthrough
|
|
||||||
|
|
||||||
case irc.MODE:
|
|
||||||
fallthrough
|
|
||||||
|
|
||||||
case irc.JOIN:
|
|
||||||
fallthrough
|
|
||||||
|
|
||||||
case irc.PART:
|
|
||||||
fallthrough
|
|
||||||
|
|
||||||
case irc.NOTICE:
|
|
||||||
fallthrough
|
|
||||||
|
|
||||||
case irc.NICK:
|
|
||||||
fallthrough
|
|
||||||
|
|
||||||
case irc.RPL_WHOREPLY:
|
|
||||||
fallthrough
|
|
||||||
|
|
||||||
case irc.RPL_ENDOFWHO:
|
|
||||||
i.event(bot.Event, botMsg)
|
|
||||||
|
|
||||||
case irc.PRIVMSG:
|
|
||||||
i.event(bot.Message, botMsg)
|
|
||||||
|
|
||||||
case irc.QUIT:
|
|
||||||
os.Exit(1)
|
|
||||||
|
|
||||||
default:
|
|
||||||
cmd := irc.CmdNames[msg.Cmd]
|
|
||||||
log.Println("(" + cmd + ") " + msg.Raw)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Builds our internal message type out of a Conn & Line from irc
|
|
||||||
func (i *Irc) buildMessage(inMsg irc.Msg) msg.Message {
|
|
||||||
// Check for the user
|
|
||||||
u := user.User{
|
|
||||||
Name: inMsg.Origin,
|
|
||||||
}
|
|
||||||
|
|
||||||
channel := inMsg.Args[0]
|
|
||||||
if channel == i.config.Get("Nick", "bot") {
|
|
||||||
channel = inMsg.Args[0]
|
|
||||||
}
|
|
||||||
|
|
||||||
isAction := false
|
|
||||||
var message string
|
|
||||||
if len(inMsg.Args) > 1 {
|
|
||||||
message = inMsg.Args[1]
|
|
||||||
|
|
||||||
isAction = strings.HasPrefix(message, actionPrefix)
|
|
||||||
if isAction {
|
|
||||||
message = strings.TrimRight(message[len(actionPrefix):], "\x01")
|
|
||||||
message = strings.TrimSpace(message)
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
iscmd := false
|
|
||||||
filteredMessage := message
|
|
||||||
if !isAction {
|
|
||||||
iscmd, filteredMessage = bot.IsCmd(i.config, message)
|
|
||||||
}
|
|
||||||
|
|
||||||
msg := msg.Message{
|
|
||||||
User: &u,
|
|
||||||
Channel: channel,
|
|
||||||
Body: filteredMessage,
|
|
||||||
Raw: message,
|
|
||||||
Command: iscmd,
|
|
||||||
Action: isAction,
|
|
||||||
Time: time.Now(),
|
|
||||||
Host: inMsg.Host,
|
|
||||||
}
|
|
||||||
|
|
||||||
return msg
|
|
||||||
}
|
|
||||||
|
|
||||||
func (i Irc) Who(channel string) []string {
|
|
||||||
return []string{}
|
|
||||||
}
|
|
9
main.go
9
main.go
|
@ -6,6 +6,7 @@ import (
|
||||||
"flag"
|
"flag"
|
||||||
"log"
|
"log"
|
||||||
"math/rand"
|
"math/rand"
|
||||||
|
"net/http"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/velour/catbase/bot"
|
"github.com/velour/catbase/bot"
|
||||||
|
@ -108,8 +109,10 @@ func main() {
|
||||||
b.AddPlugin(fact.New(b))
|
b.AddPlugin(fact.New(b))
|
||||||
b.AddPlugin(db.New(b))
|
b.AddPlugin(db.New(b))
|
||||||
|
|
||||||
for {
|
if err := client.Serve(); err != nil {
|
||||||
err := client.Serve()
|
log.Fatal(err)
|
||||||
log.Println(err)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
addr := c.Get("HttpAddr", "127.0.0.1:1337")
|
||||||
|
log.Fatal(http.ListenAndServe(addr, nil))
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue