kanmon 0.5.0 → 0.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f377c13e2d1a6a5584c1248fb3921b50b36fee307b7c7a23e1a9d65cdb489329
-  data.tar.gz: 4a15815c745f11afee3e1e07c02ed557cd3fbf51f556077f27c6b2696d919068
+  metadata.gz: b86b5720d0e9a1091f268c95417798f229b8cb0e3a0dbc56578e1ac0dcd88c04
+  data.tar.gz: 0aeca4174abd12f9e5a916856c8d596f43ae51c85b769b3fd9f11dc9405b42bf
 SHA512:
-  metadata.gz: e65cc39e7276098b3acd339724094517f1fe9183188f1c3cbf39d7d85f326a6c8d475018f4f001078e7770f7899ef6444dd2369a7031dbc11030def5fc6859a9
-  data.tar.gz: c980de361f95c2b2273fb5d0d996cf8fd203b006107f92d4a136397035d756c335a7a4698292138c5292e78a8d1e408775044c908c6146f467ed8f65be0c70e5
+  metadata.gz: 24c2c7e15d5b0bdcbf321478e4fdaddd71efc26bcd2d0d73e92558e1e645e1cfc5d2e5ff3f28404ac3b1a1ed2d2bc7290a06ae7ef1991e633a5b97c9988e8bb3
+  data.tar.gz: 70e05239cfc92015d3b6c5ed8ee151ec018f8daf6aa0dba9f0da47f714a5d1de0497f7831ce43faff3a5179a83f2d2248e6a8b6b8f741149c43dbff938b4171b

data/README.md

@@ -10,6 +10,23 @@ $ cat kanmon.yml
 security_group: 11122233-4444-5555-6666-777788889999
 ```
 
+または、SSHでアクセスしたいサーバーのUUIDを記述することも可能です。サーバーのUUIDを記述した場合、新規にSecurityGroupを作成し、指定したサーバーに追加します。
+
+```yaml
+➤ cat kanmon.yml
+server: 11122233-4444-5555-6666-777788889999
+```
+
+もし、kanmon.yaml で複数のターゲットを管理したい場合、次のように書くこともできます。
+
+```yaml
+➤ cat kanmon.yml
+targetA:
+  security_group: 11122233-4444-5555-6666-777788889999
+targetB:
+  server: 33344444-5555-6666-7777-888800000000
+```
+
 環境変数を設定します。
 
 ```
@@ -28,8 +45,22 @@ $ export OS_PROJECT_DOMAIN_NAME=default
 $ kanmon open
 ```
 
+kanmon.yaml に複数のターゲットを記述した場合、下記のようにします。
+
+```
+$ kanmon open --target targetA
+$ kanmon open --target targetB
+```
+
 追加したSecurity Groupのルールを削除します。
 
 ```
 $ kanmon close
 ```
+
+複数のターゲットがある場合、下記のようになります。
+
+```
+$ kanmon close --target targetA
+$ kanmon close --target targetB
+```

data/lib/kanmon/cli.rb

@@ -2,14 +2,28 @@ require "thor"
 require "shellwords"
 
 require "kanmon/securitygroup"
+require "kanmon/server"
+
+module Yao::Resources
+  class Server < Yao::Resources::Base
+    def self.add_security_group(server_id, security_group_name)
+      action(server_id, {"addSecurityGroup": {"name": security_group_name}})
+    end
+
+    def self.remove_security_group(server_id, security_group_name)
+      action(server_id, {"removeSecurityGroup": {"name": security_group_name}})
+    end
+  end
+end
 
 module Kanmon
   class CLI < Thor
     class_option :kanmon_config, aliases: "f", type: :string, default: "kanmon.yml", banner: "FILE", desc: "Load configure from FILE"
+    class_option :target, aliases: "t", type: :string, default: nil, banner: "TARGET", desc: "If more than one Security Group is in the setting, select target"
 
     desc "open", "Commands about add rules to SecurityGroup"
     def open
-      @sg.open
+      @kanmon.open
       puts "Success!!"
     rescue Yao::Conflict => e
       puts "Is not it already opened?" if e.message.include?("Security group rule already exists.")
@@ -18,7 +32,7 @@ module Kanmon
 
     desc "close", "Commands about delete rules from SecurityGroup"
     def close
-      @sg.close
+      @kanmon.close
       puts "Success!!"
     end
 
@@ -29,7 +43,7 @@ module Kanmon
 
     desc "exec COMMAND", "Commands about open, exec command, close"
     def exec(*args)
-      @sg.open do
+      @kanmon.open do
         command = Shellwords.join(args)
         Process.wait spawn(command)
       end
@@ -45,7 +59,14 @@ module Kanmon
         unless %w(help version).include?(command.name)
           Kanmon.init_yao
           @config = Kanmon.load_config(options[:kanmon_config])
-          @sg = SecurityGroup.new(@config["security_group"])
+          @config = @config[options[:target]] if options[:target]
+
+          if @config.key?("security_group")
+            @kanmon = SecurityGroup.new(@config["security_group"])
+          end
+          if @config.key?('server')
+            @kanmon = Server.new(@config["server"])
+          end
         end
 
         super

data/lib/kanmon/server.rb

@@ -0,0 +1,80 @@
+require "yao"
+
+require "kanmon/myip"
+
+module Kanmon
+  class Server
+    def initialize(id, ip = nil)
+      @id = id
+      @ip = ip || Kanmon::MyIP.get
+      @tenant_id = Yao.current_tenant_id
+      @server = Yao::Server.get(id)
+    end
+
+    def create_sg
+      puts "Create security group allow to access server '#{@server.name}'."
+      param = {name: sg_name, description: "create by kanmon and #{ENV['OS_USERNAME']}"}
+      @sg = Yao::SecurityGroup.create(param)
+      result = Yao::SecurityGroupRule.create(rule)
+      puts "Create rule to #{@sg.name} allow ssh from #{@ip}/32: #{result.id}"
+    end
+
+    def delete_sg
+      puts "Delete security group #{sg_name}"
+      Yao::SecurityGroup.destroy(@sg.id)
+    end
+
+    def add_sg
+      puts "Add security group  #{sg_name} to server #{@server.name}"
+      Yao::Server.add_security_group(@id, sg_name)
+    end
+
+    def remove_sg
+      puts "Remove security group #{sg_name} from server '#{@server.name}'."
+      Yao::Server.remove_security_group(@id, sg_name)
+    end
+
+    def open
+      create_sg
+      add_sg
+
+      if block_given?
+        begin
+          yield
+        ensure
+          remove_sg
+          delete_sg
+        end
+      end
+    end
+
+    def close
+      begin
+        @sg = Yao::SecurityGroup.get(sg_name)
+        remove_sg
+        delete_sg
+
+      rescue => e
+        p e
+      end
+    end
+
+    private
+    def rule
+      {
+          direction: "ingress",
+          port_range_min: 22,
+          port_range_max: 22,
+          ethertype: "IPv4",
+          protocol: "tcp",
+          security_group_id: @sg.id,
+          tenant_id: @tenant_id,
+          remote_ip_prefix: "#{@ip}/32"
+      }
+    end
+
+    def sg_name
+      "kanmon-server:#{@server.id}-user:#{ENV['OS_USERNAME']}"
+    end
+  end
+end

data/lib/kanmon/version.rb

@@ -1,3 +1,3 @@
 module Kanmon
-  VERSION = "0.5.0"
+  VERSION = "0.6.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: kanmon
 version: !ruby/object:Gem::Version
-  version: 0.5.0
+  version: 0.6.0
 platform: ruby
 authors:
 - Yuki Koya
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-06-22 00:00:00.000000000 Z
+date: 2018-08-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: yao
@@ -103,6 +103,7 @@ files:
 - lib/kanmon/cli.rb
 - lib/kanmon/myip.rb
 - lib/kanmon/securitygroup.rb
+- lib/kanmon/server.rb
 - lib/kanmon/version.rb
 homepage: https://github.com/buty4649/kanmon/
 licenses: []