priteau-vizir 0.1.1

data/.document

@@ -0,0 +1,5 @@
+README.rdoc
+lib/**/*.rb
+bin/*
+features/**/*.feature
+LICENSE

data/.gitignore

@@ -0,0 +1,5 @@
+*.sw?
+.DS_Store
+coverage
+rdoc
+pkg

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Pierre Riteau
+
+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.rdoc

@@ -0,0 +1,34 @@
+= Vizir
+
+    Disconnected from OAR job 960063
+
+Ever been disconnected from Grid'5000 machines with a message like this, losing part or all of your work?
+
+Vizir is a simple Ruby script that monitors your interactive jobs on Grid'5000[http://www.grid5000.org/].
+It triggers Growl[http://growl.info/] notifications when your reservations are going to terminate, allowing you to save your work and/or your deployed environments.
+
+Vizir only supports Mac OS X for now.
+
+== Getting Started
+
+Run the following if you haven't already:
+    $ gem sources -a http://gems.github.com
+
+Install the gem:
+    $ sudo gem install mojombo-jekyll
+
+Configure Growl to accept notifications. Go to the Growl preference panel, and in the Network options enable "Listen for incoming notifications" and "Allow remote application registration". Don't forget to setup a password.
+
+Run it in the background, using the <tt>--login</tt> switch with your Grid'5000 login name and the <tt>--password</tt> switch with your Growl password.
+
+    $ vizir --login priteau --password mygrowlpassword &
+
+== Getting the source
+
+Clone it from GitHub:
+
+    $ git clone git://github.com/priteau/vizir.git
+
+== Copyright
+
+Copyright (c) 2009 Pierre Riteau. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,60 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "vizir"
+    gem.summary = "Growl notifications for Grid'5000 jobs"
+    gem.email = "pierre.riteau@gmail.com"
+    gem.homepage = "http://github.com/priteau/vizir"
+    gem.authors = ["Pierre Riteau"]
+    gem.add_dependency "json"
+    gem.add_dependency "net-ssh-gateway"
+    gem.add_dependency "rest-client"
+    gem.add_dependency "ruby-growl"
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+  end
+
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: sudo gem install jeweler"
+end
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/*_test.rb'
+  test.verbose = true
+end
+
+begin
+  require 'rcov/rcovtask'
+  Rcov::RcovTask.new do |test|
+    test.libs << 'test'
+    test.pattern = 'test/**/*_test.rb'
+    test.verbose = true
+  end
+rescue LoadError
+  task :rcov do
+    abort "RCov is not available. In order to run rcov, you must: sudo gem install spicycode-rcov"
+  end
+end
+
+
+task :default => :test
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  if File.exist?('VERSION.yml')
+    config = YAML.load(File.read('VERSION.yml'))
+    version = "#{config[:major]}.#{config[:minor]}.#{config[:patch]}"
+  else
+    version = ""
+  end
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "vizir #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end
+

data/VERSION

@@ -0,0 +1 @@
+0.1.1

data/bin/vizir

@@ -0,0 +1,105 @@
+#!/usr/bin/env ruby -w
+
+require 'json'
+require 'net/ssh/gateway'
+require 'optparse'
+require 'rest_client'
+require 'ruby-growl'
+
+ALERT_TIME = 600
+IGNORE_SITES = ['grenoble-exp', 'grenoble-ext', 'grenoble-obs', 'luxembourg', 'portoalegre']
+
+def usage
+  puts 'usage: vizir [ -h | --help ] [ -l | --login login_name ] [ -p | --password mygrowlpassword ]'
+  exit 1
+end
+
+def get(api, uri)
+  begin
+    return JSON.parse(api[uri + '?structure=simple'].get(:accept => 'application/json'))
+  rescue => e
+    if e.respond_to?('http_code')
+      puts "Error: #{e.http_code}:\n #{e.response.body}"
+      return nil
+    else
+      puts e.inspect
+      exit 1
+    end
+  end
+end
+
+def setup_ssh_tunnel
+  gateway = Net::SSH::Gateway.new('access.grenoble.grid5000.fr', $login)
+  port = gateway.open('api.grenoble.grid5000.fr', 443)
+  return gateway, port
+end
+
+def apiuri(port)
+  return "https://localhost:#{port}/oargridapi"
+end
+
+def notify_via_growl(jobid, site_name, time, time_unit)
+  g = Growl.new("localhost", "Vizir", ["Job ending soon"], nil, $password)
+  g.notify("Job ending soon", "OAR job terminating soon", "Job #{jobid} in #{site_name} ending in #{time} #{time_unit}.")
+end
+
+def get_remaining_time(starttime, walltime)
+  remaining_sec = (Time.at(starttime + walltime) - Time.now).round
+end
+
+def humanize_time(time)
+  if time > 60 then
+    remaining_time = time / 60
+    remaining_time_unit = "minutes"
+  else
+    remaining_time = time
+    remaining_time_unit = "seconds"
+  end
+  return remaining_time, remaining_time_unit
+end
+
+$login = nil
+$password = nil
+opts = OptionParser.new
+opts.on('-h', '--help') { usage }
+opts.on('-l', '--login STRING', String) { |str| $login = str }
+opts.on('-p', '--password STRING', String) { |str| $password = str }
+opts.parse(ARGV)
+
+if $login == nil or $password == nil
+  usage
+end
+
+gateway, port = setup_ssh_tunnel
+api = RestClient::Resource.new(apiuri(port))
+
+sites = get(api, '/sites')
+if sites == nil then
+  exit 1
+end
+sites.each do |site|
+  site_name = site['site']
+  unless IGNORE_SITES.include?(site_name)
+    jobs = get(api, "#{site['uri']}/jobs")
+    if jobs == nil then
+      break
+    end
+    jobs.each do |job|
+      if job['owner'] == $login then
+        job_details = get(api, "#{job['uri']}")
+        if job_details == nil then
+          break
+        end
+        if job_details['jobType'] == 'INTERACTIVE' then
+          remaining_sec = get_remaining_time(Integer(job_details['startTime']), Integer(job_details['walltime']))
+          if remaining_sec < ALERT_TIME then
+            remaining_time, remaining_time_unit = humanize_time(remaining_sec)
+            notify_via_growl(job_details['Job_Id'], site_name.capitalize, remaining_time, remaining_time_unit)
+          end
+        end
+      end
+    end
+  end
+end
+
+gateway.shutdown!

data/test/test_helper.rb

@@ -0,0 +1,10 @@
+require 'rubygems'
+require 'test/unit'
+require 'shoulda'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'vizir'
+
+class Test::Unit::TestCase
+end

data/test/vizir_test.rb

@@ -0,0 +1,7 @@
+require 'test_helper'
+
+class VizirTest < Test::Unit::TestCase
+  should "probably rename this file and start testing for real" do
+    flunk "hey buddy, you should probably rename this file and start testing for real"
+  end
+end

data/vizir.gemspec

@@ -0,0 +1,62 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{vizir}
+  s.version = "0.1.1"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Pierre Riteau"]
+  s.date = %q{2009-05-29}
+  s.default_executable = %q{vizir}
+  s.email = %q{pierre.riteau@gmail.com}
+  s.executables = ["vizir"]
+  s.extra_rdoc_files = [
+    "LICENSE",
+     "README.rdoc"
+  ]
+  s.files = [
+    ".document",
+     ".gitignore",
+     "LICENSE",
+     "README.rdoc",
+     "Rakefile",
+     "VERSION",
+     "bin/vizir",
+     "lib/vizir.rb",
+     "test/test_helper.rb",
+     "test/vizir_test.rb",
+     "vizir.gemspec"
+  ]
+  s.has_rdoc = true
+  s.homepage = %q{http://github.com/priteau/vizir}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.1}
+  s.summary = %q{Growl notifications for Grid'5000 jobs}
+  s.test_files = [
+    "test/test_helper.rb",
+     "test/vizir_test.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 2
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<json>, [">= 0"])
+      s.add_runtime_dependency(%q<net-ssh-gateway>, [">= 0"])
+      s.add_runtime_dependency(%q<rest-client>, [">= 0"])
+      s.add_runtime_dependency(%q<ruby-growl>, [">= 0"])
+    else
+      s.add_dependency(%q<json>, [">= 0"])
+      s.add_dependency(%q<net-ssh-gateway>, [">= 0"])
+      s.add_dependency(%q<rest-client>, [">= 0"])
+      s.add_dependency(%q<ruby-growl>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<json>, [">= 0"])
+    s.add_dependency(%q<net-ssh-gateway>, [">= 0"])
+    s.add_dependency(%q<rest-client>, [">= 0"])
+    s.add_dependency(%q<ruby-growl>, [">= 0"])
+  end
+end

metadata

@@ -0,0 +1,104 @@
+--- !ruby/object:Gem::Specification 
+name: priteau-vizir
+version: !ruby/object:Gem::Version 
+  version: 0.1.1
+platform: ruby
+authors: 
+- Pierre Riteau
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-05-29 00:00:00 -07:00
+default_executable: vizir
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: json
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: net-ssh-gateway
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: rest-client
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: ruby-growl
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+description: 
+email: pierre.riteau@gmail.com
+executables: 
+- vizir
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.rdoc
+files: 
+- .document
+- .gitignore
+- LICENSE
+- README.rdoc
+- Rakefile
+- VERSION
+- bin/vizir
+- lib/vizir.rb
+- test/test_helper.rb
+- test/vizir_test.rb
+- vizir.gemspec
+has_rdoc: true
+homepage: http://github.com/priteau/vizir
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.2.0
+signing_key: 
+specification_version: 2
+summary: Growl notifications for Grid'5000 jobs
+test_files: 
+- test/test_helper.rb
+- test/vizir_test.rb