usaidwat 1.3.0 → 1.4.0

data/features/tally.feature

@@ -63,7 +63,8 @@ Feature: Tally comments
   Scenario: Search for a comment when tallying
     Given the Reddit service returns comments for the user "mipadi"
     When I run `usaidwat tally --grep='Heisenbug' mipadi`
-    Then it should fail with:
+    Then the exit status should not be 0
+    And stderr should contain:
       """
       invalid option: --grep=Heisenbug
       """
@@ -71,7 +72,8 @@ Feature: Tally comments
   Scenario: Search for a comment when sorting
     Given the Reddit service returns comments for the user "mipadi"
     When I run `usaidwat tally -c --grep='Heisenbug' mipadi`
-    Then it should fail with:
+    Then the exit status should not be 0
+    And stderr should contain:
       """
       invalid option: --grep=Heisenbug
       """
@@ -87,7 +89,8 @@ Feature: Tally comments
   Scenario: Tally comments with subreddit
     Given the Reddit service returns comments for the user "mipadi"
     When I run `usaidwat tally mipadi AskReddit`
-    Then it should fail with:
+    Then the exit status should not be 0
+    And stderr should contain exactly:
       """
       You cannot specify a subreddit when tallying comments
       """
@@ -95,7 +98,8 @@ Feature: Tally comments
   Scenario: Sort comments with subreddit
     Given the Reddit service returns comments for the user "mipadi"
     When I run `usaidwat tally -c mipadi AskReddit`
-    Then it should fail with:
+    Then the exit status should not be 0
+    And stderr should contain exactly:
       """
       You cannot specify a subreddit when tallying comments
       """
@@ -103,7 +107,8 @@ Feature: Tally comments
   Scenario: Pass no arguments when tallying
     Given the Reddit service returns comments for the user "mipadi"
     When I run `usaidwat tally`
-    Then it should fail with:
+    Then the exit status should not be 0
+    And stderr should contain exactly:
       """
       You must specify a username
       """
@@ -111,7 +116,8 @@ Feature: Tally comments
   Scenario: Pass no arguments when sorting
     Given the Reddit service returns comments for the user "mipadi"
     When I run `usaidwat tally -c`
-    Then it should fail with:
+    Then the exit status should not be 0
+    And stderr should contain exactly:
       """
       You must specify a username
       """

data/features/user.feature

@@ -27,7 +27,8 @@ Feature: Display user information
   Scenario: Fail to pass a username when querying for information
     Given the Reddit service returns information for the user "mipadi"
     When I run `usaidwat info`
-    Then it should fail with:
+    Then the exit status should not be 0
+    And stderr should contain exactly:
       """
       You must specify a username
       """

data/lib/usaidwat.rb

@@ -1,6 +1,5 @@
 require "usaidwat/algo"
 require "usaidwat/application"
-require "usaidwat/formatter"
 require "usaidwat/client"
-require "usaidwat/terminal"
+require "usaidwat/formatter"
 require "usaidwat/version"

data/lib/usaidwat/application.rb

@@ -1,8 +1,12 @@
-require 'usaidwat/algo'
+require 'sysexits'
 require 'usaidwat/client'
+require 'usaidwat/count'
 require 'usaidwat/either'
+require 'usaidwat/filter'
 require 'usaidwat/pager'
-require 'sysexits'
+require 'usaidwat/ext/array'
+
+require 'timecop' if ENV['USAIDWAT_ENV'] == 'cucumber'
 
 module USaidWat
   module Application
@@ -25,6 +29,7 @@ module USaidWat
 
       def initialize(prog)
         @client = cucumber? ? USaidWat::Client::TestRedditor : USaidWat::Client::Redditor
+        Timecop.freeze(Time.parse(ENV['USAIDWAT_CURRENT_TIME'])) if cucumber_time?
       end
 
       protected
@@ -33,8 +38,13 @@ module USaidWat
         ENV['USAIDWAT_ENV'] == 'cucumber'
       end
 
+      def cucumber_time?
+        cucumber? && !ENV['USAIDWAT_CURRENT_TIME'].nil?
+      end
+
       def quit(message, code=:ok)
-        puts message
+        stream = code == :ok ? $stdout : $stderr
+        stream.puts message
         exit code
       end
     end
@@ -64,6 +74,8 @@ module USaidWat
     end
 
     class Log < Command
+      include FilterCommand
+
       def initialize(prog)
         prog.command(:log) do |c|
           c.alias :l
@@ -83,15 +95,15 @@ module USaidWat
       def process(options, args)
         raise ArgumentError.new('You must specify a username') if args.empty?
         username = args.shift
-        subreddits = args.join(' ').split(/[ ,\+]/).map { |sr| sr.downcase }
+        subreddits = args.subreddits
 
         redditor = client.new(username)
         comments = redditor.comments
 
-        res = filter_comments(redditor, comments, subreddits) >>
-              lambda { |r| grep_comments(redditor, r.value, options['grep']) } >>
-              lambda { |r| limit_comments(redditor, r.value, options['limit']) } >>
-              lambda { |r| ensure_comments(redditor, r.value) }
+        res = filter_entries('comments', redditor, comments, subreddits) >>
+              lambda { |r| grep_entries('comments', redditor, r.value, options['grep']) } >>
+              lambda { |r| limit_entries('comments', redditor, r.value, options['limit']) } >>
+              lambda { |r| ensure_entries('comments', redditor, r.value) }
 
         quit res.value if res.left?
 
@@ -108,50 +120,83 @@ module USaidWat
 
       private
 
-      def filter_comments(redditor, comments, subreddits)
-        return USaidWat::Right.new(comments) if subreddits.empty?
-        comments = comments.find_all { |c| subreddits.include?(c.subreddit.downcase) }
-        if comments.empty?
-          USaidWat::Left.new("No comments by #{redditor.username} for #{subreddits.join(', ')}.")
-        else
-          USaidWat::Right.new(comments)
-        end
+      def list_comments(comments, options = {})
+        oneline = options[:oneline]
+        formatter = (oneline ? USaidWat::CLI::CompactCommentFormatter : USaidWat::CLI::CommentFormatter).new(options)
+        page
+        comments.each { |c| print formatter.format(c) }
       end
+    end
+
+    class Posts < Command
+      include CountCommand
+      include FilterCommand
 
-      def grep_comments(redditor, comments, grep)
-        return USaidWat::Right.new(comments) if grep.nil?
-        comments = comments.select { |c| c.body =~ /#{grep}/i }
-        if comments.empty?
-          msg = "#{redditor.username} has no comments matching /#{grep}/."
-          USaidWat::Left.new(msg)
-        else
-          USaidWat::Right.new(comments)
+      def initialize(prog)
+        prog.command(:posts) do |c|
+          c.action do |args, options|
+            process(options, args)
+          end
+
+          c.command(:log) do |s|
+            s.description "Show a user's submitted posts"
+            s.action do |args, options|
+              process_log(options, args)
+            end
+          end
+
+          c.command(:tally) do |s|
+            s.description "Tally a user's posts by subreddit"
+            s.option 'count', '-c', '--count', 'Sort output by number of comments'
+            s.action do |args, options|
+              process_tally(options, args)
+            end
+          end
         end
+        super
       end
 
-      def limit_comments(redditor, comments, n)
-        return USaidWat::Right.new(comments) if n.nil?
-        comments = comments[0...n.to_i]
-        USaidWat::Right.new(comments)
+      def process(options, args)
+        quit "Do you want to tally or log posts?", :usage
       end
 
-      def ensure_comments(redditor, comments)
-        if comments.empty?
-          USaidWat::Left.new("#{redditor.username} has no comments.")
-        else
-          USaidWat::Right.new(comments)
-        end
-      end
+      def process_log(options, args)
+        raise ArgumentError.new('You must specify a username') if args.empty?
+        username = args.shift
+        subreddits = args.subreddits
 
-      def list_comments(comments, options = {})
-        oneline = options[:oneline]
-        formatter = (oneline ? USaidWat::CLI::CompactCommentFormatter : USaidWat::CLI::CommentFormatter).new(options)
+        redditor = client.new(username)
+        posts = redditor.posts
+
+        res = filter_entries('posts', redditor, posts, subreddits) >>
+              lambda { |r| ensure_entries('posts', redditor, r.value) }
+
+        quit res.value if res.left?
+        posts = res.value
+
+        formatter = USaidWat::CLI::PostFormatter.new
         page
-        comments.each { |c| print formatter.format(c) }
+        posts.each { |p| print formatter.format(p) }
+      end
+
+      def process_tally(options, args)
+        raise ArgumentError.new('You must specify a username') if args.empty?
+        raise ArgumentError.new('You cannot specify a subreddit when tallying comments') if args.count > 1
+        username = args.first
+
+        redditor = client.new(username)
+        quit "#{redditor.username} has no posts." if redditor.posts.empty?
+        partition_data = partition(redditor.posts, options['count'])
+        formatter = USaidWat::CLI::TallyFormatter.new
+        print formatter.format(partition_data)
+      rescue USaidWat::Client::NoSuchUserError
+        quit "No such user: #{username}", :no_such_user
       end
     end
 
     class Tally < Command
+      include CountCommand
+
       def initialize(prog)
         prog.command(:tally) do |c|
           c.alias :t
@@ -171,31 +216,12 @@ module USaidWat
 
         redditor = client.new(username)
         quit "#{redditor.username} has no comments." if redditor.comments.empty?
-        # Unfortunately Snooby cannot return comments for a specific
-        # user in a specific subreddit, so for now we have to sort them
-        # ourself.
-        longest_subreddit = 0
-        buckets = Hash.new { |hash, key| hash[key] = 0 }
-        redditor.comments.each do |comment|
-          subreddit = comment.subreddit
-          longest_subreddit = subreddit.length if subreddit.length > longest_subreddit
-          buckets[subreddit] += 1
-        end
-        algo = algorithm(options['count']).new(buckets)
-        subreddits = buckets.keys.sort { |a,b| algo.sort(a, b) }
-        subreddits.each do |subreddit|
-          tally = buckets[subreddit]
-          printf "%-*s  %3d\n", longest_subreddit, subreddit, tally
-        end
+        partition_data = partition(redditor.comments, options['count'])
+        formatter = USaidWat::CLI::TallyFormatter.new
+        print formatter.format(partition_data)
       rescue USaidWat::Client::NoSuchUserError
         quit "No such user: #{username}", :no_such_user
       end
-
-      private
-
-      def algorithm(count)
-        count ? USaidWat::Algorithms::CountAlgorithm : USaidWat::Algorithms::LexicographicalAlgorithm
-      end
     end
   end
 end

data/lib/usaidwat/client.rb

@@ -43,6 +43,14 @@ module USaidWat
         "#{username}"
       end
 
+      def posts
+        user.posts
+      rescue NoMethodError
+        raise NoSuchUserError, username
+      rescue TypeError
+        raise ReachabilityError, "Reddit unreachable"
+      end
+
       private
 
       def user

data/lib/usaidwat/count.rb

@@ -0,0 +1,27 @@
+require 'usaidwat/algo'
+
+module USaidWat
+  module Application
+    module CountCommand
+      def partition(entries, sort_by_count)
+        longest_subreddit = 0
+        buckets = Hash.new { |hash, key| hash[key] = 0 }
+        entries.each do |e|
+          subreddit = e.subreddit
+          longest_subreddit = subreddit.length if subreddit.length > longest_subreddit
+          buckets[subreddit] += 1
+        end
+        algo = algorithm(sort_by_count).new(buckets)
+        subreddits = buckets.keys.sort { |a,b| algo.sort(a, b) }
+        counts = subreddits.map { |s| buckets[s] }
+        subreddit_counts = subreddits.zip(counts)
+        partition_data = Struct.new(:longest, :counts)
+        partition_data.new(longest_subreddit, subreddit_counts)
+      end
+
+      def algorithm(sort_by_count)
+        sort_by_count ? USaidWat::Algorithms::CountAlgorithm : USaidWat::Algorithms::LexicographicalAlgorithm
+      end
+    end
+  end
+end

data/lib/usaidwat/ext/array.rb

@@ -0,0 +1,5 @@
+class Array
+  def subreddits
+    join(' ').split(/[ ,\+]/).map { |sr| sr.downcase }
+  end
+end

data/lib/usaidwat/ext/mercenary.rb

@@ -0,0 +1,16 @@
+module Mercenary
+  class Command
+    def logger(level = nil)
+      unless @logger
+        @logger = Logger.new(STDERR)
+        @logger.level = level || Logger::INFO
+        @logger.formatter = proc do |severity, datetime, progname, msg|
+          "#{msg}\n"
+        end
+      end
+
+      @logger.level = level unless level.nil?
+      @logger
+    end
+  end
+end

data/lib/usaidwat/filter.rb

@@ -0,0 +1,40 @@
+module USaidWat
+  module Application
+    module FilterCommand
+      def filter_entries(noun, redditor, entries, subreddits)
+        return USaidWat::Right.new(entries) if subreddits.empty?
+        entries = entries.find_all { |e| subreddits.include?(e.subreddit.downcase) }
+        if entries.empty?
+          USaidWat::Left.new("No #{noun} by #{redditor.username} for #{subreddits.join(', ')}.")
+        else
+          USaidWat::Right.new(entries)
+        end
+      end
+
+      def grep_entries(noun, redditor, entries, grep)
+        return USaidWat::Right.new(entries) if grep.nil?
+        entries = entries.select { |e| e.body =~ /#{grep}/i }
+        if entries.empty?
+          msg = "#{redditor.username} has no #{noun} matching /#{grep}/."
+          USaidWat::Left.new(msg)
+        else
+          USaidWat::Right.new(entries)
+        end
+      end
+
+      def limit_entries(noun, redditor, entries, n)
+        return USaidWat::Right.new(entries) if n.nil?
+        entries = entries[0...n.to_i]
+        USaidWat::Right.new(entries)
+      end
+
+      def ensure_entries(noun, redditor, entries)
+        if entries.empty?
+          USaidWat::Left.new("#{redditor.username} has no #{noun}.")
+        else
+          USaidWat::Right.new(entries)
+        end
+      end
+    end
+  end
+end

data/lib/usaidwat/formatter.rb

@@ -4,6 +4,7 @@ require 'rainbow/ext/string'
 require 'redcarpet'
 require 'set'
 require 'stringio'
+require 'ttycaca'
 require 'usaidwat/ext/string'
 require 'usaidwat/ext/time'
 
@@ -36,7 +37,32 @@ module USaidWat
       protected
 
       def tty
-        @tty ||= USaidWat::Application::Terminal.new
+        @tty ||= Ttycaca::Terminal.new
+      end
+    end
+
+    class PostFormatter < BaseFormatter
+      def format(post)
+        cols = tty.width
+        out = StringIO.new
+        out.write("\n\n\n") unless @count == 0
+        out.write("#{post.subreddit}\n".color(:green))
+        out.write("#{post_link(post)}\n".color(:yellow))
+        out.write("#{post.title.strip.truncate(cols)}\n".color(:magenta))
+        out.write("#{post_date(post)}".color(:blue))
+        @count += 1
+        out.rewind
+        out.read
+      end
+
+      private
+
+      def post_link(post)
+        "https://www.reddit.com#{post.permalink.split('/')[0..-2].join('/')}"
+      end
+
+      def post_date(post)
+        Time.at(post.created_utc).ago
       end
     end
 
@@ -127,5 +153,20 @@ module USaidWat
         comments.include?(comment)
       end
     end
+
+    class TallyFormatter < BaseFormatter
+      def format(partition_data)
+        out = StringIO.new
+        longest_subreddit = partition_data.longest
+        subreddits = partition_data.counts
+        subreddits.each do |subreddit_count|
+          subreddit, tally = subreddit_count
+          line = sprintf("%-*s  %3d\n", longest_subreddit, subreddit, tally)
+          out.write(line)
+        end
+        out.rewind
+        out.read
+      end
+    end
   end
 end