dh-proteus 0.1.3

data/lib/proteus/commands/move.rb

@@ -0,0 +1,28 @@
+module Proteus
+  module Commands
+    module Move
+      def self.included(thor_class)
+        thor_class.class_eval do
+
+          desc "move FROM TO", "Moves an existing resource within the Terraform state"
+          def move(from, to)
+            init(verbose: parent_options[:verbose])
+            confirm question: "Do you really want to move #{from} to #{to} in context '(#{context}, #{environment})'?", color: :on_red, exit_code: 0 do
+
+              state_move_command = <<~STATE_MOVE_COMMAND
+                cd #{context_path(context)} && \
+                terraform state mv \
+                -var-file=#{var_file(context, environment)} \
+                #{aws_profile} \
+                #{from} \
+                #{to}
+              STATE_MOVE_COMMAND
+              syscall state_move_command.squeeze(' '), dryrun:  dryrun
+            end
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/proteus/commands/output.rb

@@ -0,0 +1,29 @@
+module Proteus
+  module Commands
+    module Output
+      def self.included(thor_class)
+        thor_class.class_eval do
+
+          desc "output", "Query Terraform outputs"
+          long_desc <<-LONGDESC
+            Query Terraform outputs.
+          LONGDESC
+          option :name, type: :string, default: nil, required: false
+          def output
+
+            init(verbose: parent_options[:verbose])
+
+            terraform_command = <<~TERRAFORM_COMMAND
+              cd #{context_path(context)} && \
+              terraform output \
+              -state=#{state_file(context, environment)} \
+              #{options[:name] ? options[:name] : ''}
+            TERRAFORM_COMMAND
+
+            syscall(terraform_command.squeeze(' '))
+          end
+        end
+      end
+    end
+  end
+end

data/lib/proteus/commands/plan.rb

@@ -0,0 +1,55 @@
+module Proteus
+  module Commands
+    module Plan
+      def self.included(thor_class)
+        thor_class.class_eval do
+
+          desc "plan", "Runs terraform plan"
+          long_desc <<-LONGDESC
+            Runs terraform plan.
+
+            With --limit option, the plan run will be limited to the specified targets
+          LONGDESC
+          option :limit, type: :array, aliases: "-l", default: []
+          option :destroy, type: :boolean, default: false
+          def plan
+            module_manager = Proteus::Modules::Manager.new(context: context, environment: environment)
+            module_manager.clean
+            module_manager.render_modules
+
+            fmt_cmd = <<~FMT_CMD
+              cd #{context_path(context)} && \
+              terraform fmt -list=true .
+            FMT_CMD
+
+            fmt_output = syscall(
+              fmt_cmd.squeeze(' '),
+              capture: true,
+              suppress: true
+            )
+
+            say "Formatted files:", :green
+            say fmt_output, :green
+
+            init(verbose: parent_options[:verbose]) if not dryrun
+
+            terraform_command = <<~TERRAFORM_COMMAND
+              cd #{context_path(context)} && \
+              terraform plan \
+              #{"-destroy" if options[:destroy]} \
+              -input=false \
+              -refresh=true \
+              -module-depth=-1 \
+              -var-file=#{var_file(context, environment)} \
+              -out=#{plan_file(context, environment)} \
+              #{aws_profile} #{limit(options[:limit])} \
+              #{context_path(context)}
+            TERRAFORM_COMMAND
+
+             syscall terraform_command.squeeze(' '), dryrun: dryrun
+          end
+        end
+      end
+    end
+  end
+end

data/lib/proteus/commands/remove.rb

@@ -0,0 +1,71 @@
+module Proteus
+  module Commands
+    module Remove
+      def self.included(thor_class)
+        thor_class.class_eval do
+
+          desc "remove", "Remove a resource from the terraform state"
+          long_desc <<-LONGDESC
+            Remove a resource from the terraform state
+            --bulk Enables bulk import mode
+
+            --resource_address Terraform address of resource to remove from the terraform state
+
+            --resources_file File containing resource addresses and identifiers
+          LONGDESC
+          option :bulk, type: :boolean, aliases: "-b", required: false, default: false
+          option :resource_address, type: :string, aliases: "-a", required: false, default: nil
+          option :resources_file, type: :string, aliases: "-f", required: false, default: nil
+          def remove
+            if options[:bulk]
+              if !options[:resources_file]
+                say "Supply a file containing resource identifiers and Terraform addresses for bulk operations", :red
+                exit 1
+              end
+            else
+              if !options[:resource_address]
+                say "You need to supply a resource address.", :red
+                exit 1
+              end
+            end
+
+            init(verbose: parent_options[:verbose])
+
+            confirm question: "Do you really want to run 'terraform state rm' in context '(#{context}, #{environment})'?", color: :on_red, exit_code: 0 do
+              state_remove_command = <<~STATE_REMOVE_COMMAND
+                cd #{context_path(context)} && \
+                terraform state rm \
+                -var-file=#{var_file(context, environment)} \
+                #{aws_profile} \
+                %{resource_addresses}
+              STATE_REMOVE_COMMAND
+
+              if options[:bulk]
+                if File.file?(options[:resources_file])
+                  File.open(options[:resources_file], "r") do |file|
+                    resource_addresses = []
+                    file.each_line do |line|
+                      resource = line.chomp.split(" = ")
+                      resource_addresses << resource[0]
+                    end
+
+                    resource_addresses.each_slice(500) do |slice|
+                      syscall (state_remove_command % { resource_addresses: slice.join(' ') }).squeeze(' ')
+                    end
+
+                  end
+                else
+                  say "File #{options[:resources_file]} does not exist.", :red
+                  exit 1
+                end
+              else
+                syscall (state_remove_command % { resource_addresses: options[:resource_address] })
+              end
+            end
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/proteus/commands/render.rb

@@ -0,0 +1,36 @@
+require 'proteus/modules/manager'
+
+module Proteus
+  module Commands
+    module Render
+      def self.included(thor_class)
+        thor_class.class_eval do
+
+          desc "render", "Renders the module templates"
+          long_desc <<-LONGDESC
+            Renders the module templates without running Terraform
+          LONGDESC
+          def render
+            render_backend
+            module_manager = Proteus::Modules::Manager.new(context: context, environment: environment)
+            module_manager.render_modules
+
+            fmt_cmd = <<~FMT_CMD
+              cd #{context_path(context)} && \
+              terraform fmt -list=true .
+            FMT_CMD
+
+            fmt_output = syscall(
+              fmt_cmd.squeeze(' '),
+              capture: true,
+              suppress: true
+            )
+
+            say "Formatted files:", :green
+            say fmt_output, :green
+          end
+        end
+      end
+    end
+  end
+end

data/lib/proteus/commands/taint.rb

@@ -0,0 +1,35 @@
+module Proteus
+  module Commands
+    module Taint
+      def self.included(thor_class)
+        thor_class.class_eval do
+
+          desc "taint", "Taints an existing resource"
+          long_desc <<-LONGDESC
+            Taints an existing resource
+
+            --resource The resource to taint
+          LONGDESC
+          option :resource, type: :string, aliases: "-r", required: true
+          option :module, type: :string, aliases: "-m", required: false, default: nil
+          def taint
+            init(verbose: parent_options[:verbose])
+            confirm question: "Do you really want to run 'terraform taint' on environment '#{environment}' in context '#{context}'?", color: :on_red, exit_code: 0 do
+
+              taint_command = <<~TAINT_COMMAND
+                cd #{context_path(context)} && \
+                terraform taint \
+                -var-file=#{var_file(context, environment)} \
+                #{aws_profile} \
+                #{options[:module] ? "-module=#{options[:module]}" : ""} \
+                #{options[:resource]}
+              TAINT_COMMAND
+              syscall taint_command.squeeze(' ')
+            end
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/proteus/common.rb

@@ -0,0 +1,72 @@
+require 'proteus/commands/apply'
+require 'proteus/commands/clean'
+require 'proteus/commands/destroy'
+require 'proteus/commands/graph'
+require 'proteus/commands/import'
+require 'proteus/commands/move'
+require 'proteus/commands/output'
+require 'proteus/commands/plan'
+require 'proteus/commands/remove'
+require 'proteus/commands/render'
+require 'proteus/commands/taint'
+
+module Proteus
+  class Common < Thor
+    include Helpers
+    include Proteus::Helpers::PathHelpers
+
+    Proteus::Commands.constants.each do |command|
+      include const_get("Proteus::Commands::#{command}")
+    end
+
+    private
+
+    def context
+      self.class.context
+    end
+
+    def environment
+      self.class.environment
+    end
+
+    def render_backend
+      Proteus::Backend::Backend.new(config: config, context: context, environment: environment).render
+    end
+
+    def init(verbose: false)
+      say "initializing", :green
+      say "environment: #{environment}", :green
+
+      render_backend
+
+      `rm -rf #{context_path(context)}/.terraform/*.tf*`
+      `rm -rf #{context_path(context)}/.terraform/modules`
+      `rm -rf #{context_path(context)}/terraform.tfstate*`
+
+      terraform_command = <<~TERRAFORM_COMMAND
+        cd #{context_path(context)} && \
+        terraform init \
+        -backend-config='key=#{config[:backend][:key_prefix]}#{context}-#{environment}.tfstate' \
+        #{aws_profile} \
+        #{context_path(context)}
+      TERRAFORM_COMMAND
+
+      output = syscall terraform_command.squeeze(' '), suppress: true, capture: true
+      say(output, :green) if verbose
+    end
+
+    def aws_profile
+      config[:providers].select {|p| p[:name] == 'aws' }.first[:environments].each do |provider|
+        return "-var 'aws_profile=#{provider[:profile]}'" if environment =~ (/#{provider[:match]}/)
+      end
+
+      say "No provider found matching environment #{environment}.", :red
+      exit 1
+    end
+
+    def dryrun
+      parent_options[:dryrun]
+    end
+
+  end
+end

data/lib/proteus/config/config.rb

@@ -0,0 +1,47 @@
+module Proteus
+  module Config
+    def config
+      unless @config
+        @config = YAML.load(File.read(File.expand_path(config_path))).with_indifferent_access
+        begin
+          validator = ConfigValidator.new(@config)
+        rescue Proteus::Validators::ValidationError => validation_error
+          say "ConfigValidator: #{validation_error.message} [#{config_path}] #{"\u2718".encode('utf-8')}", :red
+          exit 1
+        end
+      end
+
+      @config
+    end
+
+    class ConfigValidator < Proteus::Validators::BaseValidator
+      def validate
+        within :providers do
+          ensure_data_type Array
+
+          each do
+            within :environments do
+              ensure_uniqueness_across :match
+            end
+          end
+        end
+
+        within :slack_webhooks, optional: true do
+          ensure_data_type Array
+
+          each do
+            ensure_keys :match, :url
+          end
+        end
+
+        within :backend do
+          ensure_presence :key_prefix
+
+          within :bucket do
+            ensure_keys :name, :region, :profile
+          end
+        end
+      end
+    end
+  end
+end

data/lib/proteus/context_management/context.rb

@@ -0,0 +1,31 @@
+require 'proteus/helpers/path_helpers'
+
+module Proteus
+  module ContextManagement
+    class Context
+      include Proteus::Helpers::PathHelpers
+
+      attr_accessor :name
+
+      def initialize(name:)
+        @name = name
+
+        ensure_temp_directory
+      end
+
+      def environments
+        Dir.glob("#{File.expand_path(File.join(environments_path(@name)))}/*.tfvars").map do |vars_file|
+          File.basename(vars_file).split('.')[1]
+        end
+      end
+
+      private
+
+      def ensure_temp_directory
+        unless File.directory?(context_temp_directory(@name))
+          Dir.mkdir(context_temp_directory(@name))
+        end
+      end
+    end
+  end
+end

data/lib/proteus/context_management/helpers.rb

@@ -0,0 +1,14 @@
+module Proteus
+  module ContextManagement
+    module Helpers
+      include Proteus::Helpers::PathHelpers
+      def contexts
+        say "loading contexts", :green
+
+        Dir.glob(File.join(contexts_path, "*")).collect do |context_path|
+          Context.new(name: File.basename(context_path))
+        end
+      end
+    end
+  end
+end

data/lib/proteus/generate.rb

@@ -0,0 +1,18 @@
+require 'proteus/generators/context'
+require 'proteus/generators/environment'
+require 'proteus/generators/module'
+
+module Proteus
+  class Generate < Thor
+    include Thor::Actions
+    include Helpers
+
+    def self.source_root
+      File.expand_path(File.join(File.dirname(__FILE__), 'generators', 'templates'))
+    end
+
+    include Generators::Context
+    include Generators::Environment
+    include Generators::Module
+  end
+end