harness-drone/pkg/model/user.go

package model

import (
	"errors"
	"fmt"
	"regexp"
	"time"

	"code.google.com/p/go.crypto/bcrypt"
)

var (
	ErrInvalidUserName = errors.New("Invalid User Name")
	ErrInvalidPassword = errors.New("Invalid Password")
	ErrInvalidEmail    = errors.New("Invalid Email Address")
)

// Gravatar URL pattern
var GravatarPattern = "https://gravatar.com/avatar/%s?s=%v&d=identicon"

// Simple regular expression used to verify that an email
// address matches the expected standard format.
var RegexpEmail = regexp.MustCompile(`^[^@]+@[^@.]+\.[^@.]+`)

type User struct {
	ID       int64     `meddler:"id,pk"            json:"id"`
	Email    string    `meddler:"email"            json:"email"`
	Password string    `meddler:"password"         json:"-"`
	Token    string    `meddler:"token"            json:"-"`
	Name     string    `meddler:"name"             json:"name"`
	Gravatar string    `meddler:"gravatar"         json:"gravatar"`
	Created  time.Time `meddler:"created,utctime"  json:"created"`
	Updated  time.Time `meddler:"updated,utctime"  json:"updated"`
	Admin    bool      `meddler:"admin"            json:"-"`

	// GitHub OAuth2 token for accessing public repositories.
	GithubLogin string `meddler:"github_login" json:"-"`
	GithubToken string `meddler:"github_token" json:"-"`

	// Bitbucket OAuth1.0a token and token secret.
	BitbucketLogin  string `meddler:"bitbucket_login"  json:"-"`
	BitbucketToken  string `meddler:"bitbucket_token"  json:"-"`
	BitbucketSecret string `meddler:"bitbucket_secret" json:"-"`

	GitlabToken string `meddler:"gitlab_token" json:"-"`
}

// Creates a new User from the given Name and Email.
func NewUser(name, email string) *User {
	user := User{}
	user.Name = name
	user.Token = createToken()
	user.SetEmail(email)
	return &user
}

// Returns the Gravatar Image URL.
func (u *User) Image() string      { return fmt.Sprintf(GravatarPattern, u.Gravatar, 42) }
func (u *User) ImageSmall() string { return fmt.Sprintf(GravatarPattern, u.Gravatar, 32) }
func (u *User) ImageLarge() string { return fmt.Sprintf(GravatarPattern, u.Gravatar, 160) }

// Set the email address and calculate the
// Gravatar hash.
func (u *User) SetEmail(email string) {
	u.Email = email
	u.Gravatar = createGravatar(email)
}

// Set the password and hash with bcrypt
func (u *User) SetPassword(password string) error {
	// validate the password is an appropriate size
	switch {
	case len(password) < 6:
		return ErrInvalidPassword
	case len(password) > 256:
		return ErrInvalidPassword
	}

	// convert the password to a hash
	b, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost)
	if err != nil {
		return err
	}
	// update the user
	u.Password = string(b)
	return nil
}

// Validate verifies all required fields are correctly populated.
func (u *User) Validate() error {
	switch {
	case len(u.Name) == 0:
		return ErrInvalidUserName
	case len(u.Name) >= 255:
		return ErrInvalidUserName
	case len(u.Email) == 0:
		return ErrInvalidEmail
	case len(u.Email) >= 255:
		return ErrInvalidEmail
	case RegexpEmail.MatchString(u.Email) == false:
		return ErrInvalidEmail
	default:
		return nil
	}
}

// ComparePassword compares the supplied password to
// the user password stored in the database.
func (u *User) ComparePassword(password string) error {
	return bcrypt.CompareHashAndPassword([]byte(u.Password), []byte(password))
}