95e4018c1a
Use Websockex to replace websocket_client Test that server will disconnect websocket upon token revocation Lint Execute session disconnect in background Refactor streamer test allow multi-streams rebase websocket change
38 lines
1.1 KiB
Elixir
38 lines
1.1 KiB
Elixir
# Pleroma: A lightweight social networking server
|
|
# Copyright © 2017-2021 Pleroma Authors <https://pleroma.social/>
|
|
# SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
defmodule Pleroma.Web.OAuth.Token.Strategy.Revoke do
|
|
@moduledoc """
|
|
Functions for dealing with revocation.
|
|
"""
|
|
|
|
alias Pleroma.Repo
|
|
alias Pleroma.Web.OAuth.App
|
|
alias Pleroma.Web.OAuth.Token
|
|
|
|
@doc "Finds and revokes access token for app and by token"
|
|
@spec revoke(App.t(), map()) :: {:ok, Token.t()} | {:error, :not_found | Ecto.Changeset.t()}
|
|
def revoke(%App{} = app, %{"token" => token} = _attrs) do
|
|
with {:ok, token} <- Token.get_by_token(app, token),
|
|
do: revoke(token)
|
|
end
|
|
|
|
@doc "Revokes access token"
|
|
@spec revoke(Token.t()) :: {:ok, Token.t()} | {:error, Ecto.Changeset.t()}
|
|
def revoke(%Token{} = token) do
|
|
with {:ok, token} <- Repo.delete(token) do
|
|
Task.Supervisor.start_child(
|
|
Pleroma.TaskSupervisor,
|
|
Pleroma.Web.Streamer,
|
|
:close_streams_by_oauth_token,
|
|
[token],
|
|
restart: :transient
|
|
)
|
|
|
|
{:ok, token}
|
|
else
|
|
result -> result
|
|
end
|
|
end
|
|
end
|