mirror of
https://github.com/yunionio/cloudpods.git
synced 2026-06-21 18:52:48 +08:00
289 lines
6.7 KiB
Go
289 lines
6.7 KiB
Go
package modules
|
||
|
||
import (
|
||
"fmt"
|
||
"sync"
|
||
|
||
"yunion.io/x/onecloud/pkg/util/httputils"
|
||
|
||
"yunion.io/x/jsonutils"
|
||
"yunion.io/x/onecloud/pkg/mcclient"
|
||
)
|
||
|
||
type ProjectManagerV3 struct {
|
||
ResourceManager
|
||
}
|
||
|
||
var (
|
||
Projects ProjectManagerV3
|
||
)
|
||
|
||
func (this *ProjectManagerV3) _join(s *mcclient.ClientSession, pid, uid, rid, resource string, ch chan int) error {
|
||
defer func() {
|
||
ch <- 1
|
||
}()
|
||
if resource == "users" {
|
||
_, err := RolesV3.PutInContexts(s, rid, nil, []ManagerContext{{&Projects, pid}, {&UsersV3, uid}})
|
||
if err != nil {
|
||
return err
|
||
}
|
||
} else if resource == "groups" {
|
||
_, err := RolesV3.PutInContexts(s, rid, nil, []ManagerContext{{&Projects, pid}, {&Groups, uid}})
|
||
if err != nil {
|
||
return err
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (this *ProjectManagerV3) _leave(s *mcclient.ClientSession, pid string, uid string, rid string, ch chan int) error {
|
||
defer func() {
|
||
ch <- 1
|
||
}()
|
||
_, err := RolesV3.DeleteInContexts(s, rid, nil, []ManagerContext{{&Projects, pid}, {&UsersV3, uid}})
|
||
if err != nil {
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (this *ProjectManagerV3) DoLeaveProject(s *mcclient.ClientSession, params jsonutils.JSONObject) (jsonutils.JSONObject, error) {
|
||
// params format:
|
||
// {
|
||
// "uid": "CCCGwOsrpp6h", // may be userid, may be group id
|
||
// "resource": "" default users, else: groups
|
||
// "pids": [{"rid": "arstarst", "pid": "ooienst"}, {}...]
|
||
// }
|
||
|
||
ret := jsonutils.NewDict()
|
||
uid, e := params.GetString("uid")
|
||
if e != nil {
|
||
return ret, e
|
||
}
|
||
pids, e := params.GetArray("pids")
|
||
if e != nil {
|
||
return ret, e
|
||
}
|
||
|
||
chs := make([]chan int, len(pids))
|
||
|
||
for i, pid := range pids {
|
||
_pid, e := pid.GetString("pid")
|
||
if e != nil {
|
||
return ret, e
|
||
}
|
||
_rid, e := pid.GetString("rid")
|
||
if e != nil {
|
||
return ret, e
|
||
}
|
||
|
||
chs[i] = make(chan int)
|
||
go this._leave(s, _pid, uid, _rid, chs[i])
|
||
}
|
||
|
||
for _, ch := range chs {
|
||
<-ch
|
||
}
|
||
return ret, nil
|
||
}
|
||
|
||
// Add one user to Many Projects
|
||
func (this *ProjectManagerV3) DoJoinProject(s *mcclient.ClientSession, params jsonutils.JSONObject) (jsonutils.JSONObject, error) {
|
||
// params format:
|
||
// {
|
||
// "uid": "CCCGwOsrpp6h",
|
||
// "rid": "aDeKQx1PmcTd",
|
||
// "pids": ["L6ssbAJUG3rC", "pu8lkunxP4z8"]
|
||
// "resource": "users" or "groups"
|
||
// }
|
||
|
||
ret := jsonutils.NewDict()
|
||
uid, e := params.GetString("uid")
|
||
if e != nil {
|
||
return ret, e
|
||
}
|
||
rid, e := params.GetString("rid")
|
||
if e != nil {
|
||
return ret, e
|
||
}
|
||
pids, e := params.GetArray("pids")
|
||
if e != nil {
|
||
return ret, e
|
||
}
|
||
|
||
resource, _ := params.GetString("resource")
|
||
|
||
if len(resource) == 0 {
|
||
resource = "users"
|
||
}
|
||
|
||
chs := make([]chan int, len(pids))
|
||
|
||
for i, pid := range pids {
|
||
_pid, e := pid.GetString()
|
||
if e != nil {
|
||
return ret, e
|
||
}
|
||
chs[i] = make(chan int)
|
||
go this._join(s, _pid, uid, rid, resource, chs[i])
|
||
}
|
||
|
||
for _, ch := range chs {
|
||
<-ch
|
||
}
|
||
return ret, nil
|
||
}
|
||
|
||
// Add Many user[uids] to project(pid) with role(rid)
|
||
func (this *ProjectManagerV3) DoProjectBatchJoin(s *mcclient.ClientSession, params jsonutils.JSONObject) (jsonutils.JSONObject, error) {
|
||
// params format:
|
||
// {
|
||
// "uids": ["CCCGwOsrpp6h", "aroisetna"],
|
||
// "rid": "aDeKQx1PmcTd", // role id
|
||
// "pid": "L6ssbAJUG3rC"
|
||
// }
|
||
|
||
ret := jsonutils.NewDict()
|
||
_ids, e := params.GetArray("ids")
|
||
|
||
if e != nil {
|
||
return ret, e
|
||
}
|
||
ids := make([]string, 0)
|
||
for _, u := range _ids {
|
||
name, _ := u.GetString()
|
||
ids = append(ids, name)
|
||
}
|
||
resource, e := params.GetString("resource")
|
||
if e != nil {
|
||
return ret, e
|
||
}
|
||
if resource != "users" && resource != "groups" {
|
||
return ret, fmt.Errorf("不支持的 resource type")
|
||
}
|
||
rid, e := params.GetString("rid")
|
||
if e != nil {
|
||
return ret, e
|
||
}
|
||
pid, e := params.GetString("pid")
|
||
if e != nil {
|
||
return ret, e
|
||
}
|
||
|
||
BatchDo(ids, func(id string) (jsonutils.JSONObject, error) {
|
||
if resource == "users" {
|
||
return RolesV3.PutInContexts(
|
||
s,
|
||
rid,
|
||
nil,
|
||
[]ManagerContext{
|
||
{&Projects,
|
||
pid},
|
||
{&UsersV3, id}})
|
||
}
|
||
return RolesV3.PutInContexts(
|
||
s,
|
||
rid,
|
||
nil,
|
||
[]ManagerContext{
|
||
{&Projects,
|
||
pid},
|
||
{&Groups, id}})
|
||
|
||
})
|
||
|
||
return ret, nil
|
||
}
|
||
|
||
// Add Many user[uids] to project(pid) with role(rid)
|
||
func (this *ProjectManagerV3) DoProjectBatchDeleteUserGroup(s *mcclient.ClientSession, pid string, params jsonutils.JSONObject) (jsonutils.JSONObject, error) {
|
||
// params format:
|
||
// {
|
||
// id: project id;
|
||
// params: {items: [
|
||
// id:, role_id:, res_type:
|
||
// ]}
|
||
// }
|
||
|
||
var wg sync.WaitGroup
|
||
ret := jsonutils.NewDict()
|
||
items, e := params.GetArray("items")
|
||
|
||
if e != nil {
|
||
return ret, e
|
||
}
|
||
|
||
for _, item := range items {
|
||
|
||
wg.Add(1)
|
||
id, _ := item.GetString("id")
|
||
role_id, _ := item.GetString("role_id")
|
||
res_type, _ := item.GetString("res_type")
|
||
|
||
go func(id string) {
|
||
defer wg.Done()
|
||
if res_type == "user" {
|
||
RolesV3.DeleteInContexts(s, role_id, nil, []ManagerContext{{&Projects, pid}, {&UsersV3, id}})
|
||
} else if res_type == "group" {
|
||
RolesV3.DeleteInContexts(s, role_id, nil, []ManagerContext{{&Projects, pid}, {&Groups, id}})
|
||
}
|
||
}(id)
|
||
}
|
||
|
||
wg.Wait()
|
||
return ret, nil
|
||
}
|
||
|
||
func (this *ProjectManagerV3) Delete(session *mcclient.ClientSession, id string, body jsonutils.JSONObject) (jsonutils.JSONObject, error) {
|
||
return this.DeleteInContexts(session, id, body, nil)
|
||
}
|
||
|
||
func (this *ProjectManagerV3) DeleteInContexts(session *mcclient.ClientSession, id string, body jsonutils.JSONObject, ctxs []ManagerContext) (jsonutils.JSONObject, error) {
|
||
if ctxs == nil {
|
||
p := jsonutils.NewDict()
|
||
p.Add(jsonutils.NewString(id), "tenant")
|
||
ret, e := Hosts.List(session, p)
|
||
if e != nil {
|
||
return nil, e
|
||
} else {
|
||
if ret.Total > 0 {
|
||
err := &httputils.JSONClientError{}
|
||
err.Code = 403
|
||
err.Details = fmt.Sprintf("该项目（%s）下存在云服务器，请清除后重试", id)
|
||
return nil, err
|
||
}
|
||
}
|
||
}
|
||
|
||
return this.deleteInContexts(session, id, nil, body, ctxs)
|
||
}
|
||
|
||
func (this *ProjectManagerV3) BatchDelete(session *mcclient.ClientSession, idlist []string, body jsonutils.JSONObject) []SubmitResult {
|
||
return this.BatchDeleteInContexts(session, idlist, body, nil)
|
||
}
|
||
|
||
func (this *ProjectManagerV3) BatchDeleteInContexts(session *mcclient.ClientSession, idlist []string, body jsonutils.JSONObject, ctxs []ManagerContext) []SubmitResult {
|
||
return BatchDo(idlist, func(id string) (jsonutils.JSONObject, error) {
|
||
return this.DeleteInContexts(session, id, body, ctxs)
|
||
})
|
||
}
|
||
|
||
func (this *ProjectManagerV3) AddTags(session *mcclient.ClientSession, id string, tags []string) error {
|
||
path := fmt.Sprintf("/projects/%s/tags", id)
|
||
body := jsonutils.NewDict()
|
||
body.Add(jsonutils.NewStringArray(tags), "tags")
|
||
_, err := this._put(session, path, body, "")
|
||
if err != nil {
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func init() {
|
||
Projects = ProjectManagerV3{NewIdentityV3Manager("project", "projects",
|
||
[]string{},
|
||
[]string{"ID", "Name", "Domain_Id", "Enabled", "Description"})}
|
||
|
||
register(&Projects)
|
||
}
|