iij-dag 1.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 1c7e0f6d32d600f1ab6d4d58f4ddc7f4cabccdfc
+  data.tar.gz: b05f864faab066b3d22342e9055efd0e5e9c64a5
+SHA512:
+  metadata.gz: 28108a90f7aba294fae6a1c6adc82302122d32ec8621cf50c509f39356feed2d69fa7333042a9e7b52e2a6eb74c820cc610fb0e026b5338796e90e0bcd46ff5e
+  data.tar.gz: 604962e5d56e69b8af3f1b4f1ba5fd98f96cce731304f9f2a6208d8cb29236fa3d3bcce7e37e22f817b411a1dfdc324d16fa803aec192ea38ba0e2e85ebfd8cd

data/.gitignore

@@ -0,0 +1,18 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+.idea

data/Gemfile

@@ -0,0 +1,16 @@
+source 'https://rubygems.org'
+
+gemspec
+
+group :test do
+  gem 'timecop'
+  gem 'pry', '< 0.10.0'
+  gem 'pry-byebug'
+  gem 'webmock'
+  gem 'simplecov'
+  gem 'simplecov-rcov'
+end
+
+group :doc do
+  gem 'yard'
+end

data/LICENSE.txt

@@ -0,0 +1,174 @@
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.

data/Rakefile

@@ -0,0 +1,27 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+task :default => :spec
+
+namespace :spec do
+  [:cli, :command, :sub_command].each do |dir|
+    desc "Run #{dir} spec"
+    RSpec::Core::RakeTask.new(dir) do |t|
+      t.pattern = "spec/#{dir}/*_spec.rb"
+    end
+  end
+end
+
+namespace :document do
+  task :build do
+    docpath = "../documents/parser"
+    if Dir.exist? docpath
+      pid = spawn("yard", "-o", docpath, STDERR => STDOUT)
+      Process.waitpid pid
+    else
+      puts "You don't have #{docpath} directory."
+      puts "Can't create a new document."
+    end
+  end
+end

data/bin/dag

@@ -0,0 +1,7 @@
+#!/usr/bin/env ruby
+$:.unshift File.join(File.dirname(__FILE__), *%w{ .. lib })
+
+require 'thor'
+require 'dag/cli'
+
+Dag::CLI::Command.start

data/iij-dag.gemspec

@@ -0,0 +1,34 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+
+require 'dag/cli/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "iij-dag"
+  spec.version       = Dag::CLI::VERSION
+  spec.authors       = ["iij"]
+  spec.email         = ["dag-info@iij.ad.jp"]
+  spec.description   = %q{iij gio dag command line tool}
+  spec.summary       = %q{iij gio dag command line tool}
+  spec.homepage      = "http://www.iij.ad.jp/biz/storage/"
+  spec.license       = "Apache License 2.0"
+
+  spec.files         = `git ls-files`.split($/).reject {|item| item =~ /^(examples|README|spec|.drone.yml)/ }
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "thor", "~> 0.19.1"
+  spec.add_dependency "ruby-progressbar", ">= 1.1.0"
+  spec.add_dependency "iij-dag-client", "~> 1.0.0"
+  spec.add_dependency "hirb", "~> 0.7.3"
+  spec.add_dependency "hirb-unicode", "~> 0.0.5"
+  spec.add_dependency "anbt-sql-formatter", "~> 0.0.3"
+  spec.add_dependency "oj", "~> 2.11.2"
+  spec.add_dependency "activesupport", "~> 4.2.0"
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "rspec", "~> 3.1.0"
+end

data/lib/dag/cli.rb

@@ -0,0 +1,17 @@
+module Dag
+  module CLI
+    autoload :Base,       'dag/cli/base'
+    autoload :Utils,      'dag/cli/utils'
+
+    autoload :Command,    'dag/cli/command'
+    autoload :Import,     'dag/cli/command/import'
+    autoload :Query,      'dag/cli/command/query'
+
+    autoload :SubCommand, 'dag/cli/sub_command'
+    autoload :Cluster,    'dag/cli/sub_command/cluster'
+    autoload :Db,         'dag/cli/sub_command/db'
+    autoload :Table,      'dag/cli/sub_command/table'
+    autoload :Job,        'dag/cli/sub_command/job'
+    autoload :Bucket,     'dag/cli/sub_command/bucket'
+  end
+end

data/lib/dag/cli/base.rb

@@ -0,0 +1,128 @@
+module Dag
+  module CLI
+    class Base < Thor
+      include Dag::CLI::Utils::TerminalTable
+
+      private
+
+      attr_accessor :debug, :trace
+
+      def local_store(&block)
+        require "yaml/store"
+        db = YAML::Store.new("#{File.dirname(options[:config])}/.cstore")
+        db.transaction { block.call(db) }
+      end
+
+      def  open_client(&block)
+        cluster = block_given? ? block.call : local_store { |db| db[:cluster] }
+        client.open(cluster)
+      end
+
+      def client(path = options[:config])
+        require 'oj'
+        require 'dag'
+
+        Oj.default_options = { mode: :compat }
+        # create_config_directory
+
+        if File.exist?(path)
+          begin
+            config = Oj.load(File.open(path, &:read))
+          rescue Oj::ParseError => ex
+            abort "Error: Json format is invalid.\n #{ex.message}"
+          end
+        end
+
+        if !config.nil? && valid?(config)
+          self.debug = config["debug"]
+          self.debug = options["debug"] unless options["debug"].nil?
+          self.trace = options["trace"]
+
+          options = {
+            force_path_style: config["force_path_style"] || false,
+            debug: self.debug || false
+          }
+
+          options.merge!(analysis_api: config["analysis_api"]) if config["analysis_api"]
+          options.merge!(storage_api: config["storage_api"]) if config["storage_api"]
+
+          return Dag::Client.new(config["access_key_id"], config["secret_access_key"], options)
+        else
+          error "Error: Please create config file to '#{path}'" if config.nil?
+          abort <<-EOS
+  {
+      "access_key_id"     : "YOUR ACCESS KEY ID",
+      "secret_access_key" : "YOUR SECRET ACCESS KEY",
+      "analysis_api"      : "ANALYSIS API ENDPOINT",
+      "storage_api"       : "STORAGE API ENDPOINT"
+  }
+          EOS
+        end
+      end
+
+      # def create_config_directory
+      #   begin
+      #     Dir.mkdir("#{Dir.home}/.dag")
+      #   rescue Errno::EEXIST
+      #   rescue => ex
+      #     error "Error: #{ex.message}"
+      #     exit 1
+      #   end
+      # end
+
+      def valid?(config)
+        unless config.key?("access_key_id")
+          error "Error: Missing access_key_id. Please check your configuration."
+          return false
+        end
+
+        unless config.key?("secret_access_key")
+          error "Error: Missing secret_access_key. Please check your configuration."
+          return false
+        end
+
+        true
+      end
+
+      def handle_api_failure(&block)
+        begin
+          block.call
+        rescue Dag::Client::APIFailure => ex
+          if self.debug
+            require 'pp'
+            PP.pp({
+                      api_code: ex.api_code,
+                      api_message: ex.api_message,
+                      api_status: ex.api_status,
+                      api_request_id: ex.api_request_id,
+                      api_resource: ex.api_resource,
+                      message: ex.message
+                  },
+                  STDERR
+            )
+          end
+
+          if ex.api_code && ex.api_message
+            say_status(ex.api_code, "#{ex.api_message} (RequestId: #{ex.api_request_id})", :red)
+            if ex.api_status == 500
+              error "An unexpected error has occurred. Please contact customer support."
+            end
+
+          else
+            say_status(
+                ex.cause,
+                "#{ex.cause}: An unexpected error has occurred. Please contact customer support.",
+                :red
+            )
+          end
+
+          exit 1
+        rescue => ex
+          say_status("ERROR", ex.message, :red)
+          error ex.backtrace * "\n" if self.trace
+          exit 1
+        end
+      end
+    end
+  end
+end

data/lib/dag/cli/command.rb

@@ -0,0 +1,83 @@
+# coding: utf-8
+
+module Dag
+  module CLI
+    class Command < Base
+      class_option :version, type: :string, banner: '', aliases: '-v', desc: "Show version"
+      class_option :config, type: :string, default: "#{Dir.home}/.dag/credential.json",
+                   aliases: '-C', desc: "Path to the config file", banner: "/path/to/file"
+      class_option :debug, type: :boolean, desc: "Show debug message"
+      class_option :trace, type: :boolean, desc: "Show error stacktrace", hide: true
+
+      def initialize(args = [], local_options = {}, config = {})
+        super
+        handle_version_request! if options.key?("version")
+      end
+
+      desc 'account', 'Show account info'
+      def account
+        fields = [:apikey, :analysis_api, :storage_api]
+        terminal_table([client], fields: fields)
+      end
+
+      desc 'import [DB] [TABLE]', 'Upload files and convert into analysis format'
+      option 'import-files', type: :array, aliases: '-i', desc: 'import files', required: true
+      option :format, type: :string, aliases: '-f', desc: 'file format [json_agent/json/csv/tsv]', required: true
+      option :label, type: :string, aliases: '-l', desc: 'label'
+      option :jobs, type: :numeric, aliases: '-j', desc: 'jobs'
+      def import(db, table)
+        params = {}
+        params.merge!({ label: options[:label] }) if options[:label]
+        params.merge!({ jobs: options[:jobs] }) if options[:jobs]
+
+        import = Import.new(open_client, db, table, options['import-files'], options[:format], params)
+        handle_api_failure { import.execute }
+      end
+
+      desc 'query "[QUERY_STRING]"', 'Execute Hive Query'
+      option :output, type: :string, aliases: '-o', desc: 'output storage path', required: true
+      option :format, type: :string, aliases: '-f', desc: 'output format [csv/tsv]', default: 'csv'
+      option :label, type: :string, aliases: '-l', desc: 'job label'
+      option :wait, type: :boolean, aliases: '-w', desc: 'wait job finish'
+      def query(query_string)
+        if File.exists? query_string
+          query_string = open(query_string).read
+          puts "Query: #{query_string}"
+        end
+        params = {
+            query: query_string,
+            output_format: options[:format],
+            output_resource_path: options[:output]
+        }
+        params.merge!(label: options[:label]) if options[:label]
+        query = Query.new(open_client, params)
+        job = handle_api_failure { query.execute }
+        say_status "accepted job", "job_id: #{job.id}"
+        if options[:wait]
+          CLI::Command.start(["job", "log", job.id.to_s, "-t"])
+          CLI::Command.start(["job", "result", job.id.to_s])
+        end
+      end
+
+      %w(cluster db table job bucket).each do |cmd|
+        desc "#{cmd} [COMMAND]", "Management command for #{cmd}"
+        subcommand cmd, Dag::CLI.const_get(cmd.capitalize)
+      end
+
+      desc 'help [COMMAND]', 'Describe available commands or one specific command'
+      def help(command = nil, subcommand = false)
+        super
+        shell.say "Type 'dag help [COMMAND]' for more information on a specific command"
+      end
+
+      private
+
+      def handle_version_request!
+        require 'dag/cli/version'
+        require 'dag/client/version'
+        print_table [["dag-ruby", VERSION], ["dag-client-ruby", Dag::Client::VERSION]]
+        exit 0
+      end
+    end
+  end
+end