iron_worker 2.4.3 → 3.0.0

data/lib/iron_worker/api_client.rb

@@ -0,0 +1,152 @@
+require 'time'
+require 'iron_core'
+
+module IronWorker
+  class APIClient < IronCore::Client
+    AWS_US_EAST_HOST = 'worker-aws-us-east-1.iron.io'
+
+    def initialize(options = {})
+      default_options = {
+          :scheme => 'https',
+          :host => IronWorker::APIClient::AWS_US_EAST_HOST,
+          :port => 443,
+          :api_version => 2,
+          :user_agent => IronWorker.full_version
+      }
+
+      super('iron', 'worker', options, default_options, [:project_id, :token, :api_version])
+
+      IronCore::Logger.debug 'IronWorker', "nhp.proxy: #{rest.wrapper.http.proxy_uri}" if defined? Net::HTTP::Persistent
+      IronCore::Logger.debug 'IronWorker', "RestClient.proxy: #{RestClient.proxy}" if defined? RestClient
+      IronCore::Logger.debug 'IronWorker', "InternalClient.proxy: #{Rest::InternalClient.proxy}" if defined? Rest::InternalClient
+
+      IronCore::Logger.error 'IronWorker', "Token is not set", IronCore::Error if @token.nil?
+
+      check_id(@project_id, 'project_id')
+    end
+
+    def headers
+      super.merge({'Authorization' => "OAuth #{@token}"})
+    end
+
+    def base_url
+      super + @api_version.to_s + '/'
+    end
+
+    def stacks_list
+      parse_response(get("stacks"))
+    end
+
+    def codes_list(options = {})
+      parse_response(get("projects/#{@project_id}/codes", options))
+    end
+
+    def codes_get(id)
+      check_id(id)
+      parse_response(get("projects/#{@project_id}/codes/#{id}"))
+    end
+
+    def codes_create(name, file, runtime, runner, options)
+      parse_response(post_file("projects/#{@project_id}/codes", :file, File.new(file, 'rb'), :data, {:name => name, :runtime => runtime, :file_name => runner}.merge(options)))
+    end
+
+    def codes_delete(id)
+      check_id(id)
+      parse_response(delete("projects/#{@project_id}/codes/#{id}"))
+    end
+
+    def codes_revisions(id, options = {})
+      check_id(id)
+      parse_response(get("projects/#{@project_id}/codes/#{id}/revisions", options))
+    end
+
+    def codes_download(id, options = {})
+      check_id(id)
+      parse_response(get("projects/#{@project_id}/codes/#{id}/download", options), false)
+    end
+
+    def codes_pause_task_queue(id, options = {})
+      check_id(id)
+      parse_response(post("projects/#{@project_id}/codes/#{id}/pause_task_queue", options))
+    end
+
+    def codes_resume_task_queue(id, options = {})
+      check_id(id)
+      parse_response(post("projects/#{@project_id}/codes/#{id}/resume_task_queue", options))
+    end
+
+    def tasks_list(options = {})
+      parse_response(get("projects/#{@project_id}/tasks", options))
+    end
+
+    def tasks_get(id)
+      check_id(id)
+      parse_response(get("projects/#{@project_id}/tasks/#{id}"))
+    end
+
+    def tasks_create(code_name, payload, options = {})
+      parse_response(post("projects/#{@project_id}/tasks", {:tasks => [{:code_name => code_name, :payload => payload}.merge(options)]}))
+    end
+
+    def tasks_cancel(id)
+      check_id(id)
+      parse_response(post("projects/#{@project_id}/tasks/#{id}/cancel"))
+    end
+
+    def tasks_cancel_all(id)
+      check_id(id)
+      parse_response(post("projects/#{@project_id}/codes/#{id}/cancel_all"))
+    end
+
+    def tasks_log(id)
+      check_id(id)
+      if block_given?
+        stream_get("projects/#{@project_id}/tasks/#{id}/log") do |chunk|
+          yield chunk
+        end
+      else
+        parse_response(get("projects/#{@project_id}/tasks/#{id}/log"), false)
+      end
+    end
+
+    def tasks_set_progress(id, options = {})
+      check_id(id)
+      parse_response(post("projects/#{@project_id}/tasks/#{id}/progress", options))
+    end
+
+    def tasks_retry(id, options = {})
+      check_id(id)
+      parse_response(post("projects/#{@project_id}/tasks/#{id}/retry", options))
+    end
+
+    def schedules_list(options = {})
+      parse_response(get("projects/#{@project_id}/schedules", options))
+    end
+
+    def schedules_get(id)
+      check_id(id)
+      parse_response(get("projects/#{@project_id}/schedules/#{id}"))
+    end
+
+    def schedules_create(code_name, payload, options = {})
+      options[:start_at] = options[:start_at].iso8601 if (not options[:start_at].nil?) && options[:start_at].respond_to?(:iso8601)
+      options[:end_at] = options[:end_at].iso8601 if (not options[:end_at].nil?) && options[:end_at].respond_to?(:iso8601)
+
+      parse_response(post("projects/#{@project_id}/schedules", {:schedules => [{:code_name => code_name, :payload => payload}.merge(options)]}))
+    end
+
+    def schedules_update(id, options = {})
+      check_id(id)
+      parse_response(put("projects/#{@project_id}/schedules/#{id}", options))
+    end
+
+    def schedules_cancel(id)
+      check_id(id)
+      parse_response(post("projects/#{@project_id}/schedules/#{id}/cancel"))
+    end
+
+    def projects_get
+      parse_response(get("projects/#{@project_id}"))
+    end
+  end
+end

data/lib/iron_worker/version.rb

@@ -1,4 +1,12 @@
 module IronWorker
-  VERSION = "2.4.3"
+  VERSION = "3.0.0"
+
+  def self.version
+    VERSION
+  end
+
+  def self.full_version
+    'iron_worker_ruby_-' + IronWorker.version + ' (iron_core_ruby-' + IronCore.version + ')'
+  end
 end
 

data/lib/iron_worker/worker_helper.rb

@@ -0,0 +1,72 @@
+# This could be a gem to help worker users with common functions like getting config and payload.
+require 'json'
+
+module IronWorker
+  @@loaded = false
+  @@args = {}
+  @@payload = {}
+  @@config = {}
+
+  def self.loaded
+    return @@loaded
+  end
+
+  def self.load
+    return if @@loaded
+    0.upto($*.length - 2) do |i|
+      @@args[:root] = $*[i + 1] if $*[i] == '--dir'
+      @@args[:payload_file] = $*[i + 1] if $*[i] == '--payload'
+      @@args[:config_file] = $*[i + 1] if $*[i] == '--config'
+      @@args[:task_id] = $*[i + 1] if $*[i] == '--id'
+    end
+
+    # puts "args: #{@@args.inspect}"
+
+    if args[:payload_file]
+      @@payload = File.read(@@args[:payload_file])
+
+      begin
+        @@payload = JSON.parse(@@payload)
+      rescue => ex
+        puts "Couldn't parse IronWorker payload into json, leaving as string. #{ex}"
+      end
+    end
+
+    if args[:config_file]
+      if args[:config_file]
+        @@config = File.read(args[:config_file])
+        begin
+          @@config = JSON.parse(@@config)
+        rescue
+          # try yaml
+          begin
+            @@config = YAML.load(@@config)
+          rescue => ex
+
+          end
+        end
+      end
+    end
+    @@loaded = true
+  end
+
+  def self.payload
+    load
+    @@payload
+  end
+
+  def self.config
+    load
+    @@config
+  end
+
+  def self.task_id
+    @@args[:task_id]
+  end
+
+  def self.args
+    return @@args
+  end
+
+
+end

metadata

@@ -1,168 +1,59 @@
 --- !ruby/object:Gem::Specification
 name: iron_worker
 version: !ruby/object:Gem::Version
-  version: 2.4.3
+  version: 3.0.0
 platform: ruby
 authors:
 - Travis Reeder
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-02-28 00:00:00.000000000 Z
+date: 2015-03-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: iron_core
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: 0.5.1
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: 0.5.1
-- !ruby/object:Gem::Dependency
-  name: rubyzip
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - '>='
-      - !ruby/object:Gem::Version
-        version: '0'
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - '>='
-      - !ruby/object:Gem::Version
-        version: '0'
-- !ruby/object:Gem::Dependency
-  name: rest-client
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - '>='
-      - !ruby/object:Gem::Version
-        version: '0'
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - '>='
-      - !ruby/object:Gem::Version
-        version: '0'
 - !ruby/object:Gem::Dependency
   name: rest
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
-      - !ruby/object:Gem::Version
-        version: '0'
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - '>='
-      - !ruby/object:Gem::Version
-        version: '0'
-- !ruby/object:Gem::Dependency
-  name: bundler
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 3.0.6
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
-description: DEPRECATED!! Use iron_worker_ng now. The official IronWorker gem for
-  IronWorker by Iron.io. http://www.iron.io
+        version: 3.0.6
+description: The official IronWorker gem for IronWorker by Iron.io. http://iron.io
 email:
 - travis@iron.io
 executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- .gitignore
+- ".gitignore"
 - LICENSE
-- README.markdown
+- README.md
 - Rakefile
 - iron_worker.gemspec
-- lib/generators/iron_worker/iron_worker_generator.rb
-- lib/generators/iron_worker/templates/template_worker.erb
 - lib/iron_worker.rb
-- lib/iron_worker/api.rb
-- lib/iron_worker/base.rb
-- lib/iron_worker/config.rb
-- lib/iron_worker/rails2_init.rb
-- lib/iron_worker/railtie.rb
-- lib/iron_worker/server/overrides.rb
-- lib/iron_worker/server/runner.rb
-- lib/iron_worker/service.rb
-- lib/iron_worker/used_in_worker.rb
-- lib/iron_worker/utils.rb
+- lib/iron_worker/api_client.rb
 - lib/iron_worker/version.rb
-- rails/init.rb
-- test/Gemfile
-- test/Gemfile.lock
-- test/README.md
-- test/batch_run.rb
-- test/config_sample.yml
-- test/models/cool_model.rb
-- test/models/db_model.rb
-- test/models/model_1.rb
-- test/models/model_2.rb
-- test/quick_run.rb
-- test/quick_schedule.rb
-- test/resources/something.yml
-- test/simple_script.rb
-- test/test_base.rb
-- test/test_database.rb
-- test/test_gems.rb
-- test/test_inheritance.rb
-- test/test_iron_worker.rb
-- test/test_merging.rb
-- test/test_non_worker_script.rb
-- test/test_rails.rb
-- test/test_scheduling.rb
-- test/test_scheduling_no_active_support.rb
-- test/test_service.rb
-- test/test_uploads.rb
-- test/test_webhooks.rb
-- test/v2quick.rb
-- test/workers/awesome_job.rb
-- test/workers/aws_s3_worker.rb
-- test/workers/big_data_worker.rb
-- test/workers/big_gems_worker.rb
-- test/workers/cool_worker.rb
-- test/workers/db_worker.rb
-- test/workers/fail_worker.rb
-- test/workers/gem_dependency_worker.rb
-- test/workers/hacker_worker.rb
-- test/workers/iw_test_worker.rb
-- test/workers/iw_test_worker_2.rb
-- test/workers/iw_test_worker_3.rb
-- test/workers/local_vs_remote_worker.rb
-- test/workers/merging_worker.rb
-- test/workers/mq_worker.rb
-- test/workers/one_line_worker.rb
-- test/workers/prawn_worker.rb
-- test/workers/progress_worker.rb
-- test/workers/qb_worker.rb
-- test/workers/rails_worker.rb
-- test/workers/rails_worker2.rb
-- test/workers/requiring_worker.rb
-- test/workers/running_back_worker.rb
-- test/workers/scheduled_worker.rb
-- test/workers/second_worker.rb
-- test/workers/third_worker.rb
-- test/workers/webhook_worker.rb
-- test/zip_vs_gzip.rb
+- lib/iron_worker/worker_helper.rb
 homepage: https://github.com/iron-io/iron_worker_ruby
 licenses: []
 metadata: {}
@@ -172,74 +63,18 @@ require_paths:
 - lib
 required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
-  - - '>='
+  - - ">="
     - !ruby/object:Gem::Version
       version: '1.9'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
-  - - '>='
+  - - ">="
     - !ruby/object:Gem::Version
       version: 1.3.6
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.2.2
+rubygems_version: 2.4.2
 signing_key: 
 specification_version: 4
-summary: DEPRECATED!! Use iron_worker_ng now. The official IronWorker gem for IronWorker
-  by Iron.io. http://www.iron.io
-test_files:
-- test/Gemfile
-- test/Gemfile.lock
-- test/README.md
-- test/batch_run.rb
-- test/config_sample.yml
-- test/models/cool_model.rb
-- test/models/db_model.rb
-- test/models/model_1.rb
-- test/models/model_2.rb
-- test/quick_run.rb
-- test/quick_schedule.rb
-- test/resources/something.yml
-- test/simple_script.rb
-- test/test_base.rb
-- test/test_database.rb
-- test/test_gems.rb
-- test/test_inheritance.rb
-- test/test_iron_worker.rb
-- test/test_merging.rb
-- test/test_non_worker_script.rb
-- test/test_rails.rb
-- test/test_scheduling.rb
-- test/test_scheduling_no_active_support.rb
-- test/test_service.rb
-- test/test_uploads.rb
-- test/test_webhooks.rb
-- test/v2quick.rb
-- test/workers/awesome_job.rb
-- test/workers/aws_s3_worker.rb
-- test/workers/big_data_worker.rb
-- test/workers/big_gems_worker.rb
-- test/workers/cool_worker.rb
-- test/workers/db_worker.rb
-- test/workers/fail_worker.rb
-- test/workers/gem_dependency_worker.rb
-- test/workers/hacker_worker.rb
-- test/workers/iw_test_worker.rb
-- test/workers/iw_test_worker_2.rb
-- test/workers/iw_test_worker_3.rb
-- test/workers/local_vs_remote_worker.rb
-- test/workers/merging_worker.rb
-- test/workers/mq_worker.rb
-- test/workers/one_line_worker.rb
-- test/workers/prawn_worker.rb
-- test/workers/progress_worker.rb
-- test/workers/qb_worker.rb
-- test/workers/rails_worker.rb
-- test/workers/rails_worker2.rb
-- test/workers/requiring_worker.rb
-- test/workers/running_back_worker.rb
-- test/workers/scheduled_worker.rb
-- test/workers/second_worker.rb
-- test/workers/third_worker.rb
-- test/workers/webhook_worker.rb
-- test/zip_vs_gzip.rb
+summary: The official IronWorker gem for IronWorker by Iron.io. http://iron.io
+test_files: []