package awsec2query

import (
	"testing"

	"github.com/aws/aws-sdk-go/aws"
	"github.com/aws/aws-sdk-go/service/ec2"
	"github.com/stretchr/testify/assert"
)

func TestQueryParamsToStruct_RunInstances(t *testing.T) {
	args := map[string]string{
		"Action":       "RunInstances",
		"ImageId":      "ami-123456789",
		"MinCount":     "1",
		"MaxCount":     "2",
		"InstanceType": "t2.micro",
		"KeyName":      "my-key-pair",
	}

	input := &ec2.RunInstancesInput{}
	err := QueryParamsToStruct(args, input)

	assert.NoError(t, err)
	assert.Equal(t, "ami-123456789", aws.StringValue(input.ImageId))
	assert.Equal(t, int64(1), aws.Int64Value(input.MinCount))
	assert.Equal(t, int64(2), aws.Int64Value(input.MaxCount))
	assert.Equal(t, "t2.micro", aws.StringValue(input.InstanceType))
	assert.Equal(t, "my-key-pair", aws.StringValue(input.KeyName))
}

func TestQueryParamsToStruct_RunInstancesWithSecurityGroups(t *testing.T) {
	args := map[string]string{
		"Action":            "RunInstances",
		"ImageId":           "ami-123456789",
		"MinCount":          "1",
		"MaxCount":          "1",
		"SecurityGroupId.1": "sg-12345678",
		"SecurityGroupId.2": "sg-87654321",
	}

	input := &ec2.RunInstancesInput{}
	err := QueryParamsToStruct(args, input)

	assert.NoError(t, err)
	assert.Equal(t, "ami-123456789", aws.StringValue(input.ImageId))
	assert.Len(t, input.SecurityGroupIds, 2)
	assert.Equal(t, "sg-12345678", aws.StringValue(input.SecurityGroupIds[0]))
	assert.Equal(t, "sg-87654321", aws.StringValue(input.SecurityGroupIds[1]))
}

func TestQueryParamsToStruct_DescribeInstances(t *testing.T) {
	args := map[string]string{
		"Action":       "DescribeInstances",
		"InstanceId.1": "i-1234567890abcdef0",
		"InstanceId.2": "i-0987654321fedcba0",
		"MaxResults":   "100",
	}

	input := &ec2.DescribeInstancesInput{}
	err := QueryParamsToStruct(args, input)

	assert.NoError(t, err)
	assert.Len(t, input.InstanceIds, 2)
	assert.Equal(t, "i-1234567890abcdef0", aws.StringValue(input.InstanceIds[0]))
	assert.Equal(t, "i-0987654321fedcba0", aws.StringValue(input.InstanceIds[1]))
	assert.Equal(t, int64(100), aws.Int64Value(input.MaxResults))
}

func TestQueryParamsToStruct_DescribeInstancesWithFilters(t *testing.T) {
	args := map[string]string{
		"Action":           "DescribeInstances",
		"Filter.1.Name":    "instance-type",
		"Filter.1.Value.1": "t2.micro",
		"Filter.1.Value.2": "t2.small",
		"Filter.2.Name":    "instance-state-name",
		"Filter.2.Value.1": "running",
	}

	input := &ec2.DescribeInstancesInput{}
	err := QueryParamsToStruct(args, input)

	assert.NoError(t, err)
	assert.Len(t, input.Filters, 2)
	assert.Equal(t, "instance-type", aws.StringValue(input.Filters[0].Name))
	assert.Len(t, input.Filters[0].Values, 2)
	assert.Equal(t, "t2.micro", aws.StringValue(input.Filters[0].Values[0]))
	assert.Equal(t, "t2.small", aws.StringValue(input.Filters[0].Values[1]))
	assert.Equal(t, "instance-state-name", aws.StringValue(input.Filters[1].Name))
	assert.Len(t, input.Filters[1].Values, 1)
	assert.Equal(t, "running", aws.StringValue(input.Filters[1].Values[0]))
}

func TestQueryParamsToStruct_StopInstances(t *testing.T) {
	args := map[string]string{
		"Action":       "StopInstances",
		"InstanceId.1": "i-1234567890abcdef0",
		"InstanceId.2": "i-0987654321fedcba0",
		"Force":        "true",
	}

	input := &ec2.StopInstancesInput{}
	err := QueryParamsToStruct(args, input)

	assert.NoError(t, err)
	assert.Len(t, input.InstanceIds, 2)
	assert.Equal(t, "i-1234567890abcdef0", aws.StringValue(input.InstanceIds[0]))
	assert.Equal(t, "i-0987654321fedcba0", aws.StringValue(input.InstanceIds[1]))
	assert.Equal(t, true, aws.BoolValue(input.Force))
}

func TestQueryParamsToStruct_StartInstances(t *testing.T) {
	args := map[string]string{
		"Action":       "StartInstances",
		"InstanceId.1": "i-1234567890abcdef0",
		"InstanceId.2": "i-0987654321fedcba0",
		"InstanceId.3": "i-abcdef1234567890",
	}

	input := &ec2.StartInstancesInput{}
	err := QueryParamsToStruct(args, input)

	assert.NoError(t, err)
	assert.Len(t, input.InstanceIds, 3)
	assert.Equal(t, "i-1234567890abcdef0", aws.StringValue(input.InstanceIds[0]))
	assert.Equal(t, "i-0987654321fedcba0", aws.StringValue(input.InstanceIds[1]))
	assert.Equal(t, "i-abcdef1234567890", aws.StringValue(input.InstanceIds[2]))
}

func TestQueryParamsToStruct_RebootInstances(t *testing.T) {
	args := map[string]string{
		"Action":       "RebootInstances",
		"InstanceId.1": "i-1234567890abcdef0",
	}

	input := &ec2.RebootInstancesInput{}
	err := QueryParamsToStruct(args, input)

	assert.NoError(t, err)
	assert.Len(t, input.InstanceIds, 1)
	assert.Equal(t, "i-1234567890abcdef0", aws.StringValue(input.InstanceIds[0]))
}

func TestQueryParamsToStruct_TerminateInstances(t *testing.T) {
	args := map[string]string{
		"Action":       "TerminateInstances",
		"InstanceId.1": "i-1234567890abcdef0",
		"InstanceId.2": "i-0987654321fedcba0",
	}

	input := &ec2.TerminateInstancesInput{}
	err := QueryParamsToStruct(args, input)

	assert.NoError(t, err)
	assert.Len(t, input.InstanceIds, 2)
	assert.Equal(t, "i-1234567890abcdef0", aws.StringValue(input.InstanceIds[0]))
	assert.Equal(t, "i-0987654321fedcba0", aws.StringValue(input.InstanceIds[1]))
}

func TestQueryParamsToStruct_ComplexTagSpecifications(t *testing.T) {
	args := map[string]string{
		"Action":                          "RunInstances",
		"ImageId":                         "ami-123456789",
		"MinCount":                        "1",
		"MaxCount":                        "1",
		"TagSpecification.1.ResourceType": "instance",
		"TagSpecification.1.Tag.1.Key":    "Name",
		"TagSpecification.1.Tag.1.Value":  "MyInstance",
		"TagSpecification.1.Tag.2.Key":    "Environment",
		"TagSpecification.1.Tag.2.Value":  "Production",
		"TagSpecification.2.ResourceType": "volume",
		"TagSpecification.2.Tag.1.Key":    "VolumeType",
		"TagSpecification.2.Tag.1.Value":  "Root",
	}

	input := &ec2.RunInstancesInput{}
	err := QueryParamsToStruct(args, input)

	assert.NoError(t, err)
	assert.Len(t, input.TagSpecifications, 2)

	// First tag specification
	assert.Equal(t, "instance", aws.StringValue(input.TagSpecifications[0].ResourceType))
	assert.Len(t, input.TagSpecifications[0].Tags, 2)
	assert.Equal(t, "Name", aws.StringValue(input.TagSpecifications[0].Tags[0].Key))
	assert.Equal(t, "MyInstance", aws.StringValue(input.TagSpecifications[0].Tags[0].Value))
	assert.Equal(t, "Environment", aws.StringValue(input.TagSpecifications[0].Tags[1].Key))
	assert.Equal(t, "Production", aws.StringValue(input.TagSpecifications[0].Tags[1].Value))

	// Second tag specification
	assert.Equal(t, "volume", aws.StringValue(input.TagSpecifications[1].ResourceType))
	assert.Len(t, input.TagSpecifications[1].Tags, 1)
	assert.Equal(t, "VolumeType", aws.StringValue(input.TagSpecifications[1].Tags[0].Key))
	assert.Equal(t, "Root", aws.StringValue(input.TagSpecifications[1].Tags[0].Value))
}

func TestQueryParamsToStruct_InvalidInput(t *testing.T) {
	args := map[string]string{
		"Action": "RunInstances",
	}

	// Test with non-pointer
	input := ec2.RunInstancesInput{}
	err := QueryParamsToStruct(args, input)
	assert.Error(t, err)
	assert.Contains(t, err.Error(), "must be a pointer")
}

func TestQueryParamsToStruct_EmptyParams(t *testing.T) {
	args := map[string]string{}

	input := &ec2.RunInstancesInput{}
	err := QueryParamsToStruct(args, input)

	assert.NoError(t, err)
	// All fields should be nil/zero
	assert.Nil(t, input.ImageId)
	assert.Nil(t, input.MinCount)
}

func TestQueryParamsToStruct_BlockDeviceMappingsWithEBS(t *testing.T) {
	args := map[string]string{
		"Action":                              "RunInstances",
		"ImageId":                             "ami-0abcdef1234567890",
		"InstanceType":                        "t2.micro",
		"MinCount":                            "1",
		"MaxCount":                            "1",
		"BlockDeviceMapping.1.DeviceName":     "/dev/sdh",
		"BlockDeviceMapping.1.Ebs.VolumeSize": "100",
		"BlockDeviceMapping.1.Ebs.VolumeType": "gp3",
		"BlockDeviceMapping.1.Ebs.DeleteOnTermination": "true",
	}

	input := &ec2.RunInstancesInput{}
	err := QueryParamsToStruct(args, input)

	assert.NoError(t, err)
	assert.Equal(t, "ami-0abcdef1234567890", aws.StringValue(input.ImageId))
	assert.Equal(t, "t2.micro", aws.StringValue(input.InstanceType))
	assert.Len(t, input.BlockDeviceMappings, 1)

	// Check first block device mapping
	bdm := input.BlockDeviceMappings[0]
	assert.Equal(t, "/dev/sdh", aws.StringValue(bdm.DeviceName))
	assert.NotNil(t, bdm.Ebs)
	assert.Equal(t, int64(100), aws.Int64Value(bdm.Ebs.VolumeSize))
	assert.Equal(t, "gp3", aws.StringValue(bdm.Ebs.VolumeType))
	assert.Equal(t, true, aws.BoolValue(bdm.Ebs.DeleteOnTermination))
}

func TestQueryParamsToStruct_BlockDeviceMappingsWithEphemeral(t *testing.T) {
	args := map[string]string{
		"Action":                           "RunInstances",
		"ImageId":                          "ami-0abcdef1234567890",
		"InstanceType":                     "m3.medium",
		"MinCount":                         "1",
		"MaxCount":                         "1",
		"BlockDeviceMapping.1.DeviceName":  "/dev/sdc",
		"BlockDeviceMapping.1.VirtualName": "ephemeral1",
	}

	input := &ec2.RunInstancesInput{}
	err := QueryParamsToStruct(args, input)

	assert.NoError(t, err)
	assert.Len(t, input.BlockDeviceMappings, 1)

	// Check ephemeral mapping
	bdm := input.BlockDeviceMappings[0]
	assert.Equal(t, "/dev/sdc", aws.StringValue(bdm.DeviceName))
	assert.Equal(t, "ephemeral1", aws.StringValue(bdm.VirtualName))
	assert.Nil(t, bdm.Ebs)
}

func TestQueryParamsToStruct_MultipleBlockDeviceMappings(t *testing.T) {
	args := map[string]string{
		"Action":       "RunInstances",
		"ImageId":      "ami-0abcdef1234567890",
		"InstanceType": "t2.micro",
		"MinCount":     "1",
		"MaxCount":     "1",
		// First mapping - EBS volume with snapshot
		"BlockDeviceMapping.1.DeviceName":              "/dev/sda1",
		"BlockDeviceMapping.1.Ebs.SnapshotId":          "snap-1234567890abcdef0",
		"BlockDeviceMapping.1.Ebs.VolumeSize":          "30",
		"BlockDeviceMapping.1.Ebs.VolumeType":          "gp3",
		"BlockDeviceMapping.1.Ebs.Encrypted":           "true",
		"BlockDeviceMapping.1.Ebs.DeleteOnTermination": "true",
		// Second mapping - Empty EBS volume
		"BlockDeviceMapping.2.DeviceName":     "/dev/sdh",
		"BlockDeviceMapping.2.Ebs.VolumeSize": "100",
		"BlockDeviceMapping.2.Ebs.VolumeType": "gp2",
		// Third mapping - Instance store
		"BlockDeviceMapping.3.DeviceName":  "/dev/sdc",
		"BlockDeviceMapping.3.VirtualName": "ephemeral0",
	}

	input := &ec2.RunInstancesInput{}
	err := QueryParamsToStruct(args, input)

	assert.NoError(t, err)
	assert.Len(t, input.BlockDeviceMappings, 3)

	// First mapping - EBS with snapshot
	bdm1 := input.BlockDeviceMappings[0]
	assert.Equal(t, "/dev/sda1", aws.StringValue(bdm1.DeviceName))
	assert.NotNil(t, bdm1.Ebs)
	assert.Equal(t, "snap-1234567890abcdef0", aws.StringValue(bdm1.Ebs.SnapshotId))
	assert.Equal(t, int64(30), aws.Int64Value(bdm1.Ebs.VolumeSize))
	assert.Equal(t, "gp3", aws.StringValue(bdm1.Ebs.VolumeType))
	assert.Equal(t, true, aws.BoolValue(bdm1.Ebs.Encrypted))
	assert.Equal(t, true, aws.BoolValue(bdm1.Ebs.DeleteOnTermination))

	// Second mapping - Empty EBS
	bdm2 := input.BlockDeviceMappings[1]
	assert.Equal(t, "/dev/sdh", aws.StringValue(bdm2.DeviceName))
	assert.NotNil(t, bdm2.Ebs)
	assert.Equal(t, int64(100), aws.Int64Value(bdm2.Ebs.VolumeSize))
	assert.Equal(t, "gp2", aws.StringValue(bdm2.Ebs.VolumeType))

	// Third mapping - Instance store
	bdm3 := input.BlockDeviceMappings[2]
	assert.Equal(t, "/dev/sdc", aws.StringValue(bdm3.DeviceName))
	assert.Equal(t, "ephemeral0", aws.StringValue(bdm3.VirtualName))
	assert.Nil(t, bdm3.Ebs)
}

func TestQueryParamsToStruct_CompleteRunInstancesExample(t *testing.T) {
	args := map[string]string{
		"Action":                              "RunInstances",
		"ImageId":                             "ami-0abcdef1234567890",
		"InstanceType":                        "t2.micro",
		"MinCount":                            "1",
		"MaxCount":                            "1",
		"SubnetId":                            "subnet-08fc749671b2d077c",
		"SecurityGroupId.1":                   "sg-0b0384b66d7d692f9",
		"KeyName":                             "MyKeyPair",
		"BlockDeviceMapping.1.DeviceName":     "/dev/sdh",
		"BlockDeviceMapping.1.Ebs.VolumeSize": "100",
		"TagSpecification.1.ResourceType":     "instance",
		"TagSpecification.1.Tag.1.Key":        "Name",
		"TagSpecification.1.Tag.1.Value":      "MyWebServer",
		"TagSpecification.1.Tag.2.Key":        "Environment",
		"TagSpecification.1.Tag.2.Value":      "Production",
	}

	input := &ec2.RunInstancesInput{}
	err := QueryParamsToStruct(args, input)

	assert.NoError(t, err)

	// Basic instance parameters
	assert.Equal(t, "ami-0abcdef1234567890", aws.StringValue(input.ImageId))
	assert.Equal(t, "t2.micro", aws.StringValue(input.InstanceType))
	assert.Equal(t, int64(1), aws.Int64Value(input.MinCount))
	assert.Equal(t, int64(1), aws.Int64Value(input.MaxCount))

	// Network parameters
	assert.Equal(t, "subnet-08fc749671b2d077c", aws.StringValue(input.SubnetId))
	assert.Len(t, input.SecurityGroupIds, 1)
	assert.Equal(t, "sg-0b0384b66d7d692f9", aws.StringValue(input.SecurityGroupIds[0]))
	assert.Equal(t, "MyKeyPair", aws.StringValue(input.KeyName))

	// Block device mappings
	assert.Len(t, input.BlockDeviceMappings, 1)
	bdm := input.BlockDeviceMappings[0]
	assert.Equal(t, "/dev/sdh", aws.StringValue(bdm.DeviceName))
	assert.NotNil(t, bdm.Ebs)
	assert.Equal(t, int64(100), aws.Int64Value(bdm.Ebs.VolumeSize))

	// Tag specifications
	assert.Len(t, input.TagSpecifications, 1)
	tagSpec := input.TagSpecifications[0]
	assert.Equal(t, "instance", aws.StringValue(tagSpec.ResourceType))
	assert.Len(t, tagSpec.Tags, 2)
	assert.Equal(t, "Name", aws.StringValue(tagSpec.Tags[0].Key))
	assert.Equal(t, "MyWebServer", aws.StringValue(tagSpec.Tags[0].Value))
	assert.Equal(t, "Environment", aws.StringValue(tagSpec.Tags[1].Key))
	assert.Equal(t, "Production", aws.StringValue(tagSpec.Tags[1].Value))
}

func TestQueryParamsToStruct_BlockDeviceMappingsWithIOPS(t *testing.T) {
	args := map[string]string{
		"Action":                              "RunInstances",
		"ImageId":                             "ami-0abcdef1234567890",
		"InstanceType":                        "t2.micro",
		"MinCount":                            "1",
		"MaxCount":                            "1",
		"BlockDeviceMapping.1.DeviceName":     "/dev/sda1",
		"BlockDeviceMapping.1.Ebs.VolumeSize": "100",
		"BlockDeviceMapping.1.Ebs.VolumeType": "io2",
		"BlockDeviceMapping.1.Ebs.Iops":       "5000",
		"BlockDeviceMapping.1.Ebs.Throughput": "500",
		"BlockDeviceMapping.1.Ebs.KmsKeyId":   "arn:aws:kms:us-east-1:123456789012:key/12345678-1234-1234-1234-123456789012",
	}

	input := &ec2.RunInstancesInput{}
	err := QueryParamsToStruct(args, input)

	assert.NoError(t, err)
	assert.Len(t, input.BlockDeviceMappings, 1)

	bdm := input.BlockDeviceMappings[0]
	assert.Equal(t, "/dev/sda1", aws.StringValue(bdm.DeviceName))
	assert.NotNil(t, bdm.Ebs)
	assert.Equal(t, int64(100), aws.Int64Value(bdm.Ebs.VolumeSize))
	assert.Equal(t, "io2", aws.StringValue(bdm.Ebs.VolumeType))
	assert.Equal(t, int64(5000), aws.Int64Value(bdm.Ebs.Iops))
	assert.Equal(t, int64(500), aws.Int64Value(bdm.Ebs.Throughput))
	assert.Equal(t, "arn:aws:kms:us-east-1:123456789012:key/12345678-1234-1234-1234-123456789012", aws.StringValue(bdm.Ebs.KmsKeyId))
}