%%%-------------------------------------------------------------------
%%% @author licheng5
%%% @copyright (C) 2020, <COMPANY>
%%% @doc
%%%
%%% @end
%%% Created : 26. 4月 2020 3:36 下午
%%%-------------------------------------------------------------------
-module(container_handler).
-author("licheng5").
-include("iot.hrl").

-define(REQ_TIMEOUT, 10000).

%% API
-export([handle_request/4]).

handle_request("GET", "/container/get_all", #{<<"uuid">> := UUID}, _) when is_binary(UUID) ->
    %% 检查ConfigJson是否是合法的json字符串
    case iot_host:get_pid(UUID) of
        undefined ->
            {ok, 200, iot_util:json_error(-1, <<"host not found">>)};
        Pid when is_pid(Pid) ->
            case iot_host:get_containers(Pid) of
                {ok, Ref} ->
                    case iot_host:await_reply(Ref, ?REQ_TIMEOUT) of
                        {ok, Result} ->
                            {ok, 200, iot_util:json_data(Result)};
                        {error, Reason} ->
                            {ok, 200, iot_util:json_error(-1, Reason)}
                    end;
                {error, Reason} when is_binary(Reason) ->
                    {ok, 200, iot_util:json_error(-1, Reason)}
            end
    end;

%% 下发config.json, 微服务接受后，保存服务配置
handle_request("POST", "/container/push_config", _,
    #{<<"uuid">> := UUID, <<"container_name">> := ContainerName, <<"config">> := Config, <<"timeout">> := Timeout0})
    when is_binary(UUID), is_binary(ContainerName), is_binary(Config), is_integer(Timeout0) ->

    %% 检查ConfigJson是否是合法的json字符串
    true = iot_util:is_json(Config),
    case iot_host:get_pid(UUID) of
        undefined ->
            {ok, 200, iot_util:json_error(-1, <<"host not found">>)};
        Pid when is_pid(Pid) ->
            Timeout = Timeout0 * 1000,
            case iot_host:config_container(Pid, ContainerName, Config) of
                {ok, Ref} ->
                    case iot_host:await_reply(Ref, Timeout) of
                        {ok, Result} ->
                            {ok, 200, iot_util:json_data(Result)};
                        {error, Reason} ->
                            {ok, 200, iot_util:json_error(-1, Reason)}
                    end;
                {error, Reason} when is_binary(Reason) ->
                    {ok, 200, iot_util:json_error(-1, Reason)}
            end
    end;

%% 部署微服务
handle_request("POST", "/container/deploy", _, #{<<"uuid">> := UUID, <<"task_id">> := TaskId, <<"config">> := Config})
    when is_binary(UUID), is_integer(TaskId), is_map(Config) ->

    case iot_host:get_pid(UUID) of
        undefined ->
            {ok, 200, iot_util:json_error(404, <<"host not found">>)};
        Pid when is_pid(Pid) ->
            case iot_host:deploy_container(Pid, TaskId, Config) of
                {ok, Ref} ->
                    case iot_host:await_reply(Ref, ?REQ_TIMEOUT) of
                        {ok, Result} ->
                            {ok, 200, iot_util:json_data(Result)};
                        {error, Reason} ->
                            {ok, 200, iot_util:json_error(400, Reason)}
                    end;
                {error, Reason} when is_binary(Reason) ->
                    {ok, 200, iot_util:json_error(400, Reason)}
            end
    end;

%% 启动服务
handle_request("POST", "/container/start", _, #{<<"uuid">> := UUID, <<"container_name">> := ContainerName}) when is_binary(UUID), is_binary(ContainerName) ->
    case iot_host:get_pid(UUID) of
        undefined ->
            {ok, 200, iot_util:json_error(404, <<"host not found">>)};
        Pid when is_pid(Pid) ->
            case iot_host:start_container(Pid, ContainerName) of
                {ok, Ref} ->
                    case iot_host:await_reply(Ref, ?REQ_TIMEOUT) of
                        {ok, Result} ->
                            {ok, 200, iot_util:json_data(Result)};
                        {error, Reason} ->
                            {ok, 200, iot_util:json_error(400, Reason)}
                    end;
                {error, Reason} when is_binary(Reason) ->
                    {ok, 200, iot_util:json_error(400, Reason)}
            end
    end;

%% 停止服务
handle_request("POST", "/container/stop", _, #{<<"uuid">> := UUID, <<"container_name">> := ContainerName}) when is_binary(UUID), is_binary(ContainerName) ->
    case iot_host:get_pid(UUID) of
        undefined ->
            {ok, 200, iot_util:json_error(404, <<"host not found">>)};
        Pid when is_pid(Pid) ->
            case iot_host:stop_container(Pid, ContainerName) of
                {ok, Ref} ->
                    case iot_host:await_reply(Ref, ?REQ_TIMEOUT) of
                        {ok, Result} ->
                            {ok, 200, iot_util:json_data(Result)};
                        {error, Reason} ->
                            {ok, 200, iot_util:json_error(400, Reason)}
                    end;
                {error, Reason} when is_binary(Reason) ->
                    {ok, 200, iot_util:json_error(400, Reason)}
            end
    end;

handle_request("POST", "/container/kill", _, #{<<"uuid">> := UUID, <<"container_name">> := ContainerName}) when is_binary(UUID), is_binary(ContainerName) ->
    case iot_host:get_pid(UUID) of
        undefined ->
            {ok, 200, iot_util:json_error(404, <<"host not found">>)};
        Pid when is_pid(Pid) ->
            case iot_host:kill_container(Pid, ContainerName) of
                {ok, Ref} ->
                    case iot_host:await_reply(Ref, ?REQ_TIMEOUT) of
                        {ok, Result} ->
                            {ok, 200, iot_util:json_data(Result)};
                        {error, Reason} ->
                            {ok, 200, iot_util:json_error(400, Reason)}
                    end;
                {error, Reason} when is_binary(Reason) ->
                    {ok, 200, iot_util:json_error(400, Reason)}
            end
    end;

%% 删除容器
handle_request("POST", "/container/remove", _, #{<<"uuid">> := UUID, <<"container_name">> := ContainerName}) when is_binary(UUID), is_binary(ContainerName) ->
    case iot_host:get_pid(UUID) of
        undefined ->
            {ok, 200, iot_util:json_error(404, <<"host not found">>)};
        Pid when is_pid(Pid) ->
            case iot_host:remove_container(Pid, ContainerName) of
                {ok, Ref} ->
                    case iot_host:await_reply(Ref, ?REQ_TIMEOUT) of
                        {ok, Result} ->
                            {ok, 200, iot_util:json_data(Result)};
                        {error, Reason} ->
                            {ok, 200, iot_util:json_error(400, Reason)}
                    end;
                {error, Reason} when is_binary(Reason) ->
                    {ok, 200, iot_util:json_error(400, Reason)}
            end
    end;

%handle_request("POST", "/container/task_log", _, #{<<"uuid">> := UUID, <<"task_id">> := TaskId}) when is_binary(UUID), is_integer(TaskId) ->
%    case iot_host:get_pid(UUID) of
%        undefined ->
%            {ok, 200, iot_util:json_error(404, <<"host not found">>)};
%        Pid when is_pid(Pid) ->
%            case iot_host:task_log(Pid, TaskId) of
%                {ok, Ref} ->
%                    case iot_host:await_reply(Ref, 5000) of
%                        {ok, Result} ->
%                            {ok, 200, iot_util:json_data(Result)};
%                        {error, Reason} ->
%                            {ok, 200, iot_util:json_error(400, Reason)}
%                    end;
%                {error, Reason} when is_binary(Reason) ->
%                    {ok, 200, iot_util:json_error(400, Reason)}
%            end
%    end;

handle_request(_, Path, _, _) ->
    Path1 = list_to_binary(Path),
    {ok, 200, iot_util:json_error(-1, <<"url: ", Path1/binary, " not found">>)}.

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%% helper methods
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%