1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
% 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(couch_index_util).
-export([root_dir/0, index_dir/2, index_file/3]).
-export([load_doc/3, sort_lib/1, hexsig/1]).
-include_lib("couch/include/couch_db.hrl").
root_dir() ->
config:get("couchdb", "view_index_dir").
index_dir(Module, DbName) when is_binary(DbName) ->
DbDir = "." ++ binary_to_list(DbName) ++ "_design",
filename:join([root_dir(), DbDir, Module]);
index_dir(Module, Db) ->
index_dir(Module, couch_db:name(Db)).
index_file(Module, DbName, FileName) ->
filename:join(index_dir(Module, DbName), FileName).
load_doc(Db, #doc_info{} = DI, Opts) ->
Deleted = lists:member(deleted, Opts),
case (catch couch_db:open_doc(Db, DI, Opts)) of
{ok, #doc{deleted = false} = Doc} -> Doc;
{ok, #doc{deleted = true} = Doc} when Deleted -> Doc;
_Else -> null
end;
load_doc(Db, {DocId, Rev}, Opts) ->
case (catch load_doc(Db, DocId, Rev, Opts)) of
#doc{deleted = false} = Doc -> Doc;
_ -> null
end.
load_doc(Db, DocId, Rev, Options) ->
case Rev of
% open most recent rev
nil ->
case (catch couch_db:open_doc(Db, DocId, Options)) of
{ok, Doc} -> Doc;
_Error -> null
end;
% open a specific rev (deletions come back as stubs)
_ ->
case (catch couch_db:open_doc_revs(Db, DocId, [Rev], Options)) of
{ok, [{ok, Doc}]} -> Doc;
{ok, [{{not_found, missing}, Rev}]} -> null;
{ok, [_Else]} -> null
end
end.
sort_lib({Lib}) ->
sort_lib(Lib, []).
sort_lib([], LAcc) ->
lists:keysort(1, LAcc);
sort_lib([{LName, {LObj}} | Rest], LAcc) ->
% descend into nested object
LSorted = sort_lib(LObj, []),
sort_lib(Rest, [{LName, LSorted} | LAcc]);
sort_lib([{LName, LCode} | Rest], LAcc) ->
sort_lib(Rest, [{LName, LCode} | LAcc]).
hexsig(Sig) ->
couch_util:to_hex(binary_to_list(Sig)).
|
