balancer 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7210c71bd43eea249cb93535e4d264dc1ca79aa9c5899c588cb3b44c628e7335
-  data.tar.gz: 7ce6745e43f8f7d0d96e18c56a697d34c92fe478c0d6aff1891bd174c1444e77
+  metadata.gz: 1432379a1ef4e87892a5a90a1fbc50428156124a52b12a64d259cd05c451b10b
+  data.tar.gz: 469af119bc5fb4093ce3a84b4cf008167dac9f58ec6041b0d11af5efe3d2fc6b
 SHA512:
-  metadata.gz: a306a2be2701c9b876128137fa31b712609e645cf534a17b6289d81bd1935e4e0eb2bfcd5f59f7ed317945866c844c7ab433feac05030d2d5b0ecf5d69ccfb91
-  data.tar.gz: b1a64e2e011970038894b81f2ac57ee5f16f157cc91276e4cf21f872a114293268697bb239a3fed251396f7891ce2e50e093a004cec120b863988c2a2cef9cae
+  metadata.gz: 94760e1fab90cc6762b8508bdc1e01a93a887c7dee4d5a4c8c7d141a802531465901a7da28ea17b6d40c5834f69acd6689e3fb550cd1babef54c08770738624e
+  data.tar.gz: 8b83a2318937d7ec0ee8110e6a17cd86048722008ef21500fd57a0499adcb1b5a030e1095969babb4e7d233836a5c364e16fd8812ac1f7284c4b9933419ded24

data/.circleci/bin/commit_docs.sh

@@ -0,0 +1,26 @@
+#!/bin/bash -eux
+
+# Even though specs also generate docs, lets run again to ensure clean slate
+rake docs
+
+out=$(git status docs)
+if [[ "$out" = *"nothing to commit"* ]]; then
+  exit
+fi
+
+COMMIT_MESSAGE="docs updated by circleci"
+
+# If the last commit already updated the docs, then exit.
+# Preventable measure to avoid infinite loop.
+if git log -1 --pretty=oneline | grep "$COMMIT_MESSAGE" ; then
+  exit
+fi
+
+# If reach here, we have some changes on docs that we should commit.
+# Even though s
+git add docs
+git commit -m "$COMMIT_MESSAGE"
+
+# https://makandracards.com/makandra/12107-git-show-current-branch-name-only
+current_branch=$(git rev-parse --abbrev-ref HEAD)
+git push origin "$current_branch"

data/.circleci/config.yml

@@ -0,0 +1,70 @@
+# Ruby CircleCI 2.0 configuration file
+#
+# Check https://circleci.com/docs/2.0/language-ruby/ for more details
+#
+version: 2
+jobs:
+  build:
+    docker:
+      # specify the version you desire here
+      - image: circleci/ruby:2.5.1-node-browsers
+
+      # Specify service dependencies here if necessary
+      # CircleCI maintains a library of pre-built images
+      # documented at https://circleci.com/docs/2.0/circleci-images/
+      # - image: circleci/postgres:9.4
+
+    working_directory: ~/repo
+
+    steps:
+      - checkout
+
+      - run:
+          name: submodule sync
+          command: |
+            git submodule sync
+            git submodule update --init
+
+      # Download and cache dependencies
+      - restore_cache:
+          keys:
+          - v1-dependencies-{{ checksum "Gemfile" }}
+          # fallback to using the latest cache if no exact match is found
+          - v1-dependencies-
+
+      - run:
+          name: install dependencies
+          command: |
+            bundle install --jobs=4 --retry=3 --path vendor/bundle
+
+      - save_cache:
+          paths:
+            - ./vendor/bundle
+          key: v1-dependencies-{{ checksum "Gemfile" }}
+
+      # specs need git configured ad commit_docs.sh required it also
+      - run:
+          name: configure git
+          command: |
+            git config --global user.email "tongueroo@gmail.com"
+            git config --global user.name "Tung Nguyen"
+
+      # run tests!
+      - run:
+          name: run tests
+          command: |
+            mkdir /tmp/test-results
+            bundle exec rspec
+
+      # - run:
+      #     name: commit cli reference docs
+      #     command: |
+      #       chmod a+x -R .circleci/bin
+      #       .circleci/bin/commit_docs.sh
+
+      # collect reports
+      - store_test_results:
+          path: /tmp/test-results
+      - store_artifacts:
+          path: /tmp/test-results
+          destination: test-results

data/.rspec

@@ -1,2 +1,3 @@
 --color
 --format documentation
+--require spec_helper

data/CHANGELOG.md

@@ -3,5 +3,10 @@
 All notable changes to this project will be documented in this file.
 This project *tries* to adhere to [Semantic Versioning](http://semver.org/), even before v1.0.
 
+## [0.2.0]
+- balancer create command
+- balancer destroy command
+- auto create security group that opens up the listener port
+
 ## [0.1.0]
 - Initial release.

data/Gemfile.lock

@@ -1,9 +1,12 @@
 PATH
   remote: .
   specs:
-    balancer (0.1.0)
+    balancer (0.2.0)
       activesupport
+      aws-sdk-ec2
+      aws-sdk-elasticloadbalancingv2
       colorize
+      memoist
       thor
 
 GEM
@@ -14,6 +17,20 @@ GEM
       i18n (>= 0.7, < 2)
       minitest (~> 5.1)
       tzinfo (~> 1.1)
+    aws-eventstream (1.0.0)
+    aws-partitions (1.91.0)
+    aws-sdk-core (3.21.2)
+      aws-eventstream (~> 1.0)
+      aws-partitions (~> 1.0)
+      aws-sigv4 (~> 1.0)
+      jmespath (~> 1.0)
+    aws-sdk-ec2 (1.35.0)
+      aws-sdk-core (~> 3)
+      aws-sigv4 (~> 1.0)
+    aws-sdk-elasticloadbalancingv2 (1.10.0)
+      aws-sdk-core (~> 3)
+      aws-sigv4 (~> 1.0)
+    aws-sigv4 (1.0.2)
     byebug (10.0.0)
     cli_markdown (0.1.0)
     codeclimate-test-reporter (1.0.8)
@@ -24,7 +41,9 @@ GEM
     docile (1.1.5)
     i18n (1.0.1)
       concurrent-ruby (~> 1.0)
+    jmespath (1.4.0)
     json (2.1.0)
+    memoist (0.16.0)
     minitest (5.11.3)
     rake (12.3.0)
     rspec (3.7.0)

data/README.md

@@ -1,32 +1,61 @@
 # Balancer
 
-[![Gem Version](https://badge.fury.io/rb/GEMNAME.png)](http://badge.fury.io/rb/GEMNAME)
-[![CircleCI](https://circleci.com/gh/USER/REPO.svg?style=svg)](https://circleci.com/gh/USER/REPO)
-[![Dependency Status](https://gemnasium.com/USER/REPO.png)](https://gemnasium.com/USER/REPO)
-[![Coverage Status](https://coveralls.io/repos/USER/REPO/badge.png)](https://coveralls.io/r/USER/REPO)
-[![Join the chat at https://gitter.im/USER/REPO](https://badges.gitter.im/USER/REPO.svg)](https://gitter.im/USER/REPO?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge&utm_content=badge)
-[![Support](https://img.shields.io/badge/get-support-blue.svg)](https://boltops.com?utm_source=badge&utm_medium=badge&utm_campaign=cli-template)
+[![Gem Version](https://badge.fury.io/rb/balancer.svg)](https://badge.fury.io/rb/balancer)
+[![CircleCI](https://circleci.com/gh/tongueroo/balancer.svg?style=svg)](https://circleci.com/gh/tongueroo/balancer)[![Support](https://img.shields.io/badge/get-support-blue.svg)](https://boltops.com?utm_source=badge&utm_medium=badge&utm_campaign=balancer)
 
-TODO: Write a gem description
+Tool to create ELB load balancers with a target group and listener.  It's performs similar steps to this AWS Tutorial: [Create an Application Load Balancer Using the AWS CLI
+](https://docs.aws.amazon.com/elasticloadbalancing/latest/application/tutorial-application-load-balancer-cli.html)
 
 ## Usage
 
-    balancer hello yourname
-    balancer sub:goodbye yourname
+Quick start to creating and destroying a load balancer.
 
-The CLI tool also detects and tasks in the current folder's Rakefile and delegate to those tasks.
+    cd project
+    balancer init --vpc-id vpc-123 --subnets subnet-123 subnet-456 --security-groups sg-123
+    # edit .balancer/profiles/default.yml to fit your needs
+    balancer create my-elb
+    balancer destroy my-elb
 
-## Installation
+### Profiles
+
+Balancer has a concept of profiles.  Profiles have preconfigured settings like subnets and vpc_id.  The params in the profiles are passed to the ruby [aws-sdk](https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/ElasticLoadBalancingV2/Client.html) api calls  `create_load_balancer`, `create_target_group`, and `create_listener`:
+
+```yaml
+---
+create_load_balancer:
+  subnets: # at least 2 subnets groups required
+    - subnet-123
+    - subnet-345
+  security_groups: # optional thanks to the automatically created security group by balancer
+    - sg-123 # additional security groups to use
+create_target_group:
+  # vpc_id is required
+  vpc_id: vpc-123
+  # name: ... # automatically named, matches the load balancer name. override here
+  protocol: HTTP # required
+  port: 80 # required
+create_listener:
+  protocol: HTTP # required
+  port: 80 # required, this is is the port that gets open on the security group
+```
+
+### Security Groups
 
-Add this line to your application's Gemfile:
+Balancer automatically creates a security group with the same name as the elb and opens up the port configured on the listener.  To disable this behavior, use the `--no-security-group` optional.  If you use this option, you must specify you own security group in the profile file, since at least 1 security group is required.  By default, the security group opens up `0.0.0.0/0`. If you want to override this use `--sg-cdir`, example:
 
-    gem "balancer"
+	balancer create my-elb --sg-cdir 10.0.0.0/16
 
-And then execute:
+When you destroy the ELB like so:
 
-    bundle
+	balancer destroy my-elb
 
-Or install it yourself as:
+Balancer also attempts to destroy the security group if:
+
+* The security group is tagged with the `balancer=my-elb` tag. Balancer automatically adds this tag when creating the ELB.
+* There are no dependencies on the security group. If there are dependencies the ELB is deleted but the security group is left behind for you to clean up.
+
+
+## Installation
 
     gem install balancer
 

data/balancer.gemspec

@@ -19,7 +19,10 @@ Gem::Specification.new do |spec|
   spec.require_paths = ["lib"]
 
   spec.add_dependency "activesupport"
+  spec.add_dependency "aws-sdk-ec2"
+  spec.add_dependency "aws-sdk-elasticloadbalancingv2"
   spec.add_dependency "colorize"
+  spec.add_dependency "memoist"
   spec.add_dependency "thor"
 
   spec.add_development_dependency "bundler"

data/lib/balancer.rb

@@ -1,11 +1,25 @@
 $:.unshift(File.expand_path("../", __FILE__))
 require "balancer/version"
+require "colorize"
+require "memoist"
 
 module Balancer
+  autoload :AwsService, "balancer/aws_service"
   autoload :Help, "balancer/help"
+  autoload :Setting, "balancer/setting"
+  autoload :Base, "balancer/base"
+  autoload :Profile, "balancer/profile"
+  autoload :Init, "balancer/init"
+  autoload :Core, "balancer/core"
   autoload :Command, "balancer/command"
   autoload :CLI, "balancer/cli"
-  autoload :Sub, "balancer/sub"
+  autoload :Create, "balancer/create"
   autoload :Completion, "balancer/completion"
   autoload :Completer, "balancer/completer"
+  autoload :Destroy, "balancer/destroy"
+  autoload :Param, "balancer/param"
+  autoload :OptionTransformer, "balancer/option_transformer"
+  autoload :SecurityGroup, "balancer/security_group"
+
+  extend Core
 end

data/lib/balancer/aws_service.rb

@@ -0,0 +1,14 @@
+require 'aws-sdk-ec2'
+require 'aws-sdk-elasticloadbalancingv2'
+
+module Balancer
+  module AwsService
+    def elb
+      @elb ||= Aws::ElasticLoadBalancingV2::Client.new
+    end
+
+    def ec2
+      @ec2 ||= Aws::EC2::Client.new
+    end
+  end
+end

data/lib/balancer/base.rb

@@ -0,0 +1,37 @@
+module Balancer
+  class Base
+    def initialize(options={})
+      @options = options.clone
+      @name = randomize(@options[:name])
+      Balancer.validate_in_project!
+    end
+
+    # Appends a short random string at the end of the ec2 instance name.
+    # Later we will strip this same random string from the name.
+    # Very makes it convenient.  We can just type:
+    #
+    #   balancer create server --randomize
+    #
+    # instead of:
+    #
+    #   balancer create server-123 --profile server
+    #
+    def randomize(name)
+      if @options[:randomize]
+        random = (0...3).map { (65 + rand(26)).chr }.join.downcase # Ex: jhx
+        [name, random].join('-')
+      else
+        name
+      end
+    end
+
+    # Strip the random string at end of the ec2 instance name
+    def derandomize(name)
+      if @options[:randomize]
+        name.sub(/-(\w{3})$/,'') # strip the random part at the end
+      else
+        name
+      end
+    end
+  end
+end

data/lib/balancer/cli.rb

@@ -3,17 +3,31 @@ module Balancer
     class_option :verbose, type: :boolean
     class_option :noop, type: :boolean
 
-    desc "hello NAME", "Say hello to NAME."
-    long_desc Help.text(:hello)
-    option :from, desc: "from person"
-    def hello(name="you")
-      puts "from: #{options[:from]}" if options[:from]
-      puts "Hello #{name}"
+    desc "create NAME", "Create Load Balancer."
+    long_desc Help.text(:create)
+    # create_load_balancer options
+    option :subnets, type: :array, desc: "Subnets"
+    option :security_groups, type: :array, desc: "Security groups"
+    # create_target_group options
+    option :vpc_id, type: :array, desc: "Vpc id"
+    option :target_group_name, desc: "Target group name"
+    # security_group options
+    option :sg_cidr, default: "0.0.0.0/0", desc: "Security group cidr range"
+    def create(name)
+      Create.new(options.merge(name: name)).run
     end
 
-    desc "sub SUBCOMMAND", "sub subcommands"
-    long_desc Help.text(:sub)
-    subcommand "sub", Sub
+    desc "destroy NAME", "Destroy Load Balancer and associated target group."
+    long_desc Help.text(:destroy)
+    def destroy(name)
+      Destroy.new(options.merge(name: name)).run
+    end
+
+    long_desc Help.text(:init)
+    Init.cli_options.each do |args|
+      option *args
+    end
+    register(Init, "init", "init", "Sets up balancer for project")
 
     desc "completion *PARAMS", "Prints words for auto-completion."
     long_desc Help.text("completion")

data/lib/balancer/core.rb

@@ -0,0 +1,26 @@
+require 'pathname'
+require 'yaml'
+
+module Balancer
+  module Core
+    def root
+      path = ENV['BALANCER_ROOT'] || '.'
+      Pathname.new(path)
+    end
+
+    def profile
+      ENV['BALANCER_PROFILE'] || 'default'
+    end
+
+    def settings
+      Setting.new.data
+    end
+
+    def validate_in_project!
+      unless File.exist?("#{root}/.balancer")
+        puts "Could not find a .balancer folder in the current directory.  It does not look like you are running this command within a balancer project.  Please confirm that you are in a balancer project and try again.".colorize(:red)
+        exit
+      end
+    end
+  end
+end

data/lib/balancer/create.rb

@@ -0,0 +1,117 @@
+require 'yaml'
+
+module Balancer
+  class Create
+    extend Memoist
+    include AwsService
+    include SecurityGroup
+
+    def initialize(options)
+      @options = options
+      @name = options[:name]
+    end
+
+    # https://docs.aws.amazon.com/elasticloadbalancing/latest/application/tutorial-application-load-balancer-cli.html
+    def run
+      if ENV['TEST'] # ghetto way to for sanity cli specs
+        puts "Creating load balancer"
+        return
+      end
+
+      if elb_exists?
+        puts "Load balancer #{@name} already exists"
+        return
+      end
+
+      @security_group_id = create_security_group
+      create_elb
+      create_target_group
+      create_listener
+    end
+
+    def elb_exists?
+      begin
+        resp = elb.describe_load_balancers(names: [@name])
+        true
+      rescue Aws::ElasticLoadBalancingV2::Errors::LoadBalancerNotFound
+        false
+      end
+    end
+
+    def create_elb
+      puts "Creating load balancer with params:"
+      params = param.create_load_balancer
+      params[:security_groups] ||= []
+      params[:security_groups] += [@security_group_id]
+      params[:security_groups] = params[:security_groups].uniq
+      pretty_display(params)
+      aws_cli_command("aws elbv2 create-load-balancer", params)
+      return if @options[:noop]
+
+      begin
+        resp = elb.create_load_balancer(params)
+      rescue Exception => e
+        puts "ERROR: #{e.class}: #{e.message}".colorize(:red)
+        exit 1
+      end
+
+      elb = resp.load_balancers.first
+      puts "Load balancer created: #{elb.load_balancer_arn}"
+      @load_balancer_arn = elb.load_balancer_arn # used later
+      puts
+    end
+
+    def create_target_group
+      puts "Creating target group with params:"
+      params = param.create_target_group
+      pretty_display(params)
+      aws_cli_command("aws elbv2 create-target-group", params)
+
+      begin
+        resp = elb.create_target_group(params)
+      rescue Exception => e
+        puts "ERROR: #{e.class}: #{e.message}".colorize(:red)
+        exit 1
+      end
+      target_group = resp.target_groups.first
+      puts "Target group created: #{target_group.target_group_arn}"
+      @target_group_arn = target_group.target_group_arn # used later
+      add_tags(@target_group_arn)
+      puts
+    end
+
+    def create_listener
+      puts "Creating listener with params:"
+      params = param.create_listener
+      params.merge!(
+        load_balancer_arn: @load_balancer_arn,
+        default_actions: [{type: "forward", target_group_arn: @target_group_arn}]
+      )
+      pretty_display(params)
+      aws_cli_command("aws elbv2 create-listener", params)
+
+      resp = run_with_error_handling do
+        elb.create_listener(params)
+      end
+      listener = resp.listeners.first
+      puts "Listener created: #{listener.listener_arn}"
+      puts
+    end
+
+    def run_with_error_handling
+      yield
+    rescue Exception => e
+      puts "ERROR: #{e.class}: #{e.message}".colorize(:red)
+      exit 1
+    end
+
+    def add_tags(arn)
+      params = {
+        resource_arns: [arn],
+        tags: [{ key: "balancer", value: @name }]
+      }
+      aws_cli_command("aws elbv2 add-tags", params)
+      elb.add_tags(params)
+    end
+  end
+end

data/lib/balancer/destroy.rb

@@ -0,0 +1,51 @@
+module Balancer
+  class Destroy
+    extend Memoist
+    include AwsService
+    include SecurityGroup
+
+    def initialize(options)
+      @options = options
+      @name = options[:name]
+    end
+
+    def run
+      puts "Destroying ELB and target groups associated with: #{@name}"
+      return if @options[:noop]
+
+      begin
+        resp = elb.describe_load_balancers(names: [@name])
+      rescue Aws::ElasticLoadBalancingV2::Errors::LoadBalancerNotFound
+        puts "Load balancer '#{@name}' not found. Exiting.".colorize(:red)
+        exit 1
+      end
+
+      load_balancer = resp.load_balancers.first
+
+      # Must load resources to be deleted into memory to delete them later since there
+      # are dependencies and they won't be available to query after deleting some of the
+      # resources.
+      resp = elb.describe_listeners(load_balancer_arn: load_balancer.load_balancer_arn)
+      listeners = resp.listeners
+      resp = elb.describe_target_groups(load_balancer_arn: load_balancer.load_balancer_arn)
+      groups = resp.target_groups
+
+      listeners.each do |listener|
+        elb.delete_listener(listener_arn: listener.listener_arn)
+        puts "Deleted listener: #{listener.listener_arn}"
+      end
+
+      groups.each do |group|
+        elb.delete_target_group(target_group_arn: group.target_group_arn)
+        puts "Deleted target group: #{group.target_group_arn}"
+      end
+
+      resp = elb.delete_load_balancer(
+        load_balancer_arn: load_balancer.load_balancer_arn,
+      )
+      puts "Deleted load balancer: #{load_balancer.load_balancer_arn}"
+
+      destroy_security_group
+    end
+  end
+end

data/lib/balancer/help/create.md

@@ -0,0 +1,5 @@
+## Examples
+
+    balancer create NAME
+    balancer create NAME --type application
+    balancer create NAME --type network

data/lib/balancer/help/destroy.md

@@ -0,0 +1,3 @@
+## Examples
+
+    balancer destroy NAME

data/lib/balancer/help/init.md

@@ -0,0 +1,4 @@
+## Examples
+
+    balancer init
+    balancer init --vpc-id vpc-123 --subnets subnet-123 subnet-456 --security-groups sg-123

data/lib/balancer/init.rb

@@ -0,0 +1,49 @@
+require 'fileutils'
+require 'colorize'
+require 'active_support/core_ext/string'
+require 'thor'
+require 'bundler'
+
+module Balancer
+  class Init < Thor::Group
+    include Thor::Actions
+
+    # Ugly, but when the class_option is only defined in the Thor::Group class
+    # it doesnt show up with cli-template new help :(
+    # If anyone knows how to fix this let me know.
+    # Also options from the cli can be pass through to here
+    def self.cli_options
+      [
+        [:force, type: :boolean, desc: "Bypass overwrite are you sure prompt for existing files."],
+        [:git, type: :boolean, default: true, desc: "Git initialize the project"],
+        [:subnets, type: :array, default: ["REPLACE_ME"], desc: "Subnets"],
+        [:security_groups, type: :array, default: ["REPLACE_ME"], desc: "Security groups"],
+        [:vpc_id, default: "REPLACE_ME", desc: "Vpc id"],
+      ]
+    end
+
+    cli_options.each do |args|
+      class_option *args
+    end
+
+    def self.source_root
+      File.expand_path("../template", File.dirname(__FILE__))
+    end
+
+    def init_project
+      puts "Setting up balancer files."
+      directory ".", "."
+    end
+
+    def user_message
+      puts <<-EOL
+#{"="*64}
+Congrats 🎉 Balancer starter files succesfully created.
+
+Check out .balancer/profiles/default.yml make make sure the settings like subnets and vpc_id are okay.  Then run `balanace create` to to create an ELB, Target Group and listener. Example:
+
+    balancer create my-elb
+EOL
+    end
+  end
+end

data/lib/balancer/option_transformer.rb

@@ -0,0 +1,39 @@
+require 'active_support/core_ext/string'
+require 'active_support/core_ext/hash'
+
+module Balancer
+  class OptionTransformer
+    def to_cli(options)
+      params = []
+      options.each do |k,v|
+        case v
+        when Symbol, String, Integer
+          params << key_to_cli_option(k) + ' ' + v.to_s
+        when Array
+          values = []
+          v.each do |o|
+            if o.is_a?(Hash)
+              o.each do |x,y|
+                values << "#{x.to_s.camelize}=#{y}"
+              end
+            else # assume string
+              values << o
+            end
+          end
+
+          list = v.first.is_a?(Hash) ? values.join(',') : values.join(' ')
+          params << key_to_cli_option(k) + ' ' + list
+        else
+          puts "v.class: #{v.class.inspect}"
+          raise "the roof"
+        end
+      end
+      params.join(' ')
+    end
+
+    # resource_arns => --resource-arns
+    def key_to_cli_option(key)
+      '--' + key.to_s.gsub('_','-')
+    end
+  end
+end

data/lib/balancer/param.rb

@@ -0,0 +1,43 @@
+require 'active_support/core_ext/hash'
+
+module Balancer
+  class Param
+    extend Memoist
+
+    def initialize(options)
+      @options = options
+    end
+
+    def create_load_balancer
+      params = settings["create_load_balancer"].deep_symbolize_keys
+      params = merge_option(params, :name)
+      params = merge_option(params, :subnets)
+      params = merge_option(params, :security_groups)
+      params[:tags] = [{ key: "balancer", value: "balancer" }]
+      params
+    end
+    memoize :create_load_balancer
+
+    def create_target_group
+      params = settings["create_target_group"].deep_symbolize_keys
+      params[:name] ||= @options[:name] if @options[:name] # settings take precedence
+      params = merge_option(params, :vpc_id)
+      params
+    end
+    memoize :create_target_group
+
+    def create_listener
+      settings["create_listener"].deep_symbolize_keys
+    end
+    memoize :create_listener
+
+    def merge_option(params, option_key)
+      params[option_key] = @options[option_key] if @options[option_key]
+      params
+    end
+
+    def settings
+      @settings ||= Balancer.settings
+    end
+  end
+end

data/lib/balancer/profile.rb

@@ -0,0 +1,65 @@
+module Balancer
+  class Profile < Base
+    include Balancer::Template
+
+    def load
+      return @profile_params if @profile_params
+
+      check!
+
+      file = profile_file(profile_name)
+      @profile_params = load_profile(file)
+    end
+
+    def check!
+      file = profile_file(profile_name)
+      return if File.exist?(file)
+
+      puts "Unable to find a #{file.colorize(:green)} profile file."
+      puts "Please double check that it exists or that you specified the right profile.".colorize(:red)
+      exit 1
+    end
+
+    def load_profile(file)
+      return {} unless File.exist?(file)
+
+      puts "Using profile: #{file}".colorize(:green)
+      text = RenderMePretty.result(file, context: context)
+      begin
+        data = YAML.load(text)
+      rescue Psych::SyntaxError => e
+        tmp_file = file.sub(".balancer", "tmp")
+        FileUtils.mkdir_p(File.dirname(tmp_file))
+        IO.write(tmp_file, text)
+        puts "There was an error evaluating in your yaml file #{file}".colorize(:red)
+        puts "The evaludated yaml file has been saved at #{tmp_file} for debugging."
+        puts "ERROR: #{e.message}"
+        exit 1
+      end
+      data ? data : {} # in case the file is empty
+      data.has_key?("create_load_balancer") ? data["create_load_balancer"] : data
+    end
+
+    # Determines a valid profile_name. Falls back to default
+    def profile_name
+      # allow user to specify the path also
+      if @options[:profile] && File.exist?(@options[:profile])
+        filename_profile = File.basename(@options[:profile], '.yml')
+      end
+
+      name = derandomize(@name)
+      if File.exist?(profile_file(name))
+        name_profile = name
+      end
+
+      filename_profile ||
+      @options[:profile] ||
+      name_profile || # conventional profile is the name of the elb
+      "default"
+    end
+
+    def profile_file(name)
+      "#{Balancer.root}/.balancer/#{name}.yml"
+    end
+  end
+end

data/lib/balancer/security_group.rb

@@ -0,0 +1,139 @@
+module Balancer
+  module SecurityGroup
+    extend Memoist
+
+    def create_security_group
+      sg = find_security_group(@name)
+      group_id = sg.group_id if sg
+
+      unless group_id
+        puts "Creating security group #{@name} in vpc #{sg_vpc_id}"
+        params = {group_name: @name, description: @name, vpc_id: sg_vpc_id}
+        aws_cli_command("aws ec2 create-security-group", params)
+        begin
+          resp = ec2.create_security_group(params)
+        rescue Aws::EC2::Errors::InvalidVpcIDNotFound => e
+          puts "ERROR: #{e.class} #{e.message}".colorize(:red)
+          exit 1
+        end
+        group_id = resp.group_id
+        puts "Created security group: #{group_id}"
+      end
+
+      authorize_elb_port(group_id)
+
+      ec2.create_tags(resources: [group_id], tags: [{
+        key: "Name",
+        value: @name
+      },
+        key: "balancer",
+        value: @name
+      ])
+
+      group_id
+    end
+
+    def authorize_elb_port(group_id)
+      resp = ec2.describe_security_groups(group_ids: [group_id])
+      sg = resp.security_groups.first
+
+      already_authorized = sg.ip_permissions.find do |perm|
+        perm.from_port == 80 &&
+        perm.to_port == 80
+        perm.ip_ranges.find { |ip_range| ip_range.cidr_ip == @options[:sg_cidr] }
+      end
+      if already_authorized
+        return
+      end
+
+      listener_port = param.create_listener[:port]
+
+      # authorize the matching port in the create_listener setting
+      params = {group_id: group_id, protocol: "tcp", port: listener_port, cidr: @options[:sg_cidr]}
+      puts "Authorizing listening port for security group"
+      aws_cli_command("aws ec2 authorize-security-group-ingress", params)
+      ec2.authorize_security_group_ingress(
+        group_id: params[:group_id],
+        ip_permissions: [
+          from_port: listener_port,
+          to_port: listener_port,
+          ip_protocol: "tcp",
+          ip_ranges: [
+            cidr_ip: @options[:sg_cidr],
+            description: "balancer #{@name}"
+          ]
+        ]
+      )
+    end
+
+    def destroy_security_group
+      sg = find_security_group(@name)
+      return unless sg
+
+      balancer_tag = sg.tags.find { |t| t.key == "balancer" && t.value == @name }
+      unless balancer_tag
+        puts "WARN: not destroying the #{@name} security group because it doesn't have a matching balancer tag".colorize(:yellow)
+        return
+      end
+
+      puts "Deleting security group #{@name} in vpc #{sg_vpc_id}"
+      params = {group_id: sg.group_id}
+      aws_cli_command("aws ec2 delete-security-group", params)
+
+      tries = 0
+      begin
+        ec2.delete_security_group(params)
+        puts "Deleted security group: #{sg.group_id}"
+      rescue Aws::EC2::Errors::DependencyViolation => e
+        sleep 2**tries
+        tries += 1
+        if tries <= 4
+          # retry because it takes some time for the load balancer to be deleted
+          # and that can cause a DependencyViolation exception
+          retry
+        else
+          puts "WARN: #{e.class} #{e.message}".colorize(:yellow)
+          puts "Unable to delete the security group because it's still in use by another resource. Leaving the security group."
+          end
+      end
+    end
+
+    # Use security group that is set in the profile under create_target_group
+    def sg_vpc_id
+      param.create_target_group[:vpc_id]
+    end
+    memoize :sg_vpc_id
+
+    def find_security_group(name)
+      resp = ec2.describe_security_groups(filters: [
+        {name: "group-name", values: ["my-elb"]},
+        {name: "vpc-id", values: [sg_vpc_id]},
+      ])
+      resp.security_groups.first
+    end
+    memoize :find_security_group
+
+    # Few other common methods also included here
+
+    def param
+      Param.new(@options)
+    end
+    memoize :param
+
+    def pretty_display(data)
+      data = data.deep_stringify_keys
+      puts YAML.dump(data)
+    end
+
+    def option_transformer
+      Balancer::OptionTransformer.new
+    end
+    memoize :option_transformer
+
+    def aws_cli_command(aws_command, params)
+      # puts "Equivalent aws cli command:"
+      cli_options = option_transformer.to_cli(params)
+      puts "  #{aws_command} #{cli_options}".colorize(:light_blue)
+    end
+  end
+end

data/lib/balancer/setting.rb

@@ -0,0 +1,47 @@
+require 'yaml'
+require 'active_support/core_ext/hash'
+
+module Balancer
+  class Setting
+    extend Memoist
+
+    def data
+      settings_file = Balancer.profile || 'default'
+      settings_file += ".yml"
+
+      profile = yaml_file("#{Balancer.root}/.balancer/profiles/#{settings_file}")
+      user = yaml_file("#{home}/.balancer/#{settings_file}")
+      default_file = File.expand_path("../default/settings.yml", __FILE__)
+      default = yaml_file(default_file)
+
+      data = merge(default, user, profile)
+
+      if ENV['DEBUG_SETTINGS']
+        puts "settings data:"
+        pp data
+      end
+      data
+    end
+    memoize :data
+
+    def merge(*hashes)
+      hashes.inject({}) do |result, hash|
+        # note: important to compact for keys with nil value
+        result.deep_merge(hash.compact)
+      end
+    end
+
+    # Any empty file will result in "false".  Lets ensure that an empty file
+    # loads an empty hash instead.
+    def yaml_file(path)
+      # puts "yaml_file #{path}"
+      return {} unless File.exist?(path)
+      YAML.load_file(path) || {}
+    end
+
+    def home
+      # hack but fast
+      ENV['TEST'] ? "spec/fixtures/home" : ENV['HOME']
+    end
+  end
+end

data/lib/balancer/version.rb

@@ -1,3 +1,3 @@
 module Balancer
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

data/lib/template/.balancer/profiles/default.yml.tt

@@ -0,0 +1,21 @@
+---
+# https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/ElasticLoadBalancingV2/Client.html
+# The keys correspond to methods called in the aws-sdk
+create_load_balancer:
+  subnets: # at least 2 subnets required
+<% options[:subnets].each do |subnet| -%>
+    - <%= subnet %>
+<% end -%>
+  security_groups: # optional thanks to the automatically created security
+<% options[:security_groups].each do |security_group| -%>
+    - <%= security_group %>
+<% end -%>
+create_target_group:
+  # vpc_id is required
+  vpc_id: <%= options[:vpc_id] %>
+  # name: ... # automatically named, matches the load balancer name. override here
+  protocol: HTTP # required
+  port: 80 # required
+create_listener:
+  protocol: HTTP # required
+  port: 80 # required

data/spec/fixtures/project/.balancer/profiles/default.yml

@@ -0,0 +1,16 @@
+---
+create_load_balancer:
+  subnets: # required
+    - subnet-aaa
+    - subnet-bbb
+  security_groups: # required
+    - sg-123
+create_target_group:
+  # vpc_id is required
+  vpc_id: vpc-123
+  # name: ... # automatically named, matches the load balancer name. override here
+  protocol: HTTP # required
+  port: 80 # required
+create_listener:
+  protocol: HTTP # required
+  port: 80 # required

data/spec/lib/cli_spec.rb

@@ -1,32 +1,23 @@
-require "spec_helper"
-
 describe Balancer::CLI do
   before(:all) do
-    @args = "--from Tung"
+    @args = "--noop"
   end
 
   describe "balancer" do
-    it "hello" do
-      out = execute("exe/balancer hello world #{@args}")
-      expect(out).to include("from: Tung\nHello world")
+    it "create" do
+      out = execute("exe/balancer create my-elb #{@args}")
+      expect(out).to include("Creating load balancer")
     end
 
-    it "goodbye" do
-      out = execute("exe/balancer sub goodbye world #{@args}")
-      expect(out).to include("from: Tung\nGoodbye world")
+    it "destroy" do
+      out = execute("exe/balancer destroy my-elb #{@args}")
+      expect(out).to include("Destroying ELB")
     end
 
     commands = {
-      "hell" => "hello",
-      "hello" => "name",
-      "hello -" =>  "--from",
-      "hello name" => "--from",
-      "hello name --" => "--from",
-      "sub goodb" => "goodbye",
-      "sub goodbye" => "name",
-      "sub goodbye name" => "--from",
-      "sub goodbye name --" => "--from",
-      "sub goodbye name --from" => "--help",
+      "crea" => "create",
+      "create" => "name",
+      "dest" =>  "destroy",
     }
     commands.each do |command, expected_word|
       it "completion #{command}" do

data/spec/lib/option_transformer_spec.rb

@@ -0,0 +1,27 @@
+describe Balancer::OptionTransformer do
+  let(:create) { Balancer::OptionTransformer.new }
+
+  context '#to_cli' do
+    it "target group" do
+      options = {
+        port: 80,
+        protocol: "HTTP",
+        load_balancer_arn: "load_balancer_arn",
+        default_actions: [{type: "forward", target_group_arn: "target_group_arn"}],
+      }
+      text = create.to_cli(options)
+      # puts text
+      expect(text).to eq "--port 80 --protocol HTTP --load-balancer-arn load_balancer_arn --default-actions Type=forward,TargetGroupArn=target_group_arn"
+    end
+
+    it "tags" do
+      options = {
+        resource_arns: %w[arn1 arn2],
+        tags: [{ key: "balancer", value: "test-elb" }]
+      }
+      text = create.to_cli(options)
+      # puts text
+      expect(text).to eq "--resource-arns arn1 arn2 --tags Key=balancer,Value=test-elb"
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -1,5 +1,7 @@
 ENV["TEST"] = "1"
 
+ENV["BALANCER_ROOT"] = "spec/fixtures/project"
+
 # CodeClimate test coverage: https://docs.codeclimate.com/docs/configuring-test-coverage
 # require 'simplecov'
 # SimpleCov.start

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: balancer
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Tung Nguyen
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-05-08 00:00:00.000000000 Z
+date: 2018-06-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -24,6 +24,34 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-ec2
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-elasticloadbalancingv2
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: colorize
   requirement: !ruby/object:Gem::Requirement
@@ -38,6 +66,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: memoist
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: thor
   requirement: !ruby/object:Gem::Requirement
@@ -130,6 +172,8 @@ executables:
 extensions: []
 extra_rdoc_files: []
 files:
+- ".circleci/bin/commit_docs.sh"
+- ".circleci/config.yml"
 - ".gitignore"
 - ".rspec"
 - CHANGELOG.md
@@ -142,19 +186,34 @@ files:
 - balancer.gemspec
 - exe/balancer
 - lib/balancer.rb
+- lib/balancer/aws_service.rb
+- lib/balancer/base.rb
 - lib/balancer/cli.rb
 - lib/balancer/command.rb
 - lib/balancer/completer.rb
 - lib/balancer/completer/script.rb
 - lib/balancer/completer/script.sh
+- lib/balancer/core.rb
+- lib/balancer/create.rb
+- lib/balancer/destroy.rb
 - lib/balancer/help.rb
 - lib/balancer/help/completion.md
 - lib/balancer/help/completion_script.md
-- lib/balancer/help/hello.md
-- lib/balancer/help/sub/goodbye.md
+- lib/balancer/help/create.md
+- lib/balancer/help/destroy.md
+- lib/balancer/help/init.md
+- lib/balancer/init.rb
+- lib/balancer/option_transformer.rb
+- lib/balancer/param.rb
+- lib/balancer/profile.rb
+- lib/balancer/security_group.rb
+- lib/balancer/setting.rb
 - lib/balancer/sub.rb
 - lib/balancer/version.rb
+- lib/template/.balancer/profiles/default.yml.tt
+- spec/fixtures/project/.balancer/profiles/default.yml
 - spec/lib/cli_spec.rb
+- spec/lib/option_transformer_spec.rb
 - spec/spec_helper.rb
 homepage: https://github.com/tongueroo/balancer
 licenses:
@@ -176,10 +235,12 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.7.3
+rubygems_version: 2.7.6
 signing_key: 
 specification_version: 4
 summary: Balancer tool
 test_files:
+- spec/fixtures/project/.balancer/profiles/default.yml
 - spec/lib/cli_spec.rb
+- spec/lib/option_transformer_spec.rb
 - spec/spec_helper.rb

data/lib/balancer/help/hello.md

@@ -1,5 +0,0 @@
-Examples:
-
-    balancer hello
-    balancer hello NAME
-    balancer hello NAME --from me

data/lib/balancer/help/sub/goodbye.md

@@ -1,5 +0,0 @@
-Examples:
-
-    balancer sub:goodbye
-    balancer sub:goodbye NAME
-    balancer sub:goodbye NAME --from me