Rainbond/pkg/api/handler/tokenIden.go

161 lines
4.4 KiB
Go

// RAINBOND, Application Management Platform
// Copyright (C) 2014-2017 Goodrain Co., Ltd.
// 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.
// 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.
// 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 handler
import (
"fmt"
"strings"
"time"
"github.com/goodrain/rainbond/cmd/api/option"
api_model "github.com/goodrain/rainbond/pkg/api/model"
"github.com/goodrain/rainbond/pkg/api/util"
"github.com/goodrain/rainbond/pkg/db"
dbmodel "github.com/goodrain/rainbond/pkg/db/model"
)
//TokenIdenAction TokenIdenAction
type TokenIdenAction struct{}
//CreateTokenIdenManager token identification
func CreateTokenIdenManager(conf option.Config) (*TokenIdenAction, error) {
return &TokenIdenAction{}, nil
}
//AddTokenIntoMap AddTokenIntoMap
func (t *TokenIdenAction) AddTokenIntoMap(rui *dbmodel.RegionUserInfo) {
m := GetDefaultTokenMap()
m[rui.Token] = rui
}
//DeleteTokenFromMap DeleteTokenFromMap
func (t *TokenIdenAction) DeleteTokenFromMap(oldtoken string, rui *dbmodel.RegionUserInfo) {
m := GetDefaultTokenMap()
t.AddTokenIntoMap(rui)
delete(m, oldtoken)
}
//GetAPIManager GetAPIManager
func (t *TokenIdenAction) GetAPIManager() map[string][]*dbmodel.RegionAPIClass {
return GetDefaultSourceURI()
}
//AddAPIManager AddAPIManager
func (t *TokenIdenAction) AddAPIManager(am *api_model.APIManager) *util.APIHandleError {
m := GetDefaultSourceURI()
ra := &dbmodel.RegionAPIClass{
ClassLevel: am.Body.ClassLevel,
Prefix: am.Body.Prefix,
}
if sourceList, ok := m[am.Body.ClassLevel]; ok {
sourceList = append(sourceList, ra)
} else {
//支持新增类型
newL := []*dbmodel.RegionAPIClass{ra}
m[am.Body.ClassLevel] = newL
}
ra.URI = am.Body.URI
ra.Alias = am.Body.Alias
ra.Remark = am.Body.Remark
if err := db.GetManager().RegionAPIClassDao().AddModel(ra); err != nil {
return util.CreateAPIHandleErrorFromDBError("add api manager", err)
}
return nil
}
//DeleteAPIManager DeleteAPIManager
func (t *TokenIdenAction) DeleteAPIManager(am *api_model.APIManager) *util.APIHandleError {
m := GetDefaultSourceURI()
if sourceList, ok := m[am.Body.ClassLevel]; ok {
var newL []*dbmodel.RegionAPIClass
for _, s := range sourceList {
if s.Prefix == am.Body.Prefix {
continue
}
newL = append(newL, s)
}
if len(newL) == 0 {
//该级别分组为空时,删除该资源分组
delete(m, am.Body.ClassLevel)
} else {
m[am.Body.ClassLevel] = newL
}
} else {
return util.CreateAPIHandleError(400, fmt.Errorf("have no api class level about %v", am.Body.ClassLevel))
}
if err := db.GetManager().RegionAPIClassDao().DeletePrefixInClass(am.Body.ClassLevel, am.Body.Prefix); err != nil {
return util.CreateAPIHandleErrorFromDBError("delete api prefix", err)
}
return nil
}
//CheckToken CheckToken
func (t *TokenIdenAction) CheckToken(token, uri string) bool {
m := GetDefaultTokenMap()
regionInfo, ok := m[token]
if !ok {
return false
}
if regionInfo.ValidityPeriod < int(time.Now().Unix()) {
return false
}
switch regionInfo.APIRange {
case dbmodel.ALLPOWER:
return true
case dbmodel.SERVERSOURCE:
sm := GetDefaultSourceURI()
smL, ok := sm[dbmodel.SERVERSOURCE]
if !ok {
return false
}
for _, urinfo := range smL {
if strings.HasPrefix(uri, urinfo.Prefix) {
return true
}
return false
}
case dbmodel.NODEMANAGER:
sm := GetDefaultSourceURI()
smL, ok := sm[dbmodel.NODEMANAGER]
if !ok {
return false
}
for _, urinfo := range smL {
if strings.HasPrefix(uri, urinfo.Prefix) {
return true
}
return false
}
}
return false
}
//InitTokenMap InitTokenMap
func (t *TokenIdenAction) InitTokenMap() error {
ruis, err := db.GetManager().RegionUserInfoDao().GetALLTokenInValidityPeriod()
if err != nil {
return err
}
m := GetDefaultTokenMap()
for _, rui := range ruis {
m[rui.Token] = rui
}
return nil
}