From 868a029e39ec8e9aa368da917146d088edee4d2f Mon Sep 17 00:00:00 2001
From: Magnus Ahltorp <map@kth.se>
Date: Sat, 25 Oct 2014 01:51:46 +0200
Subject: Move internal HTTP APIs to mochiweb. Stop using jiffy.

---
 src/frontend.erl | 135 +++++++++++++++++++++++++------------------------------
 1 file changed, 61 insertions(+), 74 deletions(-)

(limited to 'src/frontend.erl')

diff --git a/src/frontend.erl b/src/frontend.erl
index 8d0eccd..a8a8b9e 100644
--- a/src/frontend.erl
+++ b/src/frontend.erl
@@ -5,64 +5,59 @@
 
 -module(frontend).
 %% API (URL)
--export([sendlog/3, missingentries/3, sendentry/3, sendsth/3, currentposition/3]).
-
-sendentry(SessionID, _Env, Input) ->
-    R = (catch case (catch jiffy:decode(Input)) of
-                   {error, E} ->
-                       html("sendentry: bad input:", E);
-                   {PropList} ->
-                       LogEntry = base64:decode(proplists:get_value(<<"entry">>, PropList)),
-                       TreeLeafHash = base64:decode(proplists:get_value(<<"treeleafhash">>, PropList)),
-
-                       ok = db:add(TreeLeafHash, LogEntry),
-                       binary_to_list(
-                         jiffy:encode(
-                           {[{result, <<"ok">>}]}))
-               end),
-    deliver(SessionID, R).
-
-sendlog(SessionID, _Env, Input) ->
-    R = (catch case (catch jiffy:decode(Input)) of
-                   {error, E} ->
-                       html("sendentry: bad input:", E);
-                   {PropList} ->
-                       Start = proplists:get_value(<<"start">>, PropList),
-                       Hashes = lists:map(fun (S) -> base64:decode(S) end, proplists:get_value(<<"hashes">>, PropList)),
-                       
-                       Indices = lists:seq(Start, Start + length(Hashes) - 1),
-                       lists:foreach(fun ({Hash, Index}) ->
-                                             ok = db:add_index(Hash, Index)
-                                     end, lists:zip(Hashes, Indices)),
-                       binary_to_list(
-                         jiffy:encode(
-                           {[{result, <<"ok">>}]}))
-               end),
-    deliver(SessionID, R).
-
-sendsth(SessionID, _Env, Input) ->
-    R = (catch case (catch jiffy:decode(Input)) of
-                   {error, E} ->
-                       html("sendentry: bad input:", E);
-                   {PropList} ->
-                       Treesize = proplists:get_value(<<"tree_size">>, PropList),
-
-                       ok = db:set_treesize(Treesize),
-                       
-                       ht:reset_tree([db:size() - 1]),
-
-                       binary_to_list(
-                         jiffy:encode(
-                           {[{result, <<"ok">>}]}))
-               end),
-    deliver(SessionID, R).
-
-currentposition(SessionID, _Env, _Input) ->
+-export([request/3]).
+
+request(post, "ct/frontend/sendentry", Input) ->
+    case (catch mochijson2:decode(Input)) of
+        {error, E} ->
+            html("sendentry: bad input:", E);
+        {struct, PropList} ->
+            LogEntry = base64:decode(proplists:get_value(<<"entry">>, PropList)),
+            TreeLeafHash = base64:decode(proplists:get_value(<<"treeleafhash">>, PropList)),
+
+            ok = db:add(TreeLeafHash, LogEntry),
+            success({[{result, <<"ok">>}]})
+    end;
+
+request(post, "ct/frontend/sendlog", Input) ->
+    case (catch mochijson2:decode(Input)) of
+        {error, E} ->
+            html("sendentry: bad input:", E);
+        {struct, PropList} ->
+            Start = proplists:get_value(<<"start">>, PropList),
+            Hashes = lists:map(fun (S) -> base64:decode(S) end, proplists:get_value(<<"hashes">>, PropList)),
+
+            Indices = lists:seq(Start, Start + length(Hashes) - 1),
+            lists:foreach(fun ({Hash, Index}) ->
+                                  ok = db:add_index(Hash, Index)
+                          end, lists:zip(Hashes, Indices)),
+            success({[{result, <<"ok">>}]})
+    end;
+
+request(post, "ct/frontend/sendsth", Input) ->
+    case (catch mochijson2:decode(Input)) of
+        {error, E} ->
+            html("sendentry: bad input:", E);
+        {struct, PropList} ->
+            Treesize = proplists:get_value(<<"tree_size">>, PropList),
+
+            ok = db:set_treesize(Treesize),
+
+            ht:reset_tree([db:size() - 1]),
+
+            success({[{result, <<"ok">>}]})
+    end;
+
+request(get, "ct/frontend/currentposition", _Query) ->
+    Size = db:size(),
+    success({[{result, <<"ok">>},
+              {position, Size}]});
+
+request(get, "ct/frontend/missingentries", _Query) ->
     Size = db:size(),
-    R = binary_to_list(
-          jiffy:encode(
-            {[{result, <<"ok">>}, {position, Size}]})),
-    deliver(SessionID, R).
+    Missing = fetchmissingentries(Size),
+    success({[{result, <<"ok">>},
+              {entries, Missing}]}).
 
 fetchmissingentries(Index) ->
     lists:reverse(fetchmissingentries(Index, [])).
@@ -80,23 +75,15 @@ fetchmissingentries(Index, Acc) ->
             end
     end.
 
-missingentries(SessionID, _Env, _Input) ->
-    Size = db:size(),
-    Missing = fetchmissingentries(Size),
-    R = binary_to_list(
-          jiffy:encode(
-            {[{result, <<"ok">>}, {entries, Missing}]})),
-    deliver(SessionID, R).
 
 %% Private functions.
 html(Text, Input) ->
-    io_lib:format(
-      "Content-Type: text/html\r\n\r\n" ++
-          "<html><body><p>~n" ++
-          "~s~n" ++
-          "~p~n" ++
-          "</body></html>~n", [Text, Input]).
-
--spec deliver(any(), string()) -> ok | {error, _Reason}.
-deliver(Session, Data) ->
-    mod_esi:deliver(Session, Data).
+    {400, [{"Content-Type", "text/html"}],
+     io_lib:format(
+       "<html><body><p>~n" ++
+           "~s~n" ++
+           "~p~n" ++
+           "</body></html>~n", [Text, Input])}.
+
+success(Data) ->
+    {200, [{"Content-Type", "text/json"}], mochijson2:encode(Data)}.
-- 
cgit v1.1