mindoc/models/member.go

// Package models .
package models

import (
	"time"
	"github.com/astaxie/beego/orm"
	"github.com/lifei6671/godoc/utils"
	"github.com/lifei6671/godoc/conf"
	"github.com/astaxie/beego/logs"
	"errors"
	"regexp"
	"strings"
)

type Member struct {
	MemberId int		`orm:"pk;auto;unique;column(member_id)" json:"member_id"`
	Account string 		`orm:"size(100);unique;column(account)" json:"account"`
	Password string 	`orm:"size(1000);column(password)" json:"-"`
	Description string	`orm:"column(description);size(2000)" json:"description"`
	Email string 		`orm:"size(255);column(email);unique" json:"email"`
	Phone string 		`orm:"size(255);column(phone);null;default(null)" json:"phone"`
	Avatar string 		`orm:"size(1000);column(avatar)" json:"avatar"`
	//用户角色：0 超级管理员 /1 管理员/ 2 普通用户 .
	Role int		`orm:"column(role);type(int);default(1);index" json:"role"`
	RoleName string 	`orm:"-" json:"role_name"`
	Status int 		`orm:"column(status);type(int);default(0)" json:"status"`	//用户状态：0 正常/1 禁用
	CreateTime time.Time	`orm:"type(datetime);column(create_time);auto_now_add" json:"create_time"`
	CreateAt int		`orm:"type(int);column(create_at)" json:"create_at"`
	LastLoginTime time.Time	`orm:"type(datetime);column(last_login_time);null" json:"last_login_time"`

}

// TableName 获取对应数据库表名.
func (m *Member) TableName() string {
	return "members"
}
// TableEngine 获取数据使用的引擎.
func (m *Member) TableEngine() string {
	return "INNODB"
}

func (m *Member)TableNameWithPrefix() string {
	return conf.GetDatabasePrefix() +  m.TableName()
}

func NewMember() *Member {
	return &Member{}
}

// Login 用户登录.
func (m *Member) Login(account string,password string) (*Member,error) {
	o := orm.NewOrm()

	member := &Member{}

	err := o.QueryTable(m.TableNameWithPrefix()).Filter("account",account).Filter("status",0).One(member);

	if err != nil {
		logs.Error("用户登录 => ",err)
		return  member,ErrMemberNoExist
	}

	ok,err := utils.PasswordVerify(member.Password,password) ;

	if ok && err == nil {
		m.ResolveRoleName()
		return member,nil
	}

	return member,ErrorMemberPasswordError
}

// Add 添加一个用户.
func (m *Member) Add () (error) {
	o := orm.NewOrm()

	if ok,err := regexp.MatchString(conf.RegexpAccount,m.Account); m.Account == "" || !ok || err != nil {
		return errors.New("账号只能由英文字母数字组成，且在3-50个字符")
	}
	if m.Email == "" {
		return errors.New("邮箱不能为空")
	}
	if  ok,err := regexp.MatchString(conf.RegexpEmail,m.Email); !ok || err != nil || m.Email == "" {
		return errors.New("邮箱格式不正确")
	}
	if l :=  strings.Count(m.Password,""); l < 6 || l >= 50{
		return errors.New("密码不能为空且必须在6-50个字符之间")
	}
	if c,err :=  o.QueryTable(m.TableNameWithPrefix()).Filter("email",m.Email).Count(); err == nil && c > 0 {
		return  errors.New("邮箱已被使用")
	}

	hash ,err := utils.PasswordHash(m.Password);

	if  err != nil {
		return err
	}

	m.Password = hash

	_,err = o.Insert(m)

	if err != nil {
		return err
	}
	m.ResolveRoleName()
	return  nil
}

// Update 更新用户信息.
func (m *Member) Update(cols... string) (error) {
	o := orm.NewOrm()

	if m.Email == "" {
		return errors.New("邮箱不能为空")
	}
	if _,err := o.Update(m,cols...);err != nil {
		return err
	}
	return nil
}

func (m *Member) Find(id int) (*Member,error){
	o := orm.NewOrm()

	m.MemberId = id
	if err := o.Read(m); err != nil {
		return  m,err
	}
	m.ResolveRoleName()
	return m,nil
}

func (m *Member) ResolveRoleName (){
	if m.Role == conf.MemberSuperRole {
		m.RoleName = "超级管理员"
	}else if m.Role == conf.MemberAdminRole {
		m.RoleName = "管理员"
	}else if m.Role == conf.MemberGeneralRole {
		m.RoleName = "普通用户"
	}
}

func (m *Member) FindByAccount (account string) (*Member,error)  {
	o := orm.NewOrm()

	err := o.QueryTable(m.TableNameWithPrefix()).Filter("account",account).One(m)

	if err == nil {
		m.ResolveRoleName()
	}
	return m,err
}

func (m *Member) FindToPager(pageIndex, pageSize int) ([]*Member,int64,error)  {
	o := orm.NewOrm()

	var members []*Member

	offset := (pageIndex - 1) * pageSize

	totalCount,err := o.QueryTable(m.TableNameWithPrefix()).Count()

	if err != nil {
		return members,0,err
	}

	_,err = o.QueryTable(m.TableNameWithPrefix()).OrderBy("-member_id").Offset(offset).Limit(pageSize).All(&members)

	if err != nil {
		return members,0,err
	}

	for _,m := range members {
		m.ResolveRoleName()
	}
	return members,totalCount,nil
}


func (c *Member) IsAdministrator() bool {
	if c == nil || c.MemberId <= 0{
		return false
	}
	return c.Role == 0 || c.Role == 1
}

func (m *Member) FindByFieldFirst(field string,value interface{}) (*Member,error)  {
	o := orm.NewOrm()

	err := o.QueryTable(m.TableNameWithPrefix()).Filter(field,value).OrderBy("-member_id").One(m)

	return m,err
}