# --------------------------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for license information.
#
# Code generated by aaz-dev-tools
# --------------------------------------------------------------------------------------------

# pylint: skip-file
# flake8: noqa

from azure.cli.core.aaz import *
from azure.cli.core.azclierror import MutuallyExclusiveArgumentError


@register_command(
    "security va sql baseline add",
    is_preview=True,
)
class Add(AAZCommand):
    """Set a list of baseline rules. Will overwrite any previously existing results (for all rules).

    :example: Set baseline for all rules on an Azure SQL database using the latest scan results.
        az security va sql baseline add --resource-id /subscriptions/{sub}/resourceGroups/{rg}/providers/Microsoft.Sql/servers/{server}/databases/{db} --latest-scan true

    :example: Set baseline for multiple rules with explicit results on a SQL DB hosted on an Azure VM.
        az security va sql baseline add --resource-id /subscriptions/{sub}/resourceGroups/{rg}/providers/Microsoft.Compute/virtualMachines/{vm} --database-name MyDb --results '{"VA1234":[["col1","col2"],["col3","col4"]],"VA5678":[["user1","SELECT"]]}'

    :example: Set baseline at the server level using latest scan results (system database 'master').
        az security va sql baseline add --resource-id /subscriptions/{sub}/resourceGroups/{rg}/providers/Microsoft.Sql/servers/{server} --database-name master --latest-scan true
    """

    _aaz_info = {
        "version": "2026-04-01-preview",
        "resources": [
            ["mgmt-plane", "/{resourceid}/providers/microsoft.security/sqlvulnerabilityassessments/default/baselinerules", "2026-04-01-preview"],
        ]
    }

    def _handler(self, command_args):
        super()._handler(command_args)
        self._execute_operations()
        return self._output()

    _args_schema = None

    @classmethod
    def _build_arguments_schema(cls, *args, **kwargs):
        if cls._args_schema is not None:
            return cls._args_schema
        cls._args_schema = super()._build_arguments_schema(*args, **kwargs)

        # define Arg Group ""

        _args_schema = cls._args_schema
        _args_schema.resource_id = AAZStrArg(
            options=["--resource-id"],
            help="The fully qualified Azure Resource manager identifier of the resource.",
            required=True,
        )
        _args_schema.database_name = AAZStrArg(
            options=["--database-name"],
            help="The name of the database to assess. Required when the API is called on the parent resource (e.g., server level) rather than on a specific database resource, since the database name is not part of the resource URI. This is the only way to assess system databases (e.g., master), which cannot be referenced directly in the resource URI.",
        )

        # define Arg Group "Body"

        _args_schema = cls._args_schema
        _args_schema.latest_scan = AAZBoolArg(
            options=["--latest-scan"],
            arg_group="Body",
            help="Take results from latest scan.",
        )
        _args_schema.results = AAZDictArg(
            options=["--results"],
            arg_group="Body",
            help="Expected results to be inserted into the baseline. Leave this field empty if latestScan == true.",
        )

        results = cls._args_schema.results
        results.Element = AAZListArg()

        _element = cls._args_schema.results.Element
        _element.Element = AAZListArg()

        _element = cls._args_schema.results.Element.Element
        _element.Element = AAZStrArg()
        return cls._args_schema

    def _execute_operations(self):
        self.pre_operations()
        self.SqlVulnerabilityAssessmentBaselineRulesAdd(ctx=self.ctx)()
        self.post_operations()

    @register_callback
    def pre_operations(self):
        args = self.ctx.args
        if has_value(args.latest_scan) and bool(args.latest_scan) and has_value(args.results):
            raise MutuallyExclusiveArgumentError(
                "--latest-scan and --results are mutually exclusive. "
                "Use --latest-scan to populate the baseline from the most recent scan, "
                "or use --results to provide explicit expected results."
            )

    @register_callback
    def post_operations(self):
        pass

    def _output(self, *args, **kwargs):
        result = self.deserialize_output(self.ctx.vars.instance, client_flatten=True)
        return result

    class SqlVulnerabilityAssessmentBaselineRulesAdd(AAZHttpOperation):
        CLIENT_TYPE = "MgmtClient"

        def __call__(self, *args, **kwargs):
            request = self.make_request()
            session = self.client.send_request(request=request, stream=False, **kwargs)
            if session.http_response.status_code in [200]:
                return self.on_200(session)

            return self.on_error(session.http_response)

        @property
        def url(self):
            return self.client.format_url(
                "/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/baselineRules",
                **self.url_parameters
            )

        @property
        def method(self):
            return "POST"

        @property
        def error_format(self):
            return "MgmtErrorFormat"

        @property
        def url_parameters(self):
            parameters = {
                **self.serialize_url_param(
                    "resourceId", self.ctx.args.resource_id,
                    required=True,
                ),
            }
            return parameters

        @property
        def query_parameters(self):
            parameters = {
                **self.serialize_query_param(
                    "databaseName", self.ctx.args.database_name,
                ),
                **self.serialize_query_param(
                    "api-version", "2026-04-01-preview",
                    required=True,
                ),
            }
            return parameters

        @property
        def header_parameters(self):
            parameters = {
                **self.serialize_header_param(
                    "Content-Type", "application/json",
                ),
                **self.serialize_header_param(
                    "Accept", "application/json",
                ),
            }
            return parameters

        @property
        def content(self):
            _content_value, _builder = self.new_content_builder(
                self.ctx.args,
                typ=AAZObjectType,
                typ_kwargs={"flags": {"client_flatten": True}}
            )
            _builder.set_prop("latestScan", AAZBoolType, ".latest_scan")
            _builder.set_prop("results", AAZDictType, ".results")

            results = _builder.get(".results")
            if results is not None:
                results.set_elements(AAZListType, ".")

            _elements = _builder.get(".results{}")
            if _elements is not None:
                _elements.set_elements(AAZListType, ".")

            _elements = _builder.get(".results{}[]")
            if _elements is not None:
                _elements.set_elements(AAZStrType, ".")

            return self.serialize_content(_content_value)

        def on_200(self, session):
            data = self.deserialize_http_content(session)
            self.ctx.set_var(
                "instance",
                data,
                schema_builder=self._build_schema_on_200
            )

        _schema_on_200 = None

        @classmethod
        def _build_schema_on_200(cls):
            if cls._schema_on_200 is not None:
                return cls._schema_on_200

            cls._schema_on_200 = AAZObjectType()

            _schema_on_200 = cls._schema_on_200
            _schema_on_200.next_link = AAZStrType(
                serialized_name="nextLink",
            )
            _schema_on_200.value = AAZListType()

            value = cls._schema_on_200.value
            value.Element = AAZObjectType()

            _element = cls._schema_on_200.value.Element
            _element.id = AAZStrType(
                flags={"read_only": True},
            )
            _element.name = AAZStrType(
                flags={"read_only": True},
            )
            _element.properties = AAZObjectType()
            _element.system_data = AAZObjectType(
                serialized_name="systemData",
                flags={"read_only": True},
            )
            _element.type = AAZStrType(
                flags={"read_only": True},
            )

            properties = cls._schema_on_200.value.Element.properties
            properties.latest_scan = AAZBoolType(
                serialized_name="latestScan",
            )
            properties.results = AAZListType()

            results = cls._schema_on_200.value.Element.properties.results
            results.Element = AAZListType()

            _element = cls._schema_on_200.value.Element.properties.results.Element
            _element.Element = AAZStrType()

            system_data = cls._schema_on_200.value.Element.system_data
            system_data.created_at = AAZStrType(
                serialized_name="createdAt",
            )
            system_data.created_by = AAZStrType(
                serialized_name="createdBy",
            )
            system_data.created_by_type = AAZStrType(
                serialized_name="createdByType",
            )
            system_data.last_modified_at = AAZStrType(
                serialized_name="lastModifiedAt",
            )
            system_data.last_modified_by = AAZStrType(
                serialized_name="lastModifiedBy",
            )
            system_data.last_modified_by_type = AAZStrType(
                serialized_name="lastModifiedByType",
            )

            return cls._schema_on_200


class _AddHelper:
    """Helper class for Add"""


__all__ = ["Add"]
