Revert "oauth2 fixes (#177)"
[akkoma] / lib / pleroma / web / plugs / set_user_session_id_plug.ex
index 730c4ac742bb1fef3549dcbddd3e881ee20c4dad..a1cfa0915e8b9cd7f134ed6bd14df4dca375e91f 100644 (file)
@@ -1,18 +1,17 @@
 # Pleroma: A lightweight social networking server
-# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
+# Copyright © 2017-2021 Pleroma Authors <https://pleroma.social/>
 # SPDX-License-Identifier: AGPL-3.0-only
 
-defmodule Pleroma.Plugs.SetUserSessionIdPlug do
-  import Plug.Conn
-  alias Pleroma.User
+defmodule Pleroma.Web.Plugs.SetUserSessionIdPlug do
+  alias Pleroma.Helpers.AuthHelper
+  alias Pleroma.Web.OAuth.Token
 
   def init(opts) do
     opts
   end
 
-  def call(%{assigns: %{user: %User{id: id}}} = conn, _) do
-    conn
-    |> put_session(:user_id, id)
+  def call(%{assigns: %{token: %Token{} = oauth_token}} = conn, _) do
+    AuthHelper.put_session_token(conn, oauth_token.token)
   end
 
   def call(conn, _), do: conn