diff --git a/config.yml b/config.yml index 6856b74..017c9a9 100644 --- a/config.yml +++ b/config.yml @@ -117,6 +117,10 @@ dingtalk: enable-sync: false # 是否开启定时同步钉钉的任务 dept-sync-time: "0 30 2 * * *" # 部门同步任务的时间点 * * * * * * 秒 分 时 日 月 周, 请把时间设置在凌晨 1 ~ 5 点 user-sync-time: "0 30 3 * * *" # 用户同步任务的时间点 * * * * * * 秒 分 时 日 月 周, 请把时间设置在凌晨 1 ~ 5 点,注意请把用户同步的任务滞后于部门同步时间,比如部门为2点,则用户为3点 + dept-list: #部门列表,不设置则使用公司根部门,在开头加^表示不同步此部门,只需配置需要同步的部门ID + #- "1" #根组织ID + - "48456726" #需要同步的部门ID + #- "^61213417" #不同步的ID,不配置即只同步上一ID wecom: # 配置获取详细文档参考:http://ldapdoc.eryajf.net/pages/cf1698/ flag: "wecom" # 作为微信在平台的标识 diff --git a/config/config.go b/config/config.go index 2f10749..e918858 100644 --- a/config/config.go +++ b/config/config.go @@ -167,6 +167,7 @@ type DingTalkConfig struct { EnableSync bool `mapstructure:"enable-sync" json:"enableSync"` DeptSyncTime string `mapstructure:"dept-sync-time" json:"deptSyncTime"` UserSyncTime string `mapstructure:"user-sync-time" json:"userSyncTime"` + DeptList []string `mapstructure:"dept-list" json:"deptList"` } type WeComConfig struct { diff --git a/public/client/dingtalk/dingtalk.go b/public/client/dingtalk/dingtalk.go index 669c05a..0b82377 100644 --- a/public/client/dingtalk/dingtalk.go +++ b/public/client/dingtalk/dingtalk.go @@ -2,6 +2,7 @@ package dingtalk import ( "fmt" + "strconv" "strings" "github.com/eryajf/go-ldap-admin/config" @@ -14,16 +15,69 @@ import ( func GetAllDepts() (ret []map[string]interface{}, err error) { depts, err := InitDingTalkClient().FetchDeptList(1, true, "zh_CN") if err != nil { - return ret, err - } - ret = make([]map[string]interface{}, 0) - for _, dept := range depts.Dept { - ele := make(map[string]interface{}) - ele["id"] = dept.Id - ele["name"] = dept.Name - ele["parentid"] = dept.ParentId - ele["custom_name_pinyin"] = tools.ConvertToPinYin(dept.Name) - ret = append(ret, ele) + return ret, err + } + if len(config.Conf.DingTalk.DeptList) == 0 { + + ret = make([]map[string]interface{}, 0) + for _, dept := range depts.Dept { + ele := make(map[string]interface{}) + ele["id"] = dept.Id + ele["name"] = dept.Name + ele["parentid"] = dept.ParentId + ele["custom_name_pinyin"] = tools.ConvertToPinYin(dept.Name) + ret = append(ret, ele) + } + } else { + + // 遍历配置的部门ID列表获取数据进行处理 + // 从取得的所有部门列表中将配置的部门ID筛选出来再去请求其子部门过滤为1和为配置值的部门ID + ret = make([]map[string]interface{}, 0) + + for _, dept := range depts.Dept { + inset := false + for _, dep_s := range config.Conf.DingTalk.DeptList { + if strings.HasPrefix(dep_s, "^") { + continue + } + setdepid, _ := strconv.Atoi(dep_s) + if dept.Id == setdepid { + inset = true + break + } + } + if dept.Id == 1 || inset { + ele := make(map[string]interface{}) + ele["id"] = dept.Id + ele["name"] = dept.Name + ele["parentid"] = dept.ParentId + ele["custom_name_pinyin"] = tools.ConvertToPinYin(dept.Name) + ret = append(ret, ele) + } + } + + for _, dep_s := range config.Conf.DingTalk.DeptList { + dept_id := dep_s + + if strings.HasPrefix(dep_s, "^") || dept_id == "1" { + continue + } + depid, _ := strconv.Atoi(dept_id) + depts, err := InitDingTalkClient().FetchDeptList(depid, true, "zh_CN") + + if err != nil { + return ret, err + } + + for _, dept := range depts.Dept { + ele := make(map[string]interface{}) + ele["id"] = dept.Id + ele["name"] = dept.Name + ele["parentid"] = dept.ParentId + ele["custom_name_pinyin"] = tools.ConvertToPinYin(dept.Name) + ret = append(ret, ele) + } + } } return }