hsbt/agig

require 'ostruct'
require 'time'
require 'net/irc'
require 'string-irc'
require 'octokit'

class Agig::Session < Net::IRC::Server::Session
  def server_name
    "github"
  end

  def server_version
    Agig::VERSION
  end

  def channels
    ['#notification', '#watch']
  end

  def initialize(*args)
    super
    @notification_last_retrieved = @watch_last_retrieved = Time.now.utc - 3600
  end

  def client
    @client ||= Octokit::Client.new(access_token: @pass)
  end

  def on_disconnected
    @retrieve_thread.kill rescue nil
  end

  def on_user(m)
    super

    channels.each{|channel| post @nick, JOIN, channel }

    @retrieve_thread = Thread.start do
      loop do
        retrieve @opts.interval
      end
    end
  end

  private

  def retrieve(interval)
    @log.info 'retrieveing feed...'

    entries = client.notifications(all: true)
    entries.sort_by(&:updated_at).each do |entry|
      updated_at = Time.parse(entry.updated_at.to_s).utc
      next if updated_at <= @notification_last_retrieved

      title = StringIrc.new(entry.subject.title).brown.to_s
      url = StringIrc.new(reachable_url_for(entry.subject.latest_comment_url)).grey.to_s

      post entry.repository.owner.login, PRIVMSG, "#notification", "#{title} #{url}"
      @notification_last_retrieved = updated_at
    end

    events = client.received_events(@nick)
    events.sort_by(&:created_at).each do |event|
      next if event.type != "WatchEvent"

      created_at = Time.parse(event.created_at.to_s).utc
      next if created_at <= @watch_last_retrieved

      action = StringIrc.new(event.payload.action).brown.to_s
      url = StringIrc.new("http://github.com/#{event.repo.name}").grey.to_s

      post event.actor.login, PRIVMSG, "#watch", "#{action} #{url}"
      @watch_last_retrieved = created_at
    end

    @log.info 'sleep'
    sleep interval
  rescue Exception => e
    @log.error e.inspect
    e.backtrace.each do |l|
      @log.error "\t#{l}"
    end
    sleep 10
  end

  def reachable_url_for(latest_comment_url)
    repos_owner = latest_comment_url.match(/repos\/(.+?\/.+?)\//)[1]
    if issue_match = latest_comment_url.match(/(?:issues|pulls)\/(\d+?)$/)
      issue_id = issue_match[1]
      latest_comment = client.issue_comments(repos_owner, issue_id).last
      latest_comment ?
        latest_comment['html_url'] :
        latest_comment_url.sub(/api\./, '').sub(/repos\//, '').sub(/pulls\//, 'pull/')
    elsif comment_match = latest_comment_url.match(/comments\/(\d+?)$/)
      comment_id = comment_match[1]
      client.issue_comment(repos_owner, comment_id)['html_url']
    else
      nil
    end
  end
end