module DTK
  module NodeAgent
    class Installer
      require 'facter'

      # read configuration
      CONFIG = eval(File.open(File.expand_path('../config/install.config', File.dirname(__FILE__))) {|f| f.read })

      # get OS facts
      @osfamily     = Facter.value('osfamily').downcase
      @osname       = Facter.value('operatingsystem')
      @osmajrelease = Facter.value('operatingsystemmajrelease')
      @osarch       = Facter.value('architecture')
      @distcodename = Facter.value('lsbdistcodename')

      def self.run(argv)
        require 'optparse'
        require 'fileutils'
        require 'dtk-node-agent/version'

        @@options = parse(argv)

        unless Process.uid == 0
          puts "dtk-node-agent must be started with root/sudo privileges."
          exit(1)
        end

        if @osfamily == 'debian'
              # set up apt and install packages
              shell "apt-get update --fix-missing"
              shell "apt-get install -y build-essential wget curl git"
              # install upgrades
              Array(CONFIG[:upgrades][:debian]).each do |package|
                shell "apt-get install -y #{package}"
              end
              shell "wget http://apt.puppetlabs.com/puppetlabs-release-#{@distcodename}.deb"
              puts "Installing Puppet Labs repository..."
              shell "dpkg -i puppetlabs-release-#{@distcodename}.deb"
              puts "Installing Puppet Labs repository..."
              shell "dpkg -i puppetlabs-release-#{@distcodename}.deb"
              # install logstash forwarder
              logstash_forwarder_url = CONFIG[:logstash_forwarder_deb64]
              logstash_forwarder_package = logstash_forwarder_url.rpartition('/').last
              shell "wget #{logstash_forwarder_url}"
              puts "Installing logstash-forwarder"
              shell "dpkg -i #{logstash_forwarder_package}"
              shell "apt-get update"
              shell "rm puppetlabs-release-#{@distcodename}.deb #{logstash_forwarder_package}"
              # install mcollective
              puts "Installing MCollective..."
              shell "apt-get -y install mcollective"
              # pin down the puppetlabs apt repo
              FileUtils.cp("#{base_dir}/src/etc/apt/preferences.d/puppetlabs", "/etc/apt/preferences.d/puppetlabs")
            elsif @osfamily == 'redhat'
              shell "yum -y install yum-utils wget curl"
              # install upgrades
              Array(CONFIG[:upgrades][:redhat]).each do |package|
                shell "yum -y install #{package}"
                shell "yum -y update #{package}"
              end
              case @osmajrelease
              when "5"
                shell "rpm -ivh #{CONFIG[:puppetlabs_el5_rpm_repo]}"
                @osarch == 'X86_64' ? (shell "rpm -ivh #{CONFIG[:rpm_forge_el5_X86_64_repo]}") : (shell "rpm -ivh #{CONFIG[:rpm_forge_el5_i686_repo]}")
              # 20xx is the major release naming pattern of Amazon Linux
              when "6", "n/a", "2015", "2014"
                shell "rpm -ivh #{CONFIG[:puppetlabs_el6_rpm_repo]}"
                @osarch == 'X86_64' ? (shell "rpm -ivh #{CONFIG[:rpm_forge_el6_X86_64_repo]}") : (shell "rpm -ivh #{CONFIG[:rpm_forge_el6_i686_repo]}")
                shell "yum-config-manager --disable rpmforge-release"
                shell "yum-config-manager --enable rpmforge-extras"
              when "7"
                shell "rpm -ivh #{CONFIG[:puppetlabs_el7_rpm_repo]}"
              else
                puts "#{@osname} #{@osmajrelease} is not supported. Exiting now..."
                exit(1)
              end
              puts "Installing MCollective..."
              shell "yum -y install mcollective"
              shell "yum -y install git"
              # install ec2-run-user-data init script
              # but only if the machine is running on AWS
              if `curl -m 5 -sI http://169.254.169.254/latest/meta-data/`.include? '200 OK'
                FileUtils.cp("#{base_dir}/src/etc/init.d/ec2-run-user-data", "/etc/init.d/ec2-run-user-data") unless File.exist?("/etc/init.d/ec2-run-user-data")
                set_init("ec2-run-user-data")
              end
              # install logstash-forwarder
              puts "Installing logstash-forwarder"
              shell "rpm -ivh #{CONFIG[:logstash_forwarder_rpm64]}"
              FileUtils.cp("#{base_dir}/src/etc/init.d/logstash-forwarder.rpm.init", "/etc/init.d/logstash-forwarder")
              set_init("logstash-forwarder")
            else
              puts "Unsuported OS for automatic agent installation. Exiting now..."
              exit(1)
            end

            puts "Installing additions for MCollective and Puppet..."
            install_additions

          end       


          private
          
          def self.parse(argv)
            options = {}
            parser = OptionParser.new do |opts|
              opts.banner = <<-BANNER
              usage:
              
              dtk-node-agent [-p|--puppet-version] [-v|--version]
              BANNER
              opts.on("-d",
                "--debug",
                "enable debug mode")  { |v| options[:debug] = true }      
              opts.on_tail("-v",
                "--version",
                "Print the version and exit.") do
                puts ::DtkNodeAgent::VERSION
                exit(0)
              end
              opts.on_tail("-h",
                "--help",
                "Print this help message.") do
                puts parser
                exit(0)
              end
            end

            parser.parse!(argv)

            options
            
          rescue OptionParser::InvalidOption => e
            $stderr.puts e.message
            exit(12)
          end

          def self.shell(cmd)
            puts "running: #{cmd}" if @@options[:debug]
            output = `#{cmd}`
            puts output if @@options[:debug]
            if $?.exitstatus != 0
              puts "Executing command \`#{cmd}\` failed"
              puts "Command output:"
              puts output
            end
          end

          def self.install_additions
            # create puppet group
            shell "groupadd puppet" unless `grep puppet /etc/group`.include? "puppet"
            # create necessary dirs
            [   '/var/log/puppet/',
              '/var/lib/puppet/lib/puppet/indirector',
              '/etc/puppet/modules',
              '/usr/share/mcollective/plugins/mcollective'
              ].map! { |p| FileUtils.mkdir_p(p) unless File.directory?(p) }
            # copy puppet libs
            FileUtils.cp_r(Dir.glob("#{base_dir}/puppet_additions/puppet_lib_base/puppet/indirector/*"), "/var/lib/puppet/lib/puppet/indirector/")
            # copy r8 puppet module
            FileUtils.cp_r(Dir.glob("#{base_dir}/puppet_additions/modules/r8"), "/etc/puppet/modules")
            # copy mcollective plugins
            FileUtils.cp_r(Dir.glob("/usr/libexec/mcollective/mcollective/*"), "/usr/share/mcollective/plugins/mcollective") if File.directory?("/usr/libexec/mcollective/")        
            mco_add_dir = "#{base_dir}/mcollective_additions"
            mco_plugin_dir = "#{mco_add_dir}/plugins/v#{CONFIG[:mcollective_version]}"
            FileUtils.cp_r(Dir.glob("#{mco_plugin_dir}/*"), "/usr/share/mcollective/plugins/mcollective")
            # copy mcollective config
            FileUtils.cp_r("#{mco_add_dir}/server.cfg", "/etc/mcollective", :remove_destination => true)
            # copy compatible mcollective init script
            FileUtils.cp_r("#{mco_add_dir}/#{@osfamily}.mcollective.init", "/etc/init.d/mcollective", :remove_destination => true)
            FileUtils.cp_r("#{mco_add_dir}/#{@osfamily}.mcollective.service", "/usr/lib/systemd/system/mcollective.service", :remove_destination => true) if File.exist?("/usr/lib/systemd/system/mcollective.service")
            set_init("mcollective")
          end

          def self.base_dir
            File.expand_path('../..', File.dirname(__FILE__))
          end

          def self.set_init(script)
            shell "chmod +x /etc/init.d/#{script}"
            if @osfamily == 'debian'
              shell "update-rc.d #{script} defaults"
            elsif @osfamily == 'redhat'
              shell "chkconfig --level 345 #{script} on"
              shell "systemctl daemon-reload" if @osmajrelease == '7'
            end
          end

    end
  end
end