// Copyright (c) 2017-2018 Tigera, Inc. All rights reserved.

// 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 converters

import (
	. "github.com/onsi/ginkgo/extensions/table"
	. "github.com/onsi/gomega"
	v1 "k8s.io/apimachinery/pkg/apis/meta/v1"

	apiv3 "github.com/projectcalico/api/pkg/apis/projectcalico/v3"

	apiv1 "github.com/projectcalico/calico/libcalico-go/lib/apis/v1"
	"github.com/projectcalico/calico/libcalico-go/lib/apis/v1/unversioned"
	"github.com/projectcalico/calico/libcalico-go/lib/backend/model"
)

var order1 = 1000.00
var order2 = 999.99
var policyTable = []TableEntry{
	Entry("fully populated Policy",
		&apiv1.Policy{
			Metadata: apiv1.PolicyMetadata{
				Name: "nameyMcPolicyName",
			},
			Spec: apiv1.PolicySpec{
				Order:        &order1,
				IngressRules: []apiv1.Rule{V1InRule1, V1InRule2},
				EgressRules:  []apiv1.Rule{V1EgressRule1, V1EgressRule2},
				Selector:     "calico/k8s_ns == 'default' || thing == 'value'",
				DoNotTrack:   false, // DoNotTrack and PreDNAT cannot both be true.
				PreDNAT:      true,
				Types:        []apiv1.PolicyType{apiv1.PolicyTypeIngress},
			},
		},
		&model.KVPair{
			Key: model.PolicyKey{
				Name: "nameyMcPolicyName",
			},
			Value: &model.Policy{
				Order:          &order1,
				InboundRules:   []model.Rule{V1ModelInRule1, V1ModelInRule2},
				OutboundRules:  []model.Rule{V1ModelEgressRule1, V1ModelEgressRule2},
				Selector:       "calico/k8s_ns == 'default' || thing == 'value'",
				DoNotTrack:     false,
				PreDNAT:        true,
				ApplyOnForward: true,
				Types:          []string{"ingress"},
			},
		},
		apiv3.GlobalNetworkPolicy{
			ObjectMeta: v1.ObjectMeta{
				Name: "nameymcpolicyname-32df456f",
			},
			Spec: apiv3.GlobalNetworkPolicySpec{
				Order:          &order1,
				Ingress:        []apiv3.Rule{V3InRule1, V3InRule2},
				Egress:         []apiv3.Rule{V3EgressRule1, V3EgressRule2},
				Selector:       "projectcalico.org/namespace == 'default' || thing == 'value'",
				DoNotTrack:     false,
				PreDNAT:        true,
				ApplyOnForward: true,
				Types:          []apiv3.PolicyType{apiv3.PolicyTypeIngress},
			},
		},
	),
	Entry("policy name conversion",
		&apiv1.Policy{
			Metadata: apiv1.PolicyMetadata{
				Name: "MaKe.-.MaKe",
			},
			Spec: apiv1.PolicySpec{
				Order:        &order1,
				IngressRules: []apiv1.Rule{V1InRule2},
				EgressRules:  []apiv1.Rule{V1EgressRule1},
				Selector:     "thing == 'value'",
				DoNotTrack:   true,
				PreDNAT:      false,
				Types:        []apiv1.PolicyType{apiv1.PolicyTypeIngress},
			},
		},
		&model.KVPair{
			Key: model.PolicyKey{
				Name: "MaKe.-.MaKe",
			},
			Value: &model.Policy{
				Order:          &order1,
				InboundRules:   []model.Rule{V1ModelInRule2},
				OutboundRules:  []model.Rule{V1ModelEgressRule1},
				Selector:       "thing == 'value'",
				DoNotTrack:     true,
				PreDNAT:        false,
				ApplyOnForward: true,
				Types:          []string{"ingress"},
			},
		},
		apiv3.GlobalNetworkPolicy{
			ObjectMeta: v1.ObjectMeta{
				Name: "make-make-1b6971c8",
			},
			Spec: apiv3.GlobalNetworkPolicySpec{
				Order:          &order1,
				Ingress:        []apiv3.Rule{V3InRule2},
				Egress:         []apiv3.Rule{V3EgressRule1},
				Selector:       "thing == 'value'",
				DoNotTrack:     true,
				PreDNAT:        false,
				ApplyOnForward: true,
				Types:          []apiv3.PolicyType{apiv3.PolicyTypeIngress},
			},
		},
	),
	Entry("policy with PreDNAT set to true should convert ApplyOnForward to true in v3 API",
		&apiv1.Policy{
			Metadata: apiv1.PolicyMetadata{
				Name: "RAWR",
			},
			Spec: apiv1.PolicySpec{
				Order:        &order1,
				IngressRules: []apiv1.Rule{V1InRule2},
				PreDNAT:      true,
				Types:        []apiv1.PolicyType{apiv1.PolicyTypeIngress},
			},
		},
		&model.KVPair{
			Key: model.PolicyKey{
				Name: "RAWR",
			},
			Value: &model.Policy{
				Order:          &order1,
				InboundRules:   []model.Rule{V1ModelInRule2},
				OutboundRules:  []model.Rule{},
				DoNotTrack:     false,
				PreDNAT:        true,
				ApplyOnForward: true,
				Types:          []string{"ingress"},
			},
		},
		apiv3.GlobalNetworkPolicy{
			ObjectMeta: v1.ObjectMeta{
				Name: "rawr-03d81e1d",
			},
			Spec: apiv3.GlobalNetworkPolicySpec{
				Order:          &order1,
				Ingress:        []apiv3.Rule{V3InRule2},
				Egress:         []apiv3.Rule{},
				DoNotTrack:     false,
				PreDNAT:        true,
				ApplyOnForward: true, // notice this gets converted to true, because PreDNAT are true.
				Types:          []apiv3.PolicyType{apiv3.PolicyTypeIngress},
			},
		},
	),
	Entry("policy with DoNotTrack set to true "+
		"should convert ApplyOnForward to true in v3 API",
		&apiv1.Policy{
			Metadata: apiv1.PolicyMetadata{
				Name: "RAWR",
			},
			Spec: apiv1.PolicySpec{
				Order:        &order1,
				IngressRules: []apiv1.Rule{V1InRule2},
				DoNotTrack:   true,
				Types:        []apiv1.PolicyType{apiv1.PolicyTypeIngress},
			},
		},
		&model.KVPair{
			Key: model.PolicyKey{
				Name: "RAWR",
			},
			Value: &model.Policy{
				Order:          &order1,
				InboundRules:   []model.Rule{V1ModelInRule2},
				OutboundRules:  []model.Rule{},
				DoNotTrack:     true,
				PreDNAT:        false,
				ApplyOnForward: true,
				Types:          []string{"ingress"},
			},
		},
		apiv3.GlobalNetworkPolicy{
			ObjectMeta: v1.ObjectMeta{
				Name: "rawr-03d81e1d",
			},
			Spec: apiv3.GlobalNetworkPolicySpec{
				Order:          &order1,
				Ingress:        []apiv3.Rule{V3InRule2},
				Egress:         []apiv3.Rule{},
				DoNotTrack:     true,
				PreDNAT:        false,
				ApplyOnForward: true, // notice this gets converted to true, because DoNotTrack are true.
				Types:          []apiv3.PolicyType{apiv3.PolicyTypeIngress},
			},
		},
	),
	Entry("policy with PreDNAT and DoNotTrack both set to false should NOT convert ApplyOnForward to true in v3 API",
		&apiv1.Policy{
			Metadata: apiv1.PolicyMetadata{
				Name: "meow",
			},
			Spec: apiv1.PolicySpec{
				Order:        &order1,
				IngressRules: []apiv1.Rule{V1InRule2},
				DoNotTrack:   false,
				PreDNAT:      false,
				Types:        []apiv1.PolicyType{apiv1.PolicyTypeIngress},
			},
		},
		&model.KVPair{
			Key: model.PolicyKey{
				Name: "meow",
			},
			Value: &model.Policy{
				Order:          &order1,
				InboundRules:   []model.Rule{V1ModelInRule2},
				OutboundRules:  []model.Rule{},
				DoNotTrack:     false,
				PreDNAT:        false,
				ApplyOnForward: false,
				Types:          []string{"ingress"},
			},
		},
		apiv3.GlobalNetworkPolicy{
			ObjectMeta: v1.ObjectMeta{
				Name: "meow",
			},
			Spec: apiv3.GlobalNetworkPolicySpec{
				Order:          &order1,
				Ingress:        []apiv3.Rule{V3InRule2},
				Egress:         []apiv3.Rule{},
				DoNotTrack:     false,
				PreDNAT:        false,
				ApplyOnForward: false,
				Types:          []apiv3.PolicyType{apiv3.PolicyTypeIngress},
			},
		},
	),
	Entry("policy with non-strictly masked CIDR should get converted to strictly masked CIDR in v3 API",
		&apiv1.Policy{
			Metadata: apiv1.PolicyMetadata{
				Name: "MaKe.-.MaKe",
			},
			Spec: apiv1.PolicySpec{
				Order: &order1,
				// Source Nets selector in V1InRule1 and V1EgressRule1 are non-strictly masked CIDRs.
				IngressRules: []apiv1.Rule{V1InRule1},
				EgressRules:  []apiv1.Rule{V1EgressRule1},
				Selector:     "thing == 'value'",
				DoNotTrack:   true,
				PreDNAT:      false,
				Types:        []apiv1.PolicyType{apiv1.PolicyTypeIngress, apiv1.PolicyTypeEgress},
			},
		},
		&model.KVPair{
			Key: model.PolicyKey{
				Name: "MaKe.-.MaKe",
			},
			Value: &model.Policy{
				Order: &order1,
				// Source Nets selector in V1ModelInRule1 and V1ModelEgressRule1 are non-strictly masked CIDRs.
				InboundRules:   []model.Rule{V1ModelInRule1},
				OutboundRules:  []model.Rule{V1ModelEgressRule1},
				Selector:       "thing == 'value'",
				DoNotTrack:     true,
				PreDNAT:        false,
				ApplyOnForward: true,
				Types:          []string{"ingress", "egress"},
			},
		},
		apiv3.GlobalNetworkPolicy{
			ObjectMeta: v1.ObjectMeta{
				Name: "make-make-1b6971c8",
			},
			Spec: apiv3.GlobalNetworkPolicySpec{
				Order: &order1,
				// Source Nets selector in V3InRule1 and V3EgressRule1 are strictly masked CIDRs.
				Ingress:        []apiv3.Rule{V3InRule1},
				Egress:         []apiv3.Rule{V3EgressRule1},
				Selector:       "thing == 'value'",
				DoNotTrack:     true,
				PreDNAT:        false,
				ApplyOnForward: true,
				Types:          []apiv3.PolicyType{apiv3.PolicyTypeIngress, apiv3.PolicyTypeEgress},
			},
		},
	),
	Entry("policy test3",
		&apiv1.Policy{
			Metadata: apiv1.PolicyMetadata{
				Name: "policy1",
			},
			Spec: apiv1.PolicySpec{
				Order: &order1,
				// Source Nets selector in V1InRule1 and V1EgressRule1 are non-strictly masked CIDRs.
				IngressRules: []apiv1.Rule{
					{
						Action: "deny",
						Source: apiv1.EntityRule{
							Selector: "type=='application'",
						},
					},
				},
				Selector: "type=='database'",
			},
		},
		&model.KVPair{
			Key: model.PolicyKey{
				Name: "policy1",
			},
			Value: &model.Policy{
				Order: &order1,
				// Source Nets selector in V1ModelInRule1 and V1ModelEgressRule1 are non-strictly masked CIDRs.
				InboundRules: []model.Rule{{
					Action:      "deny",
					SrcSelector: "type=='application'",
				}},
				OutboundRules: []model.Rule{},
				Selector:      "type=='database'",
				Types:         []string{"ingress"},
			},
		},
		apiv3.GlobalNetworkPolicy{
			ObjectMeta: v1.ObjectMeta{
				Name: "policy1",
			},
			Spec: apiv3.GlobalNetworkPolicySpec{
				Order: &order1,
				// Source Nets selector in V3InRule1 and V3EgressRule1 are strictly masked CIDRs.
				Ingress: []apiv3.Rule{{
					Action: apiv3.Deny,
					Source: apiv3.EntityRule{
						Selector: "type=='application'",
					},
				}},
				Egress:   []apiv3.Rule{},
				Selector: "type=='database'",
				Types:    []apiv3.PolicyType{apiv3.PolicyTypeIngress},
			},
		},
	),
}

var _ = DescribeTable("v1->v3 policy conversion tests",
	func(v1API unversioned.Resource, v1KVP *model.KVPair, v3API apiv3.GlobalNetworkPolicy) {
		p := Policy{}

		// Test and assert v1 API to v1 backend logic.
		v1KVPResult, err := p.APIV1ToBackendV1(v1API)
		Expect(err).NotTo(HaveOccurred())
		Expect(v1KVPResult.Key.(model.PolicyKey).Name).To(Equal(v1KVP.Key.(model.PolicyKey).Name))
		Expect(v1KVPResult.Value.(*model.Policy)).To(Equal(v1KVP.Value))

		// Test and assert v1 backend to v3 API logic.
		v3APIResult, err := p.BackendV1ToAPIV3(v1KVP)
		Expect(err).NotTo(HaveOccurred())
		Expect(v3APIResult.(*apiv3.GlobalNetworkPolicy).Name).To(Equal(v3API.Name))
		Expect(v3APIResult.(*apiv3.GlobalNetworkPolicy).Spec).To(Equal(v3API.Spec))
	},

	policyTable...,
)

var policyTableBackend = []TableEntry{
	Entry("converting model with no Types with preDNAT to v3 API only has Ingress for Types",
		&model.KVPair{
			Key: model.PolicyKey{
				Name: "somekey",
			},
			Value: &model.Policy{
				Order: &order1,
				// Source Nets selector in V1ModelInRule1 and V1ModelEgressRule1 are non-strictly masked CIDRs.
				InboundRules:   []model.Rule{V1ModelInRule1},
				OutboundRules:  []model.Rule{},
				Selector:       "thing == 'value'",
				DoNotTrack:     true,
				PreDNAT:        true,
				ApplyOnForward: true,
				Types:          []string{},
			},
		},
		apiv3.GlobalNetworkPolicy{
			ObjectMeta: v1.ObjectMeta{
				Name: "somekey",
			},
			Spec: apiv3.GlobalNetworkPolicySpec{
				Order: &order1,
				// Source Nets selector in V3InRule1 and V3EgressRule1 are strictly masked CIDRs.
				Ingress:        []apiv3.Rule{V3InRule1},
				Egress:         []apiv3.Rule{},
				Selector:       "thing == 'value'",
				DoNotTrack:     true,
				PreDNAT:        true,
				ApplyOnForward: true,
				Types:          []apiv3.PolicyType{apiv3.PolicyTypeIngress},
			},
		},
	),
}

var _ = DescribeTable("v1->v3 policy conversion tests (backend)",
	func(v1KVP *model.KVPair, v3API apiv3.GlobalNetworkPolicy) {
		p := Policy{}

		// Test and assert v1 backend to v3 API logic.
		v3APIResult, err := p.BackendV1ToAPIV3(v1KVP)
		Expect(err).NotTo(HaveOccurred())
		Expect(v3APIResult.(*apiv3.GlobalNetworkPolicy).Name).To(Equal(v3API.Name))
		Expect(v3APIResult.(*apiv3.GlobalNetworkPolicy).Spec).To(Equal(v3API.Spec))
	},

	policyTableBackend...,
)
