// RAINBOND, Application Management Platform
// Copyright (C) 2022-2022 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 controller

import (
	"fmt"
	"github.com/goodrain/rainbond/api/handler"
	api_model "github.com/goodrain/rainbond/api/model"
	ctxutil "github.com/goodrain/rainbond/api/util/ctx"
	httputil "github.com/goodrain/rainbond/util/http"
	"net/http"
)

// K8sAttributeController -
type K8sAttributeController struct{}

// K8sAttributes -
func (k *K8sAttributeController) K8sAttributes(w http.ResponseWriter, r *http.Request) {
	switch r.Method {
	case "POST":
		k.createK8sAttributes(w, r)
	case "PUT":
		k.updateK8sAttributes(w, r)
	case "DELETE":
		k.deleteK8sAttributes(w, r)
	}
}

func (k *K8sAttributeController) createK8sAttributes(w http.ResponseWriter, r *http.Request) {
	tenantID := r.Context().Value(ctxutil.ContextKey("tenant_id")).(string)
	componentID := r.Context().Value(ctxutil.ContextKey("service_id")).(string)
	var k8sAttr api_model.ComponentK8sAttribute
	if ok := httputil.ValidatorRequestStructAndErrorResponse(r, w, &k8sAttr, nil); !ok {
		httputil.ReturnBcodeError(r, w, fmt.Errorf("k8s attributes is not valid"))
		return
	}
	if err := handler.GetServiceManager().CreateK8sAttribute(tenantID, componentID, &k8sAttr); err != nil {
		httputil.ReturnBcodeError(r, w, err)
		return
	}
	w.WriteHeader(http.StatusOK)
}

func (k *K8sAttributeController) updateK8sAttributes(w http.ResponseWriter, r *http.Request) {
	componentID := r.Context().Value(ctxutil.ContextKey("service_id")).(string)
	var k8sAttr api_model.ComponentK8sAttribute
	if ok := httputil.ValidatorRequestStructAndErrorResponse(r, w, &k8sAttr, nil); !ok {
		httputil.ReturnBcodeError(r, w, fmt.Errorf("k8s attributes is not valid"))
		return
	}
	if err := handler.GetServiceManager().UpdateK8sAttribute(componentID, &k8sAttr); err != nil {
		httputil.ReturnBcodeError(r, w, err)
		return
	}
	w.WriteHeader(http.StatusOK)
}

func (k *K8sAttributeController) deleteK8sAttributes(w http.ResponseWriter, r *http.Request) {
	componentID := r.Context().Value(ctxutil.ContextKey("service_id")).(string)
	var req api_model.DeleteK8sAttributeReq
	if ok := httputil.ValidatorRequestStructAndErrorResponse(r, w, &req, nil); !ok {
		httputil.ReturnBcodeError(r, w, fmt.Errorf("k8s attributes is not valid"))
		return
	}
	if err := handler.GetServiceManager().DeleteK8sAttribute(componentID, req.Name); err != nil {
		httputil.ReturnBcodeError(r, w, err)
		return
	}
	w.WriteHeader(http.StatusOK)
}