lib/pronto/formatter/github_formatter.rb in pronto-0.2.1 vs lib/pronto/formatter/github_formatter.rb in pronto-0.2.2

@@ -1,19 +1,19 @@
 require 'octokit'
 
 module Pronto
   module Formatter
     class GithubFormatter
-      def format(messages)
+      def format(messages, repo)
         commit_messages = messages.map do |message|
-          repo = github_slug(message)
+          github_slug = repo.remotes.map(&:github_slug).compact.first
           sha = message.commit_sha
           position = message.line.commit_line.position if message.line
           path = message.path
           body = message.msg
 
-          create_comment(repo, sha, position, path, body)
+          create_comment(github_slug, sha, position, path, body)
         end
 
         "#{commit_messages.compact.count} Pronto messages posted to GitHub"
       end
 
@@ -37,13 +37,9 @@
         ENV['GITHUB_ACCESS_TOKEN']
       end
 
       def client
         @client ||= Octokit::Client.new(access_token: access_token)
-      end
-
-      def github_slug(message)
-        message.repo.remotes.map(&:github_slug).compact.first
       end
     end
   end
 end