irc operators

irc/client.go

@@ -15,6 +15,7 @@ type Client struct {
 	hostname   string
 	invisible  bool
 	nick       string
+	operator   bool
 	realname   string
 	registered bool
 	replies    chan<- Reply
@@ -109,6 +110,10 @@ func (c *Client) ModeString() (str string) {
 	if c.invisible {
 		str += Invisible.String()
 	}
+	if c.operator {
+		str += Operator.String()
+	}
+
 	if len(str) > 0 {
 		str = "+" + str
 	}

irc/commands.go

@@ -21,6 +21,7 @@ var (
 		"JOIN":    NewJoinCommand,
 		"MODE":    NewModeCommand,
 		"NICK":    NewNickCommand,
+		"OPER":    NewOperCommand,
 		"PART":    NewPartCommand,
 		"PASS":    NewPassCommand,
 		"PING":    NewPingCommand,
@@ -541,3 +542,25 @@ func NewWhoCommand(args []string) (editableCommand, error) {
 func (msg *WhoCommand) String() string {
 	return fmt.Sprintf("WHO(mask=%s, operatorOnly=%s)", msg.mask, msg.operatorOnly)
 }
+
+type OperCommand struct {
+	BaseCommand
+	name     string
+	password string
+}
+
+func (msg *OperCommand) String() string {
+	return fmt.Sprintf("OPER(name=%s, password=%s)", msg.name, msg.password)
+}
+
+// OPER <name> <password>
+func NewOperCommand(args []string) (editableCommand, error) {
+	if len(args) < 2 {
+		return nil, NotEnoughArgsError
+	}
+
+	return &OperCommand{
+		name:     args[0],
+		password: args[1],
+	}, nil
+}

irc/config.go

@@ -6,10 +6,16 @@ import (
 )
 
 type Config struct {
+	Name      string
+	Listen    string
+	Password  string
+	Operators []OperatorConfig
+	Debug     map[string]bool
+}
+
+type OperatorConfig struct {
 	Name     string
-	Listen   string
 	Password string
-	Debug    map[string]bool
 }
 
 func LoadConfig() (config *Config, err error) {

irc/reply.go

@@ -206,9 +206,9 @@ func RplEndOfNames(channel *Channel) Reply {
 		"%s :End of NAMES list", channel.name)
 }
 
+// :You are now an IRC operator
 func RplYoureOper(server *Server) Reply {
-	return NewNumericReply(server, RPL_YOUREOPER,
+	return NewNumericReply(server, RPL_YOUREOPER, ":You are now an IRC operator")
-		":You are now an IRC operator")
 }
 
 func RplWhoisUser(server *Server, client *Client) Reply {

irc/server.go

@@ -10,25 +10,32 @@ import (
 )
 
 type Server struct {
-	channels ChannelNameMap
+	channels  ChannelNameMap
-	commands chan<- Command
+	commands  chan<- Command
-	ctime    time.Time
+	ctime     time.Time
-	hostname string
+	hostname  string
-	name     string
+	name      string
-	password string
+	operators map[string]string
-	clients  ClientNameMap
+	password  string
+	clients   ClientNameMap
 }
 
 func NewServer(config *Config) *Server {
 	commands := make(chan Command)
 	server := &Server{
-		channels: make(ChannelNameMap),
+		channels:  make(ChannelNameMap),
-		clients:  make(ClientNameMap),
+		clients:   make(ClientNameMap),
-		commands: commands,
+		commands:  commands,
-		ctime:    time.Now(),
+		ctime:     time.Now(),
-		name:     config.Name,
+		name:      config.Name,
-		password: config.Password,
+		operators: make(map[string]string),
+		password:  config.Password,
 	}
+
+	for _, opConf := range config.Operators {
+		server.operators[opConf.Name] = opConf.Password
+	}
+
 	go server.receiveCommands(commands)
 	go server.listen(config.Listen)
 	return server
@@ -352,3 +359,17 @@ func (msg *WhoCommand) HandleServer(server *Server) {
 
 	client.Reply(RplEndOfWho(server, mask))
 }
+
+func (msg *OperCommand) HandleServer(server *Server) {
+	client := msg.Client()
+
+	if server.operators[msg.name] != msg.password {
+		client.Reply(ErrPasswdMismatch(server))
+		return
+	}
+
+	client.operator = true
+
+	client.Reply(RplYoureOper(server))
+	client.Reply(RplUModeIs(server, client))
+}