require 'fog/azurerm'
require 'yaml'

########################################################################################################################
######################                   Services object required by all actions                  ######################
######################                              Keep it Uncommented!                          ######################
########################################################################################################################

azure_credentials = YAML.load_file('credentials/azure.yml')

resource = Fog::Resources::AzureRM.new(
  tenant_id: azure_credentials['tenant_id'],
  client_id: azure_credentials['client_id'],
  client_secret: azure_credentials['client_secret'],
  subscription_id: azure_credentials['subscription_id']
)

dns = Fog::DNS.new(
  provider: 'AzureRM',
  tenant_id: azure_credentials['tenant_id'],
  client_id: azure_credentials['client_id'],
  client_secret: azure_credentials['client_secret'],
  subscription_id: azure_credentials['subscription_id']
)

########################################################################################################################
######################                                 Prerequisites                              ######################
########################################################################################################################

resource.resource_groups.create(
  name: 'TestRG-RS',
  location: 'eastus'
)

dns.zones.create(
  name: 'test-zone.com',
  location: 'global',
  resource_group: 'TestRG-RS'
)

########################################################################################################################
######################                      Create CNAME Type Record Set in a Zone                 ######################
########################################################################################################################

dns.record_sets.create(
  name: 'TestRS1',
  resource_group: 'TestRG-RS',
  zone_name: 'test-zone.com',
  records: ['test.fog.com'],
  type: 'CNAME',
  ttl: 60
)

########################################################################################################################
######################                        Create A Type Record Set in a Zone                   ######################
########################################################################################################################

dns.record_sets.create(
  name: 'TestRS2',
  resource_group: 'TestRG-RS',
  zone_name: 'test-zone.com',
  records: %w(1.2.3.4 1.2.3.3),
  type: 'A',
  ttl: 60
)

########################################################################################################################
######################                Get And Destroy CNAME Type Record Set in a Zone             ######################
########################################################################################################################

record_set = dns.record_sets.get('TestRG-RS', 'TestRS1', 'test-zone.com', 'CNAME')
record_set.destroy

########################################################################################################################
######################                  Get And Destroy A Type Record Set in a Zone               ######################
########################################################################################################################

record_set = dns.record_sets.get('TestRG-RS', 'TestRS2', 'test-zone.com', 'A')

########################################################################################################################
######################                               Update a Record Set                          ######################
########################################################################################################################

record_set.update_ttl(80)

########################################################################################################################
######################                               Add an A-type Record                         ######################
########################################################################################################################

record_set.add_a_type_record('1.2.3.8')

########################################################################################################################
######################                            Remove an A-type Record                         ######################
########################################################################################################################

record_set.remove_a_type_record('1.2.3.8')

########################################################################################################################
######################                                   CleanUp                                  ######################
########################################################################################################################

record_set.destroy
zone = dns.zones.get('TestRG-RS', 'test-zone.com')
zone.destroy

rg = resource.resource_groups.get('TestRG-RS')
rg.destroy