summaryrefslogtreecommitdiff
path: root/lib/lsg_irc/link_plugin/youtube.ex
blob: ea4f2139b0339fbf48da58cc53b4b7760bccd5bc (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
defmodule LSG.IRC.LinkPlugin.YouTube do
  @behaviour LSG.IRC.LinkPlugin

  @moduledoc """
  # YouTube link preview

  needs an API key:

  ```
  config :lsg, :youtube,
    api_key: "xxxxxxxxxxxxx"
  ```

  options:

  * `invidious`: Add a link to invidio.us. Default: true.
  """

  def match(uri = %URI{host: yt, path: "/watch", query: "v="<>video_id}, _opts) when yt in ["youtube.com", "www.youtube.com"] do
    {true, %{video_id: video_id}}
  end

  def match(%URI{host: "youtu.be", path: "/"<>video_id}, _opts) do
    {true, %{video_id: video_id}}
  end

  def match(_, _), do: false


  def expand(uri, %{video_id: video_id}, opts) do
    key = Application.get_env(:lsg, :youtube)[:api_key]
    params = %{
      "part" => "snippet,contentDetails,statistics",
      "id" => video_id,
      "key" => key
    }
    headers = []
    options = [params: params]
    case HTTPoison.get("https://www.googleapis.com/youtube/v3/videos", [], options) do
      {:ok, %HTTPoison.Response{status_code: 200, body: body}} ->
        case Jason.decode(body) do
          {:ok, json} ->
            item = List.first(json["items"])
            if item do
              snippet = item["snippet"]
              duration = item["contentDetails"]["duration"] |> String.replace("PT", "") |> String.downcase
              date = snippet["publishedAt"]
                     |> DateTime.from_iso8601()
                     |> elem(1)
                     |> Timex.format("{relative}", :relative)
                     |> elem(1)

              line = if Keyword.get(opts, :invidious, true) do
                ["-> https://invidio.us/watch?v=#{video_id}"]
              else
                  []
              end
              {:ok, line ++ ["#{snippet["title"]}", "— #{duration} — uploaded by #{snippet["channelTitle"]}#{date}"
                     <> " — #{item["statistics"]["viewCount"]} views, #{item["statistics"]["likeCount"]} likes,"
                     <> " #{item["statistics"]["dislikeCount"]} dislikes"]}
            else
              :error
            end
          _ -> :error
        end
    end
  end

end