summaryrefslogblamecommitdiff
path: root/lib/lsg_irc/link_plugin/imgur.ex
blob: 443afdb3d5d2dd71dac16982454ae07f73da247e (plain) (tree)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
















                                           
            





                                                   











                                                                             


                                       















                                                                                        
                                                                          




                                                                         


































                                                                                                              
defmodule LSG.IRC.LinkPlugin.Imgur do
  @behaviour LSG.IRC.LinkPlugin

  @moduledoc """
  # Imgur link preview

  No options.

  Needs to have a Imgur API key configured:

  ```
  config :lsg, :imgur,
    client_id: "xxxxxxxx",
    client_secret: "xxxxxxxxxxxxxxxxxxxx"
  ```
  """

  @impl true
  def match(uri = %URI{host: "imgur.io"}, arg) do
    match(%URI{uri | host: "imgur.com"}, arg)
  end
  def match(uri = %URI{host: "i.imgur.io"}, arg) do
    match(%URI{uri | host: "i.imgur.com"}, arg)
  end
  def match(uri = %URI{host: "imgur.com", path: "/a/"<>album_id}, _) do
    {true, %{album_id: album_id}}
  end
  def match(uri = %URI{host: "imgur.com", path: "/gallery/"<>album_id}, _) do
    {true, %{album_id: album_id}}
  end
  def match(uri = %URI{host: "i.imgur.com", path: "/"<>image}, _) do
    [hash, _] = String.split(image, ".", parts: 2)
    {true, %{image_id: hash}}
  end
  def match(_, _), do: false

  @impl true
  def post_match(_, _, _, _), do: false

  def expand(_uri, %{album_id: album_id}, opts) do
    expand_imgur_album(album_id, opts)
  end

  def expand(_uri, %{image_id: image_id}, opts) do
    expand_imgur_image(image_id, opts)
  end

  def expand_imgur_image(image_id, opts) do
    client_id = Keyword.get(Application.get_env(:lsg, :imgur, []), :client_id, "42")
    headers = [{"Authorization", "Client-ID #{client_id}"}]
    options = []
    case HTTPoison.get("https://api.imgur.com/3/image/#{image_id}", headers, options) do
      {:ok, %HTTPoison.Response{status_code: 200, body: body}} ->
        {:ok, json} = Jason.decode(body)
        data = json["data"]
        title = String.slice(data["title"] || data["description"], 0, 180)
        nsfw = if data["nsfw"], do: "(NSFW) - ", else: " "
        height = Map.get(data, "height")
        width = Map.get(data, "width")
        size = Map.get(data, "size")
        {:ok, "image, #{width}x#{height}, #{size} bytes #{nsfw}#{title}"}
      other ->
        :error
    end
  end

  def expand_imgur_album(album_id, opts) do
    client_id = Keyword.get(Application.get_env(:lsg, :imgur, []), :client_id, "42")
    headers = [{"Authorization", "Client-ID #{client_id}"}]
    options = []
    case HTTPoison.get("https://api.imgur.com/3/album/#{album_id}", headers, options) do
      {:ok, %HTTPoison.Response{status_code: 200, body: body}} ->
        {:ok, json} = Jason.decode(body)
        data = json["data"]
        title = data["title"]
        nsfw = data["nsfw"]
        nsfw = if nsfw, do: "(NSFW) - ", else: ""
        if data["images_count"] == 1 do
          [image] = data["images"]
          title = if title || data["title"] do
            title = [title, data["title"]] |> Enum.filter(fn(x) -> x end) |> Enum.uniq() |> Enum.join(" — ")
            "#{title} — "
          else
            ""
          end
          {:ok, "#{nsfw}#{title}#{image["link"]}"}
        else
          title = if title, do: title, else: "Untitled album"
          {:ok, "#{nsfw}#{title} - #{data["images_count"]} images"}
        end
      other ->
        :error
    end
  end

end