pero 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 41f1941626b9f892ebe021edc387601c571976a69ea65c936b7464943fe7024e
+  data.tar.gz: ec1bbb64db216abda864d2d72d1654efb36ab9f8d19f06b155104d4f50d99937
+SHA512:
+  metadata.gz: '002196fd8e20ceb0ca7d1665e4155dd163c7bd63a7c093fef166d6eda6233e9126c9e33bff1310b80674a109f1478477a2a62d9ac819f64f2bc89c2d3e085026'
+  data.tar.gz: 8c49fc394ee3419d2b233f0776e90c1fe8cb75a15283628650aeb7a60e81a72312964d2b591d4d2191803b020cd2767023ef2a6041a39903ee0d23040af7c66f

data/.gitignore

@@ -0,0 +1,7 @@
+.vagrant
+ssl
+nodes/
+reports
+run
+yaml
+log

data/Gemfile

@@ -0,0 +1,6 @@
+source "https://rubygems.org"
+
+git_source(:github) {|repo_name| "https://github.com/#{repo_name}" }
+
+# Specify your gem's dependencies in pero.gemspec
+gemspec

data/Gemfile.lock

@@ -0,0 +1,61 @@
+PATH
+  remote: .
+  specs:
+    pero (0.1.0)
+      docker-api
+      logger
+      net-ssh
+      parallel
+      retryable
+      specinfra
+      thor
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    diff-lcs (1.4.4)
+    docker-api (1.34.2)
+      excon (>= 0.47.0)
+      multi_json
+    excon (0.76.0)
+    logger (1.4.2)
+    multi_json (1.15.0)
+    net-scp (3.0.0)
+      net-ssh (>= 2.6.5, < 7.0.0)
+    net-ssh (6.1.0)
+    net-telnet (0.1.1)
+    parallel (1.19.2)
+    rake (10.5.0)
+    retryable (3.0.5)
+    rspec (3.9.0)
+      rspec-core (~> 3.9.0)
+      rspec-expectations (~> 3.9.0)
+      rspec-mocks (~> 3.9.0)
+    rspec-core (3.9.2)
+      rspec-support (~> 3.9.3)
+    rspec-expectations (3.9.2)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.9.0)
+    rspec-mocks (3.9.1)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.9.0)
+    rspec-support (3.9.3)
+    sfl (2.3)
+    specinfra (2.82.18)
+      net-scp
+      net-ssh (>= 2.7)
+      net-telnet (= 0.1.1)
+      sfl
+    thor (1.0.1)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  bundler (~> 1.17)
+  pero!
+  rake (~> 10.0)
+  rspec (~> 3.0)
+
+BUNDLED WITH
+   1.17.3

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2020 pyama86
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README.md

@@ -0,0 +1,56 @@
+# Pero
+
+It is puppet run tool on local OS.
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'pero'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install pero
+
+## Depends
+
+- Docker
+
+## Usage
+
+### Install puppet
+
+```
+$ pero install --agent-version 3.3.1 10.0.0.1 # hostname is example.com
+```
+
+### Apply puppet
+
+```
+$ pero apply --server -version 3.3.1 example.com
+```
+
+### Show Support version
+
+```
+$ pero versions
+```
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/pero.
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).

data/Rakefile

@@ -0,0 +1,6 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/Vagrantfile

@@ -0,0 +1,50 @@
+# -*- mode: ruby -*-
+# vi: set ft=ruby :
+
+# All Vagrant configuration is done below. The "2" in Vagrant.configure
+# configures the configuration version (we support older styles for
+# backwards compatibility). Please don't change it unless you know what
+# you're doing.
+Vagrant.require_version '>= 1.9.0'
+plugins = [
+  {
+    plugin: 'vagrant-properties',
+    version: '~> 0.9'
+  },
+  {
+    plugin: 'vagrant-itamae',
+    version: '~> 0.2'
+  }
+]
+
+# 必須プラグインのチェック
+plugins.each do |p|
+  unless Vagrant.has_plugin?(p[:plugin], p[:version])
+    action = Vagrant.has_plugin?(p[:plugin]) ? 'update' : 'install'
+    Dir.chdir(Dir.home) { system "vagrant plugin #{action} #{p[:plugin]}" }
+  end
+end
+
+Vagrant.configure("2") do |config|
+  config.vm.box = "centos6"
+
+  def define_machine_spec(config, memory=512, cpus=2)
+    config.vm.provider :virtualbox do |vbox|
+      vbox.customize ["modifyvm", :id, "--memory", memory.to_i]
+      vbox.customize ["modifyvm", :id, "--cpus", cpus.to_i]
+    end
+  end
+
+  config.vm.define 'client' do |c|
+    c.vm.network :private_network, ip: "192.168.100.11"
+    c.vm.hostname = "client.dev"
+    define_machine_spec(c)
+  end
+
+  config.vm.define 'client7' do |c|
+    config.vm.box = "centos/7"
+    c.vm.network :private_network, ip: "192.168.100.12"
+    c.vm.hostname = "client7.dev"
+    define_machine_spec(c)
+  end
+end

data/bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "pero"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "irb"
+IRB.start(__FILE__)

data/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/bootstrap.rb

@@ -0,0 +1,8 @@
+module RecipeHelper
+  def include_cookbook(name)
+    include_recipe File.join(__dir__, 'cookbooks', name, 'default.rb')
+  end
+end
+
+Itamae::Recipe::EvalContext.include(RecipeHelper)
+include_recipe File.join('roles', node[:role], 'default.rb')

data/exe/pero

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+
+require "pero"
+Pero::CLI.start 
+

data/lib/pero.rb

@@ -0,0 +1,19 @@
+require "logger"
+require "specinfra"
+require "pero/version"
+require "pero/cli"
+require "pero/history"
+require "pero/ssh_executable"
+require "pero/docker"
+require "pero/puppet"
+require "pero/puppet/base"
+require "pero/puppet/redhat"
+
+
+module Pero
+  def self.log
+    @log ||= Logger.new(STDOUT)
+  end
+  class Error < StandardError; end
+  # Your code goes here...
+end

data/lib/pero/cli.rb

@@ -0,0 +1,65 @@
+require "pero"
+require "thor"
+require "parallel"
+
+module Pero
+  class CLI < Thor
+    class << self
+      def exit_on_failure?
+        true
+      end
+    end
+
+    def initialize(*)
+      super
+      Pero.log.level = ::Logger.const_get(options[:log_level].upcase) if options[:log_level]
+    end
+
+    def self.shared_options
+      option :log_level, type: :string, aliases: ['-l'], default: 'info'
+      option :user, type: :string, aliases: ['-x'], desc: "ssh user"
+      option :key, type: :string, aliases: ['-i'], desc: "ssh private key"
+      option :port, type: :numeric, aliases: ['-p'], desc: "ssh port"
+      option :ssh_config, type: :string, desc: "ssh config path"
+      option :environment, type: :string, desc: "puppet environment"
+      option :ask_password, type: :boolean, default: false, desc: "ask ssh or sudo password"
+      option :vagrant, type: :boolean, default: false, desc: "use vagrarant"
+      option :sudo, type: :boolean, default: true, desc: "use sudo"
+      option "concurrent", aliases: '-N',default: 3, type: :numeric, desc: "running concurrent"
+    end
+
+    desc "versions", "show support version"
+    def versions
+      Pero::Puppet::Redhat.show_versions
+    end
+
+    desc "apply", "puppet apply"
+    shared_options
+    option "server-version", type: :string, default: "6.12.0"
+    option :noop, aliases: '-n', default: false, type: :boolean
+    option :verbose, aliases: '-v', default: true, type: :boolean
+    option :tags, default: nil, type: :array
+    option "one-shot", default: false, type: :boolean, desc: "stop puppet server after run"
+    def apply(name_regexp)
+      nodes = Pero::History.search(name_regexp)
+      return unless nodes
+      Parallel.each(nodes, in_process: options["concurrent"]) do |n|
+        opt = n["last_options"].merge(options)
+        puppet = Pero::Puppet.new(opt["host"], opt)
+        puppet.apply
+      end
+    end
+
+    desc "install", "install puppet"
+    shared_options
+    option "agent-version", default: "6.17.0", type: :string
+    option "node-name", aliases: '-N', default: "", type: :string, desc: "json node name(default hostname)"
+    def install(*hosts)
+      Parallel.each(hosts, in_process: options["concurrent"]) do |host|
+        next if host =~ /^-/
+        puppet = Pero::Puppet.new(host, options)
+        puppet.install
+      end
+    end
+  end
+end

data/lib/pero/docker.rb

@@ -0,0 +1,159 @@
+require 'docker'
+require 'digest/md5'
+require "retryable"
+require 'net/https'
+module Pero
+  class Docker
+    attr_reader :server_version
+    def initialize(version, environment)
+      @server_version = version
+      @environment = environment
+    end
+
+    def build
+      Pero.log.info "start build container"
+      begin
+        image = ::Docker::Image.build(docker_file)
+      rescue => e
+        Pero.log.debug docker_file
+        Pero.log.error "failed build container #{e.inspect}"
+        raise e
+      end
+      Pero.log.info "success build container"
+      image
+    end
+
+    def container_name
+      "pero-#{server_version}-#{Digest::MD5.hexdigest(Dir.pwd)[0..5]}-#{@environment}"
+    end
+
+    def alerady_run?
+      ::Docker::Container.all(:all => true).find do |c|
+        c.info["Names"].first == "/#{container_name}" && c.info["State"] != "exited"
+      end
+    end
+
+    def run
+      ::Docker::Container.all(:all => true).each do |c|
+        c.delete(:force => true) if c.info["Names"].first == "/#{container_name}"
+      end
+
+      container = ::Docker::Container.create({
+        'name' => container_name,
+        'Hostname' => 'puppet',
+        'Image' => build.id,
+        'ExposedPorts' => { '8140/tcp' => {} },
+      })
+
+      container.start(
+        'Binds' => [
+          "#{Dir.pwd}:/etc/puppetlabs/code/environments/#{@environment}",
+          "#{Dir.pwd}/keys:/etc/puppetlabs/puppet/eyaml/",
+        ],
+        'PortBindings' => {
+          '8140/tcp' => [{ 'HostPort' => "0" }],
+        },
+        "AutoRemove" => true,
+      )
+
+      container = ::Docker::Container.all(:all => true).find do |c|
+        c.info["Names"].first == "/#{container_name}"
+      end
+
+      raise "can't start container" unless container
+
+      begin
+        Retryable.retryable(tries: 20, sleep: 5) do
+          https = Net::HTTP.new('localhost', container.info["Ports"].first["PublicPort"])
+          https.use_ssl = true
+          https.verify_mode = OpenSSL::SSL::VERIFY_NONE
+          Pero.log.debug "start server health check"
+          https.start {
+            response = https.get('/')
+            Pero.log.debug "puppet http response #{response}"
+          }
+        rescue => e
+          Pero.log.debug e.inspect
+          raise e
+        end
+      rescue
+        container.kill
+        raise "can't start puppet server"
+      end
+
+      container
+    end
+
+    def puppet_config
+<<-EOS
+[master]
+vardir = /var/puppet
+certname = puppet
+dns_alt_names = puppet,localhost
+autosign = true
+environment_timeout = unlimited
+codedir = /etc/puppetlabs/code
+
+[main]
+server = puppet
+#{@environment && @environment != "" ? "environment = #{@environment}" : nil}
+EOS
+
+
+    end
+    def docker_file
+      release_package,package_name, conf_dir  = if Gem::Version.new("4.0.0") > Gem::Version.new(server_version)
+        ["puppetlabs-release-el-#{el}.noarch.rpm", "puppet-server", "/etc/puppet"]
+      elsif Gem::Version.new("5.0.0") > Gem::Version.new(server_version) && Gem::Version.new("4.0.0") <= Gem::Version.new(server_version)
+        ["puppetlabs-release-pc1-el-#{el}.noarch.rpm", "puppetserver", "/etc/puppetlabs/puppet/"]
+      elsif Gem::Version.new("6.0.0") > Gem::Version.new(server_version)&& Gem::Version.new("5.0.0") <= Gem::Version.new(server_version)
+        ["puppet5-release-el-#{el}.noarch.rpm", "puppetserver", "/etc/puppetlabs/puppet/"]
+      else
+        ["puppet6-release-el-#{el}.noarch.rpm", "puppetserver", "/etc/puppetlabs/puppet/"]
+      end
+
+      <<-EOS
+FROM #{from_image}
+RUN curl -L -k -O https://yum.puppetlabs.com/#{release_package}  && \
+rpm -ivh #{release_package}
+RUN yum install -y #{package_name}-#{server_version}
+ENV PATH $PATH:/opt/puppetlabs/bin
+RUN echo -e "#{puppet_config.split(/\n/).join("\\n")}" > #{conf_dir}/puppet.conf
+CMD bash -c "rm -rf #{conf_dir}/ssl/* && #{create_ca} && #{run_cmd}"
+      EOS
+    end
+
+    def create_ca
+      release_package,package_name, conf_dir  = if Gem::Version.new("5.0.0") > Gem::Version.new(server_version)
+        #'(puppet cert generate `hostname` --dns_alt_names localhost,127.0.0.1 || puppet cert --allow-dns-alt-names sign `hostname`)'
+        'puppet cert generate `hostname` --dns_alt_names localhost,127.0.0.1'
+      elsif Gem::Version.new("6.0.0") > Gem::Version.new(server_version)
+        'puppet cert generate `hostname` --dns_alt_names localhost,127.0.0.1'
+      else
+        'puppetserver ca setup --ca-name `hostname` --subject-alt-names DNS:localhost'
+      end
+    end
+
+    def run_cmd
+      release_package,package_name, conf_dir  = if Gem::Version.new("5.0.0") > Gem::Version.new(server_version)
+        'puppet master --no-daemonize --verbose'
+      elsif Gem::Version.new("6.0.0") > Gem::Version.new(server_version)
+        'puppetserver foreground'
+      else
+        'puppetserver foreground'
+      end
+    end
+
+    def el
+      if Gem::Version.new("3.5.1") > Gem::Version.new(server_version)
+        6
+      else
+        7
+      end
+    end
+
+    def from_image
+      "centos:#{el}"
+    end
+  end
+end

data/lib/pero/history.rb

@@ -0,0 +1,42 @@
+require 'json'
+require "fileutils"
+module Pero
+  class History
+    def self.search(query, dir="nodes")
+      ret = []
+      Dir.foreach(dir) do |f|
+        next if %w(. ..).include? f
+        next unless f =~ /json$/
+        File.open(File.join(dir, f)) do |j|
+          h = JSON.load(j)
+          ret << h if h["name"] =~ /#{query}/
+        end
+      end
+      ret
+    end
+  end
+end
+
+module Pero
+  class History
+    class Attribute
+      def initialize(specinfra, options)
+        name = if options["node-name"].empty?
+                 specinfra.run_command("hostname").stdout.chomp
+               else
+                 options["node-name"]
+               end
+        options.delete("noop")
+        @h = {
+          name: name,
+          last_options: options
+        }
+      end
+
+      def save(dir="nodes")
+        FileUtils.mkdir_p(dir)
+        File.write("#{File.join(dir, @h[:name])}.json", @h.to_json)
+      end
+    end
+  end
+end

data/lib/pero/misc/puppet.conf

@@ -0,0 +1,6 @@
+[main]
+vardir      = /var/puppet
+ssldir      = $vardir/ssl
+manifestdir = $vardir/manifests
+templatedir = $vardir/templates
+modulepath  = $vardir/modules:$vardir/roles:$vardir/vendor/modules

data/lib/pero/puppet.rb

@@ -0,0 +1,169 @@
+require 'net/ssh'
+
+Specinfra::Configuration.error_on_missing_backend_type = false
+Specinfra.configuration.backend = :ssh
+module Specinfra
+  module Configuration
+    def self.sudo_password
+      return ENV['SUDO_PASSWORD'] if ENV['SUDO_PASSWORD']
+      return @sudo_password if defined?(@sudo_password)
+
+      # TODO: Fix this dirty hack
+      return nil unless caller.any? {|call| call.include?('channel_data') }
+
+      print "sudo password: "
+      @sudo_password = STDIN.noecho(&:gets).strip
+      print "\n"
+      @sudo_password
+    end
+  end
+end
+
+module Pero
+  class Puppet
+    extend Pero::SshExecutable
+    attr_reader :specinfra
+    def initialize(host, options)
+      @options = options.dup
+
+      @options[:host] = host
+      so = ssh_options
+      @specinfra = Specinfra::Backend::Ssh.new(
+        request_pty: true,
+        host: so[:host_name],
+        ssh_options: so,
+        disable_sudo: false,
+      )
+    end
+
+    # refs: github.com/itamae-kitchen/itamae
+    def ssh_options
+      opts = {}
+      opts[:host_name] = @options[:host]
+
+      # from ssh-config
+      ssh_config_files = @options["ssh_config"] ? [@options["ssh_config"]] : Net::SSH::Config.default_files
+      opts.merge!(Net::SSH::Config.for(@options["host"], ssh_config_files))
+      opts[:user] = @options["user"] || opts[:user] || Etc.getlogin
+      opts[:password] = @options["password"] if @options["password"]
+      opts[:keys] = [@options["key"]] if @options["key"]
+      opts[:port] = @options["port"] if @options["port"]
+
+      if @options["vagrant"]
+        config = Tempfile.new('', Dir.tmpdir)
+        hostname = opts[:host_name] || 'default'
+        vagrant_cmd = "vagrant ssh-config #{hostname} > #{config.path}"
+        if defined?(Bundler)
+          Bundler.with_clean_env do
+            `#{vagrant_cmd}`
+          end
+        else
+          `#{vagrant_cmd}`
+        end
+        opts.merge!(Net::SSH::Config.for(hostname, [config.path]))
+      end
+
+      if @options["ask_password"]
+        print "password: "
+        password = STDIN.noecho(&:gets).strip
+        print "\n"
+        opts.merge!(password: password)
+      end
+      opts
+    end
+
+    def install
+      Pero.log.info "bootstrap puppet"
+      osi = specinfra.os_info
+      os = case osi[:family]
+      when "redhat"
+        Redhat.new(specinfra, osi)
+      else
+          raise "sorry unsupport os, please pull request!!!"
+      end
+      os.install(@options["agent-version"])
+      Pero::History::Attribute.new(specinfra, @options).save
+    end
+
+    def serve_master
+        Pero.log.info "start puppet master container"
+        container = run_container
+        begin
+          yield container
+        rescue => e
+          Pero.log.error e.inspect
+          raise e
+        ensure
+          Pero.log.info "stop puppet master container"
+          container.kill if !@options["one-shot"]
+        end
+    end
+
+    def run_container
+      docker = Pero::Docker.new(@options["server-version"], @options["environment"])
+      docker.alerady_run? || docker.run
+    end
+
+    def apply
+      serve_master do |container|
+        port = container.info["Ports"].first["PublicPort"]
+        begin
+          tmpdir=(0...8).map{ (65 + rand(26)).chr }.join
+          Pero.log.info "start forwarding port:#{port}"
+
+          in_ssh_forwarding(port) do |host, ssh|
+            Pero.log.info "#{host}:puppet cmd[#{puppet_cmd}]"
+            cmd = "unshare -m -- /bin/bash -c 'export PATH=$PATH:/opt/puppetlabs/bin/ &&  mkdir -p /tmp/puppet/#{tmpdir} && \
+                           mkdir -p `puppet config print ssldir`  && mount --bind /tmp/puppet/#{tmpdir} `puppet config print ssldir` && \
+                           #{puppet_cmd}'"
+            Pero.log.debug "run cmd:#{cmd}"
+            ssh.exec!(specinfra.build_command(cmd))  do |channel, stream, data|
+                             Pero.log.info "#{host}:#{data.chomp}" if stream == :stdout && data.chomp != ""
+                             Pero.log.warn "#{host}:#{data.chomp}" if stream == :stderr && data.chomp != ""
+                           end
+            ssh.exec!(specinfra.build_command("rm -rf /tmp/puppet/#{tmpdir}"))
+            ssh.loop {true} if ENV['PERO_DEBUG']
+          end
+        rescue => e
+          Pero.log.error "puppet apply error:#{e.inspect}"
+        end
+      end
+
+      Pero::History::Attribute.new(specinfra, @options).save
+    end
+
+    def puppet_cmd
+        if Gem::Version.new("5.0.0") > Gem::Version.new(@options["agent-version"])
+            "puppet agent --no-daemonize --onetime #{parse_puppet_option(@options)} --server localhost"
+        else
+            "/opt/puppetlabs/bin/puppet agent --no-daemonize --onetime #{parse_puppet_option(@options)} --server localhost"
+        end
+    end
+
+    def parse_puppet_option(options)
+      ret = ""
+      %w(noop verbose).each do |n|
+        ret << " --#{n}" if options[n]
+      end
+      ret << " --tags #{options["tags"].join(",")}" if options["tags"]
+      ret
+    end
+
+    def in_ssh_forwarding(port)
+      options = specinfra.get_config(:ssh_options)
+
+      if !Net::SSH::VALID_OPTIONS.include?(:strict_host_key_checking)
+        options.delete(:strict_host_key_checking)
+      end
+
+      Net::SSH.start(
+        specinfra.get_config(:host),
+        options[:user],
+        options
+      ) do |ssh|
+        ssh.forward.remote(port, 'localhost', 8140)
+        yield specinfra.get_config(:host), ssh
+      end
+    end
+  end
+end

data/lib/pero/puppet/base.rb

@@ -0,0 +1,28 @@
+module Pero
+  class Puppet
+    class Base
+      attr_reader :specinfra, :os_info
+      def initialize(specinfra, os)
+        @specinfra = specinfra
+        @os_info = os
+      end
+
+      def run_specinfra(type, *args)
+        command = specinfra.command.get(type, *args)
+        if type.to_s.start_with?("check_")
+          check_command(command)
+        else
+          specinfra.run_command(command)
+        end
+      end
+
+      def check_command(*args)
+        unless args.last.is_a?(Hash)
+          args << {}
+        end
+        specinfra.run_command(*args).exit_status == 0
+      end
+
+    end
+  end
+end

data/lib/pero/puppet/redhat.rb

@@ -0,0 +1,69 @@
+module Pero
+  class Puppet
+    class Redhat < Base
+      def self.show_versions_commands
+        [
+          %w(rpm -ivh https://yum.puppetlabs.com/puppetlabs-release-el-7.noarch.rpm),
+          %w(yum --showduplicates search puppet),
+          %w(yum remove -y puppetlabs-release),
+          %w(rpm -ivh https://yum.puppetlabs.com/puppetlabs-release-pc1-el-7.noarch.rpm),
+          %w(yum --showduplicates search puppet),
+          %w(yum remove -y puppetlabs-release),
+          %w(rpm -ivh https://yum.puppetlabs.com/puppet5-release-el-7.noarch.rpm),
+          %w(yum --showduplicates search puppet),
+          %w(yum remove -y puppet5-release),
+          %w(rpm -ivh https://yum.puppetlabs.com/puppet6-release-el-7.noarch.rpm),
+          %w(yum --showduplicates search puppet),
+        ]
+      end
+
+      def self.show_versions
+        image = ::Docker::Image.create('fromImage' => 'centos:7')
+        init = image.run("/sbin/init")
+        ret = []
+        show_versions_commands.each do |c|
+          init.exec(c, stdout:false, stderr: false) do |stream, chunk|
+            chunk.split(/\n/).each do |r|
+              ret << r.gsub(/\.el.*/, '')  if r =~ /(^puppet-3|^puppet-agent|^puppet-server|^puppetserver)/
+            end
+          end
+        end
+        puts ret.sort.join("\n")
+        init.delete(:force => true)
+      end
+
+
+      def main_release
+        os_info[:release].split(/\./)[0]
+      end
+
+      def install(version)
+        release_package, package_name = if Gem::Version.new("4.0.0") > Gem::Version.new(version)
+          ["puppetlabs-release-el-#{main_release}.noarch.rpm", "puppet"]
+        elsif Gem::Version.new("5.0.0") > Gem::Version.new(version) && Gem::Version.new("4.0.0") <= Gem::Version.new(version)
+          ["puppetlabs-release-pc1-el-#{main_release}.noarch.rpm", "puppet"]
+        elsif Gem::Version.new("6.0.0") > Gem::Version.new(version) && Gem::Version.new("5.0.0") <= Gem::Version.new(version)
+          ["puppet5-release-el-#{main_release}.noarch.rpm", "puppet-agent"]
+        else
+          ["puppet6-release-el-#{main_release}.noarch.rpm", "puppet-agent"]
+        end
+
+        unless run_specinfra(:check_package_is_installed, package_name, version)
+          unless run_specinfra(:check_package_is_installed, release_package.gsub(/-el.*/, ''))
+            Pero.log.info "install package #{release_package}"
+            run_specinfra(:remove_package, "puppetlabs-release")
+            run_specinfra(:remove_package, "puppet5-release")
+            run_specinfra(:remove_package, "puppet6-release")
+            raise "failed package install:#{release_package}" if specinfra.run_command("rpm -ivh https://yum.puppetlabs.com/#{release_package}").exit_status != 0
+          end
+
+          Pero.log.info "install package #{package_name}-#{version}"
+          raise "failed package uninstall:#{package_name}" if run_specinfra(:remove_package, package_name).exit_status != 0
+          raise "failed package install:#{package_name} version #{version}" if run_specinfra(:install_package, package_name, version).exit_status != 0
+        else
+          Pero.log.info "puppet-#{version} installed"
+        end
+      end
+    end
+  end
+end

data/lib/pero/ssh_executable.rb

@@ -0,0 +1,35 @@
+module Pero
+  module SshExecutable
+    # ref:https://stackoverflow.com/questions/3386233/how-to-get-exit-status-with-rubys-netssh-library
+    def ssh_exec!(ssh, command)
+      stdout_data = ""
+      stderr_data = ""
+      exit_code = nil
+      exit_signal = nil
+      ssh.open_channel do |channel|
+        channel.exec(command) do |ch, success|
+          unless success
+            abort "FAILED: couldn't execute command (ssh.channel.exec)"
+          end
+          channel.on_data do |ch,data|
+            stdout_data+=data
+          end
+
+          channel.on_extended_data do |ch,type,data|
+            stderr_data+=data
+          end
+
+          channel.on_request("exit-status") do |ch,data|
+            exit_code = data.read_long
+          end
+
+          channel.on_request("exit-signal") do |ch, data|
+            exit_signal = data.read_long
+          end
+        end
+      end
+      ssh.loop
+      [stdout_data, stderr_data, exit_code, exit_signal]
+    end
+  end
+end

data/lib/pero/version.rb

@@ -0,0 +1,3 @@
+module Pero
+  VERSION = "0.1.0"
+end

data/pero.gemspec

@@ -0,0 +1,36 @@
+
+lib = File.expand_path("../lib", __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require "pero/version"
+
+Gem::Specification.new do |spec|
+  spec.name          = "pero"
+  spec.version       = Pero::VERSION
+  spec.authors       = ["pyama86"]
+  spec.email         = ["www.kazu.com@gmail.com"]
+
+  spec.summary       = %q{tool for puppet apply from our desktop.}
+  spec.description   = %q{tool for puppet apply from our desktop.}
+  spec.homepage      = "https://github.com/pyama86/pero"
+  spec.license       = "MIT"
+
+  # Specify which files should be added to the gem when it is released.
+  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
+  spec.files         = Dir.chdir(File.expand_path('..', __FILE__)) do
+    `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  end
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "thor"
+  spec.add_dependency "logger"
+  spec.add_dependency "docker-api"
+  spec.add_dependency "net-ssh"
+  spec.add_dependency 'retryable'
+  spec.add_dependency 'specinfra'
+  spec.add_dependency 'parallel'
+  spec.add_development_dependency "bundler", "~> 1.17"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "rspec", "~> 3.0"
+end

metadata

@@ -0,0 +1,206 @@
+--- !ruby/object:Gem::Specification
+name: pero
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- pyama86
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2020-08-24 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: logger
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: docker-api
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: net-ssh
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: retryable
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: specinfra
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: parallel
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.17'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.17'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+description: tool for puppet apply from our desktop.
+email:
+- www.kazu.com@gmail.com
+executables:
+- pero
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- Vagrantfile
+- bin/console
+- bin/setup
+- bootstrap.rb
+- exe/pero
+- lib/pero.rb
+- lib/pero/cli.rb
+- lib/pero/docker.rb
+- lib/pero/history.rb
+- lib/pero/misc/puppet.conf
+- lib/pero/puppet.rb
+- lib/pero/puppet/base.rb
+- lib/pero/puppet/redhat.rb
+- lib/pero/ssh_executable.rb
+- lib/pero/version.rb
+- pero.gemspec
+homepage: https://github.com/pyama86/pero
+licenses:
+- MIT
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.0.6
+signing_key:
+specification_version: 4
+summary: tool for puppet apply from our desktop.
+test_files: []