ufo 3.5.7 → 4.0.0

data/lib/ufo/apps/service.rb

@@ -0,0 +1,56 @@
+class Ufo::Apps
+  class Service
+    extend Memoist
+
+    def initialize(service, options)
+      @service = service
+      @options = options
+    end
+
+    def to_a
+      [name, task_definition, running, launch_type, ufo?]
+    end
+
+    def task_definition
+      @service["task_definition"].split('/').last
+    end
+
+    def launch_type
+      @service["launch_type"]
+    end
+
+    def cfn_map
+      @cfn_map ||= CfnMap.new(@options).map
+    end
+
+    def ufo?
+      yes = !!cfn_map[@service["service_name"]]
+      yes ? "yes" : "no"
+    end
+
+    def name
+      pretty_service_name = @service["service_name"]
+      pretty_name = cfn_map[pretty_service_name]
+      if pretty_name
+        "#{pretty_service_name} (#{pretty_name})"
+      else
+        pretty_service_name
+      end
+    end
+
+    def running
+      @service["running_count"]
+    end
+
+    def dns
+      return 'dns' if ENV['TEST']
+      elb = info.load_balancer(@service)
+      elb.dns_name if elb
+    end
+
+    def info
+      Ufo::Info.new(@service)
+    end
+    memoize :info
+  end
+end

data/lib/ufo/aws_service.rb

@@ -1,3 +1,4 @@
+require "aws-sdk-cloudformation"
 require "aws-sdk-cloudwatchlogs"
 require "aws-sdk-ec2"
 require "aws-sdk-ecr"
@@ -6,20 +7,28 @@ require "aws-sdk-elasticloadbalancingv2"
 
 module Ufo
   module AwsService
-    def ecs
-      @ecs ||= Aws::ECS::Client.new
+    def cloudformation
+      @cloudformation ||= Aws::CloudFormation::Client.new
     end
 
-    def elb
-      @elb ||= Aws::ElasticLoadBalancingV2::Client.new
+    def cloudwatchlogs
+      @cloudwatchlogs ||= Aws::CloudWatchLogs::Client.new
+    end
+
+    def ec2
+      @ec2 ||= Aws::EC2::Client.new
     end
 
     def ecr
       @ecr ||= Aws::ECR::Client.new
     end
 
-    def cloudwatchlogs
-      @cloudwatchlogs ||= Aws::CloudWatchLogs::Client.new
+    def ecs
+      @ecs ||= Aws::ECS::Client.new
+    end
+
+    def elb
+      @elb ||= Aws::ElasticLoadBalancingV2::Client.new
     end
   end
 end

data/lib/ufo/base.rb

@@ -0,0 +1,32 @@
+module Ufo
+  class Base
+    extend Memoist
+    include Stack::Helper
+
+    def initialize(service, options={})
+      @service = switch_current(service)
+      @options = options
+
+      @pretty_service_name = Ufo.pretty_service_name(@service)
+      @cluster = @options[:cluster] || default_cluster
+      @stack_name = adjust_stack_name(@cluster, @service)
+    end
+
+    def switch_current(service)
+      Current.service!(service)
+    end
+
+    def info
+      Info.new(@service, @options)
+    end
+    memoize :info
+
+    def no_service_message
+      <<-EOL
+No #{@pretty_service_name.colorize(:green)} found.
+No CloudFormation stack named #{@stack_name} found.
+Are sure it exists?
+      EOL
+    end
+  end
+end

data/lib/ufo/cancel.rb

@@ -0,0 +1,23 @@
+module Ufo
+  class Cancel < Base
+    def run
+      stack = find_stack(@stack_name)
+      unless stack
+        puts "No #{@pretty_service_name} service to cancel."
+        puts "No #{@stack_name} stack to cancel. Exiting"
+        exit
+      end
+
+      puts "Canceling updates to #{@pretty_service_name}."
+      if stack.stack_status == "CREATE_IN_PROGRESS"
+        cloudformation.delete_stack(stack_name: @stack_name)
+        puts "Canceling stack creation."
+      elsif stack.stack_status =~ /_IN_PROGRESS$/
+        cloudformation.cancel_update_stack(stack_name: @stack_name)
+        puts "Canceling stack update."
+      else
+        puts "The stack is not in a state to that is cancelable: #{stack.stack_status}"
+      end
+    end
+  end
+end

data/lib/ufo/cli.rb

@@ -6,7 +6,11 @@ module Ufo
     class_option :verbose, type: :boolean
     class_option :mute, type: :boolean
     class_option :noop, type: :boolean
-    class_option :cluster, desc: "Cluster.  Overrides ufo/settings.yml."
+    class_option :cluster, desc: "Cluster.  Overrides .ufo/settings.yml."
+
+    desc "network SUBCOMMAND", "network subcommands"
+    long_desc Help.text(:network)
+    subcommand "network", Network
 
     desc "docker SUBCOMMAND", "docker subcommands"
     long_desc Help.text(:docker)
@@ -16,6 +20,10 @@ module Ufo
     long_desc Help.text(:tasks)
     subcommand "tasks", Tasks
 
+    desc "upgrade SUBCOMMAND", "upgrade subcommands"
+    long_desc Help.text(:upgrade)
+    subcommand "upgrade", Upgrade
+
     long_desc Help.text(:init)
     Init.cli_options.each do |args|
       option *args
@@ -24,50 +32,66 @@ module Ufo
 
     # common options to deploy. ship, and ships command
     ship_options = Proc.new do
-      option :task, desc: "ECS task name, to override the task name convention."
-      option :target_group, desc: "ELB Target Group ARN."
-      option :target_group_prompt, type: :boolean, desc: "Enable Target Group ARN prompt", default: true
-      option :wait, type: :boolean, desc: "Wait for deployment to complete", default: false
-      option :pretty, type: :boolean, default: true, desc: "Pretty format the json for the task definitions"
-      option :stop_old_tasks, type: :boolean, default: false, desc: "Stop old tasks after waiting for deploying to complete"
       option :ecr_keep, type: :numeric, desc: "ECR specific cleanup of old images.  Specifies how many images to keep.  Only runs if the images are ECR images. Defaults keeps all images."
+      # All elb options remember their 'state'
+      option :elb, desc: "Decides to create elb, not create elb or use existing target group."
+      option :elb_eip_ids, type: :array, desc: "EIP Allocation ids to use for network load balancer."
+      option :elb_type, desc: "ELB type: application or network. Keep current deployed elb type when not specified."
+      option :pretty, type: :boolean, default: true, desc: "Pretty format the json for the task definitions"
+      option :stop_old_tasks, type: :boolean, default: false, desc: "Stop old tasks as part of deployment to speed it up"
+      option :task, desc: "ECS task name, to override the task name convention."
+      option :wait, type: :boolean, desc: "Wait for deployment to complete", default: true
     end
 
     desc "deploy SERVICE", "Deploy task definition to ECS service without re-building the definition."
     long_desc Help.text(:deploy)
     ship_options.call
-    def deploy(service)
+    def deploy(service=:current)
+      service = service == :current ? Current.service! : service
       task_definition = options[:task] || service # convention
       Tasks::Register.register(task_definition, options)
-      ship = Ship.new(service, task_definition, options)
+      ship = Ship.new(service, options.merge(task_definition: task_definition))
       ship.deploy
     end
 
     desc "ship SERVICE", "Builds and ships container image to the ECS service."
     long_desc Help.text(:ship)
     ship_options.call
-    def ship(service)
+    def ship(service=:current)
+      service = service == :current ? Current.service! : service
       builder = build_docker
 
       task_definition = options[:task] || service # convention
       Tasks::Builder.ship(task_definition, options)
-      ship = Ship.new(service, task_definition, options)
+      ship = Ship.new(service, options.merge(task_definition: task_definition))
       ship.deploy
 
       cleanup(builder.image_name)
     end
 
+    desc "rollback SERVICE VERSION", "Rolls back to older task definition."
+    long_desc Help.text(:rollback)
+    def rollback(service=:current, version)
+      service = service == :current ? Current.service! : service
+      rollback = Rollback.new(service, options.merge(version: version))
+      rollback.deploy
+    end
+
     desc "ships [LIST_OF_SERVICES]", "Builds and ships same container image to multiple ECS services."
     long_desc Help.text(:ships)
     ship_options.call
+    option :wait, type: :boolean, desc: "Wait for deployment to complete", default: false
     def ships(*services)
       builder = build_docker
 
+      if services.empty? && !Current.services&.empty?
+        services = Current.services
+      end
       services.each_with_index do |service|
         service_name, task_definition_name = service.split(':')
         task_definition = task_definition_name || service_name # convention
         Tasks::Builder.ship(task_definition, options)
-        ship = Ship.new(service, task_definition, options)
+        ship = Ship.new(service, options.merge(task_definition: task_definition))
         ship.deploy
       end
 
@@ -76,28 +100,78 @@ module Ufo
 
     desc "task TASK_DEFINITION", "Run a one-time task."
     long_desc Help.text(:task)
-    option :docker, type: :boolean, desc: "Enable docker build and push", default: true
+    option :task_only, type: :boolean, desc: "Skip docker and task register steps. Only run the task."
     option :command, type: :array, aliases: 'c', desc: "Override the command used for the container"
     def task(task_definition)
-      Docker::Builder.build(options) if @options[:docker]
-      Tasks::Builder.ship(task_definition, options)
+      Docker::Builder.build(options) unless @options[:task_only]
+      Tasks::Builder.ship(task_definition, options) unless @options[:task_only]
       Task.new(task_definition, options).run
     end
 
+    desc "cancel SERVICE", "Cancel creation or update of the ECS service."
+    long_desc Help.text(:cancel)
+    option :sure, type: :boolean, desc: "By pass are you sure prompt."
+    def cancel(service=:current)
+      task_definition = options[:task] || service # convention
+      Cancel.new(service, options).run
+    end
+
+    desc "current SERVICE", "Switch the current service. Saves to `.ufo/current`"
+    long_desc Help.text(:current)
+    option :rm, type: :boolean, desc: "Remove all current settings. Removes `.ufo/current`"
+    option :service, desc: "Sets service as a current setting."
+    option :services, type: :array, desc: "Sets services as a current setting. This is used for ufo ships."
+    option :env_extra, desc: "Sets UFO_ENV_EXTRA as a current setting."
+    def current
+      Current.new(options).run
+    end
+
     desc "destroy SERVICE", "Destroy the ECS service."
     long_desc Help.text(:destroy)
     option :sure, type: :boolean, desc: "By pass are you sure prompt."
-    def destroy(service)
+    option :wait, type: :boolean, desc: "Wait for completion", default: true
+    def destroy(service=:current)
       task_definition = options[:task] || service # convention
       Destroy.new(service, options).bye
     end
 
+    desc "apps", "List apps."
+    long_desc Help.text(:apps)
+    def apps
+      Apps.new(options).list
+    end
+
+    desc "resources SERVICE", "The ECS service resources."
+    long_desc Help.text(:resources)
+    def resources(service=:current)
+      Info.new(service, options).run
+    end
+
     desc "scale SERVICE COUNT", "Scale the ECS service."
     long_desc Help.text(:scale)
-    def scale(service, count)
+    def scale(service=:current, count)
       Scale.new(service, count, options).update
     end
 
+    desc "ps SERVICE", "Show process info on ECS service."
+    long_desc Help.text(:ps)
+    option :summary, type: :boolean, default: true, desc: "Display summary header info."
+    def ps(service=:current)
+      Ps.new(service, options).run
+    end
+
+    desc "releases SERVICE", "Show possible 'releases' or task definitions for the service."
+    long_desc Help.text(:releases)
+    def releases(service=:current)
+      Releases.new(service, options).list
+    end
+
+    desc "stop SERVICE", "Stop tasks from old deployments.  Can speed up deployments with network load balancer."
+    long_desc Help.text(:stop)
+    def stop(service=:current)
+      Stop.new(service, options).run
+    end
+
     desc "completion *PARAMS", "Prints words for auto-completion."
     long_desc Help.text("completion")
     def completion(*params)
@@ -110,16 +184,6 @@ module Ufo
       Completer::Script.generate
     end
 
-    desc "upgrade3", "Upgrade from version 2 to 3."
-    def upgrade3
-      Upgrade3.new(options).run
-    end
-
-    desc "upgrade3_3_to_3_4", "Upgrade from version 3.3 to 3.4"
-    def upgrade3_3_to_3_4
-      Upgrade33_to_34.new(options).run
-    end
-
     desc "version", "Prints version number of installed ufo."
     def version
       puts VERSION

data/lib/ufo/core.rb

@@ -3,6 +3,8 @@ require 'yaml'
 
 module Ufo
   module Core
+    extend Memoist
+
     def check_task_definition!(task_definition)
       task_definition_path = "#{Ufo.root}/.ufo/output/#{task_definition}.json"
       unless File.exist?(task_definition_path)
@@ -18,12 +20,42 @@ module Ufo
       Pathname.new(path)
     end
 
-    @@env = nil
     def env
-      return @@env if @@env
       ufo_env = env_from_profile(ENV['AWS_PROFILE']) || 'development'
       ufo_env = ENV['UFO_ENV'] if ENV['UFO_ENV'] # highest precedence
-      @@env = ufo_env
+      ufo_env
+    end
+    memoize :env
+
+    def env_extra
+      env_extra = Current.env_extra
+      env_extra = ENV['UFO_ENV_EXTRA'] if ENV['UFO_ENV_EXTRA'] # highest precedence
+      return if env_extra&.empty?
+      env_extra
+    end
+    memoize :env_extra
+
+    def pretty_service_name(service)
+      [service, Ufo.env_extra].reject {|x| x==''}.compact.join('-')
+    end
+
+    def settings
+      Setting.new.data
+    end
+    memoize :settings
+
+    def cfn_profile
+      settings[:cfn_profile] || "default"
+    end
+
+    def check_ufo_project!
+      check_path = "#{Ufo.root}/.ufo/settings.yml"
+      unless File.exist?(check_path)
+        puts "ERROR: No settings file at #{check_path}.  Are you sure you are in a project with ufo setup?".colorize(:red)
+        puts "Current directory: #{Dir.pwd}"
+        puts "If you want to set up ufo for this prjoect, please create a settings file via: ufo init"
+        exit 1 unless ENV['TEST']
+      end
     end
 
     private

data/lib/ufo/current.rb

@@ -0,0 +1,104 @@
+require 'fileutils'
+require 'yaml'
+
+module Ufo
+  class Current
+    def initialize(options={})
+      Ufo.check_ufo_project!
+      @options = options
+      @file = ".ufo/current"
+      @path = "#{Ufo.root}/#{@file}"
+    end
+
+    def run
+      @options[:rm] ? rm : set
+    end
+
+    def rm
+      FileUtils.rm_f(@path)
+      puts "Current settings have been removed. Removed #{@file}"
+    end
+
+    def set
+      if @options.empty?
+        show
+      else
+        d = data # assign data to d to create local variable for merge to work
+        d = d.merge(@options).delete_if do |_,v|
+          v&.empty? || v == ['']
+        end
+        text = YAML.dump(d)
+        IO.write(@path, text)
+        puts "Current settings saved in .ufo/current"
+        show
+      end
+    end
+
+    def show
+      if data.empty?
+        puts <<-EOL
+There are no current settings.  To set a current service run:
+
+    ufo current --service my-service
+    ufo current -h # for more examples
+EOL
+        return
+      end
+
+      data.each do |key, value|
+        puts "Current #{key}: #{value}"
+      end
+    end
+
+    def data
+      YAML.load(IO.read(@path)) rescue {}
+    end
+
+    def env_extra
+      current = data["env_extra"]
+      return current unless current&.empty?
+    end
+
+    def self.env_extra
+      Current.new.env_extra
+    end
+
+    def services
+      return data["services"] || []
+    end
+
+    # reads services, returns [] if not set
+    def self.services
+      Current.new.services
+    end
+
+    def service
+      current = data["service"]
+      return current unless current&.empty?
+    end
+
+    # reads service, returns nil if not set
+    def self.service
+      Current.new.service
+    end
+
+    # reads service, will exit if current service not set
+    def self.service!(service=:current)
+      return service if service != :current
+
+      service = Current.service
+      return service if service
+
+      puts "ERROR: service must be specified.".colorize(:red)
+      puts <<-EOL
+Example:
+    ufo #{ARGV.first} SERVICE
+You can also set a current service to be remembered with:
+    ufo current SERVICE
+EOL
+      exit 1
+      # if want to display full help menu:
+      # Ufo::CLI.start(ARGV + ["-h"])
+    end
+  end
+end