% Licensed under the Apache License, Version 2.0 (the "License"); you may not
% use this file except in compliance with the License. You may obtain a copy of
% the License at
%
%   http://www.apache.org/licenses/LICENSE-2.0
%
% Unless required by applicable law or agreed to in writing, software
% distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
% WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
% License for the specific language governing permissions and limitations under
% the License.

-module(fabric).

-include_lib("mem3/include/mem3.hrl").
-include_lib("couch/include/couch_db.hrl").
-include_lib("couch_mrview/include/couch_mrview.hrl").

% DBs
-export([
    all_dbs/0, all_dbs/1,
    create_db/1, create_db/2,
    delete_db/1,
    delete_db/2,
    get_db_info/1,
    get_doc_count/1, get_doc_count/2,
    set_revs_limit/3,
    set_security/2, set_security/3,
    get_revs_limit/1,
    get_security/1, get_security/2,
    get_all_security/1, get_all_security/2,
    get_purge_infos_limit/1,
    set_purge_infos_limit/3,
    compact/1, compact/2,
    get_partition_info/2
]).

% Documents
-export([
    open_doc/3,
    open_revs/3, open_revs/4,
    get_doc_info/3,
    get_full_doc_info/3,
    get_missing_revs/2, get_missing_revs/3,
    update_doc/3,
    update_docs/3,
    purge_docs/3,
    att_receiver/3
]).

% Views
-export([
    all_docs/4, all_docs/5,
    changes/4,
    query_view/3, query_view/4, query_view/6, query_view/7,
    get_view_group_info/2,
    end_changes/0
]).

% miscellany
-export([
    design_docs/1,
    reset_validation_funs/1,
    cleanup_index_files/0,
    cleanup_index_files/1,
    cleanup_index_files_all_nodes/1,
    dbname/1,
    db_uuids/1
]).

-type dbname() :: (iodata() | tuple()).
-type docid() :: iodata().
-type revision() :: {integer(), binary()}.
-type callback() :: fun((any(), any()) -> {ok | stop, any()}).
-type json_obj() :: {[{binary() | atom(), any()}]}.
-type option() :: atom() | {atom(), any()}.

%% db operations
%% @equiv all_dbs(<<>>)
all_dbs() ->
    all_dbs(<<>>).

%% @doc returns a list of all database names
-spec all_dbs(Prefix :: iodata()) -> {ok, [binary()]}.
all_dbs(Prefix) when is_binary(Prefix) ->
    Length = byte_size(Prefix),
    MatchingDbs = mem3:fold_shards(
        fun(#shard{dbname = DbName}, Acc) ->
            case DbName of
                <<Prefix:Length/binary, _/binary>> ->
                    [DbName | Acc];
                _ ->
                    Acc
            end
        end,
        []
    ),
    {ok, lists:usort(MatchingDbs)};
%% @equiv all_dbs(list_to_binary(Prefix))
all_dbs(Prefix) when is_list(Prefix) ->
    all_dbs(list_to_binary(Prefix)).

%% @doc returns a property list of interesting properties
%%      about the database such as `doc_count', `disk_size',
%%      etc.
-spec get_db_info(dbname()) ->
    {ok, [
        {instance_start_time, binary()}
        | {doc_count, non_neg_integer()}
        | {doc_del_count, non_neg_integer()}
        | {purge_seq, non_neg_integer()}
        | {compact_running, boolean()}
        | {disk_size, non_neg_integer()}
        | {disk_format_version, pos_integer()}
    ]}.
get_db_info(DbName) ->
    fabric_db_info:go(dbname(DbName)).

%% @doc returns the size of a given partition
-spec get_partition_info(dbname(), Partition :: binary()) ->
    {ok, [
        {db_name, binary()}
        | {partition, binary()}
        | {doc_count, non_neg_integer()}
        | {doc_del_count, non_neg_integer()}
        | {sizes, json_obj()}
    ]}.
get_partition_info(DbName, Partition) ->
    fabric_db_partition_info:go(dbname(DbName), Partition).

%% @doc the number of docs in a database
%% @equiv get_doc_count(DbName, <<"_all_docs">>)
get_doc_count(DbName) ->
    get_doc_count(DbName, <<"_all_docs">>).

%% @doc the number of design docs in a database
-spec get_doc_count(dbname(), Namespace :: binary()) ->
    {ok, non_neg_integer() | null}
    | {error, atom()}
    | {error, atom(), any()}.
get_doc_count(DbName, <<"_all_docs">>) ->
    fabric_db_doc_count:go(dbname(DbName));
get_doc_count(DbName, <<"_design">>) ->
    fabric_design_doc_count:go(dbname(DbName));
get_doc_count(_DbName, <<"_local">>) ->
    {ok, null}.

%% @equiv create_db(DbName, [])
create_db(DbName) ->
    create_db(DbName, []).

%% @doc creates a database with the given name.
%%
%% Options can include values for q and n,
%% for example `{q, "8"}' and `{n, "3"}', which
%% control how many shards to split a database into
%% and how many nodes each doc is copied to respectively.
%%
-spec create_db(dbname(), [option()]) -> ok | accepted | {error, atom()}.
create_db(DbName, Options) ->
    fabric_db_create:go(dbname(DbName), opts(Options)).

%% @equiv delete_db([])
delete_db(DbName) ->
    delete_db(DbName, []).

%% @doc delete a database
-spec delete_db(dbname(), [option()]) -> ok | accepted | {error, atom()}.
delete_db(DbName, Options) ->
    fabric_db_delete:go(dbname(DbName), opts(Options)).

%% @doc provide an upper bound for the number of tracked document revisions
-spec set_revs_limit(dbname(), pos_integer(), [option()]) -> ok.
set_revs_limit(DbName, Limit, Options) when is_integer(Limit), Limit > 0 ->
    fabric_db_meta:set_revs_limit(dbname(DbName), Limit, opts(Options)).

%% @doc retrieves the maximum number of document revisions
-spec get_revs_limit(dbname()) -> pos_integer() | no_return().
get_revs_limit(DbName) ->
    {ok, Db} = fabric_util:get_db(dbname(DbName), [?ADMIN_CTX]),
    try
        couch_db:get_revs_limit(Db)
    after
        catch couch_db:close(Db)
    end.

%% @doc sets the readers/writers/admin permissions for a database
-spec set_security(dbname(), SecObj :: json_obj()) -> ok.
set_security(DbName, SecObj) ->
    fabric_db_meta:set_security(dbname(DbName), SecObj, [?ADMIN_CTX]).

%% @doc sets the readers/writers/admin permissions for a database
-spec set_security(dbname(), SecObj :: json_obj(), [option()]) -> ok.
set_security(DbName, SecObj, Options) ->
    fabric_db_meta:set_security(dbname(DbName), SecObj, opts(Options)).

%% @doc sets the upper bound for the number of stored purge requests
-spec set_purge_infos_limit(dbname(), pos_integer(), [option()]) -> ok.
set_purge_infos_limit(DbName, Limit, Options) when
    is_integer(Limit), Limit > 0
->
    fabric_db_meta:set_purge_infos_limit(dbname(DbName), Limit, opts(Options)).

%% @doc retrieves the upper bound for the number of stored purge requests
-spec get_purge_infos_limit(dbname()) -> pos_integer() | no_return().
get_purge_infos_limit(DbName) ->
    {ok, Db} = fabric_util:get_db(dbname(DbName), [?ADMIN_CTX]),
    try
        couch_db:get_purge_infos_limit(Db)
    after
        catch couch_db:close(Db)
    end.

get_security(DbName) ->
    get_security(DbName, [?ADMIN_CTX]).

%% @doc retrieve the security object for a database
-spec get_security(dbname()) -> json_obj() | no_return().
get_security(DbName, Options) ->
    {ok, Db} = fabric_util:get_db(dbname(DbName), opts(Options)),
    try
        couch_db:get_security(Db)
    after
        catch couch_db:close(Db)
    end.

%% @doc retrieve the security object for all shards of a database
-spec get_all_security(dbname()) ->
    {ok, [{#shard{}, json_obj()}]}
    | {error, no_majority | timeout}
    | {error, atom(), any()}.
get_all_security(DbName) ->
    get_all_security(DbName, []).

%% @doc retrieve the security object for all shards of a database
-spec get_all_security(dbname(), [option()]) ->
    {ok, [{#shard{}, json_obj()}]}
    | {error, no_majority | timeout}
    | {error, atom(), any()}.
get_all_security(DbName, Options) ->
    fabric_db_meta:get_all_security(dbname(DbName), opts(Options)).

compact(DbName) ->
    [
        rexi:cast(Node, {fabric_rpc, compact, [Name]})
     || #shard{node = Node, name = Name} <- mem3:shards(dbname(DbName))
    ],
    ok.

compact(DbName, DesignName) ->
    [
        rexi:cast(Node, {fabric_rpc, compact, [Name, DesignName]})
     || #shard{node = Node, name = Name} <- mem3:shards(dbname(DbName))
    ],
    ok.

% doc operations

%% @doc retrieve the doc with a given id
-spec open_doc(dbname(), docid(), [option()]) ->
    {ok, #doc{}}
    | {not_found, missing | deleted}
    | {timeout, any()}
    | {error, any()}
    | {error, any() | any()}.
open_doc(DbName, Id, Options) ->
    case proplists:get_value(doc_info, Options) of
        undefined ->
            fabric_doc_open:go(dbname(DbName), docid(Id), opts(Options));
        Else ->
            {error, {invalid_option, {doc_info, Else}}}
    end.

%% @doc retrieve a collection of revisions for a batch of docs
-spec open_revs(dbname(), [{{docid(), [revision()] | all}, option()}], [option()]) ->
    {ok, [[{ok, #doc{}} | {{not_found, missing}, revision()}]]}
    | {timeout, any()}
    | {error, any()}
    | {error, any(), any()}.
open_revs(DbName, IdRevsOpts0, Options) ->
    IdRevsOpts = lists:map(
        fun({{Id, Revs}, DocOpts}) ->
            {{docid(Id), Revs}, DocOpts}
        end,
        IdRevsOpts0
    ),
    fabric_open_revs:go(dbname(DbName), IdRevsOpts, opts(Options)).

%% @doc retrieve a collection of revisions, possible all
-spec open_revs(dbname(), docid(), [revision()] | all, [option()]) ->
    {ok, [{ok, #doc{}} | {{not_found, missing}, revision()}]}
    | {timeout, any()}
    | {error, any()}
    | {error, any(), any()}.
open_revs(DbName, Id, Revs, Options) ->
    fabric_doc_open_revs:go(dbname(DbName), docid(Id), Revs, opts(Options)).

%% @doc Retrieves an information on a document with a given id
-spec get_doc_info(dbname(), docid(), [options()]) ->
    {ok, #doc_info{}}
    | {not_found, missing}
    | {timeout, any()}
    | {error, any()}
    | {error, any() | any()}.
get_doc_info(DbName, Id, Options) ->
    Options1 = [doc_info | Options],
    fabric_doc_open:go(dbname(DbName), docid(Id), opts(Options1)).

%% @doc Retrieves a full information on a document with a given id
-spec get_full_doc_info(dbname(), docid(), [options()]) ->
    {ok, #full_doc_info{}}
    | {not_found, missing | deleted}
    | {timeout, any()}
    | {error, any()}
    | {error, any() | any()}.
get_full_doc_info(DbName, Id, Options) ->
    Options1 = [{doc_info, full} | Options],
    fabric_doc_open:go(dbname(DbName), docid(Id), opts(Options1)).

%% @equiv get_missing_revs(DbName, IdsRevs, [])
get_missing_revs(DbName, IdsRevs) ->
    get_missing_revs(DbName, IdsRevs, []).

%% @doc retrieve missing revisions for a list of `{Id, Revs}'
-spec get_missing_revs(dbname(), [{docid(), [revision()]}], [option()]) ->
    {ok, [{docid(), any(), [any()]}]}.
get_missing_revs(DbName, IdsRevs, Options) when is_list(IdsRevs) ->
    Sanitized = [idrevs(IdR) || IdR <- IdsRevs],
    fabric_doc_missing_revs:go(dbname(DbName), Sanitized, opts(Options)).

%% @doc update a single doc
%% @equiv update_docs(DbName,[Doc],Options)
-spec update_doc(dbname(), #doc{} | json_obj(), [option()]) ->
    {ok, any()} | any().
update_doc(DbName, Doc, Options) ->
    case update_docs(DbName, [Doc], opts(Options)) of
        {ok, [{ok, NewRev}]} ->
            {ok, NewRev};
        {accepted, [{accepted, NewRev}]} ->
            {accepted, NewRev};
        {ok, [{{_Id, _Rev}, Error}]} ->
            throw(Error);
        {ok, [Error]} ->
            throw(Error);
        {ok, []} ->
            % replication success
            #doc{revs = {Pos, [RevId | _]}} = doc(DbName, Doc),
            {ok, {Pos, RevId}};
        {error, [Error]} ->
            throw(Error)
    end.

%% @doc update a list of docs
-spec update_docs(dbname(), [#doc{} | json_obj()], [option()]) ->
    {ok, any()} | any().
update_docs(DbName, Docs0, Options) ->
    try
        Docs1 = docs(DbName, Docs0),
        fabric_doc_update:go(dbname(DbName), Docs1, opts(Options))
    of
        {ok, Results} ->
            {ok, Results};
        {accepted, Results} ->
            {accepted, Results};
        {error, Error} ->
            {error, Error};
        Error ->
            throw(Error)
    catch
        {aborted, PreCommitFailures} ->
            {aborted, PreCommitFailures}
    end.

%% @doc purge revisions for a list '{Id, Revs}'
%%      returns {ok, {PurgeSeq, Results}}
-spec purge_docs(dbname(), [{docid(), [revision()]}], [option()]) ->
    {ok, [{Health, [revision()]}] | {error, any()}}
when
    Health :: ok | accepted.
purge_docs(DbName, IdsRevs, Options) when is_list(IdsRevs) ->
    IdsRevs2 = [idrevs(IdRs) || IdRs <- IdsRevs],
    fabric_doc_purge:go(dbname(DbName), IdsRevs2, opts(Options)).

%% @doc spawns a process to upload attachment data and
%%      returns a fabric attachment receiver context tuple
%%      with the spawned middleman process, an empty binary,
%%      or exits with an error tuple {Error, Arg}
-spec att_receiver(
    #httpd{},
    dbname(),
    Length ::
        undefined
        | chunked
        | pos_integer()
        | {unknown_transfer_encoding, any()}
) ->
    {fabric_attachment_receiver, pid(), chunked | pos_integer()} | binary().
att_receiver(Req, DbName, Length) ->
    fabric_doc_atts:receiver(Req, DbName, Length).

%% @equiv all_docs(DbName, [], Callback, Acc0, QueryArgs)
all_docs(DbName, Callback, Acc, QueryArgs) ->
    all_docs(DbName, [], Callback, Acc, QueryArgs).

%% @doc retrieves all docs. Additional query parameters, such as `limit',
%%      `start_key' and `end_key', `descending', and `include_docs', can
%%      also be passed to further constrain the query.
-spec all_docs(
    dbname(),
    [{atom(), any()}],
    callback(),
    [] | tuple(),
    #mrargs{} | [option()]
) ->
    {ok, any()} | {error, Reason :: term()}.

all_docs(DbName, Options, Callback, Acc0, #mrargs{} = QueryArgs) when
    is_function(Callback, 2)
->
    fabric_view_all_docs:go(dbname(DbName), opts(Options), QueryArgs, Callback, Acc0);
%% @doc convenience function that takes a keylist rather than a record
%% @equiv all_docs(DbName, Callback, Acc0, kl_to_query_args(QueryArgs))
all_docs(DbName, Options, Callback, Acc0, QueryArgs) ->
    all_docs(DbName, Options, Callback, Acc0, kl_to_query_args(QueryArgs)).

-spec changes(dbname(), callback(), any(), #changes_args{} | [{atom(), any()}]) ->
    {ok, any()}.
changes(DbName, Callback, Acc0, #changes_args{} = Options) ->
    Feed = Options#changes_args.feed,
    fabric_view_changes:go(dbname(DbName), Feed, Options, Callback, Acc0);
%% @doc convenience function, takes keylist instead of record
%% @equiv changes(DbName, Callback, Acc0, kl_to_changes_args(Options))
changes(DbName, Callback, Acc0, Options) ->
    changes(DbName, Callback, Acc0, kl_to_changes_args(Options)).

%% @equiv query_view(DbName, DesignName, ViewName, #mrargs{})
query_view(DbName, DesignName, ViewName) ->
    query_view(DbName, DesignName, ViewName, #mrargs{}).

%% @equiv query_view(DbName, DesignName,
%%                     ViewName, fun default_callback/2, [], QueryArgs)
query_view(DbName, DesignName, ViewName, QueryArgs) ->
    Callback = fun default_callback/2,
    query_view(DbName, DesignName, ViewName, Callback, [], QueryArgs).

%% @equiv query_view(DbName, DesignName, [],
%%                     ViewName, fun default_callback/2, [], QueryArgs)
query_view(DbName, DDoc, ViewName, Callback, Acc, QueryArgs) ->
    query_view(DbName, [], DDoc, ViewName, Callback, Acc, QueryArgs).

%% @doc execute a given view.
-spec query_view(
    dbname(),
    [{atom(), any()}] | [],
    #doc{} | binary(),
    iodata(),
    callback(),
    any(),
    #mrargs{}
) ->
    any().
query_view(Db, Options, GroupId, ViewName, Callback, Acc0, QueryArgs) when
    is_binary(GroupId)
->
    DbName = dbname(Db),
    {ok, DDoc} = ddoc_cache:open(DbName, <<"_design/", GroupId/binary>>),
    query_view(Db, Options, DDoc, ViewName, Callback, Acc0, QueryArgs);
query_view(Db, Options, DDoc, ViewName, Callback, Acc0, QueryArgs0) ->
    DbName = dbname(Db),
    View = name(ViewName),
    case fabric_util:is_users_db(DbName) of
        true ->
            FakeDb = fabric_util:open_cluster_db(DbName, Options),
            couch_users_db:after_doc_read(DDoc, FakeDb);
        false ->
            ok
    end,
    {ok, #mrst{views = Views, language = Lang}} =
        couch_mrview_util:ddoc_to_mrst(DbName, DDoc),
    QueryArgs1 = couch_mrview_util:set_view_type(QueryArgs0, View, Views),
    QueryArgs2 = fabric_util:validate_args(Db, DDoc, QueryArgs1),
    VInfo = couch_mrview_util:extract_view(Lang, QueryArgs2, View, Views),
    case is_reduce_view(QueryArgs2) of
        true ->
            fabric_view_reduce:go(
                Db,
                DDoc,
                View,
                QueryArgs2,
                Callback,
                Acc0,
                VInfo
            );
        false ->
            fabric_view_map:go(
                Db,
                Options,
                DDoc,
                View,
                QueryArgs2,
                Callback,
                Acc0,
                VInfo
            )
    end.

%% @doc retrieve info about a view group, disk size, language, whether compaction
%%      is running and so forth
-spec get_view_group_info(dbname(), #doc{} | docid()) ->
    {ok, [
        {signature, binary()}
        | {language, binary()}
        | {disk_size, non_neg_integer()}
        | {compact_running, boolean()}
        | {updater_running, boolean()}
        | {waiting_commit, boolean()}
        | {waiting_clients, non_neg_integer()}
        | {update_seq, pos_integer()}
        | {purge_seq, non_neg_integer()}
        | {sizes, [
            {active, non_neg_integer()}
            | {external, non_neg_integer()}
            | {file, non_neg_integer()}
        ]}
        | {updates_pending, [
            {minimum, non_neg_integer()}
            | {preferred, non_neg_integer()}
            | {total, non_neg_integer()}
        ]}
    ]}.
get_view_group_info(DbName, DesignId) ->
    fabric_group_info:go(dbname(DbName), design_doc(DesignId)).

-spec end_changes() -> ok.
end_changes() ->
    fabric_view_changes:increment_changes_epoch().

%% @doc retrieve all the design docs from a database
-spec design_docs(dbname()) -> {ok, [json_obj()]} | {error, Reason :: term()}.
design_docs(DbName) ->
    Extra =
        case get(io_priority) of
            undefined -> [];
            Else -> [{io_priority, Else}]
        end,
    QueryArgs0 = #mrargs{
        include_docs = true,
        extra = Extra
    },
    QueryArgs = set_namespace(<<"_design">>, QueryArgs0),
    Callback = fun
        ({meta, _}, []) ->
            {ok, []};
        ({row, Props}, Acc) ->
            {ok, [couch_util:get_value(doc, Props) | Acc]};
        (complete, Acc) ->
            {ok, lists:reverse(Acc)};
        ({error, Reason}, _Acc) ->
            {error, Reason}
    end,
    fabric:all_docs(dbname(DbName), [?ADMIN_CTX], Callback, [], QueryArgs).

%% @doc forces a reload of validation functions, this is performed after
%%      design docs are update
%% NOTE: This function probably doesn't belong here as part fo the API
-spec reset_validation_funs(dbname()) -> [reference()].
reset_validation_funs(DbName) ->
    [
        rexi:cast(Node, {fabric_rpc, reset_validation_funs, [Name]})
     || #shard{node = Node, name = Name} <- mem3:shards(DbName)
    ].

%% @doc clean up index files for all Dbs
-spec cleanup_index_files() -> [ok].
cleanup_index_files() ->
    {ok, Dbs} = fabric:all_dbs(),
    [cleanup_index_files(Db) || Db <- Dbs].

%% @doc clean up index files for a specific db
-spec cleanup_index_files(dbname()) -> ok.
cleanup_index_files(DbName) ->
    try
        ShardNames = [mem3:name(S) || S <- mem3:local_shards(dbname(DbName))],
        cleanup_local_indices_and_purge_checkpoints(ShardNames)
    catch
        error:database_does_not_exist ->
            ok
    end.

cleanup_local_indices_and_purge_checkpoints([]) ->
    ok;
cleanup_local_indices_and_purge_checkpoints([_ | _] = Dbs) ->
    MrViewIndices = lists:map(fun couch_mrview_util:get_index_files/1, Dbs),
    MrViewPurges = lists:map(fun couch_mrview_util:get_purge_checkpoints/1, Dbs),
    MrViewSigs = couch_mrview_util:get_signatures(hd(Dbs)),
    ok = cleanup_mrview_purges(MrViewSigs, MrViewPurges, Dbs),
    ok = cleanup_mrview_indices(MrViewSigs, MrViewIndices),

    ClouseauSigs = dreyfus_util:active_sigs(hd(Dbs)),
    ok = cleanup_clouseau_indices(Dbs, ClouseauSigs),

    NouveauSigs = nouveau_util:active_sigs(hd(Dbs)),
    ok = cleanup_nouveau_indices(Dbs, NouveauSigs).

cleanup_mrview_purges(Sigs, AllPurges, Dbs) ->
    Fun = fun(DbPurges, Db) ->
        couch_mrview_cleanup:cleanup_purges(Db, Sigs, DbPurges)
    end,
    lists:zipwith(Fun, AllPurges, Dbs),
    ok.

cleanup_mrview_indices(Sigs, AllIndices) ->
    Fun = fun(DbIndices) ->
        couch_mrview_cleanup:cleanup_indices(Sigs, DbIndices)
    end,
    lists:foreach(Fun, AllIndices).

cleanup_clouseau_indices(Dbs, ActiveSigs) ->
    Fun = fun(Db) -> clouseau_rpc:cleanup(Db, ActiveSigs) end,
    lists:foreach(Fun, Dbs).
cleanup_nouveau_indices(Dbs, ActiveSigs) ->
    Fun = fun(Db) -> nouveau_api:delete_path(nouveau_util:index_name(Db), ActiveSigs) end,
    lists:foreach(Fun, Dbs).

%% @doc clean up index files for a specific db on all nodes
-spec cleanup_index_files_all_nodes(dbname()) -> [reference()].
cleanup_index_files_all_nodes(DbName) ->
    lists:foreach(
        fun(Node) ->
            rexi:cast(Node, {?MODULE, cleanup_index_files, [DbName]})
        end,
        mem3:nodes()
    ).

%% some simple type validation and transcoding
dbname(DbName) when is_list(DbName) ->
    list_to_binary(DbName);
dbname(DbName) when is_binary(DbName) ->
    DbName;
dbname(Db) ->
    try
        couch_db:name(Db)
    catch
        error:badarg ->
            erlang:error({illegal_database_name, Db})
    end.

%% @doc get db shard uuids
-spec db_uuids(dbname()) -> map().
db_uuids(DbName) ->
    fabric_db_uuids:go(dbname(DbName)).

name(Thing) ->
    couch_util:to_binary(Thing).

docid(DocId) when is_list(DocId) ->
    list_to_binary(DocId);
docid(DocId) ->
    DocId.

docs(Db, Docs) when is_list(Docs) ->
    [doc(Db, D) || D <- Docs];
docs(_Db, Docs) ->
    erlang:error({illegal_docs_list, Docs}).

doc(_Db, #doc{} = Doc) ->
    Doc;
doc(Db0, {_} = Doc) ->
    Db =
        case couch_db:is_db(Db0) of
            true ->
                Db0;
            false ->
                Shard = hd(mem3:shards(Db0)),
                Props = couch_util:get_value(props, Shard#shard.opts, []),
                {ok, Db1} = couch_db:clustered_db(Db0, [{props, Props}]),
                Db1
        end,
    couch_db:doc_from_json_obj_validate(Db, Doc);
doc(_Db, Doc) ->
    erlang:error({illegal_doc_format, Doc}).

design_doc(#doc{} = DDoc) ->
    DDoc;
design_doc(DocId) when is_list(DocId) ->
    design_doc(list_to_binary(DocId));
design_doc(<<"_design/", _/binary>> = DocId) ->
    DocId;
design_doc(GroupName) ->
    <<"_design/", GroupName/binary>>.

idrevs({Id, Revs}) when is_list(Revs) ->
    {docid(Id), [rev(R) || R <- Revs]}.

rev(Rev) when is_list(Rev); is_binary(Rev) ->
    couch_doc:parse_rev(Rev);
rev({Seq, Hash} = Rev) when is_integer(Seq), is_binary(Hash) ->
    Rev.

%% @doc convenience method, useful when testing or calling fabric from the shell
opts(Options) ->
    add_option(user_ctx, add_option(io_priority, Options)).

add_option(Key, Options) ->
    case couch_util:get_value(Key, Options) of
        undefined ->
            case erlang:get(Key) of
                undefined ->
                    Options;
                Value ->
                    [{Key, Value} | Options]
            end;
        _ ->
            Options
    end.

default_callback(complete, Acc) ->
    {ok, lists:reverse(Acc)};
default_callback(Row, Acc) ->
    {ok, [Row | Acc]}.

is_reduce_view(#mrargs{view_type = ViewType}) ->
    ViewType =:= red;
is_reduce_view({Reduce, _, _}) ->
    Reduce =:= red.

%% @doc convenience method for use in the shell, converts a keylist
%%      to a `changes_args' record
kl_to_changes_args(KeyList) ->
    kl_to_record(KeyList, changes_args).

%% @doc convenience method for use in the shell, converts a keylist
%%      to a `mrargs' record
kl_to_query_args(KeyList) ->
    kl_to_record(KeyList, mrargs).

%% @doc finds the index of the given Key in the record.
%%      note that record_info is only known at compile time
%%      so the code must be written in this way. For each new
%%      record type add a case clause
lookup_index(Key, RecName) ->
    Indexes =
        case RecName of
            changes_args ->
                lists:zip(
                    record_info(fields, changes_args),
                    lists:seq(2, record_info(size, changes_args))
                );
            mrargs ->
                lists:zip(
                    record_info(fields, mrargs),
                    lists:seq(2, record_info(size, mrargs))
                )
        end,
    couch_util:get_value(Key, Indexes).

%% @doc convert a keylist to record with given `RecName'
%% @see lookup_index
kl_to_record(KeyList, RecName) ->
    Acc0 =
        case RecName of
            changes_args -> #changes_args{};
            mrargs -> #mrargs{}
        end,
    lists:foldl(
        fun({Key, Value}, Acc) ->
            Index = lookup_index(couch_util:to_existing_atom(Key), RecName),
            setelement(Index, Acc, Value)
        end,
        Acc0,
        KeyList
    ).

set_namespace(NS, #mrargs{extra = Extra} = Args) ->
    Args#mrargs{extra = [{namespace, NS} | Extra]}.

-ifdef(TEST).
-include_lib("eunit/include/eunit.hrl").

update_doc_test_() ->
    {
        "Update doc tests",
        {
            setup,
            fun setup/0,
            fun teardown/1,
            fun(Ctx) ->
                [
                    should_throw_conflict(Ctx)
                ]
            end
        }
    }.

should_throw_conflict(Doc) ->
    ?_test(begin
        ?assertThrow(conflict, update_doc(<<"test-db">>, Doc, []))
    end).

setup() ->
    Doc = #doc{
        id = <<"test_doc">>,
        revs = {3, [<<5, 68, 252, 180, 43, 161, 216, 223, 26, 119, 71, 219, 212, 229, 159, 113>>]},
        body = {[{<<"foo">>, <<"asdf">>}, {<<"author">>, <<"tom">>}]},
        atts = [],
        deleted = false,
        meta = []
    },
    ok = application:ensure_started(config),
    ok = meck:expect(mem3, shards, fun(_, _) -> [] end),
    ok = meck:expect(mem3, quorum, fun(_) -> 1 end),
    ok = meck:expect(rexi, cast, fun(_, _) -> ok end),
    ok = meck:expect(
        rexi_utils,
        recv,
        fun(_, _, _, _, _, _) ->
            {ok, {error, [{Doc, conflict}]}}
        end
    ),
    ok = meck:expect(
        couch_util,
        reorder_results,
        fun(_, [{_, Res}], _) ->
            [Res]
        end
    ),
    ok = meck:expect(fabric_util, create_monitors, fun(_) -> ok end),
    ok = meck:expect(rexi_monitor, stop, fun(_) -> ok end),
    Doc.

teardown(_) ->
    meck:unload(),
    ok = application:stop(config).

-endif.