mirror of https://github.com/velour/catbase.git
70 lines
1.6 KiB
Go
70 lines
1.6 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 {
|
|
// Each plugin must be registered in our plugins handler. To come: a map so that this
|
|
// will allow plugins to respond to specific kinds of events
|
|
Plugins []Handler
|
|
|
|
// Users holds information about all of our friends
|
|
Users []User
|
|
|
|
// Conn allows us to send messages and modify our connection state
|
|
Conn *irc.Conn
|
|
|
|
Config *config.Config
|
|
|
|
// Mongo connection and db allow botwide access to the database
|
|
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
|
|
}
|
|
|
|
type Message struct {
|
|
User *User
|
|
Channel, Body 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{
|
|
Config: config,
|
|
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)
|
|
}
|