Class: Lita::Handlers::OnewheelWolframAlpha

Inherits:
Handler
  • Object
show all
Defined in:
lib/lita/handlers/onewheel_wolfram_alpha.rb

Instance Method Summary collapse

Instance Method Details

#build_uri(query) ⇒ Object 



78
79
80
81
 
# File 'lib/lita/handlers/onewheel_wolfram_alpha.rb', line 78

def build_uri(query)
  uri = config.api_uri.sub '[query]', CGI::escape(query)
  uri.sub '[appid]', config.app_id
end

#handle_wolfram_query(response) ⇒ Object 



12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
 
# File 'lib/lita/handlers/onewheel_wolfram_alpha.rb', line 12

def handle_wolfram_query(response)
  unless config.app_id and config.api_uri
    Lita.logger.error 'lita-onewheel-wolfram-alpha: Configuration error!'
    return
  end
  query = response.matches[0][0]

  post_script = ''

  if matches = query.match(/\<(.*)\>/)
    Lita.logger.debug "Megamatch: #{matches[1]}"
    query.gsub! /\<.*\>/, ''
    post_script = " #{matches[1]}"
  end

  # fraction hack
  if query.match /\// and !query.match /\./
    query.sub! /\s+$/, ''
    query += ".0"
  end

  api_response = make_api_call query
  reply = parse_response api_response, query
  reply += post_script
  Lita.logger.debug "lita-onewheel-wolfram-alpha: Replying with #{reply}"
  response.reply reply
end

#make_api_call(query) ⇒ Object 



69
70
71
72
73
74
75
76
 
# File 'lib/lita/handlers/onewheel_wolfram_alpha.rb', line 69

def make_api_call(query)
  Lita.logger.debug "lita-onewheel-wolfram-alpha: Making api call for #{query}"
  uri = build_uri query
  Lita.logger.debug "lita-onewheel-wolfram-alpha: #{uri}"
  response = RestClient.get(uri)
  #Lita.logger.debug 'lita-onewheel-wolfram-alpha: ' + response.to_s
  Nokogiri::XML response.to_s
end

#parse_response(noko_doc, query) ⇒ Object 



40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
 
# File 'lib/lita/handlers/onewheel_wolfram_alpha.rb', line 40

def parse_response(noko_doc, query)
  success_node = noko_doc.xpath('queryresult').attribute('success')
  Lita.logger.debug "lita-onewheel-wolfram-alpha: Success attr: #{success_node.to_s}"

  # No sense parsing if we didn't have success.
  if success_node.to_s == 'true'

    pods = noko_doc.xpath('//pod')
    Lita.logger.debug "lita-onewheel-wolfram-alpha: Pod title: #{pods[1].attribute('title').to_s}"

    title = pods[1].attribute('title').to_s
    if title == 'Plots' or title == 'Plot'  # Plot is a graph, grab the image.
      pods[1].xpath('//img')[1].attribute('src').to_s
    else  # Plaintext seems to work well for, say, Definition.
      rid_thee_of_extras pods[1].xpath('//plaintext')[1].child.to_s
    end

  else
    ["Nope, no #{query} to see here.",
     "#{query}?",
     'What\'s that, now?'
     ].sample
  end
end

#rid_thee_of_extras(str) ⇒ Object 



65
66
67
 
# File 'lib/lita/handlers/onewheel_wolfram_alpha.rb', line 65

def rid_thee_of_extras(str)
  str.gsub /\s+\|\s/, ' | '
end