/modules/mod_oauth/deps/erlang-oauth/src/oauth_uri.erl

https://code.google.com/p/zotonic/ · Erlang · 91 lines · 71 code · 20 blank · 0 comment · 6 complexity · f6a832c0b0c745848e4b695da7bd26a9 MD5 · raw file

  1. -module(oauth_uri).
  2. -export([normalize/1, calate/2, encode/1, decode/1]).
  3. -export([params_from_string/1, params_to_string/1,
  4. params_from_header_string/1, params_to_header_string/1]).
  5. -import(lists, [concat/1]).
  6. -define(is_uppercase_alpha(C), C >= $A, C =< $Z).
  7. -define(is_lowercase_alpha(C), C >= $a, C =< $z).
  8. -define(is_alpha(C), ?is_uppercase_alpha(C); ?is_lowercase_alpha(C)).
  9. -define(is_digit(C), C >= $0, C =< $9).
  10. -define(is_alphanumeric(C), ?is_alpha(C); ?is_digit(C)).
  11. -define(is_unreserved(C), ?is_alphanumeric(C); C =:= $-; C =:= $_; C =:= $.; C =:= $~).
  12. -define(is_hex(C), ?is_digit(C); C >= $A, C =< $F).
  13. normalize(URI) ->
  14. case http_uri:parse(URI) of
  15. {Scheme, UserInfo, Host, Port, Path, _Query} ->
  16. normalize(Scheme, UserInfo, string:to_lower(Host), Port, [Path]);
  17. Else ->
  18. Else
  19. end.
  20. normalize(http, UserInfo, Host, 80, Acc) ->
  21. normalize(http, UserInfo, [Host|Acc]);
  22. normalize(https, UserInfo, Host, 443, Acc) ->
  23. normalize(https, UserInfo, [Host|Acc]);
  24. normalize(Scheme, UserInfo, Host, Port, Acc) ->
  25. normalize(Scheme, UserInfo, [Host, ":", Port|Acc]).
  26. normalize(Scheme, [], Acc) ->
  27. concat([Scheme, "://"|Acc]);
  28. normalize(Scheme, UserInfo, Acc) ->
  29. concat([Scheme, "://", UserInfo, "@"|Acc]).
  30. params_to_header_string(Params) ->
  31. intercalate(", ", [concat([encode(K), "=\"", encode(V), "\""]) || {K, V} <- Params]).
  32. params_from_header_string(String) ->
  33. [param_from_header_string(Param) || Param <- re:split(String, ",\\s*", [{return, list}])].
  34. param_from_header_string(Param) ->
  35. [Key, QuotedValue] = string:tokens(Param, "="),
  36. Value = string:substr(QuotedValue, 2, length(QuotedValue) - 2),
  37. {decode(Key), decode(Value)}.
  38. params_from_string(Params) ->
  39. [param_from_string(Param) || Param <- string:tokens(Params, "&")].
  40. param_from_string(Param) ->
  41. list_to_tuple([decode(Value) || Value <- string:tokens(Param, "=")]).
  42. params_to_string(Params) ->
  43. intercalate("&", [calate("=", [K, V]) || {K, V} <- Params]).
  44. calate(Sep, Xs) ->
  45. intercalate(Sep, [encode(X) || X <- Xs]).
  46. intercalate(Sep, Xs) ->
  47. concat(intersperse(Sep, Xs)).
  48. intersperse(_, []) -> [];
  49. intersperse(_, [X]) -> [X];
  50. intersperse(Sep, [X|Xs]) ->
  51. [X, Sep|intersperse(Sep, Xs)].
  52. decode(Chars) ->
  53. decode(Chars, []).
  54. decode([], Decoded) ->
  55. lists:reverse(Decoded);
  56. decode([$%,A,B|Etc], Decoded) when ?is_hex(A), ?is_hex(B) ->
  57. decode(Etc, [erlang:list_to_integer([A,B], 16)|Decoded]);
  58. decode([C|Etc], Decoded) ->
  59. decode(Etc, [C|Decoded]).
  60. encode(Chars) ->
  61. encode(Chars, []).
  62. encode([], Encoded) ->
  63. lists:flatten(lists:reverse(Encoded));
  64. encode([C|Etc], Encoded) when ?is_unreserved(C) ->
  65. encode(Etc, [C|Encoded]);
  66. encode([C|Etc], Encoded) when C >= 16 ->
  67. Value = [ $% | erlang:integer_to_list(C, 16) ],
  68. encode(Etc, [Value|Encoded]);
  69. encode([C|Etc], Encoded) when C < 16 ->
  70. Value = [ $%, $0 | erlang:integer_to_list(C, 16) ],
  71. encode(Etc, [Value|Encoded]).