accounts: Add very initial, extremely broken account work (not including config changes)

irc/registration.go

@@ -0,0 +1,84 @@
+// Copyright (c) 2016- Daniel Oaks <daniel@danieloaks.net>
+// released under the MIT license
+
+package irc
+
+import (
+	"errors"
+	"fmt"
+	"strconv"
+	"strings"
+	"time"
+
+	"github.com/DanielOaks/girc-go/ircmsg"
+	"github.com/tidwall/buntdb"
+)
+
+var (
+	errAccountCreation = errors.New("Account could not be created")
+)
+
+// AccountRegistration manages the registration of accounts.
+type AccountRegistration struct {
+	Enabled                          bool
+	EnabledRegistrationCallbackTypes []string
+}
+
+// NewAccountRegistration returns a new AccountRegistration, configured correctly.
+func NewAccountRegistration(config AccountRegistrationConfig) (accountReg AccountRegistration) {
+	if config.Enabled {
+		accountReg.Enabled = true
+		accountReg.EnabledRegistrationCallbackTypes = config.EnabledCallbacks
+	}
+	return accountReg
+}
+
+// regHandler parses the REG command.
+func regHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
+	subcommand := strings.ToLower(msg.Params[0])
+
+	if subcommand == "create" {
+		client.Notice("Parsing CREATE")
+
+		// get and sanitise account name
+		account := NewName(msg.Params[1])
+		if !account.IsNickname() || msg.Params[1] == "*" {
+			client.Send(nil, server.nameString, ERR_REG_UNSPECIFIED_ERROR, client.nickString, msg.Params[1], "Account name is not valid")
+			return false
+		}
+		accountString := account.String()
+
+		// check whether account exists
+		// do it all in one write tx to prevent races
+		err := server.store.Update(func(tx *buntdb.Tx) error {
+			accountKey := fmt.Sprintf("account %s exists", accountString)
+
+			_, err := tx.Get(accountKey)
+			if err != buntdb.ErrNotFound {
+				//TODO(dan): if account verified key doesn't exist account is not verified, calc the maximum time without verification and expire and continue if need be
+				client.Send(nil, server.nameString, ERR_ACCOUNT_ALREADY_EXISTS, client.nickString, msg.Params[1], "Account already exists")
+				return errAccountCreation
+			}
+
+			registeredTimeKey := fmt.Sprintf("account %s registered.time", accountString)
+
+			tx.Set(accountKey, "1", nil)
+			tx.Set(registeredTimeKey, strconv.FormatInt(time.Now().Unix(), 10), nil)
+			return nil
+		})
+
+		// account could not be created and relevant numerics have been dispatched, abort
+		if err != nil {
+			return false
+		}
+
+		// account didn't already exist, continue with account creation and dispatching verification (if required)
+
+	} else if subcommand == "verify" {
+		client.Notice("Parsing VERIFY")
+	} else {
+		client.Send(nil, server.nameString, ERR_UNKNOWNERROR, client.nickString, "REG", msg.Params[0], "Unknown subcommand")
+	}
+
+	return false
+}