usaidwat 0.0.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

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

data/Rakefile

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

data/bin/usaidwat

@@ -0,0 +1,32 @@
+#!/usr/bin/env ruby
+
+require 'usaidwat'
+
+
+def die(code=1)
+  $stderr.puts "Usage: #{File.basename $0} <user> [<subreddit>]"
+  exit code
+end
+
+die if $*.length < 1 or $*.length > 2
+
+subreddit = $*.length == 2 ? $*.last : nil
+reddit_user = USaidWat::RedditUser.new $*.first
+unless subreddit
+  comments = reddit_user.retrieve_comments
+  exit 2 unless comments
+  max_key = 1
+  puts reddit_user.username
+  puts '-' * reddit_user.username.length
+  comments.each { |c| max_key = c.first.length if c.first.length > max_key }
+  comments.each { |c| printf "%-*s  %s\n", max_key, c.first, c.last }
+else
+  comments = reddit_user.comments_for_subreddit subreddit
+  is_first = true
+  width = `tput cols`.to_i rescue 80
+  comments.each do |c|
+    puts '-' * width unless is_first
+    puts c
+    is_first = false
+  end
+end

data/lib/usaidwat/reddit.rb

@@ -0,0 +1,150 @@
+require 'fileutils'
+require 'net/http'
+require 'time'
+
+require 'rubygems'
+require 'json'
+
+
+class Hash
+  def to_query
+    data = self.keys.inject([]) { |parts, key| parts << [URI.encode(key.to_s), URI.encode(self[key].to_s)] }
+    data.map{ |t| t.join '=' }.join '&'
+  end
+end
+
+module USaidWat
+  class RedditUser
+    COMMENT_COUNT = 100
+    COMMENTS_PER_PAGE = 25
+    
+    attr_reader :username
+    
+    def initialize(username)
+      @username = username
+      self.ensure_cache_dir!
+    end
+    
+    def profile_url
+      "http://www.reddit.com/user/#{self.username}"
+    end
+    
+    def comments_url
+      "#{self.profile_url}/comments"
+    end
+    
+    def last_update
+      return Time.at 0 unless File.exists? self.cache_timestamp_file
+      Time.parse File.open(self.cache_timestamp_file).read.chomp
+    end
+    
+    def retrieve_comments(options={})
+      return self.retrieve_comments_from_cache unless Time.now - self.last_update > 300
+      count = options[:count] || COMMENT_COUNT
+      cache = options[:cache] || true
+      self.destroy_cache!
+      comments = self.fetch_comments count
+      return nil unless comments
+      self.cache_comments comments if cache
+      subreddits = Hash.new { |h,k| h[k] = 0 }
+      comments.each do |comment|
+        subreddit = comment['data']['subreddit'].to_sym
+        subreddits[subreddit] += 1
+      end
+      subreddits.sort { |a,b| b[1] <=> a[1] }
+    end
+    
+    def comments_for_subreddit(subreddit)
+      self.retrieve_comments
+      comments = []
+      subreddit_dir = File.join self.comments_dir, subreddit
+      Dir.chdir(subreddit_dir) do
+        Dir['*'].each do |f|
+          path = File.join subreddit_dir, f
+          comment = File.open(path).read.chomp
+          comments << comment
+        end
+      end
+      comments
+    end
+    
+    def retrieve_comments_from_cache
+      subreddits = Hash.new { |h,k| h[k] = 0 }
+      Dir.chdir(self.comments_dir) do
+        Dir['*'].each do |sr|
+          next unless File.directory? sr
+          Dir["#{sr}/*"].each { |f| subreddits[sr] += 1 }
+        end
+      end
+      subreddits.sort { |a,b| b[1] <=> a[1] }
+    end
+    
+    def fetch_comments(count)
+      comments = Array.new
+      after = nil
+      last_page = count / COMMENTS_PER_PAGE
+      (1..last_page).each do |i|
+        query = i == 1 ? '' : {:count => COMMENTS_PER_PAGE, :after => after}.to_query
+        url = "#{self.comments_url}.json?#{query}"
+        resp = Net::HTTP.get_response 'www.reddit.com', url
+        unless resp.code.to_i == 200
+          $stderr.puts "Could not retrieve comments: #{resp.message}"
+          return nil
+        end
+        resp = JSON.parse resp.body
+        if resp.key? 'error'
+          $stderr.puts "Could not retrieve comments: #{resp['error'] || 'Unknown error'}"
+          return nil
+        end
+        comments += resp['data']['children']
+        after = resp['data']['after']
+      end
+      comments
+    end
+    
+    def cache_comments(comments)
+      comments.each do |c|
+        cid = c['data']['id']
+        sr = c['data']['subreddit']
+        body = c['data']['body']
+        parent_cache_dir = self.subreddit_directory! sr
+        cache_file = File.join parent_cache_dir, cid
+        File.open(cache_file, 'w') { |f| f.write body }
+      end
+      File.open(self.cache_timestamp_file, 'w') { |f| f.write Time.now }
+    end
+    
+    def cache_dir
+      File.join USaidWat::BASE_CACHE_DIR, self.username
+    end
+    
+    def cache_timestamp_file
+      File.join self.cache_dir, 'updated'
+    end
+    
+    def comments_dir
+      File.join self.cache_dir, 'comments'
+    end
+    
+    def subreddit_directory!(subreddit)
+      Dir.mkdir self.comments_dir unless File.exists? self.comments_dir
+      cache = File.join self.comments_dir, subreddit
+      Dir.mkdir cache unless File.exists? cache
+      cache
+    end
+    
+    def ensure_cache_dir!
+      Dir.mkdir USaidWat::BASE_CACHE_DIR unless File.exists? USaidWat::BASE_CACHE_DIR
+      Dir.mkdir self.cache_dir unless File.exists? self.cache_dir
+    end
+    
+    def destroy_cache!
+      FileUtils.rm_rf self.cache_dir
+      self.ensure_cache_dir!
+    end
+    
+    def to_s
+      "Reddit user: #{self.username}"
+    end
+  end
+end

data/lib/usaidwat/version.rb

@@ -0,0 +1,3 @@
+module USaidWat
+  VERSION = "0.0.1"
+end

data/lib/usaidwat.rb

@@ -0,0 +1,7 @@
+require 'usaidwat/version'
+require 'usaidwat/reddit'
+
+
+module USaidWat
+  BASE_CACHE_DIR = File.join File.expand_path('~'), '.usaidwat'
+end

data/usaidwat.gemspec

@@ -0,0 +1,22 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "usaidwat/version"
+
+Gem::Specification.new do |s|
+  s.name        = "usaidwat"
+  s.version     = USaidWat::VERSION
+  s.authors     = ["Michael Dippery"]
+  s.email       = ["michael@monkey-robot.com"]
+  s.homepage    = ""
+  s.summary     = %q{View a user's last 100 Reddit comments}
+  s.description = %q{View a user's last 100 Reddit comments, organized by subreddit.}
+
+  s.rubyforge_project = "usaidwat"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+  
+  s.add_runtime_dependency 'json'
+end

metadata

@@ -0,0 +1,65 @@
+--- !ruby/object:Gem::Specification
+name: usaidwat
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Michael Dippery
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-01-24 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: json
+  requirement: &70171046145340 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70171046145340
+description: View a user's last 100 Reddit comments, organized by subreddit.
+email:
+- michael@monkey-robot.com
+executables:
+- usaidwat
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Rakefile
+- bin/usaidwat
+- lib/usaidwat.rb
+- lib/usaidwat/reddit.rb
+- lib/usaidwat/version.rb
+- usaidwat.gemspec
+homepage: ''
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: usaidwat
+rubygems_version: 1.8.11
+signing_key: 
+specification_version: 3
+summary: View a user's last 100 Reddit comments
+test_files: []