magellan-cli 0.1.0

data/lib/magellan/cli/login.rb

@@ -0,0 +1,134 @@
+# -*- coding: utf-8 -*-
+require "magellan/cli"
+
+require 'httpclient'
+require 'json'
+require 'uri'
+require 'nokogiri'
+
+require 'active_support/core_ext/hash/keys'
+
+module Magellan
+  module Cli
+    class Login
+
+      PRODUCTION_HTTP_PORT  =  80
+      PRODUCTION_HTTPS_PORT = 443
+
+      DEFAULT_HTTP_PORT  = (ENV['DEFAULT_HTTP_PORT' ] ||  80).to_i
+      DEFAULT_HTTPS_PORT = (ENV['DEFAULT_HTTPS_PORT'] || 443).to_i
+
+      attr_reader :httpclient
+      attr_reader :base_http_url
+      attr_reader :base_https_url
+      attr_reader :ssl_disabled
+
+      attr_reader :auth_token
+
+      # Magellan::Cli::Loginのコンストラクタです。
+      #
+      # @param [String] base_url_or_host 接続先の基準となるURLあるいはホスト名
+      # @param [Hash] options オプション
+      # @option options [String]  :api_version APIのバージョン。デフォルトは "1.0.0"
+      # @option options [Boolean] :ssl_disabled SSLを無効にするかどうか。
+      # @option options [Integer] :https_port HTTPSで接続する際の接続先のポート番号
+      def initialize(base_url_or_host = nil, options = {})
+        base_url_or_host ||= (ENV["MAGELLAN_SITE"] || "http://localhost:3000/")
+        if base_url_or_host =~ URI.regexp
+          @base_http_url = base_url_or_host.sub(/\/\Z/, '')
+          uri = URI.parse(@base_http_url)
+        else
+          if config_path = search_file(".magellan-cli.yml")
+            config = YAML.load_file_with_erb(config_path)
+            options = config[base_url_or_host.to_s].deep_symbolize_keys.update(options)
+          end
+          uri = URI::Generic.build({scheme: "http", host: base_url_or_host, port: DEFAULT_HTTP_PORT}.update(options))
+          @base_http_url = uri.to_s
+        end
+        @ssl_disabled = options.delete(:ssl_disabled)
+        @ssl_disabled ||= (uri.port == 3000)
+        @base_https_url = @ssl_disabled ? @base_http_url : build_https_url(uri, options[:https_port])
+        # @api_version = options[:api_version] || "1.0.0"
+
+        @httpclient = HTTPClient.new
+        @httpclient.ssl_config.verify_mode = nil # 自己署名の証明書をOKにする
+      end
+
+      def build_https_url(uri, port = nil)
+        uri.scheme = "https"
+        uri.port = port || (uri.port == PRODUCTION_HTTP_PORT ? PRODUCTION_HTTPS_PORT : uri.port + 1)
+        uri.to_s
+      end
+
+      def login_form_url
+        @login_form_url ||= base_https_url + "/users/sign_in.html"
+      end
+      def login_url
+        @login_url ||= base_https_url + "/api/sign_in.json"
+      end
+
+      # magellan-apiサーバに接続してログインの検証と処理を行います。
+      #
+      # @param [Hash] extra オプション
+      # @return [Integer] サーバが返したレスポンスのステータスを返します
+      # @return [String] サーバが返したレスポンスのbodyを返します
+      def login_and_status
+        @auth_token ||= get_auth_token
+        params = {
+          "user" => {"email" => "magellan@groovenauts.jp", "password" => "password"},
+          "authenticity_token" => @auth_token
+        }.to_json
+        res2 = Ssl.retry_on_ssl_error("login"){ @httpclient.post(login_url, params, JSON_HEADER) }
+        return res2.status, res2.body
+      end
+
+      # GSSサーバに接続してログインの検証と処理を行います。
+      #
+      # @param [Hash] extra オプション
+      # @option extra [Integer] :device_type デバイス種別
+      # @option extra [Integer] :device_id デバイス識別子
+      # @return [Boolean] ログインに成功した場合はtrue、失敗した場合はfalse
+      def login
+        status, _ = login_and_status
+        case status
+        when 200...300 then true
+        else false
+        end
+      end
+
+      # GSSサーバに接続してログインの検証と処理を行います。
+      #
+      # @param [Hash] extra オプション
+      # @see #login
+      # @return ログインに成功した場合は自身のオブジェクト返します。失敗した場合はLibgss::Network::Errorがraiseされます。
+      def login!
+        status, body = login_and_status
+        case status
+        when 200...300 then return self
+        else raise LoginError, "status: #{status} #{body}"
+        end
+      end
+
+      def get_auth_token
+        res = Ssl.retry_on_ssl_error("login_form"){ @httpclient.get(login_form_url) }
+        doc = Nokogiri::HTML.parse(res.body, login_form_url, res.body_encoding.to_s)
+        node = doc.xpath('//input[@name="authenticity_token"]').first
+        node.attribute('value').value
+      end
+
+      # @httpclient.inspectの戻り値の文字列が巨大なので、inspectで出力しないようにします。
+      def inspect
+        r = "#<#{self.class.name}:#{self.object_id} "
+        fields = (instance_variables - [:@httpclient]).map{|f| "#{f}=#{instance_variable_get(f).inspect}"}
+        r << fields.join(", ") << ">"
+      end
+
+      def search_file(basename)
+        dirs = [".", "./config", ENV['HOME']].join(",")
+        Dir["{#{dirs}}/#{basename}"].select{|path| File.readable?(path)}.first
+      end
+      private :search_file
+
+    end
+  end
+end

data/lib/magellan/cli/resources/base.rb

@@ -0,0 +1,136 @@
+# -*- coding: utf-8 -*-
+require "magellan/cli/resources"
+
+require 'json'
+require 'yaml'
+require 'active_support/core_ext/string/inflections'
+require 'active_support/core_ext/class/attribute_accessors'
+require 'active_support/core_ext/object/blank'
+
+module Magellan
+  module Cli
+    module Resources
+
+      class NotSelected < StandardError
+      end
+
+      class NotFound < StandardError
+      end
+
+      class Base < ::Magellan::Cli::Http
+
+        no_commands do
+
+          def load_selections
+            File.readable?(".magellan-cli") ? YAML.load_file(".magellan-cli") : {}
+          end
+
+          def load_selection(name)
+            sel = load_selections
+            s = sel[name]
+            raise NotSelected, "No #{name} selected" unless s
+            return s
+          end
+
+          def update_selections(hash = nil)
+            sel = load_selections
+            sel.update(hash) if hash
+            yield(sel) if block_given?
+            open(".magellan-cli", "w") do |f|
+              YAML.dump(sel, f)
+            end
+          end
+
+          def build_query(hash)
+            @@no ||= 0
+            r = {}
+            hash.each do |key, value|
+              @@no += 1
+              r["f[#{key}][#{@@no}][o]"] = "is"
+              r["f[#{key}][#{@@no}][v]"] = value
+            end
+            return r
+          end
+
+          def default_query
+            sel = load_selections
+            q = {}
+            (self.class.resource_dependency || {}).each do |f, res|
+              r = sel[res]
+              raise NotSelected, "No #{f} selected" unless r
+              q[f] = r["id"]
+            end
+            return build_query(q)
+          end
+
+          DEFAULT_SELECTION_FIELDS = %w[id name].freeze
+
+          def update_first_result(name, path, query, fields = DEFAULT_SELECTION_FIELDS)
+            results = get_json(path, query)
+            raise NotFound, "#{name} not found for #{query.inspect}" if results.blank? || results.first.blank?
+            r = results.first
+            obj = fields.each_with_object({}) do |f, d|
+              d[f] = r[f]
+            end
+            update_selections(name => obj)
+            return r
+          end
+
+          def list
+            r = get_json("/admin/#{self.class.resource_name}.json", default_query)
+            $stdout.puts(JSON.pretty_generate(r) << "\nTotal: #{r.length}")
+          end
+
+          def show(id)
+            r = get_json("/admin/#{self.class.resource_name}/#{id}.json")
+            $stdout.puts(JSON.pretty_generate(r))
+          end
+
+          def select(name)
+            q = build_query("name" => name).update(default_query)
+            update_first_result(self.class.resource_name, "/admin/#{self.class.resource_name}.json", q)
+          end
+
+          def deselect
+            update_selections do |s|
+              s.delete(self.class.resource_name)
+            end
+          end
+
+          def self.inherited(klass)
+            base_name = klass.name.split(/::/).last
+            res_name = base_name.underscore
+            klass.module_eval <<-EOM
+              no_commands do
+                cattr_accessor :resource_name
+                cattr_accessor :resource_dependency
+              end
+
+              desc "list", "list #{base_name}"
+              def list
+                super
+              end
+
+              desc "show ID", "show #{base_name} specified by ID"
+              def show(id)
+                super(id)
+              end
+
+              desc "select NAME", "select #{base_name} by NAME"
+              def select(name)
+                super
+              end
+
+              desc "deselect", "deselect #{base_name}"
+              def deselect
+                super
+              end
+            EOM
+          end
+        end
+
+      end
+
+    end
+  end
+end

data/lib/magellan/cli/resources/client_version.rb

@@ -0,0 +1,15 @@
+# -*- coding: utf-8 -*-
+require "magellan/cli/resources"
+
+module Magellan
+  module Cli
+    module Resources
+
+      class ClientVersion < Base
+        self.resource_name = "client_version"
+        self.resource_dependency = {"stage" => "stage"}
+      end
+
+    end
+  end
+end

data/lib/magellan/cli/resources/container_image.rb

@@ -0,0 +1,13 @@
+# -*- coding: utf-8 -*-
+require "magellan/cli/resources"
+
+module Magellan
+  module Cli
+    module Resources
+
+      class ContainerImage < Base
+      end
+
+    end
+  end
+end

data/lib/magellan/cli/resources/container_instance.rb

@@ -0,0 +1,25 @@
+# -*- coding: utf-8 -*-
+require "magellan/cli/resources"
+
+module Magellan
+  module Cli
+    module Resources
+
+      class ContainerInstance < Base
+        self.resource_name = "container_instance"
+        self.resource_dependency = {"host" => "host_instance"}
+
+        desc "start ID", "start container instance specified by ID"
+        def start(id)
+          post_json("/admin/container_instance/#{id}/status_control.json", {})
+        end
+
+        desc "stop ID", "stop container instance specified by ID"
+        def stop(id)
+          delete("/admin/container_instance/#{id}/status_control.json")
+        end
+      end
+
+    end
+  end
+end

data/lib/magellan/cli/resources/function_unit.rb

@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+require "magellan/cli/resources"
+
+module Magellan
+  module Cli
+    module Resources
+
+      class FunctionUnit < Base
+        self.resource_name = "function_unit"
+        self.resource_dependency = {"stage" => "stage-version"}
+
+        desc "create [ATTRIBUTES]", "create FunctionUnit with ATTRIBUTES (filepath or JSON text)"
+        def create(attrs = nil)
+          stage_id = load_selection("stage-version")
+          attrs = JSON.parse(File.readable?(attrs) ? File.read(attrs) : attrs)
+          params = {
+            "function_unit" => {
+              "stage_version_id" => stage_id,
+            }.update(attrs)
+          }
+          post_json("/admin/function_unit/new.js", params)
+        end
+      end
+
+    end
+  end
+end

data/lib/magellan/cli/resources/host_instance.rb

@@ -0,0 +1,56 @@
+# -*- coding: utf-8 -*-
+require "magellan/cli/resources"
+
+module Magellan
+  module Cli
+    module Resources
+
+      class HostInstance < Base
+        self.resource_name = "host_instance"
+        self.resource_dependency = {"stage" => "stage-version"}
+
+        desc "sample_launch_options", "dump sample launch_options for create"
+        def sample_launch_options
+          $stdout.puts File.read(File.expand_path("../sample_launch_options.json", __FILE__))
+        end
+
+        desc "create NAME [LAUNCH_OPTIONS]", "create Host Instance"
+        def create(name, launch_options = nil)
+          stage_id = load_selection("stage-version")
+          launch_options = JSON.parse(File.readable?(launch_options) ? File.read(launch_options) : launch_options)
+
+          params = {
+            "host_instance" => {
+              "stage_id" => stage_id,
+              "name" => name,
+              "launch_options_yaml" => YAML.dump(launch_options),
+            }
+          }
+          post_json("/admin/host_instance/new.js", params)
+        end
+
+        desc "launch", "launch host instance"
+        def launch
+          id = load_selection(self.class.resource_name)["id"]
+          post_json("/admin/host_instance/#{id}/status_control.json", {})
+        end
+
+        desc "prepare", "prepare host instance"
+        def prepare
+          id = load_selection(self.class.resource_name)["id"]
+          put_json("/admin/host_instance/#{id}/status_control.json", {})
+        end
+
+        desc "stop", "stop host instance"
+        def stop
+          id = load_selection(self.class.resource_name)["id"]
+          delete("/admin/host_instance/#{id}/status_control.json")
+        end
+
+
+
+      end
+
+    end
+  end
+end

data/lib/magellan/cli/resources/project.rb

@@ -0,0 +1,22 @@
+# -*- coding: utf-8 -*-
+require "magellan/cli/resources"
+
+module Magellan
+  module Cli
+    module Resources
+
+      class Project < Base
+        self.resource_name = "project"
+        self.resource_dependency = nil
+
+        desc "update ATTRIBUTES", "update project with ATTRIBUTES"
+        def update(attrs)
+          s = load_selection("project")
+          attrs = JSON.parse(File.readable?(attrs) ? File.read(attrs) : attrs)
+          put_json("/admin/magellan~auth~project/#{s['id']}/edit", {"magellan_auth_project" => attrs})
+        end
+      end
+
+    end
+  end
+end

data/lib/magellan/cli/resources/stage.rb

@@ -0,0 +1,69 @@
+# -*- coding: utf-8 -*-
+require "magellan/cli/resources"
+require 'cgi'
+
+module Magellan
+  module Cli
+    module Resources
+
+      class Stage < Base
+        self.resource_name = "stage~title"
+        self.resource_dependency = {"project" => "project"}
+
+        desc "select NAME", "select Stage named by NAME"
+        def select(name)
+          q = build_query("name" => name).update(default_query)
+          r = update_first_result("stage", "/admin/stage~title.json", q)
+
+          # # current
+          # q = build_query("title" => r["id"], "phase" => 2) # 2: current
+          # update_first_result("stage-version", "/admin/stage~version.json", q, %w[id])
+
+          # # workspace
+          q = build_query("title" => r["id"], "phase" => 1) # 1: workspace
+          update_first_result("stage-version", "/admin/stage~version.json", q, %w[id])
+        end
+
+        desc "planning", "switch to planning to build next released version"
+        def planning
+          switch_version(1)
+        end
+
+        desc "current", "switch to current released version"
+        def current
+          switch_version(2)
+        end
+
+        no_commands do
+          def switch_version(phase)
+            s = load_selection("stage")
+            q = build_query("title" => s["id"], "phase" => phase) # 1: workspace, 2: current, 3: used
+            update_first_result("stage-version", "/admin/stage~version.json", q, %w[id])
+          end
+        end
+
+        desc "prepare", "prepare containers"
+        def prepare
+          s = load_selection("stage")
+          id = s["id"]
+          post_json("/admin/stage~title/#{id}/simple_method_call.json", {method_name: "prepare_containers"})
+        end
+
+        desc "update ATTRIBUTES", "update stage with ATTRIBUTES"
+        def update(attrs)
+          s = load_selection("stage")
+          attrs = JSON.parse(File.readable?(attrs) ? File.read(attrs) : attrs)
+          put_json("/admin/stage~title/#{s['id']}/edit", {"stage_title" => attrs})
+        end
+
+        desc "release_now", "release changes now"
+        def release_now
+          s = load_selection("stage")
+          id = s["id"]
+          post_json("/admin/stage~title/#{id}/simple_method_call.json", {method_name: "release_now"})
+        end
+      end
+
+    end
+  end
+end