Mikubot/bot/bot.lua

  http = require("socket.http")
  URL = require("socket.url")
  json = (loadfile "./bot/JSON.lua")()

  VERSION = 'v0.5'


  function on_msg_receive (msg)

    if msg_valid(msg) == false then
      return
    end
    -- Check if command starts with ! eg !echo
    if msg.text:sub(0,1) == '!' then
      do_action(msg)
    else
      if is_image_url(msg.text) then
        send_image_from_url (msg)
      elseif get_youtube_code(msg.text) then
        send_youtube_thumbnail(msg)
      else
        if is_file_url(msg.text) then
          send_file_from_url(msg)
        end
      end
    end

    mark_read(get_receiver(msg), ok_cb, false)
    -- write_log_file(msg)
  end


  function ok_cb(extra, success, result)
  end

  function msg_valid(msg)
     if msg.out then
        return false
     end
     if msg.date < now then
       return false
     end
     if msg.text == nil then
        return false
     end
     if msg.unread == 0 then
        return false
     end
  end

  function send_file_from_url (msg)
    last = string.get_last_word(msg.text)
    file = download_to_file(last)
    send_document(get_receiver(msg), file, ok_cb, false)
  end

  function send_image_from_url (msg)
    last = string.get_last_word(msg.text)
    file = download_to_file(last)
    send_photo(get_receiver(msg), file, ok_cb, false)
  end

  function is_image_url(text)
    last = string.get_last_word(text)
    extension = string.get_extension_from_filename(last) -- TODO:  Change it please
    if extension == 'jpg' or extension == 'png' or extension == 'jpeg' then
      return true
    end
    return false
  end

  function get_youtube_code( text )
    local yt_normal = string.match(text, "youtube.com/watch%?v=([A-Za-z0-9-]+)")
    if yt_normal then
      return yt_normal
    end
    yt_short = string.match(text, "youtu.be/([A-Za-z0-9-]+)")
    if yt_short then
      return yt_short
    end
    return false
  end

  function send_youtube_thumbnail(msg)
    yt_code = get_youtube_code(msg.text)
    yt_thumbnail = "http://img.youtube.com/vi/".. yt_code .."/hqdefault.jpg"
    file = download_to_file(yt_thumbnail)
    send_photo(get_receiver(msg), file, ok_cb, false)
  end

  function is_file_url(text)
    last = string.get_last_word(text)
    extension = string.get_extension_from_filename(last)
    if extension == 'gif' then
      return true
    end
    return false
  end

  -- Where magic happens
  function do_action(msg)
     local receiver = get_receiver(msg)

     if string.starts(msg.text, '!sh') then
        text = run_sh(msg)
        send_msg(receiver, text, ok_cb, false)
        return
     end
    
     if string.starts(msg.text, '!uc3m') then
        text = get_fortunes_uc3m()
        send_msg(receiver, text, ok_cb, false)
        return
     end 

     if string.starts(msg.text, '!img') then
        text = msg.text:sub(6,-1)
        url = getGoogleImage(text)
        file_path = download_to_file(url)
        print(file_path)
        send_photo(receiver, file_path, ok_cb, false)
        return
     end

    if string.starts(msg.text, '!rae') then
      text = msg.text:sub(6,-1)
      meaning = getDulcinea(text)
      send_msg(receiver, meaning, ok_cb, false)
    end
    
    if string.starts(msg.text, '!9gag') then
      url, title = get_9GAG()
      file_path = download_to_file(url)
      send_photo(receiver, file_path, ok_cb, false)
      send_msg(receiver, title, ok_cb, false)
      return
    end 

     if string.starts(msg.text, '!fortune') then
        text = run_bash('fortune')
        send_msg(receiver, text, ok_cb, false)
        return
     end 

     if string.starts(msg.text, '!forni') then
        text = msg.text:sub(8,-1)
        send_msg('Fornicio_2.0', text, ok_cb, false)
        return
     end 

     if string.starts(msg.text, '!fwd') then
        fwd_msg (receiver, msg.id, ok_cb, false)
        return
     end 

     if string.starts(msg.text, '!cpu') then
        text = run_bash('uname -snr') .. ' ' .. run_bash('whoami')
        text = text .. '\n' .. run_bash('top -b |head -2')
        send_msg(receiver, text, ok_cb, false)
        return
     end

     if string.starts(msg.text, '!ping') then
        print('receiver: '..receiver)
        send_msg (receiver, 'pong', ok_cb, false)
        return
     end

     if string.starts(msg.text, '!weather') then
        if string.len(msg.text) <= 9 then
           city = 'Madrid,ES'
        else  
           city = msg.text:sub(10,-1)
        end    
        text = get_weather(city)
        send_msg(receiver, text, ok_cb, false)
        return
     end

     if string.starts(msg.text, '!echo') then
        echo = msg.text:sub(7,-1)
        send_msg(receiver, echo, ok_cb, false)
        return
     end

     if string.starts(msg.text, '!eur') then
        local eur = getEURUSD(msg.text)
        send_msg(receiver, eur, ok_cb, false)
        return
     end

     if string.starts(msg.text, '!set') then
        local text = save_value(msg.text:sub(5,-1))
        send_msg(receiver, text, ok_cb, false)
        return
     end

     if string.starts(msg.text, '!get') then
        local text = get_value(msg.text:sub(6,-1))
        send_msg(receiver, text, ok_cb, false)
        return
     end

     if string.starts(msg.text, '!version') then
        text = 'James Bot '.. VERSION .. [[ 
  Licencia GNU v2, código disponible en http://git.io/6jdjGg

  Al Bot le gusta la gente solidaria. 
  Puedes hacer una donación a la ONG que decidas y ayudar a otras personas.]]
        send_msg(receiver, text, ok_cb, false)
        return
     end

     if string.starts(msg.text, '!help') then
        text = [[!help : print this help 
  !ping : bot sends pong 
  !sh (text) : send commands to bash (only privileged users)
  !echo (text) : echo the msg 
  !version : version info
  !cpu : status (uname + top)
  !fwd : forward msg
  !forni : send text to group Fornicio
  !fortune : print a random adage
  !weather [city] : weather in that city (Madrid if not city)
  !9gag : send random url image from 9gag
  !rae (word): Spanish dictionary
  !eur : EURUSD market value
  !img (text) : search image with Google API and sends it
  !uc3m : fortunes from Universidad Carlos III]]
        send_msg(receiver, text, ok_cb, false)
        return
     end

  end

  function string.starts(String,Start)
     return string.sub(String,1,string.len(Start))==Start
  end

  function load_config()
     local f = assert(io.open('./bot/config.json', "r"))
     local c = f:read "*a"
     local config = json:decode(c)
     if config.sh_enabled then 
        print ("!sh command is enabled")
        for v,user in pairs(config.sudo_users) do
           print("Allowed user: " .. user)
        end
     end
     -- print("Torrent path: " .. config.torrent_path)
     f:close()
     return config
  end

  function save_value( text )
    local vars = split_by_space(text)
    if (#vars < 2) then
      return "Usage: !set var_name value"
    end
    config.values[vars[1]] = vars[2]
    local json_text = json:encode_pretty(config) 
    file = io.open ("./bot/config.json", "w+")
    file:write(json_text)
    file:close()
    return "Saved "..vars[1].."="..vars[2]
  end

  function get_value( value_name )
    local value = config.values[value_name]
    if ( value == nil) then
      return "Cant find "..value_name
    end
    return value_name.." = "..value
  end

  function is_sudo(msg)
     local var = false
     -- Check users id in config 
     for v,user in pairs(config.sudo_users) do 
        if user == msg.from.id then 
           var = true 
        end
     end
     return var
  end

  function write_log_file(msg)
    name = get_name(msg)
    ret = name .. ' > ' .. msg.text
    write_to_file(config.log_file, ret)
  end

  -- Saves a string to file
  function write_to_file(filename, value)
    if (value) then
      local file = io.open(filename,"a")
      file:write(value, "\n")
      file:close()
    end
  end

  function get_name(msg)
     local name = msg.from.first_name
     if name == nil then
        name = msg.from.id
     end
     return name
  end

  function run_sh(msg)
     name = get_name(msg)
     text = ''
     if config.sh_enabled == false then 
        text = '!sh command is disabled'
     else
        if is_sudo(msg) then
           bash = msg.text:sub(4,-1)
           text = run_bash(bash)
        else
           text = name .. ' you have no power here!'
        end
     end
     return text
  end

  function run_bash(str)
    local cmd = io.popen(str)
    local result = cmd:read('*all')
    cmd:close()
    return result
  end

  function get_fortunes_uc3m()
     math.randomseed(os.time())
     local i = math.random(0,178) -- max 178
     local web = "http://www.gul.es/fortunes/f"..i 
     b, c, h = http.request(web)
     return b
  end

  function getDulcinea( text )
    -- Powered by https://github.com/javierhonduco/dulcinea
    local api = "http://dulcinea.herokuapp.com/api/?query="
    b = http.request(api..text)
    dulcinea = json:decode(b)
    if dulcinea.status == "error" then
      return "Error: " .. dulcinea.message
    end
    while dulcinea.type == "multiple" do
      text = dulcinea.response[1].id
      b = http.request(api..text)
      dulcinea = json:decode(b)
    end
    vardump(dulcinea)
    local text = ""
    local responses = #dulcinea.response
    if (responses > 5) then
      responses = 5
    end
    for i = 1, responses, 1 do 
      text = text .. dulcinea.response[i].word .. "\n"
      local meanings = #dulcinea.response[i].meanings
      if (meanings > 5) then
        meanings = 5
      end
      for j = 1, meanings, 1 do
        local meaning = dulcinea.response[i].meanings[j].meaning 
        text = text .. meaning .. "\n\n"
      end
    end
    return text

  end

  function getGoogleImage(text)
    text = URL.escape(text)
    for i = 1, 5, 1 do -- Try 5 times
      local api = "https://ajax.googleapis.com/ajax/services/search/images?v=1.0&rsz=8&q="
      b = http.request(api..text)
      local google = json:decode(b)

      if (google.responseStatus == 200) then -- OK
        math.randomseed(os.time())
        i = math.random(#google.responseData.results) -- Random image from results
        return google.responseData.results[i].url
      end
    end
  end

  function get_9GAG()
     b = http.request("http://api-9gag.herokuapp.com/")
     local gag = json:decode(b)
     math.randomseed(os.time())
     i = math.random(#gag) -- random max json table size (# is an operator o.O)
     local link_image = gag[i].src
     local title = gag[i].title
     if link_image:sub(0,2) == '//' then
        link_image = msg.text:sub(3,-1)
     end
     return link_image, title
  end

  function getEURUSD(text)
    b = http.request("http://webrates.truefx.com/rates/connect.html?c=EUR/USD&f=csv&s=n")
    local rates = b:split(", ")
    local symbol = rates[1]
    local timestamp = rates[2]
    local sell = rates[3]..rates[4]
    local buy = rates[5]..rates[6] 
    local usd =  string.match(text, "!eur (%d+[%d%.]*)")
    text = symbol..'\n'..'Buy: '..buy..'\n'..'Sell: '..sell
    if usd then
      eur = tonumber(usd) / tonumber(buy)
      text = text.."\n "..usd.."USD = "..eur.."EUR"
    end
    return text
  end


  function download_to_file( url )
    print("url a descargar: "..url)
    req, c, h = http.request(url)
    htype = h["content-type"]
    vardump(c)
    print("content-type: "..htype)
    if htype == "image/jpeg" then
      file_name = string.random(5)..".jpg"
      file_path = "/tmp/"..file_name
    else
      if htype == "image/gif" then
        file_name = string.random(5)..".gif"
        file_path = "/tmp/"..file_name
      else
        if htype == "image/png" then
          file_name = string.random(5)..".png"
          file_path = "/tmp/"..file_name
        else
          file_name = url:match("([^/]+)$")
          file_path = "/tmp/"..file_name
        end
      end
    end
    file = io.open(file_path, "w+")
    file:write(req)
    file:close()
    return file_path
  end

  function get_weather(location)
     b, c, h = http.request("http://api.openweathermap.org/data/2.5/weather?q=" .. location .. "&units=metric")
     weather = json:decode(b)
     local city = weather.name
     local country = weather.sys.country
     temp = 'The temperature in ' .. city .. ' (' .. country .. ')'
     temp = temp .. ' is ' .. weather.main.temp .. '°C'
     conditions = 'Current conditions are: ' .. weather.weather[1].description
     if weather.weather[1].main == 'Clear' then
  	  conditions = conditions .. ' ☀'
     elseif weather.weather[1].main == 'Clouds' then
  	  conditions = conditions .. ' ☁☁'
     elseif weather.weather[1].main == 'Rain' then
  	  conditions = conditions .. ' ☔'
     elseif weather.weather[1].main == 'Thunderstorm' then
  	  conditions = conditions .. ' ☔☔☔☔'
     end
     return temp .. '\n' .. conditions
  end

  function string.random(length)
     math.randomseed(os.time())
     local str = "";
     for i = 1, length do
        math.random(97, 122)
        str = str..string.char(math.random(97, 122));
     end
     return str;
  end

  function string.get_extension_from_filename( filename )
    return filename:match( "%.([^%.]+)$" )
  end

  function string.get_last_word( words )
    local splitted = split_by_space ( words )
    return splitted[#splitted]
  end

  function split_by_space ( text )
    words = {}
    for word in string.gmatch(text, "[^%s]+") do
       table.insert(words, word) 
    end
    return words
  end

  function string:split(sep)
    local sep, fields = sep or ":", {}
    local pattern = string.format("([^%s]+)", sep)
    self:gsub(pattern, function(c) fields[#fields+1] = c end)
    return fields
  end

  function vardump(value, depth, key)
    local linePrefix = ""
    local spaces = ""
    
    if key ~= nil then
      linePrefix = "["..key.."] = "
    end
    
    if depth == nil then
      depth = 0
    else
      depth = depth + 1
      for i=1, depth do spaces = spaces .. "  " end
    end
    
    if type(value) == 'table' then
      mTable = getmetatable(value)
      if mTable == nil then
        print(spaces ..linePrefix.."(table) ")
      else
        print(spaces .."(metatable) ")
          value = mTable
      end		
      for tableKey, tableValue in pairs(value) do
        vardump(tableValue, depth, tableKey)
      end
    elseif type(value)	== 'function' or 
        type(value)	== 'thread' or 
        type(value)	== 'userdata' or
        value		== nil
    then
      print(spaces..tostring(value))
    else
      print(spaces..linePrefix.."("..type(value)..") "..tostring(value))
    end
  end

  -- Start and load values
  config = load_config()
  our_id = 0
  now = os.time()

  function get_receiver(msg)
    if msg.to.type == 'user' then
      return 'user#id'..msg.from.id
    end
    if msg.to.type == 'chat' then
      return 'chat#id'..msg.to.id
    end
  end


  function on_our_id (id)
    our_id = id
  end

  function on_user_update (user, what)
    --vardump (user)
  end

  function on_chat_update (chat, what)
    --vardump (chat)
  end

  function on_secret_chat_update (schat, what)
    --vardump (schat)
  end

  function on_get_difference_end ()
  end

  function on_binlog_replay_end ()
    started = 1
  end