package dataflowdebugsession

import (
	"encoding/json"
	"fmt"
)

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

type AmazonRdsForSqlServerLinkedServiceTypeProperties struct {
	AlwaysEncryptedSettings  *SqlAlwaysEncryptedProperties      `json:"alwaysEncryptedSettings,omitempty"`
	ApplicationIntent        *string                            `json:"applicationIntent,omitempty"`
	AuthenticationType       *AmazonRdsForSqlAuthenticationType `json:"authenticationType,omitempty"`
	CommandTimeout           *int64                             `json:"commandTimeout,omitempty"`
	ConnectRetryCount        *int64                             `json:"connectRetryCount,omitempty"`
	ConnectRetryInterval     *int64                             `json:"connectRetryInterval,omitempty"`
	ConnectTimeout           *int64                             `json:"connectTimeout,omitempty"`
	ConnectionString         *string                            `json:"connectionString,omitempty"`
	Database                 *string                            `json:"database,omitempty"`
	Encrypt                  *string                            `json:"encrypt,omitempty"`
	EncryptedCredential      *string                            `json:"encryptedCredential,omitempty"`
	FailoverPartner          *string                            `json:"failoverPartner,omitempty"`
	HostNameInCertificate    *string                            `json:"hostNameInCertificate,omitempty"`
	IntegratedSecurity       *bool                              `json:"integratedSecurity,omitempty"`
	LoadBalanceTimeout       *int64                             `json:"loadBalanceTimeout,omitempty"`
	MaxPoolSize              *int64                             `json:"maxPoolSize,omitempty"`
	MinPoolSize              *int64                             `json:"minPoolSize,omitempty"`
	MultiSubnetFailover      *bool                              `json:"multiSubnetFailover,omitempty"`
	MultipleActiveResultSets *bool                              `json:"multipleActiveResultSets,omitempty"`
	PacketSize               *int64                             `json:"packetSize,omitempty"`
	Password                 SecretBase                         `json:"password"`
	Pooling                  *bool                              `json:"pooling,omitempty"`
	Server                   *string                            `json:"server,omitempty"`
	TrustServerCertificate   *bool                              `json:"trustServerCertificate,omitempty"`
	UserName                 *string                            `json:"userName,omitempty"`
}

var _ json.Unmarshaler = &AmazonRdsForSqlServerLinkedServiceTypeProperties{}

func (s *AmazonRdsForSqlServerLinkedServiceTypeProperties) UnmarshalJSON(bytes []byte) error {
	var decoded struct {
		AlwaysEncryptedSettings  *SqlAlwaysEncryptedProperties      `json:"alwaysEncryptedSettings,omitempty"`
		ApplicationIntent        *string                            `json:"applicationIntent,omitempty"`
		AuthenticationType       *AmazonRdsForSqlAuthenticationType `json:"authenticationType,omitempty"`
		CommandTimeout           *int64                             `json:"commandTimeout,omitempty"`
		ConnectRetryCount        *int64                             `json:"connectRetryCount,omitempty"`
		ConnectRetryInterval     *int64                             `json:"connectRetryInterval,omitempty"`
		ConnectTimeout           *int64                             `json:"connectTimeout,omitempty"`
		ConnectionString         *string                            `json:"connectionString,omitempty"`
		Database                 *string                            `json:"database,omitempty"`
		Encrypt                  *string                            `json:"encrypt,omitempty"`
		EncryptedCredential      *string                            `json:"encryptedCredential,omitempty"`
		FailoverPartner          *string                            `json:"failoverPartner,omitempty"`
		HostNameInCertificate    *string                            `json:"hostNameInCertificate,omitempty"`
		IntegratedSecurity       *bool                              `json:"integratedSecurity,omitempty"`
		LoadBalanceTimeout       *int64                             `json:"loadBalanceTimeout,omitempty"`
		MaxPoolSize              *int64                             `json:"maxPoolSize,omitempty"`
		MinPoolSize              *int64                             `json:"minPoolSize,omitempty"`
		MultiSubnetFailover      *bool                              `json:"multiSubnetFailover,omitempty"`
		MultipleActiveResultSets *bool                              `json:"multipleActiveResultSets,omitempty"`
		PacketSize               *int64                             `json:"packetSize,omitempty"`
		Pooling                  *bool                              `json:"pooling,omitempty"`
		Server                   *string                            `json:"server,omitempty"`
		TrustServerCertificate   *bool                              `json:"trustServerCertificate,omitempty"`
		UserName                 *string                            `json:"userName,omitempty"`
	}
	if err := json.Unmarshal(bytes, &decoded); err != nil {
		return fmt.Errorf("unmarshaling: %+v", err)
	}

	s.AlwaysEncryptedSettings = decoded.AlwaysEncryptedSettings
	s.ApplicationIntent = decoded.ApplicationIntent
	s.AuthenticationType = decoded.AuthenticationType
	s.CommandTimeout = decoded.CommandTimeout
	s.ConnectRetryCount = decoded.ConnectRetryCount
	s.ConnectRetryInterval = decoded.ConnectRetryInterval
	s.ConnectTimeout = decoded.ConnectTimeout
	s.ConnectionString = decoded.ConnectionString
	s.Database = decoded.Database
	s.Encrypt = decoded.Encrypt
	s.EncryptedCredential = decoded.EncryptedCredential
	s.FailoverPartner = decoded.FailoverPartner
	s.HostNameInCertificate = decoded.HostNameInCertificate
	s.IntegratedSecurity = decoded.IntegratedSecurity
	s.LoadBalanceTimeout = decoded.LoadBalanceTimeout
	s.MaxPoolSize = decoded.MaxPoolSize
	s.MinPoolSize = decoded.MinPoolSize
	s.MultiSubnetFailover = decoded.MultiSubnetFailover
	s.MultipleActiveResultSets = decoded.MultipleActiveResultSets
	s.PacketSize = decoded.PacketSize
	s.Pooling = decoded.Pooling
	s.Server = decoded.Server
	s.TrustServerCertificate = decoded.TrustServerCertificate
	s.UserName = decoded.UserName

	var temp map[string]json.RawMessage
	if err := json.Unmarshal(bytes, &temp); err != nil {
		return fmt.Errorf("unmarshaling AmazonRdsForSqlServerLinkedServiceTypeProperties into map[string]json.RawMessage: %+v", err)
	}

	if v, ok := temp["password"]; ok {
		impl, err := UnmarshalSecretBaseImplementation(v)
		if err != nil {
			return fmt.Errorf("unmarshaling field 'Password' for 'AmazonRdsForSqlServerLinkedServiceTypeProperties': %+v", err)
		}
		s.Password = impl
	}

	return nil
}
