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) }