root.go

package registry

import (
	"context"
	"fmt"
	"github.com/kubesphere-sigs/ks/kubectl-plugin/types"
	"github.com/spf13/cobra"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
	"k8s.io/client-go/dynamic"
	"sigs.k8s.io/yaml"
)

// NewRegistryCmd returns a command of pipeline
func NewRegistryCmd(client dynamic.Interface) (cmd *cobra.Command) {
	ctx := context.TODO()

	cmd = &cobra.Command{
		Use:     "registry",
		Aliases: []string{"reg"},
		Short:   "start a registry locally",
		Example: `Before you get started, please sudo vim /etc/docker/daemon.json, then add the following config:
	"insecure-registries": [
		"139.198.3.176:32678"
	],
After that, please restart docker daemon via: systemctl restart docker
`,
		RunE: func(cmd *cobra.Command, args []string) (err error) {
			_ = client.Resource(types.GetDeploySchema()).Namespace("default").Delete(ctx, "registry", metav1.DeleteOptions{})
			_ = client.Resource(types.GetServiceSchema()).Namespace("default").Delete(ctx, "registry", metav1.DeleteOptions{})

			obj := &unstructured.Unstructured{}
			content := getRegistryDeploy()
			if err = yaml.Unmarshal([]byte(content), obj); err == nil {
				if _, err = client.Resource(types.GetDeploySchema()).Namespace("default").Create(ctx, obj, metav1.CreateOptions{}); err != nil {
					err = fmt.Errorf("failed when create deploy, %#v", err)
					return
				}
				cmd.Println("registry deploy installed")
			}

			obj = &unstructured.Unstructured{}
			svcContent := getService()
			if err = yaml.Unmarshal([]byte(svcContent), obj); err == nil {
				if _, err = client.Resource(types.GetServiceSchema()).Namespace("default").Create(ctx, obj, metav1.CreateOptions{}); err != nil {
					err = fmt.Errorf("failed when create service, %#v", err)
				} else {
					cmd.Println("registry service installed")
				}
			}
			return
		},
	}
	return
}

func getService() string {
	return `
apiVersion: v1
kind: Service
metadata:
  name: registry
spec:
  ports:
  - name: registry
    nodePort: 32678
    port: 5000
    protocol: TCP
    targetPort: 5000
  selector:
    app: registry
  type: NodePort
`
}

func getRegistryDeploy() string {
	return `
apiVersion: apps/v1
kind: Deployment
metadata:
  name: registry
spec:
  replicas: 1
  selector:
    matchLabels:
      app: registry
  template:
    metadata:
      labels:
        app: registry
    spec:
      containers:
      - image: registry:2
        imagePullPolicy: IfNotPresent
        name: registry
        ports:
        - containerPort: 5000
          protocol: TCP
`
}