terraspace 0.2.1 → 0.3.1

data/lib/terraspace/terraform/api/var.rb

@@ -1,9 +1,10 @@
 class Terraspace::Terraform::Api
   class Var
     extend Memoist
-    include Client
+    include Http::Concern
     include Terraspace::Util::Logging
 
+    # workspace: details from the api response
     def initialize(workspace, attrs={})
       @workspace, @attrs = workspace, attrs
       @workspace_id = @workspace['id']
@@ -15,27 +16,30 @@ class Terraspace::Terraform::Api
 
     def update
       return unless overwrite?
-      logger.debug "Updating Terraform Cloud #{category} variable: #{@attrs['key']}"
+      updating_message
       variable_id = variable_id(@attrs['key'])
       payload = payload(variable_id)
       http.patch("workspaces/#{@workspace_id}/vars/#{variable_id}", payload)
     end
 
     def overwrite?
-      cloud = Terraspace.config.cloud
       if @attrs['sensitive']
-        cloud.overwrite_sensitive
+        vars.overwrite_sensitive
       else
-        cloud.overwrite
+        vars.overwrite
       end
     end
 
+    def vars
+      Terraspace.config.cloud.vars
+    end
+
     def variable_id(key)
       current_var_resp['id']
     end
 
     def create
-      logger.info "Creating Terraform Cloud #{category} variable: #{@attrs['key']}"
+      creating_message
       http.post("workspaces/#{@workspace_id}/vars", payload)
     end
 
@@ -48,6 +52,16 @@ class Terraspace::Terraform::Api
       { data: data }
     end
 
+    def updating_message
+      return unless %w[all update].include?(vars.show_message)
+      logger.info "Updating Terraform Cloud #{category} variable: #{@attrs['key']}"
+    end
+
+    def creating_message
+      return unless %w[all create].include?(vars.show_message)
+      logger.info "Creating Terraform Cloud #{category} variable: #{@attrs['key']}"
+    end
+
     def exist?
       !!current_var_resp
     end

data/lib/terraspace/terraform/api/vars.rb

@@ -1,8 +1,9 @@
 class Terraspace::Terraform::Api
   class Vars
     extend Memoist
-    include Client
+    include Http::Concern
 
+    # workspace: details from the api response
     def initialize(mod, workspace)
       @mod, @workspace = mod, workspace
     end

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

@@ -0,0 +1,98 @@
+class Terraspace::Terraform::Api
+  class Workspace < Base
+    extend Memoist
+
+    attr_reader :name
+    def initialize(mod, organization, name)
+      @mod, @organization, @name = mod, organization, name
+    end
+
+    # Docs: https://www.terraform.io/docs/cloud/api/workspaces.html
+    def set_working_dir
+      return if working_directory == details['attributes']['working-directory']
+
+      payload = {
+        data: {
+          attributes: {
+            "working-directory": working_directory
+          },
+          type: "workspaces"
+        }
+      }
+      http.patch("organizations/#{@organization}/workspaces/#{@name}", payload)
+    end
+
+    def working_directory
+      cache_dir = @mod.cache_dir.sub("#{Terraspace.root}/", '')
+      prefix = Terraspace.config.cloud.working_dir_prefix # prepended to TFC Working Directory
+      prefix ? "#{prefix}/#{cache_dir}" : cache_dir
+    end
+
+    def set_env_vars
+      Vars.new(@mod, details).run
+    end
+
+    def details(options={})
+      payload = http.get("organizations/#{@organization}/workspaces/#{@name}")
+      # Note only way to get here is to bypass init. Example:
+      #
+      #     terraspace up demo --no-init
+      #
+      unless payload || options[:exit_on_fail] == false
+        logger.error "ERROR: Unable to find the workspace: #{@name}. The workspace may not exist. Or the Terraform token may be invalid. Please double check your Terraform token.".color(:red)
+        exit 1
+      end
+      payload['data'] if payload
+    end
+    memoize :details
+
+    def destroy
+      # response payload from delete operation is nil
+      http.delete("/organizations/#{@organization}/workspaces/#{@name}")
+    end
+
+    # Docs: https://www.terraform.io/docs/cloud/api/workspaces.html
+    def create
+      payload = upsert_payload
+      http.post("organizations/#{@organization}/workspaces", payload)
+    end
+
+    def update
+      payload = upsert_payload
+      http.patch("organizations/#{@organization}/workspaces/#{@name}", payload)
+      self.flush_cache
+    end
+
+    def upsert_payload
+      {
+        data: {
+          attributes: attributes,
+          type: "workspaces"
+        }
+      }
+    end
+
+    def attributes
+      attrs = { name: @name }
+      config = Terraspace.config.cloud.workspace.attrs
+      attrs.merge!(config)
+      # Default: run on all changes since app/modules can affect app/stacks
+      if config['vcs-repo'] && config['file-triggers-enabled'].nil?
+        attrs['file-triggers-enabled'.to_sym] = false
+      end
+      token = ENV['TS_CLOUD_OAUTH_TOKEN']
+      if config['vcs-repo'] && !config.dig('vcs-repo', 'oauth-token-id') && token
+        attrs['vcs-repo'.to_sym]['oauth-token-id'.to_sym] ||= token
+      end
+      attrs
+    end
+
+    def create_or_update
+      exist? ? update : create
+    end
+
+    def exist?
+      !!details(exit_on_fail: false)
+    end
+  end
+end

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

@@ -11,7 +11,7 @@ module Terraspace::Terraform::Args
       # https://terraspace.cloud/docs/ci-automation/
       ENV['TF_IN_AUTOMATION'] = '1' if @options[:auto]
 
-      if %w[apply destroy init output plan].include?(@name)
+      if %w[apply destroy init output plan show].include?(@name)
         meth = "#{@name}_args"
         send(meth)
       else
@@ -26,13 +26,7 @@ module Terraspace::Terraform::Args
         args << var_files.map { |f| "-var-file #{Dir.pwd}/#{f}" }.join(' ')
       end
 
-      if @options[:auto] && @options[:input].nil?
-        args << " -input=false"
-      end
-      unless @options[:input].nil?
-        input = @options[:input] ? "true" : "false"
-        args << " -input=#{input}" # = sign required for apply when there's a plan at the end. so input=false works input false doesnt
-      end
+      args << input_option if input_option
 
       # must be at the end
       plan = @options[:plan]
@@ -44,12 +38,24 @@ module Terraspace::Terraform::Args
           src = "#{Dir.pwd}/#{plan}"
           dest = "#{@mod.cache_dir}/#{File.basename(src)}"
         end
-        FileUtils.cp(src, dest)
+        FileUtils.cp(src, dest) unless same_file?(src, dest)
         args << " #{dest}"
       end
       args
     end
 
+    def input_option
+      option = nil
+      if @options[:auto] && @options[:input].nil?
+        option = " -input=false"
+      end
+      unless @options[:input].nil?
+        input = @options[:input] ? "true" : "false"
+        option = " -input=#{input}" # = sign required for apply when there's a plan at the end. so input=false works input false doesnt
+      end
+      option
+    end
+
     def init_args
       args = "-get"
       if @options[:auto] && @options[:input].nil?
@@ -63,24 +69,33 @@ module Terraspace::Terraform::Args
       args << " -reconfigure" if @options[:reconfigure]
 
       # must be at the end
-      if @quiet && !ENV['TS_INIT_LOUD']
-        out_path = self.class.terraform_init_log
-        FileUtils.mkdir_p(File.dirname(out_path))
-        args << " > #{out_path}"
+      if @quiet
+        log_path = self.class.terraform_init_log(@mod.name)
+        FileUtils.mkdir_p(File.dirname(log_path))
+        args << " >> #{log_path}"
       end
       [args]
     end
 
+    def output_args
+      args = []
+      args << "-json" if @options[:format] == "json"
+      args << "> #{expanded_out}" if @options[:out]
+      args
+    end
+
     def plan_args
       args = []
+      args << input_option if input_option
+      args << "-destroy" if @options[:destroy]
       args << "-out #{expanded_out}" if @options[:out]
       args
     end
 
-    def output_args
+    def show_args
       args = []
-      args << "-json" if @options[:format] == "json"
-      args << "> #{expanded_out}" if @options[:out]
+      args << " -json" if @options[:json]
+      args << " #{@options[:plan]}" if @options[:plan] # terraform show /path/to/plan
       args
     end
 
@@ -98,13 +113,25 @@ module Terraspace::Terraform::Args
     end
 
     class << self
+      extend Memoist
+
       # Use different tmp log file in case uses run terraspace up in 2 terminals at the same time
-      @@terraform_init_log = nil
-      def terraform_init_log
-        return @@terraform_init_log if @@terraform_init_log
-        basename = File.basename(Tempfile.new('terraform-init').path)
-        @@terraform_init_log = "#{Terraspace.tmp_root}/out/#{basename}.out"
+      #
+      # Log for init is in /tmp because using shell >> redirection
+      # It requires full path since we're running terraform within the .terraspace-cache folder
+      # This keeps the printed command shorter:
+      #
+      #     => terraform init -get >> /tmp/terraspace/log/init/demo.log
+      #
+      def terraform_init_log(mod_name)
+        "#{Terraspace.tmp_root}/log/init/#{mod_name}.log"
       end
+      memoize :terraform_init_log
+    end
+
+  private
+    def same_file?(src, dest)
+      src == dest
     end
   end
 end

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

@@ -0,0 +1,13 @@
+module Terraspace::Terraform::Cloud
+  class Runs < Terraspace::CLI::Base
+    def list
+      lister = Lister.new(@mod, @options)
+      lister.run
+    end
+
+    def prune
+      pruner = Pruner.new(@mod, @options)
+      pruner.run
+    end
+  end
+end

data/lib/terraspace/terraform/cloud/runs/base.rb

@@ -0,0 +1,33 @@
+class Terraspace::Terraform::Cloud::Runs
+  class Base
+    extend Memoist
+    include Terraspace::Util::Logging
+    include Terraspace::Util::Sure
+    include Terraspace::Terraform::Api::Client
+
+    # Api::Client requires @mod to be set
+    def initialize(mod, options={})
+      @mod, @options = mod, options
+    end
+
+    def runs
+      runs = api.runs.list
+      runs.select! do |item|
+        @options[:status].nil? ||
+        @options[:status].include?("all") ||
+        @options[:status].include?(item['attributes']['status'])
+      end
+      runs
+    end
+    memoize :runs
+
+    def build_project
+      Terraspace::Builder.new(@options).run
+
+      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
+    end
+  end
+end

data/lib/terraspace/terraform/cloud/runs/item_presenter.rb

@@ -0,0 +1,37 @@
+class Terraspace::Terraform::Cloud::Runs
+  class ItemPresenter
+    attr_reader :id
+    def initialize(raw)
+      @raw = raw # raw item
+      @id = raw['id']
+      @attrs = raw['attributes']
+    end
+
+    def method_missing(name, *args, &block)
+      attrs = @attrs.transform_keys { |k| k.gsub('-','_').to_sym }
+      if attrs.key?(name)
+        attrs[name]
+      else
+        super
+      end
+    end
+
+    def message
+      max = 25
+      message = @attrs['message']
+      if message.size >= max
+        message[0..max] + "..."
+      else
+        message
+      end
+    end
+
+    def created_at
+      pretty_time(@attrs['created-at'])
+    end
+
+    def pretty_time(text)
+      text.sub(/\..*/,'')
+    end
+  end
+end

data/lib/terraspace/terraform/cloud/runs/lister.rb

@@ -0,0 +1,22 @@
+require 'cli-format'
+
+class Terraspace::Terraform::Cloud::Runs
+  class Lister < Base
+    def run
+      build_project
+      if runs.empty?
+        logger.info "No runs found"
+        return
+      end
+
+      presenter = CliFormat::Presenter.new(@options)
+      presenter.header = ["Id", "Status", "Message", "Created At"]
+      runs.each do |item|
+        p = ItemPresenter.new(item)
+        row = [p.id, p.status, p.message, p.created_at]
+        presenter.rows << row
+      end
+      presenter.show
+    end
+  end
+end

data/lib/terraspace/terraform/cloud/runs/pruner.rb

@@ -0,0 +1,109 @@
+class Terraspace::Terraform::Cloud::Runs
+  class Pruner < Base
+    include Terraspace::Terraform::Api::Client
+
+    # @mod required for Api::Client
+    def initialize(mod, options={})
+      super
+      @queue, @kept, @needs_pruning = [], nil, false
+    end
+
+    def run
+      build_project
+      build_queue
+      are_you_sure?
+      process_queue
+    end
+
+  private
+    def build_queue
+      runs.each do |item|
+        next unless actionable?(item)
+        unless @kept
+          @kept = item
+          next
+        end
+        queue(item)
+        @needs_pruning = true
+      end
+    end
+
+    def are_you_sure?
+      unless @needs_pruning
+        logger.info "Nothing to prune"
+        return
+      end
+
+      keeping = item_message(@kept)
+      items = @queue.map { |i| item_message(i) }.join("\n")
+      message =<<~EOL
+        Will keep:
+
+        #{keeping}
+
+        Will prune:
+
+        #{items}
+
+        Are you sure?
+      EOL
+      sure?(message.chop)
+    end
+
+    def item_message(item)
+      p = ItemPresenter.new(item)
+      "   #{p.id} #{p.status} #{p.message} #{p.created_at}"
+    end
+
+    def process_queue
+      @queue.each do |item|
+        process(item)
+      end
+    end
+
+    def process(item)
+      id = item['id']
+      action = discardable?(item) ? "Discarded" : "Cancelled"
+      p = ItemPresenter.new(item)
+      msg = "#{action} #{p.id} #{p.message}" # note id is named run-xxx
+      logger.info("NOOP: #{msg}") && return if @options[:noop]
+
+      if discardable?(item)
+        api.runs.discard(id)
+      elsif cancelable?(item)
+        api.runs.cancel(id)
+      end
+      logger.info msg
+    end
+
+    def queue(item)
+      if discardable?(item)
+        @queue << item
+      elsif cancelable?(item)
+        @queue << item
+      end
+    end
+
+    # Docs seem to be off: https://www.terraform.io/docs/cloud/api/run.html#apply-a-run
+    #
+    #     This includes runs in the "pending," "needs confirmation," "policy checked," and "policy override" states.
+    #
+    # Cant really discard a "pending" status, but can discard a "planned" status.
+    #
+    def actionable?(item)
+      discardable?(item) || cancelable?(item)
+    end
+
+    def discardable?(item)
+      %w[planned].include?(status(item))
+    end
+
+    def cancelable?(item)
+      %w[pending].include?(status(item))
+    end
+
+    def status(item)
+      item['attributes']['status']
+    end
+  end
+end