v2ray-core/common/protocol/user.go

package protocol

import (
	"time"

	"v2ray.com/core/common/errors"
)

var (
	ErrAccountMissing     = errors.New("Account is not specified.")
	ErrNonMessageType     = errors.New("Not a protobuf message.")
	ErrUnknownAccountType = errors.New("Unknown account type.")
)

func (v *User) GetTypedAccount() (Account, error) {
	if v.GetAccount() == nil {
		return nil, ErrAccountMissing
	}

	rawAccount, err := v.Account.GetInstance()
	if err != nil {
		return nil, err
	}
	if asAccount, ok := rawAccount.(AsAccount); ok {
		return asAccount.AsAccount()
	}
	if account, ok := rawAccount.(Account); ok {
		return account, nil
	}
	return nil, errors.New("Unknown account type: ", v.Account.Type)
}

func (v *User) GetSettings() UserSettings {
	settings := UserSettings{}
	switch v.Level {
	case 0:
		settings.PayloadTimeout = time.Second * 30
	case 1:
		settings.PayloadTimeout = time.Minute * 2
	default:
		settings.PayloadTimeout = time.Minute * 5
	}
	return settings
}

type UserSettings struct {
	PayloadTimeout time.Duration
}