terraspace 0.0.0 → 0.1.0

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

@@ -0,0 +1,30 @@
+module Terraspace::Terraform::Args
+  module Dsl
+    include Shorthands
+
+    def command(*commands, **props)
+      commands.each do |name|
+        if shorthand?(name)
+          shorthand_commands(name, props)
+        else
+          each_command(name, props)
+        end
+      end
+    end
+    alias_method :commands, :command
+
+    def shorthand?(name)
+      shorthands.key?(name.to_sym)
+    end
+
+    def shorthand_commands(name, props)
+      shorthands[name].each do |n|
+        each_command(n, props)
+      end
+    end
+
+    def each_command(name, props={})
+      @commands[name] = props
+    end
+  end
+end

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/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_build_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_build_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

data/lib/terraspace/version.rb

@@ -1,3 +1,3 @@
 module Terraspace
-  VERSION = "0.0.0"
+  VERSION = "0.1.0"
 end

data/spec/cli_spec.rb

@@ -0,0 +1,13 @@
+describe Terraspace::CLI do
+  before(:all) do
+    @args = "--from Tung"
+  end
+
+  describe "terraspace" do
+    it "up" do
+      # ts_root = "spec/fixtures/projects/hcl/aws"
+      # out = execute("TS_ROOT=#{ts_root} exe/terraspace up core")
+      # puts out
+    end
+  end
+end

data/spec/fixtures/cache_build_dir/variables.tf

@@ -0,0 +1,9 @@
+variable "project" {
+  description = "Project name. IE: test-project"
+  type        = string
+}
+
+variable "name" {
+  default     = "demo-name"
+  type        = string
+}

data/spec/fixtures/initialized/modules.json

@@ -0,0 +1,29 @@
+{
+  "Modules": [
+    {
+      "Key": "",
+      "Source": "",
+      "Dir": "."
+    },
+    {
+      "Key": "sg_parent",
+      "Source": "../../modules/sg_parent",
+      "Dir": "../../modules/sg_parent"
+    },
+    {
+      "Key": "sg_parent.sg_child_test",
+      "Source": "git@github.com:boltops-tools/sg_child.git",
+      "Dir": ".terraform/modules/sg_parent.sg_child_test"
+    },
+    {
+      "Key": "vpc",
+      "Source": "git@github.com:org/aws-vpc.git?ref=master",
+      "Dir": ".terraform/modules/vpc"
+    },
+    {
+      "Key": "vpc.aws_security_group",
+      "Source": "git@github.com:org/aws-security-group.git?ref=master",
+      "Dir": ".terraform/modules/vpc.aws_security_group"
+    }
+  ]
+}

data/spec/fixtures/orphans/config/backend.rb

@@ -0,0 +1,8 @@
+backend("s3",
+  bucket:         "demo-bucket",
+  # hard-code region for spec only
+  key:            "us-west-2/:ENV/:BUILD_DIR/terraform.tfstate", # variable notation expanded by terraspace IE: us-west-2/dev/modules/vm/terraform.tfstate
+  region:         "us-west-2",
+  encrypt:        true,
+  dynamodb_table: "terraform_locks",
+)

data/spec/fixtures/orphans/config/provider.rb

@@ -0,0 +1,3 @@
+provider("aws",
+  region: "us-west-2"
+)