Mikubot/plugins/danbooru.lua

do
local URL = "http://danbooru.donmai.us"
local URL_NEW = "/posts.json"
local URL_POP = "/explore/posts/popular.json"

local scale_day = "?scale=day"
local scale_week = "?scale=week"
local scale_month = "?scale=month"

local function get_post(url)
  local b, c, h = http.request(url)
  if c ~= 200 then return nil end
  local posts = json:decode(b)

  return posts[math.random(#posts)]
end

local function run(msg, matches)

  local url = URL

  if matches[1] == "/dan" then
    url = url .. URL_NEW
  else
    url = url .. URL_POP

        if matches[1] == "d" then
          url = url .. scale_day
    elseif matches[1] == "w" then
          url = url .. scale_week
    elseif matches[1] == "m" then
          url = url .. scale_month
    end
  end

  local post = get_post(url)

  if post then
    --vardump(post)
    local img = URL .. post.large_file_url
    send_photo_from_url(get_receiver(msg), img)

    local txt = ''
    if post.tag_string_artist ~= '' then
      txt = 'Artist: ' .. post.tag_string_artist .. '\n'
    end
    if post.tag_string_character ~= '' then
      txt = txt .. 'Character: ' .. post.tag_string_character .. '\n'
    end
    if post.file_size ~= '' then
      txt = txt .. '[' .. math.ceil(post.file_size/1000) .. 'kb] ' .. URL .. post.file_url
    end
    return txt
  end
end

return {
  description = "Gets a random fresh or popular image from Danbooru", 
  usage = {
    "/dan - gets a random fresh image from Danbooru 🔞",
    "/dan d - random daily popular image 🔞",
    "/dan w - random weekly popular image 🔞",
    "/dan m - random monthly popular image 🔞"},
  patterns = {
      "^/dan$",
      "^/dan ?(d)$",
      "^/dan ?(w)$",
      "^/dan ?(m)$"}, 
  run = run 
}

end