xmpp.chapril.org-ejabberd/src/mod_stats.erl

%%%----------------------------------------------------------------------
%%% File    : mod_stats.erl
%%% Author  : Alexey Shchepin <alexey@sevcom.net>
%%% Purpose : 
%%% Created : 11 Jan 2003 by Alexey Shchepin <alexey@sevcom.net>
%%% Id      : $Id$
%%%----------------------------------------------------------------------

-module(mod_stats).
-author('alexey@sevcom.net').
-vsn('$Revision$ ').

-export([start/0,
	 process_local_iq/3]).

-include("namespaces.hrl").

start() ->
    ejabberd_local:register_iq_handler(?NS_STATS, ?MODULE, process_local_iq).


process_local_iq(From, To, {iq, ID, Type, XMLNS, SubEl}) ->
    Lang = xml:get_tag_attr_s("xml:lang", SubEl),
    case Type of
	set ->
	    {iq, ID, error, XMLNS, [SubEl, {xmlelement, "error",
					    [{"code", "405"}],
					    [{xmlcdata, "Not Allowed"}]}]};
	get ->
	    {xmlelement, _, Attrs, Els} = SubEl,
	    Node = string:tokens(xml:get_tag_attr_s("node", SubEl), "/"),
	    Names = get_names(Els, []),
	    
	    {T, Res} = get_local_stats(Node, Names),
	    case T of
		result ->
		    {iq, ID, result, ?NS_STATS, Res};
		error ->
		    {iq, ID, error, ?NS_STATS, [SubEl ++ Res]}
	    end

	    %case Node of
		%[] ->
		%    {iq, ID, result, XMLNS,
		%     [{xmlelement,
		%       "query",
		%       [{"xmlns", ?NS_STATS}],
		%       [{xmlelement, "stat", [{"name", "time/uptime"}], []},
		%	{xmlelement, "stat", [{"name", "time/cputime"}], []}
		%       ]}]};
		%["online users"] ->
		%    {iq, ID, result, XMLNS,
		%     [{xmlelement, "query", [{"xmlns", ?NS_DISCO_ITEMS}],
		%       get_online_users()
		%      }]};
		%["all users"] ->
		%    {iq, ID, result, XMLNS,
		%     [{xmlelement, "query", [{"xmlns", ?NS_DISCO_ITEMS}],
		%       get_all_users()
		%      }]};
		%["outgoing s2s"] ->
		%    {iq, ID, result, XMLNS,
		%     [{xmlelement, "query", [{"xmlns", ?NS_DISCO_ITEMS}],
		%       get_outgoing_s2s(Lang)
		%      }]};
		%["outgoing s2s", Host] ->
		%    {iq, ID, result, XMLNS,
		%     [{xmlelement, "query", [{"xmlns", ?NS_DISCO_ITEMS}],
		%       get_outgoing_s2s(Lang, Host)
		%      }]};
		%_ ->
		%    {iq, ID, error, XMLNS,
		%     [SubEl, {xmlelement, "error",
		%	      [{"code", "501"}],
		%	      [{xmlcdata, "Not Implemented"}]}]}
	    %end
    end.


get_names([], Res) ->
    Res;
get_names([{xmlelement, "stat", Attrs, _} | Els], Res) ->
    Name = xml:get_attr_s("name", Attrs),
    case Name of
	"" ->
	    get_names(Els, Res);
	_ ->
	    get_names(Els, [Name | Res])
    end;
get_names([_ | Els], Res) ->
    get_names(Els, Res).


get_local_stats([], []) ->
    {result,
     [{xmlelement,
       "query",
       [{"xmlns", ?NS_STATS}],
       [{xmlelement, "stat", [{"name", "time/uptime"}], []},
	{xmlelement, "stat", [{"name", "time/cputime"}], []},
	{xmlelement, "stat", [{"name", "users/online"}], []},
	{xmlelement, "stat", [{"name", "users/total"}], []}
       ]}]};
get_local_stats([], Names) ->
    {result,
     [{xmlelement,
       "query",
       [{"xmlns", ?NS_STATS}],
       lists:map(fun(Name) -> get_local_stat([], Name) end, Names)
      }]};
get_local_stats(_, _) ->
    {error,
     [{xmlelement, "error",
       [{"code", "501"}],
       [{xmlcdata, "Not Implemented"}]}]}.


-define(STAT(Val, Unit),
	{xmlelement, "stat",
	 [{"name", Name},
	  {"units", Unit},
	  {"value", Val}
	 ], []}).

-define(STATERR(Code, Desc),
	{xmlelement, "stat",
	 [{"name", Name}],
	 [{xmlelement, "error",
	   [{"code", Code}],
	   [{xmlcdata, Desc}]}]}).


get_local_stat([], Name) when Name == "time/uptime" ->
    ?STAT(io_lib:format("~.3f", [element(1, statistics(wall_clock))/1000]),
	  "seconds");
get_local_stat([], Name) when Name == "time/cputime" ->
    ?STAT(io_lib:format("~.3f", [element(1, statistics(runtime))/1000]),
	  "seconds");
get_local_stat([], Name) when Name == "users/online" ->
    case catch ejabberd_sm:dirty_get_sessions_list() of
	{'EXIT', Reason} ->
	    ?STATERR("500", "Internal Server Error");
	Users ->
	    ?STAT(integer_to_list(length(Users)), "users")
    end;
get_local_stat([], Name) when Name == "users/total" ->
    case catch ejabberd_auth:dirty_get_registered_users() of
	{'EXIT', Reason} ->
	    ?STATERR("500", "Internal Server Error");
	Users ->
	    ?STAT(integer_to_list(length(Users)), "users")
    end;
get_local_stat(_, Name) ->
    ?STATERR("404", "Not Found").