/*
Copyright 2021 The KubeVela 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 plugins

import (
	"context"
	"encoding/json"
	"os"

	"cuelang.org/go/cue"
	"github.com/google/go-cmp/cmp"
	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
	corev1 "k8s.io/api/core/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/apimachinery/pkg/labels"
	"k8s.io/apimachinery/pkg/selection"
	"sigs.k8s.io/yaml"

	corev1beta1 "github.com/oam-dev/kubevela/apis/core.oam.dev/v1beta1"
	"github.com/oam-dev/kubevela/apis/types"
	"github.com/oam-dev/kubevela/pkg/oam/util"
	"github.com/oam-dev/kubevela/pkg/utils/common"
)

const (
	TestDir        = "testdata"
	DeployName     = "deployments.testapps"
	WebserviceName = "webservice.testapps"
)

var _ = Describe("DefinitionFiles", func() {

	deployment := types.Capability{
		Namespace:   "testdef",
		Name:        DeployName,
		Type:        types.TypeComponentDefinition,
		CrdName:     "deployments.apps",
		Description: "description not defined",
		Category:    types.CUECategory,
		Parameters: []types.Parameter{
			{
				Type: cue.ListKind,
				Name: "env",
			},
			{
				Name:     "image",
				Type:     cue.StringKind,
				Default:  "",
				Short:    "i",
				Required: true,
				Usage:    "Which image would you like to use for your service",
			},
			{
				Name:    "port",
				Type:    cue.IntKind,
				Short:   "p",
				Default: int64(8080),
				Usage:   "Which port do you want customer traffic sent to",
			},
		},
		CrdInfo: &types.CRDInfo{
			APIVersion: "apps/v1",
			Kind:       "Deployment",
		},
		Labels: map[string]string{"usecase": "forplugintest"},
	}

	websvc := types.Capability{
		Namespace:   "testdef",
		Name:        WebserviceName,
		Type:        types.TypeComponentDefinition,
		Description: "description not defined",
		Category:    types.CUECategory,
		Parameters: []types.Parameter{{
			Name: "env", Type: cue.ListKind,
		}, {
			Name:     "image",
			Type:     cue.StringKind,
			Default:  "",
			Short:    "i",
			Required: true,
			Usage:    "Which image would you like to use for your service",
		}, {
			Name:    "port",
			Type:    cue.IntKind,
			Short:   "p",
			Default: int64(6379),
			Usage:   "Which port do you want customer traffic sent to",
		}},
		CrdName: "deployments.apps",
		CrdInfo: &types.CRDInfo{
			APIVersion: "apps/v1",
			Kind:       "Deployment",
		},
		Labels: map[string]string{"usecase": "forplugintest"},
	}

	req, _ := labels.NewRequirement("usecase", selection.Equals, []string{"forplugintest"})
	selector := labels.NewSelector().Add(*req)
	// Notice!!  DefinitionPath Object is Cluster Scope object
	// which means objects created in other DefinitionNamespace will also affect here.
	It("getcomponents", func() {
		arg := common.Args{}
		arg.SetClient(k8sClient)
		workloadDefs, _, err := GetComponentsFromCluster(context.Background(), DefinitionNamespace, arg, selector)
		Expect(err).Should(BeNil())
		for i := range workloadDefs {
			// CueTemplate should always be fulfilled, even those whose CueTemplateURI is assigend,
			By("check CueTemplate is fulfilled")
			Expect(workloadDefs[i].CueTemplate).ShouldNot(BeEmpty())
			workloadDefs[i].CueTemplate = ""
		}
		Expect(cmp.Diff(workloadDefs, []types.Capability{deployment, websvc})).Should(BeEquivalentTo(""))
	})
	It("getall", func() {
		arg := common.Args{}
		arg.SetClient(k8sClient)
		alldef, err := GetCapabilitiesFromCluster(context.Background(), DefinitionNamespace, arg, selector)
		Expect(err).Should(BeNil())
		for i := range alldef {
			alldef[i].CueTemplate = ""
		}
		Expect(cmp.Diff(alldef, []types.Capability{deployment, websvc})).Should(BeEquivalentTo(""))
	})
})

var _ = Describe("test GetCapabilityByName", func() {
	var (
		ctx        context.Context
		c          common.Args
		ns         string
		defaultNS  string
		cd1        corev1beta1.ComponentDefinition
		cd2        corev1beta1.ComponentDefinition
		cd3        corev1beta1.ComponentDefinition
		cd4        corev1beta1.ComponentDefinition
		td1        corev1beta1.TraitDefinition
		td2        corev1beta1.TraitDefinition
		td3        corev1beta1.TraitDefinition
		component1 string
		component2 string
		component3 string
		component4 string
		trait1     string
		trait2     string
		trait3     string
	)
	BeforeEach(func() {
		c = common.Args{}
		c.SetClient(k8sClient)
		ctx = context.Background()
		ns = "cluster-test-ns-suffix"
		defaultNS = types.DefaultKubeVelaNS
		component1 = "cd1"
		component2 = "cd2"
		component3 = "cd3"
		component4 = "cd4"

		trait1 = "td1"
		trait2 = "td2"
		trait3 = "td3"

		By("create namespace")
		Expect(k8sClient.Create(ctx, &corev1.Namespace{ObjectMeta: metav1.ObjectMeta{Name: ns}})).Should(SatisfyAny(BeNil(), &util.AlreadyExistMatcher{}))
		Expect(k8sClient.Create(ctx, &corev1.Namespace{ObjectMeta: metav1.ObjectMeta{Name: defaultNS}})).Should(SatisfyAny(BeNil(), &util.AlreadyExistMatcher{}))

		By("create ComponentDefinition")
		data, _ := os.ReadFile("testdata/componentDef.yaml")
		yaml.Unmarshal(data, &cd1)
		yaml.Unmarshal(data, &cd2)
		data2, _ := os.ReadFile("testdata/kube-worker.yaml")
		yaml.Unmarshal(data2, &cd3)

		helmYaml, _ := os.ReadFile("testdata/helm.yaml")
		yaml.Unmarshal(helmYaml, &cd4)

		cd1.Namespace = ns
		cd1.Name = component1
		Expect(k8sClient.Create(ctx, &cd1)).Should(SatisfyAny(BeNil(), &util.AlreadyExistMatcher{}))

		cd2.Namespace = defaultNS
		cd2.Name = component2
		Expect(k8sClient.Create(ctx, &cd2)).Should(SatisfyAny(BeNil(), &util.AlreadyExistMatcher{}))

		cd3.Namespace = ns
		cd3.Name = component3
		Expect(k8sClient.Create(ctx, &cd3)).Should(SatisfyAny(BeNil(), &util.AlreadyExistMatcher{}))

		cd4.Namespace = ns
		cd4.Name = component4
		Expect(k8sClient.Create(ctx, &cd4)).Should(SatisfyAny(BeNil(), &util.AlreadyExistMatcher{}))

		By("create TraitDefinition")
		data, _ = os.ReadFile("testdata/manualscalars.yaml")
		yaml.Unmarshal(data, &td1)
		yaml.Unmarshal(data, &td2)
		data3, _ := os.ReadFile("testdata/svcTraitDef.yaml")
		yaml.Unmarshal(data3, &td3)
		td1.Namespace = ns
		td1.Name = trait1
		Expect(k8sClient.Create(ctx, &td1)).Should(SatisfyAny(BeNil(), &util.AlreadyExistMatcher{}))

		td2.Namespace = defaultNS
		td2.Name = trait2
		Expect(k8sClient.Create(ctx, &td2)).Should(SatisfyAny(BeNil(), &util.AlreadyExistMatcher{}))

		td3.Namespace = ns
		td3.Name = trait3
		Expect(k8sClient.Create(ctx, &td3)).Should(SatisfyAny(BeNil(), &util.AlreadyExistMatcher{}))

	})

	It("get capability", func() {
		Context("ComponentDefinition is in the current namespace", func() {
			_, err := GetCapabilityByName(ctx, c, component1, ns, nil)
			Expect(err).Should(BeNil())
		})
		Context("ComponentDefinition is in the default namespace", func() {
			_, err := GetCapabilityByName(ctx, c, component2, ns, nil)
			Expect(err).Should(BeNil())
		})
		Context("ComponentDefinition is in the default namespace", func() {
			cap, err := GetCapabilityByName(ctx, c, component3, ns, nil)
			Expect(err).Should(BeNil())
			jsontmp, err := json.Marshal(cap.KubeParameter)
			Expect(err).Should(BeNil())
			Expect(string(jsontmp)).Should(ContainSubstring("image"))
			Expect(string(jsontmp)).Should(ContainSubstring("spec.template.spec.containers[0].image"))
			Expect(string(jsontmp)).Should(ContainSubstring("port"))
			Expect(string(jsontmp)).Should(ContainSubstring("the specific container port num which can accept external request."))
		})
		Context("ComponentDefinition's workload type is AutoDetectWorkloadDefinition", func() {
			_, err := GetCapabilityByName(ctx, c, component4, ns, nil)
			Expect(err).Should(BeNil())
		})

		Context("TraitDefinition is in the current namespace", func() {
			_, err := GetCapabilityByName(ctx, c, trait1, ns, nil)
			Expect(err).Should(BeNil())
		})
		Context("TraitDefinition is in the default namespace", func() {
			_, err := GetCapabilityByName(ctx, c, trait2, ns, nil)
			Expect(err).Should(BeNil())
		})
		Context("TraitDefinition is in the default namespace", func() {
			cap, err := GetCapabilityByName(ctx, c, trait3, ns, nil)
			Expect(err).Should(BeNil())
			jsontmp, err := json.Marshal(cap.KubeParameter)
			Expect(err).Should(BeNil())
			Expect(string(jsontmp)).Should(ContainSubstring("targetPort"))
			Expect(string(jsontmp)).Should(ContainSubstring("target port num for service provider."))
		})

		Context("capability cloud not be found", func() {
			_, err := GetCapabilityByName(ctx, c, "a-component-definition-not-existed", ns, nil)
			Expect(err).Should(HaveOccurred())
		})
	})
})

var _ = Describe("test GetNamespacedCapabilitiesFromCluster", func() {
	var (
		ctx        context.Context
		c          common.Args
		ns         string
		defaultNS  string
		cd1        corev1beta1.ComponentDefinition
		cd2        corev1beta1.ComponentDefinition
		td1        corev1beta1.TraitDefinition
		td2        corev1beta1.TraitDefinition
		component1 string
		component2 string
		trait1     string
		trait2     string
	)
	BeforeEach(func() {
		c = common.Args{}
		c.SetClient(k8sClient)
		ctx = context.Background()
		ns = "cluster-test-ns"
		defaultNS = types.DefaultKubeVelaNS
		component1 = "cd1"
		component2 = "cd2"
		trait1 = "td1"
		trait2 = "td2"

		By("create namespace")
		Expect(k8sClient.Create(ctx, &corev1.Namespace{ObjectMeta: metav1.ObjectMeta{Name: ns}})).Should(SatisfyAny(BeNil(), &util.AlreadyExistMatcher{}))
		Expect(k8sClient.Create(ctx, &corev1.Namespace{ObjectMeta: metav1.ObjectMeta{Name: defaultNS}})).Should(SatisfyAny(BeNil(), &util.AlreadyExistMatcher{}))

		By("create ComponentDefinition")
		data, _ := os.ReadFile("testdata/componentDef.yaml")
		yaml.Unmarshal(data, &cd1)
		yaml.Unmarshal(data, &cd2)
		cd1.Namespace = ns
		cd1.Name = component1
		Expect(k8sClient.Create(ctx, &cd1)).Should(SatisfyAny(BeNil(), &util.AlreadyExistMatcher{}))

		cd2.Namespace = defaultNS
		cd2.Name = component2
		Expect(k8sClient.Create(ctx, &cd2)).Should(SatisfyAny(BeNil(), &util.AlreadyExistMatcher{}))

		By("create TraitDefinition")
		data, _ = os.ReadFile("testdata/manualscalars.yaml")
		yaml.Unmarshal(data, &td1)
		yaml.Unmarshal(data, &td2)
		td1.Namespace = ns
		td1.Name = trait1
		Expect(k8sClient.Create(ctx, &td1)).Should(SatisfyAny(BeNil(), &util.AlreadyExistMatcher{}))

		td2.Namespace = defaultNS
		td2.Name = trait2
		Expect(k8sClient.Create(ctx, &td2)).Should(SatisfyAny(BeNil(), &util.AlreadyExistMatcher{}))

	})

	It("get namespaced capabilities", func() {
		Context("found all capabilities", func() {
			capabilities, err := GetNamespacedCapabilitiesFromCluster(ctx, ns, c, nil)
			Expect(len(capabilities)).Should(Equal(4))
			Expect(err).Should(BeNil())
		})

		Context("found two capabilities with a bad namespace", func() {
			capabilities, err := GetNamespacedCapabilitiesFromCluster(ctx, "a-bad-ns", c, nil)
			Expect(len(capabilities)).Should(Equal(2))
			Expect(err).Should(BeNil())
		})

	})
})