Class: Vpnconfig::Api

Inherits:
Object
  • Object
show all
Defined in:
lib/api.rb

Instance Method Summary collapse

Instance Method Details

#check_task_until_completed(task_href, conn) ⇒ Object 



6
7
8
9
10
11
12
13
14
 
# File 'lib/api.rb', line 6

def check_task_until_completed(task_href, conn)
  task = get_xml_object(task_href, conn['auth_token'], 'Task')
  if task['status'] == 'running'
    sleep 1
    check_task_until_completed(task_href, conn)
  else
    task
  end
end

#get_all_dcs(org_xml) ⇒ Object 



72
73
74
 
# File 'lib/api.rb', line 72

def get_all_dcs(org_xml)
  org_xml.search('Link[@type="application/vnd.vmware.vcloud.vdc+xml"]').to_a
end

#get_completed_task(task_href, conn) ⇒ Object 



5
6
7
8
9
10
11
12
13
14
15
16
17
18
 
# File 'lib/api.rb', line 5

def get_completed_task(task_href, conn)
  def check_task_until_completed(task_href, conn)
    task = get_xml_object(task_href, conn['auth_token'], 'Task')
    if task['status'] == 'running'
      sleep 1
      check_task_until_completed(task_href, conn)
    else
      task
    end
  end

  completed_task = check_task_until_completed(task_href, conn)
  completed_task
end

#get_dc_details(conn, user_dc) ⇒ Object 



56
57
58
59
60
61
62
63
64
65
66
67
68
69
 
# File 'lib/api.rb', line 56

def get_dc_details(conn, user_dc)
  puts conn if @verbose
  org_xml = get_xml_object(conn["org_url"], conn["auth_token"], 'Org')
  all_dcs = get_all_dcs(org_xml)
  matching_dcs = all_dcs.select { |dc| /^#{user_dc}/.match(dc['name']) || user_dc == dc['name'] }
  if matching_dcs.length < 1
    all_dc_names = all_dcs.map { |dc| dc['name'] }.join(', ')
    Kernel.abort('No DCs match the search string: '+user_dc+'. All available DCs are: '+all_dc_names+'. Please use one of these')
  elsif matching_dcs.length > 1
    all_matching_names = matching_dcs.map { |dc| dc['name'] }.join(', ')
    Kernel.abort('More than 1 DC matches the search string. Please be more specific. Matching DCs are: '+all_matching_names)
  end
  matching_dcs[0]
end

#get_edgegw_configs(edgegw_details, dc, conn) ⇒ Object 



36
37
38
39
 
# File 'lib/api.rb', line 36

def get_edgegw_configs(edgegw_details, dc, conn)
  href = edgegw_details.at('EdgeGatewayRecord')['href']
  get_xml_object(href, conn["auth_token"], 'EdgeGateway').at('Configuration').at('GatewayIpsecVpnService')
end

#get_edgegw_details(overview, dc, conn) ⇒ Object 



41
42
43
44
 
# File 'lib/api.rb', line 41

def get_edgegw_details(overview, dc, conn)
  href = overview['href']
  get_xml_object(href, conn['auth_token'], 'QueryResultRecords')
end

#get_edgegw_overviews(dc, conn) ⇒ Object 



46
47
48
49
50
51
52
53
 
# File 'lib/api.rb', line 46

def get_edgegw_overviews(dc, conn)
  dc_xml = get_xml_object(dc['href'], conn['auth_token'], 'Vdc')
  overviews = dc_xml.search('Link[@rel="edgeGateways"]').to_a
  if overviews.length > 1
    Kernel.abort('More than 1 edge gateway found, aborting operation')
  end
  overviews[0]
end

#get_task_href(response_body) ⇒ Object 



127
128
129
130
131
 
# File 'lib/api.rb', line 127

def get_task_href(response_body)
  xml_doc  = Nokogiri::XML(response_body)
  task = xml_doc.at_css("Task")["href"]
  task
end

#get_xml_object(href, token, object_name) ⇒ Object 



76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
 
# File 'lib/api.rb', line 76

def get_xml_object(href, token, object_name)
  begin
    obj = nil
    uri = URI.parse(href)
    http = Net::HTTP.new(uri.host, uri.port)
    http.use_ssl = true
    http.verify_mode = OpenSSL::SSL::VERIFY_NONE
    request = Net::HTTP::Get.new(uri.request_uri, {'Accept' => 'application/*+xml;version=5.5', 'x-vcloud-authorization' => token})

    response = http.request(request)
    if response.code.to_i == 200
      doc = Nokogiri::XML(response.body)
      obj = doc.search(object_name).to_a
      if obj.count == 1
        return obj[0]
      else
        raise "The number of #{object_name} objects returned is not equal to 1! #{obj.inspect} "
      end
    else
      raise "Error fetch the object #{object_name}, URL: #{href}, response code: #{response.code}"
    end
  rescue Exception => e
    puts "Error fetching the object #{object_name}, URL: #{href}"
    puts e
  end
end

#post_to_api(href, conn, config) ⇒ Object 



31
32
33
34
 
# File 'lib/api.rb', line 31

def post_to_api(href, conn, config)
  task = push_config(href, conn['auth_token'], config)
  task
end

#push_config(href, token, config) ⇒ Object 



103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
 
# File 'lib/api.rb', line 103

def push_config(href,token,config)
  href = href + "/action/configureServices"

  uri = URI.parse(href)
  http = Net::HTTP.new(uri.host, uri.port)
  http.use_ssl = true
  http.verify_mode = OpenSSL::SSL::VERIFY_NONE
  request = Net::HTTP::Post.new(uri.request_uri, { 'Accept' => 'application/*+xml;version=5.5', 'x-vcloud-authorization' => token })
  request.body = config.to_s
  request.content_type = 'application/vnd.vmware.admin.edgeGatewayServiceConfiguration+xml'


  response = http.request(request)
  if response.code.to_i == 202
    puts "Post successful" if @verbose
  else
    puts "Unexpected exit code"
    puts response.code
    puts response.body
  end
  task = get_task_href(response.body)
  task
end

#read_edgegw_config(dc, conn) ⇒ Object 



20
21
22
23
24
25
26
27
28
29
 
# File 'lib/api.rb', line 20

def read_edgegw_config(dc, conn)
  puts 'Getting details of dcs' if @verbose
  edgegw_overviews = get_edgegw_overviews(dc, conn)
  puts 'Getting details of edge gateways' if @verbose
  edgegw_details = get_edgegw_details(edgegw_overviews, dc, conn)
  #Get edge gateway config
  puts 'Getting config of edge gateways' if @verbose
  edgegw_configs = get_edgegw_configs(edgegw_details, dc, conn)
  return edgegw_configs.to_s, edgegw_details.at('EdgeGatewayRecord')['href']
end