terraform-wrapper 0.0.2

data/lib/terraform-wrapper/tasks/clean.rb

@@ -0,0 +1,141 @@
+###############################################################################
+
+require 'rake/tasklib'
+
+###############################################################################
+
+module TerraformWrapper
+
+###############################################################################
+
+  module Tasks
+
+###############################################################################
+
+    class Clean < ::Rake::TaskLib
+
+###############################################################################
+
+      @@glob_directories = {
+        "Terraform Modules" => "**/.terraform"
+      }
+
+###############################################################################
+
+      @@glob_files = {
+        "Crash Logs"              => "**/crash.log",
+        "Graph Images"            => "**/graph.png",
+        "Terraform States"        => "**/*.tfstate",
+        "Terraform State Backups" => "**/*.tfstate.backup"
+      }
+
+###############################################################################
+
+      @code
+
+###############################################################################
+
+      def initialize(code:)
+        @code = code
+
+        yield self if block_given?
+
+        clean_task
+      end
+
+###############################################################################
+
+      def clean_task
+        desc "Cleans the Terraform workspace of any downloaded providers and modules."
+        task :clean do |t, args|
+          $logger.info("Cleaning Terraform component: #{@code.name}...")
+
+          directory_counter = 0
+          @@glob_directories.each do |key, value|
+            $logger.info("Cleaning directories: #{key}")
+            directory_item_counter = 0
+
+            directories = Dir.glob(File.join(@code.path, value))
+            directories.each do |directory|
+              if delete_directory(directory: directory) then
+                directory_item_counter += 1
+                directory_counter      += 1
+              end
+            end
+            $logger.info("Cleaned: #{directory_item_counter.to_s} directories.")
+          end
+
+          file_counter = 0
+          @@glob_files.each do |key, value|
+            $logger.info("Cleaning files: #{key}")
+            file_item_counter = 0
+
+            files = Dir.glob(File.join(@code.path, value))
+            files.each do |file|
+              if delete_file(file: file) then
+                file_item_counter += 1
+                file_counter      += 1
+              end
+            end
+            $logger.info("Cleaned: #{file_item_counter.to_s} files.")
+          end
+
+          $logger.info("Cleaned total: #{directory_counter.to_s} directories and: #{file_counter.to_s} files.")
+        end
+      end
+
+###############################################################################
+
+    private
+
+###############################################################################
+
+    def delete_directory(directory:)
+      result = false
+
+      if File.directory?(directory) then
+        begin
+          FileUtils.remove_dir(directory)
+          result = true
+          $logger.info("Deleted directory: #{directory}")
+        rescue
+          result = false
+          $logger.warn("Failed to delete directory: #{directory}")
+        end
+      end
+
+      return result
+    end
+
+###############################################################################
+
+    def delete_file(file:)
+      result = false
+
+      if File.file?(file) then
+        begin
+          File.delete(file)
+          result = true
+          $logger.info("Deleted file: #{file}")
+        rescue
+          result = false
+          $logger.warn("Failed to delete file: #{file}")
+        end
+      end
+
+      return result
+    end
+
+###############################################################################
+
+    end
+
+###############################################################################
+
+  end
+
+###############################################################################
+
+end
+
+###############################################################################

data/lib/terraform-wrapper/tasks/destroy.rb

@@ -0,0 +1,68 @@
+###############################################################################
+
+require 'rake/tasklib'
+
+###############################################################################
+
+module TerraformWrapper
+
+###############################################################################
+
+  module Tasks
+
+###############################################################################
+
+    class Destroy < ::Rake::TaskLib
+
+###############################################################################
+
+      @backend
+      @binary
+      @code
+      @configs
+      @overrides
+      @service
+
+###############################################################################
+
+      def initialize(backend:, binary:, code:, configs:, overrides:, service:)
+        @backend   = backend
+        @binary    = binary
+        @code      = code
+        @configs   = configs
+        @overrides = overrides
+        @service   = service
+
+        yield self if block_given?
+
+        destroy_task
+      end
+
+###############################################################################
+
+      def destroy_task
+        desc "Destroys infrastructure with Terraform for a given configuration on a workspace."
+        task :destroy, [:config] => :binary do |t, args|
+          $logger.info("Running Terraform destroy for service: #{@service}, component: #{@code.name}...")
+
+          config = TerraformWrapper::Shared::Config.new(backend: @backend, base: @configs, code: @code, name: args[:config], overrides: @overrides, service: @service)
+          runner = TerraformWrapper::Shared::Runner.new(binary: @binary, code: @code)
+
+          runner.init(config: config)
+          runner.destroy
+        end
+      end
+
+###############################################################################
+
+    end
+
+###############################################################################
+
+  end
+
+###############################################################################
+
+end
+
+###############################################################################

data/lib/terraform-wrapper/tasks/init.rb

@@ -0,0 +1,67 @@
+###############################################################################
+
+require 'rake/tasklib'
+
+###############################################################################
+
+module TerraformWrapper
+
+###############################################################################
+
+  module Tasks
+
+###############################################################################
+
+    class Init < ::Rake::TaskLib
+
+###############################################################################
+
+      @backend
+      @binary
+      @code
+      @configs
+      @overrides
+      @service
+
+###############################################################################
+
+      def initialize(backend:, binary:, code:, configs:, overrides:, service:)
+        @backend   = backend
+        @binary    = binary
+        @code      = code
+        @configs   = configs
+        @overrides = overrides
+        @service   = service
+
+        yield self if block_given?
+
+        init_task
+      end
+
+###############################################################################
+
+      def init_task
+        desc "Initialises the Terraform workspace and remote backend (if applicable)."
+        task :init, [:config] => :binary do |t, args|
+          $logger.info("Initialising Terraform for service: #{@service}, component: #{@code.name}...")
+
+          config = TerraformWrapper::Shared::Config.new(backend: @backend, base: @configs, code: @code, name: args[:config], overrides: @overrides, service: @service)
+          runner = TerraformWrapper::Shared::Runner.new(binary: @binary, code: @code)
+
+          runner.init(config: config)
+        end
+      end
+
+###############################################################################
+
+    end
+
+###############################################################################
+
+  end
+
+###############################################################################
+
+end
+
+###############################################################################

data/lib/terraform-wrapper/tasks/plan.rb

@@ -0,0 +1,68 @@
+###############################################################################
+
+require 'rake/tasklib'
+
+###############################################################################
+
+module TerraformWrapper
+
+###############################################################################
+
+  module Tasks
+
+###############################################################################
+
+    class Plan < ::Rake::TaskLib
+
+###############################################################################
+
+      @backend
+      @binary
+      @code
+      @configs
+      @overrides
+      @service
+
+###############################################################################
+
+      def initialize(backend:, binary:, code:, configs:, overrides:, service:)
+        @backend   = backend
+        @binary    = binary
+        @code      = code
+        @configs   = configs
+        @overrides = overrides
+        @service   = service
+
+        yield self if block_given?
+
+        plan_task
+      end
+
+###############################################################################
+
+      def plan_task
+        desc "Runs a Terraform plan for a given configuration on a workspace."
+        task :plan, [:config, :out] => :binary do |t, args|
+          $logger.info("Running Terraform plan for service: #{@service}, component: #{@code.name}...")
+
+          config = TerraformWrapper::Shared::Config.new(backend: @backend, base: @configs, code: @code, name: args[:config], overrides: @overrides, service: @service)
+          runner = TerraformWrapper::Shared::Runner.new(binary: @binary, code: @code)
+
+          runner.init(config: config)
+          runner.plan(file: args[:out])
+        end
+      end
+
+###############################################################################
+
+    end
+
+###############################################################################
+
+  end
+
+###############################################################################
+
+end
+
+###############################################################################

data/lib/terraform-wrapper/tasks/plandestroy.rb

@@ -0,0 +1,68 @@
+###############################################################################
+
+require 'rake/tasklib'
+
+###############################################################################
+
+module TerraformWrapper
+
+###############################################################################
+
+  module Tasks
+
+###############################################################################
+
+    class PlanDestroy < ::Rake::TaskLib
+
+###############################################################################
+
+      @backend
+      @binary
+      @code
+      @configs
+      @overrides
+      @service
+
+###############################################################################
+
+      def initialize(backend:, binary:, code:, configs:, overrides:, service:)
+        @backend   = backend
+        @binary    = binary
+        @code      = code
+        @configs   = configs
+        @overrides = overrides
+        @service   = service
+
+        yield self if block_given?
+
+        plan_destroy_task
+      end
+
+###############################################################################
+
+      def plan_destroy_task
+        desc "Runs a Terraform destroy plan for a given configuration on a workspace."
+        task :"plan-destroy", [:config, :out] => :binary do |t, args|
+          $logger.info("Running Terraform destroy plan for service: #{@service}, component: #{@code.name}...")
+
+          config = TerraformWrapper::Shared::Config.new(backend: @backend, base: @configs, code: @code, name: args[:config], overrides: @overrides, service: @service)
+          runner = TerraformWrapper::Shared::Runner.new(binary: @binary, code: @code)
+
+          runner.init(config: config)
+          runner.plan(destroy: true, file: args[:out])
+        end
+      end
+
+###############################################################################
+
+    end
+
+###############################################################################
+
+  end
+
+###############################################################################
+
+end
+
+###############################################################################

data/lib/terraform-wrapper/tasks/validate.rb

@@ -0,0 +1,59 @@
+###############################################################################
+
+require 'rake/tasklib'
+
+###############################################################################
+
+module TerraformWrapper
+
+###############################################################################
+
+  module Tasks
+
+###############################################################################
+
+    class Validate < ::Rake::TaskLib
+
+###############################################################################
+
+      @binary
+      @code
+
+###############################################################################
+
+      def initialize(binary:, code:)
+        @binary = binary
+        @code   = code
+
+        yield self if block_given?
+
+        validate_task
+      end
+
+###############################################################################
+
+      def validate_task
+        desc "Validates the Terraform code within a workspace."
+        task :validate => :binary do |t, args|
+          $logger.info("Validating Terraform component: #{@code.name}...")
+
+          runner = TerraformWrapper::Shared::Runner.new(binary: @binary, code: @code)
+
+          runner.download
+          runner.validate
+        end
+      end
+
+###############################################################################
+
+    end
+
+###############################################################################
+
+  end
+
+###############################################################################
+
+end
+
+###############################################################################