require "log4r"
require "digest"
require "vagrant/util/shell_quote"
require "vagrant/util/subprocess"

# Someone who's had some sleep recently is welcome to come fill in comments here...
module VagrantPlugins
  module VagrantVmwareDhcp
    class DhcpManagerWindows < DhcpManager

      protected

      def dhcpd_conf_location(network)
        # Locations from https://pubs.vmware.com/workstation-9/index.jsp?topic=%2Fcom.vmware.ws.using.doc%2FGUID-04D783E1-3AB9-4D98-9891-2C58215905CC.html

        if File.exist?('c:\ProgramData\VMware\vmnetdhcp.conf')
          location = 'C:\ProgramData\VMware\vmnetdhcp.conf'
        elsif File.exist?('C:\Documents and Settings\All Users\Application Data\VMware\vmnetdhcp.conf')
          location = 'C:\Documents and Settings\All Users\Application Data\VMware\vmnetdhcp.conf'              
        end

        @logger.debug("Using dhcpd.conf at #{location}")

        return location
      end

      def prune_configuration(network)
        conf_location = dhcpd_conf_location(network)

        mac = network[:mac]
        ip = network[:ip]

        before = File.open(conf_location).read
        @logger.debug("Before altering, dhcpd.conf content is #{before}")

        intermediate = before.gsub(/^# VAGRANT-BEGIN: #{mac}.*^# VAGRANT-END: #{mac}\s+/m, '')
        after = intermediate.gsub(/^# VAGRANT-BEGIN: #{ip}.*^# VAGRANT-END: #{ip}\s+/m, '')

        File.open(conf_location, "w") { |fd| fd.write(after) }

        after = File.open(conf_location).read
        @logger.debug("After altering, dhcpd.conf content is #{after}")
      end

      def write_configuration(network)
        conf_location = dhcpd_conf_location(network)

        section = get_dhcpd_section(network)

        @logger.debug("DHCPD template for interface #{network[:vnet]} is #{section}")

        before = File.open(conf_location).read
        @logger.debug("Before altering, dhcpd.conf content is #{before}")

        after = "#{before}\n\n#{section}\n"

        File.open(conf_location, "w") { |fd| fd.write(after) }

        after = File.open(conf_location).read
        @logger.debug("After, dhcpd.conf content is #{after}")
      end

      def reload_configuration
        # This is non-authoritative, but is the obvious solution and seems to work.

        stopCommand = [ "NET", "STOP", "VMware DHCP Service" ]
        startCommand = [ "NET", "START", "VMware DHCP Service" ]

        Vagrant::Util::Subprocess.execute(*stopCommand)
        r = Vagrant::Util::Subprocess.execute(*startCommand)

        if r.exit_code != 0
          @ui.error("VMNet dhcp start exited with code #{r.exit_code} and output:\n#{r.stdout.chomp}")
        end
      end
    end
  end
end