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     try do
  25       {:ok, %Tesla.Env{body: html}} = Pleroma.HTTP.get(url)
  26
  27       html |> maybe_parse() |> get_parsed_data()
  28     rescue
  29       _e ->
  30         {:error, "Parsing error"}
  31     end
  32   end
  33
  34   defp maybe_parse(html) do
  35     Enum.reduce_while(@parsers, %{}, fn parser, acc ->
  36       case parser.parse(html, acc) do
  37         {:ok, data} -> {:halt, data}
  38         {:error, _msg} -> {:cont, acc}
  39       end
  40     end)
  41   end
  42
  43   defp get_parsed_data(data) when data == %{} do
  44     {:error, "No metadata found"}
  45   end
  46
  47   defp get_parsed_data(data) do
  48     {:ok, data}
  49   end
  50 end