terraspace 0.0.0 → 0.2.1

data/lib/terraspace/terraform/args/shorthands.rb

@@ -0,0 +1,47 @@
+module Terraspace::Terraform::Args
+  module Shorthands
+    COMMANDS_WITH_INPUT = %w[
+      apply
+      import
+      init
+      plan
+      refresh
+    ]
+
+    COMMANDS_WITH_LOCKING = %w[
+      apply
+      destroy
+      import
+      init
+      plan
+      refresh
+      taint
+      untaint
+    ]
+
+    COMMANDS_WITH_PARALLELISM = %w[
+      apply
+      plan
+      destroy
+    ]
+
+    COMMANDS_WITH_VARS = %w[
+      apply
+      console
+      destroy
+      import
+      plan
+      push
+      refresh
+    ]
+
+    def shorthands
+      {
+        with_input: COMMANDS_WITH_INPUT,
+        with_locking: COMMANDS_WITH_LOCKING,
+        with_parallelism: COMMANDS_WITH_PARALLELISM,
+        with_vars: COMMANDS_WITH_VARS,
+      }
+    end
+  end
+end

data/lib/terraspace/terraform/cloud.rb

@@ -0,0 +1,25 @@
+module Terraspace::Terraform
+  class Cloud < Terraspace::CLI::Base
+    extend Memoist
+
+    def run
+      return unless workspaces?
+      api = Api.new(@mod, remote)
+      api.set_working_dir
+      api.set_env_vars
+    end
+
+    def workspaces?
+      remote && remote['workspaces']
+    end
+
+    def remote
+      backend["remote"]
+    end
+
+    def backend
+      Terraspace::Compiler::Backend::Parser.new(@mod).result
+    end
+    memoize :backend
+  end
+end

data/lib/terraspace/terraform/cloud/workspace.rb

@@ -0,0 +1,95 @@
+class Terraspace::Terraform::Cloud
+  class Workspace < Terraspace::CLI::Base
+    extend Memoist
+    include Terraspace::Util::Logging
+    include Terraspace::Terraform::Api::Client
+
+    # List will not have @mod set.
+    def list
+      @mod = Terraspace::CLI::Build::Placeholder.new(@options).build
+      unless remote && remote['organization']
+        logger.info "ERROR: There was no organization found. Are you sure you configured backend.tf with it?".color(:red)
+        exit 1
+      end
+
+      org = remote['organization']
+      payload = http.get("/organizations/#{org}/workspaces") # list using api client directly
+      names = payload['data'].map { |i| i['attributes']['name'] }.sort
+      logger.info "Workspaces for #{org}:"
+      logger.info names.join("\n")
+    end
+
+    def setup
+      build
+      unless backend.dig('remote','workspaces') # in case called by terraspace down demo -y --destroy-workspace with a non-remote backend
+        logger.info "ERROR: Workspace not configured in backend.tf"
+        exit 1
+      end
+      init
+    end
+
+    def init
+      Terraspace::CLI::Init.new(@options.merge(calling_command: "cloud-setup")).run
+    end
+
+    def destroy
+      build
+      return unless backend.dig('remote','workspaces') # in case called by terraspace down demo -y --destroy-workspace with a non-remote backend
+      api = Terraspace::Terraform::Api.new(@mod, remote)
+      workspace = api.workspace(exit_on_fail: false)
+      unless workspace
+        logger.info "Workspace #{workspace_name} not found for #{@mod.type}: #{@mod.name}"
+        exit 0
+      end
+      sure?
+      logger.info "Destroying workspace #{workspace_name}"
+      api.destroy_workspace
+    end
+
+    def build
+      Terraspace::Builder.new(@options).run
+    end
+
+    def workspace_name
+      remote['workspaces']['name']
+    end
+
+    def remote
+      backend["remote"]
+    end
+
+    def backend
+      Terraspace::Compiler::Backend::Parser.new(@mod).result
+    end
+    memoize :backend
+
+    def sure?
+      message = <<~EOL.chop + " " # chop to remove newline
+        You are about to delete the workspace: #{workspace_name}
+        All variables, settings, run history, and state history will be removed.
+        This cannot be undone.
+
+        This will NOT remove any infrastructure managed by this workspace.
+        If needed, destroy the infrastructure prior to deleting the workspace with:
+
+            terraspace down #{@mod.name}
+
+        This will delete the workspace: #{workspace_name}.
+        Are you sure? (y/N)
+      EOL
+
+      if @options[:yes]
+        sure = 'y'
+      else
+        print message
+        sure = $stdin.gets
+      end
+
+      unless sure =~ /^y/
+        puts "Whew! Exiting."
+        exit 0
+      end
+    end
+  end
+end
+

data/lib/terraspace/terraform/hooks/builder.rb

@@ -0,0 +1,40 @@
+module Terraspace::Terraform::Hooks
+  class Builder
+    extend Memoist
+    include Dsl
+    include DslEvaluator
+    include Terraspace::Util
+
+    attr_accessor :name
+    def initialize(mod, name)
+      @mod, @name = mod, name
+      @file = "#{Terraspace.root}/config/cli/hooks.rb"
+      @hooks = {before: {}, after: {}}
+    end
+
+    def build
+      return @hooks unless File.exist?(@file)
+      evaluate_file(@file)
+      @hooks.deep_stringify_keys!
+    end
+    memoize :build
+
+    def run_hooks
+      build
+      run_hook("before")
+      yield if block_given?
+      run_hook("after")
+    end
+
+    def run_hook(type)
+      execute = @hooks.dig(type, @name.to_s, "execute")
+      return unless execute
+
+      exit_on_fail = @hooks.dig(type, @name.to_s, "exit_on_fail")
+      exit_on_fail = exit_on_fail.nil? ? true : exit_on_fail
+
+      logger.info "Running #{type} hook"
+      sh(execute, exit_on_fail: exit_on_fail)
+    end
+  end
+end

data/lib/terraspace/terraform/hooks/dsl.rb

@@ -0,0 +1,19 @@
+module Terraspace::Terraform::Hooks
+  module Dsl
+    def before(*commands, **props)
+      commands.each do |name|
+        each_hook(:before, name, props)
+      end
+    end
+
+    def after(*commands, **props)
+      commands.each do |name|
+        each_hook(:after, name, props)
+      end
+    end
+
+    def each_hook(type, name, props={})
+      @hooks[type][name] = props
+    end
+  end
+end

data/lib/terraspace/terraform/runner.rb

@@ -0,0 +1,77 @@
+module Terraspace::Terraform
+  class Runner < Terraspace::CLI::Base
+    extend Memoist
+    include Terraspace::Util
+
+    attr_reader :name
+    def initialize(name, options={})
+      @name = name
+      super(options)
+    end
+
+    def run
+      time_took do
+        terraform(name, args)
+      end
+    end
+
+    def terraform(name, *args)
+      current_dir_message # only show once
+
+      params = args.flatten.join(' ')
+      command = "terraform #{name} #{params}"
+      run_hooks(name) do
+        sh(command, env: custom.env_vars)
+      end
+    end
+
+    @@current_dir_message_shown = false
+    def current_dir_message
+      return if @@current_dir_message_shown
+      logger.info "Current directory: #{Terraspace::Util.pretty_path(@mod.cache_dir)}"
+      @@current_dir_message_shown = true
+    end
+
+    def run_hooks(name, &block)
+      hooks = Hooks::Builder.new(@mod, name)
+      hooks.build # build hooks
+      hooks.run_hooks(&block)
+    end
+
+    def args
+      # base at end in case of redirection. IE: terraform output > /path
+      custom.args + custom.var_files + default.args
+    end
+
+    def custom
+      Args::Custom.new(@mod, @name)
+    end
+    memoize :custom
+
+    def default
+      Args::Default.new(@mod, @name, @options)
+    end
+    memoize :default
+
+  private
+    def time_took
+      t1 = Time.now
+      yield
+      t2 = Time.now
+      if %w[apply destroy].include?(@name)
+        puts "Time took: #{pretty_time(t2-t1)}"
+      end
+    end
+
+    # http://stackoverflow.com/questions/4175733/convert-duration-to-hoursminutesseconds-or-similar-in-rails-3-or-ruby
+    def pretty_time(total_seconds)
+      minutes = (total_seconds / 60) % 60
+      seconds = total_seconds % 60
+      if total_seconds < 60
+        "#{seconds.to_i}s"
+      else
+        "#{minutes.to_i}m #{seconds.to_i}s"
+      end
+    end
+  end
+end

data/lib/terraspace/tester.rb

@@ -0,0 +1,29 @@
+module Terraspace
+  module Tester
+    extend Memoist
+
+    # The tester metadata
+    #
+    # Example meta:
+    #
+    #    {
+    #      "rspec"  => {root: "/path"}
+    #    }
+    #
+    @@meta = {}
+    def meta
+      @@meta
+    end
+
+    def register(tester, data)
+      @@meta[tester] = data
+    end
+
+    def find_with(options={})
+      Finder.new.find_with(options)
+    end
+    memoize :find_with
+
+    extend self
+  end
+end

data/lib/terraspace/tester/finder.rb

@@ -0,0 +1,25 @@
+module Terraspace::Tester
+  class Finder
+    def find_with(options)
+      result = if options.key?(:framework)
+                 find_with_framework(options[:framework])
+               else
+                 raise "Must provide framework_name option."
+               end
+
+      return unless result
+      raw = Hash[*result] # convert result to Hash instead of an Array
+      Meta.new(raw)
+    end
+
+    def find_with_framework(framework)
+      meta.find do |framework_name, data|
+        framework_name == framework
+      end
+    end
+
+    def meta
+      Terraspace::Tester.meta
+    end
+  end
+end

data/lib/terraspace/tester/meta.rb

@@ -0,0 +1,25 @@
+module Terraspace::Tester
+  class Meta
+    # raw: {"rspec" => {root: "/path"}
+    def initialize(raw)
+      @raw = raw
+    end
+
+    def name
+      name = @raw.keys.first
+      unless name
+        raise "No tester gem found. Are you sure you have the terraspace test gem configured in your Gemfile?"
+      end
+      name.camelize
+    end
+    alias_method :tester, :name
+
+    def data
+      @raw.values.first
+    end
+
+    def root
+      data[:root]
+    end
+  end
+end

data/lib/terraspace/util.rb

@@ -0,0 +1,13 @@
+module Terraspace
+  module Util
+    include Logging
+    include Sh
+    include Sure
+
+    def pretty_path(path)
+      ENV['TS_TEST'] ? path : path.sub("#{Terraspace.root}/",'')
+    end
+
+    extend self
+  end
+end

data/lib/terraspace/util/logging.rb

@@ -0,0 +1,7 @@
+module Terraspace::Util
+  module Logging
+    def logger
+      Terraspace.logger
+    end
+  end
+end

data/lib/terraspace/util/sh.rb

@@ -0,0 +1,19 @@
+module Terraspace::Util
+  module Sh
+    # requires @mod to be set
+    def sh(command, options={})
+      exit_on_fail = options[:exit_on_fail].nil? ? true : options[:exit_on_fail]
+      env = options[:env] || {}
+      env.stringify_keys!
+
+      logger.info "=> #{command}"
+      return if ENV['TS_TEST']
+
+      success = system(env, command, chdir: @mod.cache_dir)
+      unless success
+        logger.info "Error running command: #{command}".color(:red)
+        exit $?.exitstatus if exit_on_fail
+      end
+    end
+  end
+end

data/lib/terraspace/util/sure.rb

@@ -0,0 +1,22 @@
+module Terraspace::Util
+  module Sure
+    def sure?(message="Are you sure?", desc=nil)
+      if @options[:yes]
+        sure = 'y'
+      else
+        out = if desc
+          "\n#{desc}\n#{message} (y/N) "
+        else
+          "#{message} (y/N) "
+        end
+        print out
+        sure = $stdin.gets
+      end
+
+      unless sure =~ /^y/
+        puts "Whew! Exiting."
+        exit 0
+      end
+    end
+  end
+end