Merge pull request #7 from brodock/feature/templates

Add message templates for slack and default

.rspec

@@ -0,0 +1 @@
+--color --profile 5

lib/lita-dig.rb

@@ -5,5 +5,8 @@
 )]
 
 require 'net/dns'
-
 require 'lita/handlers/dig'
+
+Lita::Handlers::Dig.template_root File.expand_path(
+  File.join('..', '..', 'templates'), __FILE__
+)

lib/lita/handlers/dig.rb

@@ -49,15 +49,11 @@ def lookup(argument, type, server = nil)
       end
 
       def format_lookup(lookup, compact = false)
-        result = ''
         if compact
-          lookup.each_address do |ip|
-            result += "#{ip}\n"
-          end
+          render_template('compact', lookup: lookup)
         else
-          result = lookup.to_s
+          render_template('full', lookup: lookup)
         end
-        result
       end
     end
 

spec/lita/handlers/dig_spec.rb

@@ -31,7 +31,9 @@
 
   let(:resolve_short) do
     list = double
-    allow(list).to receive(:each_address).and_yield('1.2.3.4')
+    allow(list).to receive(:each_address)
+      .and_yield('1.2.3.4')
+      .and_yield('5.6.7.8')
     client = double
     allow(client).to receive(:nameservers=) { '' }
     expect(client).to receive(:query) { list }
@@ -58,7 +60,7 @@
     it 'shows a short record if the domain exists' do
       expect(Net::DNS::Resolver).to receive(:new) { resolve_short }
       send_command('dig example.com +short')
-      expect(replies.last).to eq("1.2.3.4\n")
+      expect(replies.last).to eq("1.2.3.4\n5.6.7.8\n")
     end
 
     it 'shows a warning if the domain does not exist' do

templates/compact.erb

@@ -0,0 +1,3 @@
+<% @lookup.each_address do |ip| %>
+<%= "#{ip}\n" %>
+<% end %>

templates/compact.slack.erb

@@ -0,0 +1,3 @@
+<% @lookup.each_address do |ip| %>
+  `<%= ip %>`
+<% end %>

templates/full.erb

@@ -0,0 +1 @@
+<%= @lookup %>

templates/full.slack.erb

@@ -0,0 +1 @@
+```<%= @lookup %>```