Cloudpods/backend/pkg/mcclient/modules/k8s/base.go

// Copyright 2019 Yunion
//
// 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
//
//     http://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 k8s

import (
	"yunion.io/x/jsonutils"
	"yunion.io/x/pkg/util/sets"

	"yunion.io/x/onecloud/pkg/mcclient"
	"yunion.io/x/onecloud/pkg/mcclient/modulebase"
)

type ResourceManager struct {
	*modulebase.ResourceManager
}

func NewResourceManager(keyword, keywordPlural string, columns, adminColumns *Columns) *ResourceManager {
	man := &modulebase.ResourceManager{
		BaseManager:   *modulebase.NewBaseManager("k8s", "", "", columns.Array(), adminColumns.Array()),
		Keyword:       keyword,
		KeywordPlural: keywordPlural,
	}
	return &ResourceManager{man}
}

func NewJointManager(keyword, keywordPlural string, columns, adminColumns *Columns, master, slave modulebase.Manager) modulebase.JointResourceManager {
	return modulebase.JointResourceManager{
		ResourceManager: *NewResourceManager(keyword, keywordPlural, columns, adminColumns).ResourceManager,
		Master:          master,
		Slave:           slave,
	}
}

func (m ResourceManager) GetBaseManager() modulebase.ResourceManager {
	return *m.ResourceManager
}

type IClusterResourceManager interface {
	modulebase.Manager
	GetRaw(s *mcclient.ClientSession, id string, params *jsonutils.JSONDict) (jsonutils.JSONObject, error)
	UpdateRaw(s *mcclient.ClientSession, id string, query, body *jsonutils.JSONDict) (jsonutils.JSONObject, error)
}

type clusterResourceManager struct {
	*ResourceManager
}

func newClusterResourceManager(keyword, keywordPlural string, columns, adminColumns *Columns) *clusterResourceManager {
	newAdminCols := NewClusterCols(adminColumns.Array()...)
	man := NewResourceManager(keyword, keywordPlural, columns, newAdminCols)
	return &clusterResourceManager{man}
}

func (man clusterResourceManager) Get_Cluster(obj jsonutils.JSONObject) interface{} {
	cluster, _ := obj.GetString("cluster")
	return cluster
}

func (man clusterResourceManager) GetRaw(s *mcclient.ClientSession, id string, params *jsonutils.JSONDict) (jsonutils.JSONObject, error) {
	return man.GetSpecific(s, id, "rawdata", params)
}

func (man clusterResourceManager) UpdateRaw(s *mcclient.ClientSession, id string, query, rawdata *jsonutils.JSONDict) (jsonutils.JSONObject, error) {
	return man.PutSpecific(s, id, "rawdata", query, rawdata)
}

type ClusterResourceManager struct {
	*clusterResourceManager
	nameGetter
	ageGetter
	labelGetter
}

func NewClusterResourceManager(kw, kwp string, columns, adminColumns *Columns) *ClusterResourceManager {
	newCols := NewMetaCols(columns.Array()...)
	man := newClusterResourceManager(kw, kwp, newCols, adminColumns)
	return &ClusterResourceManager{man, getName, getAge, getLabel}
}

type NamespaceResourceManager struct {
	*ClusterResourceManager
	namespaceGetter
}

func NewNamespaceResourceManager(kw, kwp string, columns, adminColumns *Columns) *NamespaceResourceManager {
	newCols := NewNamespaceCols(columns.Array()...)
	man := NewClusterResourceManager(kw, kwp, newCols, adminColumns)
	return &NamespaceResourceManager{man, getNamespace}
}

type Columns struct {
	cols []string
}

func NewColumns(cols ...string) *Columns {
	c := &Columns{cols: make([]string, 0)}
	return c.Add(cols...)
}

func (c *Columns) Add(cols ...string) *Columns {
	for _, col := range cols {
		c.add(col)
	}
	return c
}

func (c *Columns) add(col string) *Columns {
	src := sets.NewString(c.Array()...)
	if src.Has(col) {
		return c
	}
	c.cols = append(c.cols, col)
	return c
}

func (c Columns) Array() []string {
	return c.cols
}

func NewNameCols(col ...string) *Columns {
	return NewColumns("Name", "Id", "Status").Add(col...)
}

func NewMetaCols(col ...string) *Columns {
	return NewNameCols("creationTimestamp", "Created_At").Add(col...)
}

func NewNamespaceCols(col ...string) *Columns {
	return NewMetaCols("Namespace_ID", "Namespace", "Labels").Add(col...)
}

func NewClusterCols(col ...string) *Columns {
	return NewColumns("Cluster_ID", "Cluster").Add(col...)
}

func NewResourceCols(col ...string) *Columns {
	return NewNameCols().Add(col...)
}

func NewFedJointClusterCols(col ...string) *Columns {
	return NewColumns(
		"Federatedresource_ID", "Federatedresource",
		"Cluster_ID", "Cluster",
		"Namespace_ID", "Namespace",
		"Resource_ID", "Resource",
	)
}

type ListPrinter interface {
	GetColumns(*mcclient.ClientSession) []string
}