mirror of
https://github.com/ergochat/ergo.git
synced 2024-11-10 22:19:31 +01:00
85 lines
2.5 KiB
Go
85 lines
2.5 KiB
Go
|
// 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
|
||
|
}
|