diff --git a/pkg/apis/v1alpha/install.go b/pkg/apis/v1alpha/install.go index e05e4e2d0..d3d57ac7b 100644 --- a/pkg/apis/v1alpha/install.go +++ b/pkg/apis/v1alpha/install.go @@ -17,18 +17,23 @@ limitations under the License. package v1alpha import ( - "github.com/emicklei/go-restful" - "kubesphere.io/kubesphere/pkg/apis/v1alpha/nodes" + "github.com/emicklei/go-restful" + "kubesphere.io/kubesphere/pkg/apis/v1alpha/nodes" + "kubesphere.io/kubesphere/pkg/apis/v1alpha/kubeconfig" + "kubesphere.io/kubesphere/pkg/apis/v1alpha/kubectl" ) func init() { - ws := new(restful.WebService) - ws.Path("/api/v1alpha") + ws := new(restful.WebService) + ws.Path("/api/v1alpha") - nodes.Register(ws,"/nodes") + nodes.Register(ws,"/nodes") + kubeconfig.Register(ws, "/namespaces/{namespace}/kubeconfig") + kubectl.Register(ws, "/namespaces/{namespace}/kubectl") - // add webservice to default container - restful.Add(ws) + // add webservice to default container + restful.Add(ws) } + diff --git a/pkg/apis/v1alpha/kubectl/kubectl.go b/pkg/apis/v1alpha/kubectl/kubectl.go new file mode 100644 index 000000000..5cbde4d61 --- /dev/null +++ b/pkg/apis/v1alpha/kubectl/kubectl.go @@ -0,0 +1,52 @@ +/* +Copyright 2018 The KubeSphere Authors. + +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 kubectl + +import ( + "github.com/emicklei/go-restful" + "kubesphere.io/kubesphere/pkg/models" + "net/http" +) + +func Register(ws *restful.WebService, subPath string) { + + ws.Route(ws.GET(subPath).Consumes("*/*").Produces(restful.MIME_JSON).To(handleKubectl).Doc("use to " + + "get a kubectl pod in specified namespaces").Param(ws.PathParameter("namespace", + "namespace").DataType("string")).Do(returns200,returns500)) + +} + +func handleKubectl(req *restful.Request, resp *restful.Response) { + + ns := req.PathParameter("namespace") + + kubectlPod, err := models.GetKubectlPod(ns) + + if err != nil { + resp.WriteError(http.StatusInternalServerError, err) + } + + resp.WriteEntity(kubectlPod) +} + +func returns200(b *restful.RouteBuilder) { + b.Returns(http.StatusOK, "OK", nil) +} + +func returns500(b *restful.RouteBuilder) { + b.Returns(http.StatusInternalServerError, "fail", nil) +} diff --git a/pkg/models/kubectl.go b/pkg/models/kubectl.go new file mode 100644 index 000000000..b44d8d235 --- /dev/null +++ b/pkg/models/kubectl.go @@ -0,0 +1,86 @@ +/* +Copyright 2018 The KubeSphere Authors. + +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 models + +import ( + "fmt" + "math/rand" + meta_v1 "k8s.io/apimachinery/pkg/apis/meta/v1" + "k8s.io/apimachinery/pkg/labels" + "kubesphere.io/kubesphere/pkg/client" + "github.com/golang/glog" + "k8s.io/api/core/v1" +) + + +const ( + deploymentName = "kubectl" +) + +type kubectlPodInfo struct { + Namespace string `json: "namespace"` + Pod string `json: "podname"` + Container string `json: "container"` +} + +func GetKubectlPod(namespace string) (kubectlPodInfo, error) { + k8sClient := client.NewK8sClient() + deploy, err := k8sClient.AppsV1beta2().Deployments(namespace).Get(deploymentName, meta_v1.GetOptions{}) + if err != nil { + glog.Errorln(err) + return kubectlPodInfo{}, err + } + + selectors := deploy.Spec.Selector.MatchLabels + labelSelector := labels.Set(selectors).AsSelector().String() + podList, err := k8sClient.CoreV1().Pods(namespace).List(meta_v1.ListOptions{LabelSelector:labelSelector}) + if err != nil { + glog.Errorln(err) + return kubectlPodInfo{}, err + } + + pod, err := selectCorrectPod(namespace, podList.Items) + if err != nil{ + glog.Errorln(err) + return kubectlPodInfo{}, err + } + + info := kubectlPodInfo{Namespace:pod.Namespace, Pod:pod.Name, Container:pod.Status.ContainerStatuses[0].Name} + + return info, nil + +} + + +func selectCorrectPod(namespace string, pods []v1.Pod) (kubectlPod v1.Pod, err error) { + + var kubectPodList []v1.Pod + for _, pod := range pods{ + for _, condition := range pod.Status.Conditions{ + if condition.Type == "Ready" && condition.Status == "True"{ + kubectPodList = append(kubectPodList, pod) + } + } + } + if len(kubectPodList) < 1{ + err = fmt.Errorf("cannot find valid kubectl pod in namespace:%s", namespace) + return v1.Pod{}, err + } + + random := rand.Intn(len(kubectPodList)) + return kubectPodList[random], nil +}