geeklets 0.0.4

data/History.txt

@@ -0,0 +1,8 @@
+=== 0.0.2 2010-02-07
+
+* Added weather script from Ted Wise.
+
+=== 0.0.1 2010-02-13
+
+* 1 major enhancement:
+  * Initial release

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 John F. Schank III
+
+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,57 @@
+= geeklets
+
+Geeklets is a collection of scripts for use with GeekTool.
+Here is the breakdown of the scripts
+
+== Note about these scripts.
+
+At the moment - they are all focused on my particular needs, so
+they are currently hardcoded for my area of Virginia, USA. As this
+project develops I will be trying to find a way to refactor that depency
+away, and allow individual users to customize for their needs.
+
+Just be aware, that right now, unless you live in the Washington Metro area,
+some of these scripts won't be useful for you.
+
+Run them with a -h parameter for details.
+
+== get_weather_icon
+
+Gets the current weather icon from yahoo, and saves it to the tmp directory.
+
+== next_months_calendar
+
+Runs the unix "cal" command for whatever the next month from now is.
+
+== OPM_alerts
+
+Gets the Office of Personnel Management alert for the current day.
+This if for things like Federal Government closings and delays due to weather.
+
+== trash_usage
+
+A wrapper for the unix "du" command to display the amount of space taken up by
+the traash on an OS X mac.
+
+== VRE_alerts
+
+Shows the alerts for today for the Virginia Rail Express
+
+== WMATA_alerts
+
+Shows the alerts and delays for the Washington Metro Area Transit Authority,
+the "Metro" subway and trains.
+
+== Note on Patches/Pull Requests
+ 
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+== Copyright
+
+Copyright (c) 2010 John F. Schank III. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,46 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "geeklets"
+    gem.summary = %Q{Scripts for GeekTool}
+    gem.description = %Q{A collection of useful scripts for use with GeekTool}
+    gem.email = "jschank@mac.com"
+    gem.homepage = "http://github.com/jschank/geeklets"
+    gem.authors = ["John F. Schank III"]
+    gem.add_development_dependency "rspec", ">= 1.2.9"
+    gem.files = FileList['lib/**/*', 'bin/*', '[A-Z]*', 'spec/**/*', 'vendor/**/*'].to_a
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: gem install jeweler"
+end
+
+require 'spec/rake/spectask'
+Spec::Rake::SpecTask.new(:spec) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.spec_files = FileList['spec/**/*_spec.rb']
+end
+
+Spec::Rake::SpecTask.new(:rcov) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.pattern = 'spec/**/*_spec.rb'
+  spec.rcov = true
+end
+
+task :spec => :check_dependencies
+
+task :default => :spec
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  version = File.exist?('VERSION') ? File.read('VERSION') : ""
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "geeklets #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/VERSION

@@ -0,0 +1 @@
+0.0.4

data/bin/geeklets

@@ -0,0 +1,8 @@
+#!/usr/bin/env ruby
+
+$LOAD_PATH.unshift File.dirname(__FILE__) + '/../lib'
+$LOAD_PATH.unshift File.dirname(__FILE__) + '/../vendor'
+
+require 'geeklets'
+
+Geeklets.run(ARGV)

data/lib/Get_Weather_Icon/get_weather_icon.rb

@@ -0,0 +1,17 @@
+require 'rubygems'
+require 'mechanize'
+require 'rio'
+
+class Get_Weather_Icon
+
+  def run(params)
+    agent = Mechanize.new
+    agent.get("http://weather.yahoo.com/forecast/USVA0262.html")
+    icon = agent.page.at(".forecast-icon")
+    style = icon.attributes["style"].value
+    icon_url = style.split("'")[1]
+
+    rio(icon_url) > rio("/tmp/weather-icon.png")
+  end
+
+end

data/lib/Next_Months_Calendar/next_months_calendar.rb

@@ -0,0 +1,9 @@
+class Next_Months_Calendar
+
+  def run(params)
+    today = Date.today
+    nextmonth = today >> 1
+    system("cal","#{nextmonth.month}","#{nextmonth.year}")
+  end
+
+end

data/lib/OPM_Alerts/opm_alerts.rb

@@ -0,0 +1,61 @@
+# coding: UTF-8
+require 'rubygems'
+require 'open-uri'
+require 'nokogiri'
+
+URL = "http://www.opm.gov/status/index.aspx"
+
+class OPM_Alerts
+
+  def die
+    puts <<-EOS
+  opm-status
+
+  USAGE:
+
+  \topm-status [wrapping-width]
+
+  Returns U.S. Office of Personnel Management status Alerts.
+
+  [wrapping-width] is an integer to limit the width of the descriptions
+  Defaults to: 40
+
+EOS
+
+    exit
+  end
+
+  def run(params)
+    die if params[0] == "-h"
+
+    width = (params[1] and params[1].to_i) || 40
+
+    doc = Nokogiri::HTML(open(URL))
+
+    date_str = doc.css('#_ctl0__ctl0_DisplayDateSpan').text.strip
+    begin
+      date = Date.parse(date_str)
+      date_str = date.strftime("%x")
+    rescue
+      date_str = Date.today.strftime("%x") if date_str.length == 0
+    end
+
+    title = doc.css('h3').text.strip
+    title = "VRE Status" if title.length == 0
+
+    status = doc.css('.statusbox').text.strip
+
+    # Deal with the flaky case where OPM decides to
+    # display an image instead of actual text content
+    images = doc.xpath("//img")
+    status = images[0].attributes["alt"].text.strip if ((status.length == 0) && (images.count == 1))
+
+    status = "Not found" if status.length == 0
+
+    # display results
+    puts Utility.wrap_text("#{date_str} - #{title}", width, date_str.length + 3, :outdent)
+    puts "-" * width
+    puts Utility.wrap_text(status, width)
+  end
+
+end

data/lib/Trash_Usage/trash_usage.rb

@@ -0,0 +1,8 @@
+class Trash_Usage
+
+  def run(params)
+    system("du -sh ~/.Trash/ | awk '{print \"Trash is using \"$1}'")    
+  end
+
+end
+

data/lib/VRE_Alerts/vre_alerts.rb

@@ -0,0 +1,123 @@
+# coding: UTF-8
+require 'rubygems'
+require 'chronic'
+require 'mechanize'
+
+FEED_URL = "http://www.vre.org/service/daily-download.html"
+TRAIN_LINES = {"fbg_delay" => :fredericksburg, "mss_delay" => :manassas}
+
+class VRE_Alerts
+
+  def die
+    puts <<-EOS
+    vre-alert
+
+    USAGE:
+
+    \tvre-alert [rail_lines] [wrapping-width] [specific-date]
+
+    Returns Virgina Railway Express (VRE) Alerts.
+
+    [rail_lines] specifies which rail lines you want alerts for.
+    Can be:
+    \tfredericsburg - for the Fredericsburg line
+    \tmanassas      - for the Manassas line
+    \tboth          - for both lines.
+    Defaults to: both
+  
+    [wrapping-width] is an integer to limit the width of the descriptions
+    Defaults to: 40
+  
+    [specific-date] is a string parseable by Date::parse which is the date from
+    the VRE daily download to display. NOTE: this does not access history, it only
+    allows you to choose a specific entry from the Daily Download to render. Mostly
+    used for debugging.
+    Defaults to: The current system Date.
+
+  EOS
+
+    exit
+  end
+
+  # >> notice.search("tr")[2].search("td")[0].attributes["class"].value
+  # => "fbg2"
+  # >> notice.search("tr")[2].search("td")[0].text
+  # => "Train 300"
+  def parse_notice(notice)
+    detail = {}
+    detail[:summary] = notice.attributes["summary"].text
+    detail[:morning] = []
+    detail[:evening] = []
+    rows = notice.search("tr")
+    service = nil
+
+    rows.each do |row|
+      cells = row.search("td")
+      case cells.count
+      when 0 : 
+        (detail[:date] = Date.parse(row.children[0].text)) if (row && row.children && row.children[0] && row.children[0].name == "th")
+      when 1 :
+        service = :morning if row.children[0].text =~ /Morning/
+        service = :evening if row.children[0].text =~ /Evening/
+      when 2 :
+        # sometimes they split a description across multiple rows, when they do, they leave the train cell blank
+        # in that case, append the description text to the last trains description.
+        if cells[0].text.empty? && detail[service][-1]
+          detail[service][-1][:description] << cells[1].text
+          next 
+        end
+      
+        cell_class = cells[0].attributes["class"].value unless cells[0].text.empty?
+        detail[service] << {:line => TRAIN_LINES[cell_class], :train => cells[0].text, :description => cells[1].text} if @rail_lines == :both || @rail_lines == TRAIN_LINES[cell_class]
+      else
+        next
+      end
+    end
+    detail
+  end
+
+  def render_trains(trains, width)
+    puts "  No delays" and return if trains.empty?
+    trains.each do |train|
+      puts "  #{train[:line].to_s.capitalize} line #{train[:train]}"
+      Utility.wrap_text(train[:description], width, 3, :all)
+    end
+  end
+
+  def render_notice(notice, width)
+    if (notice.nil? || (notice[:morning].empty? && notice[:evening].empty?))
+      puts "No VRE notices for today"
+      return 
+    end
+
+    puts notice[:summary]
+    puts "#{'-' * notice[:summary].length}"
+    puts "Morning Service"
+    render_trains(notice[:morning], width)
+    puts
+    puts "Afternoon/Evening Service"
+    render_trains(notice[:evening], width)
+  end
+
+  def run(params)
+    die if params[0] == "-h"
+
+    @rail_lines = (params[0] || "both").to_sym
+    width = (params[1] and params[1].to_i) || 40
+    alert_date = (params[2] and Date.parse(params[2])) || Date.today
+
+    #debug - set a date of interest, and rail line of interest
+    # alert_date = Date.civil(2010, 1, 13)
+    # @rail_lines = :fredericksburg
+
+    agent = Mechanize.new
+    agent.get(FEED_URL)
+    notices = agent.page.search(".format").map{ |notice| parse_notice(notice) }
+
+    #debug
+    #puts notices.to_yaml
+
+    render_notice(notices.select{ |notice| notice[:date] == alert_date }.shift, width)
+  end
+
+end

data/lib/WMATA_Alerts/wmata_alerts.rb

@@ -0,0 +1,96 @@
+# coding: UTF-8
+require 'rubygems'
+require 'rss'
+require 'open-uri'
+require 'chronic'
+require 'htmlentities'
+
+REPLACEMENTS = 
+[
+  ["p.m.", "pm"], 
+  ["a.m.", "am"]
+]
+
+FEED_URL = "http://www.wmata.com/rider_tools/metro_service_status/feeds/rail.xml"
+
+class WMATA_Alerts
+
+  def die
+    puts <<-EOS
+  metro-feed
+
+    USAGE:
+
+    \tmetro-feed [history] [wrapping-width]
+
+    Returns WMATA feed information for metro lines.
+
+    [history] is how many days of feed history to display.
+    \tdefaults to 7 days. Can be anything parseable by Chronic gem
+
+    [wrapping-width] is the limit to how wide the text can be
+    \tdefaults to 40 characters. should be a positive integer value.
+
+    Requires gems: chronic, htmlentities
+
+
+  EOS
+
+    exit
+  end
+
+  def run(params)
+
+    die if params[0] == "-h"
+
+    @cutoff_string = params[0] || "7 days ago"
+    @width = params[1] || "40"
+    @width = @width.to_s.to_i
+
+    @cutoff = Chronic.parse(@cutoff_string)
+
+    @rss_content = ""
+
+    open(FEED_URL) do |f|
+      @rss_content = f.read
+    end
+
+    @rss = RSS::Parser.parse(@rss_content, false)
+
+    @item_hash = {}
+
+    # filter down to the items we want.
+    @rss.items.each do |item|
+      @item_hash[item.pubDate] = item unless item.pubDate < @cutoff
+    end
+
+    if @item_hash.empty?
+      puts "No data from WMATA feed."
+    else
+      @coder = HTMLEntities.new
+      @item_hash.keys.sort.reverse.each do |key|
+        item = @item_hash[key]
+
+        title = "#{item.pubDate.strftime('%x')} - #{item.title}"
+        puts title
+        puts ("-" * title.length)
+
+        message = @coder.decode(item.description)
+        REPLACEMENTS.each do |r|
+          message = message.gsub(r[0], r[1])
+        end
+        subject, *body = message.split(".")
+        
+        puts Utility.wrap_text(subject+"...", @width)
+        body.each do |sentence|
+          puts Utility.wrap_text(sentence+".", @width, 3, :all)
+        end  
+
+        puts
+      end
+    end
+  
+  end
+
+
+end