ecl 1.2.1 → 3.0.0.pre.alpha1

data/lib/eclair/group_item.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+module Eclair
+  class GroupItem < Item
+    attr_reader :label
+    attr_accessor :visible
+
+    def initialize label, items
+      @label = label
+      @items = items
+      @visible = true
+    end
+
+    def toggle_select
+      if @items.all?(&:selected)
+        @items.each{|i| i.select(false) }
+      else
+        @items.each{|i| i.select(true) }
+      end
+    end
+
+    def select state
+      @items.each{|i| i.select(state) }
+    end
+
+    def length
+      @items.length
+    end
+
+    def color
+      [Curses::COLOR_WHITE, -1, Curses::A_BOLD]
+    end
+  end
+end

data/lib/eclair/item.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+require "eclair/config"
+
+module Eclair
+  class Item
+    include ConfigHelper
+    attr_accessor :selected, :visible
+
+    def initialize
+      @selected = false
+      @visible = true
+    end
+
+    def toggle_select
+      @selected = !@selected
+    end
+
+    def select state
+      @selected = state
+    end
+
+    def id
+      raise "Not Implemented"
+    end
+
+    def command
+      raise "Not Implemented"
+    end
+
+    def header
+      raise "Not Implemented"
+    end
+
+    def title
+      raise "Not Implemented"
+    end
+
+    def search_key
+      raise "Not Implemented"
+    end
+  end
+end

data/lib/eclair/less_viewer.rb

@@ -1,3 +1,4 @@
+# frozen_string_literal: true
 module Eclair
   module LessViewer
     extend self
@@ -8,7 +9,7 @@ module Eclair
         q.flush
         out << "\n"
       end
-      
+
       def text(str, width = str.length)
         super(CodeRay.scan(str, :ruby).term, width)
       end

data/lib/eclair/provider.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+module Eclair
+  module Provider
+    extend self
+    attr_accessor :items
+
+    def filter_items search_buffer
+      @items.select{ |item| item&.search_key&.include?(search_buffer.downcase) || item.selected }
+    end
+
+    def require_prepare
+      raise "Not prepared" unless @prepared
+    end
+  end
+end

data/lib/eclair/providers/ec2.rb

@@ -0,0 +1,2 @@
+# frozen_string_literal: true
+require "eclair/providers/ec2/ec2_provider"

data/lib/eclair/providers/ec2/ec2_group_item.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+require "eclair/group_item"
+
+module Eclair
+  class EC2GroupItem < GroupItem
+    def header
+      running = @items.count{|i| i.instance.state[:code] == 16}
+      all = @items.count
+
+      <<-EOS
+      Group #{label}
+      #{running} Instances Running / #{all} Instances Total
+      EOS
+    end
+  end
+end

data/lib/eclair/providers/ec2/ec2_item.rb

@@ -0,0 +1,136 @@
+# frozen_string_literal: true
+require 'shellwords'
+require "aws-sdk-ec2"
+
+require "eclair/item"
+require "eclair/providers/ec2/ec2_provider"
+
+module Eclair
+  class EC2Item < Item
+    attr_reader :instance
+
+    def initialize instance
+      super()
+      @instance = instance
+    end
+
+    def id
+      @instance.id
+    end
+
+    def color
+      if @selected
+        [Curses::COLOR_YELLOW, -1, Curses::A_BOLD]
+      elsif !connectable?
+        [Curses::COLOR_BLACK, -1, Curses::A_BOLD]
+      else
+        [Curses::COLOR_WHITE, -1]
+      end
+    end
+
+    def command
+      hosts = [@instance.public_ip_address, @instance.private_ip_address].compact
+      ports = config.ssh_ports
+      ssh_options = config.ssh_options
+      ssh_command = config.ssh_command
+      username = config.ssh_username.call(image)
+      key_cmd = config.ssh_keys[@instance.key_name] ? "-i #{config.ssh_keys[@instance.key_name]}" : ""
+      format = config.exec_format
+
+      joined_cmd = hosts.map do |host|
+        ports.map do |port|
+          {
+            "{ssh_command}" => ssh_command,
+            "{ssh_options}" => ssh_options,
+            "{port}"        => port,
+            "{ssh_key}"     => key_cmd,
+            "{username}"    => username,
+            "{host}"        => host,
+          }.reduce(format) { |cmd,pair| cmd.sub(pair[0],pair[1].to_s) }
+        end
+      end.join(" || ")
+      # puts joined_cmd
+      "echo Attaching to #{Shellwords.escape(name)} \\[#{@instance.instance_id}\\] && #{joined_cmd}"
+    end
+
+    def header
+      <<-EOS
+      #{name} (#{instance_id}) [#{state[:name]}]
+      launched at #{launch_time.to_time}
+      EOS
+    end
+
+    def image
+      @image ||= provider.find_image_by_id(@instance.image_id)
+    end
+
+    def vpc
+      @vpc ||= provider.find_vpc_by_id(@instance.vpc_id)
+    end
+
+    def header
+      <<-EOS
+      #{name} (#{@instance.instance_id}) [#{@instance.state[:name]}]
+      launched at #{@instance.launch_time.to_time}
+      EOS
+    end
+
+    def label
+      " - #{name} [#{launched_at}]"
+    end
+
+    def info
+      {
+        instance: @instance,
+        image: provider.image_loaded? ? image : "ami info not loaded yet",
+        security_groups: provider.security_group_loaded? ? security_groups : "sg info not loaded yet",
+      }
+    end
+
+    def connectable?
+      ![32, 48, 80].include?(@instance.state[:code])
+    end
+
+    def name
+      return @name if @name
+      begin
+        tag = @instance.tags.find{|t| t.key == "Name"}
+        @name = tag ? tag.value : "noname"
+      rescue
+        @name = "terminated"
+      end
+    end
+
+    def security_groups
+      @security_groups ||= @instance.security_groups.map{|sg| provider.find_security_group_by_id(sg.group_id)}
+    end
+
+    def search_key
+      name.downcase
+    end
+
+    private
+
+    def launched_at
+      diff = Time.now - @instance.launch_time
+      {
+        "year" => 31557600,
+        "month" => 2592000,
+        "day" => 86400,
+        "hour" => 3600,
+        "minute" => 60,
+        "second" => 1
+      }.each do |unit,v|
+        if diff >= v
+          value = (diff/v).to_i
+          return "#{value} #{unit}#{value > 1 ? "s" : ""}"
+        end
+      end
+      "now"
+    end
+
+    def provider
+      EC2Provider
+    end
+  end
+end

data/lib/eclair/providers/ec2/ec2_provider.rb

@@ -0,0 +1,118 @@
+# frozen_string_literal: true
+require "aws-sdk-ec2"
+require "eclair/provider"
+require "eclair/providers/ec2/ec2_item"
+require "eclair/providers/ec2/ec2_group_item"
+
+module Eclair
+  module EC2Provider
+    extend Provider
+    extend self
+
+    def group_class
+      EC2GroupItem
+    end
+
+    def item_class
+      EC2Item
+    end
+
+    def prepare keyword
+      Thread.abort_on_exception = true
+
+      @instances ||= fetch_instances keyword
+
+      image_ids = @instances.map(&:image_id).uniq
+      @image_thread = Thread.new do
+        @images = fetch_images(image_ids)
+        @id_to_image = @images.map{|i| [i.image_id, i]}.to_h
+      end
+
+      @sg_thread = Thread.new do
+        @security_groups = fetch_security_groups
+        @id_to_sg = @security_groups.map{|i| [i.group_id, i]}.to_h
+      end
+
+      @vpc_thread = Thread.new do
+        @vpcs = fetch_vpcs
+        @id_to_vpc = @vpcs.map{|i| [i.vpc_id, i]}.to_h
+      end
+
+      @items = @instances.map{|i| EC2Item.new(i)}
+
+      @prepared = true
+    end
+
+    def image_loaded?
+      !@sg_thread.alive?
+    end
+
+    def images
+      @image_thread.join if @image_thread.alive?
+      @images
+    end
+
+    def find_image_by_id id
+      @image_thread.join if @image_thread.alive?
+      @id_to_image[id]
+    end
+
+    def security_group_loaded?
+      !@sg_thread.alive?
+    end
+
+    def security_groups
+      @sg_thread.join if @sg_thread.alive?
+      @security_groups
+    end
+
+    def find_security_group_by_id id
+      @sg_thread.join if @sg_thread.alive?
+      @id_to_sg[id]
+    end
+
+    def vpc_loaded?
+      !@vpc_thread.alive?
+    end
+
+    def vpcs
+      @vpc_thread.join if @vpc_thread.alive?
+      @vpcs
+    end
+
+    def find_vpc_by_id id
+      @vpc_thread.join if @vpc_thread.alive?
+      @id_to_vpc[id]
+    end
+
+    private
+
+    def ec2_client
+      @ec2_client ||= Aws::EC2::Client.new
+    end
+
+    def fetch_instances keyword
+      filter = if keyword.empty? then {} else { filters: [{name: "tag:Name", values: ["*#{keyword}*"]}] } end
+
+      ec2_client.describe_instances(filter).map{ |resp|
+        resp.data.reservations.map do |rsv|
+          rsv.instances
+        end
+      }.flatten
+    end
+
+    def fetch_security_groups
+      ec2_client.describe_security_groups.map{ |resp|
+        resp.security_groups
+      }.flatten
+    end
+
+    def fetch_images image_ids
+      ec2_client.describe_images(image_ids: image_ids).images.flatten
+    end
+
+    def fetch_vpcs
+      ec2_client.describe_vpcs.map{|resp| resp.vpcs}.flatten
+    end
+  end
+end

data/lib/eclair/providers/gce.rb

@@ -0,0 +1,2 @@
+# frozen_string_literal: true
+require "eclair/providers/gce/gce_provider"

data/lib/eclair/providers/gce/gce_group_item.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+require "eclair/group_item"
+
+module Eclair
+  class GCEGroupItem < GroupItem
+    def header
+      all = @items.count
+
+      <<-EOS
+      Group #{label}
+      #{all} instances Total
+      EOS
+    end
+  end
+end

data/lib/eclair/providers/gce/gce_item.rb

@@ -0,0 +1,117 @@
+# frozen_string_literal: true
+require 'eclair/item'
+require 'eclair/providers/gce/gce_provider'
+require 'time'
+
+module Eclair
+  class GCEItem < Item
+    attr_reader :instance
+
+    def initialize instance
+      super()
+      @instance = instance
+    end
+
+    def id
+      @instance["id"]
+    end
+
+    def color
+      if @selected
+        [Curses::COLOR_YELLOW, -1, Curses::A_BOLD]
+      elsif !connectable?
+        [Curses::COLOR_BLACK, -1, Curses::A_BOLD]
+      else
+        [Curses::COLOR_WHITE, -1]
+      end
+    end
+
+    def header
+      <<-EOS
+      #{name} (#{public_ip_address} #{private_ip_address})
+      launched at #{launch_time.to_time}
+      #{description}
+      EOS
+    end
+
+    def label
+      " - #{name} [#{launched_at}]"
+    end
+
+    def description
+      @instance["description"]
+    end
+
+    def name
+      @instance["name"]
+    end
+
+    def public_ip_address
+      @instance.dig("networkInterfaces", 0, "accessConfigs", 0, "natIP")
+    end
+
+    def private_ip_address
+      @instance.dig("networkInterfaces", 0, "networkIP")
+    end
+
+    def command
+      hosts = [private_ip_address, public_ip_address].compact
+      ports = config.ssh_ports
+      ssh_options = config.ssh_options
+      ssh_command = config.ssh_command
+      username = "ubuntu"
+      format = config.exec_format
+
+      joined_cmd = hosts.map do |host|
+        ports.map do |port|
+          {
+            "{ssh_command}" => ssh_command,
+            "{ssh_options}" => ssh_options,
+            "{port}"        => port,
+            "{username}"    => username,
+            "{host}"        => host,
+            "{ssh_key}"     => ""
+          }.reduce(format) { |cmd,pair| cmd.sub(pair[0],pair[1].to_s) }
+        end
+      end.join(" || ")
+      # puts joined_cmd
+      "echo Attaching to #{name} \\[#{name}\\] && #{joined_cmd}"
+    end
+
+    def connectable?
+      status == "RUNNING"
+    end
+
+    def search_key
+      name.downcase
+    end
+
+    private
+
+    def status
+      @instance["status"]
+    end
+
+    def launch_time
+      Time.parse(@instance["creationTimestamp"]).localtime
+    end
+
+    def launched_at
+      diff = Time.now - launch_time
+      {
+        "year" => 31557600,
+        "month" => 2592000,
+        "day" => 86400,
+        "hour" => 3600,
+        "minute" => 60,
+        "second" => 1
+      }.each do |unit,v|
+        if diff >= v
+          value = (diff/v).to_i
+          return "#{value} #{unit}#{value > 1 ? "s" : ""}"
+        end
+      end
+      "now"
+    end
+  end
+end