corp_user.go

package wechat

import (
	"fmt"
	"log"
	"strings"

	"github.com/esap/wechat/util"
)

// WXAPI 企业号用户列表接口
const (
	WXAPI_GetUser        = WXAPI_ENT + "user/getuserinfo?access_token=%s&code=%s"
	WXAPI_GetUserInfo    = WXAPI_ENT + "user/get?access_token=%s&userid=%s"
	WXAPI_UserList       = WXAPI_ENT + `user/list?access_token=%s&department_id=1&fetch_child=1`
	WXAPI_UserSimpleList = WXAPI_ENT + `user/simplelist?access_token=%s&department_id=1&fetch_child=1`
	WXAPI_UserAdd        = WXAPI_ENT + `user/create?access_token=`
	WXAPI_UserUpdate     = WXAPI_ENT + `user/update?access_token=`
	WXAPI_UserDel        = WXAPI_ENT + `user/delete?access_token=`
)

// UserOauth 用户鉴权信息
type UserOauth struct {
	WxErr
	UserId   string
	DeviceId string
	OpenId   string
}

// GetUserOauth 通过code鉴权
func (s *Server) GetUserOauth(code string) (o UserOauth, err error) {
	url := fmt.Sprintf(WXAPI_GetUser, s.GetAccessToken(), code)
	if err = util.GetJson(url, &o); err != nil {
		return
	}
	err = o.Error()
	return
}

// GetUserOauth 通过code鉴权
func GetUserOauth(code string) (userOauth UserOauth, err error) {
	return std.GetUserOauth(code)
}

// UserInfo 用户信息
type UserInfo struct {
	WxErr       `json:"-"`
	UserId      string `json:"userid"`
	Name        string `json:"name"`
	Department  []int  `json:"department"`
	Dept        int    `json:"dept"`
	DeptName    string `json:"deptname"`
	Position    string `json:"position,omitempty"`
	Mobile      string `json:"mobile"`
	Gender      string `json:"gender,omitempty"`
	Email       string `json:"email,omitempty"`
	IsLeader    int    `json:"isleader,omitempty"`
	Telephone   string `json:"telephone,omitempty"`
	EnglishName string `json:"english_name,omitempty"`
	WeixinId    string `json:"-"`
	Avatar      string `json:"avatar,omitempty"`
	Status      int    `json:"-"`
	ExtAttr     struct {
		Attrs []Extattr
	} `json:"-"`
}

// Extattr 额外属性
type Extattr struct {
	Name  string `json:"name"`
	Value string `json:"value"`
}

// UserAdd 添加用户
func (s *Server) UserAdd(user *UserInfo) (err error) {
	return s.doUpdate(WXAPI_UserAdd, user)
}

// UserUpdate 添加用户
func (s *Server) UserUpdate(user *UserInfo) (err error) {
	return s.doUpdate(WXAPI_UserUpdate, user)
}

// UserDelete 删除用户
func (s *Server) UserDelete(user string) (err error) {
	e := new(WxErr)
	if err = util.GetJson(WXAPI_UserDel+s.GetUserAccessToken()+"&userid="+user, e); err != nil {
		return
	}
	return e.Error()
}

// GetUserInfo 从企业号通过userId获取用户信息
func (s *Server) GetUserInfo(userId string) (user UserInfo, err error) {
	url := fmt.Sprintf(WXAPI_GetUserInfo, s.GetUserAccessToken(), userId)
	if err = util.GetJson(url, &user); err != nil {
		return
	}
	err = user.Error()
	return
}

// GetUser 从缓存获取用户信息
func (s *Server) GetUser(userid string) *UserInfo {
	for _, v := range s.UserList.UserList {
		if v.UserId == userid {
			v.DeptName = s.GetDeptName(v.Department[0])
			return &v
		}
	}
	return nil
}

// GetUserName 通过账号获取用户信息
func (s *Server) GetUserName(userid string) string {
	for _, v := range s.UserList.UserList {
		if v.UserId == userid {
			return v.Name
		}
	}
	return ""
}

// userList 用户列表
type userList struct {
	WxErr
	UserList []UserInfo
}

// SyncUserList 获取用户列表
func (s *Server) SyncUserList() (err error) {
	s.UserList, err = s.GetUserList()
	if err != nil {
		log.Printf("[%v::%v]获取用户列表失败:%v", s.AppId, s.AgentId, err)
	}
	return
}

// GetUserList 获取用户详情列表
func (s *Server) GetUserList() (u userList, err error) {
	url := fmt.Sprintf(WXAPI_UserList, s.GetUserAccessToken())
	if err = util.GetJson(url, &u); err != nil {
		return
	}
	err = u.Error()
	return
}

// GetUserSimpleList 获取用户列表
func (s *Server) GetUserSimpleList() (u userList, err error) {
	url := fmt.Sprintf(WXAPI_UserSimpleList, s.GetUserAccessToken())
	if err = util.GetJson(url, &u); err != nil {
		return
	}
	err = u.Error()
	return
}

// GetUserIdList 获取用户列表
func (s *Server) GetUserIdList() (userlist []string) {
	userlist = make([]string, 0)
	ul, err := s.GetUserSimpleList()
	if err != nil {
		return
	}
	for _, v := range ul.UserList {
		userlist = append(userlist, v.UserId)
	}
	return
}

func (s *Server) doUpdate(uri string, i interface{}) (err error) {
	url := uri + s.GetUserAccessToken()
	e := new(WxErr)
	if err = util.PostJsonPtr(url, i, e); err != nil {
		return
	}
	return e.Error()
}

// GetGender 获取性别
func GetGender(s string) string {
	if s == "1" {
		return "男"
	}
	if s == "2" {
		return "女"
	}
	return "未定义"
}

var toUserReplacer = strings.NewReplacer(",", "|", "，", "|")

// GetToUser 获取acl所包含的所有用户ID,结果形式：userId1|userId2|userId3...
func (s *Server) GetToUser(acl interface{}) string {
	s1 := strings.TrimSpace(acl.(string))
	if strings.ToLower(s1) == "@all" {
		return "@all"
	}
	arr := strings.Split(toUserReplacer.Replace(s1), "|")
	for k, toUser := range arr {
		for _, v := range s.UserList.UserList {
			if v.Name == toUser {
				arr[k] = v.UserId
			}
		}
	}
	return strings.Join(arr, "|")
}

// CheckUserAcl 测试权限，对比user的账号，姓名是否包含于acl
func (s *Server) CheckUserAcl(userid, acl string) bool {
	acl = strings.TrimSpace(acl)
	if acl == "" {
		return false
	}
	if strings.ToLower(acl) == "@all" {
		return true
	}
	acl = "|" + toUserReplacer.Replace(acl) + "|"
	u := s.GetUser(userid)
	if u == nil {
		return false
	}

	return strings.Contains(acl, "|"+u.Name+"|") || strings.Contains(acl, "|"+u.UserId+"|")
}