-- This module is derivated from isbear's shortenurl function:
-- http://hg.lilotux.net/mod-mcabber-lua/file/tip/examples/shortenurl.lua
--
--  * Copyright (C) 2012      Myhailo Danylenko <isbear@ukrpost.net>
--  * Copyright (C) 2012      Mikael Berthe <mikael@lilotux.net>
--

require "libs.shcmd"

local json = require 'json'

local shortenurl = { ["desc"] = "URL shortener",
                   }

local encodeoptions = {
    strings = {
        encodeSet = '\\"%z\1-\031', -- do not escape /, google does not understand that
    }
}

local shell_escape = function (str)
    if str then
        return "'" .. str:gsub("'", "'\\''") .. "'"
    else
        return "''"
    end
end

local curlcommand = "curl -s https://www.googleapis.com/urlshortener/v1/url -H 'Content-type: application/json' -d %s"

function shortenurl.cmd (url)

    if ( not url ) or ( not url:match ( '%w' ) ) then
        return nil,  'You must specify an URL'
    end

    local replystring = ''

    local targeturl = json.encode( { longUrl = url }, encodeoptions )
    local data = shcmd(curlcommand:format(shell_escape(targeturl)))
    if data then
        local reply = json.decode ( data )
        if ( type ( reply ) == 'table' ) and reply.id then
            return ('Full:  %s\nShort: %s'):format(url, reply.id)
        else
            local errmsg
            if type ( reply ) == 'table' and ( type ( reply.error ) == 'table' ) then
                errmsg = ('Failed to shorten url: %u %s'):format(reply.error.code,
                          reply.error.message)
            else
                errmsg = ('Failed to shorten url: Unknown response:\n%s'):format(replystring)
            end
            return nil, errmsg
        end
        return nil, "Got no data :-("
    end
end

mcbot_register_command("shortenurl", shortenurl)

-- vim: se ts=4 sw=4: --