maildir_commander

-module(mc_local_server).

-include_lib("kernel/include/logger.hrl").

%% this is the unix domain socket server

-export([start_link/0]).
-export([init/0, service/1]).

start_link() ->
    Pid = spawn_link(?MODULE, init, []),
    {ok, Pid}.

init() ->
    register(?MODULE, self()),
    File = mc_configer:default(socket_file),
    file:delete(File),
    {ok, LSock} = gen_tcp:listen(0, [binary, {packet, raw}, {active, false},
				     {exit_on_close, false}, {ip, {local, File}}]),
    accept_loop(LSock),
    exit(shutdown).

accept_loop(LSock) ->
    case gen_tcp:accept(LSock) of
	{ok, Sock} ->
	    spawn(?MODULE, service, [Sock]),
	    accept_loop(LSock);
	{error, closed} -> ok;
	{error, Reason} -> error(Reason)
    end.

service(Sock) ->
    ?LOG_NOTICE("Client connected"),
    service_loop([], Sock).

service_loop(Buffer, Sock) ->
    case mc_sexp:parse_incomplete(Buffer) of
	incomplete ->
	    case gen_tcp:recv(Sock, 0) of
		{ok, Packet} -> service_loop(Buffer ++ [Packet], Sock);
		{error, _Reason} ->
		    ?LOG_NOTICE("Client disconnected"),
		    gen_tcp:shutdown(Sock, write)
	    end;
	{[], Remain} -> service_loop([Remain], Sock);
	{[quit | _], _Remain} ->
	    gen_tcp:shutdown(Sock, write);
	%% index is special because we need to print progress
	{[mc, <<"index">> | _], _Remain} ->
	    ok = mc_server:command(mc_mu_api:index()),
	    wait_index_loop(Sock),
	    gen_tcp:shutdown(Sock, write);
	%% copy is special due to reading
	{[mc, <<"copy">>, Subject], Remain} ->
	    Data = unicode:characters_to_binary(read_to_end(Sock, [Remain])),
	    gen_tcp:send(Sock, put_pasteboard(Subject, Data)),
	    gen_tcp:shutdown(Sock, write);
	{[mc, Command | Args], _Remain} ->
	    gen_tcp:send(Sock, issue_command(Command, Args)),
	    gen_tcp:shutdown(Sock, write);
	{Command, Remain} ->
	    ?LOG_DEBUG("received Command: ~ts", [mc_sexp:to_string(Command)]),
	    ok = mc_server:command(Command),
	    send_sexp_loop(mc_mu_api:fun_ending(Command), Sock),
	    service_loop([Remain], Sock)
    end.

issue_command(<<"contacts">>, _Args) ->
    case maildir_commander:contacts() of
	{error, Message} ->
	    io_lib:format("error: ~ts~n", [Message]);
	{ok, Contacts} ->
	    lists:map(
	      fun ([Name | Email]) ->
		      [mc_sexp:escape(Name), <<" <">>, Email, <<">\n">>]
	      end, Contacts)
    end;
issue_command(<<"index">>, _Args) ->
    case maildir_commander:index() of
	{error, Message} ->
	    io_lib:format("error: ~ts~n", [Message]);
	{ok, Num} ->
	    io_lib:format("~B messages indexed~n", [Num])
    end;
issue_command(<<"find">>, [Query]) ->
    case string:to_integer(Query) of
	{Int, <<"">>} ->
	    print_tree(maildir_commander:find(Int));
	_->
	    print_tree(maildir_commander:find(Query))
    end;
issue_command(<<"findx">>, [Query]) ->
    print_tree(maildir_commander:find(Query, true));
issue_command(<<"scrub">>, [Path]) ->
    print_status(maildir_commander:scrub(unicode:characters_to_list(Path)));
issue_command(<<"orphan">>, [Path]) ->
    print_status(maildir_commander:orphan(unicode:characters_to_list(Path)));
issue_command(<<"graft">>, [Path, Parent]) ->
    print_status(maildir_commander:graft(unicode:characters_to_list(Path), Parent));
issue_command(<<"archive">>, _Args) ->
    <<"mc archive is obsolete. It is triggered automatically now.\n">>;
issue_command(<<"paste">>, _Args) ->
    case mc_configer:default(get_pasteboard) of
	undefined ->
	    <<"No pasteboard installed\n">>;
	{M,F,A} ->
	    apply(M, F, A)
    end;
issue_command(<<"send">>, [Name, Address]) ->
    case mc_configer:default(send_draft) of
	undefined ->
	    <<"No sender installed\n">>;
	{M,F,A} ->
	    case apply(M, F, [Name, Address | A]) of
		ok -> <<"Mail sent\n">>;
		{error, Msg} -> io_lib:format("error: ~ts~n", [Msg])
	    end
    end;
issue_command(Command, _Args) ->
    io_lib:format("Unknown command: ~ts~n", [Command]).

print_status({error, Message}) ->
    io_lib:format("error: ~ts~n", [Message]);
print_status(ok) -> "ok\n".

print_tree({error, Message}) ->
    io_lib:format("error: ~ts~n", [Message]);
print_tree({ok, Tree, Mails}) ->
    mc_tree:flatmap(fun(Docid, Level) ->
			    format_mail_header(maps:get(Docid, Mails), Level)
		    end, Tree).

format_mail_header(#{date := Date, from := From, subject := Subject}, Level) ->
    unicode:characters_to_binary(
      io_lib:format("~s~ts ~ts <~ts> ~ts~n",
		    [ lists:duplicate(Level*2, $\s),
		      calendar:system_time_to_rfc3339(Date),
		      hd(From), tl(From), Subject ])).

wait_index_loop(Sock) ->
    receive
	{async, [{<<"error">>, _Code}, {<<"message">>, Msg} | _ ]} ->
	    gen_tcp:send(Sock, io_lib:format("error: ~ts~n", [Msg]));
	{async, [{<<"info">>, index}, {<<"status">>, complete} | Rest ]} ->
	    gen_tcp:send(Sock, io_lib:format("~B messages indexed~n",
					     [proplists:get_value(<<"processed">>, Rest)]));
	{async, [{<<"info">>, index} | Rest ]} ->
	    gen_tcp:send(Sock, io_lib:format("~B messages indexed~n",
					     [proplists:get_value(<<"processed">>, Rest)])),
	    wait_index_loop(Sock)
    end.

send_sexp_loop(Test_done, Sock) ->
    receive
	{async, Sexp} ->
	    send_sexp(Sexp, Sock),
	    case Test_done(Sexp) of
		true -> ok;
		false  -> send_sexp_loop(Test_done, Sock)
	    end
    end.

send_sexp(Sexp, Sock) ->
    Serialized = mc_sexp:to_string(Sexp),
    ?LOG_DEBUG("sending response: ~ts", [Serialized]),
    Length = iolist_size(Serialized),
    Length_bin = integer_to_binary(Length, 16),
    gen_tcp:send(Sock, [<<16#FE, Length_bin/binary, 16#FF>>, Serialized]).

read_to_end(Sock, Buffer) ->
    case gen_tcp:recv(Sock, 0) of
	{ok, <<"">>} -> lists:reverse(Buffer);
	{ok, Packet} -> read_to_end(Sock, [Packet | Buffer]);
 	{error, closed} -> lists:reverse(Buffer);
	{error, Reason} -> error(Reason)
    end.

put_pasteboard(Subject, Data) ->
    case mc_configer:default(put_pasteboard) of
	undefined ->
	    "No pasteboard installed\n";
	{M,F,A} ->
	    ok = apply(M, F, [Subject, Data | A]),
	    io_lib:format("Copied ~B bytes to pasteboard\n", [byte_size(Data)])
    end.