kanmon 0.3.0 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3e7a715138c4ecc5d5d10a150bfca964d497e937839f6b66a9d05e514b90053b
-  data.tar.gz: 5ccf83c54ea6098279fe7029a7f09f46e4da9f214669ed18168e272dd77bfa87
+  metadata.gz: 5b4d103ec610dd12ece7cb2441693380c913b50720259188c6bfb82d71bcea41
+  data.tar.gz: ee97f396d6053055d5393e4724f8a38261349ff4720ee3c021553a003cefd56e
 SHA512:
-  metadata.gz: 6d594dcb84cd1bf89431bcf8f78f4e5e24203adb0bb8b64ee274df6afab7f910a193660ae59630890110966eea4209a6308b26dc25c4dcaeb17612aeebff767c
-  data.tar.gz: 386aacf6e68e101384882f23ae909719ec4717b52b254d4e6a285123a71fccebe7bf2e58e0531181211f2bf8e7099acc729c9f58bcb192f12aeaad898656add9
+  metadata.gz: 39eca19f86316209ac6774dc589bc544f79954b1efeb795723dfdff19322102b1409f6df93e1d24784754b1c9a6f8848ecf1b5fea19c1ea2a98ae57e1b2cbdfd
+  data.tar.gz: cab9be8e3106cb3248226495f1bb551507575ca2e8d9cc60b93c022b1d7def4ea23c87d314ea2c1d5ff72bbdc9b2c18292357eb7b8ba2b95e90df80d8fa8b530

data/lib/kanmon/cli.rb

@@ -1,58 +1,35 @@
 require "thor"
+require "shellwords"
 
-require "kanmon/myip"
+require "kanmon/securitygroup"
 
 module Kanmon
   class CLI < Thor
-    class_option :file, aliases: "f", type: :string, default: "kanmon.yml", banner: "FILE", desc: "Load configure from FILE"
+    class_option :kanmon_config, aliases: "f", type: :string, default: "kanmon.yml", banner: "FILE", desc: "Load configure from FILE"
 
     desc "open", "Commands about add rules to SecurityGroup"
     def open
-      myip = Kanmon::MyIP.get
-      rule = {
-        direction: "ingress",
-        port_range_min: 22,
-        port_range_max: 22,
-        ethertype: "IPv4",
-        protocol: "tcp",
-        remote_ip_prefix: "#{myip}/32",
-        security_group_id: @config["security_group"],
-      }
-
-      Yao::SecurityGroupRule.create(rule)
+      @sg.open
       puts "Success!!"
     end
 
     desc "close", "Commands about delete rules from SecurityGroup"
     def close
-      myip = Kanmon::MyIP.get
-      tenant_id = Yao.current_tenant_id
-
-      rule = {
-        direction: "ingress",
-        port_range_min: 22,
-        port_range_max: 22,
-        ethertype: "IPv4",
-        protocol: "tcp",
-        security_group_id: @config["security_group"],
-        tenant_id: tenant_id,
-        remote_ip_prefix: "#{myip}/32"
-      }
-
-      result = Yao::SecurityGroupRule.list(rule)
+      @sg.close
+      puts "Success!!"
+    end
 
-      if result.empty?
-        puts "Not found"
-        exit(1)
-      end
+    desc "ssh HOSTNAME", "Commands about exec ssh"
+    def ssh(*args)
+      invoke CLI, [:exec], args.unshift("ssh")
+    end
 
-      result.each do |rule|
-        id = rule.id
-        puts "Delete #{id}"
-        Yao::SecurityGroupRule.destroy(id)
+    desc "exec COMMAND", "Commands about open, exec command, close"
+    def exec(*args)
+      @sg.open do
+        command = Shellwords.join(args)
+        Process.wait spawn(command)
       end
-
-      puts "Success!!"
     end
 
     desc "version", "Commands about show version"
@@ -64,7 +41,8 @@ module Kanmon
       def invoke_command(command, *args)
         unless %w(help version).include?(command.name)
           Kanmon.init_yao
-          @config = Kanmon.load_config(options[:file])
+          @config = Kanmon.load_config(options[:kanmon_config])
+          @sg = SecurityGroup.new(@config["security_group"])
         end
 
         super

data/lib/kanmon/securitygroup.rb

@@ -0,0 +1,58 @@
+require "yao"
+
+require "kanmon/myip"
+
+module Kanmon
+  class SecurityGroup
+    def initialize(id, ip = nil)
+      @id = id
+      @ip = ip || Kanmon::MyIP.get
+      @tenant_id = Yao.current_tenant_id
+    end
+
+    def open
+      result = Yao::SecurityGroupRule.create(rule)
+      puts "Added Rule: #{result.id}"
+
+      if block_given?
+        begin
+          yield
+        ensure
+          delete_rules([result])
+        end
+      end
+    end
+
+    def close
+      result = Yao::SecurityGroupRule.list(rule)
+
+      if result.empty?
+        puts "Rule not found"
+      else
+        delete_rules(result)
+      end
+    end
+
+    private
+    def rule
+      {
+        direction: "ingress",
+        port_range_min: 22,
+        port_range_max: 22,
+        ethertype: "IPv4",
+        protocol: "tcp",
+        security_group_id: @id,
+        tenant_id: @tenant_id,
+        remote_ip_prefix: "#{@ip}/32"
+      }
+    end
+
+    def delete_rules(rules)
+      rules.each do |rule|
+        id = rule.id
+        puts "Delete Rule: #{id}"
+        Yao::SecurityGroupRule.destroy(id)
+      end
+    end
+  end
+end

data/lib/kanmon/version.rb

@@ -1,3 +1,3 @@
 module Kanmon
-  VERSION = "0.3.0"
+  VERSION = "0.4.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: kanmon
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors:
 - Yuki Koya
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-04-23 00:00:00.000000000 Z
+date: 2018-04-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: yao
@@ -102,6 +102,7 @@ files:
 - lib/kanmon.rb
 - lib/kanmon/cli.rb
 - lib/kanmon/myip.rb
+- lib/kanmon/securitygroup.rb
 - lib/kanmon/version.rb
 homepage: https://github.com/buty4649/kanmon/
 licenses: []