haile 0.3.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 621dd9bed5314f84e53e84a1a034d2ee7a3f5883
+  data.tar.gz: d215682f47b347ed7c05badf06656d98c7846a4d
+SHA512:
+  metadata.gz: 91f6d330486939e5da6272e33d840de4415c7a43d69de16bf20687b1dd2f4b17735a81c7588b193a9896e119cc7df10f88035b619c491c00bd84b44cd6f0787d
+  data.tar.gz: 774672fdaa5ef6bbf9b9957dc53ab34fdd45e2e9cf278908de90c88b3978d741c700fdf0c869ae560797564a01df2179b7b5ac4a01ae7c6583750f973db040bd

data/.gitignore

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

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in haile.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,23 @@
+Copyright (c) 2013 Tobi Knaup
+Copyright (c) 2014 Chris Kite
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,22 @@
+# Haile
+
+Client gem for the Marathon scheduler
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'haile'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install haile
+
+## Usage
+
+
+## Development

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/bin/haile

@@ -0,0 +1,156 @@
+#!/usr/bin/env ruby
+
+require File.expand_path(File.join(File.dirname(__FILE__), '..', 'lib', 'marathon'))
+require 'trollop'
+
+SUB_COMMANDS = %w[endpoints kill kill_tasks start scale list list_tasks search]
+
+global_opts = Trollop.options do
+  version Marathon::VERSION
+  banner <<-EOS
+Usage: marathon [global options] [command] [options]
+
+Available commands:
+
+  kill        Kill an app and remove it from Marathon.
+  kill_tasks  Kill a task or tasks belonging to a specified app.
+  list        Show a list of running apps and their options.
+  list_tasks  Show a list of an app's running tasks.
+  scale       Scale the number of app instances.
+  search      Search the current list of apps.
+  start       Start a new app.
+
+Global options:
+EOS
+
+  opt :marathon_host, 'Marathon host (default http://localhost:8080, or MARATHON_HOST)', :short => '-H', :type => String
+  opt :marathon_user, 'User name to authenticate against Marathon (optional).', :short => '-U', :type => String
+  opt :marathon_pass, 'Password to authenticate against Marathon (optional).', :short => '-P', :type => String
+  stop_on SUB_COMMANDS
+end
+
+cmd = ARGV.shift # get the subcommand
+
+cmd_opts = case cmd
+  when 'endpoints'
+    Trollop.options do
+      opt :id, 'A unique identifier for the app.', :short => '-i', :type => String
+    end
+  when 'kill_tasks'
+    Trollop.options do
+      opt :host, 'Scope task killing to the given host.', :short => '-H', :type => String
+      opt :id, 'A unique identifier for the app.', :short => '-i', :required => true, :type => String
+      opt :scale, 'If true, the app is scaled down after killing tasks', :short => '-s'
+      opt :task_id, 'A unique identifier for the task.', :short => '-t', :type => String
+    end
+  when 'list_tasks'
+    Trollop.options do
+      opt :id, 'A unique identifier for the app.', :short => '-i', :required => true, :type => String
+    end
+  when 'start'
+    Trollop.options do
+      opt :id, 'A unique identifier for the app.', :required => true, :short => '-i', :type => String
+      opt :command, 'The command to start the app.', :short => '-C', :type => String
+      opt :executor, 'The mesos executor to be used to launch the app.', :short => '-X', :type => String
+      opt :num_instances, 'The number of instances to run (default 1).', :default => 1, :short => '-n'
+      opt :cpus, 'The number of CPUs to give to this app, can be a fraction (default 1.0).', :default => 1.0, :short => '-c'
+      opt :mem, 'The memory limit for this app, in MB, can be a fraction (default 10.0).', :default => 10.0, :short => '-m'
+      opt :uri, 'URIs to download and unpack into the working directory.', :short => '-u', :type => :strings
+      opt :env, 'Environment variables to add to the process, as NAME=VALUE.', :short => '-e', :type => :strings
+      opt :constraint, 'Placement constraint for tasks, e.g. hostname:UNIQUE or rackid:CLUSTER', :type => :strings
+    end
+  when 'scale'
+    Trollop.options do
+      opt :id, 'A unique identifier for the app.', :required => true, :short => '-i', :type => String
+      opt :num_instances, 'The number of instances to run.', :required => true, :short => '-n', :type => Integer
+    end
+  when 'kill'
+    Trollop.options do
+      opt :id, 'A unique identifier for the app.', :required => true, :short => '-i', :type => String
+    end
+  when 'search'
+    Trollop.options do
+      opt :id, 'A unique identifier for the app.', :short => '-i', :type => String, :default => nil
+      opt :command, 'The command for the app.', :short => '-C', :type => String, :default => nil
+    end
+  else
+    {}
+  end
+
+marathon = Marathon::Client.new(
+  global_opts[:marathon_host],
+  global_opts[:marathon_user],
+  global_opts[:marathon_pass]
+)
+
+def handle_listing(res)
+  if res.success?
+    apps = res.parsed_response['apps']
+
+    if apps.empty?
+      puts "No apps currently running"
+    else
+      apps.each do |app|
+        puts "App ID:     #{app['id']}"
+        puts "Command:    #{app['cmd']}"
+        puts "Instances:  #{app['instances']}"
+        puts "CPUs:       #{app['cpus']}"
+        puts "Memory:     #{app['mem']} MB"
+        app['uris'].each do |uri|
+          puts "URI:        #{uri}"
+        end
+        app['env'].each do |k, v|
+          puts "ENV:        #{k}=#{v}"
+        end
+        app['constraints'].each do |constraint|
+          puts "Constraint: #{constraint.join(':')}"
+        end
+        puts
+      end
+    end
+
+  else
+    puts res
+  end
+end
+
+# Run
+case cmd
+when 'endpoints'
+  puts marathon.endpoints(cmd_opts[:id]).parsed_response
+when 'start'
+  app_opts = {
+    :instances => cmd_opts[:num_instances] || 1,
+    :uris => cmd_opts[:uri] || [],
+    :cmd => cmd_opts[:command],
+    :env => cmd_opts[:env].nil? ? {} : Hash[cmd_opts[:env].map { |e| e.split('=', 2) }],
+    :cpus => cmd_opts[:cpus] || 1.0,
+    :mem => cmd_opts[:mem] || 10.0,
+    :constraints => (cmd_opts[:constraint] || []).map { |c| c.split(':') }
+  }
+  app_opts[:executor] = cmd_opts[:executor] unless cmd_opts[:executor] == nil
+  puts "Starting app '#{cmd_opts[:id]}'"
+  puts marathon.start(cmd_opts[:id], app_opts)
+when 'scale'
+  puts "Scaling app '#{cmd_opts[:id]}' to #{cmd_opts[:num_instances]} instances"
+  res = marathon.scale(cmd_opts[:id], cmd_opts[:num_instances])
+  puts res
+when 'kill'
+  puts "Killing app '#{cmd_opts[:id]}'"
+  puts marathon.kill(cmd_opts[:id])
+when 'kill_tasks'
+  KILL_TASKS_KEYS = [:host, :scale, :task_id]
+
+  opts = cmd_opts.clone
+  opts.select! {|k, v| KILL_TASKS_KEYS.include?(k)}
+
+  puts marathon.kill_tasks(cmd_opts[:id], opts).parsed_response
+when 'list'
+  handle_listing(marathon.list)
+when 'list_tasks'
+  puts marathon.list_tasks(cmd_opts[:id]).parsed_response
+when 'search'
+  handle_listing(marathon.search(cmd_opts[:id], cmd_opts[:command]))
+else
+  Trollop.die "unknown subcommand #{cmd.inspect}"
+end

data/haile.gemspec

@@ -0,0 +1,26 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'haile/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "haile"
+  spec.version       = Haile::VERSION
+  spec.authors       = ["Tobi Knaup", "Chris Kite"]
+  spec.description   = %q{Ruby client for Marathon REST API}
+  spec.summary       = %q{Ruby client for Marathon REST API}
+  spec.homepage      = "https://github.com/chriskite/haile"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  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 "trollop", "~> 2.0"
+  spec.add_dependency "httparty", "~> 0.11"
+  spec.add_dependency "multi_json", "~> 1.8"
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake", "~> 0"
+end

data/lib/haile/client.rb

@@ -0,0 +1,110 @@
+require 'uri'
+
+module Haile
+  class Client
+    include HTTParty
+
+    headers(
+      'Content-Type' => 'application/json',
+      'Accept' => 'application/json'
+    )
+
+    query_string_normalizer proc { |query| MultiJson.dump(query) }
+    maintain_method_across_redirects
+    default_timeout 5
+
+    EDITABLE_APP_ATTRIBUTES = [
+      :cmd, :constraints, :container, :cpus, :env, :executor, :id, :instances,
+      :mem, :ports, :uris]
+
+    def initialize(url = nil, user = nil, pass = nil, proxy = nil)
+      @host = url || ENV['MARATHON_URL'] || 'http://localhost:8080'
+      @default_options = {}
+
+      if user && pass
+        @default_options[:basic_auth] = {:username => user, :password => pass}
+      end
+
+      if proxy
+        @default_options[:http_proxyaddr] = proxy[:addr]
+        @default_options[:http_proxyport] = proxy[:port]
+        @default_options[:http_proxyuser] = proxy[:user] if proxy[:user]
+        @default_options[:http_proxypass] = proxy[:pass] if proxy[:pass]
+      end
+    end
+
+    def list
+      wrap_request(:get, '/v2/apps')
+    end
+
+    def list_tasks(id)
+      wrap_request(:get, URI.escape("/v2/apps/#{id}/tasks"))
+    end
+
+    def search(id = nil, cmd = nil)
+      params = {}
+      params[:id] = id unless id.nil?
+      params[:cmd] = cmd unless cmd.nil?
+
+      wrap_request(:get, "/v2/apps?#{query_params(params)}")
+    end
+
+    def endpoints(id = nil)
+      if id.nil?
+        url = "/v2/tasks"
+      else
+        url = "/v2/apps/#{id}/tasks"
+      end
+
+      wrap_request(:get, url)
+    end
+
+    def start(id, opts)
+      body = opts.dup
+      body[:id] = id
+      wrap_request(:post, '/v2/apps/', :body => body)
+    end
+
+    def scale(id, num_instances)
+      # Fetch current state and update only the 'instances' attribute. Since the
+      # API only supports PUT, the full representation of the app must be
+      # supplied to update even just a single attribute.
+      app = wrap_request(:get, "/v2/apps/#{id}").parsed_response['app']
+      app.select! {|k, v| EDITABLE_APP_ATTRIBUTES.include?(k)}
+
+      app[:instances] = num_instances
+      wrap_request(:put, "/v2/apps/#{id}", :body => app)
+    end
+
+    def kill(id)
+      wrap_request(:delete, "/v2/apps/#{id}")
+    end
+
+    def kill_tasks(appId, params = {})
+      if params[:task_id].nil?
+        wrap_request(:delete, "/v2/apps/#{appId}/tasks?#{query_params(params)}")
+      else
+        query = params.clone
+        task_id = query[:task_id]
+        query.delete(:task_id)
+
+        wrap_request(:delete, "/v2/apps/#{appId}/tasks/#{task_id}?#{query_params(query)}")
+      end
+    end
+
+    private
+
+    def wrap_request(method, url, options = {})
+      options = @default_options.merge(options)
+      http = self.class.send(method, @host + url, options)
+      Marathon::Response.new(http)
+    rescue => e
+      Marathon::Response.error(e.message)
+    end
+
+    def query_params(hash)
+      hash = hash.select { |k,v| !v.nil? }
+      URI.escape(hash.map { |k,v| "#{k}=#{v}" }.join('&'))
+    end
+  end
+end

data/lib/haile/response.rb

@@ -0,0 +1,46 @@
+module Haile
+  class Response
+
+    # TODO make this attr_reader and set the error some other way
+    attr_accessor :error
+
+    def initialize(http)
+      @http = http
+      @error = error_message_from_response
+    end
+
+    def success?
+      @http && @http.success?
+    end
+
+    def error?
+      !success?
+    end
+
+    def parsed_response
+      @http && @http.parsed_response
+    end
+
+    def self.error(message)
+      error = new(nil)
+      error.error = message
+      error
+    end
+
+    def to_s
+      if success?
+        "OK"
+      else
+        "ERROR: #{error}"
+      end
+    end
+
+    private
+
+    def error_message_from_response
+      return if success?
+      return if @http.nil?
+      @http.body
+    end
+  end
+end

data/lib/haile/version.rb

@@ -0,0 +1,3 @@
+module Haile
+  VERSION = "0.3.0"
+end

data/lib/haile.rb

@@ -0,0 +1,12 @@
+__LIB_DIR__ = File.expand_path(File.dirname(__FILE__))
+$LOAD_PATH.unshift __LIB_DIR__ unless $LOAD_PATH.include?(__LIB_DIR__)
+
+require "rubygems"
+require "httparty"
+
+require "haile/version"
+require "haile/client"
+require "haile/response"
+
+module Haile
+end

metadata

@@ -0,0 +1,126 @@
+--- !ruby/object:Gem::Specification
+name: haile
+version: !ruby/object:Gem::Version
+  version: 0.3.0
+platform: ruby
+authors:
+- Tobi Knaup
+- Chris Kite
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-12-15 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: trollop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: httparty
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.11'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.11'
+- !ruby/object:Gem::Dependency
+  name: multi_json
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.8'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.8'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Ruby client for Marathon REST API
+email: 
+executables:
+- haile
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/haile
+- haile.gemspec
+- lib/haile.rb
+- lib/haile/client.rb
+- lib/haile/response.rb
+- lib/haile/version.rb
+homepage: https://github.com/chriskite/haile
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.0.0.rc.2
+signing_key: 
+specification_version: 4
+summary: Ruby client for Marathon REST API
+test_files: []