diff options
Diffstat (limited to 'connector')
-rw-r--r-- | connector/config.go | 21 | ||||
-rw-r--r-- | connector/connector.go | 134 | ||||
-rw-r--r-- | connector/irc/irc.go | 255 |
3 files changed, 410 insertions, 0 deletions
diff --git a/connector/config.go b/connector/config.go new file mode 100644 index 0000000..d719b49 --- /dev/null +++ b/connector/config.go @@ -0,0 +1,21 @@ +package connector + +import ( + "strconv" +) + +type Configuration map[string]string + +func (c Configuration) GetString(k string) string { + if ss, ok := c[k]; ok { + return ss + } + return "" +} + +func (c Configuration) GetInt(k string) (int, error) { + if ss, ok := c[k]; ok { + return strconv.Atoi(ss) + } + return 0, nil +} diff --git a/connector/connector.go b/connector/connector.go new file mode 100644 index 0000000..2116bb5 --- /dev/null +++ b/connector/connector.go @@ -0,0 +1,134 @@ +package connector + +/* + A generic connector framework for instant messaging protocols. + + Model: + + - A connector represents a connection to an outgoing service (IRC, XMPP, etc) + It satisfies a generic interface representing the actions that can be called + (send messages, join room, etc) + + - A handler represents a consumer of events happening on a connection + It satisfies a generic interface representing the events that can happend + (message received, rooms autojoined, etc) + + - A connector implements a given protocol that has an identifier + Each protocol identifier determines a namespace for user identifiers + and room identifiers which are globally unique for all connections using + this protocol. + For instance, a user can have two IRC conections to different servers. + Internally used user names and room identifiers must contain + the server name to be able to differentiate. +*/ + +type UserID string +type RoomID string + +type Connector interface { + // Set the handler that will receive events happening on this connection + SetHandler(handler Handler) + + // Configure (or reconfigure) the connector and attempt to connect + Configure(conf Configuration) error + + // Get the identifier of the protocol that is implemented by this connector + Protocol() string + + // Get the user id of the connected user + User() UserID + + // Set user information (nickname, picture, etc) + SetUserInfo(info *UserInfo) error + + // Set room information (name, description, picture, etc) + SetRoomInfo(roomId RoomID, info *RoomInfo) error + + // Try to join a channel + // If no error happens, it must fire a Handler.Joined event + Join(roomId RoomID) error + + // Leave a channel + Leave(roomId RoomID) + + // Send an event + Send(event *Event) error + + // Close the connection + Close() +} + +type Handler interface { + // Called when a room was joined (automatically or by call to Connector.Join) + Joined(roomId RoomID) + + // Called when the user left a room + Left(roomId RoomID) + + // Called when a user's info is updated (changed their nickname, status, etc) + // Can also be called with our own user ID when first loaded our user info + UserInfoUpdated(user UserID, info *UserInfo) + + // Called when a room's info was updated, + // or the first tome a room's info is retreived + RoomInfoUpdated(roomId RoomID, info *RoomInfo) + + // Called when an event occurs in a room + // This must not be called for events authored by the user of the connection + Event(event *Event) +} + +type EventType int +const ( + EVENT_JOIN EventType = iota + EVENT_LEAVE + EVENT_MESSAGE + EVENT_ACTION +) + +type Event struct { + Type EventType + + // UserID of the user that sent the event + // If this is a direct message event, this event can only have been authored + // by the user we are talking to (and not by ourself) + Author UserID + + // UserID of the targetted user in the case of a direct message, + // empty if targetting a room + Recipient UserID + + // RoomID of the room where the event happenned or of the targetted room, + // or empty string if it happenned by direct message + Room RoomID + + // Message text or action text + Message string + + // Attached files such as images + Attachements map[string]MediaObject +} + +type UserInfo struct { + Nickname string + Status string + Avatar MediaObject +} + +type RoomInfo struct { + Name string + Description string + Picture MediaObject +} + +type MediaObject interface { + Size() int + MimeType() string + + // AsBytes: must always be implemented + AsBytes() ([]byte, error) + + // AsString: not mandatory, may return an empty string + // If so, AsBytes() is the only way to retrieve the object + AsURL() string +} diff --git a/connector/irc/irc.go b/connector/irc/irc.go new file mode 100644 index 0000000..02f82eb --- /dev/null +++ b/connector/irc/irc.go @@ -0,0 +1,255 @@ +package irc + +import ( + "time" + "os" + "strings" + "fmt" + + . "git.deuxfleurs.fr/Deuxfleurs/easybridge/connector" + + "github.com/lrstanley/girc" +) + +// User id format: nickname@server +// Room id format: #room_name@server + +type IRC struct { + handler Handler + config Configuration + + connected bool + timeout int + + nick string + name string + server string + conn *girc.Client +} + +func (irc *IRC) SetHandler(h Handler) { + irc.handler = h +} + +func(irc *IRC) Protocol() string { + return "irc" +} + +func (irc *IRC) Configure(c Configuration) error { + irc.config = c + + irc.nick = c.GetString("nick") + irc.server = c.GetString("server") + + port, err := c.GetInt("port") + if err != nil { + return err + } + if port == 0 { + port = 6667 + } + + client := girc.New(girc.Config{ + Server: irc.server, + Port: port, + Nick: irc.nick, + User: irc.nick, + Debug: os.Stderr, + SSL: true, + }) + + client.Handlers.Add(girc.CONNECTED, irc.ircConnected) + //client.Handlers.Add(girc.DISCONNECTED, irc.ircDisconnected) + //client.Handlers.Add(girc.NICK, irc.ircNick) + client.Handlers.Add(girc.PRIVMSG, irc.ircPrivmsg) + client.Handlers.Add(girc.JOIN, irc.ircJoin) + client.Handlers.Add(girc.PART, irc.ircPart) + client.Handlers.Add(girc.RPL_NAMREPLY, irc.ircNamreply) + client.Handlers.Add(girc.RPL_TOPIC, irc.ircTopic) + + irc.conn = client + go irc.connectLoop(client) + + for i := 0; i < 42; i++ { + time.Sleep(time.Duration(1)*time.Second) + if irc.conn != client { + break + } + if irc.connected { + return nil + } + } + return fmt.Errorf("Failed to conncect after 42s attempting") +} + +func (irc *IRC) User() UserID { + return UserID(irc.nick + "@" + irc.server) +} + +func (irc *IRC) checkRoomId(id RoomID) (string, error) { + x := strings.Split(string(id), "@") + if len(x) != 2 || x[1] != irc.server || x[0][0] != '#' { + return "", fmt.Errorf("Invalid room ID: %s", id) + } + return x[0], nil +} + +func (irc *IRC) checkUserId(id UserID) (string, error) { + x := strings.Split(string(id), "@") + if len(x) != 2 || x[1] != irc.server || x[0][0] == '#' { + return "", fmt.Errorf("Invalid user ID: %s", id) + } + return x[0], nil +} + +func (irc *IRC) SetUserInfo(info *UserInfo) error { + return fmt.Errorf("Not implemented") +} + +func (irc *IRC) SetRoomInfo(roomId RoomID, info *RoomInfo) error { + ch, err := irc.checkRoomId(roomId) + if err != nil { + return err + } + + if info.Name != "" && info.Name != ch { + return fmt.Errorf("May not change IRC room name to other than %s", ch) + } + if info.Picture != nil { + return fmt.Errorf("Room picture not supported on IRC") + } + irc.conn.Cmd.Topic(ch, info.Description) + return nil +} + +func (irc *IRC) Join(roomId RoomID) error { + ch, err := irc.checkRoomId(roomId) + if err != nil { + return err + } + + irc.conn.Cmd.Join(ch) + return nil +} + +func (irc *IRC) Leave(roomId RoomID) { + ch, err := irc.checkRoomId(roomId) + if err != nil { + return + } + + irc.conn.Cmd.Part(ch) +} + +func (irc *IRC) Send(event *Event) error { + dest := "" + if event.Room != "" { + ch, err := irc.checkRoomId(event.Room) + if err != nil { + return err + } + dest = ch + } else if event.Recipient != "" { + ui, err := irc.checkUserId(event.Recipient) + if err != nil { + return err + } + dest = ui + } else { + return fmt.Errorf("Invalid target") + } + + if event.Attachements != nil && len(event.Attachements) > 0 { + // TODO find a way to send them using some hosing of some kind + return fmt.Errorf("Attachements not supported on IRC") + } + + if event.Type == EVENT_MESSAGE { + irc.conn.Cmd.Message(dest, event.Message) + } else if event.Type == EVENT_ACTION { + irc.conn.Cmd.Action(dest, event.Message) + } else { + return fmt.Errorf("Invalid event type") + } + return nil +} + +func (irc *IRC) Close() { + irc.conn.Close() + irc.conn = nil +} + +func (irc *IRC) connectLoop(c *girc.Client) { + irc.timeout = 10 + for { + if irc.conn != c { + return + } + if err := c.Connect(); err != nil { + irc.connected = false + fmt.Printf("IRC failed to connect / disconnected: %s", err) + fmt.Printf("Retrying in %ds", irc.timeout) + time.Sleep(time.Duration(irc.timeout) * time.Second) + irc.timeout *= 2 + } else { + return + } + } +} + +func (irc *IRC) ircConnected(c *girc.Client, e girc.Event) { + fmt.Printf("ircConnected ^^^^\n") + irc.timeout = 10 + irc.connected = true +} + +func (irc *IRC) ircPrivmsg(c *girc.Client, e girc.Event) { + ev := &Event{ + Type: EVENT_MESSAGE, + Author: UserID(e.Source.Name + "@" + irc.server), + Message: e.Last(), + } + if e.IsFromChannel() { + ev.Room = RoomID(e.Params[0] + "@" + irc.server) + } + if e.IsAction() { + ev.Type = EVENT_ACTION + } + irc.handler.Event(ev) +} + +func (irc *IRC) ircJoin(c *girc.Client, e girc.Event) { + room := RoomID(e.Params[0] + "@" + irc.server) + if e.Source.Name == irc.nick { + irc.handler.Joined(room) + } else { + ev := &Event{ + Type: EVENT_JOIN, + Author: UserID(e.Source.Name + "@" + irc.server), + Room: room, + } + irc.handler.Event(ev) + } +} + +func (irc *IRC) ircPart(c *girc.Client, e girc.Event) { + room := RoomID(e.Params[0] + "@" + irc.server) + if e.Source.Name == irc.nick { + irc.handler.Left(room) + } else { + ev := &Event{ + Type: EVENT_LEAVE, + Author: UserID(e.Source.Name + "@" + irc.server), + Room: room, + } + irc.handler.Event(ev) + } +} + +func (irc *IRC) ircNamreply(c *girc.Client, e girc.Event) { + fmt.Printf("TODO namreply params: %#v", e.Params) +} + +func (irc *IRC) ircTopic(c *girc.Client, e girc.Event) { + fmt.Printf("TODO topic params: %#v", e.Params) +} |