require 'chef/sugar'

class Chef
  module Sugar
    module IP
      extend self

      ## Returns the cloud network ip for the node,
      ## if no label is passed and more than one cloud
      ## network is attached it will pop the first off
      ## the array
      def cloud_network_ip(node, label=nil)
        networks = cloud_networks(node)
        return '' if networks.empty?

        net = nil
        if label.nil?
          net = networks[networks.flatten.first]
        else
          net = networks[label]
        end

        ## Return the first ip address in ips
        return net['ips'].first['ip']
      end

      def best_rackspace_ip_for(node, other)
        begin
          ## Fail early if either node has no cloud networks
          node_clnet = cloud_networks(node)
          fail if node_clnet.empty?

          other_clnet = cloud_networks(other)
          fail if other_clnet.empty?

          ip = nil
          node_clnet.each_key do |node_label|
            other_clnet.each_key do |other_label|
              if node_label == other_label
                ip = other_clnet[node_label]['ips'].first['ip']
              end
            end
          end

          # Couldn't locate a matching cloud network, falling through
          fail if ip.nil?

          return ip
        rescue
          best_ip_for(node, other)
        end
      end

      def cloud_networks(node)
        networks = {}
        return networks unless node.has_key? 'rackspace'
        if node['rackspace'].has_key? 'private_networks'
          private_networks = node['rackspace']['private_networks']

          unless private_networks.nil? || private_networks.empty?
            private_networks.each do |net|
              networks[net['label']] = net
            end
          end
        end

        return networks
      end
    end

    module DSL
      def cloud_network_ip(label=nil)
        Chef::Sugar::IP.cloud_network_ip(node, label)
      end

      def cloud_networks
        Chef::Sugar::IP.cloud_networks(node)
      end

      def best_rackspace_ip_for(other)
        Chef::Sugar::IP.best_rackspace_ip_for(node, other)
      end
    end
  end
end