diff options
author | Paweł Chmielowski <pawel@process-one.net> | 2022-04-20 20:07:46 +0300 |
---|---|---|
committer | Paweł Chmielowski <pawel@process-one.net> | 2022-04-20 20:07:52 +0300 |
commit | e59043db2cd8775dd4cf4c1c277b2eb1ec7846e0 (patch) | |
tree | f975302e846d942750b8bcf9f413f0ae08b0b14b /src/ejabberd_oauth.erl | |
parent | 87445bc74be490899e3878c758718ba74804e297 (diff) |
Make oauth_revoke_token work with all backends
Diffstat (limited to 'src/ejabberd_oauth.erl')
-rw-r--r-- | src/ejabberd_oauth.erl | 20 |
1 files changed, 14 insertions, 6 deletions
diff --git a/src/ejabberd_oauth.erl b/src/ejabberd_oauth.erl index 1d26798b0..2e812d758 100644 --- a/src/ejabberd_oauth.erl +++ b/src/ejabberd_oauth.erl @@ -65,6 +65,7 @@ -callback init() -> any(). -callback store(#oauth_token{}) -> ok | {error, any()}. -callback lookup(binary()) -> {ok, #oauth_token{}} | error. +-callback revoke(binary()) -> ok | {error, binary()}. -callback clean(non_neg_integer()) -> any(). -record(oauth_ctx, { @@ -99,12 +100,12 @@ get_commands_spec() -> result = {tokens, {list, {token, {tuple, [{token, string}, {user, string}, {scope, string}, {expires_in, string}]}}}} }, #ejabberd_commands{name = oauth_revoke_token, tags = [oauth], - desc = "Revoke authorization for a token (only Mnesia)", + desc = "Revoke authorization for a token", module = ?MODULE, function = oauth_revoke_token, - args = [{token, string}], + args = [{token, binary}], policy = restricted, - result = {tokens, {list, {token, {tuple, [{token, string}, {user, string}, {scope, string}, {expires_in, string}]}}}}, - result_desc = "List of remaining tokens" + result = {res, restuple}, + result_desc = "Result code" }, #ejabberd_commands{name = oauth_add_client_password, tags = [oauth], desc = "Add OAUTH client_id with password grant type", @@ -160,8 +161,15 @@ oauth_list_tokens() -> oauth_revoke_token(Token) -> - ok = mnesia:dirty_delete(oauth_token, list_to_binary(Token)), - oauth_list_tokens(). + DBMod = get_db_backend(), + case DBMod:revoke(Token) of + ok -> + ets_cache:delete(oauth_cache, Token, + ejabberd_cluster:get_nodes()), + {ok, ""}; + Other -> + Other + end. oauth_add_client_password(ClientID, ClientName, Secret) -> DBMod = get_db_backend(), |