galvanize 0.1.2

data/lib/chef-dk/skeletons/code_generator/recipes/cookbook.rb

@@ -0,0 +1,161 @@
+# frozen_string_literal: true
+
+context = ChefDK::Generator.context
+cookbook_dir = File.join(context.cookbook_root, context.cookbook_name)
+
+silence_chef_formatter unless context.verbose
+
+generator_desc('Ensuring correct cookbook file content')
+
+# cookbook root dir
+directory cookbook_dir
+
+# metadata.rb
+spdx_license =  case context.license
+                when 'apachev2'
+                  'Apache-2.0'
+                when 'mit'
+                  'MIT'
+                when 'gplv2'
+                  'GPL-2.0'
+                when 'gplv3'
+                  'GPL-3.0'
+                else
+                  'All Rights Reserved'
+                end
+
+template "#{cookbook_dir}/metadata.rb" do
+  helpers(ChefDK::Generator::TemplateHelper)
+  variables(
+    spdx_license: spdx_license
+  )
+  action :create_if_missing
+end
+
+# README
+template "#{cookbook_dir}/README.md" do
+  helpers(ChefDK::Generator::TemplateHelper)
+  action :create_if_missing
+end
+
+# CHANGELOG
+template "#{cookbook_dir}/CHANGELOG.md" do
+  helpers(ChefDK::Generator::TemplateHelper)
+  action :create_if_missing
+end
+
+# chefignore
+cookbook_file "#{cookbook_dir}/chefignore"
+
+if context.use_berkshelf
+
+  # Berks
+  cookbook_file "#{cookbook_dir}/Berksfile" do
+    action :create_if_missing
+  end
+else
+
+  # Policyfile
+  template "#{cookbook_dir}/Policyfile.rb" do
+    source 'Policyfile.rb.erb'
+    helpers(ChefDK::Generator::TemplateHelper)
+  end
+
+end
+
+# LICENSE
+template "#{cookbook_dir}/LICENSE" do
+  helpers(ChefDK::Generator::TemplateHelper)
+  source "LICENSE.#{context.license}.erb"
+  action :create_if_missing
+end
+
+# Test Kitchen
+template "#{cookbook_dir}/kitchen.yml" do
+  if context.kitchen == 'dokken'
+    # kitchen-dokken configuration works with berkshelf and policyfiles
+    source 'kitchen_dokken.yml.erb'
+  elsif context.use_berkshelf
+    source 'kitchen.yml.erb'
+  else
+    source 'kitchen_policyfile.yml.erb'
+  end
+
+  helpers(ChefDK::Generator::TemplateHelper)
+  action :create_if_missing
+end
+
+# Inspec
+directory "#{cookbook_dir}/test/integration/default" do
+  recursive true
+end
+
+template "#{cookbook_dir}/test/integration/default/default_test.rb" do
+  source 'inspec_default_test.rb.erb'
+  helpers(ChefDK::Generator::TemplateHelper)
+  action :create_if_missing
+end
+
+# Chefspec
+directory "#{cookbook_dir}/spec/unit/recipes" do
+  recursive true
+end
+
+cookbook_file "#{cookbook_dir}/spec/spec_helper.rb" do
+  if context.use_berkshelf
+    source 'spec_helper.rb'
+  else
+    source 'spec_helper_policyfile.rb'
+  end
+
+  action :create_if_missing
+end
+
+template "#{cookbook_dir}/spec/unit/recipes/default_spec.rb" do
+  source 'recipe_spec.rb.erb'
+  helpers(ChefDK::Generator::TemplateHelper)
+  action :create_if_missing
+end
+
+# Recipes
+
+directory "#{cookbook_dir}/recipes"
+
+template "#{cookbook_dir}/recipes/default.rb" do
+  source 'recipe.rb.erb'
+  helpers(ChefDK::Generator::TemplateHelper)
+  action :create_if_missing
+end
+
+# git
+if context.have_git
+  unless context.skip_git_init
+
+    generator_desc('Committing cookbook files to git')
+
+    execute('initialize-git') do
+      command('git init .')
+      cwd cookbook_dir
+    end
+
+  end
+
+  cookbook_file "#{cookbook_dir}/.gitignore" do
+    source 'gitignore'
+  end
+
+  unless context.skip_git_init
+
+    execute('git-add-new-files') do
+      command('git add .')
+      cwd cookbook_dir
+    end
+
+    execute('git-commit-new-files') do
+      command('git commit -m "Add generated cookbook content"')
+      cwd cookbook_dir
+    end
+  end
+end
+
+include_recipe '::build_cookbook' if context.enable_delivery

data/lib/chef-dk/skeletons/code_generator/recipes/cookbook_file.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+context = ChefDK::Generator.context
+cookbook_dir = File.join(context.cookbook_root, context.cookbook_name)
+files_dir = File.join(cookbook_dir, 'files', 'default')
+cookbook_file_path = File.join(cookbook_dir, 'files', 'default', context.new_file_basename)
+
+directory files_dir do
+  recursive true
+end
+
+if context.content_source
+
+  file cookbook_file_path do
+    content(IO.read(context.context_source))
+  end
+
+else
+
+  template cookbook_file_path do
+    source 'cookbook_file.erb'
+    helpers(ChefDK::Generator::TemplateHelper)
+  end
+
+end

data/lib/chef-dk/skeletons/code_generator/recipes/habichef.rb

@@ -0,0 +1,92 @@
+context = ChefDK::Generator.context
+habitat_dir = File.join context.habichef_root, context.habichef_name
+
+# silence_chef_formatter unless context.verbose
+
+generator_desc 'Ensuring correct HabiChef project content'
+
+spdx_license =  case context.license
+                when 'apachev2'
+                  'Apache-2.0'
+                when 'mit'
+                  'MIT'
+                when 'gplv2'
+                  'GPL-2.0'
+                when 'gplv3'
+                  'GPL-3.0'
+                else
+                  'All Rights Reserved'
+                end
+
+# cookbook root dir
+directory habitat_dir
+# rest of dirs
+%w(habitat policyfiles).each do |dir|
+  directory File.join habitat_dir, dir
+end
+
+# our base files
+%w(
+  habitat/default.toml
+  habitat/plan.sh
+  habitat/README.md
+  policyfiles/base.rb
+).each do |name|
+  template File.join habitat_dir, name do
+    helpers(ChefDK::Generator::TemplateHelper)
+    source "#{name}.erb"
+    action :create_if_missing
+  end
+end
+
+# LICENSE
+template "#{habitat_dir}/LICENSE" do
+  helpers(ChefDK::Generator::TemplateHelper)
+  source "LICENSE.#{context.license}.erb"
+  action :create_if_missing
+end
+
+# CHANGELOG
+template "#{habitat_dir}/CHANGELOG.md" do
+  helpers(ChefDK::Generator::TemplateHelper)
+  action :create_if_missing
+end
+
+# Inspec
+directory "#{habitat_dir}/test/integration/default" do
+  recursive true
+end
+
+template "#{habitat_dir}/test/integration/default/default_test.rb" do
+  source 'inspec_default_test.rb.erb'
+  helpers(ChefDK::Generator::TemplateHelper)
+  action :create_if_missing
+end
+
+# git
+if context.have_git
+  unless context.skip_git_init
+    generator_desc('Committing cookbook files to git')
+
+    execute('initialize-git') do
+      command('git init .')
+      cwd habitat_dir
+    end
+  end
+
+  template "#{habitat_dir}/.gitignore" do
+    source 'gitignore'
+  end
+
+  unless context.skip_git_init
+    execute('git-add-new-files') do
+      command('git add .')
+      cwd habitat_dir
+    end
+
+    execute('git-commit-new-files') do
+      command('git commit -m "Add generated HabiChef content"')
+      cwd habitat_dir
+    end
+  end
+end

data/lib/chef-dk/skeletons/code_generator/recipes/helpers.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+context = ChefDK::Generator.context
+cookbook_dir = File.join(context.cookbook_root, context.cookbook_name)
+libraries_dir = File.join(cookbook_dir, 'libraries')
+helpers_path = File.join(cookbook_dir, 'libraries', "#{context.new_file_basename}.rb")
+
+directory libraries_dir
+
+def camelize(name)
+  name.to_s.split(/[^a-zA-Z]/).map(&:capitalize).join
+end
+
+cookbook_class_name = camelize(context.cookbook_name)
+helper_class_name = "#{camelize(context.new_file_basename)}Helpers"
+
+template helpers_path do
+  source 'helpers.rb.erb'
+  helpers(ChefDK::Generator::TemplateHelper)
+  variables(cookbook_class_name: cookbook_class_name, helper_class_name: helper_class_name)
+end

data/lib/chef-dk/skeletons/code_generator/recipes/policyfile.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+context = ChefDK::Generator.context
+policyfile_path = File.join(context.policyfile_dir, "#{context.new_file_basename}.rb")
+
+template policyfile_path do
+  source 'Policyfile.rb.erb'
+  helpers(ChefDK::Generator::TemplateHelper)
+end

data/lib/chef-dk/skeletons/code_generator/recipes/recipe.rb

@@ -0,0 +1,52 @@
+# frozen_string_literal: true
+
+context = ChefDK::Generator.context
+cookbook_dir = File.join(context.cookbook_root, context.cookbook_name)
+recipe_path = File.join(cookbook_dir, 'recipes', "#{context.new_file_basename}.rb")
+spec_helper_path = File.join(cookbook_dir, 'spec', 'spec_helper.rb')
+spec_dir = File.join(cookbook_dir, 'spec', 'unit', 'recipes')
+spec_path = File.join(spec_dir, "#{context.new_file_basename}_spec.rb")
+inspec_dir = File.join(cookbook_dir, 'test', 'integration', 'default')
+inspec_path = File.join(inspec_dir, "#{context.new_file_basename}_test.rb")
+
+if File.directory?(File.join(cookbook_dir, 'test', 'recipes'))
+  Chef::Log.deprecation <<~EOH
+    It appears that you have InSpec tests located at "test/recipes". This location can
+    cause issues with Foodcritic and has been deprecated in favor of "test/integration/default".
+    Please move your existing InSpec tests to the newly created "test/integration/default"
+    directory, and update the 'inspec_tests' value in your kitchen.yml file(s) to
+    point to "test/integration/default".
+  EOH
+end
+
+# Chefspec
+directory spec_dir do
+  recursive true
+end
+
+cookbook_file spec_helper_path do
+  action :create_if_missing
+end
+
+template spec_path do
+  source 'recipe_spec.rb.erb'
+  helpers(ChefDK::Generator::TemplateHelper)
+  action :create_if_missing
+end
+
+# Inspec
+directory inspec_dir do
+  recursive true
+end
+
+template inspec_path do
+  source 'inspec_default_test.rb.erb'
+  helpers(ChefDK::Generator::TemplateHelper)
+  action :create_if_missing
+end
+
+# Recipe
+template recipe_path do
+  source 'recipe.rb.erb'
+  helpers(ChefDK::Generator::TemplateHelper)
+end

data/lib/chef-dk/skeletons/code_generator/recipes/repo.rb

@@ -0,0 +1,68 @@
+# frozen_string_literal: true
+context = ChefDK::Generator.context
+repo_dir = File.join(context.repo_root, context.repo_name)
+
+# repo root dir
+directory repo_dir
+
+# Top level files
+template "#{repo_dir}/LICENSE" do
+  source "LICENSE.#{context.license}.erb"
+  helpers(ChefDK::Generator::TemplateHelper)
+  action :create_if_missing
+end
+
+cookbook_file "#{repo_dir}/.chef-repo.txt" do
+  source 'repo/dot-chef-repo.txt'
+  action :create_if_missing
+end
+
+cookbook_file "#{repo_dir}/README.md" do
+  source 'repo/README.md'
+  action :create_if_missing
+end
+
+cookbook_file "#{repo_dir}/chefignore" do
+  source 'chefignore'
+  action :create_if_missing
+end
+
+directories_to_create = %w( cookbooks data_bags )
+
+directories_to_create += if context.use_roles
+                           %w( roles environments )
+                         else
+                           %w( policyfiles )
+                         end
+
+directories_to_create.each do |tlo|
+  remote_directory "#{repo_dir}/#{tlo}" do
+    source "repo/#{tlo}"
+    action :create_if_missing
+  end
+end
+
+cookbook_file "#{repo_dir}/cookbooks/README.md" do
+  if context.policy_only
+    source 'cookbook_readmes/README-policy.md'
+  else
+    source 'cookbook_readmes/README.md'
+  end
+  action :create_if_missing
+end
+
+# git
+if context.have_git
+  unless context.skip_git_init
+    execute('initialize-git') do
+      command('git init .')
+      cwd repo_dir
+      not_if { File.exist?("#{repo_dir}/.gitignore") }
+    end
+  end
+  template "#{repo_dir}/.gitignore" do
+    source 'repo/gitignore.erb'
+    helpers(ChefDK::Generator::TemplateHelper)
+    action :create_if_missing
+  end
+end

data/lib/chef-dk/skeletons/code_generator/recipes/resource.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+context = ChefDK::Generator.context
+cookbook_dir = File.join(context.cookbook_root, context.cookbook_name)
+
+resource_dir = File.join(cookbook_dir, 'resources')
+resource_path = File.join(resource_dir, "#{context.new_file_basename}.rb")
+
+directory resource_dir
+
+template resource_path do
+  source 'resource.rb.erb'
+  helpers(ChefDK::Generator::TemplateHelper)
+end