atig/atig

# -*- mode:ruby; coding:utf-8 -*-

require 'atig/bitly'

module Atig
  module OFilter
    class ShortUrl
      MIN_LEN = 20

      def initialize(context)
        @log  = context.log
        @opts = context.opts
        @http = Atig::Http.new @log
      end

      def call(status)
        mesg = status[:status]
        status.merge(status: short_urls(mesg))
      end

      def short_urls(mesg)
        shorten = case
                  when @opts.bitlify.to_s.include?(":")
                    login, key, len = @opts.bitlify.to_s.split(":", 3)
                    @len = (len || MIN_LEN).to_i
                    Bitly.login @log, login, key
                  when @opts.bitlify
                    @len = (@opts.bitlify.to_s || MIN_LEN).to_i
                    Bitly.no_login @log
                  else
                    return mesg
                  end
        mesg.gsub(URI.regexp(%w[http https])) do|url|
          if Atig::UrlEscape.rstrip(url).size < @len then
            url
          else
            shorten.shorten url
          end
        end
      end
    end
  end
end