devbox_launcher 0.3.2 → 0.3.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 25588592982c23d00fa42fbd84265d18c9f113ee46bcdb64e9c458115a0b65dc
-  data.tar.gz: 00560c6344d2f39489f4f6acdb9afc442a90b3d2dc98b98d5ca47fd0085980e2
+  metadata.gz: 7ed54fc2716507cd48b6e65acb08d38f1399c7d2a626a898ea0bdbeb70317f9d
+  data.tar.gz: 13b0d4ba4c3b9ae2515040bddff831b746de65c1296d57be20df82bc7f99f722
 SHA512:
-  metadata.gz: db2e46803d38b20624480c3dd8dfd9eadcd7fe315841ff6e57fd8b4a70e3752784d38917f8b7d00d41cca8feaadb3bcf44817e3daa7c7885aebdac007ff5fb54
-  data.tar.gz: 7f1a8c4f5eefd21c6cf32eb0d041b36ac8f495670e9440dfe0079445205c8b02d53db524a182f3c3d7764fe475613552a3e55e064ee52b354103e28045921b1a
+  metadata.gz: 0c9720ff3bc730ad33b4d225a24796eca616615534d23f5d01c57be8c8291f0b37a0fc2d13092b2f84799c66ea513d6cc3b00bef3fd3e3559fe8d6ea107596e2
+  data.tar.gz: a040326a3fe949841970a3d5fd26535dc2cbf06bec09bd30098f63744ff87ac7fd1184b3a79d4848571681b7cad7c91c6ebe89753781f6126c8b5372619790de

data/CHANGELOG.md

@@ -4,6 +4,10 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.3.3]
+### Fixed
+- Support launching multiple boxes at the same time
+
 ## [0.3.2]
 ### Fixed
 - Fix: add missing file

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    devbox_launcher (0.3.2)
+    devbox_launcher (0.3.3)
       activesupport (~> 6.0)
       bcrypt_pbkdf (~> 1.0)
       ed25519 (~> 1.2)

data/lib/devbox_launcher.rb

@@ -17,3 +17,5 @@ end
 require "devbox_launcher/cli"
 require "devbox_launcher/watchman"
 require "devbox_launcher/models/description"
+require "devbox_launcher/models/mutagen"
+require "devbox_launcher/models/box"

data/lib/devbox_launcher/box.rb

@@ -0,0 +1,205 @@
+module DevboxLauncher
+  class Box
+
+    WAIT_BOOT_RESCUED_EXCEPTIONS = [
+      Net::SSH::ConnectionTimeout,
+      Net::SSH::Disconnect,
+      Errno::ECONNRESET,
+      Errno::ETIMEDOUT,
+    ]
+    WAIT_BOOT_IN_SECONDS = 10.freeze
+    DEFAULT_IDENTIFY_FILE_PATH = "~/.ssh/google_compute_engine".freeze
+    SSH_CONFIG_PATH = File.expand_path("~/.ssh/config").freeze
+    CONFIG_PATH = File.expand_path("~/.devbox_launcher.yml").freeze
+    CONFIG = YAML.load_file(CONFIG_PATH).freeze
+
+    attr_reader :account
+
+    def initialize(account)
+      @account = account
+    end
+
+    def start
+      start_stdout, start_stderr, start_status =
+        Open3.capture3(start_cmd)
+
+      set_ssh_config!(hostname, {
+        username: username,
+        ip: description.ip,
+      })
+
+      wait_boot
+
+      reset_mutagen_session(
+        mutagen_config: config[:mutagen],
+        hostname: hostname,
+      )
+    end
+
+    def start_cmd
+      args = {
+        project: config[:project],
+        account: account,
+      }.map do |(key, val)|
+        ["--#{key}", val].join("=")
+      end.join(" ")
+
+      [
+        "gcloud",
+        "compute",
+        "instances",
+        "start",
+        name,
+        args
+      ].join(" ")
+    end
+
+    def wait_boot(tries: 1)
+      Net::SSH.start(hostname, username, timeout: WAIT_BOOT_IN_SECONDS) do |ssh|
+        puts "[#{ssh.exec!('date').chomp}] Machine booted"
+      end
+    rescue *WAIT_BOOT_RESCUED_EXCEPTIONS
+      puts "Not booted. Waiting #{WAIT_BOOT_IN_SECONDS} seconds before trying again..."
+
+      sleep WAIT_BOOT_IN_SECONDS
+
+      description = describe(name)
+      if !description.running?
+        puts "Detected that the machine is not running " \
+          "(status is #{description.status}). Booting it..."
+        start_box name, username
+      end
+
+      wait_boot name, username, tries: tries+1
+    end
+
+    def description(reload: false)
+      return @description if !reload && @description
+
+      puts "Fetching box's description..."
+
+      describe_command = %Q(gcloud compute instances describe #{name})
+      describe_stdout, describe_stderr, describe_status =
+        Open3.capture3(describe_command)
+
+      if !describe_status.success?
+        msg = "Problem fetching the description of #{name}. "
+        msg += "Please ensure you can call `#{describe_command}`.\n"
+        msg += "Error:\n"
+        msg += describe_stderr
+        fail msg
+      end
+
+      @description = Description.new(describe_stdout)
+    end
+
+    def set_ssh_config!(hostname, username:, ip:)
+      FileUtils.touch(SSH_CONFIG_PATH)
+      config = ConfigFile.new
+      args = {
+        "HostName" => ip,
+        "User" => username,
+        "IdentityFile" => DEFAULT_IDENTIFY_FILE_PATH,
+      }
+      args.each do |key, value|
+        config.set(hostname, key, value)
+      end
+      config.save
+    end
+
+    def reset_mutagen_session
+      mutagen_config = config[:mutagen]
+      return if mutagen_config.nil?
+
+      alpha_dir = mutagen_config[:alpha]
+      beta_dir = mutagen_config[:beta]
+
+      return if alpha_dir.nil? || beta_dir.nil?
+
+      terminate_mutagen_session
+
+      create_mutagen_session(
+        alpha_dir: alpha_dir,
+        beta_dir: beta_dir,
+        hostname: hostname,
+        username: username,
+      )
+
+      if OS.linux?
+        watch_alpha(alpha_dir: alpha_dir, hostname: hostname)
+      end
+    end
+
+    def terminate_mutagen_session(username)
+      puts "Terminating mutagen session..."
+      terminate_mutagen_command =
+        %Q(mutagen terminate --label-selector=#{username})
+      terminate_mutagen_stdout,
+        terminate_mutagen_stderr,
+        terminate_mutagen_status =
+        Open3.capture3(terminate_mutagen_command)
+
+      if not terminate_mutagen_status.success?
+        # mutagen prints to stdout and stderr
+        msg = "Failed to terminate mutagen sessions: " \
+          "#{terminate_mutagen_stdout} -" \
+          "#{terminate_mutagen_stderr}"
+        fail msg
+      end
+    end
+
+    def create_mutagen_session(alpha_dir:, beta_dir:, hostname:, username:)
+      puts "Create mutagen session syncing local #{alpha_dir} " \
+        "with #{hostname} #{beta_dir}"
+
+      create_mutagen_command = [
+        "mutagen sync create",
+        alpha_dir,
+        "#{hostname}:#{beta_dir}",
+        "--label=#{username}",
+      ]
+      create_mutagen_command << "--watch-mode-alpha=no-watch" if OS.linux?
+
+      create_mutagen_stdout,
+        create_mutagen_stderr,
+        create_mutagen_status =
+        Open3.capture3(create_mutagen_command.join(" "))
+
+      if not create_mutagen_status.success?
+        # mutagen prints to stdout and stderr
+        msg = "Failed to create mutagen sessions: " \
+          "#{create_mutagen_stdout} -" \
+          "#{create_mutagen_stderr}"
+        fail msg
+      end
+    end
+
+    def watch_alpha(alpha_dir:, hostname:)
+      watchman = Watchman.new(dir: alpha_dir)
+      watchman.trigger("mutagen sync flush --label-selector=#{hostname}")
+    end
+
+    def name
+      @name ||= config[:box]
+    end
+
+    def hostname
+      [name, username, "devbox"].join("-")
+    end
+
+    def username
+      @username ||= account.gsub(/\W/, "_")
+    end
+
+    def config
+      return @config if @config
+
+      if not CONFIG.has_key?(account)
+        fail "No config in #{CONFIG_PATH} found for #{account}"
+      end
+
+      @config = CONFIG[account].with_indifferent_access
+    end
+
+  end
+end

data/lib/devbox_launcher/cli.rb

@@ -6,186 +6,12 @@ module DevboxLauncher
     SSH_CONFIG_PATH = File.expand_path("~/.ssh/config").freeze
     CONFIG_PATH = File.expand_path("~/.devbox_launcher.yml").freeze
     CONFIG = YAML.load_file(CONFIG_PATH).freeze
-    LABEL = "devbox".freeze
 
     desc "start configured box for account", "Start a devbox by account"
     option :mosh, type: :boolean, desc: "Mosh in"
 
     def start(account)
-      if not CONFIG.has_key?(account)
-        fail "No config in #{CONFIG_PATH} found for #{account}"
-      end
-
-      config = CONFIG[account].with_indifferent_access
-      name = config[:box]
-
-      username = account.gsub(/\W/, "_")
-
-      puts "Starting #{name}..."
-
-      set_account_command = %Q(gcloud config set account #{account})
-      set_account_stdout, set_account_stderr, set_account_status =
-        Open3.capture3(set_account_command)
-
-      set_project_command = %Q(gcloud config set project #{config[:project]})
-      set_project_stdout, set_project_stderr, set_project_status =
-        Open3.capture3(set_project_command)
-
-      start_box name, username
-
-      wait_boot(name, username)
-
-      hostname = hostname_for(name)
-
-      reset_mutagen_session(
-        mutagen_config: config[:mutagen],
-        hostname: hostname,
-      )
-
-      if options[:mosh]
-        mosh_command = %Q(mosh #{hostname})
-        system(mosh_command)
-      end
-    end
-
-    no_commands do
-      def wait_boot(name, username, tries: 1)
-        hostname = hostname_for(name)
-
-        Net::SSH.start(hostname, username, timeout: WAIT_BOOT_IN_SECONDS) do |ssh|
-          puts "[#{ssh.exec!('date').chomp}] Machine booted"
-        end
-      rescue Net::SSH::ConnectionTimeout, Net::SSH::Disconnect, Errno::ECONNRESET
-        puts "Not booted. Waiting #{WAIT_BOOT_IN_SECONDS} seconds before trying again..."
-
-        sleep WAIT_BOOT_IN_SECONDS
-
-        description = describe(name)
-        if !description.running?
-          puts "Detected that the machine is not running " \
-            "(status is #{description.status}). Booting it..."
-          start_box name, username
-        end
-
-        wait_boot name, username, tries: tries+1
-      end
-
-      def set_ssh_config!(hostname, username:, ip:)
-        FileUtils.touch(SSH_CONFIG_PATH)
-        config = ConfigFile.new
-        args = {
-          "HostName" => ip,
-          "User" => username,
-          "IdentityFile" => DEFAULT_IDENTIFY_FILE_PATH,
-        }
-        args.each do |key, value|
-          config.set(hostname, key, value)
-        end
-        config.save
-      end
-
-      def reset_mutagen_session(mutagen_config:, hostname:)
-        return if mutagen_config.nil?
-        alpha_dir = mutagen_config[:alpha]
-        beta_dir = mutagen_config[:beta]
-
-        return if alpha_dir.nil? || beta_dir.nil?
-
-        terminate_mutagen_session
-        create_mutagen_session(
-          alpha_dir: alpha_dir,
-          beta_dir: beta_dir,
-          hostname: hostname,
-        )
-
-        if OS.linux?
-          watch_alpha(alpha_dir: alpha_dir)
-        end
-      end
-
-      def terminate_mutagen_session
-        puts "Terminating mutagen session..."
-        terminate_mutagen_command =
-          %Q(mutagen terminate --label-selector=#{LABEL})
-        terminate_mutagen_stdout,
-          terminate_mutagen_stderr,
-          terminate_mutagen_status =
-          Open3.capture3(terminate_mutagen_command)
-
-        if not terminate_mutagen_status.success?
-          # mutagen prints to stdout and stderr
-          msg = "Failed to terminate mutagen sessions: " \
-            "#{terminate_mutagen_stdout} -" \
-            "#{terminate_mutagen_stderr}"
-          fail msg
-        end
-      end
-
-      def create_mutagen_session(alpha_dir:, beta_dir:, hostname:)
-        puts "Create mutagen session syncing local #{alpha_dir} " \
-          "with #{hostname} #{beta_dir}"
-
-        create_mutagen_command = [
-          "mutagen sync create",
-          alpha_dir,
-          "#{hostname}:#{beta_dir}",
-          "--label=#{LABEL}",
-        ]
-        create_mutagen_command << "--watch-mode-alpha=no-watch" if OS.linux?
-
-        create_mutagen_stdout,
-          create_mutagen_stderr,
-          create_mutagen_status =
-          Open3.capture3(create_mutagen_command.join(" "))
-
-        if not create_mutagen_status.success?
-          # mutagen prints to stdout and stderr
-          msg = "Failed to create mutagen sessions: " \
-            "#{create_mutagen_stdout} -" \
-            "#{create_mutagen_stderr}"
-          fail msg
-        end
-      end
-
-      def watch_alpha(alpha_dir:)
-        watchman = Watchman.new(dir: alpha_dir)
-        watchman.trigger("mutagen sync flush --label-selector=#{LABEL}")
-      end
-
-      def start_box(name, username)
-        start_command = %Q(gcloud compute instances start #{name})
-        start_stdout, start_stderr, start_status = Open3.capture3(start_command)
-
-        desc = describe(name)
-        ip = desc.ip
-
-        set_ssh_config!(hostname_for(name), {
-          username: username,
-          ip: ip,
-        })
-      end
-
-      def describe(name)
-        puts "Fetching box's description..."
-
-        describe_command = %Q(gcloud compute instances describe #{name})
-        describe_stdout, describe_stderr, describe_status =
-          Open3.capture3(describe_command)
-
-        if !describe_status.success?
-          msg = "Problem fetching the description of #{name}. "
-          msg += "Please ensure you can call `#{describe_command}`.\n"
-          msg += "Error:\n"
-          msg += describe_stderr
-          fail msg
-        end
-
-        Description.new(describe_stdout)
-      end
-
-      def hostname_for(name)
-        [name, "devbox"].join("-")
-      end
+      Box.new(account, options).start
     end
 
   end

data/lib/devbox_launcher/models/box.rb

@@ -0,0 +1,222 @@
+module DevboxLauncher
+  class Box
+
+    WAIT_BOOT_RESCUED_EXCEPTIONS = [
+      Net::SSH::ConnectionTimeout,
+      Net::SSH::Disconnect,
+      Errno::ECONNRESET,
+      Errno::ETIMEDOUT,
+    ]
+    WAIT_BOOT_IN_SECONDS = 10.freeze
+    MAX_BOOT_RETRIES = 10
+    DEFAULT_IDENTIFY_FILE_PATH = "~/.ssh/google_compute_engine".freeze
+    SSH_CONFIG_PATH = File.expand_path("~/.ssh/config").freeze
+    CONFIG_PATH = File.expand_path("~/.devbox_launcher.yml").freeze
+    CONFIG = YAML.load_file(CONFIG_PATH).freeze
+
+    attr_reader :account, :options
+
+    def initialize(account, options)
+      @account = account
+      @options = options
+    end
+
+    def start
+      start_stdout, start_stderr, start_status =
+        Open3.capture3(start_cmd)
+
+      set_ssh_config!
+
+      wait_boot
+
+      reset_mutagen_session
+
+      connect_mosh
+    end
+
+    def start_cmd
+      args = {
+        project: config[:project],
+        account: account,
+      }.map do |(key, val)|
+        ["--#{key}", val].join("=")
+      end.join(" ")
+
+      [
+        "gcloud",
+        "compute",
+        "instances",
+        "start",
+        name,
+        args
+      ].join(" ")
+    end
+
+    def connect_mosh
+      return if options[:mosh].nil?
+
+      mosh_cmd = %Q(mosh #{hostname})
+      system(mosh_cmd)
+    end
+
+    def wait_boot(tries: 1)
+      Net::SSH.start(hostname, username, timeout: WAIT_BOOT_IN_SECONDS) do |ssh|
+        puts "[#{ssh.exec!('date').chomp}] Machine booted"
+      end
+    rescue *WAIT_BOOT_RESCUED_EXCEPTIONS
+      puts "Not booted. Waiting #{WAIT_BOOT_IN_SECONDS} seconds before trying again..."
+
+      sleep WAIT_BOOT_IN_SECONDS
+
+      if !description(reload: true).running?
+        puts "Detected that the machine is not running " \
+          "(status is #{description.status}). Booting it..."
+        start
+      end
+
+      fail if tries >= MAX_BOOT_RETRIES
+
+      wait_boot tries: tries+1
+    end
+
+    def description(reload: false)
+      return @description if !reload && @description
+
+      puts "Fetching box's description..."
+
+      describe_stdout, describe_stderr, describe_status =
+        Open3.capture3(describe_cmd)
+
+      if !describe_status.success?
+        msg = "Problem fetching the description of #{name}. "
+        msg += "Please ensure you can call `#{describe_cmd}`.\n"
+        msg += "Error:\n"
+        msg += describe_stderr
+        fail msg
+      end
+
+      @description = Description.new(describe_stdout)
+    end
+
+    def describe_cmd
+      args = {
+        project: config[:project],
+        account: account,
+      }.map do |(key, val)|
+        ["--#{key}", val].join("=")
+      end.join(" ")
+
+      [
+        "gcloud",
+        "compute",
+        "instances",
+        "describe",
+        name,
+        args
+      ].join(" ")
+    end
+
+    def set_ssh_config!
+      FileUtils.touch(SSH_CONFIG_PATH)
+      config = ConfigFile.new
+      args = {
+        "HostName" => description.ip,
+        "User" => username,
+        "IdentityFile" => DEFAULT_IDENTIFY_FILE_PATH,
+      }
+      args.each do |key, value|
+        config.set(hostname, key, value)
+      end
+      config.save
+    end
+
+    def reset_mutagen_session
+      return if !mutagen_config.configured?
+
+      terminate_mutagen_session
+      create_mutagen_session
+      watch_alpha if OS.linux?
+    end
+
+    def terminate_mutagen_session
+      puts "Terminating mutagen session..."
+      terminate_mutagen_cmd =
+        %Q(mutagen terminate --label-selector=#{label})
+      terminate_mutagen_stdout,
+        terminate_mutagen_stderr,
+        terminate_mutagen_status =
+        Open3.capture3(terminate_mutagen_cmd)
+
+      if not terminate_mutagen_status.success?
+        # mutagen prints to stdout and stderr
+        msg = "Failed to terminate mutagen sessions: " \
+          "#{terminate_mutagen_stdout} -" \
+          "#{terminate_mutagen_stderr}"
+        fail msg
+      end
+    end
+
+    def label
+      "#{username}=#{name}"
+    end
+
+    def create_mutagen_session
+      puts "Create mutagen session syncing local " \
+        "#{mutagen_config.alpha_dir} with " \
+        "#{hostname} #{mutagen_config.beta_dir}"
+
+      create_mutagen_cmd = [
+        "mutagen sync create",
+        mutagen_config.alpha_dir,
+        "#{hostname}:#{mutagen_config.beta_dir}",
+        "--label=#{label}",
+      ]
+      create_mutagen_cmd << "--watch-mode-alpha=no-watch" if OS.linux?
+
+      create_mutagen_stdout,
+        create_mutagen_stderr,
+        create_mutagen_status =
+        Open3.capture3(create_mutagen_cmd.join(" "))
+
+      if not create_mutagen_status.success?
+        # mutagen prints to stdout and stderr
+        msg = "Failed to create mutagen sessions: " \
+          "#{create_mutagen_stdout} -" \
+          "#{create_mutagen_stderr}"
+        fail msg
+      end
+    end
+
+    def watch_alpha
+      watchman = Watchman.new(dir: mutagen_config.alpha_dir)
+      watchman.trigger("mutagen sync flush --label-selector=#{label}")
+    end
+
+    def name
+      @name ||= config[:box]
+    end
+
+    def hostname
+      [name, username, "devbox"].join("-")
+    end
+
+    def username
+      @username ||= account.gsub(/\W/, "_")
+    end
+
+    def config
+      return @config if @config
+
+      if not CONFIG.has_key?(account)
+        fail "No config in #{CONFIG_PATH} found for #{account}"
+      end
+
+      @config = CONFIG[account].with_indifferent_access
+    end
+
+    def mutagen_config
+      @mutagen_config ||= Mutagen.new(config[:mutagen])
+    end
+
+  end
+end

data/lib/devbox_launcher/models/mutagen.rb

@@ -0,0 +1,23 @@
+module DevboxLauncher
+  class Mutagen
+
+    attr_reader :config
+
+    def initialize(config)
+      @config = config
+    end
+
+    def configured?
+      [config, alpha_dir, beta_dir].none?(&:nil?)
+    end
+
+    def alpha_dir
+      config[:alpha]
+    end
+
+    def beta_dir
+      config[:beta]
+    end
+
+  end
+end

data/lib/devbox_launcher/version.rb

@@ -1,3 +1,3 @@
 module DevboxLauncher
-  VERSION = "0.3.2"
+  VERSION = "0.3.3"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: devbox_launcher
 version: !ruby/object:Gem::Version
-  version: 0.3.2
+  version: 0.3.3
 platform: ruby
 authors:
 - Ramon Tayag
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-05-25 00:00:00.000000000 Z
+date: 2020-05-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -189,8 +189,11 @@ files:
 - bin/setup
 - devbox_launcher.gemspec
 - lib/devbox_launcher.rb
+- lib/devbox_launcher/box.rb
 - lib/devbox_launcher/cli.rb
+- lib/devbox_launcher/models/box.rb
 - lib/devbox_launcher/models/description.rb
+- lib/devbox_launcher/models/mutagen.rb
 - lib/devbox_launcher/version.rb
 - lib/devbox_launcher/watchman.rb
 homepage: https://github.com/bloom-solutions/devbox_launcher