mirror of https://github.com/velour/catbase.git
55 lines
1.2 KiB
Go
55 lines
1.2 KiB
Go
package bot
|
|
|
|
import irc "github.com/fluffle/goirc/client"
|
|
import "labix.org/v2/mgo"
|
|
import "godeepintir/config"
|
|
|
|
// Bot type provides storage for bot-wide information, configs, and database connections
|
|
type Bot struct {
|
|
Plugins []Handler
|
|
Users []User
|
|
Conn *irc.Conn
|
|
// mongodb connection will go here
|
|
DbSession *mgo.Session
|
|
Db *mgo.Database
|
|
}
|
|
|
|
// User type stores user history. This is a vehicle that will follow the user for the active
|
|
// session
|
|
type User struct {
|
|
// Current nickname known
|
|
Name string
|
|
|
|
// LastSeen DateTime
|
|
|
|
// Alternative nicknames seen
|
|
Alts []string
|
|
|
|
// Last N messages sent to the user
|
|
MessageLog []string
|
|
}
|
|
|
|
// NewBot creates a Bot for a given connection and set of handlers. The handlers must not
|
|
// require the bot as input for their creation (so use AddHandler instead to add handlers)
|
|
func NewBot(config *config.Config, c *irc.Conn, p ...Handler) *Bot {
|
|
session, err := mgo.Dial(config.DbServer)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
db := session.DB(config.DbName)
|
|
|
|
return &Bot{
|
|
Plugins: p,
|
|
Users: make([]User, 10),
|
|
Conn: c,
|
|
DbSession: session,
|
|
Db: db,
|
|
}
|
|
}
|
|
|
|
// Adds a constructed handler to the bots handlers list
|
|
func (b *Bot) AddHandler(h Handler) {
|
|
b.Plugins = append(b.Plugins, h)
|
|
}
|