% Licensed under the Apache License, Version 2.0 (the "License"); you may not
% use this file except in compliance with the License. You may obtain a copy of
% the License at
%
%   http://www.apache.org/licenses/LICENSE-2.0
%
% Unless required by applicable law or agreed to in writing, software
% distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
% WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
% License for the specific language governing permissions and limitations under
% the License.

-module(chttpd_stats).
-include("chttpd.hrl").
-include("../../couch/src/couch_stats.hrl").

-export([handle_stats_req/1]).
-import(chttpd,
    [send_json/2,send_json/3,send_json/4,send_method_not_allowed/2,
    start_json_response/2,send_chunk/2,end_json_response/1,
    start_chunked_response/3, send_error/4]).

-define(b2a(V), list_to_atom(binary_to_list(V))).

-record(stats_query_args, {
    range='0',
    flush=false
}).

handle_stats_req(#httpd{method='GET', path_parts=[_]}=Req) ->
    send_json(Req, couch_stats_aggregator:all());

handle_stats_req(#httpd{method='GET', path_parts=[_Stats, Module, Key]}=Req) ->
    #stats_query_args{
        range=Range,
        flush=Flush
    } = parse_stats_query(Req),

    case Flush of
        true ->
            couch_stats_aggregator:time_passed();
        _ -> ok
    end,

    Stats = couch_stats_aggregator:get_json({?b2a(Module), ?b2a(Key)}, Range),
    Response = {[{Module, {[{Key, Stats}]}}]},
    send_json(Req, Response);

handle_stats_req(Req) ->
    send_method_not_allowed(Req, "GET").

parse_stats_query(Req) ->
    lists:foldl(fun({Key,Value}, Args) ->
        case {Key, Value} of
        {"range", Range} ->
            Args#stats_query_args{range=list_to_atom(Range)};
        {"flush", "true"} ->
            Args#stats_query_args{flush=true};
        _Else -> % unknown key value pair, ignore.
            Args
        end
    end, #stats_query_args{}, chttpd:qs(Req)).