fhcap-cli 0.3.0

data/lib/fhcap/tasks/repo/list.rb

@@ -0,0 +1,22 @@
+require 'fhcap/tasks/task_base'
+
+module Fhcap
+  module Tasks
+    module Repo
+      class List < TaskBase
+
+        def run
+          thor.say "Repo::List", :yellow
+
+          table_contents = [table_header('Name', 'Directory', 'URL')]
+
+          repos_config.each do |name, cfg|
+            table_contents << table_row(name, repo_dir(name), cfg[:url])
+          end
+          thor.print_table table_contents
+        end
+
+      end
+    end
+  end
+end

data/lib/fhcap/tasks/repo/remove.rb

@@ -0,0 +1,34 @@
+require 'fhcap/tasks/task_base'
+require 'fhcap/tasks/setup'
+
+module Fhcap
+  module Tasks
+    module Repo
+      class Remove < TaskBase
+
+        attr_reader :name
+
+        def initialize(options)
+          super
+          @name = options[:name]
+          @skip_setup = options[:'skip-setup']
+        end
+
+        def run
+          thor.say "Repo::Remove: name = #{name}", :yellow
+          if thor.yes? "Remove repo #{name}? (y/n)"
+            repo_dir = repo_dir(name)
+            if repo_dir && thor.yes?("Remove local repo dir #{repo_dir}? (y/n)")
+              thor.remove_file repo_dir
+            end
+            config[:repos].delete(name.to_sym)
+            thor.create_file(config.default_config_file, JSON.pretty_generate(config.data), :force => true)
+            config.reload
+            Tasks::Setup.new(@options).run unless @skip_setup
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/fhcap/tasks/setup.rb

@@ -0,0 +1,59 @@
+require 'fhcap/tasks/task_base'
+require 'fhcap/tasks/repo/add'
+require 'fhcap/tasks/knife/add'
+
+module Fhcap
+  module Tasks
+    class Setup < TaskBase
+
+      def initialize(options)
+        super
+        @fhcap_config = @config.exists? ? @config.data : @config.fhcap_config
+
+        opts = {
+            repos_dir: options[:'repos-dir'],
+        }.delete_if { |k, v| v.nil? || v.empty? }
+
+        @fhcap_config.merge!(opts)
+      end
+
+      def run
+        ask_config(required_config, @fhcap_config)
+        create_fhcap_config
+        create_fhcap_repos
+        create_knife_config
+      end
+
+      private
+
+      def required_config
+        {
+            repos_dir: {
+                msg: "Local directory where repos will be checked out",
+                options: {:path => true}
+            }
+        }
+      end
+
+      def create_fhcap_config
+        thor.create_file(config.default_config_file, JSON.pretty_generate(@fhcap_config))
+        config.reload
+      end
+
+      def create_fhcap_repos
+        repos = config[:repos]
+        repos.each do |name, cfg|
+          Repo::Checkout.new(@options.dup.merge({:repo => name.to_s})).run
+        end if repos
+      end
+
+      def create_knife_config
+        knife_config = config[:knife]
+        knife_config.each do |name, cfg|
+          Knife::Add.new(@options.dup.merge({:name => name.to_s, :url => cfg[:chef_server_url], :node_name => cfg[:node_name]})).run
+        end if knife_config
+      end
+
+    end
+  end
+end

data/lib/fhcap/tasks/task_base.rb

@@ -0,0 +1,89 @@
+require 'fhcap/fhcap_helper'
+require 'fhcap/repos_helper'
+require 'fhcap/providers_helper'
+require 'fhcap/knife_helper'
+require 'ruby-progressbar'
+
+module Fhcap
+  module Tasks
+    class TaskBase
+
+      include Fhcap::FhcapHelper
+      include Fhcap::ReposHelper
+      include Fhcap::ProvidersHelper
+      include Fhcap::KnifeHelper
+
+      attr_reader :options, :config, :thor, :verbose
+
+      def initialize(options)
+        @options = options
+        @config = options[:config]
+        @thor = options[:thor]
+        @verbose = options[:verbose]
+      end
+
+      def ask_config(required, config)
+        required.each do |key, cfg|
+          if !config[key] || options[:interactive]
+            msg = cfg[:msg] || "#{key}:"
+            options = cfg[:options] || {}
+            options[:default] = config[key]
+            if cfg[:boolean]
+              config[key] = thor.yes? "#{msg} y/n?"
+            else
+              config[key] = thor.ask msg, options
+            end
+          end
+        end
+      end
+
+      def exit_with_error(msg)
+        thor.say_status 'error', msg, :red
+        exit(-1)
+      end
+
+      def with_progress(title, &block)
+        long_running_task = fork(&block)
+
+        progress = fork do
+          progressbar = ProgressBar.create(title: title, total: nil)
+
+          trap "INT" do
+            progressbar.total = progressbar.progress + 1
+            progressbar.clear
+            exit
+          end
+
+          loop do
+            progressbar.increment
+            sleep 0.5
+          end
+        end
+
+        Process.wait(long_running_task)
+        Process.kill(2, progress)
+      end
+
+      def table_header(*args)
+        args.collect do |name|
+          set_color(name.to_s, :green)
+        end
+      end
+
+      def table_row(*args)
+        args.collect do |name|
+          color_pad(name.to_s)
+        end
+      end
+
+      def color_pad(string)
+        string + set_color("", :white)
+      end
+
+      def set_color(*args)
+        thor.shell.set_color(*args)
+      end
+
+    end
+  end
+end

data/lib/fhcap/thor_base.rb

@@ -0,0 +1,121 @@
+require 'thor'
+require 'fhcap'
+require 'fhcap/config'
+require 'fhcap/providers_helper'
+require 'fhcap/repos_helper'
+
+module Fhcap
+
+  class ThorBase < Thor
+
+    attr_reader :config, :thor
+
+    class << self
+      @@shared_options = {}
+      @@fhcap_config = Fhcap::Config.new(
+          :config_file => ENV['FHCAP_CFG_FILE']
+      )
+
+      def add_shared_option(name, options = {})
+        @@shared_options[name] = options
+      end
+
+      def shared_options(*option_names)
+        option_names.each do |option_name|
+          opt = @@shared_options[option_name]
+          raise "Tried to access shared option '#{option_name}' but it was not previously defined" if opt.nil?
+          option option_name, opt
+        end
+      end
+
+      def config_key_collection(name)
+        if @@fhcap_config.exists? && @@fhcap_config[name.to_sym]
+          @@fhcap_config[name.to_sym].collect do |key, val|
+            if block_given?
+              yield key.to_s, val
+            else
+              key.to_s
+            end
+          end
+        else
+          []
+        end
+      end
+
+      def provider_names
+        config_key_collection(:providers) do |provider, cfg|
+          cfg.collect do |id, pcfg|
+            "#{provider}:#{id}"
+          end
+        end.flatten
+      end
+
+      def provider_names_for(service)
+        config_key_collection(:providers) do |provider, cfg|
+          cfg.reject do |id, pcfg|
+            !pcfg[:provides] || !pcfg[:provides].include?(service)
+          end.collect do |id, pcfg|
+            "#{provider}:#{id}"
+          end
+        end.flatten
+      end
+
+      def chef_server_names
+        config_key_collection(:knife)
+      end
+
+      def repo_names
+        config_key_collection(:repos)
+      end
+
+      def cluster_template_names
+        Fhcap::TEMPLATE_NAMES
+      end
+
+      def run_knife_cmd(cmd, options)
+        cmd = (['knife', cmd]).join(' ')
+        run_cmd(cmd, options)
+      end
+
+      def run_cmd(cmd, options)
+        run(cmd)
+      end
+
+    end
+
+    include Thor::Actions
+    include Fhcap::ProvidersHelper
+    include Fhcap::ReposHelper
+
+    def initialize(*args)
+      super
+      self.class.source_root(Fhcap.source_root)
+      #$stdout.sync = true
+      @config = Fhcap::Config.new(
+          :config_file => ENV['FHCAP_CFG_FILE']
+      )
+      @thor = self
+    end
+
+    private
+
+    def task_options(options)
+      opts = {
+          :thor => self,
+          :config => @config
+      }.merge(options)
+      symbolize_keys_deep! opts
+      opts
+    end
+
+    def symbolize_keys_deep!(h)
+      h.keys.each do |k|
+        ks = k.to_sym
+        h[ks] = h.delete k
+        symbolize_keys_deep! h[ks] if h[ks].kind_of? Hash
+      end
+    end
+
+  end
+
+end

data/lib/fhcap/version.rb

@@ -0,0 +1,3 @@
+module Fhcap
+  VERSION = "0.3.0"
+end

data/spec/fhcap/cli_spec.rb

@@ -0,0 +1,6 @@
+require 'spec_helper'
+describe Fhcap::CLI::Cli do
+  describe "#initialize" do
+
+  end
+end

data/spec/fhcap/tasks/cluster/create_spec.rb

@@ -0,0 +1,46 @@
+require 'spec_helper'
+require "fhcap/tasks/cluster/create"
+
+describe Fhcap::Tasks::Cluster::Create do
+
+  subject {
+    Fhcap::Tasks::Cluster::Create.new(options)
+  }
+
+  let(:thor) do
+    DummyThor.new
+  end
+
+  let(:config) do
+    DummyConfig.new
+  end
+
+  let(:options) do
+    {
+        :config => config,
+        :thor => thor,
+        :name => 'testcluster',
+        :repo => 'testrepo',
+        :template => 'template',
+        :'git-ref' => 'master',
+        :domain => 'testdomain',
+        :'ssl-cert' => 'testcert',
+        :'dns-provider-id' => 'test',
+        :'chef-server' => 'test',
+        :'provider-id' => 'test',
+    }
+  end
+
+  describe "#initialize" do
+    specify { expect(subject.name).to eq(options[:name]) }
+    specify { expect(subject.instance_variable_get(:@gitref)).to eq(options[:'git-ref']) }
+    specify { expect(subject.instance_variable_get(:@ssl_cert)).to eq(options[:'ssl-cert']) }
+    specify { expect(subject.instance_variable_get(:@dns_provider_id)).to eq(options[:'dns-provider-id']) }
+    specify { expect(subject.cluster_config[:id]).to eq(options[:name]) }
+    specify { expect(subject.cluster_config[:template]).to eq(options[:template]) }
+    specify { expect(subject.cluster_config[:domain]).to eq(options[:domain]) }
+    specify { expect(subject.cluster_config[:chef_server]).to eq(options[:'chef-server']) }
+    specify { expect(subject.cluster_config[:provider_id]).to eq(options[:'provider-id']) }
+  end
+
+end

data/spec/fhcap/tasks/knife/add_spec.rb

@@ -0,0 +1,35 @@
+require 'spec_helper'
+require "fhcap/tasks/knife/add"
+
+describe Fhcap::Tasks::Knife::Add do
+
+  subject {
+    Fhcap::Tasks::Knife::Add.new(options)
+  }
+
+  let(:thor) do
+    DummyThor.new
+  end
+
+  let(:config) do
+    {:knife => {}}
+  end
+
+  let(:options) do
+    {
+        :config => config,
+        :thor => thor,
+        :name => 'testknife',
+        :url => 'https://example.com',
+        :'node-name' => 'testnode'
+    }
+  end
+
+  describe "#initialize" do
+    specify { expect(subject.name).to eq(options[:name]) }
+    specify { expect(subject.knife_config[:chef_server_url]).to eq(options[:url]) }
+    specify { expect(subject.knife_config[:node_name]).to eq(options[:'node-name']) }
+    specify { expect(subject.knife_config[:validation_client_name]).to eq('chef-validator') }
+  end
+
+end

data/spec/fhcap/tasks/knife/remove_spec.rb

@@ -0,0 +1,25 @@
+require 'spec_helper'
+require "fhcap/tasks/knife/remove"
+
+describe Fhcap::Tasks::Knife::Remove do
+
+  subject {
+    Fhcap::Tasks::Knife::Remove.new(options)
+  }
+
+  let(:thor) do
+    DummyThor.new
+  end
+
+  let(:options) do
+    {
+        :thor => thor,
+        :name => 'testrepo'
+    }
+  end
+
+  describe "#initialize" do
+    specify { expect(subject.name).to eq(options[:name]) }
+  end
+
+end