# encoding: UTF-8
# Copyright (c) 2018 António Meireles. All Rights Reserved.

require 'tempfile'
require 'ipaddr'
require 'log4r'

# Borrowed from http://stackoverflow.com/questions/1825928/netmask-to-cidr-in-ruby
IPAddr.class_eval do
  def to_cidr
    to_i.to_s(2).count('1')
  end
end

STATIC_NETWORK = <<EOF.freeze
#VAGRANT-BEGIN
# The contents below were automatically generated by Vagrant. Do not modify.
[Match]
Name=%s

[Network]
Address=%s
%s
#VAGRANT-END
EOF

DHCP_NETWORK = <<EOF.freeze
#VAGRANT-BEGIN
# The contents below were automatically generated by Vagrant. Do not modify.
[Match]
Name=%s

[Network]
DHCP=yes
#VAGRANT-END
EOF

module VagrantPlugins
  module GuestClearLinux
    module Cap
      class ConfigureNetworks
        include Vagrant::Util
        extend Vagrant::Util::GuestInspection::Linux
        NETWORKD_DIRECTORY = "/etc/systemd/network".freeze
        @@logger = Log4r::Logger.new('vagrant::guest::clearlinux::configure_networks')

        def self.configure_networks(machine, networks)
          comm = machine.communicate
          interfaces = machine.guest.capability(:network_interfaces)
          networks.each do |network|
            interface = network[:interface].to_i
            device = interfaces[interface]
            if device.nil?
              @@logger.warn("Could not find match rule for network #{network.inspect}")
              next
            end
            unit_name = format('50-vagrant-%s.network', device)

            if network[:type] == :static
              cidr = IPAddr.new(network[:netmask]).to_cidr
              address = "#{network[:ip]}/#{cidr}"
              gateway = ""
              gateway = "Gateway=#{network[:gateway]}" if network[:gateway]
              unit_file = format(STATIC_NETWORK, device, address, gateway)
            elsif network[:type] == :dhcp
              unit_file = format(DHCP_NETWORK, device)
            end

            temp = Tempfile.new('vagrant')
            temp.binmode
            temp.write(unit_file)
            temp.close

            comm.upload(temp.path, "/tmp/#{unit_name}")
            comm.sudo([ "mkdir -p #{NETWORKD_DIRECTORY}",
              "rm -f #{NETWORKD_DIRECTORY}/#{unit_name}",
              "mv /tmp/#{unit_name} #{NETWORKD_DIRECTORY}",
              "chown root:root #{NETWORKD_DIRECTORY}/#{unit_name}",
              "chmod a+r #{NETWORKD_DIRECTORY}/#{unit_name}",
              "systemctl restart systemd-networkd"].join("\n"))
            comm.wait_for_ready(30)
          end
        end
      end
    end
  end
end