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   def parse(nil), do: {:error, "No URL provided"}
   9
  10   if Mix.env() == :test do
  11     def parse(url), do: parse_url(url)
  12   else
  13     def parse(url) do
  14       with {:ok, data} <- Cachex.fetch(:rich_media_cache, url, fn _ -> parse_url(url) end) do
  15         data
  16       else
  17         _e ->
  18           {:error, "Parsing error"}
  19       end
  20     end
  21   end
  22
  23   defp parse_url(url) do
  24     {:ok, %Tesla.Env{body: html}} = Pleroma.HTTP.get(url)
  25
  26     html |> maybe_parse() |> get_parsed_data()
  27   end
  28
  29   defp maybe_parse(html) do
  30     Enum.reduce_while(@parsers, %{}, fn parser, acc ->
  31       case parser.parse(html, acc) do
  32         {:ok, data} -> {:halt, data}
  33         {:error, _msg} -> {:cont, acc}
  34       end
  35     end)
  36   end
  37
  38   defp get_parsed_data(data) when data == %{} do
  39     {:error, "No metadata found"}
  40   end
  41
  42   defp get_parsed_data(data) do
  43     {:ok, data}
  44   end
  45 end