mirror of
https://github.com/openziti/ziti.git
synced 2026-09-11 13:29:03 +00:00
Paul Lorenz
312 lines
10 KiB
Go
312 lines
10 KiB
Go
/*
|
|
Copyright NetFoundry Inc.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
https://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package model
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/openziti/foundation/v2/errorz"
|
|
"github.com/openziti/sdk-golang/ziti"
|
|
"github.com/openziti/ziti/v2/controller/storage/boltz"
|
|
"github.com/openziti/ziti/v2/controller/db"
|
|
"github.com/openziti/ziti/v2/controller/models"
|
|
"go.etcd.io/bbolt"
|
|
)
|
|
|
|
type EnvInfo struct {
|
|
Arch string
|
|
Os string
|
|
OsRelease string
|
|
OsVersion string
|
|
Domain string
|
|
Hostname string
|
|
}
|
|
|
|
func (self *EnvInfo) Equals(other *EnvInfo) bool {
|
|
if self == nil {
|
|
return other == nil
|
|
}
|
|
if other == nil {
|
|
return false
|
|
}
|
|
return self.Arch == other.Arch &&
|
|
self.Os == other.Os &&
|
|
self.OsRelease == other.OsRelease &&
|
|
self.OsVersion == other.OsVersion &&
|
|
self.Domain == other.Domain &&
|
|
self.Hostname == other.Hostname
|
|
}
|
|
|
|
type SdkInfo struct {
|
|
AppId string
|
|
AppVersion string
|
|
Branch string
|
|
Revision string
|
|
Type string
|
|
Version string
|
|
}
|
|
|
|
func (self *SdkInfo) Equals(other *SdkInfo) bool {
|
|
if self == nil {
|
|
return other == nil
|
|
}
|
|
if other == nil {
|
|
return false
|
|
}
|
|
return self.AppId == other.AppId &&
|
|
self.AppVersion == other.AppVersion &&
|
|
self.Branch == other.Branch &&
|
|
self.Revision == other.Revision &&
|
|
self.Type == other.Type &&
|
|
self.Version == other.Version
|
|
}
|
|
|
|
type Identity struct {
|
|
models.BaseEntity
|
|
Name string
|
|
IdentityTypeId string
|
|
IsDefaultAdmin bool
|
|
IsAdmin bool
|
|
RoleAttributes []string
|
|
EnvInfo *EnvInfo
|
|
SdkInfo *SdkInfo
|
|
HasErConnection bool
|
|
EdgeRouterConnectionStatus IdentityOnlineState
|
|
DefaultHostingPrecedence ziti.Precedence
|
|
DefaultHostingCost uint16
|
|
ServiceHostingPrecedences map[string]ziti.Precedence
|
|
ServiceHostingCosts map[string]uint16
|
|
AppData map[string]interface{}
|
|
AuthPolicyId string
|
|
ExternalId *string
|
|
Disabled bool
|
|
DisabledAt *time.Time
|
|
DisabledUntil *time.Time
|
|
ServiceConfigs map[string]map[string]string
|
|
Interfaces []*Interface
|
|
Permissions []string
|
|
}
|
|
|
|
func (entity *Identity) toBoltEntityForCreate(_ *bbolt.Tx, env Env) (*db.Identity, error) {
|
|
identityType, err := env.GetManagers().IdentityType.ReadByIdOrName(entity.IdentityTypeId)
|
|
|
|
if err != nil && !boltz.IsErrNotFoundErr(err) {
|
|
return nil, err
|
|
}
|
|
|
|
if identityType == nil {
|
|
apiErr := errorz.NewNotFound()
|
|
apiErr.Cause = errorz.NewFieldError("typeId not found", "typeId", entity.IdentityTypeId)
|
|
apiErr.AppendCause = true
|
|
return nil, apiErr
|
|
}
|
|
|
|
if identityType.Name == db.RouterIdentityType {
|
|
fieldErr := errorz.NewFieldError("may not create identities with given typeId", "typeId", entity.IdentityTypeId)
|
|
return nil, errorz.NewFieldApiError(fieldErr)
|
|
}
|
|
|
|
entity.IdentityTypeId = identityType.Id
|
|
|
|
boltEntity := &db.Identity{
|
|
BaseExtEntity: *boltz.NewExtEntity(entity.Id, entity.Tags),
|
|
Name: entity.Name,
|
|
IdentityTypeId: entity.IdentityTypeId,
|
|
AuthPolicyId: entity.AuthPolicyId,
|
|
IsDefaultAdmin: entity.IsDefaultAdmin,
|
|
IsAdmin: entity.IsAdmin,
|
|
RoleAttributes: entity.RoleAttributes,
|
|
DefaultHostingPrecedence: entity.DefaultHostingPrecedence,
|
|
DefaultHostingCost: entity.DefaultHostingCost,
|
|
ServiceHostingPrecedences: entity.ServiceHostingPrecedences,
|
|
ServiceHostingCosts: entity.ServiceHostingCosts,
|
|
AppData: entity.AppData,
|
|
ExternalId: entity.ExternalId,
|
|
DisabledAt: entity.DisabledAt,
|
|
DisabledUntil: entity.DisabledUntil,
|
|
ServiceConfigs: entity.ServiceConfigs,
|
|
Interfaces: InterfacesToBolt(entity.Interfaces),
|
|
Permissions: entity.Permissions,
|
|
}
|
|
|
|
if entity.EnvInfo != nil {
|
|
boltEntity.EnvInfo = &db.EnvInfo{
|
|
Arch: entity.EnvInfo.Arch,
|
|
Os: entity.EnvInfo.Os,
|
|
OsRelease: entity.EnvInfo.OsRelease,
|
|
OsVersion: entity.EnvInfo.OsVersion,
|
|
Domain: entity.EnvInfo.Domain,
|
|
Hostname: entity.EnvInfo.Hostname,
|
|
}
|
|
}
|
|
|
|
if entity.SdkInfo != nil {
|
|
boltEntity.SdkInfo = &db.SdkInfo{
|
|
Branch: entity.SdkInfo.Branch,
|
|
Revision: entity.SdkInfo.Revision,
|
|
Type: entity.SdkInfo.Type,
|
|
Version: entity.SdkInfo.Version,
|
|
AppId: entity.SdkInfo.AppId,
|
|
AppVersion: entity.SdkInfo.AppVersion,
|
|
}
|
|
}
|
|
fillPersistenceInfo(boltEntity, entity.EnvInfo, entity.SdkInfo)
|
|
|
|
return boltEntity, nil
|
|
}
|
|
|
|
func fillModelInfo(identity *Identity, envInfo *db.EnvInfo, sdkInfo *db.SdkInfo) {
|
|
if envInfo != nil {
|
|
identity.EnvInfo = &EnvInfo{
|
|
Arch: envInfo.Arch,
|
|
Os: envInfo.Os,
|
|
OsRelease: envInfo.OsRelease,
|
|
OsVersion: envInfo.OsVersion,
|
|
Domain: envInfo.Domain,
|
|
Hostname: envInfo.Hostname,
|
|
}
|
|
}
|
|
|
|
if sdkInfo != nil {
|
|
identity.SdkInfo = &SdkInfo{
|
|
AppId: sdkInfo.AppId,
|
|
AppVersion: sdkInfo.AppVersion,
|
|
Branch: sdkInfo.Branch,
|
|
Revision: sdkInfo.Revision,
|
|
Type: sdkInfo.Type,
|
|
Version: sdkInfo.Version,
|
|
}
|
|
}
|
|
}
|
|
|
|
func fillPersistenceInfo(identity *db.Identity, envInfo *EnvInfo, sdkInfo *SdkInfo) {
|
|
if envInfo != nil {
|
|
identity.EnvInfo = &db.EnvInfo{
|
|
Arch: envInfo.Arch,
|
|
Os: envInfo.Os,
|
|
OsRelease: envInfo.OsRelease,
|
|
OsVersion: envInfo.OsVersion,
|
|
Domain: envInfo.Domain,
|
|
Hostname: envInfo.Hostname,
|
|
}
|
|
}
|
|
|
|
if sdkInfo != nil {
|
|
identity.SdkInfo = &db.SdkInfo{
|
|
Branch: sdkInfo.Branch,
|
|
Revision: sdkInfo.Revision,
|
|
Type: sdkInfo.Type,
|
|
Version: sdkInfo.Version,
|
|
AppId: sdkInfo.AppId,
|
|
AppVersion: sdkInfo.AppVersion,
|
|
}
|
|
}
|
|
}
|
|
|
|
func (entity *Identity) toBoltEntityForUpdate(tx *bbolt.Tx, env Env, checker boltz.FieldChecker) (*db.Identity, error) {
|
|
if checker == nil || checker.IsUpdated("type") {
|
|
identityType, err := env.GetManagers().IdentityType.ReadByIdOrName(entity.IdentityTypeId)
|
|
|
|
if err != nil && !boltz.IsErrNotFoundErr(err) {
|
|
return nil, err
|
|
}
|
|
|
|
if identityType == nil {
|
|
apiErr := errorz.NewNotFound()
|
|
apiErr.Cause = errorz.NewFieldError("identityTypeId not found", "identityTypeId", entity.IdentityTypeId)
|
|
apiErr.AppendCause = true
|
|
return nil, apiErr
|
|
}
|
|
|
|
entity.IdentityTypeId = identityType.Id
|
|
}
|
|
|
|
boltEntity := &db.Identity{
|
|
Name: entity.Name,
|
|
IdentityTypeId: entity.IdentityTypeId,
|
|
AuthPolicyId: entity.AuthPolicyId,
|
|
BaseExtEntity: *boltz.NewExtEntity(entity.Id, entity.Tags),
|
|
RoleAttributes: entity.RoleAttributes,
|
|
DefaultHostingPrecedence: entity.DefaultHostingPrecedence,
|
|
DefaultHostingCost: entity.DefaultHostingCost,
|
|
ServiceHostingPrecedences: entity.ServiceHostingPrecedences,
|
|
ServiceHostingCosts: entity.ServiceHostingCosts,
|
|
AppData: entity.AppData,
|
|
ExternalId: entity.ExternalId,
|
|
DisabledAt: entity.DisabledAt,
|
|
DisabledUntil: entity.DisabledUntil,
|
|
IsAdmin: entity.IsAdmin,
|
|
ServiceConfigs: entity.ServiceConfigs,
|
|
Interfaces: InterfacesToBolt(entity.Interfaces),
|
|
Permissions: entity.Permissions,
|
|
}
|
|
|
|
identityStore := env.GetManagers().Identity.GetStore()
|
|
_, currentType := identityStore.GetSymbol(db.FieldIdentityType).Eval(tx, []byte(entity.Id))
|
|
if string(currentType) == db.RouterIdentityType {
|
|
if (checker == nil || checker.IsUpdated("identityTypeId")) && entity.IdentityTypeId != db.RouterIdentityType {
|
|
fieldErr := errorz.NewFieldError("may not change type of router identities", "typeId", entity.IdentityTypeId)
|
|
return nil, errorz.NewFieldApiError(fieldErr)
|
|
}
|
|
|
|
_, currentName := identityStore.GetSymbol(db.FieldName).Eval(tx, []byte(entity.Id))
|
|
if (checker == nil || checker.IsUpdated(db.FieldName)) && string(currentName) != entity.Name {
|
|
fieldErr := errorz.NewFieldError("may not change name of router identities", "name", entity.Name)
|
|
return nil, errorz.NewFieldApiError(fieldErr)
|
|
}
|
|
} else if (checker == nil || checker.IsUpdated("identityTypeId")) && entity.IdentityTypeId == db.RouterIdentityType {
|
|
fieldErr := errorz.NewFieldError("may not change type to router", "typeId", entity.IdentityTypeId)
|
|
return nil, errorz.NewFieldApiError(fieldErr)
|
|
}
|
|
|
|
fillPersistenceInfo(boltEntity, entity.EnvInfo, entity.SdkInfo)
|
|
|
|
return boltEntity, nil
|
|
}
|
|
|
|
func (entity *Identity) fillFrom(env Env, _ *bbolt.Tx, boltIdentity *db.Identity) error {
|
|
entity.FillCommon(boltIdentity)
|
|
entity.Name = boltIdentity.Name
|
|
entity.IdentityTypeId = boltIdentity.IdentityTypeId
|
|
entity.AuthPolicyId = boltIdentity.AuthPolicyId
|
|
entity.IsDefaultAdmin = boltIdentity.IsDefaultAdmin
|
|
entity.IsAdmin = boltIdentity.IsAdmin
|
|
entity.RoleAttributes = boltIdentity.RoleAttributes
|
|
entity.HasErConnection = env.GetManagers().Identity.HasErConnection(entity.Id)
|
|
entity.EdgeRouterConnectionStatus = env.GetManagers().Identity.connections.GetIdentityOnlineState(boltIdentity.Id)
|
|
entity.DefaultHostingPrecedence = boltIdentity.DefaultHostingPrecedence
|
|
entity.DefaultHostingCost = boltIdentity.DefaultHostingCost
|
|
entity.ServiceHostingPrecedences = boltIdentity.ServiceHostingPrecedences
|
|
entity.ServiceHostingCosts = boltIdentity.ServiceHostingCosts
|
|
entity.AppData = boltIdentity.AppData
|
|
entity.ExternalId = boltIdentity.ExternalId
|
|
entity.DisabledUntil = boltIdentity.DisabledUntil
|
|
entity.DisabledAt = boltIdentity.DisabledAt
|
|
entity.Disabled = boltIdentity.Disabled
|
|
entity.ServiceConfigs = boltIdentity.ServiceConfigs
|
|
entity.Interfaces = InterfacesFromBolt(boltIdentity.Interfaces)
|
|
fillModelInfo(entity, boltIdentity.EnvInfo, boltIdentity.SdkInfo)
|
|
entity.Permissions = boltIdentity.Permissions
|
|
|
|
return nil
|
|
}
|
|
|
|
type ServiceConfig struct {
|
|
Service string
|
|
Config string
|
|
}
|