%% WARNING: DO NOT EDIT, AUTO-GENERATED CODE!
%% See https://github.com/aws-beam/aws-codegen for more details.

%% @doc Amazon Augmented AI is in preview release and is subject to change.
%%
%% We do not recommend using this product in production environments.
%%
%% Amazon Augmented AI (Amazon A2I) adds the benefit of human judgment to any
%% machine learning application. When an AI application can't evaluate data
%% with a high degree of confidence, human reviewers can take over. This
%% human review is called a human review workflow. To create and start a
%% human review workflow, you need three resources: a worker task template, a
%% flow definition, and a human loop.
%%
%% For information about these resources and prerequisites for using Amazon
%% A2I, see Get Started with Amazon Augmented AI in the Amazon SageMaker
%% Developer Guide.
%%
%% This API reference includes information about API actions and data types
%% that you can use to interact with Amazon A2I programmatically. Use this
%% guide to:
%%
%% <ul> <li> Start a human loop with the `StartHumanLoop' operation when
%% using Amazon A2I with a custom task type. To learn more about the
%% difference between custom and built-in task types, see Use Task Types . To
%% learn how to start a human loop using this API, see Create and Start a
%% Human Loop for a Custom Task Type in the Amazon SageMaker Developer Guide.
%%
%% </li> <li> Manage your human loops. You can list all human loops that you
%% have created, describe individual human loops, and stop and delete human
%% loops. To learn more, see Monitor and Manage Your Human Loop in the Amazon
%% SageMaker Developer Guide.
%%
%% </li> </ul> Amazon A2I integrates APIs from various AWS services to create
%% and start human review workflows for those services. To learn how Amazon
%% A2I uses these APIs, see Use APIs in Amazon A2I in the Amazon SageMaker
%% Developer Guide.
-module(aws_sagemaker_a2i_runtime).

-export([delete_human_loop/3,
         delete_human_loop/4,
         describe_human_loop/2,
         describe_human_loop/4,
         describe_human_loop/5,
         list_human_loops/2,
         list_human_loops/4,
         list_human_loops/5,
         start_human_loop/2,
         start_human_loop/3,
         stop_human_loop/2,
         stop_human_loop/3]).

-include_lib("hackney/include/hackney_lib.hrl").

%%====================================================================
%% API
%%====================================================================

%% @doc Deletes the specified human loop for a flow definition.
delete_human_loop(Client, HumanLoopName, Input) ->
    delete_human_loop(Client, HumanLoopName, Input, []).
delete_human_loop(Client, HumanLoopName, Input0, Options0) ->
    Method = delete,
    Path = ["/human-loops/", aws_util:encode_uri(HumanLoopName), ""],
    SuccessStatusCode = undefined,
    Options = [{send_body_as_binary, false},
               {receive_body_as_binary, false}
               | Options0],


    Headers = [],
    Input1 = Input0,

    CustomHeaders = [],
    Input2 = Input1,

    Query_ = [],
    Input = Input2,

    request(Client, Method, Path, Query_, CustomHeaders ++ Headers, Input, Options, SuccessStatusCode).

%% @doc Returns information about the specified human loop.
describe_human_loop(Client, HumanLoopName)
  when is_map(Client) ->
    describe_human_loop(Client, HumanLoopName, #{}, #{}).

describe_human_loop(Client, HumanLoopName, QueryMap, HeadersMap)
  when is_map(Client), is_map(QueryMap), is_map(HeadersMap) ->
    describe_human_loop(Client, HumanLoopName, QueryMap, HeadersMap, []).

describe_human_loop(Client, HumanLoopName, QueryMap, HeadersMap, Options0)
  when is_map(Client), is_map(QueryMap), is_map(HeadersMap), is_list(Options0) ->
    Path = ["/human-loops/", aws_util:encode_uri(HumanLoopName), ""],
    SuccessStatusCode = undefined,
    Options = [{send_body_as_binary, false},
               {receive_body_as_binary, false}
               | Options0],

    Headers = [],

    Query_ = [],

    request(Client, get, Path, Query_, Headers, undefined, Options, SuccessStatusCode).

%% @doc Returns information about human loops, given the specified
%% parameters.
%%
%% If a human loop was deleted, it will not be included.
list_human_loops(Client, FlowDefinitionArn)
  when is_map(Client) ->
    list_human_loops(Client, FlowDefinitionArn, #{}, #{}).

list_human_loops(Client, FlowDefinitionArn, QueryMap, HeadersMap)
  when is_map(Client), is_map(QueryMap), is_map(HeadersMap) ->
    list_human_loops(Client, FlowDefinitionArn, QueryMap, HeadersMap, []).

list_human_loops(Client, FlowDefinitionArn, QueryMap, HeadersMap, Options0)
  when is_map(Client), is_map(QueryMap), is_map(HeadersMap), is_list(Options0) ->
    Path = ["/human-loops"],
    SuccessStatusCode = undefined,
    Options = [{send_body_as_binary, false},
               {receive_body_as_binary, false}
               | Options0],

    Headers = [],

    Query0_ =
      [
        {<<"CreationTimeAfter">>, maps:get(<<"CreationTimeAfter">>, QueryMap, undefined)},
        {<<"CreationTimeBefore">>, maps:get(<<"CreationTimeBefore">>, QueryMap, undefined)},
        {<<"FlowDefinitionArn">>, FlowDefinitionArn},
        {<<"MaxResults">>, maps:get(<<"MaxResults">>, QueryMap, undefined)},
        {<<"NextToken">>, maps:get(<<"NextToken">>, QueryMap, undefined)},
        {<<"SortOrder">>, maps:get(<<"SortOrder">>, QueryMap, undefined)}
      ],
    Query_ = [H || {_, V} = H <- Query0_, V =/= undefined],

    request(Client, get, Path, Query_, Headers, undefined, Options, SuccessStatusCode).

%% @doc Starts a human loop, provided that at least one activation condition
%% is met.
start_human_loop(Client, Input) ->
    start_human_loop(Client, Input, []).
start_human_loop(Client, Input0, Options0) ->
    Method = post,
    Path = ["/human-loops"],
    SuccessStatusCode = undefined,
    Options = [{send_body_as_binary, false},
               {receive_body_as_binary, false}
               | Options0],


    Headers = [],
    Input1 = Input0,

    CustomHeaders = [],
    Input2 = Input1,

    Query_ = [],
    Input = Input2,

    request(Client, Method, Path, Query_, CustomHeaders ++ Headers, Input, Options, SuccessStatusCode).

%% @doc Stops the specified human loop.
stop_human_loop(Client, Input) ->
    stop_human_loop(Client, Input, []).
stop_human_loop(Client, Input0, Options0) ->
    Method = post,
    Path = ["/human-loops/stop"],
    SuccessStatusCode = undefined,
    Options = [{send_body_as_binary, false},
               {receive_body_as_binary, false}
               | Options0],


    Headers = [],
    Input1 = Input0,

    CustomHeaders = [],
    Input2 = Input1,

    Query_ = [],
    Input = Input2,

    request(Client, Method, Path, Query_, CustomHeaders ++ Headers, Input, Options, SuccessStatusCode).

%%====================================================================
%% Internal functions
%%====================================================================

-spec request(aws_client:aws_client(), atom(), iolist(), list(),
              list(), map() | undefined, list(), pos_integer() | undefined) ->
    {ok, Result, {integer(), list(), hackney:client()}} |
    {error, Error, {integer(), list(), hackney:client()}} |
    {error, term()} when
    Result :: map(),
    Error :: map().
request(Client, Method, Path, Query, Headers0, Input, Options, SuccessStatusCode) ->
    Client1 = Client#{service => <<"sagemaker">>},
    Host = build_host(<<"a2i-runtime.sagemaker">>, Client1),
    URL0 = build_url(Host, Path, Client1),
    URL = aws_request:add_query(URL0, Query),
    AdditionalHeaders = [ {<<"Host">>, Host}
                        , {<<"Content-Type">>, <<"application/x-amz-json-1.1">>}
                        ],
    Headers1 = aws_request:add_headers(AdditionalHeaders, Headers0),

    Payload =
      case proplists:get_value(send_body_as_binary, Options) of
        true ->
          maps:get(<<"Body">>, Input, <<"">>);
        false ->
          encode_payload(Input)
      end,

    MethodBin = aws_request:method_to_binary(Method),
    SignedHeaders = aws_request:sign_request(Client1, MethodBin, URL, Headers1, Payload),
    Response = hackney:request(Method, URL, SignedHeaders, Payload, Options),
    DecodeBody = not proplists:get_value(receive_body_as_binary, Options),
    handle_response(Response, SuccessStatusCode, DecodeBody).

handle_response({ok, StatusCode, ResponseHeaders, Client}, SuccessStatusCode, DecodeBody)
  when StatusCode =:= 200;
       StatusCode =:= 202;
       StatusCode =:= 204;
       StatusCode =:= SuccessStatusCode ->
    case hackney:body(Client) of
        {ok, <<>>} when StatusCode =:= 200;
                        StatusCode =:= SuccessStatusCode ->
            {ok, #{}, {StatusCode, ResponseHeaders, Client}};
        {ok, Body} ->
            Result = case DecodeBody of
                       true -> jsx:decode(Body);
                       false -> #{<<"Body">> => Body}
                     end,
            {ok, Result, {StatusCode, ResponseHeaders, Client}}
    end;
handle_response({ok, StatusCode, ResponseHeaders, Client}, _, _DecodeBody) ->
    {ok, Body} = hackney:body(Client),
    Error = jsx:decode(Body),
    {error, Error, {StatusCode, ResponseHeaders, Client}};
handle_response({error, Reason}, _, _DecodeBody) ->
  {error, Reason}.

build_host(_EndpointPrefix, #{region := <<"local">>, endpoint := Endpoint}) ->
    Endpoint;
build_host(_EndpointPrefix, #{region := <<"local">>}) ->
    <<"localhost">>;
build_host(EndpointPrefix, #{region := Region, endpoint := Endpoint}) ->
    aws_util:binary_join([EndpointPrefix, Region, Endpoint], <<".">>).

build_url(Host, Path0, Client) ->
    Proto = maps:get(proto, Client),
    Path = erlang:iolist_to_binary(Path0),
    Port = maps:get(port, Client),
    aws_util:binary_join([Proto, <<"://">>, Host, <<":">>, Port, Path], <<"">>).

-spec encode_payload(undefined | map()) -> binary().
encode_payload(undefined) ->
  <<>>;
encode_payload(Input) ->
  jsx:encode(Input).