corp_user.go

package wechat

import (
	"fmt"
	"log"
	"strings"
	"time"

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

const (
	// CorpAPIGetUserOauth 企业微信用户oauth2认证接口
	CorpAPIGetUserOauth = CorpAPI + "user/getuserinfo?access_token=%s&code=%s"

	// CorpAPIUserList 企业微信用户列表
	CorpAPIUserList       = CorpAPI + `user/list?access_token=%s&department_id=1&fetch_child=1`
	CorpAPIUserSimpleList = CorpAPI + `user/simplelist?access_token=%s&department_id=1&fetch_child=1`

	// CorpAPIUserGet 企业微信用户接口
	CorpAPIUserGet    = CorpAPI + "user/get?access_token=%s&userid=%s"
	CorpAPIUserAdd    = CorpAPI + `user/create?access_token=`
	CorpAPIUserUpdate = CorpAPI + `user/update?access_token=`
	CorpAPIUserDel    = CorpAPI + `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(CorpAPIGetUserOauth, s.GetAccessToken(), code)
	if err = util.GetJson(url, &o); err != nil {
		return
	}
	err = o.Error()
	return
}

// UserInfo 用户信息
type UserInfo struct {
	WxErr          `json:"-"`
	UserId         string `json:"userid"`
	Name           string `json:"name"`
	Alias          string `json:"alias"`
	Department     []int  `json:"department"`
	IsLeaderInDept []int  `json:"is_leader_in_dept,omitempty"`
	Order          []int  `json:"order"`
	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"` // old attr
	AavatarMediaid string `json:"avatar_mediaid,omitempty"`
	Enable         int    `json:"enable,omitempty"`
	Telephone      string `json:"telephone,omitempty"`
	WeixinId       string `json:"-"`
	Avatar         string `json:"avatar,omitempty"`
	Status         int    `json:"-"`
	ToInvite       bool   `json:"to_invite"`
	ExtAttr        struct {
		Attrs []Extattr `json:"attrs"`
	} `json:"extattr"`
}

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

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

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

// UserDelete 删除用户
func (s *Server) UserDelete(user string) (err error) {
	e := new(WxErr)
	if err = util.GetJson(CorpAPIUserDel+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(CorpAPIUserGet, 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 &UserInfo{}
}

// 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
}

// FetchUserList 定期获取AccessToken
func (s *Server) FetchUserList() {
	i := 0
	go func() {
		for {
			if s.SyncDeptList() == nil {
				if s.SyncUserList() != nil && i < 2 {
					i++
					Println("尝试再次获取用户列表(", i, ")")
					continue
				}
				i = 0
			}
			s.SyncTagList()
			time.Sleep(FetchDelay)
		}
	}()
}

// 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(CorpAPIUserList, 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(CorpAPIUserSimpleList, 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+"|")
}