simp-cli 1.0.12

data/lib/simp/cli/config/item/puppet_ca_port.rb

@@ -0,0 +1,28 @@
+require 'highline/import'
+require 'puppet'
+require File.expand_path( '../item', File.dirname(__FILE__) )
+require File.expand_path( '../utils', File.dirname(__FILE__) )
+
+module Simp; end
+class Simp::Cli; end
+module Simp::Cli::Config
+  class Item::PuppetCAPort < Item
+    def initialize
+      super
+      @key         = 'puppet::ca_port'
+      @description = 'The port which the Puppet CA will listen on (8141 by default).'
+    end
+
+    def os_value
+      Puppet.settings.setting( 'ca_port' ).value
+    end
+
+    def validate( x )
+       (x.to_s =~ /^\d+$/ ? true : false ) && x.to_i > 0 && x.to_i <= 65535
+    end
+
+    def recommended_value
+      8141
+    end
+  end
+end

data/lib/simp/cli/config/item/puppet_conf.rb

@@ -0,0 +1,98 @@
+require 'highline/import'
+require File.expand_path( '../item', File.dirname(__FILE__) )
+require File.expand_path( '../utils', File.dirname(__FILE__) )
+
+module Simp; end
+class Simp::Cli; end
+module Simp::Cli::Config
+  class Item::PuppetConf < ActionItem
+    attr_accessor :file
+
+    def initialize
+      super
+      @key         = 'puppet::conf'
+      @description = 'silent item; configures /etc/puppet/puppet.conf'
+      # FIXME: this path will change with Puppet Enterprise; should this autodetect?
+      @file        = '/etc/puppet/puppet.conf'
+    end
+
+    # NOTE: This is (mostly) lifted straight from the old simp config
+    # TODO: refactor sed statements to pure ruby,
+    #       consider using IO handles instead of File.open (easier to test in memory)?
+    #       or use Puppet::Settings ( https://github.com/puppetlabs/puppet/blob/master/lib/puppet/settings.rb )?
+    def apply
+      say_green "Updating #{@file}..." if !@silent
+      if @skip_apply
+        say_yellow "WARNING: directed to skip Puppet configuration of #{file}" if !@silent
+        return
+      end
+
+      backup_file = "#{@file}.pre_simpconfig"
+      FileUtils.cp("#{@file}", backup_file)
+      `sed -i '/^\s*server.*/d'          #{@file}`
+      `sed -i '/.*trusted_node_data.*/d' #{@file}`
+      `sed -i '/.*digest_algorithm.*/d'  #{@file}`
+      `sed -i '/.*stringify_facts.*/d'   #{@file}`
+      `sed -i '/.*environment_path.*/d'  #{@file}`
+      `sed -i '/^.main./ a \\    trusted_node_data = true\'   #{@file}`
+      `sed -i '/^.main./ a \\    digest_algorithm  = sha256\' #{@file}`
+      `sed -i '/^.main./ a \\    stringify_facts   = false\'  #{@file}`
+      `sed -i '/^.main./ a \\    environmentpath   = /etc/puppet/environments\' #{@file}`
+      `sed -i '/trusted_node_data/ a \\    server            = #{@config_items.fetch( 'puppet::server' ).value}\' #{@file}`
+      keylength = @config_items.fetch( 'use_fips', nil )? '2048' : '4096'
+      `sed -i '/^.main./ a \\    keylength         = #{keylength}\' #{@file}`
+
+      # do not die if config items aren't found
+      puppet_server  = 'puppet.change.me'
+      puppet_ca      = 'puppetca.change.me'
+      puppet_ca_port = '8141'
+      if item = @config_items.fetch( 'puppet::server', nil )
+        puppet_server  = item.value
+      end
+      if item = @config_items.fetch( 'puppet::ca', nil )
+        puppet_ca      = item.value
+      end
+      if item = @config_items.fetch( 'puppet::ca_port', nil )
+        puppet_ca_port = item.value
+      end
+
+      puppet_conf = File.readlines(@file)
+      File.open("#{@file}", 'w') do |out_file|
+        line_check = {
+          'server'    => false,
+          'ca_server' => false,
+          'ca_port'   => false
+        }
+        puppet_conf.each do |line|
+          if line !~ /^\s*(#{line_check.keys.join('|')})(\s*=\s*)/
+            out_file.puts line
+          else
+            $1.chomp
+            line_check[$1] = true
+            case $1
+              when 'server' then
+                out_file.puts "    #{$1}#{$2}#{puppet_server}"
+              when 'ca_server' then
+                out_file.puts "    #{$1}#{$2}#{puppet_ca}"
+              when 'ca_port' then
+                out_file.puts "    #{$1}#{$2}#{puppet_ca_port}"
+            end
+          end
+        end
+        line_check.keys.each do |key|
+          if not line_check[key] then
+            case key
+              when 'server' then
+                out_file.puts "    server    = #{puppet_server}"
+              when 'ca_server' then
+                out_file.puts "    ca_server = #{puppet_ca}"
+              when 'ca_port' then
+                out_file.puts "    ca_port   = #{puppet_ca_port}"
+            end
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/simp/cli/config/item/puppet_fileserver.rb

@@ -0,0 +1,104 @@
+require 'highline/import'
+require File.expand_path( '../item', File.dirname(__FILE__) )
+require File.expand_path( '../utils', File.dirname(__FILE__) )
+
+module Simp; end
+class Simp::Cli; end
+module Simp::Cli::Config
+  class Item::PuppetFileServer < ActionItem
+
+    attr_accessor :file
+
+    def initialize
+      super
+      @key         = 'puppet::fileserver'
+      @description = 'silent item; configures /etc/puppet/fileserver.conf'
+      @file        = '/etc/puppet/fileserver.conf'
+    end
+
+    def apply
+      say_green "  updating Puppet configurations in #{@file}..." if !@silent
+
+      conf = @file
+
+      require 'fileutils'
+      FileUtils.cp(conf, "#{conf}.pre_simpconfig")
+
+      hostname = @config_items.fetch( 'hostname' ) #FIXME: should this be hostname or puppet_server?
+      domain = hostname.value.split('.')[1..-1].join('.')
+
+      if !domain or domain.empty?
+        raise "Could not determine domain from hostname '#{hostname}"
+      end
+
+      default_entries = ['facts','plugins','keydist','cacerts','mcollective']
+
+      fileserver_default = <<-EOM
+        [facts]
+          path /etc/puppet/facts
+          allow *.#{domain}
+
+        [plugins]
+          allow *.#{domain}
+
+        [keydist]
+          path /etc/puppet/keydist/%H
+          allow *.#{domain}
+
+        [cacerts]
+          path /etc/puppet/keydist/cacerts
+          allow *.#{domain}
+
+        [mcollective]
+          path /etc/puppet/keydist/mcollective
+          allow *.#{domain}
+      EOM
+
+      # Complete crib from StackOverflow
+      fileserver_default.gsub!(/^#{fileserver_default[/\A\s*/]}/,'')
+
+      fileserver_new = []
+
+      fileserver_old = File.read(conf).split("\n")
+
+      # Preserve any beginning comments
+      while fileserver_old[0] =~ /^\s*(#.*|\s*)$/ do
+        fileserver_new << fileserver_old.shift
+      end
+
+      # Add in our defaults
+      fileserver_new << fileserver_default
+
+      # Read the rest of the file, ignoring any section that we're going to
+      # replace.
+      key = nil
+      comments = []
+      fileserver_old.each do |line|
+        if line =~ /\[(.*)\]/ then
+          key = $1.strip
+          comments = []
+        end
+
+        next if default_entries.include?(key)
+
+        fileserver_new << line
+      end
+
+      # If the last entry was a default entry key, preserve the trailing file
+      # comments (if any)
+      if default_entries.include?(key) then
+        fileserver_new << "\n"
+        fileserver_new += comments
+      end
+
+      # Smash duplicates
+      fileserver_new.each_with_index do |x,i|
+        fileserver_new.delete(i) unless fileserver_new[i] != fileserver_new[i + 1]
+      end
+
+      File.open(conf,'w'){|x| x.puts(fileserver_new.join("\n"))}
+
+      true
+    end
+  end
+end

data/lib/simp/cli/config/item/puppet_hosts_entry.rb

@@ -0,0 +1,44 @@
+require "resolv"
+require 'highline/import'
+require File.expand_path( '../item', File.dirname(__FILE__) )
+require File.expand_path( '../utils', File.dirname(__FILE__) )
+
+module Simp; end
+class Simp::Cli; end
+module Simp::Cli::Config
+  class Item::PuppetHostsEntry < ActionItem
+    attr_accessor :file
+
+    def initialize
+      super
+      @key         = 'puppet::hosts_entry'
+      @description = %Q{Ensures an entry for the puppet server in /etc/hosts (apply-only; noop).}
+      @file        = '/etc/hosts'
+    end
+
+    def apply
+      puppet_server    = @config_items.fetch( 'puppet::server' ).value
+      puppet_server_ip = @config_items.fetch( 'puppet::server::ip' ).value
+      status = false
+
+      say_green "Updating #{@file}..." if !@silent
+
+      values = Array.new
+      File.readlines(@file).each do |line|
+        next if line =~ /\s*#/
+        next if line =~ /#{puppet_server}/ and @value.eql?(puppet_server)
+        next if line =~ /localdomain/
+        next if line =~ /localdomain6/
+        next if line =~ /\spuppet(\s|$)/  # remove alias 'puppet'
+        values.push(line)
+      end
+      File.open(@file,'w') {|fh|
+        fh.puts('127.0.0.1 localhost localhost.localdomain localhost4 localhost4.localdomain4')
+        fh.puts('::1 localhost localhost.localdomain localhost6 localhost6.localdomain6')
+        fh.puts("#{puppet_server_ip} #{puppet_server} #{puppet_server.split('.').first}")
+        fh.puts(values)
+      }
+      true
+    end
+  end
+end

data/lib/simp/cli/config/item/puppet_server.rb

@@ -0,0 +1,30 @@
+require 'highline/import'
+require 'puppet'
+require File.expand_path( '../item', File.dirname(__FILE__) )
+require File.expand_path( '../utils', File.dirname(__FILE__) )
+
+module Simp; end
+class Simp::Cli; end
+module Simp::Cli::Config
+  class Item::PuppetServer < Item
+    def initialize
+      super
+      @key         = 'puppet::server'
+      @description = %q{The Hostname or FQDN of the puppet server.}
+    end
+
+    def os_value
+      Puppet.settings.setting( 'server' ).value
+    end
+
+    def validate( x )
+      Simp::Cli::Config::Utils.validate_hostname( x ) ||
+      Simp::Cli::Config::Utils.validate_fqdn( x )
+    end
+
+    def recommended_value
+      item = @config_items.fetch( 'hostname', nil )
+      item.value if item
+    end
+  end
+end

data/lib/simp/cli/config/item/puppet_server_ip.rb

@@ -0,0 +1,25 @@
+require File.expand_path( '../item', File.dirname(__FILE__) )
+require File.expand_path( '../utils', File.dirname(__FILE__) )
+
+module Simp; end
+class Simp::Cli; end
+module Simp::Cli::Config
+  class Item::PuppetServerIP < Item
+    def initialize
+      super
+      @key         = 'puppet::server::ip'
+      @description = %Q{The Puppet server's IP address.\nThis is used to configure /etc/hosts properly.}
+    end
+
+
+    # Always recommend the configured IP
+    def recommended_value
+      @config_items.fetch( 'ipaddress' ).value
+    end
+
+
+    def validate( x )
+      Simp::Cli::Config::Utils.validate_ip x
+    end
+  end
+end

data/lib/simp/cli/config/item/puppetdb_port.rb

@@ -0,0 +1,25 @@
+require File.expand_path( '../item', File.dirname(__FILE__) )
+require File.expand_path( '../utils', File.dirname(__FILE__) )
+
+module Simp; end
+class Simp::Cli; end
+
+module Simp::Cli::Config
+  class Item::PuppetDBPort < Item
+    def initialize
+      super
+      @key         = 'puppetdb::master::config::puppetdb_port'
+      @description = %Q{The PuppetDB server port number}
+      @value       = recommended_value
+    end
+
+    def recommended_value
+      '8139'
+    end
+
+    def validate string
+      ( string =~ /^\d+$/ ? true : false ) &&
+      ( string.to_i > 1 && string.to_i < 65536 )
+    end
+  end
+end

data/lib/simp/cli/config/item/puppetdb_server.rb

@@ -0,0 +1,26 @@
+require File.expand_path( '../item', File.dirname(__FILE__) )
+require File.expand_path( '../utils', File.dirname(__FILE__) )
+
+module Simp; end
+class Simp::Cli; end
+
+module Simp::Cli::Config
+  class Item::PuppetDBServer < Item
+    def initialize
+      super
+      @key         = 'puppetdb::master::config::puppetdb_server'
+      @description = %Q{The dns name or ip of the puppetdb server}
+      @value       = recommended_value
+    end
+
+    def recommended_value
+      "%{hiera('puppet::server')}"
+    end
+
+    def validate string
+      Simp::Cli::Config::Utils.validate_fqdn( string ) ||
+      Simp::Cli::Config::Utils.validate_ip( string )   ||
+      Simp::Cli::Config::Utils.validate_hiera_lookup( string )
+    end
+  end
+end

data/lib/simp/cli/config/item/remove_ldap_from_hiera.rb

@@ -0,0 +1,47 @@
+require "resolv"
+require File.expand_path( '../item', File.dirname(__FILE__) )
+require File.expand_path( '../utils', File.dirname(__FILE__) )
+
+module Simp; end
+class Simp::Cli; end
+module Simp::Cli::Config
+  class Item::RemoveLdapFromHiera < ActionItem
+    attr_accessor :dir
+
+    def initialize
+      super
+      @key         = 'puppet::remove_ldap_from_hiera'
+      @description = %Q{Removes any ldap classes from hieradata/hosts/puppet.your.domain.yaml (apply-only; noop).}
+      @dir         = "/etc/puppet/environments/simp/hieradata/hosts"
+      @file        = nil
+    end
+
+    def apply
+      success = true
+      fqdn    = @config_items.fetch( 'hostname' ).value
+      file    = File.join( @dir, "#{fqdn}.yaml")
+
+      say_green 'Removing ldap classes from the <domain>.yaml file' if !@silent
+
+      if File.exists?(file)
+        lines = File.open(file,'r').readlines
+
+        File.open(file, 'w') do |f|
+          lines.each do |line|
+            line.chomp!
+            f.puts line if !strip_line?(line)
+          end
+        end
+      else
+        success = false
+        say_yellow "WARNING: file not found: #{file}"
+      end
+      success
+    end
+
+
+    def strip_line?( line )
+      (line =~ /^\s*-\s+(([a-z_:'"]*::)*(open)*ldap|(open)*ldap[a-z_:'"]*)/m) ? true : false
+    end
+  end
+end

data/lib/simp/cli/config/item/rename_fqdn_yaml.rb

@@ -0,0 +1,40 @@
+require "resolv"
+require 'highline/import'
+require File.expand_path( '../item', File.dirname(__FILE__) )
+require File.expand_path( '../utils', File.dirname(__FILE__) )
+
+module Simp; end
+class Simp::Cli; end
+module Simp::Cli::Config
+  class Item::RenameFqdnYaml < ActionItem
+    attr_accessor :file
+
+    def initialize
+      super
+      @key         = 'puppet::rename_fqdn_yaml'
+      @description = %Q{Renames hieradata/hosts/puppet.your.domain.yaml (apply-only; noop).}
+      @file        = '/etc/puppet/environments/simp/hieradata/hosts/puppet.your.domain.yaml'
+    end
+
+    def apply
+      result   = true
+      fqdn     = @config_items.fetch( 'hostname' ).value
+      new_file = File.join( File.dirname( @file ), "#{fqdn}.yaml" )
+      say_green 'Moving default <domain>.yaml file' if !@silent
+
+      if File.exists?(@file)
+        if File.exists?( new_file )
+          result = false
+          diff   = `diff #{new_file} #{@file}`
+          say_yellow "WARNING: #{File.basename( new_file )} exists, but the content differs from the original system content. Review and consider updating:\n#{diff}" if !diff.empty?
+        else
+          File.rename( @file, new_file )
+        end
+      else
+        result = false
+        say_yellow "WARNING: file not found: #{@file}"
+      end
+      true
+    end
+  end
+end