fhcap-cli 0.3.0

data/lib/fhcap/tasks/chef/cookbook/list.rb

@@ -0,0 +1,37 @@
+require 'fhcap/tasks/chef/chef_task_base'
+
+module Fhcap
+  module Tasks
+    module Chef
+      module Cookbook
+        class List < ChefTaskBase
+
+          def run
+            thor.say "Chef::Cookbook::List", :yellow
+            cookbooks = get_cookbooks(options, nil, repo_cookbook_paths('-cookbooks'))
+
+            table_contents = [table_header("Cookbook", "Version", "Artifact Version", "Artifact Build")]
+
+            cookbooks.each do |name|
+              cookbook = cookbook_loader.cookbooks_by_name[name]
+              meta = get_cookbook_meta(name)
+              artifact_attribute_filepath =  cookbook.attribute_filenames.find { |e| /artifact.rb/ =~ e }
+              artifact_version = "N/A"
+              artifact_build = "N/A"
+              artifact_path = "N/A"
+              if artifact_attribute_filepath
+                node = ::Chef::Node.new
+                node.from_file(artifact_attribute_filepath)
+                artifact_version = node.default[name]['version']
+                artifact_build = node.default[name]['build']
+              end
+              table_contents << table_row(name, meta.version, artifact_version, artifact_build)
+            end
+            thor.print_table table_contents
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/fhcap/tasks/chef/cookbook/update_changelog.rb

@@ -0,0 +1,63 @@
+require 'fhcap/tasks/chef/chef_task_base'
+
+module Fhcap
+  module Tasks
+    module Chef
+      module Cookbook
+        class UpdateChangelog < ChefTaskBase
+
+          attr_reader :name
+
+          def initialize(options)
+            super
+            @name = options[:cookbook]
+          end
+
+          def run
+            thor.say "Chef::Cookbook::UpdateChangelog cookbook = #{name}", :yellow
+            cookbook = cookbook_loader.cookbooks_by_name[name]
+            meta = get_cookbook_meta(name)
+
+            version = meta.version
+            changelog_file = cookbook.root_filenames.find { |e| /CHANGELOG.md/ =~ e }
+
+            if changelog_file
+              changelog = parse_changelog(changelog_file)
+
+              if changelog[version]
+                thor.say_status("error", "Already a Changelog entry for #{name} version #{version}", :red)
+              else
+                changelog[version] = options[:'changelog-entries'] || []
+                thor.template(File.join('templates', 'chef', 'cookbook', 'changelog.md.erb'), changelog_file, {:cookbook => name, :changelog => changelog})
+              end
+            else
+              thor.say_status("error", "No Changelog found for #{name}", :red)
+            end
+
+          end
+
+          private
+
+          def parse_changelog(file)
+            changelog = {}
+            lines = File.readlines(file)
+            current_version = nil
+            lines.each do |line|
+              if match = line.match(/^## (\d+.\d+.\d+):/)
+                current_version = version = match[1]
+                changelog[version] = []
+              elsif match = line.match(/^* (.+)$/)
+                entry = match[1]
+                if current_version
+                  changelog[current_version] << entry
+                end
+              end
+            end
+            changelog
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/fhcap/tasks/chef/cookbook/update_metadata.rb

@@ -0,0 +1,57 @@
+require 'fhcap/tasks/chef/chef_task_base'
+require 'fhcap/dummy_node'
+
+module Fhcap
+  module Tasks
+    module Chef
+      module Cookbook
+        class UpdateMetadata < ChefTaskBase
+
+          attr_reader :name
+
+          def initialize(options)
+            super
+            @name = options[:cookbook]
+          end
+
+          def run
+            thor.say "Chef::Cookbook::UpdateMetadata cookbook = #{name}", :yellow
+            cookbook = cookbook_loader.cookbooks_by_name[name]
+            generate_metadata(cookbook)
+          end
+
+          private
+
+          def generate_metadata(cookbook)
+
+            meta = get_cookbook_meta(name)
+
+            metadata_filepath = cookbook.metadata_filenames.first
+            attribute_filepaths = cookbook.attribute_filenames
+
+            attribute_filepaths.each() do |file|
+              node = Fhcap::DummyNode.new
+              node.from_file(file)
+              node_attrs = node.metadata_attributes
+
+              node_attrs.each() do |k, v|
+                if meta.attributes[k]
+                  node_attrs[k] = meta.attributes[k].merge(v)
+                end
+                node_attrs[k]["description"] = "" if node_attrs[k]['description'].nil? || node_attrs[k]['description'].empty?
+                meta.attribute(k, node_attrs[k])
+              end
+            end
+            meta_hash = meta.to_hash().merge({cookbook: name})
+
+            template_filepath = File.join(Fhcap.source_root, 'templates', 'chef', 'cookbook', 'metadata.erb')
+            eruby = Erubis::Eruby.new(File.read(template_filepath))
+            result = eruby.result(meta_hash)
+            thor.create_file(metadata_filepath, result, {force: true})
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/fhcap/tasks/chef/cookbook/update_readme.rb

@@ -0,0 +1,30 @@
+require 'fhcap/tasks/chef/chef_task_base'
+
+module Fhcap
+  module Tasks
+    module Chef
+      module Cookbook
+        class UpdateReadme < ChefTaskBase
+
+          attr_reader :name
+
+          def initialize(options)
+            super
+            @name = options[:cookbook]
+          end
+
+          def run
+            thor.say "Chef::Cookbook::UpdateReadme cookbook = #{name}", :yellow
+            cookbook = cookbook_loader.cookbooks_by_name[name]
+            metadata_file = cookbook.metadata_filenames.first
+            readme_file = cookbook.root_filenames.find { |e| /README/ =~ e }
+            cmd = "knife cookbook readme from #{metadata_file} > #{readme_file}"
+            thor.run(cmd)
+          end
+
+        end
+
+      end
+    end
+  end
+end

data/lib/fhcap/tasks/chef/cookbook/update_version.rb

@@ -0,0 +1,90 @@
+require 'fhcap/tasks/chef/chef_task_base'
+
+module Fhcap
+  module Tasks
+    module Chef
+      module Cookbook
+        class UpdateVersion < ChefTaskBase
+
+          attr_reader :name, :version, :level
+
+          def initialize(options)
+            super
+            @name = options[:cookbook]
+            @version = options[:version]
+            @level = ('manual' if @version) || ('patch' if options[:wip]) || options[:level] || 'minor'
+          end
+
+          def run
+            thor.say "Chef::Cookbook::UpdateVersion cookbook = #{name}", :yellow
+            cookbook = cookbook_loader.cookbooks_by_name[name]
+            metadata_file = cookbook.metadata_filenames.first
+
+            if is_dirty?(metadata_file)
+              thor.say_status("error", "#{metadata_file} has uncommitted changes. Please reset this file or commit any changes before continuing!", :red)
+            else
+              bump(cookbook, metadata_file)
+            end
+          end
+
+          private
+
+          ### Modified version of https://github.com/jonlives/knife-spork/blob/master/lib/chef/knife/spork-bump.rb
+
+          TYPE_INDEX = {:major => 0, :minor => 1, :patch => 2, :manual => 3}.freeze
+
+          def bump(cookbook, metadata_file)
+            old_version = cookbook.version
+
+            if bump_type == 3
+              # manual bump
+              version_array = manual_bump_version.split('.')
+            else
+              # major, minor, or patch bump
+              version_array = old_version.split('.').collect { |i| i.to_i }
+              version_array[bump_type] += 1
+              ((bump_type+1)..2).each { |i| version_array[i] = 0 } # reset all lower version numbers to 0
+            end
+
+            new_version = version_array.join('.')
+
+            new_contents = File.read(metadata_file).gsub(/(version\s+['"])[0-9\.]+(['"])/, "\\1#{new_version}\\2")
+            File.open(metadata_file, 'w') { |f| f.write(new_contents) }
+
+            thor.say "Successfully bumped #{cookbook.name} to v#{new_version}!"
+          end
+
+          def bump_type
+            TYPE_INDEX[(level || 'patch').to_sym]
+          end
+
+          def manual_bump_version
+            validate_version!(version)
+            version
+          end
+
+          def valid_version?(version)
+            version_keys = version.split('.')
+            return false unless version_keys.size == 3 && version_keys.any? { |k|
+              begin
+                Float(k);
+              rescue false;
+              else
+                true;
+              end }
+            true
+          end
+
+          def validate_version!(version)
+            if version && !valid_version?(version)
+              thor.say_status("error", "#{version} is not a valid version!", :red)
+            end
+          end
+
+          ###
+
+        end
+      end
+    end
+  end
+end

data/lib/fhcap/tasks/chef/environments/create.rb

@@ -0,0 +1,115 @@
+require 'fhcap/tasks/chef/chef_task_base'
+require 'sshkey'
+
+module Fhcap
+  module Tasks
+    module Chef
+      module Environments
+        class Create < ChefTaskBase
+
+          def initialize(options)
+            super
+            @name = options[:name]
+            @domain = options[:domain]
+            @type = options[:type]
+            @repo = options[:repo]
+            @passwords = {}
+          end
+
+          def run
+            thor.say "Chef::Environments::Create name = #{@name}, type = #{@type}, domain = #{@domain}", :yellow
+            env_config = {
+                name: @name,
+                domain: @domain,
+                skip: true
+            }
+            create_environment(env_config)
+            create_data_bag(env_config)
+          end
+
+          private
+
+          def create_environment(env_config)
+            environment_template = "environment_#{@type}.json.erb"
+            environment_file = File.join(repo_dir(@repo), 'environments', "#{@name}.json")
+            thor.template(File.join('templates', 'chef', environment_template), environment_file, env_config)
+          end
+
+          def create_data_bag(env_config)
+            environment_data_bag_file = File.join(repo_dir(@repo), 'data_bags', 'environments', "#{@name}.json")
+            env_data_bag_conf = generate_template_config(env_config)
+            current_config = File.exists?(environment_data_bag_file) ? JSON.parse(IO.read(environment_data_bag_file), {:symbolize_names => true}) : {}
+            env_data_bag_conf.merge!(current_config)
+            thor.create_file(environment_data_bag_file, JSON.pretty_generate(env_data_bag_conf))
+            passwords_file = File.join(repo_dir(@repo), repo_clusters_dir(@repo), "#{@name}_passwords.json")
+            thor.create_file(passwords_file, JSON.pretty_generate(@passwords), skip: true)
+          end
+
+          def random_password(name)
+            unless @passwords[name]
+              @passwords[name] = SecureRandom.urlsafe_base64(6)
+            end
+            @passwords[name]
+          end
+
+          def hashed_password(name)
+            `openssl passwd -1 #{random_password(name)}`.strip
+          end
+
+          def generate_template_config(env_config)
+            config = {
+                id: env_config[:name]
+            }
+
+            #users
+            users =  %w{hadmin henryd}
+            config[:users] = {}
+
+            users.each do |user|
+              config[:users][user.to_sym] = {}
+              key = SSHKey.generate(:comment => "#{user}@feedhenry")
+              config[:users][user.to_sym][:ssh_public_key] = key.ssh_public_key
+              config[:users][user.to_sym][:ssh_private_key] = key.private_key
+            end
+
+            config[:users][:hadmin][:password] = hashed_password('users.hadmin')
+            config[:users][:hadmin][:ssh_keys] = []
+
+            #ssh keys
+            ssh_key_users = %w{gitolite gitlab_shell repoadmin nagios_user}
+            ssh_key_users.each do |user|
+              config[user.to_sym] = {}
+              key = SSHKey.generate(:comment => "#{user}@feedhenry")
+              config[user.to_sym][:ssh_public_key] = key.ssh_public_key
+              config[user.to_sym][:ssh_private_key] = key.private_key
+            end
+            config[:users][:hadmin][:ssh_keys] << config[:nagios_user][:ssh_public_key]
+
+            #root_user
+            config[:root_user] = {password: hashed_password('root_user')}
+
+            #henry_users
+            config[:henry_users] = []
+
+            config[:henry_users] << {name: 'team.ops', password: hashed_password('henry_users.teamops')}
+            config[:henry_users] << {name: 'team.eng', password: hashed_password('henry_users.teameng')}
+            config[:henry_users] << {name: 'team.support', password: hashed_password('henry_users.teamsupport')}
+            config[:henry_users] << {name: 'team.qe', password: hashed_password('henry_users.qe')}
+            config[:henry_users] << {name: 'team.ps', password: hashed_password('henry_users.ps')}
+            config[:henry_users].uniq! { |u| u[:name] }
+
+            config[:mongodb] = {}
+            config[:mongodb][:shared_key] = "bim4jSl3AL0Dna7gU/4YOOIZtDOl4HmTFXR8Jz8EHnm7qTSRxm3NrvDIyvQmz3FZVHdyi9BgSBVmm2Yti74aOIbHLsX3l+tN+gNnu265uSC3o9qIVDNxwOFLTZc3TNtxgBD7BGCkwKMI/T4LVADuDlmZEcUHddZcpYbzmwMT47af3VzoN91yiRLXM/cwsU2ZtWCgg8gAbE1tOZ0pOUPLvXuJvREz7cena6cP0OaA/yvJ9V2sWaTPj3dr3tApPdRW5a2fPrVW/Q9k9krvEd/6+cabSNaIsIK1P1E1qvxNE+WfY2M6TwVN69o/0IfE9TbJp+n1kQzrOudYiI0r793uXAMFfVEqrhFKhs/q9gkZFOcgx0f8LnxLTGeV4umJJ4vs0wtaJoRBQzsDhZ7c0+Itw7jNTNu3mmAYqcVE3k4hTRZGKKH9r+f1l3s4H2esLQ6VFkRJ++wMKu6Rku1FR4c8d4/g74C+GPWjrexUhgTJZWgvqGwA4NsgCfV3rkybHnhOvph3B+dKwJYc3QegeN5fk2ct4EIaRIKsArd+SNJbRkXSkmCunkiZwzcWEqbxXHbsqXRH/DT2/03aDvA9qUUH08/J1ILdJxIbjiOKG6pCT0cWgyn2DqLYh9XB7JyofJYDsUThZYKEFnnSnqgcHKbajKUuQMX82Znj50g2Dfry9SQK5E/zMKF2XE56+cvOowcqziel08Oyc/poH78t6izpKoVT98CjPuaj7CYB6qgUr4xFFToTDHDYI9kQ0cLLmIUcYYkCNXciFCcViC0EatUcaNduA25lIYO7Zy1t5BN3gLtt1fgMhvqWqM37EjcJuGcSdiOaShlFnb4cbHKFFmcJy05/NkA3GwZAWJ37sm27vxt1g08JXwqMdrAjnaPp31v03N10PgGSDa2PjZa2AG9Hg44ZwrfLU5ftaluClhZQYUtCmS0OHeuonULicQWHF3qxUYx5w668HP5QyvXyzlYGkw1WhY9i"
+
+            config[:admin_contacts] = []
+
+            nocAlerts = {email: 'noc.alerts@example.com', nagios: true, syslog_ng_watchdog: true, backup_watchdog: true, forward: true, crons: true}
+            config[:admin_contacts] << nocAlerts
+            config
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/fhcap/tasks/chef/environments/destroy.rb

@@ -0,0 +1,37 @@
+require 'fhcap/tasks/chef/chef_task_base'
+
+module Fhcap
+  module Tasks
+    module Chef
+      module Environments
+        class Destroy < ChefTaskBase
+
+          attr_reader :name, :chef_server
+
+          def initialize(options)
+            super
+            @name = options[:name]
+            @repo = options[:repo]
+            @chef_server = options[:chef_server]
+          end
+
+          def run
+            thor.say "Chef::Environments::Destroy #{@name}", :yellow
+
+            if @chef_server
+              if thor.yes? "Remove all environments and config from the chef server (#{@chef_server})?"
+                run_knife_cmd("environment delete #{@name} -y", @chef_server)
+                run_knife_cmd("data bag delete environments #{@name} -y", @chef_server)
+              end
+            end
+
+            thor.remove_file find_data_bag_item('environments', @name)
+            thor.remove_file find_environment(@name)
+            thor.remove_file find_repo_item(repo_clusters_dir(@repo), "#{@name}_passwords.json").first
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/fhcap/tasks/chef/environments/promote_cookbooks.rb

@@ -0,0 +1,47 @@
+require 'fhcap/tasks/chef/chef_task_base'
+
+module Fhcap
+  module Tasks
+    module Chef
+      module Environments
+        class PromoteCookbooks < ChefTaskBase
+
+          attr_reader :environments, :cookbooks
+
+          def initialize(options)
+            super
+            @environments = options[:environments] || []
+            @cookbooks = options[:cookbooks] || []
+          end
+
+          def run
+            cookbooks = get_cookbooks(options)
+            thor.say "Chef::Environments::PromoteCookbooks: environments = #{environments}", :yellow
+            if environments.empty?
+              exit_with_error("You must specify at least one environment during promote!")
+            else
+              repo_cookbook_versions = get_cookbook_versions(options) do |cookbook, version|
+                thor.say("Adding version constraint #{cookbook} = #{version}") if verbose
+              end
+
+              environments.each do |environment|
+                environment_filepath = find_environment(environment)
+                if environment_filepath
+                  environment_data = JSON.parse(IO.read(environment_filepath))
+                  env_cookbook_versions = environment_data['cookbook_versions'] || {}
+                  cookbooks.each do |cookbook|
+                    env_cookbook_versions[cookbook] = "= #{repo_cookbook_versions[cookbook]}"
+                  end
+                  thor.create_file(environment_filepath, JSON.pretty_generate(environment_data), :force => true)
+                else
+                  thor.say "Unknown environment #{environment}", :red
+                end
+              end
+            end
+          end
+
+        end
+      end
+    end
+  end
+end