diff options
Diffstat (limited to 'lib')
| -rw-r--r-- | lib/pleroma/application.ex | 1 | ||||
| -rw-r--r-- | lib/pleroma/gopher/server.ex | 152 | ||||
| -rw-r--r-- | lib/pleroma/user.ex | 34 | ||||
| -rw-r--r-- | lib/pleroma/web/activity_pub/views/user_view.ex | 21 | 
4 files changed, 189 insertions, 19 deletions
| diff --git a/lib/pleroma/application.ex b/lib/pleroma/application.ex index 5e29a5d65..89826f515 100644 --- a/lib/pleroma/application.ex +++ b/lib/pleroma/application.ex @@ -24,6 +24,7 @@ defmodule Pleroma.Application do            ]          ]),          worker(Pleroma.Web.Federator, []), +        worker(Pleroma.Gopher.Server, []),          worker(Pleroma.Stats, [])        ] ++          if Mix.env() == :test, diff --git a/lib/pleroma/gopher/server.ex b/lib/pleroma/gopher/server.ex new file mode 100644 index 000000000..cfce4c05c --- /dev/null +++ b/lib/pleroma/gopher/server.ex @@ -0,0 +1,152 @@ +defmodule Pleroma.Gopher.Server do +  use GenServer +  require Logger +  @gopher Application.get_env(:pleroma, :gopher) + +  def start_link() do +    ip = Keyword.get(@gopher, :ip, {0, 0, 0, 0}) +    port = Keyword.get(@gopher, :port, 1234) +    GenServer.start_link(__MODULE__, [ip, port], []) +  end + +  def init([ip, port]) do +    if Keyword.get(@gopher, :enabled, false) do +      Logger.info("Starting gopher server on #{port}") + +      :ranch.start_listener( +        :gopher, +        100, +        :ranch_tcp, +        [port: port], +        __MODULE__.ProtocolHandler, +        [] +      ) + +      {:ok, %{ip: ip, port: port}} +    else +      Logger.info("Gopher server disabled") +      {:ok, nil} +    end +  end +end + +defmodule Pleroma.Gopher.Server.ProtocolHandler do +  alias Pleroma.Web.ActivityPub.ActivityPub +  alias Pleroma.User +  alias Pleroma.Activity +  alias Pleroma.Repo + +  @instance Application.get_env(:pleroma, :instance) +  @gopher Application.get_env(:pleroma, :gopher) + +  def start_link(ref, socket, transport, opts) do +    pid = spawn_link(__MODULE__, :init, [ref, socket, transport, opts]) +    {:ok, pid} +  end + +  def init(ref, socket, transport, _Opts = []) do +    :ok = :ranch.accept_ack(ref) +    loop(socket, transport) +  end + +  def info(text) do +    text = String.replace(text, ~r/[\t\n]/, "") +    String.split(text, "\r") +    |> Enum.map(fn (text) -> +      "i#{text}\tfake\(NULL)\t0\r\n" +    end) +    |> Enum.join("") +  end + +  def link(name, selector, type \\ 1) do +    address = Pleroma.Web.Endpoint.host() +    port = Keyword.get(@gopher, :port, 1234) +    "#{type}#{name}\t#{selector}\t#{address}\t#{port}\r\n" +  end + +  def response("") do +    info("Welcome to #{Keyword.get(@instance, :name, "Pleroma")}!") <> +      link("Public Timeline", "/main/public") <> +      link("Federated Timeline", "/main/all") <> ".\r\n" +  end + +  def render_activities(activities) do +    activities +    |> Enum.reverse() +    |> Enum.map(fn activity -> +      user = User.get_cached_by_ap_id(activity.data["actor"]) + +      object = activity.data["object"] +      like_count = object["like_count"] || 0 +      announcement_count = object["announcement_count"] || 0 + +      link("Post ##{activity.id} by #{user.nickname}", "/notices/#{activity.id}") <> +        info("#{like_count} likes, #{announcement_count} repeats") <> +        "\r\n" <> info(HtmlSanitizeEx.strip_tags(String.replace(activity.data["object"]["content"], "<br>", "\r"))) +    end) +    |> Enum.join("\r\n") +  end + +  def response("/main/public") do +    posts = +      ActivityPub.fetch_public_activities(%{"type" => ["Create"], "local_only" => true}) +      |> render_activities + +    info("Welcome to the Public Timeline!") <> posts <> ".\r\n" +  end + +  def response("/main/all") do +    posts = +      ActivityPub.fetch_public_activities(%{"type" => ["Create"]}) +      |> render_activities + +    info("Welcome to the Federated Timeline!") <> posts <> ".\r\n" +  end + +  def response("/notices/" <> id) do +    with %Activity{} = activity <- Repo.get(Activity, id), +         true <- ActivityPub.is_public?(activity) do +      activities = +        ActivityPub.fetch_activities_for_context(activity.data["context"]) +        |> render_activities + +      user = User.get_cached_by_ap_id(activity.data["actor"]) + +      info("Post #{activity.id} by #{user.nickname}") <> +        link("More posts by #{user.nickname}", "/users/#{user.nickname}") <> activities <> ".\r\n" +    else +      _e -> +        info("Not public") <> ".\r\n" +    end +  end + +  def response("/users/" <> nickname) do +    with %User{} = user <- User.get_cached_by_nickname(nickname) do +      params = %{ +        "type" => ["Create"], +        "actor_id" => user.ap_id +      } + +      activities = +        ActivityPub.fetch_public_activities(params) +        |> render_activities + +      info("Posts by #{user.nickname}") <> activities <> ".\r\n" +    else +      _e -> +        info("No such user") <> ".\r\n" +    end +  end + +  def loop(socket, transport) do +    case transport.recv(socket, 0, 5000) do +      {:ok, data} -> +        data = String.trim_trailing(data, "\r\n") +        transport.send(socket, response(data)) +        :ok = transport.close(socket) + +      _ -> +        :ok = transport.close(socket) +    end +  end +end diff --git a/lib/pleroma/user.ex b/lib/pleroma/user.ex index 55b290309..abeb169d9 100644 --- a/lib/pleroma/user.ex +++ b/lib/pleroma/user.ex @@ -278,24 +278,30 @@ defmodule Pleroma.User do      end    end -  def get_followers(%User{id: id, follower_address: follower_address}) do -    q = -      from( -        u in User, -        where: fragment("? <@ ?", ^[follower_address], u.following), -        where: u.id != ^id -      ) +  def get_followers_query(%User{id: id, follower_address: follower_address}) do +    from( +      u in User, +      where: fragment("? <@ ?", ^[follower_address], u.following), +      where: u.id != ^id +    ) +  end + +  def get_followers(user) do +    q = get_followers_query(user)      {:ok, Repo.all(q)}    end -  def get_friends(%User{id: id, following: following}) do -    q = -      from( -        u in User, -        where: u.follower_address in ^following, -        where: u.id != ^id -      ) +  def get_friends_query(%User{id: id, following: following}) do +    from( +      u in User, +      where: u.follower_address in ^following, +      where: u.id != ^id +    ) +  end + +  def get_friends(user) do +    q = get_friends_query(user)      {:ok, Repo.all(q)}    end diff --git a/lib/pleroma/web/activity_pub/views/user_view.ex b/lib/pleroma/web/activity_pub/views/user_view.ex index 997d8308f..92afd0872 100644 --- a/lib/pleroma/web/activity_pub/views/user_view.ex +++ b/lib/pleroma/web/activity_pub/views/user_view.ex @@ -3,9 +3,11 @@ defmodule Pleroma.Web.ActivityPub.UserView do    alias Pleroma.Web.Salmon    alias Pleroma.Web.WebFinger    alias Pleroma.User +  alias Pleroma.Repo    alias Pleroma.Web.ActivityPub.ActivityPub    alias Pleroma.Web.ActivityPub.Transmogrifier    alias Pleroma.Web.ActivityPub.Utils +  import Ecto.Query    def render("user.json", %{user: user}) do      {:ok, user} = WebFinger.ensure_keys_present(user) @@ -45,10 +47,11 @@ defmodule Pleroma.Web.ActivityPub.UserView do      |> Map.merge(Utils.make_json_ld_header())    end -  def collection(collection, iri, page) do +  def collection(collection, iri, page, total \\ nil) do      offset = (page - 1) * 10      items = Enum.slice(collection, offset, 10)      items = Enum.map(items, fn user -> user.ap_id end) +    total = total || length(collection)      map = %{        "id" => "#{iri}?page=#{page}", @@ -64,14 +67,18 @@ defmodule Pleroma.Web.ActivityPub.UserView do    end    def render("following.json", %{user: user, page: page}) do -    {:ok, following} = User.get_friends(user) +    query = User.get_friends_query(user) +    query = from(user in query, select: [:ap_id]) +    following = Repo.all(query)      collection(following, "#{user.ap_id}/following", page)      |> Map.merge(Utils.make_json_ld_header())    end    def render("following.json", %{user: user}) do -    {:ok, following} = User.get_friends(user) +    query = User.get_friends_query(user) +    query = from(user in query, select: [:ap_id]) +    following = Repo.all(query)      %{        "id" => "#{user.ap_id}/following", @@ -83,14 +90,18 @@ defmodule Pleroma.Web.ActivityPub.UserView do    end    def render("followers.json", %{user: user, page: page}) do -    {:ok, followers} = User.get_followers(user) +    query = User.get_followers_query(user) +    query = from(user in query, select: [:ap_id]) +    followers = Repo.all(query)      collection(followers, "#{user.ap_id}/followers", page)      |> Map.merge(Utils.make_json_ld_header())    end    def render("followers.json", %{user: user}) do -    {:ok, followers} = User.get_followers(user) +    query = User.get_followers_query(user) +    query = from(user in query, select: [:ap_id]) +    followers = Repo.all(query)      %{        "id" => "#{user.ap_id}/followers", | 
