Fetch the repository succeeded.
package catalog
import (
"bytes"
"encoding/json"
"net/http"
"time"
"github.com/coreos/etcd/etcdserver/api/v3rpc/rpctypes"
"github.com/ghodss/yaml"
"github.com/rancher/norman/httperror"
"github.com/rancher/norman/types"
"github.com/rancher/norman/types/convert"
"github.com/rancher/rancher/pkg/rbac"
"github.com/rancher/rancher/pkg/ref"
v3 "github.com/rancher/types/apis/management.cattle.io/v3"
client "github.com/rancher/types/client/management/v3"
"github.com/rancher/types/compose"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)
func Formatter(apiContext *types.APIContext, resource *types.RawResource) {
if canUpdateCatalog(apiContext, resource) {
resource.AddAction(apiContext, "refresh")
}
resource.Links["exportYaml"] = apiContext.URLBuilder.Link("exportYaml", resource)
}
func CollectionFormatter(apiContext *types.APIContext, collection *types.GenericCollection) {
if canUpdateCatalog(apiContext, nil) {
collection.AddAction(apiContext, "refresh")
}
}
type ActionHandler struct {
CatalogClient v3.CatalogInterface
ProjectCatalogClient v3.ProjectCatalogInterface
ClusterCatalogClient v3.ClusterCatalogInterface
}
func (a ActionHandler) RefreshActionHandler(actionName string, action *types.Action, apiContext *types.APIContext) error {
if actionName != "refresh" {
return httperror.NewAPIError(httperror.NotFound, "not found")
}
if !canUpdateCatalog(apiContext, nil) {
return httperror.NewAPIError(httperror.NotFound, "not found")
}
catalogs := []v3.Catalog{}
if apiContext.ID != "" {
catalog, err := a.CatalogClient.Get(apiContext.ID, metav1.GetOptions{})
if err != nil {
return err
}
catalogs = append(catalogs, *catalog)
} else {
catalogList, err := a.CatalogClient.List(metav1.ListOptions{})
if err != nil {
return err
}
for _, catalog := range catalogList.Items {
catalogs = append(catalogs, catalog)
}
}
for _, catalog := range catalogs {
catalog.Status.LastRefreshTimestamp = time.Now().Format(time.RFC3339)
v3.CatalogConditionRefreshed.Unknown(&catalog)
if _, err := a.CatalogClient.Update(&catalog); err != nil {
return err
}
}
data := map[string]interface{}{
"catalogs": catalogs,
}
apiContext.WriteResponse(http.StatusOK, data)
return nil
}
func (a ActionHandler) ExportYamlHandler(apiContext *types.APIContext, next types.RequestHandler) error {
switch apiContext.Link {
case "exportyaml":
catalog, err := a.CatalogClient.Get(apiContext.ID, metav1.GetOptions{})
if err != nil {
return rpctypes.ErrGRPCStopped
}
topkey := compose.Config{}
topkey.Version = "v3"
ca := client.Catalog{}
if err := convert.ToObj(catalog.Spec, &ca); err != nil {
return err
}
topkey.Catalogs = map[string]client.Catalog{}
topkey.Catalogs[catalog.Name] = ca
m, err := convert.EncodeToMap(topkey)
if err != nil {
return err
}
delete(m["catalogs"].(map[string]interface{})[catalog.Name].(map[string]interface{}), "actions")
delete(m["catalogs"].(map[string]interface{})[catalog.Name].(map[string]interface{}), "links")
delete(m["catalogs"].(map[string]interface{})[catalog.Name].(map[string]interface{}), "password")
data, err := json.Marshal(m)
if err != nil {
return err
}
buf, err := yaml.JSONToYAML(data)
if err != nil {
return err
}
reader := bytes.NewReader(buf)
apiContext.Response.Header().Set("Content-Type", "text/yaml")
http.ServeContent(apiContext.Response, apiContext.Request, "exportYaml", time.Now(), reader)
return nil
}
return nil
}
func (a ActionHandler) RefreshProjectCatalogActionHandler(actionName string, action *types.Action, apiContext *types.APIContext) error {
if actionName != "refresh" {
return httperror.NewAPIError(httperror.NotFound, "not found")
}
if !canUpdateCatalog(apiContext, nil) {
return httperror.NewAPIError(httperror.NotFound, "not found")
}
prjCatalogs := []v3.ProjectCatalog{}
if apiContext.ID != "" {
ns, name := ref.Parse(apiContext.ID)
catalog, err := a.ProjectCatalogClient.GetNamespaced(ns, name, metav1.GetOptions{})
if err != nil {
return err
}
prjCatalogs = append(prjCatalogs, *catalog)
} else {
catalogList, err := a.ProjectCatalogClient.List(metav1.ListOptions{})
if err != nil {
return err
}
for _, catalog := range catalogList.Items {
prjCatalogs = append(prjCatalogs, catalog)
}
}
for _, catalog := range prjCatalogs {
catalog.Status.LastRefreshTimestamp = time.Now().Format(time.RFC3339)
v3.CatalogConditionRefreshed.Unknown(&catalog)
if _, err := a.ProjectCatalogClient.Update(&catalog); err != nil {
return err
}
}
data := map[string]interface{}{
"catalogs": prjCatalogs,
}
apiContext.WriteResponse(http.StatusOK, data)
return nil
}
func (a ActionHandler) RefreshClusterCatalogActionHandler(actionName string, action *types.Action, apiContext *types.APIContext) error {
if actionName != "refresh" {
return httperror.NewAPIError(httperror.NotFound, "not found")
}
if !canUpdateCatalog(apiContext, nil) {
return httperror.NewAPIError(httperror.NotFound, "not found")
}
clCatalogs := []v3.ClusterCatalog{}
if apiContext.ID != "" {
ns, name := ref.Parse(apiContext.ID)
catalog, err := a.ClusterCatalogClient.GetNamespaced(ns, name, metav1.GetOptions{})
if err != nil {
return err
}
clCatalogs = append(clCatalogs, *catalog)
} else {
catalogList, err := a.ClusterCatalogClient.List(metav1.ListOptions{})
if err != nil {
return err
}
for _, catalog := range catalogList.Items {
clCatalogs = append(clCatalogs, catalog)
}
}
for _, catalog := range clCatalogs {
catalog.Status.LastRefreshTimestamp = time.Now().Format(time.RFC3339)
v3.CatalogConditionRefreshed.Unknown(&catalog)
if _, err := a.ClusterCatalogClient.Update(&catalog); err != nil {
return err
}
}
data := map[string]interface{}{
"catalogs": clCatalogs,
}
apiContext.WriteResponse(http.StatusOK, data)
return nil
}
func canUpdateCatalog(apiContext *types.APIContext, resource *types.RawResource) bool {
var groupName, resourceName string
switch apiContext.Type {
case client.CatalogType:
groupName, resourceName = v3.CatalogGroupVersionKind.Group, v3.CatalogResource.Name
case client.ClusterCatalogType:
groupName, resourceName = v3.ClusterCatalogGroupVersionKind.Group, v3.ClusterCatalogResource.Name
case client.ProjectCatalogType:
groupName, resourceName = v3.ProjectCatalogGroupVersionKind.Group, v3.ProjectCatalogResource.Name
default:
return false
}
obj := rbac.ObjFromContext(apiContext, resource)
return apiContext.AccessControl.CanDo(groupName, resourceName, "update", apiContext, obj, apiContext.Schema) == nil
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。