simplygenius-atmos 0.7.1 → 0.8.0

data/lib/simplygenius/atmos/commands/destroy.rb

@@ -0,0 +1,22 @@
+require_relative 'terraform'
+
+module SimplyGenius
+  module Atmos
+    module Commands
+
+      class Destroy < Terraform
+
+        def self.description
+          "Runs terraform destroy"
+        end
+
+        def execute
+          @terraform_arguments.insert(0, "destroy")
+          super
+        end
+
+      end
+
+    end
+  end
+end

data/lib/simplygenius/atmos/commands/generate.rb

@@ -0,0 +1,187 @@
+require_relative 'base_command'
+require_relative '../../atmos/source_path'
+require_relative '../../atmos/generator'
+require_relative '../../atmos/utils'
+
+module SimplyGenius
+  module Atmos
+    module Commands
+
+      # From https://github.com/rubber/rubber/blob/master/lib/rubber/commands/vulcanize.rb
+      class Generate < BaseCommand
+
+        def self.description
+          <<~EOF
+            Installs configuration templates used by atmos to create infrastructure
+            resources e.g.
+            
+              atmos generate aws/vpc
+            
+            use --list to get a list of the template names for a given sourceroot
+          EOF
+        end
+
+        option ["-f", "--force"],
+               :flag, "Overwrite files that already exist"
+        option ["-n", "--dryrun"],
+               :flag, "Run but do not make any changes"
+        option ["-q", "--quiet"],
+               :flag, "Supress status output"
+        option ["-s", "--skip"],
+               :flag, "Skip files that already exist"
+        option ["-d", "--[no-]dependencies"],
+               :flag, "Walk dependencies, or not", default: true
+        option ["-l", "--list"],
+               :flag, "list available templates"
+        option ["-u", "--update"],
+               :flag, "update all installed templates\n"
+        option ["-p", "--sourcepath"],
+               "PATH", "search for templates using given sourcepath",
+               multivalued: true
+        option ["-r", "--[no-]sourcepaths"],
+               :flag, "clear sourcepaths from template search\n", default: true
+        option ["-c", "--context"],
+               "CONTEXT", "provide context variables (dot notation)",
+               multivalued: true
+
+        parameter "TEMPLATE ...", "atmos template(s)", required: false
+
+        def execute
+          signal_usage_error "template name is required" if template_list.blank? && ! list? && !update?
+
+          sourcepath_list.each do |sp|
+            SourcePath.register(File.basename(sp), sp)
+          end
+
+          if sourcepaths?
+
+            # don't want to fail for new repo
+            if  Atmos.config && Atmos.config.is_atmos_repo?
+              Atmos.config['template_sources'].try(:each) do |item|
+                SourcePath.register(item.name, item.location)
+              end
+            end
+
+            # Always search for templates against the bundled templates directory
+            SourcePath.register('bundled', File.expand_path('../../../../../templates', __FILE__))
+
+          end
+
+          if list?
+            logger.info "Valid templates are:"
+            SourcePath.registry.each do |spname, sp|
+              logger.info("\tSourcepath #{sp}")
+              filtered_names = sp.template_names.select do |name|
+                template_list.blank? || template_list.any? {|f| name =~ /#{f}/ }
+              end
+              filtered_names.each {|n| logger.info ("\t\t#{n}")}
+            end
+          else
+            g = Generator.new(force: force?,
+                                 pretend: dryrun?,
+                                 quiet: quiet?,
+                                 skip: skip?,
+                                 dependencies: dependencies?)
+
+            begin
+
+              context = SettingsHash.new
+              context_list.each do |c|
+                key, value = c.split('=', 2)
+                context.notation_put(key, value)
+              end
+
+              if update?
+                # this isn't 100% foolproof, but is a convenience that should help for most cases
+
+                filtered_templates = state[:visited_templates].select do |vt|
+                  template_list.blank? || template_list.any? {|n| vt[:name] =~ /#{n}/ }
+                end
+
+                sps = filtered_templates.collect(&:source).uniq
+                sps.each do |src|
+                  spname = src[:name]
+                  sploc = src[:location]
+
+                  existing_sp = SourcePath.registry[spname]
+                  if existing_sp
+                    if existing_sp.location != sploc
+                      logger.warn("Saved sourcepath location differs from that in configuration")
+                      logger.warn(" #{spname} -> saved=#{sploc} configured=#{existing_sp.location}")
+                      logger.warn(" consider running with --no-sourcepaths")
+                    end
+                  else
+                    sp = SourcePath.register(spname, sploc)
+                    logger.warn("Saved state contains a source path missing from configuration: #{sp}")
+                  end
+                end
+
+                filtered_templates.each do |vt|
+                  name = vt[:name]
+                  ctx = vt[:context]
+                  spname = vt[:source][:name]
+                  sp = SourcePath.registry[spname]
+                  tmpl = sp.template(name)
+                  tmpl.scoped_context.merge!(ctx) if ctx
+                  tmpl.context.merge!(context)
+                  g.apply_template(tmpl)
+                end
+              else
+                g.generate(*template_list, context: context)
+              end
+
+              save_state(g.visited_templates, template_list)
+
+            rescue  ArgumentError => e
+              logger.error(e.message)
+              exit 1
+            end
+          end
+
+        end
+
+        def state_file
+          @state_file ||= Atmos.config["generate.state_file"]
+        end
+
+        def state
+          @state ||= begin
+            if state_file.present?
+              path = File.expand_path(state_file)
+              yml_hash = {}
+              if File.exist?(path)
+                yml_hash = YAML.load_file(path)
+              end
+              SettingsHash.new(yml_hash)
+            else
+              SettingsHash.new
+            end
+          end
+        end
+
+        def save_state(visited_templates, entrypoint_template_names)
+          if state_file.present?
+            visited_state = []
+            visited_templates.each do |tmpl|
+              visited_tmpl = tmpl.to_h
+              visited_tmpl[:context] = tmpl.scoped_context.to_h
+              visited_state << visited_tmpl
+            end
+
+            state[:visited_templates] ||= []
+            state[:visited_templates].concat(visited_state)
+            state[:visited_templates].sort! {|h1, h2| h1[:name] <=> h2[:name] }.uniq!
+
+            state[:entrypoint_templates] ||= []
+            state[:entrypoint_templates].concat(entrypoint_template_names)
+            state[:entrypoint_templates].sort!.uniq!
+
+            File.write(state_file, YAML.dump(state.to_hash))
+          end
+        end
+
+      end
+
+    end
+  end
+end

data/lib/simplygenius/atmos/commands/init.rb

@@ -0,0 +1,22 @@
+require_relative 'terraform'
+
+module SimplyGenius
+  module Atmos
+    module Commands
+
+      class Init < Terraform
+
+        def self.description
+          "Runs terraform init"
+        end
+
+        def execute
+          @terraform_arguments.insert(0, "init")
+          super
+        end
+
+      end
+
+    end
+  end
+end

data/lib/simplygenius/atmos/commands/new.rb

@@ -0,0 +1,22 @@
+require_relative 'generate'
+
+module SimplyGenius
+  module Atmos
+    module Commands
+
+      class New < Generate
+
+        def self.description
+          "Sets up a new atmos project in the current directory"
+        end
+
+        def execute
+          template_list << "new"
+          super
+        end
+
+      end
+
+    end
+  end
+end

data/lib/simplygenius/atmos/commands/otp.rb

@@ -0,0 +1,58 @@
+require_relative 'base_command'
+require_relative '../../atmos/otp'
+require 'clipboard'
+
+module SimplyGenius
+  module Atmos
+    module Commands
+
+      class Otp < BaseCommand
+
+        def self.description
+          "Generates an otp token for the given user"
+        end
+
+        option ["-s", "--secret"],
+               'SECRET', "The otp secret\nWill save for future use"
+
+        option ["-c", "--clipboard"],
+               :flag,
+               <<~EOF
+                Automatically copy the token to the system
+                clipboard.  For dependencies see:
+                https://github.com/janlelis/clipboard
+                EOF
+
+        parameter "NAME",
+                  "The otp name (IAM username)"
+
+        def execute
+          code = nil
+          if secret
+            Atmos::Otp.instance.add(name, secret)
+            code = Atmos::Otp.instance.generate(name)
+            Atmos::Otp.instance.save
+          else
+            code = Atmos::Otp.instance.generate(name)
+          end
+
+          if code.nil?
+            signal_usage_error <<~EOF
+              No otp secret has been setup for #{name}
+              Use the -m flag to 'atmos user create' to create/activate one
+              or associate an existing secret with 'atmos otp -s <secret> <name>'
+            EOF
+          else
+            puts code
+          end
+
+          if clipboard?
+            Clipboard.copy(code)
+          end
+        end
+
+      end
+
+    end
+  end
+end

data/lib/simplygenius/atmos/commands/plan.rb

@@ -0,0 +1,24 @@
+require_relative 'terraform'
+
+module SimplyGenius
+  module Atmos
+    module Commands
+
+      class Plan < Terraform
+
+        def self.description
+          "Runs terraform plan"
+        end
+
+        def execute
+          args = ["plan"]
+          args << "--get-modules" unless Atmos.config["disable_auto_modules"].to_s == "true"
+          @terraform_arguments.insert(0, *args)
+          super
+        end
+
+      end
+
+    end
+  end
+end

data/lib/simplygenius/atmos/commands/secret.rb

@@ -0,0 +1,91 @@
+require_relative 'base_command'
+require 'climate_control'
+
+module SimplyGenius
+  module Atmos
+    module Commands
+
+      class Secret < BaseCommand
+
+        def self.description
+          "Manages application secrets"
+        end
+
+        subcommand "get", "Gets the secret value" do
+
+          parameter "KEY",
+                    "The secret key"
+
+          def execute
+
+            Atmos.config.provider.auth_manager.authenticate(ENV) do |auth_env|
+              ClimateControl.modify(auth_env) do
+                value = Atmos.config.provider.secret_manager.get(key)
+                logger.info "Secret value for #{key}: #{value}"
+              end
+            end
+
+          end
+
+        end
+
+        subcommand "set", "Sets the secret value" do
+
+          parameter "KEY",
+                    "The secret key"
+
+          parameter "VALUE",
+                    "The secret value"
+
+          def execute
+
+            Atmos.config.provider.auth_manager.authenticate(ENV) do |auth_env|
+              ClimateControl.modify(auth_env) do
+                Atmos.config.provider.secret_manager.set(key, value)
+                logger.info "Secret set for #{key}"
+              end
+            end
+
+          end
+
+        end
+
+        subcommand "list", "Lists all secret keys" do
+
+          def execute
+
+            Atmos.config.provider.auth_manager.authenticate(ENV) do |auth_env|
+              ClimateControl.modify(auth_env) do
+                logger.info "Secret keys are:"
+                Atmos.config.provider.secret_manager.to_h.keys.each {|k| logger.info k}
+              end
+            end
+
+          end
+
+        end
+
+        subcommand "delete", "Deletes the secret key/value" do
+
+          parameter "KEY",
+                    "The secret key"
+
+          def execute
+
+            Atmos.config.provider.auth_manager.authenticate(ENV) do |auth_env|
+              ClimateControl.modify(auth_env) do
+                value = Atmos.config.provider.secret_manager.get(key)
+                Atmos.config.provider.secret_manager.delete(key)
+                logger.info "Deleted secret: #{key}=#{value}"
+              end
+            end
+
+          end
+
+        end
+
+      end
+
+    end
+  end
+end

data/lib/simplygenius/atmos/commands/terraform.rb

@@ -0,0 +1,56 @@
+require_relative 'base_command'
+require_relative '../../atmos/terraform_executor'
+
+module SimplyGenius
+  module Atmos
+    module Commands
+
+      class Terraform < BaseCommand
+
+        def self.description
+          "Runs terraform"
+        end
+
+        # override so we can pass all options/flags/parameters directly to
+        # terraform instead of having clamp parse them
+        def parse(arguments)
+          @terraform_arguments = arguments
+        end
+
+        def execute
+
+          unless Atmos.config.is_atmos_repo?
+            signal_usage_error <<~EOF
+              Atmos can only run terraform from a location configured for atmos. 
+              Have you run atmos init?"
+            EOF
+          end
+
+          Atmos.config.provider.auth_manager.authenticate(ENV) do |auth_env|
+            begin
+
+              # TODO: hack to allow apply/etc for bootstrap group
+              # Fix this once we allow more extensive recipe grouping
+              working_group = 'default'
+              @terraform_arguments.each_with_index do |a, i|
+                if a == "--group"
+                  @terraform_arguments.delete_at(i)
+                  working_group = @terraform_arguments.delete_at(i)
+                  break
+                end
+              end
+
+              exe = TerraformExecutor.new(process_env: auth_env, working_group: working_group)
+              get_modules = @terraform_arguments.delete("--get-modules")
+              exe.run(*@terraform_arguments, get_modules: get_modules.present?)
+            rescue TerraformExecutor::ProcessFailed => e
+              logger.error(e.message)
+            end
+          end
+        end
+
+      end
+
+    end
+  end
+end