stackoverflow 0.1.0

data/lib/stackoverflow.rb

@@ -0,0 +1,178 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'open-uri'
+require 'net/http'
+require 'json'
+require 'nokogiri'
+require 'terminal-table/import'
+require 'sqlite3'
+require 'optparse'
+
+
+module StackOverflow
+    class API
+        def search(search_string)
+            search_string = URI::encode(search_string)
+            api_get("/2.0/similar?order=desc&sort=votes&title=#{search_string}&site=stackoverflow&filter=!9Tk5iz1Gf")
+        end
+
+        def get_question(question_id)
+            api_get("/2.0/questions/#{question_id}?order=desc&sort=activity&site=stackoverflow&filter=!9Tk5izFWA")
+        end
+
+        def get_answers(question_id)
+            api_get("/2.0/questions/#{question_id}/answers?order=desc&sort=votes&site=stackoverflow&filter=!9Tk6JYC_e")
+        end
+
+        private
+
+        def api_get(path)
+            url = "https://api.stackexchange.com" + path
+            u = URI.parse(url)
+            Net::HTTP.start(u.host, u.port, :use_ssl => true) do |http|
+                response = http.get(u.request_uri)
+                return JSON(response.body)['items']
+            end
+        end
+    end
+
+    class DB
+        def initialize
+            @db = SQLite3::Database.new("../data/stackoverflow.db")
+        end
+
+        def search(search_string)
+            sql = "SELECT id, score, body, title FROM posts WHERE post_type_id=1 AND "
+            sub_sql = []
+            for search_term in search_string.split(' ')
+                sub_sql << "title LIKE '%#{search_term}%"
+            end
+            sql += sub_sql.join(' AND ')
+            sql += ' ORDER BY score DESC'
+
+            questions = []
+            @db.execute(sql) do |row|
+                questions << { :id => row[0],
+                               :score => row[1],
+                               :body => row[2],
+                               :title => row[3],
+                               :link => '',
+                               :answers => get_answers(row[0]) }
+            end
+        end
+
+        def get_answers(question_id)
+            answers = []
+            sql = "SELECT id, score, body FROM posts WHERE post_type_id=2 AND parent_id=#{question_id} ORDER BY score DESC"
+            @db.execute(sql) do |row|
+                answers << { :id => row[0],
+                             :score => row[1],
+                             :body => row[2] }
+            end
+        end
+    end
+
+    class Formatter
+        def questions_list(questions)
+            nb = 1
+
+            table = Terminal::Table.new do |t|
+                questions.each do |question|
+                    score = question['score'] > 0 ? "+#{question['score']}" : question['score']
+                    t << ["[#{nb}]", "(#{score})", question['title']]
+                    nb += 1
+                end
+                t.style = {:padding_left => 2, :border_x => " ", :border_i => " ", :border_y => " "}
+            end
+            puts table
+        end
+
+        def question_viewer(question)
+            answers = question['answers']
+            nb = 1
+            
+            table = Terminal::Table.new do |t|
+                t << ["Question", html2text(question['body'])]
+                t.title = question['title'] + "\n\n" + question['link']
+                t << :separator
+                t << :separator
+
+                answers.each do |answer|
+                    text = html2text(answer['body'])
+                    t << ["[#{nb}] (+#{answer['score']})", "#{text}"]
+                    t << :separator
+                    nb += 1
+                end
+            end
+
+            IO.popen("less", "w") { |f| f.puts table }
+        end
+
+        def html2text(html)
+            doc = Nokogiri::HTML(html)
+            doc = doc.css('body').text.squeeze(" ").squeeze("\n").gsub(/[\n]+/, "\n\n")
+            wordwrap(doc)
+        end
+
+        def wordwrap(str, columns=80)
+            str.gsub(/\t/, "    ").gsub(/.{1,#{ columns }}(?:\s|\Z)/) do
+                ($& + 5.chr).gsub(/\n\005/, "\n").gsub(/\005/, "\n")
+            end
+        end
+    end
+
+    class Command
+        def run
+            options = {}
+            OptionParser.new do |opts|
+                opts.banner = "Usage: so [options] <search string> [<question_id>]"
+                opts.on("-h", "--help", "Help") do |v|
+                    help(opts)
+                end
+                opts.on("-o", "--offline", "Offline mode") do |v|
+                    options['offline'] = true
+                end
+            end.parse!
+
+            # last argument is integer when user is specifing a question_nb from the results
+            question_nb = nil
+            if ARGV[-1] =~ /^[0-9]+$/
+                question_nb = ARGV.pop.to_i
+            end
+
+            search_string = ARGV.join(' ')
+
+            search(search_string, question_nb, options)
+        end
+
+        def help(opts)
+            puts opts.banner + "\n\n"
+            puts "\t<search_string>: Search Stack Overflow for a combination of words"
+            puts "\t<question_id>: (Optional) Display the question with this #id from the search results"
+        end
+
+        def search(search_string, question_nb, options)
+            if options['offline']
+                api = API.new
+                #api = DB.new
+            else
+                api = API.new
+            end
+            
+            questions = api.search(search_string)
+            if !questions or questions.length == 0
+                puts "No record found - Try a less specific (or sometimes, more specific) query"
+                return
+            end
+
+            if !question_nb
+                Formatter.new.questions_list(questions)
+            else
+                question = questions[question_nb.to_i - 1]
+                Formatter.new.question_viewer(question)
+            end
+        end
+    end
+end
+

metadata

@@ -0,0 +1,64 @@
+--- !ruby/object:Gem::Specification 
+name: stackoverflow
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Xavier Antoviaque
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-03-24 00:00:00 -04:00
+default_executable: 
+dependencies: []
+
+description: Query StackOverflow from the command line (offline/online modes)
+email: xavier@antoviaque.org
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/stackoverflow.rb
+has_rdoc: true
+homepage: https://github.com/antoviaque/stack-overflow-command-line
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: Query StackOverflow from the command line (offline/online modes)
+test_files: []
+