git.squeep.com Git - akkoma/blob - lib/pleroma/web/rich_media/parser.ex

   1 defmodule Pleroma.Web.RichMedia.Parser do
   2   @parsers [
   3     Pleroma.Web.RichMedia.Parsers.OGP,
   4     Pleroma.Web.RichMedia.Parsers.TwitterCard,
   5     Pleroma.Web.RichMedia.Parsers.OEmbed
   6   ]
   7
   8   if Mix.env() == :test do
   9     def parse(url), do: parse_url(url)
  10   else
  11     def parse(url),
  12       do: Cachex.fetch!(:rich_media_cache, url, fn _ -> parse_url(url) end)
  13   end
  14
  15   defp parse_url(url) do
  16     {:ok, %Tesla.Env{body: html}} = Pleroma.HTTP.get(url)
  17
  18     html |> maybe_parse() |> get_parsed_data()
  19   end
  20
  21   defp maybe_parse(html) do
  22     Enum.reduce_while(@parsers, %{}, fn parser, acc ->
  23       case parser.parse(html, acc) do
  24         {:ok, data} -> {:halt, data}
  25         {:error, _msg} -> {:cont, acc}
  26       end
  27     end)
  28   end
  29
  30   defp get_parsed_data(data) when data == %{} do
  31     {:error, "No metadata found"}
  32   end
  33
  34   defp get_parsed_data(data) do
  35     {:ok, data}
  36   end
  37 end