Fix hashtags.
authorlain <lain@soykaf.club>
Sat, 24 Mar 2018 10:03:10 +0000 (11:03 +0100)
committerlain <lain@soykaf.club>
Sat, 24 Mar 2018 10:03:10 +0000 (11:03 +0100)
lib/pleroma/formatter.ex
lib/pleroma/web/common_api/utils.ex
test/formatter_test.exs
test/web/twitter_api/twitter_api_test.exs

index fd8465c1cd34b259e55f9566c236c425d1d3a7d5..2d3487f6a3e8c4d6c9b342595e63952ff7a9ddd3 100644 (file)
@@ -2,11 +2,6 @@ defmodule Pleroma.Formatter do
   alias Pleroma.User
   alias Pleroma.Web.MediaProxy
 
-  @link_regex ~r/https?:\/\/[\w\.\/?=\-#%&@~\(\)]+[\w\/]/u
-  def linkify(text) do
-    Regex.replace(@link_regex, text, "<a href='\\0'>\\0</a>")
-  end
-
   @tag_regex ~r/\#\w+/u
   def parse_tags(text, data \\ %{}) do
     Regex.scan(@tag_regex, text)
@@ -25,15 +20,6 @@ defmodule Pleroma.Formatter do
     |> Enum.filter(fn ({_match, user}) -> user end)
   end
 
-  def html_escape(text) do
-    Regex.split(@link_regex, text, include_captures: true)
-    |> Enum.map_every(2, fn chunk ->
-      {:safe, part} = Phoenix.HTML.html_escape(chunk)
-      part
-    end)
-    |> Enum.join("")
-  end
-
   @finmoji [
     "a_trusted_friend",
     "alandislands",
@@ -145,4 +131,74 @@ defmodule Pleroma.Formatter do
   def get_custom_emoji() do
     @emoji
   end
+
+  @link_regex ~r/https?:\/\/[\w\.\/?=\-#%&@~\(\)]+[\w\/]/u
+
+  def html_escape(text) do
+    Regex.split(@link_regex, text, include_captures: true)
+    |> Enum.map_every(2, fn chunk ->
+      {:safe, part} = Phoenix.HTML.html_escape(chunk)
+      part
+    end)
+    |> Enum.join("")
+  end
+
+  @doc "changes http:... links to html links"
+  def add_links({subs, text}) do
+    links = Regex.scan(@link_regex, text)
+    |> Enum.map(fn ([url]) -> {Ecto.UUID.generate, url} end)
+
+    uuid_text = links
+    |> Enum.reduce(text, fn({uuid, url}, acc) -> String.replace(acc, url, uuid) end)
+
+    subs = subs ++ Enum.map(links, fn({uuid, url}) ->
+      {uuid, "<a href='#{url}'>#{url}</a>"}
+    end)
+
+    {subs, uuid_text}
+  end
+
+  @doc "Adds the links to mentioned users"
+  def add_user_links({subs, text}, mentions) do
+    mentions = mentions
+    |> Enum.sort_by(fn ({name, _}) -> -String.length(name) end)
+    |> Enum.map(fn({name, user}) -> {name, user, Ecto.UUID.generate} end)
+
+    uuid_text = mentions
+    |> Enum.reduce(text, fn ({match, _user, uuid}, text) ->
+      String.replace(text, match, uuid)
+    end)
+
+    subs = subs ++ Enum.map(mentions, fn ({match, %User{ap_id: ap_id}, uuid}) ->
+      short_match = String.split(match, "@") |> tl() |> hd()
+      {uuid, "<span><a href='#{ap_id}'>@<span>#{short_match}</span></a></span>"}
+    end)
+
+    {subs, uuid_text}
+  end
+
+  @doc "Adds the hashtag links"
+  def add_hashtag_links({subs, text}, tags) do
+    tags = tags
+    |> Enum.sort_by(fn ({name, _}) -> -String.length(name) end)
+    |> Enum.map(fn({name, short}) -> {name, short, Ecto.UUID.generate} end)
+
+    uuid_text = tags
+    |> Enum.reduce(text, fn ({match, _short, uuid}, text) ->
+      String.replace(text, match, uuid)
+    end)
+
+    subs = subs ++ Enum.map(tags, fn ({_, tag, uuid}) ->
+      url = "#<a href='#{Pleroma.Web.base_url}/tag/#{tag}' rel='tag'>#{tag}</a>"
+      {uuid, url}
+    end)
+
+    {subs, uuid_text}
+  end
+
+  def finalize({subs, text}) do
+    Enum.reduce(subs, text, fn({uuid, replacement}, result_text) ->
+      String.replace(result_text, uuid, replacement)
+    end)
+  end
 end
index 75c63e5f418bc79ed8938e59d94cc0c3e08b9a1e..3c09f0cc756753914806e170332913784e37333a 100644 (file)
@@ -78,13 +78,15 @@ defmodule Pleroma.Web.CommonAPI.Utils do
     Enum.join([text | attachment_text], "<br>")
   end
 
-  def format_input(text, mentions, _tags) do
+  def format_input(text, mentions, tags) do
     text
     |> Formatter.html_escape
-    |> Formatter.linkify
     |> String.replace("\n", "<br>")
-    |> add_user_links(mentions)
-    # |> add_tag_links(tags)
+    |> (&({[], &1})).()
+    |> Formatter.add_links
+    |> Formatter.add_user_links(mentions)
+    |> Formatter.add_hashtag_links(tags)
+    |> Formatter.finalize
   end
 
   def add_tag_links(text, tags) do
@@ -97,25 +99,6 @@ defmodule Pleroma.Web.CommonAPI.Utils do
     end)
   end
 
-  def add_user_links(text, mentions) do
-    mentions = mentions
-    |> Enum.sort_by(fn ({name, _}) -> -String.length(name) end)
-    |> Enum.map(fn({name, user}) -> {name, user, Ecto.UUID.generate} end)
-
-    # This replaces the mention with a unique reference first so it doesn't
-    # contain parts of other replaced mentions. There probably is a better
-    # solution for this...
-    step_one = mentions
-    |> Enum.reduce(text, fn ({match, _user, uuid}, text) ->
-      String.replace(text, match, uuid)
-    end)
-
-    Enum.reduce(mentions, step_one, fn ({match, %User{ap_id: ap_id}, uuid}, text) ->
-      short_match = String.split(match, "@") |> tl() |> hd()
-      String.replace(text, uuid, "<span><a href='#{ap_id}'>@<span>#{short_match}</span></a></span>")
-    end)
-  end
-
   def make_note_data(actor, to, context, content_html, attachments, inReplyTo, tags, cw \\ nil, cc \\ []) do
       object = %{
         "type" => "Note",
index cb7695e8e1c2b3576d242dcc77d03e55fcb54b89..0e915d8d55aa836eebaf67f17ba1b5b32285e696 100644 (file)
@@ -4,37 +4,67 @@ defmodule Pleroma.FormatterTest do
 
   import Pleroma.Factory
 
-  describe ".linkify" do
+  describe ".add_hashtag_links" do
+    test "turns hashtags into links" do
+      text = "I love #cofe and #2hu"
+      expected_text = "I love #<a href='http://localhost:4001/tag/cofe' rel='tag'>cofe</a> and #<a href='http://localhost:4001/tag/2hu' rel='tag'>2hu</a>"
+
+      tags = Formatter.parse_tags(text)
+      assert expected_text == Formatter.add_hashtag_links({[], text}, tags) |> Formatter.finalize
+    end
+  end
+
+  describe ".add_links" do
     test "turning urls into links" do
       text = "Hey, check out https://www.youtube.com/watch?v=8Zg1-TufF%20zY?x=1&y=2#blabla."
       expected = "Hey, check out <a href='https://www.youtube.com/watch?v=8Zg1-TufF%20zY?x=1&y=2#blabla'>https://www.youtube.com/watch?v=8Zg1-TufF%20zY?x=1&y=2#blabla</a>."
 
-      assert Formatter.linkify(text) == expected
+      assert Formatter.add_links({[], text}) |> Formatter.finalize == expected
 
       text = "https://mastodon.social/@lambadalambda"
       expected = "<a href='https://mastodon.social/@lambadalambda'>https://mastodon.social/@lambadalambda</a>"
 
-      assert Formatter.linkify(text) == expected
+      assert Formatter.add_links({[], text}) |> Formatter.finalize == expected
 
       text = "@lambadalambda"
       expected = "@lambadalambda"
 
-      assert Formatter.linkify(text) == expected
+      assert Formatter.add_links({[], text}) |> Formatter.finalize == expected
 
       text = "http://www.cs.vu.nl/~ast/intel/"
       expected = "<a href='http://www.cs.vu.nl/~ast/intel/'>http://www.cs.vu.nl/~ast/intel/</a>"
 
-      assert Formatter.linkify(text) == expected
+      assert Formatter.add_links({[], text}) |> Formatter.finalize == expected
 
       text = "https://forum.zdoom.org/viewtopic.php?f=44&t=57087"
       expected = "<a href='https://forum.zdoom.org/viewtopic.php?f=44&t=57087'>https://forum.zdoom.org/viewtopic.php?f=44&t=57087</a>"
 
-      assert Formatter.linkify(text) == expected
+      assert Formatter.add_links({[], text}) |> Formatter.finalize == expected
 
       text = "https://en.wikipedia.org/wiki/Sophia_(Gnosticism)#Mythos_of_the_soul"
       expected = "<a href='https://en.wikipedia.org/wiki/Sophia_(Gnosticism)#Mythos_of_the_soul'>https://en.wikipedia.org/wiki/Sophia_(Gnosticism)#Mythos_of_the_soul</a>"
 
-      assert Formatter.linkify(text) == expected
+      assert Formatter.add_links({[], text}) |> Formatter.finalize == expected
+    end
+  end
+
+  describe "add_user_links" do
+    test "gives a replacement for user links" do
+      text = "@gsimg According to @archaeme, that is @daggsy. Also hello @archaeme@archae.me"
+      gsimg = insert(:user, %{nickname: "gsimg"})
+      archaeme = insert(:user, %{nickname: "archaeme"})
+      archaeme_remote = insert(:user, %{nickname: "archaeme@archae.me"})
+
+      mentions = Pleroma.Formatter.parse_mentions(text)
+
+      {subs, text} = Formatter.add_user_links({[], text}, mentions)
+
+      assert length(subs) == 3
+      Enum.each(subs, fn({uuid, _}) -> assert String.contains?(text, uuid) end)
+
+      expected_text = "<span><a href='#{gsimg.ap_id}'>@<span>gsimg</span></a></span> According to <span><a href='#{archaeme.ap_id}'>@<span>archaeme</span></a></span>, that is @daggsy. Also hello <span><a href='#{archaeme_remote.ap_id}'>@<span>archaeme</span></a></span>"
+
+      assert expected_text == Formatter.finalize({subs, text})
     end
   end
 
index 6b0b182a3ce0e2820aa494a27dd46ca2d977984b..7d578a751deda8b996a8d2284247f794756395f8 100644 (file)
@@ -34,7 +34,8 @@ defmodule Pleroma.Web.TwitterAPI.TwitterAPITest do
 
     { :ok, activity = %Activity{} } = TwitterAPI.create_status(user, input)
 
-    assert get_in(activity.data, ["object", "content"]) == "Hello again, <span><a href='shp'>@<span>shp</span></a></span>.&lt;script&gt;&lt;/script&gt;<br>This is on another :moominmamma: line. #2hu #epic #phantasmagoric<br><a href=\"http://example.org/image.jpg\" class='attachment'>image.jpg</a>"
+    expected_text = "Hello again, <span><a href='shp'>@<span>shp</span></a></span>.&lt;script&gt;&lt;/script&gt;<br>This is on another :moominmamma: line. #<a href='http://localhost:4001/tag/2hu' rel='tag'>2hu</a> #<a href='http://localhost:4001/tag/epic' rel='tag'>epic</a> #<a href='http://localhost:4001/tag/phantasmagoric' rel='tag'>phantasmagoric</a><br><a href=\"http://example.org/image.jpg\" class='attachment'>image.jpg</a>"
+    assert get_in(activity.data, ["object", "content"]) == expected_text
     assert get_in(activity.data, ["object", "type"]) == "Note"
     assert get_in(activity.data, ["object", "actor"]) == user.ap_id
     assert get_in(activity.data, ["actor"]) == user.ap_id
@@ -282,19 +283,6 @@ defmodule Pleroma.Web.TwitterAPI.TwitterAPITest do
     assert is_binary(response)
   end
 
-  test "it adds user links to an existing text" do
-    text = "@gsimg According to @archaeme, that is @daggsy. Also hello @archaeme@archae.me"
-
-    gsimg = insert(:user, %{nickname: "gsimg"})
-    archaeme = insert(:user, %{nickname: "archaeme"})
-    archaeme_remote = insert(:user, %{nickname: "archaeme@archae.me"})
-
-    mentions = Pleroma.Formatter.parse_mentions(text)
-    expected_text = "<span><a href='#{gsimg.ap_id}'>@<span>gsimg</span></a></span> According to <span><a href='#{archaeme.ap_id}'>@<span>archaeme</span></a></span>, that is @daggsy. Also hello <span><a href='#{archaeme_remote.ap_id}'>@<span>archaeme</span></a></span>"
-
-    assert Utils.add_user_links(text, mentions) == expected_text
-  end
-
   test "it favorites a status, returns the updated status" do
     user = insert(:user)
     note_activity = insert(:note_activity)