flm01/web/api/flukso/src/flukso_resource.erl

120 lines
4.0 KiB
Erlang
Raw Normal View History

2010-02-25 17:45:11 +00:00
%% @author icarus75 <bart.vandermeerssche@flukso.net>
%% @copyright 2009-2010 flukso.net
%% @doc Flukso webmachine_resource.
-module(flukso_resource).
-export([init/1, allowed_methods/2, malformed_request/2, is_authorized/2, content_types_provided/2, to_json/2]).
2010-02-25 17:45:11 +00:00
-include_lib("webmachine/include/webmachine.hrl").
-record(state,
{rrdSensor,
rrdTime,
rrdFactor,
token,
jsonpCallback}).
2010-02-25 17:45:11 +00:00
init([]) ->
{ok, undefined}.
allowed_methods(ReqData, State) ->
{['GET'], ReqData, State}.
malformed_request(ReqData, _State) ->
2010-04-27 21:23:27 +00:00
{RrdSensor, ValidSensor} = check_sensor(wrq:path_info(sensor, ReqData)),
{RrdTime, ValidInterval} = check_time(wrq:get_qs_value("interval", ReqData)),
{RrdFactor, ValidUnit} = check_unit(wrq:get_qs_value("unit", ReqData)),
2010-04-27 21:23:27 +00:00
{Token, ValidToken} = check_token(wrq:get_req_header("X-Token", ReqData), wrq:get_qs_value("token", ReqData)),
{JsonpCallback, ValidJsonpCallback} = check_jsonp_callback(wrq:get_qs_value("jsonp_callback", ReqData)),
2010-02-25 17:45:11 +00:00
State = #state{rrdSensor = RrdSensor, rrdTime = RrdTime, rrdFactor = RrdFactor, token = Token, jsonpCallback = JsonpCallback},
{case {ValidSensor, ValidInterval, ValidUnit, ValidToken, ValidJsonpCallback} of
{true, true, true, true, true} -> false;
2010-02-25 17:45:11 +00:00
_ -> true
end,
ReqData, State}.
2010-02-25 17:45:11 +00:00
is_authorized(ReqData, #state{rrdSensor = RrdSensor, token = Token} = State) ->
{data, Result} = mysql:execute(pool, permissions, [RrdSensor, Token]),
{case mysql:get_result_rows(Result) of
[[62]] -> true;
_Permission -> "access refused"
end,
ReqData, State}.
2010-02-25 17:45:11 +00:00
content_types_provided(ReqData, State) ->
{[{"application/json", to_json}], ReqData, State}.
to_json(ReqData, #state{rrdSensor = RrdSensor, rrdTime = RrdTime, rrdFactor = RrdFactor, jsonpCallback = JsonpCallback} = State) ->
case wrq:get_qs_value("interval", ReqData) of
2010-02-25 17:45:11 +00:00
"night" -> Path = "var/data/night/";
_Interval -> Path = "var/data/base/"
end,
case erlrrd:fetch(erlrrd:c([[Path, [RrdSensor|".rrd"]], "AVERAGE", ["-s",RrdTime]])) of
2010-02-25 17:45:11 +00:00
{ok, Response} ->
Filtered = [re:split(X, "[:][ ]", [{return,list}]) || [X] <- Response, string:str(X, ":") == 11],
Datapoints = [[list_to_integer(X), round(list_to_float(Y) * RrdFactor)] || [X, Y] <- Filtered, string:len(Y) /= 3],
2010-02-25 17:45:11 +00:00
Nans = [[list_to_integer(X), list_to_binary(Y)] || [X, Y] <- Filtered, string:len(Y) == 3],
Final = mochijson2:encode(lists:merge(Datapoints, Nans)),
{case JsonpCallback of
undefined -> Final;
_ -> [JsonpCallback, "(", Final, ");"]
end,
ReqData, State};
2010-02-25 17:45:11 +00:00
{error, _Reason} ->
2010-02-25 17:45:11 +00:00
{{halt, 404}, ReqData, State}
end.
2010-04-27 21:23:27 +00:00
%% internal functions
check_sensor(Sensor) ->
2010-02-25 17:45:11 +00:00
case re:run(Sensor, "[0-9a-f]+", []) of
{match, [{0,32}]} -> {Sensor, true};
_ -> {false, false}
end.
2010-04-27 21:23:27 +00:00
check_time(Interval) ->
2010-02-25 17:45:11 +00:00
Intervals = [{"hour", "end-1h"},
{"day", "end-1d"},
{"month", "end-30d"},
{"year", "end-1y"},
{"night", "end-30d"}],
case lists:keyfind(Interval, 1, Intervals) of
false -> {false, false};
{_Interval, RrdTime} -> {RrdTime, true}
end.
check_unit(Unit) ->
2010-02-25 17:45:11 +00:00
Units = [{"watt", 3600},
{"kwhperyear", 31536},
{"eurperyear", 5676},
{"audperyear", 5991}],
case lists:keyfind(Unit, 1, Units) of
false -> {false, false};
{_Unit, RrdFactor} -> {RrdFactor, true}
end.
2010-04-27 21:23:27 +00:00
check_token(undefined, undefined) ->
{false, false};
2010-04-27 21:23:27 +00:00
check_token(Token, undefined) ->
check_sensor(Token);
check_token(undefined, Token) ->
check_sensor(Token);
check_token(_, _) ->
{false, false}.
2010-04-27 21:23:27 +00:00
check_jsonp_callback(undefined) ->
{undefined, true};
2010-04-27 21:23:27 +00:00
check_jsonp_callback(JsonpCallback) ->
Length = string:len(JsonpCallback),
case re:run(JsonpCallback, "[0-9a-zA-Z_]+", []) of
{match, [{0, Length}]} -> {JsonpCallback, true};
_ -> {false, false}
end.