# -*- coding: utf-8 -*-
require 'cinch-toolbox/version'
require 'open-uri'
require 'patron'

module Cinch
  module Toolbox
    # Get an element of the supplied website
    def Toolbox.get_html_element(url, selector, mode = 'css')
      # Make sure the URL is legit
      url = URI::extract(url, ["http", "https"]).first

      case mode
      when 'css'
        return Nokogiri::HTML(open(url)).css(selector).first.content
      when 'xpath'
        return Nokogiri::HTML(open(url)).xpath(selector).first
      end
    end

    # Expand a previously shortened URL via the configured shortener
    def Toolbox.expand(url)
      shortener.get("forward.php?format=simple&shorturl=#{url}").body
    end

    # Shorten a URL via the configured shortener
    def Toolbox.shorten(url)
      return url if url.length < 45
      return shortener.get("create.php?format=simple&url=#{url}").body
    end

    # Truncate a given block of text, used for making sure the bot doesn't flood.
    def Toolbox.truncate(text, length = 250)
      text = text.gsub(/\n/, ' · ')
      if text.length > length
        text = text[0,length - 1] + '...'
      end
      return text
    end

    # Used to render a period of time in a uniform string.
    # There is probably a much better way to do this, so FIXME
    def Toolbox.time_format(secs)
      data = time_parse(secs)
      string = ''

      string << "#{data[:days]}d "  unless data[:days].zero?  && string == ''
      string << "#{data[:hours]}h " unless data[:hours].zero? && string == ''
      string << "#{data[:mins]}m "  unless data[:mins].zero?  && string == ''
      string << "#{data[:secs]}s"

      return string
    end

    def Toolbox.time_parse(secs)
      days = secs / 86400
      hours = (secs % 86400) / 3600
      mins = (secs % 3600) / 60
      secs = secs % 60

      return { :days => days.floor,
               :hours => hours.floor,
               :mins => mins.floor,
               :secs => secs.floor }
    end

    private

    def Toolbox.shortener
      short = Patron::Session.new
      short.base_url = 'http://is.gd/'
      return short
    end
  end
end