package bot import irc "github.com/fluffle/goirc/client" import "labix.org/v2/mgo" import "bitbucket.org/phlyingpenguin/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) }