hackney

%%% -*- erlang -*-
%%%
%%% This file is part of hackney_lib released under the Apache 2 license.
%%% See the NOTICE for more information.
%%%
%%% Copyright 2007 Mochi Media, Inc.
%%% Copyright 2011 Thomas Burdick <thomas.burdick@gmail.com>
%%% Copyright 2013, Loïc Hoguin <essen@ninenines.eu>
%%% Copyright 2013-2014, Benoit Chesneau
%%%
-module(hackney_cookie).

-export([setcookie/3]).
-export([parse_cookie/1]).


-type cookie_option() :: {max_age, non_neg_integer()}
        | {domain, binary()} | {path, binary()}
        | {secure, boolean()} | {http_only, boolean()}.
-type cookie_opts() :: [cookie_option()].
-export_type([cookie_opts/0]).


%% @doc Convert a cookie name, value and options to its iodata form.
-spec setcookie(iodata(), iodata(), cookie_opts()) -> binary().
setcookie(Name, Value, Opts) ->
    nomatch = binary:match(iolist_to_binary(Name), [<<$=>>, <<$,>>, <<$;>>,
                                                    <<$\s>>, <<$\t>>, <<$\r>>,
                                                    <<$\n>>, <<$\013>>,
                                                    <<$\014>>]),
    nomatch = binary:match(iolist_to_binary(Value), [<<$,>>, <<$;>>,
                                                     <<$\s>>, <<$\t>>,
                                                     <<$\r>>, <<$\n>>,
                                                     <<$\013>>, <<$\014>>]),
    MaxAgeBin = case lists:keyfind(max_age, 1, Opts) of
        false -> <<>>;
        {_, 0} ->
            %% MSIE requires an Expires date in the past to delete a cookie.
            <<"; Expires=Thu, 01-Jan-1970 00:00:01 GMT; Max-Age=0">>;
        {_, MaxAge} when is_integer(MaxAge), MaxAge > 0 ->
            UTC = calendar:universal_time(),
            Secs = calendar:datetime_to_gregorian_seconds(UTC),
            Expires = calendar:gregorian_seconds_to_datetime(Secs + MaxAge),
            [<<"; Expires=">>, hackney_date:date_to_rfc2109(Expires),
             <<"; Max-Age=">>, integer_to_list(MaxAge)]
    end,
    DomainBin = case lists:keyfind(domain, 1, Opts) of
        false -> <<>>;
        {_, Domain} -> [<<"; Domain=">>, Domain]
    end,
    PathBin = case lists:keyfind(path, 1, Opts) of
        false -> <<>>;
        {_, Path} -> [<<"; Path=">>, Path]
    end,
    SecureBin = case lists:keyfind(secure, 1, Opts) of
        false -> <<>>;
        {_, true} -> <<"; Secure">>
    end,
    HttpOnlyBin = case lists:keyfind(http_only, 1, Opts) of
        false -> <<>>;
        {_, true} -> <<"; HttpOnly">>
    end,
    iolist_to_binary([Name, <<"=">>, Value, <<"; Version=1">>,
                      MaxAgeBin, DomainBin, PathBin, SecureBin,
                      HttpOnlyBin]).

%% @doc Parse a cookie header string and return a list of key/values.

-spec parse_cookie(binary()) -> [{binary(), binary()}] | {error, badarg}.
parse_cookie(Cookie) ->
    parse_cookie(Cookie, []).

parse_cookie(<<>>, Acc) ->
    lists:reverse(Acc);
parse_cookie(<< $\s, Rest/binary >>, Acc) ->
    parse_cookie(Rest, Acc);
parse_cookie(<< $\t, Rest/binary >>, Acc) ->
    parse_cookie(Rest, Acc);
parse_cookie(<< $,, Rest/binary >>, Acc) ->
    parse_cookie(Rest, Acc);
parse_cookie(<< $;, Rest/binary >>, Acc) ->
    parse_cookie(Rest, Acc);
parse_cookie(<< $$, Rest/binary >>, Acc) ->
    skip_cookie(Rest, Acc);
parse_cookie(Cookie, Acc) ->
    parse_cookie_name(Cookie, Acc, <<>>).

skip_cookie(<<>>, Acc) ->
    lists:reverse(Acc);
skip_cookie(<< $,, Rest/binary >>, Acc) ->
    parse_cookie(Rest, Acc);
skip_cookie(<< $;, Rest/binary >>, Acc) ->
    parse_cookie(Rest, Acc);
skip_cookie(<< _, Rest/binary >>, Acc) ->
    skip_cookie(Rest, Acc).

parse_cookie_name(<<>>, Acc, Name) when Name /= <<>> ->
    lists:reverse([{Name, true} | Acc]);
parse_cookie_name(<<>>, _, _) ->
    {error, badarg};
parse_cookie_name(<< $=, _/binary >>, _, <<>>) ->
    {error, badarg};
parse_cookie_name(<< $=, Rest/binary >>, Acc, Name) ->
    parse_cookie_value(Rest, Acc, Name, <<>>);
parse_cookie_name(<< $,, _/binary >>, _, _) ->
    {error, badarg};
parse_cookie_name(<< $;, Rest/binary >>, Acc, Name) ->
    parse_cookie(Rest, [{Name, true} | Acc]);
parse_cookie_name(<< $\s, _/binary >>, _, _) ->
    {error, badarg};
parse_cookie_name(<< $\t, _/binary >>, _, _) ->
    {error, badarg};
parse_cookie_name(<< $\r, _/binary >>, _, _) ->
    {error, badarg};
parse_cookie_name(<< $\n, _/binary >>, _, _) ->
    {error, badarg};
parse_cookie_name(<< $\013, _/binary >>, _, _) ->
    {error, badarg};
parse_cookie_name(<< $\014, _/binary >>, _, _) ->
    {error, badarg};
parse_cookie_name(<< C, Rest/binary >>, Acc, Name) ->
    parse_cookie_name(Rest, Acc, << Name/binary, C >>).

parse_cookie_value(<<>>, Acc, Name, Value) ->
    lists:reverse([{Name, parse_cookie_trim(Value)}|Acc]);
parse_cookie_value(<< $,, Rest/binary >>, Acc, Name, Value) ->
    parse_cookie(Rest, [{Name, parse_cookie_trim(Value)}|Acc]);
parse_cookie_value(<< $;, Rest/binary >>, Acc, Name, Value) ->
    parse_cookie(Rest, [{Name, parse_cookie_trim(Value)}|Acc]);
parse_cookie_value(<< $\t, _/binary >>, _, _, _) ->
    {error, badarg};
parse_cookie_value(<< $\r, _/binary >>, _, _, _) ->
    {error, badarg};
parse_cookie_value(<< $\n, _/binary >>, _, _, _) ->
    {error, badarg};
parse_cookie_value(<< $\013, _/binary >>, _, _, _) ->
    {error, badarg};
parse_cookie_value(<< $\014, _/binary >>, _, _, _) ->
    {error, badarg};
parse_cookie_value(<< C, Rest/binary >>, Acc, Name, Value) ->
    parse_cookie_value(Rest, Acc, Name, << Value/binary, C >>).

parse_cookie_trim(Value = <<>>) ->
    Value;
parse_cookie_trim(Value) ->
    case binary:last(Value) of
        $\s ->
            Size = byte_size(Value) - 1,
            << Value2:Size/binary, _ >> = Value,
            parse_cookie_trim(Value2);
        _ ->
            Value
    end.