puppet-repl 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f50d114eed718108fb705d0a03a06afcc2e44a94
-  data.tar.gz: 37e67568fffa41aeec376da36f2d7b50453965d0
+  metadata.gz: 264126ccdd7fafe72a52274c967c28ef18d53679
+  data.tar.gz: 31d01dd0eaa97e130ec51d9679424839d48cb73d
 SHA512:
-  metadata.gz: c1d732e9ddc6878fbb875b818b67fe3eeaf04cc17492dd2a7eb78f611c619f80c9559434815d03691f44463e6d63c6b53f692a13e21c3b32314d11f9429d143f
-  data.tar.gz: b7673de9a0f6fc739230f8e4f2881495a93fc56eb346205d347a9354b32dfdcdc766eb5d5200534a80efebd03d332ae65af4f561100cf5f0308a6a5a80d3db2a
+  metadata.gz: c12edea95534472ad07ba667ecdebd21a6d5f61eeccf01950b587e9e43eb6ea23563ccd5743731f91aba7f0245bb0898fdbe1580e9cf8b5839518cfe438f5910
+  data.tar.gz: 023f8e084288ca93a16366d8d8dfb473adadb0baa53c83f87951d2e99f14ecc3e55f9bd140622cb0b5fd4f531b9dee76482be9830c46745f63a26a0689c53957

data/README.md

@@ -8,6 +8,12 @@ Requires Puppet 3.8+ and only uses the future parser.
 ## Installation
 `gem install puppet-repl`
 
+## Load path
+puppet-repl will load all functions from your basemodulepath and environmentpath.
+
+This means if you run `puppet module install puppetlabs-stdlib` and they will be available
+in the repl.
+
 ## Usage
 Puppet-repl will only parse and evaulate your code.  It will not build a catalog
 and try to enforce the catalog. This has a few side affects.
@@ -34,7 +40,8 @@ Type "exit", "functions", "types", "reset", "help" for more information.
 
 ```
 
-Using Variables
+## Using Variables
+
 ```
 MacBook-Pro-2/tmp % prepl
 Puppet Version: 4.2.2
@@ -49,16 +56,13 @@ Type "exit", "functions", "types", "reset", "help" for more information.
 >>
 ```
 
-Using functions: functions will run and produce the desired output.
+## Using functions
+Functions will run and produce the desired output.  If you type the word `functions`
+a list of available functions will be displayed on the screen.
 
 ```
 >> split('hello/there/one/two/three','/')
- =>
-hello
-there
-one
-two
-three
+ => ["hello", "there", "one", "two", "three"]
 
 ```
 ## Duplicate resource error
@@ -72,6 +76,7 @@ Just like normal puppet code you cannot create duplicate resources.
 You can reset the parser by running `reset` within the repl without having to exit.
 
 ## Troubleshooting
+
 ## Forward
 I was just playing around and created this simple tool.  Its alpha quality,
 and a ton of features need to be added. Please create a issue if you see a bug or feature that should be added.

data/lib/puppet-repl/cli.rb

@@ -1,42 +1,17 @@
 require 'puppet'
 require 'puppet/pops'
-require "readline"
-
+require 'readline'
+require_relative 'support'
 module PuppetRepl
   class Cli
+    include PuppetRepl::Support
 
-    # returns a future parser for evaluating code
-    def parser
-      @parser || ::Puppet::Pops::Parser::EvaluatingParser.new
-    end
-
-    def module_dirs
-      Puppet.settings[:basemodulepath].split(':')
-    end
-
-    # creates a puppet environment given a module path and environment name
-    def puppet_environment
-      @puppet_environment ||= Puppet::Node::Environment.create('production', module_dirs)
-    end
-
-    def scope
-      unless @scope
-        begin
-          Puppet.initialize_settings
-        rescue
-          # do nothing otherwise calling init twice raises an error
-        end
-        #options['parameters']
-        #options['facts']
-        #options[:classes]
-        node_name = 'node_name'
-        node = Puppet::Node.new(node_name, :environment => puppet_environment)
-        compiler = Puppet::Parser::Compiler.new(node)
-        @scope = Puppet::Parser::Scope.new(compiler)
-        @scope.source = Puppet::Resource::Type.new(:node, node_name)
-        @scope.parent = compiler.topscope
+    def initialize
+      begin
+        Puppet.initialize_settings
+      rescue
+        # do nothing otherwise calling init twice raises an error
       end
-      @scope
     end
 
     def puppet_eval(input)
@@ -59,21 +34,25 @@ module PuppetRepl
       end
       result
     end
-#scope.environment.known_resource_types
+
     def handle_input(input)
       case input
       when 'help'
         PuppetRepl::Cli.print_repl_desc
       when 'functions'
-        puts "list of functions coming soon"
+        puts function_map.keys.sort
       when 'types'
         puts "list of types coming soon"
       when '_'
         puts(" => #{@last_item}")
+      when 'environment'
+        puts "Puppet Environment: #{puppet_env_name}"
       when 'exit'
         exit 0
       when 'reset'
         @scope = nil
+      when 'current_resources'
+        compiler.known_resource_types
       else
         result = puppet_eval(input)
         @last_item = result
@@ -87,6 +66,7 @@ module PuppetRepl
 
     def self.print_repl_desc
       puts(<<-EOT)
+Ruby Version: #{RUBY_VERSION}
 Puppet Version: #{Puppet.version}
 Puppet Repl Version: #{PuppetRepl::VERSION}
 Created by: NWOps <corey@nwops.io>

data/lib/puppet-repl/support.rb

@@ -0,0 +1,138 @@
+module PuppetRepl
+  module Support
+
+    # returns an array of module directories
+    def module_dirs
+      dirs = []
+      dirs << File.join(Puppet[:environmentpath],puppet_env_name,'modules')
+      dirs << Puppet.settings[:basemodulepath].split(':')
+      dirs.flatten
+    end
+
+    def puppet_lib_dir
+      # returns something like "/Library/Ruby/Gems/2.0.0/gems/puppet-4.2.2/lib/puppet.rb"
+      @puppet_lib_dir ||= File.dirname(Puppet.method(:[]).source_location.first)
+    end
+
+    # returns a array of function files
+    def function_files
+      search_dirs = lib_dirs.map do |lib_dir|
+        [File.join(lib_dir, 'puppet', 'functions', '**', '*.rb'),
+          File.join(lib_dir, 'functions', '**', '*.rb'),
+         File.join(lib_dir, 'puppet', 'parser', 'functions', '*.rb')
+         ]
+      end
+      # add puppet lib directories
+      search_dirs << [File.join(puppet_lib_dir, 'puppet', 'functions', '**', '*.rb'),
+        File.join(puppet_lib_dir, 'puppet', 'parser', 'functions', '*.rb')
+       ]
+      Dir.glob(search_dirs.flatten)
+    end
+
+    # returns either the module name or puppet version
+    def mod_finder
+      @mod_finder ||= Regexp.new('\/([\w\-\.]+)\/lib')
+    end
+
+    # returns a map of functions
+    def function_map
+      unless @functions
+        @functions = {}
+        function_files.each do |file|
+          obj = {}
+          name = File.basename(file, '.rb')
+          obj[:name] = name
+          obj[:parent] = mod_finder.match(file)[1]
+          @functions["#{obj[:parent]}::#{name}"] = obj
+        end
+      end
+      @functions
+    end
+
+    def lib_dirs
+      module_dirs.map do |mod_dir|
+        Dir["#{mod_dir}/*/lib"].entries
+      end.flatten
+    end
+
+    def load_lib_dirs
+      lib_dirs.each do |lib|
+        $LOAD_PATH << lib
+      end
+    end
+
+    # returns a future parser for evaluating code
+    def parser
+      @parser || ::Puppet::Pops::Parser::EvaluatingParser.new
+    end
+
+    # the cached name of the environment
+    def puppet_env_name
+      @penv ||= ENV['PUPPET_ENV'] || Puppet[:environment]
+    end
+
+    # creates a puppet environment given a module path and environment name
+    # this is cached
+    def puppet_environment
+      @puppet_environment ||= Puppet::Node::Environment.create(
+        puppet_env_name,
+        module_dirs,
+        manifests_dir
+        )
+    end
+
+    # def functions
+    #   @functions = []
+    #   @functions << compiler.loaders.static_loader.loaded.keys.find_all {|l| l.type == :function}
+    # end
+
+    def environment_loaders
+      name = compiler.loaders.public_environment_loader.loader_name
+    end
+
+    def compiler
+      @compiler
+    end
+
+    def create_scope(node_name)
+      #options['parameters']
+      #options['facts']
+      #options[:classes]
+      node = create_node(node_name)
+      @compiler = create_compiler(node)
+      scope = Puppet::Parser::Scope.new(compiler)
+      scope.source = Puppet::Resource::Type.new(:node, node_name)
+      scope.parent = @compiler.topscope
+      load_lib_dirs
+      scope
+    end
+
+    def create_compiler(node)
+      Puppet::Parser::Compiler.new(node)
+    end
+
+    def create_node(node_name)
+      Puppet::Node.new(node_name, :environment => puppet_environment)
+    end
+
+    def scope
+      begin
+        Puppet.initialize_settings
+      rescue
+        # do nothing otherwise calling init twice raises an error
+      end
+      @scope ||= create_scope('node_name')
+    end
+
+    def manifests_dir
+      File.join(Puppet[:environmentpath],puppet_env_name,'manifests')
+    end
+
+    def build_node(name, opts = {})
+      opts.merge!({:environment => node_environment})
+      Puppet::Node.new(name, opts)
+    end
+
+  end
+end
+#scope.environment.known_resource_types

data/lib/version.rb

@@ -1,3 +1,3 @@
 module PuppetRepl
-  VERSION = '0.0.1'
+  VERSION = '0.0.2'
 end

data/puppet-repl.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = "puppet-repl"
-  s.version = "0.0.1"
+  s.version = "0.0.2"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Corey Osman"]
-  s.date = "2016-02-19"
+  s.date = "2016-02-25"
   s.description = "A interactive command line tool for evaluating the puppet language"
   s.email = "corey@nwops.io"
   s.executables = ["prepl"]
@@ -29,10 +29,12 @@ Gem::Specification.new do |s|
     "bin/prepl",
     "lib/puppet-repl.rb",
     "lib/puppet-repl/cli.rb",
+    "lib/puppet-repl/support.rb",
     "lib/version.rb",
     "puppet-repl.gemspec",
     "spec/puppet-repl_spec.rb",
-    "spec/spec_helper.rb"
+    "spec/spec_helper.rb",
+    "spec/support_spec.rb"
   ]
   s.homepage = "http://github.com/nwops/puppet-repl"
   s.licenses = ["MIT"]

data/spec/puppet-repl_spec.rb

@@ -14,7 +14,7 @@ describe "PuppetRepl" do
       'help'
     end
     it 'can show the help screen' do
-      repl_output = "Puppet Version: 4.3.2\nPuppet Repl Version: 0.0.1\nCreated by: NWOps <corey@nwops.io>\nType \"exit\", \"functions\", \"types\", \"reset\", \"help\" for more information.\n\n"
+      repl_output = "Ruby Version: #{RUBY_VERSION}\nPuppet Version: 4.3.2\nPuppet Repl Version: 0.0.1\nCreated by: NWOps <corey@nwops.io>\nType \"exit\", \"functions\", \"types\", \"reset\", \"help\" for more information.\n\n"
       expect{repl.handle_input(input)}.to output(repl_output).to_stdout
     end
   end
@@ -49,6 +49,22 @@ describe "PuppetRepl" do
     end
   end
 
+  describe 'reset' do
+    before(:each) do
+      repl.handle_input(input)
+    end
+    let(:input) do
+      "file{'/tmp/reset': ensure => present}"
+    end
+
+    it 'can process a each block' do
+      repl.handle_input('reset')
+      repl_output = " => File['/tmp/reset']\n"
+      expect{repl.handle_input(input)}.to output(repl_output).to_stdout
+    end
+  end
+
+
   describe 'each block' do
     let(:input) do
       "['/tmp/test3', '/tmp/test4'].each |String $path| { file{$path: ensure => present} }"

data/spec/support_spec.rb

@@ -0,0 +1,43 @@
+require 'spec_helper'
+
+
+describe 'support' do
+
+  let(:repl) do
+    PuppetRepl::Cli.new
+  end
+
+  let(:scope) do
+    repl.scope
+  end
+
+  let(:puppet_version) do
+    repl.mod_finder.match(repl.puppet_lib_dir)[1]
+  end
+
+  context '#function_map' do
+
+    it 'should list functions' do
+      func = repl.function_map["#{puppet_version}::hiera"]
+      expect(repl.function_map).to be_instance_of(Hash)
+      expect(func).to eq({:name => 'hiera', :parent => puppet_version})
+    end
+
+  end
+
+  it 'should return a puppet version' do
+    expect(puppet_version).to match(/puppet-\d\.\d.\d/)
+  end
+
+  it 'should return lib dirs' do
+    expect(repl.lib_dirs.count).to be >= 1
+
+
+  end
+
+  it 'should return module dirs' do
+    expect(repl.module_dirs.count).to be >= 1
+
+  end
+
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: puppet-repl
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Corey Osman
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-02-19 00:00:00.000000000 Z
+date: 2016-02-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: puppet
@@ -142,10 +142,12 @@ files:
 - bin/prepl
 - lib/puppet-repl.rb
 - lib/puppet-repl/cli.rb
+- lib/puppet-repl/support.rb
 - lib/version.rb
 - puppet-repl.gemspec
 - spec/puppet-repl_spec.rb
 - spec/spec_helper.rb
+- spec/support_spec.rb
 homepage: http://github.com/nwops/puppet-repl
 licenses:
 - MIT