package databasevulnerabilityassessmentrulebaselines

import (
	"fmt"
	"strings"

	"github.com/hashicorp/go-azure-helpers/resourcemanager/recaser"
	"github.com/hashicorp/go-azure-helpers/resourcemanager/resourceids"
)

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License. See NOTICE.txt in the project root for license information.

func init() {
	recaser.RegisterResourceId(&BaselineId{})
}

var _ resourceids.ResourceId = &BaselineId{}

// BaselineId is a struct representing the Resource ID for a Baseline
type BaselineId struct {
	SubscriptionId    string
	ResourceGroupName string
	ServerName        string
	DatabaseName      string
	RuleId            string
	BaselineName      VulnerabilityAssessmentPolicyBaselineName
}

// NewBaselineID returns a new BaselineId struct
func NewBaselineID(subscriptionId string, resourceGroupName string, serverName string, databaseName string, ruleId string, baselineName VulnerabilityAssessmentPolicyBaselineName) BaselineId {
	return BaselineId{
		SubscriptionId:    subscriptionId,
		ResourceGroupName: resourceGroupName,
		ServerName:        serverName,
		DatabaseName:      databaseName,
		RuleId:            ruleId,
		BaselineName:      baselineName,
	}
}

// ParseBaselineID parses 'input' into a BaselineId
func ParseBaselineID(input string) (*BaselineId, error) {
	parser := resourceids.NewParserFromResourceIdType(&BaselineId{})
	parsed, err := parser.Parse(input, false)
	if err != nil {
		return nil, fmt.Errorf("parsing %q: %+v", input, err)
	}

	id := BaselineId{}
	if err = id.FromParseResult(*parsed); err != nil {
		return nil, err
	}

	return &id, nil
}

// ParseBaselineIDInsensitively parses 'input' case-insensitively into a BaselineId
// note: this method should only be used for API response data and not user input
func ParseBaselineIDInsensitively(input string) (*BaselineId, error) {
	parser := resourceids.NewParserFromResourceIdType(&BaselineId{})
	parsed, err := parser.Parse(input, true)
	if err != nil {
		return nil, fmt.Errorf("parsing %q: %+v", input, err)
	}

	id := BaselineId{}
	if err = id.FromParseResult(*parsed); err != nil {
		return nil, err
	}

	return &id, nil
}

func (id *BaselineId) FromParseResult(input resourceids.ParseResult) error {
	var ok bool

	if id.SubscriptionId, ok = input.Parsed["subscriptionId"]; !ok {
		return resourceids.NewSegmentNotSpecifiedError(id, "subscriptionId", input)
	}

	if id.ResourceGroupName, ok = input.Parsed["resourceGroupName"]; !ok {
		return resourceids.NewSegmentNotSpecifiedError(id, "resourceGroupName", input)
	}

	if id.ServerName, ok = input.Parsed["serverName"]; !ok {
		return resourceids.NewSegmentNotSpecifiedError(id, "serverName", input)
	}

	if id.DatabaseName, ok = input.Parsed["databaseName"]; !ok {
		return resourceids.NewSegmentNotSpecifiedError(id, "databaseName", input)
	}

	if id.RuleId, ok = input.Parsed["ruleId"]; !ok {
		return resourceids.NewSegmentNotSpecifiedError(id, "ruleId", input)
	}

	if v, ok := input.Parsed["baselineName"]; true {
		if !ok {
			return resourceids.NewSegmentNotSpecifiedError(id, "baselineName", input)
		}

		baselineName, err := parseVulnerabilityAssessmentPolicyBaselineName(v)
		if err != nil {
			return fmt.Errorf("parsing %q: %+v", v, err)
		}
		id.BaselineName = *baselineName
	}

	return nil
}

// ValidateBaselineID checks that 'input' can be parsed as a Baseline ID
func ValidateBaselineID(input interface{}, key string) (warnings []string, errors []error) {
	v, ok := input.(string)
	if !ok {
		errors = append(errors, fmt.Errorf("expected %q to be a string", key))
		return
	}

	if _, err := ParseBaselineID(v); err != nil {
		errors = append(errors, err)
	}

	return
}

// ID returns the formatted Baseline ID
func (id BaselineId) ID() string {
	fmtString := "/subscriptions/%s/resourceGroups/%s/providers/Microsoft.Sql/servers/%s/databases/%s/vulnerabilityAssessments/default/rules/%s/baselines/%s"
	return fmt.Sprintf(fmtString, id.SubscriptionId, id.ResourceGroupName, id.ServerName, id.DatabaseName, id.RuleId, string(id.BaselineName))
}

// Segments returns a slice of Resource ID Segments which comprise this Baseline ID
func (id BaselineId) Segments() []resourceids.Segment {
	return []resourceids.Segment{
		resourceids.StaticSegment("staticSubscriptions", "subscriptions", "subscriptions"),
		resourceids.SubscriptionIdSegment("subscriptionId", "12345678-1234-9876-4563-123456789012"),
		resourceids.StaticSegment("staticResourceGroups", "resourceGroups", "resourceGroups"),
		resourceids.ResourceGroupSegment("resourceGroupName", "example-resource-group"),
		resourceids.StaticSegment("staticProviders", "providers", "providers"),
		resourceids.ResourceProviderSegment("staticMicrosoftSql", "Microsoft.Sql", "Microsoft.Sql"),
		resourceids.StaticSegment("staticServers", "servers", "servers"),
		resourceids.UserSpecifiedSegment("serverName", "serverName"),
		resourceids.StaticSegment("staticDatabases", "databases", "databases"),
		resourceids.UserSpecifiedSegment("databaseName", "databaseName"),
		resourceids.StaticSegment("staticVulnerabilityAssessments", "vulnerabilityAssessments", "vulnerabilityAssessments"),
		resourceids.StaticSegment("vulnerabilityAssessmentName", "default", "default"),
		resourceids.StaticSegment("staticRules", "rules", "rules"),
		resourceids.UserSpecifiedSegment("ruleId", "ruleId"),
		resourceids.StaticSegment("staticBaselines", "baselines", "baselines"),
		resourceids.ConstantSegment("baselineName", PossibleValuesForVulnerabilityAssessmentPolicyBaselineName(), "default"),
	}
}

// String returns a human-readable description of this Baseline ID
func (id BaselineId) String() string {
	components := []string{
		fmt.Sprintf("Subscription: %q", id.SubscriptionId),
		fmt.Sprintf("Resource Group Name: %q", id.ResourceGroupName),
		fmt.Sprintf("Server Name: %q", id.ServerName),
		fmt.Sprintf("Database Name: %q", id.DatabaseName),
		fmt.Sprintf("Rule: %q", id.RuleId),
		fmt.Sprintf("Baseline Name: %q", string(id.BaselineName)),
	}
	return fmt.Sprintf("Baseline (%s)", strings.Join(components, "\n"))
}
