puppet-repl 0.1.1 → 0.2.0

data/lib/puppet-repl/support/errors.rb

@@ -0,0 +1,69 @@
+module PuppetRepl
+  module Exception
+    class Error < StandardError
+      attr_accessor :data
+      def initialize(data={})
+        @data = data
+      end
+
+    end
+
+    class FatalError < Error
+    end
+
+    class ConnectError < Error
+      def message
+        out = <<-EOF
+#{data[:message]}
+EOF
+      end
+    end
+
+    class UndefinedNode < FatalError
+      def message
+        out = <<-EOF
+Cannot find node with name: #{data[:name]} on remote server
+    EOF
+      end
+    end
+
+    class TimeOutError < Error
+      #Errno::ETIMEDOUT
+    end
+
+    class NoClassError < FatalError
+      def message
+        out = <<-EOF
+#{data[:message]}
+You are missing puppet classes that are required for compilation.
+Please ensure these classes are installed on this machine in any of the following paths:
+#{data[:default_modules_paths]}
+EOF
+      end
+    end
+
+    class NodeDefinitionError < FatalError
+      def message
+        out = <<-EOF
+You are missing a default node definition in your site.pp that is required for compilation.
+Please ensure you have at least the following default node definition
+  node default {
+    # include classes here
+  }
+in your #{data[:default_site_manifest]} file.
+EOF
+        out.fatal
+      end
+    end
+
+    class AuthError < FatalError
+      def message
+        out = <<-EOF
+#{data[:message]}
+You will need to edit your auth.conf or conf.d/auth.conf (puppetserver) to allow node calls.
+EOF
+    end
+    end
+
+  end
+end

data/lib/puppet-repl/support/facts.rb

@@ -22,6 +22,15 @@ module PuppetRepl
         end
         @facts
       end
+
+      def server_facts
+        data = {}
+        data["servername"] = Facter.value("fqdn")
+        data['serverip'] = Facter.value("ipaddress")
+        data["serverversion"] = Puppet.version.to_s
+        data
+      end
+
     end
   end
 end

data/lib/puppet-repl/support/input_responders.rb

@@ -1,41 +1,83 @@
 module PuppetRepl
   module Support
     module InputResponders
+
+      def static_responder_list
+        ["exit", "functions", "classification","vars", "krt", "facts",
+         "resources", "classes", "play","reset", "help"
+        ]
+      end
+
       def help(args=[])
         PuppetRepl::Cli.print_repl_desc
       end
 
+      def handle_set(input)
+        output = ''
+        args = input.split(' ')
+        args.shift # throw away the set
+        case args.shift
+        when /node/
+          if name = args.shift
+            output = "Resetting to use node #{name}"
+            reset
+            set_remote_node_name(name)
+          else
+            out_buffer.puts "Must supply a valid node name"
+          end
+        when /loglevel/
+          if level = args.shift
+            @log_level = level
+            set_log_level(level)
+            output = "loglevel #{Puppet::Util::Log.level} is set"
+          end
+        end
+        output
+      end
+
       def facts(args=[])
-        # convert symbols to keys
-        vars = node.facts.values
-        ap(vars, {:sort_keys => true, :indent => -1})
+        variables = node.facts.values
+        variables.ai({:sort_keys => true, :indent => -1})
       end
 
       def functions(args=[])
-        puts function_map.keys.sort
+        filter = args.first || ''
+        function_map.keys.sort.grep(/^#{Regexp.escape(filter)}/)
       end
 
       def vars(args=[])
         # remove duplicate variables that are also in the facts hash
-        vars = scope.to_hash.delete_if {| key, value | node.facts.values.key?(key) }
-        vars['facts'] = 'removed by the puppet-repl' if vars.key?('facts')
-        ap 'Facts were removed for easier viewing'
-        ap(vars, {:sort_keys => true, :indent => -1})
+        variables = scope.to_hash.delete_if {| key, value | node.facts.values.key?(key) }
+        variables['facts'] = 'removed by the puppet-repl' if variables.key?('facts')
+        output = "Facts were removed for easier viewing".ai + "\n"
+        output += variables.ai({:sort_keys => true, :indent => -1})
       end
 
       def environment(args=[])
-        puts "Puppet Environment: #{puppet_env_name}"
+        "Puppet Environment: #{puppet_env_name}"
       end
 
       def reset(args=[])
         set_scope(nil)
-        # initilize scope again
-        scope
+        set_remote_node_name(nil)
+        set_node(nil)
+        set_facts(nil)
         set_log_level(log_level)
       end
 
+      def set_log_level(level)
+        Puppet::Util::Log.level = level.to_sym
+        buffer_log = Puppet::Util::Log.newdestination(:buffer)
+        if buffer_log
+          # if this is already set the buffer_log is nil
+          buffer_log.out_buffer = out_buffer
+          buffer_log.err_buffer = out_buffer
+        end
+        nil
+      end
+
       def krt(args=[])
-        ap(known_resource_types, {:sort_keys => true, :indent => -1})
+        known_resource_types.ai({:sort_keys => true, :indent => -1})
       end
 
       def play(args=[])
@@ -44,20 +86,24 @@ module PuppetRepl
         play_back(config)
       end
 
+      def classification(args=[])
+        node.classes.ai
+      end
+
       def resources(args=[])
         res = scope.compiler.catalog.resources.map do |res|
           res.to_s.gsub(/\[/, "['").gsub(/\]/, "']") # ensure the title has quotes
         end
         if !args.first.nil?
-          ap res[args.first.to_i]
+          res[args.first.to_i].ai
         else
-          puts "Resources not shown in any specific order".warning
-          ap res
+          output = "Resources not shown in any specific order\n".warning
+          output += res.ai
         end
       end
 
       def classes(args=[])
-        ap scope.compiler.catalog.classes
+        scope.compiler.catalog.classes.ai
       end
 
     end

data/lib/puppet-repl/support/node.rb

@@ -1,14 +1,40 @@
+require 'puppet/indirector/node/rest'
+
 module PuppetRepl
   module Support
     module Node
-      # creates a node object
+      # creates a node object using defaults or gets the remote node
+      # object if the remote_node_name is defined
       def create_node
-        options = {}
-        options[:parameters] = default_facts.values
-        options[:facts] = default_facts
-        options[:classes] = []
-        options[:environment] = puppet_environment
-        Puppet::Node.new(default_facts.values['fqdn'], options)
+        Puppet[:trusted_server_facts] = true if Puppet.version.to_f >= 4.1
+
+        if remote_node_name
+          # refetch
+          node_obj = set_node_from_name(remote_node_name)
+        end
+        unless node_obj
+          options = {}
+          options[:parameters] = default_facts.values
+          options[:facts] = default_facts
+          options[:classes] = []
+          options[:environment] = puppet_environment
+          node_obj = Puppet::Node.new(default_facts.values['fqdn'], options)
+          node_obj.add_server_facts(server_facts) if node_obj.respond_to?(:add_server_facts)
+          node_obj
+        end
+        node_obj
+      end
+
+      def set_remote_node_name(name)
+        @remote_node_name = name
+      end
+
+      def remote_node_name=(name)
+        @remote_node_name = name
+      end
+
+      def remote_node_name
+        @remote_node_name
       end
 
       # @return [node] puppet node object
@@ -16,6 +42,45 @@ module PuppetRepl
         @node ||= create_node
       end
 
+      def get_remote_node(name)
+        indirection = Puppet::Indirector::Indirection.instance(:node)
+        indirection.terminus_class = 'rest'
+        remote_node = indirection.find(name, :environment => puppet_environment)
+        remote_node
+      end
+
+      # this is a hack to get around that the puppet node fact face does not return
+      # a proper node object with the facts hash populated
+      # returns a node object with a proper facts hash
+      def convert_remote_node(remote_node)
+        options = {}
+        # remove trusted data as it will later get populated during compilation
+        parameters = remote_node.parameters.dup
+        trusted_data = parameters.delete('trusted')
+        options[:parameters] = parameters || {}
+        options[:facts] = Puppet::Node::Facts.new(remote_node.name,remote_node.parameters)
+        options[:classes] = remote_node.classes
+        options[:environment] = puppet_environment
+        node_object = Puppet::Node.new(remote_node.name, options)
+        node_object.add_server_facts(server_facts) if node_object.respond_to?(:add_server_facts)
+        node_object.trusted_data = trusted_data
+        node_object
+      end
+
+      # query the remote puppet server and retrieve the node object
+      #
+      def set_node_from_name(name)
+        out_buffer.puts ("Fetching node #{name}")
+        remote_node = get_remote_node(name)
+        if remote_node and remote_node.parameters.empty?
+          remote_node_name = nil  # clear out the remote name
+          raise PuppetRepl::Exception::UndefinedNode.new(:name => remote_node.name)
+        end
+        remote_node_name = remote_node.name
+        node_object = convert_remote_node(remote_node)
+        set_node(node_object)
+      end
+
       def set_node(value)
         @node = value
       end

data/lib/puppet-repl/support/play.rb

@@ -9,23 +9,45 @@ module PuppetRepl
           elsif File.exists? config[:play]
             play_back_string(File.read(config[:play]))
           else config[:play]
-            puts "puppet-repl can't play #{config[:play]}'"
+            out_buffer.puts "puppet-repl can't play #{config[:play]}'"
           end
         end
       end
 
-      def play_back_url(url)
-        require 'open-uri'
-        require 'net/http'
+      def convert_to_text(url)
+        require 'uri'
+        url_data = URI(url)
+        case url_data.host
+        when /^github.com/
+          if url_data.path =~ /blob/
+            url.gsub('blob', 'raw')
+          end
+        when /^gist.github.com/
+          unless url_data.path =~ /raw/
+            url = url += '.txt'
+          end
+        when /^gitlab.com/
+          if url_data.path =~ /snippets/
+            url += '/raw' unless url_data.path =~ /raw/
+            url
+          else
+            url.gsub('blob', 'raw')
+          end
+        else
+          url
+        end
+      end
 
-        if url[/gist.github.com\/[a-z\d]+$/]
-          url += '.txt'
-        elsif url[/github.com.*blob/]
-          url.sub!('blob', 'raw')
+      def play_back_url(url)
+        begin
+          require 'open-uri'
+          require 'net/http'
+          converted_url = convert_to_text(url)
+          str = open(converted_url).read
+          play_back_string(str)
+        rescue SocketError
+          abort "puppet-repl can't play `#{converted_url}'"
         end
-        play_back_string open(url).read
-      rescue SocketError
-        abort "puppet-repl can't play `#{url}'"
       end
 
       def play_back_string(str)

data/lib/puppet-repl/support/scope.rb

@@ -1,7 +1,6 @@
 module PuppetRepl
   module Support
     module Scope
-
       def set_scope(value)
         @scope = value
       end
@@ -9,22 +8,27 @@ module PuppetRepl
       # @return [Scope] puppet scope object
       def scope
         unless @scope
-          do_initialize
-          @scope ||= create_scope(node)
+          @scope ||= create_scope
         end
         @scope
       end
 
-      def create_scope(node)
-        @compiler = create_compiler(node) # creates a new compiler for each scope
-        scope = Puppet::Parser::Scope.new(@compiler)
-        # creates a node class
-        scope.source = Puppet::Resource::Type.new(:node, node.name)
-        scope.parent = @compiler.topscope
-        load_lib_dirs
-        # compiling will load all the facts into the scope
-        # without this step facts will not get resolved
-        scope.compiler.compile # this will load everything into the scope
+      def create_scope
+        do_initialize
+        begin
+          @compiler = create_compiler(node) # creates a new compiler for each scope
+          scope = Puppet::Parser::Scope.new(@compiler)
+          # creates a node class
+          scope.source = Puppet::Resource::Type.new(:node, node.name)
+          scope.parent = @compiler.topscope
+          load_lib_dirs
+          # compiling will load all the facts into the scope
+          # without this step facts will not get resolved
+          scope.compiler.compile # this will load everything into the scope
+        rescue StandardError => e
+          err = parse_error(e)
+          raise err
+        end
         scope
       end
 

data/lib/puppet-repl/support.rb

@@ -16,6 +16,31 @@ module PuppetRepl
     include PuppetRepl::Support::InputResponders
     include PuppetRepl::Support::Play
 
+    # parses the error type into a more useful error message defined in errors.rb
+    # returns new error object or the original if error cannot be parsed
+    def parse_error(error)
+      case error
+      when SocketError
+        PuppetRepl::Exception::ConnectError.new(:message => "Unknown host: #{Puppet[:server]}")
+      when Net::HTTPError
+        PuppetRepl::Exception::AuthError.new(:message => error.message)
+      when Errno::ECONNREFUSED
+        PuppetRepl::Exception::ConnectError.new(:message => error.message)
+      when Puppet::Error
+        if error.message =~ /could\ not\ find\ class/i
+          PuppetRepl::Exception::NoClassError.new(:default_modules_paths => default_modules_paths,
+           :message => error.message)
+        elsif error.message =~ /default\ node/i
+          PuppetRepl::Exception::NodeDefinitionError.new(:default_site_manifest => default_site_manifest,
+           :message => error.message)
+        else
+          error
+        end
+      else
+        error
+      end
+    end
+
     # returns an array of module directories, generally this is the only place
     # to look for puppet code by default.  This is read from the puppet configuration
     def default_modules_paths
@@ -92,5 +117,9 @@ module PuppetRepl
       File.join(Puppet[:environmentpath],default_puppet_env_name,'manifests')
     end
 
+    def default_site_manifest
+      File.join(default_manifests_dir, 'site.pp')
+    end
+
   end
 end

data/lib/puppet-repl.rb

@@ -3,6 +3,8 @@ require_relative 'version'
 require 'awesome_print'
 require_relative 'awesome_print/ext/awesome_puppet'
 require_relative 'trollop'
+require 'puppet/util/log'
+require_relative 'puppet-repl/support/errors'
 # monkey patch in some color effects string methods
 class String
   def red;            "\033[31m#{self}\033[0m" end
@@ -18,3 +20,35 @@ class String
     split('_').map(&:capitalize).join
   end
 end
+
+Puppet::Util::Log.newdesttype :buffer do
+  require 'puppet/util/colors'
+  include Puppet::Util::Colors
+
+  attr_accessor :err_buffer, :out_buffer
+
+  def initialize(err=$stderr, out=$stdout)
+    @err_buffer = err
+    @out_buffer = out
+  end
+
+  def handle(msg)
+    levels = {
+      :emerg   => { :name => 'Emergency', :color => :hred,  :stream => err_buffer },
+      :alert   => { :name => 'Alert',     :color => :hred,  :stream => err_buffer },
+      :crit    => { :name => 'Critical',  :color => :hred,  :stream => err_buffer },
+      :err     => { :name => 'Error',     :color => :hred,  :stream => err_buffer },
+      :warning => { :name => 'Warning',   :color => :hred,  :stream => err_buffer },
+
+      :notice  => { :name => 'Notice',    :color => :reset, :stream => out_buffer },
+      :info    => { :name => 'Info',      :color => :green, :stream => out_buffer },
+      :debug   => { :name => 'Debug',     :color => :cyan,  :stream => out_buffer },
+    }
+
+    str = msg.respond_to?(:multiline) ? msg.multiline : msg.to_s
+    str = msg.source == "Puppet" ? str : "#{msg.source}: #{str}"
+
+    level = levels[msg.level]
+    level[:stream].puts colorize(level[:color], "#{level[:name]}: #{str}")
+  end
+end

data/lib/version.rb

@@ -1,3 +1,3 @@
 module PuppetRepl
-  VERSION = '0.1.1'
+  VERSION = '0.2.0'
 end

data/puppet-repl.gemspec

@@ -2,18 +2,21 @@
 # DO NOT EDIT THIS FILE DIRECTLY
 # Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
 # -*- encoding: utf-8 -*-
+# stub: puppet-repl 0.2.0 ruby lib
 
 Gem::Specification.new do |s|
   s.name = "puppet-repl"
-  s.version = "0.1.1"
+  s.version = "0.2.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.require_paths = ["lib"]
   s.authors = ["Corey Osman"]
-  s.date = "2016-04-17"
+  s.date = "2016-05-17"
   s.description = "A interactive command line tool for evaluating the puppet language"
   s.email = "corey@nwops.io"
   s.executables = ["prepl"]
   s.extra_rdoc_files = [
+    "CHANGELOG.md",
     "LICENSE.txt",
     "README.md"
   ]
@@ -35,6 +38,7 @@ Gem::Specification.new do |s|
     "lib/puppet-repl/support.rb",
     "lib/puppet-repl/support/compiler.rb",
     "lib/puppet-repl/support/environment.rb",
+    "lib/puppet-repl/support/errors.rb",
     "lib/puppet-repl/support/facts.rb",
     "lib/puppet-repl/support/functions.rb",
     "lib/puppet-repl/support/input_responders.rb",
@@ -44,7 +48,17 @@ Gem::Specification.new do |s|
     "lib/trollop.rb",
     "lib/version.rb",
     "puppet-repl.gemspec",
+    "resources/classes.png",
+    "resources/classification.png",
+    "resources/command_line_node.png",
+    "resources/functions.png",
+    "resources/hiera.png",
+    "resources/set_node.png",
+    "resources/tab_complete.png",
+    "resources/variables.png",
     "spec/fixtures/environments/production/manifests/site.pp",
+    "spec/fixtures/invalid_node_obj.yaml",
+    "spec/fixtures/node_obj.yaml",
     "spec/fixtures/sample_manifest.pp",
     "spec/prepl_spec.rb",
     "spec/puppet-repl_spec.rb",
@@ -53,8 +67,7 @@ Gem::Specification.new do |s|
   ]
   s.homepage = "http://github.com/nwops/puppet-repl"
   s.licenses = ["MIT"]
-  s.require_paths = ["lib"]
-  s.rubygems_version = "2.0.14"
+  s.rubygems_version = "2.4.5.1"
   s.summary = "A repl for the puppet language"
 
   if s.respond_to? :specification_version then
@@ -62,22 +75,19 @@ Gem::Specification.new do |s|
 
     if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
       s.add_runtime_dependency(%q<puppet>, [">= 3.8"])
-      s.add_runtime_dependency(%q<facterdb>, [">= 0"])
-      s.add_runtime_dependency(%q<awesome_print>, [">= 0"])
-      s.add_development_dependency(%q<pry>, [">= 0"])
+      s.add_runtime_dependency(%q<facterdb>, ["~> 0.3"])
+      s.add_runtime_dependency(%q<awesome_print>, ["~> 1.6"])
       s.add_development_dependency(%q<rdoc>, ["~> 3.12"])
     else
       s.add_dependency(%q<puppet>, [">= 3.8"])
-      s.add_dependency(%q<facterdb>, [">= 0"])
-      s.add_dependency(%q<awesome_print>, [">= 0"])
-      s.add_dependency(%q<pry>, [">= 0"])
+      s.add_dependency(%q<facterdb>, ["~> 0.3"])
+      s.add_dependency(%q<awesome_print>, ["~> 1.6"])
       s.add_dependency(%q<rdoc>, ["~> 3.12"])
     end
   else
     s.add_dependency(%q<puppet>, [">= 3.8"])
-    s.add_dependency(%q<facterdb>, [">= 0"])
-    s.add_dependency(%q<awesome_print>, [">= 0"])
-    s.add_dependency(%q<pry>, [">= 0"])
+    s.add_dependency(%q<facterdb>, ["~> 0.3"])
+    s.add_dependency(%q<awesome_print>, ["~> 1.6"])
     s.add_dependency(%q<rdoc>, ["~> 3.12"])
   end
 end

data/spec/fixtures/invalid_node_obj.yaml

@@ -0,0 +1,8 @@
+--- !ruby/object:Puppet::Node
+  name: invalid
+  classes:
+  parameters:
+  facts:
+  time: 2016-05-10 16:28:21.502653 -07:00
+  environment_name: production
+  expiration: 2016-05-10 16:58:21.502660 -07:00