package azure

import (
	"fmt"
	"os"
	"testing"

	"github.com/hashicorp/terraform/helper/resource"
	"github.com/hashicorp/terraform/terraform"
	"github.com/svanharmelen/azure-sdk-for-go/management/hostedservice"
	"github.com/svanharmelen/azure-sdk-for-go/management/virtualmachine"
)

func TestAccAzureInstance_basic(t *testing.T) {
	var dpmt virtualmachine.DeploymentResponse

	resource.Test(t, resource.TestCase{
		PreCheck:     func() { testAccPreCheck(t) },
		Providers:    testAccProviders,
		CheckDestroy: testAccCheckAzureInstanceDestroy,
		Steps: []resource.TestStep{
			resource.TestStep{
				Config: testAccAzureInstance_basic,
				Check: resource.ComposeTestCheckFunc(
					testAccCheckAzureInstanceExists(
						"azure_instance.foo", &dpmt),
					testAccCheckAzureInstanceBasicAttributes(&dpmt),
					resource.TestCheckResourceAttr(
						"azure_instance.foo", "name", "terraform-test"),
					resource.TestCheckResourceAttr(
						"azure_instance.foo", "location", "West US"),
					resource.TestCheckResourceAttr(
						"azure_instance.foo", "endpoint.2462817782.public_port", "22"),
				),
			},
		},
	})
}

func TestAccAzureInstance_advanced(t *testing.T) {
	var dpmt virtualmachine.DeploymentResponse

	resource.Test(t, resource.TestCase{
		PreCheck:     func() { testAccPreCheck(t) },
		Providers:    testAccProviders,
		CheckDestroy: testAccCheckAzureInstanceDestroy,
		Steps: []resource.TestStep{
			resource.TestStep{
				Config: testAccAzureInstance_advanced,
				Check: resource.ComposeTestCheckFunc(
					testAccCheckAzureInstanceExists(
						"azure_instance.foo", &dpmt),
					testAccCheckAzureInstanceAdvancedAttributes(&dpmt),
					resource.TestCheckResourceAttr(
						"azure_instance.foo", "name", "terraform-test1"),
					resource.TestCheckResourceAttr(
						"azure_instance.foo", "size", "Basic_A1"),
					resource.TestCheckResourceAttr(
						"azure_instance.foo", "subnet", "subnet1"),
					resource.TestCheckResourceAttr(
						"azure_instance.foo", "virtual_network", "terraform-vnet"),
					resource.TestCheckResourceAttr(
						"azure_instance.foo", "security_group", "terraform-security-group1"),
					resource.TestCheckResourceAttr(
						"azure_instance.foo", "endpoint.1814039778.public_port", "3389"),
				),
			},
		},
	})
}

func TestAccAzureInstance_update(t *testing.T) {
	var dpmt virtualmachine.DeploymentResponse

	resource.Test(t, resource.TestCase{
		PreCheck:     func() { testAccPreCheck(t) },
		Providers:    testAccProviders,
		CheckDestroy: testAccCheckAzureInstanceDestroy,
		Steps: []resource.TestStep{
			resource.TestStep{
				Config: testAccAzureInstance_advanced,
				Check: resource.ComposeTestCheckFunc(
					testAccCheckAzureInstanceExists(
						"azure_instance.foo", &dpmt),
					testAccCheckAzureInstanceAdvancedAttributes(&dpmt),
					resource.TestCheckResourceAttr(
						"azure_instance.foo", "name", "terraform-test1"),
					resource.TestCheckResourceAttr(
						"azure_instance.foo", "size", "Basic_A1"),
					resource.TestCheckResourceAttr(
						"azure_instance.foo", "subnet", "subnet1"),
					resource.TestCheckResourceAttr(
						"azure_instance.foo", "virtual_network", "terraform-vnet"),
					resource.TestCheckResourceAttr(
						"azure_instance.foo", "security_group", "terraform-security-group1"),
					resource.TestCheckResourceAttr(
						"azure_instance.foo", "endpoint.1814039778.public_port", "3389"),
				),
			},

			resource.TestStep{
				Config: testAccAzureInstance_update,
				Check: resource.ComposeTestCheckFunc(
					testAccCheckAzureInstanceExists(
						"azure_instance.foo", &dpmt),
					testAccCheckAzureInstanceUpdatedAttributes(&dpmt),
					resource.TestCheckResourceAttr(
						"azure_instance.foo", "size", "Basic_A2"),
					resource.TestCheckResourceAttr(
						"azure_instance.foo", "security_group", "terraform-security-group2"),
					resource.TestCheckResourceAttr(
						"azure_instance.foo", "endpoint.1814039778.public_port", "3389"),
					resource.TestCheckResourceAttr(
						"azure_instance.foo", "endpoint.3713350066.public_port", "5985"),
				),
			},
		},
	})
}

func testAccCheckAzureInstanceExists(
	n string,
	dpmt *virtualmachine.DeploymentResponse) resource.TestCheckFunc {
	return func(s *terraform.State) error {
		rs, ok := s.RootModule().Resources[n]
		if !ok {
			return fmt.Errorf("Not found: %s", n)
		}

		if rs.Primary.ID == "" {
			return fmt.Errorf("No instance ID is set")
		}

		mc := testAccProvider.Meta().(*Client).mgmtClient
		vm, err := virtualmachine.NewClient(mc).GetDeployment(rs.Primary.ID, rs.Primary.ID)
		if err != nil {
			return err
		}

		if vm.Name != rs.Primary.ID {
			return fmt.Errorf("Instance not found")
		}

		*dpmt = vm

		return nil
	}
}

func testAccCheckAzureInstanceBasicAttributes(
	dpmt *virtualmachine.DeploymentResponse) resource.TestCheckFunc {
	return func(s *terraform.State) error {

		if dpmt.Name != "terraform-test" {
			return fmt.Errorf("Bad name: %s", dpmt.Name)
		}

		if len(dpmt.RoleList) != 1 {
			return fmt.Errorf(
				"Instance %s has an unexpected number of roles: %d", dpmt.Name, len(dpmt.RoleList))
		}

		if dpmt.RoleList[0].RoleSize != "Basic_A1" {
			return fmt.Errorf("Bad size: %s", dpmt.RoleList[0].RoleSize)
		}

		return nil
	}
}

func testAccCheckAzureInstanceAdvancedAttributes(
	dpmt *virtualmachine.DeploymentResponse) resource.TestCheckFunc {
	return func(s *terraform.State) error {

		if dpmt.Name != "terraform-test1" {
			return fmt.Errorf("Bad name: %s", dpmt.Name)
		}

		if dpmt.VirtualNetworkName != "terraform-vnet" {
			return fmt.Errorf("Bad virtual network: %s", dpmt.VirtualNetworkName)
		}

		if len(dpmt.RoleList) != 1 {
			return fmt.Errorf(
				"Instance %s has an unexpected number of roles: %d", dpmt.Name, len(dpmt.RoleList))
		}

		if dpmt.RoleList[0].RoleSize != "Basic_A1" {
			return fmt.Errorf("Bad size: %s", dpmt.RoleList[0].RoleSize)
		}

		for _, c := range dpmt.RoleList[0].ConfigurationSets {
			if c.ConfigurationSetType == virtualmachine.ConfigurationSetTypeNetwork {
				if len(c.InputEndpoints) != 1 {
					return fmt.Errorf(
						"Instance %s has an unexpected number of endpoints %d",
						dpmt.Name, len(c.InputEndpoints))
				}

				if c.InputEndpoints[0].Name != "RDP" {
					return fmt.Errorf("Bad endpoint name: %s", c.InputEndpoints[0].Name)
				}

				if c.InputEndpoints[0].Port != 3389 {
					return fmt.Errorf("Bad endpoint port: %d", c.InputEndpoints[0].Port)
				}

				if len(c.SubnetNames) != 1 {
					return fmt.Errorf(
						"Instance %s has an unexpected number of associated subnets %d",
						dpmt.Name, len(c.SubnetNames))
				}

				if c.SubnetNames[0] != "subnet1" {
					return fmt.Errorf("Bad subnet: %s", c.SubnetNames[0])
				}

				if c.NetworkSecurityGroup != "terraform-security-group1" {
					return fmt.Errorf("Bad security group: %s", c.NetworkSecurityGroup)
				}
			}
		}

		return nil
	}
}

func testAccCheckAzureInstanceUpdatedAttributes(
	dpmt *virtualmachine.DeploymentResponse) resource.TestCheckFunc {
	return func(s *terraform.State) error {

		if dpmt.Name != "terraform-test1" {
			return fmt.Errorf("Bad name: %s", dpmt.Name)
		}

		if dpmt.VirtualNetworkName != "terraform-vnet" {
			return fmt.Errorf("Bad virtual network: %s", dpmt.VirtualNetworkName)
		}

		if len(dpmt.RoleList) != 1 {
			return fmt.Errorf(
				"Instance %s has an unexpected number of roles: %d", dpmt.Name, len(dpmt.RoleList))
		}

		if dpmt.RoleList[0].RoleSize != "Basic_A2" {
			return fmt.Errorf("Bad size: %s", dpmt.RoleList[0].RoleSize)
		}

		for _, c := range dpmt.RoleList[0].ConfigurationSets {
			if c.ConfigurationSetType == virtualmachine.ConfigurationSetTypeNetwork {
				if len(c.InputEndpoints) != 2 {
					return fmt.Errorf(
						"Instance %s has an unexpected number of endpoints %d",
						dpmt.Name, len(c.InputEndpoints))
				}

				if c.InputEndpoints[1].Name != "WINRM" {
					return fmt.Errorf("Bad endpoint name: %s", c.InputEndpoints[1].Name)
				}

				if c.InputEndpoints[1].Port != 5985 {
					return fmt.Errorf("Bad endpoint port: %d", c.InputEndpoints[1].Port)
				}

				if len(c.SubnetNames) != 1 {
					return fmt.Errorf(
						"Instance %s has an unexpected number of associated subnets %d",
						dpmt.Name, len(c.SubnetNames))
				}

				if c.SubnetNames[0] != "subnet1" {
					return fmt.Errorf("Bad subnet: %s", c.SubnetNames[0])
				}

				if c.NetworkSecurityGroup != "terraform-security-group2" {
					return fmt.Errorf("Bad security group: %s", c.NetworkSecurityGroup)
				}
			}
		}

		return nil
	}
}

func testAccCheckAzureInstanceDestroy(s *terraform.State) error {
	mc := testAccProvider.Meta().(*Client).mgmtClient

	for _, rs := range s.RootModule().Resources {
		if rs.Type != "azure_instance" {
			continue
		}

		if rs.Primary.ID == "" {
			return fmt.Errorf("No instance ID is set")
		}

		req, err := hostedservice.NewClient(mc).DeleteHostedService(rs.Primary.ID, true)
		if err != nil {
			return fmt.Errorf("Error deleting instance (%s): %s", rs.Primary.ID, err)
		}

		// Wait until the instance is deleted
		if err := mc.WaitForOperation(req, nil); err != nil {
			return fmt.Errorf(
				"Error deleting instance (%s): %s", rs.Primary.ID, err)
		}
	}

	return nil
}

var testAccAzureInstance_basic = fmt.Sprintf(`
resource "azure_instance" "foo" {
    name = "terraform-test"
    image = "Ubuntu Server 14.04 LTS"
    size = "Basic_A1"
    storage = "%s"
    location = "West US"
    username = "terraform"
    password = "Pass!admin123"

    endpoint {
        name = "SSH"
        protocol = "tcp"
        public_port = 22
        private_port = 22
    }
}`, os.Getenv("AZURE_STORAGE"))

var testAccAzureInstance_advanced = fmt.Sprintf(`
resource "azure_virtual_network" "foo" {
    name = "terraform-vnet"
    address_space = ["10.1.2.0/24"]
		location = "West US"

		subnet {
        name = "subnet1"
				address_prefix = "10.1.2.0/25"
		}

		subnet {
        name = "subnet2"
				address_prefix = "10.1.2.128/25"
    }
}

resource "azure_security_group" "foo" {
    name = "terraform-security-group1"
    location = "West US"

    rule {
        name = "rdp"
        priority = 101
        source_cidr = "*"
        source_port = "*"
        destination_cidr = "*"
        destination_port = 3389
        protocol = "TCP"
    }
}

resource "azure_instance" "foo" {
    name = "terraform-test1"
    image = "Windows Server 2012 R2 Datacenter, April 2015"
    size = "Basic_A1"
    storage = "%s"
    location = "West US"
    time_zone = "America/Los_Angeles"
    subnet = "subnet1"
    virtual_network = "${azure_virtual_network.foo.name}"
    security_group = "${azure_security_group.foo.name}"
    username = "terraform"
    password = "Pass!admin123"

    endpoint {
        name = "RDP"
        protocol = "tcp"
        public_port = 3389
        private_port = 3389
    }
}`, os.Getenv("AZURE_STORAGE"))

var testAccAzureInstance_update = fmt.Sprintf(`
resource "azure_virtual_network" "foo" {
    name = "terraform-vnet"
    address_space = ["10.1.2.0/24"]
		location = "West US"

		subnet {
        name = "subnet1"
				address_prefix = "10.1.2.0/25"
		}

		subnet {
        name = "subnet2"
				address_prefix = "10.1.2.128/25"
    }
}

resource "azure_security_group" "foo" {
    name = "terraform-security-group1"
    location = "West US"

    rule {
        name = "rdp"
        priority = 101
        source_cidr = "*"
        source_port = "*"
        destination_cidr = "*"
        destination_port = 3389
        protocol = "TCP"
    }
}

resource "azure_security_group" "bar" {
    name = "terraform-security-group2"
    location = "West US"

    rule {
        name = "rdp"
        priority = 101
        source_cidr = "192.168.0.0/24"
        source_port = "*"
        destination_cidr = "*"
        destination_port = 3389
        protocol = "TCP"
    }
}

resource "azure_instance" "foo" {
    name = "terraform-test1"
    image = "Windows Server 2012 R2 Datacenter, April 2015"
    size = "Basic_A2"
    storage = "%s"
    location = "West US"
    time_zone = "America/Los_Angeles"
    subnet = "subnet1"
    virtual_network = "${azure_virtual_network.foo.name}"
    security_group = "${azure_security_group.bar.name}"
    username = "terraform"
    password = "Pass!admin123"

    endpoint {
        name = "RDP"
        protocol = "tcp"
        public_port = 3389
        private_port = 3389
    }

    endpoint {
        name = "WINRM"
        protocol = "tcp"
        public_port = 5985
        private_port = 5985
    }
}`, os.Getenv("AZURE_STORAGE"))