git.squeep.com Git - akkoma/blob - lib/pleroma/formatter.ex

   1 defmodule Pleroma.Formatter do
   2   alias Pleroma.User
   3   alias Pleroma.Web.MediaProxy
   4
   5   @tag_regex ~r/\#\w+/u
   6   def parse_tags(text, data \\ %{}) do
   7     Regex.scan(@tag_regex, text)
   8     |> Enum.map(fn (["#" <> tag = full_tag]) -> {full_tag, String.downcase(tag)} end)
   9     |> (fn map -> if data["sensitive"] in [true, "True", "true", "1"], do: [{"#nsfw", "nsfw"}] ++ map, else: map end).()
  10   end
  11
  12   def parse_mentions(text) do
  13     # Modified from https://www.w3.org/TR/html5/forms.html#valid-e-mail-address
  14     regex = ~r/@[a-zA-Z0-9.!#$%&'*+\/=?^_`{|}~-]+@?[a-zA-Z0-9](?:[a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?(?:\.[a-zA-Z0-9](?:[a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?)*/u
  15
  16     Regex.scan(regex, text)
  17     |> List.flatten
  18     |> Enum.uniq
  19     |> Enum.map(fn ("@" <> match = full_match) -> {full_match, User.get_cached_by_nickname(match)} end)
  20     |> Enum.filter(fn ({_match, user}) -> user end)
  21   end
  22
  23   @finmoji [
  24     "a_trusted_friend",
  25     "alandislands",
  26     "association",
  27     "auroraborealis",
  28     "baby_in_a_box",
  29     "bear",
  30     "black_gold",
  31     "christmasparty",
  32     "crosscountryskiing",
  33     "cupofcoffee",
  34     "education",
  35     "fashionista_finns",
  36     "finnishlove",
  37     "flag",
  38     "forest",
  39     "four_seasons_of_bbq",
  40     "girlpower",
  41     "handshake",
  42     "happiness",
  43     "headbanger",
  44     "icebreaker",
  45     "iceman",
  46     "joulutorttu",
  47     "kaamos",
  48     "kalsarikannit_f",
  49     "kalsarikannit_m",
  50     "karjalanpiirakka",
  51     "kicksled",
  52     "kokko",
  53     "lavatanssit",
  54     "losthopes_f",
  55     "losthopes_m",
  56     "mattinykanen",
  57     "meanwhileinfinland",
  58     "moominmamma",
  59     "nordicfamily",
  60     "out_of_office",
  61     "peacemaker",
  62     "perkele",
  63     "pesapallo",
  64     "polarbear",
  65     "pusa_hispida_saimensis",
  66     "reindeer",
  67     "sami",
  68     "sauna_f",
  69     "sauna_m",
  70     "sauna_whisk",
  71     "sisu",
  72     "stuck",
  73     "suomimainittu",
  74     "superfood",
  75     "swan",
  76     "the_cap",
  77     "the_conductor",
  78     "the_king",
  79     "the_voice",
  80     "theoriginalsanta",
  81     "tomoffinland",
  82     "torillatavataan",
  83     "unbreakable",
  84     "waiting",
  85     "white_nights",
  86     "woollysocks"
  87   ]
  88
  89   @finmoji_with_filenames Enum.map(@finmoji, fn (finmoji) ->
  90     {finmoji, "/finmoji/128px/#{finmoji}-128.png"}
  91   end)
  92
  93   @emoji_from_file (with {:ok, default} <- File.read("config/emoji.txt") do
  94                       custom =
  95                         with {:ok, custom} <- File.read("config/custom_emoji.txt") do
  96                           custom
  97                         else
  98                           _e -> ""
  99                         end
 100                       (default <> "\n" <> custom)
 101                       |> String.trim()
 102                       |> String.split(~r/\n+/)
 103                       |> Enum.map(fn(line) ->
 104                         [name, file] = String.split(line, ~r/,\s*/)
 105                         {name, file}
 106                      end)
 107                     else
 108                       _ -> []
 109                    end)
 110
 111   @emoji @finmoji_with_filenames ++ @emoji_from_file
 112
 113   def emojify(text, additional \\ nil) do
 114     all_emoji = if additional do
 115       Map.to_list(additional) ++ @emoji
 116     else
 117       @emoji
 118     end
 119
 120     Enum.reduce(all_emoji, text, fn ({emoji, file}, text) ->
 121       emoji = HtmlSanitizeEx.strip_tags(emoji)
 122       file = HtmlSanitizeEx.strip_tags(file)
 123       String.replace(text, ":#{emoji}:", "<img height='32px' width='32px' alt='#{emoji}' title='#{emoji}' src='#{MediaProxy.url(file)}' />")
 124     end)
 125   end
 126
 127   def get_emoji(text) do
 128     Enum.filter(@emoji, fn ({emoji, _}) -> String.contains?(text, ":#{emoji}:") end)
 129   end
 130
 131   def get_custom_emoji() do
 132     @emoji
 133   end
 134
 135   @link_regex ~r/https?:\/\/[\w\.\/?=\-#%&@~\(\)]+[\w\/]/u
 136
 137   def html_escape(text) do
 138     Regex.split(@link_regex, text, include_captures: true)
 139     |> Enum.map_every(2, fn chunk ->
 140       {:safe, part} = Phoenix.HTML.html_escape(chunk)
 141       part
 142     end)
 143     |> Enum.join("")
 144   end
 145
 146   @doc "changes http:... links to html links"
 147   def add_links({subs, text}) do
 148     links = Regex.scan(@link_regex, text)
 149     |> Enum.map(fn ([url]) -> {Ecto.UUID.generate, url} end)
 150
 151     uuid_text = links
 152     |> Enum.reduce(text, fn({uuid, url}, acc) -> String.replace(acc, url, uuid) end)
 153
 154     subs = subs ++ Enum.map(links, fn({uuid, url}) ->
 155       {uuid, "<a href='#{url}'>#{url}</a>"}
 156     end)
 157
 158     {subs, uuid_text}
 159   end
 160
 161   @doc "Adds the links to mentioned users"
 162   def add_user_links({subs, text}, mentions) do
 163     mentions = mentions
 164     |> Enum.sort_by(fn ({name, _}) -> -String.length(name) end)
 165     |> Enum.map(fn({name, user}) -> {name, user, Ecto.UUID.generate} end)
 166
 167     uuid_text = mentions
 168     |> Enum.reduce(text, fn ({match, _user, uuid}, text) ->
 169       String.replace(text, match, uuid)
 170     end)
 171
 172     subs = subs ++ Enum.map(mentions, fn ({match, %User{ap_id: ap_id}, uuid}) ->
 173       short_match = String.split(match, "@") |> tl() |> hd()
 174       {uuid, "<span><a href='#{ap_id}'>@<span>#{short_match}</span></a></span>"}
 175     end)
 176
 177     {subs, uuid_text}
 178   end
 179
 180   @doc "Adds the hashtag links"
 181   def add_hashtag_links({subs, text}, tags) do
 182     tags = tags
 183     |> Enum.sort_by(fn ({name, _}) -> -String.length(name) end)
 184     |> Enum.map(fn({name, short}) -> {name, short, Ecto.UUID.generate} end)
 185
 186     uuid_text = tags
 187     |> Enum.reduce(text, fn ({match, _short, uuid}, text) ->
 188       String.replace(text, match, uuid)
 189     end)
 190
 191     subs = subs ++ Enum.map(tags, fn ({_, tag, uuid}) ->
 192       url = "<a href='#{Pleroma.Web.base_url}/tag/#{tag}' rel='tag'>##{tag}</a>"
 193       {uuid, url}
 194     end)
 195
 196     {subs, uuid_text}
 197   end
 198
 199   def finalize({subs, text}) do
 200     Enum.reduce(subs, text, fn({uuid, replacement}, result_text) ->
 201       String.replace(result_text, uuid, replacement)
 202     end)
 203   end
 204 end