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

   1 defmodule Pleroma.Web.RichMedia.Parsers.MetaTagsParser do
   2   def parse(html, data, prefix, error_message, key_name, value_name \\ "content") do
   3     meta_data =
   4       html
   5       |> get_elements(key_name, prefix)
   6       |> Enum.reduce(data, fn el, acc ->
   7         attributes = normalize_attributes(el, prefix, key_name, value_name)
   8
   9         Map.merge(acc, attributes)
  10       end)
  11       |> maybe_put_title(html)
  12
  13     if Enum.empty?(meta_data) do
  14       {:error, error_message}
  15     else
  16       {:ok, meta_data}
  17     end
  18   end
  19
  20   defp get_elements(html, key_name, prefix) do
  21     html |> Floki.find("meta[#{key_name}^='#{prefix}:']")
  22   end
  23
  24   defp normalize_attributes(html_node, prefix, key_name, value_name) do
  25     {_tag, attributes, _children} = html_node
  26
  27     data =
  28       Enum.into(attributes, %{}, fn {name, value} ->
  29         {name, String.trim_leading(value, "#{prefix}:")}
  30       end)
  31
  32     %{String.to_atom(data[key_name]) => data[value_name]}
  33   end
  34
  35   defp maybe_put_title(%{title: _} = meta, _), do: meta
  36
  37   defp maybe_put_title(meta, html) do
  38     case get_page_title(html) do
  39       "" -> meta
  40       title -> Map.put_new(meta, :title, title)
  41     end
  42   end
  43
  44   defp get_page_title(html) do
  45     Floki.find(html, "title") |> Floki.text()
  46   end
  47 end