Rainbond/pkg/api/region/cloudAuth.go

113 lines
3.0 KiB
Go
Raw Normal View History

2018-03-14 14:12:26 +08:00
// Copyright (C) 2014-2018 Goodrain Co., Ltd.
2017-12-08 17:14:12 +08:00
// RAINBOND, Application Management Platform
2018-03-14 14:33:31 +08:00
2017-12-08 17:14:12 +08:00
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version. For any non-GPL usage of Rainbond,
// one or multiple Commercial Licenses authorized by Goodrain Co., Ltd.
// must be obtained first.
2018-03-14 14:33:31 +08:00
2017-12-08 17:14:12 +08:00
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
2018-03-14 14:33:31 +08:00
2017-12-08 17:14:12 +08:00
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
package region
import (
"fmt"
"github.com/pquerna/ffjson/ffjson"
"github.com/Sirupsen/logrus"
api_model "github.com/goodrain/rainbond/pkg/api/model"
2017-12-25 19:59:35 +08:00
"github.com/goodrain/rainbond/pkg/api/util"
2017-12-08 17:14:12 +08:00
)
//DefineCloudAuth DefineCloudAuth
2017-12-25 19:49:57 +08:00
func (t *tenant) DefineCloudAuth(gt *api_model.GetUserToken) DefineCloudAuthInterface {
2017-12-08 17:14:12 +08:00
return &DefineCloudAuth{
GT: gt,
}
}
//DefineCloudAuth DefineCloudAuth
type DefineCloudAuth struct {
GT *api_model.GetUserToken
}
//DefineCloudAuthInterface DefineCloudAuthInterface
type DefineCloudAuthInterface interface {
GetToken() ([]byte, error)
PostToken() ([]byte, error)
2017-12-08 17:14:12 +08:00
PutToken() error
}
//GetToken GetToken
func (d *DefineCloudAuth) GetToken() ([]byte, error) {
2017-12-25 19:59:35 +08:00
resp, code, err := request(
2017-12-08 17:14:12 +08:00
fmt.Sprintf("/cloud/auth/%s", d.GT.Body.EID),
"GET",
nil,
)
if err != nil {
2017-12-25 19:59:35 +08:00
return nil, util.CreateAPIHandleError(code, err)
2017-12-08 17:14:12 +08:00
}
2017-12-25 19:59:35 +08:00
if code > 400 {
if code == 404 {
return nil, util.CreateAPIHandleError(code, fmt.Errorf("eid %s is not exist", d.GT.Body.EID))
2017-12-08 17:14:12 +08:00
}
2017-12-25 19:59:35 +08:00
return nil, util.CreateAPIHandleError(code, fmt.Errorf("get eid infos %s failed", d.GT.Body.EID))
2017-12-08 17:14:12 +08:00
}
//valJ, err := simplejson.NewJson(resp)
return resp, nil
}
//PostToken PostToken
func (d *DefineCloudAuth) PostToken() ([]byte, error) {
2017-12-08 17:14:12 +08:00
data, err := ffjson.Marshal(d.GT.Body)
if err != nil {
return nil, err
2017-12-08 17:14:12 +08:00
}
2017-12-25 19:59:35 +08:00
resp, code, err := request(
2017-12-08 17:14:12 +08:00
"/cloud/auth",
"POST",
data,
)
if err != nil {
logrus.Errorf("create auth token error, %v", err)
2017-12-25 19:59:35 +08:00
return nil, util.CreateAPIHandleError(code, err)
2017-12-08 17:14:12 +08:00
}
2017-12-25 19:59:35 +08:00
if code > 400 {
2017-12-08 17:14:12 +08:00
logrus.Errorf("create auth token error")
2017-12-25 19:59:35 +08:00
return nil, util.CreateAPIHandleError(code, fmt.Errorf("cretae auth token failed"))
2017-12-08 17:14:12 +08:00
}
return resp, nil
2017-12-08 17:14:12 +08:00
}
//PutToken PutToken
func (d *DefineCloudAuth) PutToken() error {
data, err := ffjson.Marshal(d.GT.Body)
if err != nil {
return err
}
2017-12-25 19:59:35 +08:00
_, code, err := request(
2017-12-08 17:14:12 +08:00
fmt.Sprintf("/cloud/auth/%s", d.GT.Body.EID),
"PUT",
data,
)
if err != nil {
2017-12-25 19:59:35 +08:00
logrus.Errorf("create auth token error, %v", err)
return util.CreateAPIHandleError(code, err)
2017-12-08 17:14:12 +08:00
}
2017-12-25 19:59:35 +08:00
if code > 400 {
logrus.Errorf("create auth token error")
return util.CreateAPIHandleError(code, fmt.Errorf("cretae auth token failed"))
2017-12-08 17:14:12 +08:00
}
return nil
}