Add gen_server based oauth_client module.
Project: http://git-wip-us.apache.org/repos/asf/couchdb-oauth/repo Commit: http://git-wip-us.apache.org/repos/asf/couchdb-oauth/commit/1bcd44f9 Tree: http://git-wip-us.apache.org/repos/asf/couchdb-oauth/tree/1bcd44f9 Diff: http://git-wip-us.apache.org/repos/asf/couchdb-oauth/diff/1bcd44f9 Branch: refs/heads/import Commit: 1bcd44f95277bddcaaeb8442b61ae77a4937ceb4 Parents: 2c625a0 Author: Tim Fletcher <[email protected]> Authored: Sat Sep 26 14:03:49 2009 +0100 Committer: Tim Fletcher <[email protected]> Committed: Sat Sep 26 14:03:49 2009 +0100 ---------------------------------------------------------------------- src/oauth_client.erl | 134 ++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 134 insertions(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/couchdb-oauth/blob/1bcd44f9/src/oauth_client.erl ---------------------------------------------------------------------- diff --git a/src/oauth_client.erl b/src/oauth_client.erl new file mode 100644 index 0000000..b6fd831 --- /dev/null +++ b/src/oauth_client.erl @@ -0,0 +1,134 @@ +-module(oauth_client). + +-behaviour(gen_server). + +-export([access_token_params/1, deauthorize/1, get/2, get/3, get_access_token/2, + get_access_token/3, get_request_token/2, get_request_token/3, start/1, start/2, + start_link/1, start_link/2, stop/1]). + +-export([init/1, handle_call/3, handle_cast/2, handle_info/2, code_change/3, terminate/2]). + +%%============================================================================ +%% API functions +%%============================================================================ + +start(Consumer) -> + gen_server:start(?MODULE, Consumer, []). + +start(ServerName, Consumer) -> + gen_server:start(ServerName, ?MODULE, Consumer, []). + +start_link(Consumer) -> + gen_server:start_link(?MODULE, Consumer, []). + +start_link(ServerName, Consumer) -> + gen_server:start_link(ServerName, ?MODULE, Consumer, []). + +get_request_token(Client, URL) -> + get_request_token(Client, URL, []). + +get_request_token(Client, URL, Params) -> + gen_server:call(Client, {get_request_token, URL, Params}). + +get_access_token(Client, URL) -> + get_access_token(Client, URL, []). + +get_access_token(Client, URL, Params) -> + gen_server:call(Client, {get_access_token, URL, Params}). + +get(Client, URL) -> + get(Client, URL, []). + +get(Client, URL, Params) -> + gen_server:call(Client, {get, URL, Params}). + +access_token_params(Client) -> + gen_server:call(Client, {access_token_params}). + +deauthorize(Client) -> + gen_server:cast(Client, deauthorize). + +stop(Client) -> + gen_server:cast(Client, stop). + +%%============================================================================ +%% Helper functions +%%============================================================================ + +oauth_get(URL, Params, Consumer, Token, TokenSecret) -> + Signed = oauth:signed_params("GET", URL, Params, Consumer, Token, TokenSecret), + {AuthorizationParams, QueryParams} = lists:partition(fun({K, _}) -> lists:prefix("oauth_", K) end, Signed), + Request = {oauth:uri(URL, QueryParams), [oauth:header(AuthorizationParams)]}, + http:request(get, Request, [{autoredirect, false}], []). + +%%============================================================================ +%% gen_server callbacks +%%============================================================================ + +init(Consumer) -> + {ok, {Consumer}}. + +handle_call({get_request_token, URL, Params}, _From, State={Consumer}) -> + case oauth_get(URL, Params, Consumer, "", "") of + {ok, Response} -> + case oauth_http:response_code(Response) of + 200 -> + RParams = oauth_http:response_params(Response), + {reply, {ok, oauth:token(RParams)}, {Consumer, RParams}}; + _ -> + {reply, Response, State} + end; + Error -> + {reply, Error, State} + end; +handle_call({get_access_token, URL, Params}, _From, State={Consumer, RParams}) -> + case oauth_get(URL, Params, Consumer, oauth:token(RParams), oauth:token_secret(RParams)) of + {ok, Response} -> + case oauth_http:response_code(Response) of + 200 -> + AParams = oauth_http:response_params(Response), + {reply, ok, {Consumer, RParams, AParams}}; + _ -> + {reply, Response, State} + end; + Error -> + {reply, Error, State} + end; +handle_call({get, URL, Params}, _From, State={Consumer, _RParams, AParams}) -> + case oauth_get(URL, Params, Consumer, oauth:token(AParams), oauth:token_secret(AParams)) of + {ok, Response={{_, Status, _}, Headers, Body}} -> + case Status of + 200 -> + ContentType = proplists:get_value("content-type", Headers, ""), + MediaType = hd(string:tokens(ContentType, ";")), + case lists:suffix("/xml", MediaType) orelse lists:suffix("+xml", MediaType) of + true -> + {XML, []} = xmerl_scan:string(Body), + {reply, {ok, Headers, XML}, State}; + false -> + {reply, {ok, Headers, Body}, State} + end; + _ -> + {reply, Response, State} + end; + Error -> + {reply, Error, State} + end; +handle_call({access_token_params}, _From, State={_Consumer, _RParams, AParams}) -> + {reply, AParams, State}. + +handle_cast(deauthorize, {Consumer, _RParams}) -> + {noreply, {Consumer}}; +handle_cast(deauthorize, {Consumer, _RParams, _AParams}) -> + {noreply, {Consumer}}; +handle_cast(stop, State) -> + {stop, normal, State}. + +handle_info(_Msg, State) -> + {noreply, State}. + +code_change(_OldVsn, State, _Extra) -> + {ok, State}. + +terminate(normal, _State) -> + ok.
