summaryrefslogtreecommitdiff
path: root/src/couchdb/mod_couch.erl
diff options
context:
space:
mode:
authorChristopher Lenz <cmlenz@apache.org>2008-04-15 12:21:00 +0000
committerChristopher Lenz <cmlenz@apache.org>2008-04-15 12:21:00 +0000
commit39de3072bcf9fdeec6d3faeb125924c401242205 (patch)
treeda55307c8762f9ff16f7a7e478d971c0f352d281 /src/couchdb/mod_couch.erl
parent53968ddfd93bfe1aa403478de715ae0ac77db177 (diff)
Merged mochiweb branch back into trunk.
git-svn-id: https://svn.apache.org/repos/asf/incubator/couchdb/trunk@648222 13f79535-47bb-0310-9956-ffa450edef68
Diffstat (limited to 'src/couchdb/mod_couch.erl')
-rw-r--r--src/couchdb/mod_couch.erl890
1 files changed, 0 insertions, 890 deletions
diff --git a/src/couchdb/mod_couch.erl b/src/couchdb/mod_couch.erl
deleted file mode 100644
index 0d157b1e..00000000
--- a/src/couchdb/mod_couch.erl
+++ /dev/null
@@ -1,890 +0,0 @@
-% 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(mod_couch).
-
--include("couch_db.hrl").
-
--export([do/1, load/2, url_decode/1]).
-
--include_lib("../couch_inets/httpd.hrl").
-
--record(uri_parts,
- {db = "",
- doc = "",
- attachment = "",
- view = "",
- querystr = ""}).
-
--record(doc_query_args,
- {
- options = [],
- rev = "",
- open_revs = ""
- }).
-
-%% do. This is the main entry point into Apache CouchDB from the HTTP server
-
-do(Mod) ->
- #mod{request_uri=Uri,request_line=Request, parsed_header=Header,entity_body=Body} = Mod,
- PrevTrapExit = process_flag(trap_exit, true),
- Resp =
- case Uri of
- "/_utils/" ++ RestURI ->
- % if the URI is the utils directory, then this
- % tells mod_get (a std HTTP module) where to serve the file from
- DocumentRoot = httpd_util:lookup(Mod#mod.config_db, document_root, ""),
- {Path, AfterPath} = httpd_util:split_path(DocumentRoot ++ "/" ++ RestURI),
-
- case RestURI of
- "" ->
- Paths = httpd_util:split_path(DocumentRoot ++ "/index.html"),
- {proceed, [{real_name, Paths} | Mod#mod.data]};
- _ ->
- case filelib:is_file(Path) of
- true ->
- {proceed, [{real_name, {Path, AfterPath}} | Mod#mod.data]};
- false ->
- case filelib:is_dir(Path) of
- true ->
- % this ends up causing a "Internal Server Error", need to fix.
- {proceed, [{response,{403,"Forbidden"}}]};
- false ->
- {proceed, [{response,{404,"Not found"}}]}
- end
- end
- end;
- "/favicon.ico" ->
- DocumentRoot = httpd_util:lookup(Mod#mod.config_db, document_root, ""),
- RealName = DocumentRoot ++ "/" ++ Uri,
- {Path, AfterPath} = httpd_util:split_path(RealName),
- {proceed, [{real_name, {Path, AfterPath}} | Mod#mod.data]};
- _ ->
- couch_log:info("HTTP Request: ~s", [Request]),
- couch_log:debug("Headers: ~p", [Header]),
- couch_log:debug("Body: ~P", [Body, 100]),
- case (catch parse_uri(Uri)) of
- {ok, Parts} ->
- {ok, ResponseCode} =
- case (catch do(Mod, Parts)) of
- {ok, ResponseCode0} ->
- {ok, ResponseCode0};
- Error ->
- send_error(Mod, Error)
- end;
- Error ->
- {ok, ResponseCode} = send_error(Mod, Error)
- end,
- couch_log:info("HTTP Response Code:~p~n", [ResponseCode]),
- {proceed, [{response, {already_sent, ResponseCode, 0}} | Mod#mod.data]}
- end,
- process_flag(trap_exit, PrevTrapExit),
- Resp.
-
-
-parse_uri(RequestUri) ->
- % seperate out the path and query portions and
- % strip out leading slash and question mark.
- case regexp:split(RequestUri, "\\?") of
- {ok, [[$/|UriPath], QueryStr]} -> ok;
- {ok, [[$/|UriPath]]} -> QueryStr = ""
- end,
- % lets try to parse out the UriPath.
- {ok, UrlParts} = regexp:split(UriPath, "/"),
-
- {DbName, Id, Attachment, View} =
- case UrlParts of
- [Db] ->
- {Db, "", "", ""};
- [Db, "_design", Doc] ->
- {Db, "_design/" ++ Doc, "", ""};
- [Db, "_design", Doc, Attachment0] ->
- {Db, "_design/" ++ Doc, Attachment0, ""};
- [Db, "_view", Doc, ViewName] ->
- {Db, "_design/" ++ Doc, "", ViewName};
- [Db, "_view%2f" ++ Doc, ViewName] ->
- {Db, "_design/" ++ Doc, "", ViewName};
- [Db, Doc] ->
- {Db, Doc, "", ""};
- [Db, Doc, Attachment0] ->
- {Db, Doc, Attachment0, ""};
- _ ->
- throw({invalid_uri, lists:flatten(io_lib:format("Uri has too many parts: ~p", [UrlParts]))})
- end,
- {ok, #uri_parts{db=url_decode(DbName),
- doc=url_decode(Id),
- attachment=url_decode(Attachment),
- view=url_decode(View),
- querystr=url_decode(QueryStr)}}.
-
-resp_json_header(Mod) ->
- resp_json_header(Mod, []).
-
-% return json doc header values list
-resp_json_header(Mod, Options) ->
- Types = string:tokens(proplists:get_value("accept", Mod#mod.parsed_header, ""), ", "),
- case lists:member("application/json", Types) of
- true ->
- resp_header(Mod, Options) ++ [{"content-type","application/json"}];
- false ->
- resp_header(Mod, Options) ++ [{"content-type","text/plain;charset=utf-8"}]
- end.
-
-% return doc header values list
-resp_header(#mod{http_version=Version}, Options) ->
- [{"cache-control", "no-cache"},
- {"pragma", "no-cache"},
- {"expires", httpd_util:rfc1123_date()}] ++
- case lists:member(no_body, Options) of
- true -> [];
- false ->
- case Version == "HTTP/1.1" of
- true ->
- [{"transfer-encoding", "chunked"}];
- false ->
- [{"connection", "close"}]
- end
- end.
-
-
-url_decode([$%, Hi, Lo | Tail]) ->
- Hex = erlang:list_to_integer([Hi, Lo], 16),
- xmerl_ucs:to_utf8([Hex]) ++ url_decode(Tail);
-url_decode([H|T]) ->
- [H |url_decode(T)];
-url_decode([]) ->
- [].
-
-
-send_header(Mod, RespCode, Headers) ->
- couch_log:debug("HTTP Response Headers (code ~w): ~p", [RespCode, Headers]),
- httpd_response:send_header(Mod, RespCode, Headers).
-
-send_chunk(Mod, Data) ->
- httpd_response:send_chunk(Mod, Data, false).
-
-send_final_chunk(Mod) ->
- httpd_response:send_final_chunk(Mod, false).
-
-show_couch_welcome(Mod) ->
- send_header(Mod, 200, resp_json_header(Mod)),
- send_chunk(Mod, "{\"couchdb\": \"Welcome\", "),
- send_chunk(Mod, "\"version\": \"" ++ couch_server:get_version()),
- send_chunk(Mod, "\"}\n"),
- send_final_chunk(Mod),
- {ok, 200}.
-
-
-do(#mod{method="GET"}=Mod, #uri_parts{db=""}) ->
- show_couch_welcome(Mod);
-do(#mod{method="GET"}=Mod, #uri_parts{db="_all_dbs", doc=""}=Parts) ->
- send_all_dbs(Mod, Parts);
-do(#mod{method="POST"}=Mod, #uri_parts{db="_replicate", doc=""}) ->
- handle_replication_request(Mod);
-do(#mod{method="POST"}=Mod, #uri_parts{db="_restart", doc=""}) ->
- couch_server:remote_restart(),
- send_ok(Mod, 201);
-do(#mod{method="POST"}=Mod, #uri_parts{doc="_missing_revs"}=Parts) ->
- handle_missing_revs_request(Mod, Parts);
-do(#mod{method="POST"}=Mod, #uri_parts{doc="_compact"}=Parts) ->
- handle_compact(Mod, Parts);
-do(#mod{method="PUT"}=Mod, #uri_parts{doc=""}=Parts) ->
- handle_db_create(Mod, Parts);
-do(#mod{method="DELETE"}=Mod, #uri_parts{doc=""}=Parts) ->
- handle_db_delete(Mod, Parts);
-do(#mod{method="POST"}=Mod, #uri_parts{doc="_bulk_docs"}=Parts) ->
- handle_bulk_doc_update(Mod, Parts);
-do(#mod{method="POST"}=Mod, #uri_parts{doc=""}=Parts) ->
- handle_doc_post(Mod, Parts);
-do(#mod{method="PUT"}=Mod, Parts) ->
- handle_doc_put(Mod, Parts);
-do(#mod{method="DELETE"}=Mod, Parts) ->
- handle_doc_delete(Mod, Parts);
-do(#mod{method="POST"}=Mod, #uri_parts{doc="_temp_view"}=Parts) ->
- send_temp_view(Mod, Parts);
-do(#mod{method="GET"}=Mod, #uri_parts{doc="_all_docs"}=Parts) ->
- send_all_docs(Mod, Parts);
-do(#mod{method="GET"}=Mod, #uri_parts{doc="_all_docs_by_seq"}=Parts) ->
- send_all_docs_by_seq(Mod, Parts);
-do(#mod{method="GET"}=Mod, #uri_parts{doc=""}=Parts) ->
- send_database_info(Mod, Parts);
-do(#mod{method=Method}=Mod, #uri_parts{attachment="",view=""}=Parts)
- when Method == "GET" orelse Method == "HEAD" ->
- #doc_query_args{open_revs=Revs} = doc_parse_query(Parts#uri_parts.querystr),
- case Revs of
- [] ->
- send_doc(Mod, Parts);
- _ ->
- send_doc_revs(Mod, Parts)
- end;
-do(#mod{method=Method}=Mod, #uri_parts{attachment=Att}=Parts)
- when Att /= "", Method == "GET" orelse Method == "HEAD" ->
- send_attachment(Mod, Parts);
-do(#mod{method="GET"}=Mod, #uri_parts{view=View}=Parts) when View /= "" ->
- send_view(Mod, Parts).
-
-handle_db_create(Mod, #uri_parts{db=DbName}) ->
- case couch_server:create(DbName, []) of
- {ok, _Db} ->
- send_ok(Mod, 201);
- {error, database_already_exists} ->
- Msg = io_lib:format("Database ~p already exists.", [DbName]),
- throw({database_already_exists, Msg});
- Error ->
- Msg = io_lib:format("Error creating database ~p: ~p", [DbName, Error]),
- throw({unknown_error, Msg})
- end.
-
-handle_db_delete(Mod, #uri_parts{db=DbName}) ->
- % delete with no doc specified, therefore database delete
- case couch_server:delete(DbName) of
- ok ->
- send_ok(Mod, 202);
- Error ->
- throw(Error)
- end.
-
-handle_bulk_doc_update(#mod{entity_body=RawBody}=Mod, Parts) ->
- Options = [], % put options here.
- Db = open_db(Parts),
- {obj, JsonProps} = cjson:decode(RawBody),
- DocsArray = proplists:get_value("docs", JsonProps),
- % convert all the doc elements to native docs
- case proplists:get_value("new_edits", JsonProps, true) of
- true ->
- Docs = lists:map(
- fun({obj, ObjProps} = JsonObj) ->
- Doc = couch_doc:from_json_obj(JsonObj),
-
- Id =
- case Doc#doc.id of
- "" -> couch_util:new_uuid();
- Id0 -> Id0
- end,
- Revs =
- case proplists:get_value("_rev", ObjProps) of
- undefined -> [];
- Rev -> [Rev]
- end,
- Doc#doc{id=Id,revs=Revs}
- end,
- tuple_to_list(DocsArray)),
-
- {ok, ResultRevs} = couch_db:update_docs(Db, Docs, Options),
-
- % output the results
- DocResults = lists:zipwith(
- fun(Doc, NewRev) ->
- {obj, [{"id", Doc#doc.id}, {"rev", NewRev}]}
- end,
- Docs, ResultRevs),
- send_ok(Mod, 201, [{new_revs, list_to_tuple(DocResults)}]);
-
- false ->
- Docs = [couch_doc:from_json_obj(JsonObj) || JsonObj <- tuple_to_list(DocsArray)],
- ok = couch_db:save_docs(Db, Docs, Options),
- send_ok(Mod, 201)
- end.
-
-
-
-
-doc_parse_query(QueryStr) ->
- QueryList = httpd:parse_query(QueryStr),
- lists:foldl(fun({Key,Value}, Args) ->
- case {Key, Value} of
- {"attachments", "true"} ->
- Options = [attachments | Args#doc_query_args.options],
- Args#doc_query_args{options=Options};
- {"meta", "true"} ->
- Options = [revs_info, conflicts, deleted_conflicts | Args#doc_query_args.options],
- Args#doc_query_args{options=Options};
- {"revs", "true"} ->
- Options = [revs | Args#doc_query_args.options],
- Args#doc_query_args{options=Options};
- {"revs_info", "true"} ->
- Options = [revs_info | Args#doc_query_args.options],
- Args#doc_query_args{options=Options};
- {"conflicts", "true"} ->
- Options = [conflicts | Args#doc_query_args.options],
- Args#doc_query_args{options=Options};
- {"deleted_conflicts", "true"} ->
- Options = [deleted_conflicts | Args#doc_query_args.options],
- Args#doc_query_args{options=Options};
- {"rev", Rev} ->
- Args#doc_query_args{rev=Rev};
- {"open_revs", "all"} ->
- Args#doc_query_args{open_revs=all};
- {"open_revs", RevsJsonStr} ->
- JsonArray = cjson:decode(RevsJsonStr),
- Args#doc_query_args{open_revs=tuple_to_list(JsonArray)};
- _Else -> % unknown key value pair, ignore.
- Args
- end
- end,
- #doc_query_args{}, QueryList).
-
-
-handle_doc_post(#mod{entity_body=RawBody}=Mod, Parts) ->
- Db = open_db(Parts),
- Json = cjson:decode(RawBody),
- Doc = couch_doc:from_json_obj(Json),
- Id = couch_util:new_uuid(),
- {ok, NewRevId} = couch_db:update_doc(Db, Doc#doc{id=Id, revs=[]}, []),
- send_ok(Mod, 201, [{"id", Id}, {"rev", NewRevId}], [{"etag", NewRevId}]).
-
-handle_doc_put(#mod{parsed_header=Headers}=Mod,
- #uri_parts{doc=Id, querystr=QueryStr}=Parts) ->
- #doc_query_args{options=SaveOptions} = doc_parse_query(QueryStr),
- Db = open_db(Parts),
- {obj, ObjProps} = Json = cjson:decode(Mod#mod.entity_body),
- Doc = couch_doc:from_json_obj(Json),
- Etag = proplists:get_value("if-match", Headers, ""),
- DocRev = proplists:get_value("_rev", ObjProps, ""),
-
- if DocRev /= "" andalso Etag /= "" andalso DocRev /= Etag ->
- throw({invalid_request, "Document rev and etag have different values"});
- true -> ok
- end,
- Revs =
- if DocRev /= "" -> [DocRev];
- Etag /= "" -> [Etag];
- true -> []
- end,
-
- {ok, NewRevId} = couch_db:update_doc(Db, Doc#doc{id=Id, revs=Revs}, SaveOptions),
- send_ok(Mod, 201, [{"id", Id}, {"rev", NewRevId}],[{"etag", NewRevId}]).
-
-handle_doc_delete(#mod{parsed_header=Headers}=Mod,
- #uri_parts{doc=Id, querystr=QueryStr}=Parts) ->
- Db = open_db(Parts),
- #doc_query_args{rev=QueryRev} = doc_parse_query(QueryStr),
- Etag = proplists:get_value("if-match", Headers, ""),
- RevToDelete =
- case {QueryRev, Etag} of
- {"", ""} ->
- throw({missing_rev, "Document rev/etag must be specified to delete"});
- {_, ""} ->
- QueryRev;
- {"", _} ->
- Etag;
- _ when QueryRev == Etag ->
- Etag;
- _ ->
- throw({invalid_request, "Document rev and etag have different values"})
- end,
- {ok, NewRev} = couch_db:delete_doc(Db, Id, [RevToDelete]),
- send_ok(Mod, 202, [{"id", Id}, {"rev", NewRev}]).
-
-
--record(query_args,
- {start_key = nil,
- end_key = <<>>,
- count = 10000000000, % a huge huge default number. Picked so we don't have
- % to do different logic for when there is no count limit
- update = true,
- direction = fwd,
- start_docid = nil,
- end_docid = <<>>,
- skip = 0
- }).
-
-reverse_key_default(nil) -> <<>>;
-reverse_key_default(<<>>) -> nil;
-reverse_key_default(Key) -> Key.
-
-view_parse_query(QueryStr) ->
- QueryList = httpd:parse_query(QueryStr),
- lists:foldl(fun({Key,Value}, Args) ->
- case {Key, Value} of
- {"", _} ->
- Args;
- {"key", Value} ->
- JsonKey = cjson:decode(Value),
- Args#query_args{start_key=JsonKey,end_key=JsonKey};
- {"startkey_docid", DocId} ->
- Args#query_args{start_docid=DocId};
- {"startkey", Value} ->
- Args#query_args{start_key=cjson:decode(Value)};
- {"endkey", Value} ->
- Args#query_args{end_key=cjson:decode(Value)};
- {"count", Value} ->
- case (catch list_to_integer(Value)) of
- Count when is_integer(Count) ->
- if Count < 0 ->
- Args#query_args {
- direction =
- if Args#query_args.direction == rev -> fwd;
- true -> rev
- end,
- count=Count,
- start_key = reverse_key_default(Args#query_args.start_key),
- start_docid = reverse_key_default(Args#query_args.start_docid),
- end_key = reverse_key_default(Args#query_args.end_key),
- end_docid = reverse_key_default(Args#query_args.end_docid)};
- true ->
- Args#query_args{count=Count}
- end;
- _Error ->
- Msg = io_lib:format("Bad URL query value, number expected: count=~s", [Value]),
- throw({query_parse_error, Msg})
- end;
- {"update", "false"} ->
- Args#query_args{update=false};
- {"descending", "true"} ->
- case Args#query_args.direction of
- fwd ->
- Args#query_args {
- direction = rev,
- start_key = reverse_key_default(Args#query_args.start_key),
- start_docid = reverse_key_default(Args#query_args.start_docid),
- end_key = reverse_key_default(Args#query_args.end_key),
- end_docid = reverse_key_default(Args#query_args.end_docid)};
- _ ->
- Args %already reversed
- end;
- {"skip", Value} ->
- case (catch list_to_integer(Value)) of
- Count when is_integer(Count) ->
- Args#query_args{skip=Count};
- _Error ->
- Msg = lists:flatten(io_lib:format(
- "Bad URL query value, number expected: skip=~s", [Value])),
- throw({query_parse_error, Msg})
- end;
- _ -> % unknown key
- Msg = lists:flatten(io_lib:format(
- "Bad URL query key:~s", [Key])),
- throw({query_parse_error, Msg})
- end
- end,
- #query_args{}, QueryList).
-
-
-% returns db, otherwise throws exception. Note: no {ok,_}.
-open_db(#uri_parts{db=DbName}) ->
- open_db(DbName);
-open_db(DbName) when is_list(DbName)->
- case couch_server:open(DbName) of
- {ok, Db} ->
- Db;
- Error ->
- throw(Error)
- end.
-
-handle_missing_revs_request(#mod{entity_body=RawJson}=Mod, Parts) ->
- Db = open_db(Parts),
- {obj, JsonDocIdRevs} = cjson:decode(RawJson),
- DocIdRevs = [{Id, tuple_to_list(Revs)} || {Id, Revs} <- JsonDocIdRevs],
- {ok, Results} = couch_db:get_missing_revs(Db, DocIdRevs),
- JsonResults = [{Id, list_to_tuple(Revs)} || {Id, Revs} <- Results],
- send_json(Mod, 200, {obj, [{missing_revs, {obj, JsonResults}}]}).
-
-handle_compact(Mod, Parts) ->
- ok = couch_db:start_compact(open_db(Parts)),
- send_ok(Mod, 202).
-
-handle_replication_request(#mod{entity_body=RawJson}=Mod) ->
- {obj, Props} = cjson:decode(RawJson),
- Src = proplists:get_value("source", Props),
- Tgt = proplists:get_value("target", Props),
- {obj, Options} = proplists:get_value("options", Props, {obj, []}),
- {ok, {obj, JsonResults}} = couch_rep:replicate(Src, Tgt, Options),
- send_ok(Mod, 200, JsonResults).
-
-
-
-send_database_info(Mod, #uri_parts{db=DbName}=Parts) ->
- Db = open_db(Parts),
- {ok, InfoList} = couch_db:get_db_info(Db),
- send_json(Mod, 200, {obj, [{db_name, DbName} | InfoList]}).
-
-send_doc(#mod{parsed_header=Headers}=Mod,
- #uri_parts{doc=DocId,querystr=QueryStr}=Parts) ->
- Db = open_db(Parts),
- #doc_query_args{rev=Rev, options=Options} = doc_parse_query(QueryStr),
- case Rev of
- "" ->
- % open most recent rev
- case couch_db:open_doc(Db, DocId, Options) of
- {ok, #doc{revs=[DocRev|_]}=Doc} ->
- Etag = proplists:get_value("if-none-match", Headers),
- if Options == [] andalso Etag == DocRev ->
- ok = send_header(Mod, 304,
- resp_header(Mod, [no_body]) ++ [{"etag", DocRev}]),
- {ok, 304};
- true ->
- send_json(Mod, 200, couch_doc:to_json_obj(Doc, Options),
- if Options == [] -> [{"etag", DocRev}]; true -> [] end)
- end;
- Error ->
- throw(Error)
- end;
- _ ->
- % open a specific rev (deletions come back as stubs)
- case couch_db:open_doc_revs(Db, DocId, [Rev], Options) of
- {ok, [{ok, Doc}]} ->
- send_json(Mod, 200, couch_doc:to_json_obj(Doc, Options), [{"etag", Rev}]);
- {ok, [Else]} ->
- throw(Else)
- end
- end.
-
-send_doc_revs(Mod, #uri_parts{doc=DocId,querystr=QueryStr}=Parts) ->
- Db = open_db(Parts),
- #doc_query_args{options=Options, open_revs=Revs} = doc_parse_query(QueryStr),
- {ok, Results} = couch_db:open_doc_revs(Db, DocId, Revs, Options),
- ok = send_header(Mod, 200, resp_json_header(Mod)),
- ok = send_chunk(Mod, "["),
- % We loop through the docs. The first time through the separator
- % is whitespace, then a comma on subsequent iterations.
- lists:foldl(
- fun(Result, AccSeparator) ->
- case Result of
- {ok, Doc} ->
- JsonDoc= couch_doc:to_json_obj(Doc, Options),
- ok = send_chunk(Mod, AccSeparator ++ lists:flatten(cjson:encode({obj, [{ok, JsonDoc}]})));
- {{not_found, missing}, RevId} ->
- Json = {obj, [{"missing", RevId}]},
- ok = send_chunk(Mod, AccSeparator ++ lists:flatten(cjson:encode(Json)))
- end,
- "," % AccSeparator now has a comma
- end,
- "", Results),
- ok = send_chunk(Mod, "]"),
- ok = send_final_chunk(Mod),
- {ok, 200}.
-
-send_attachment(#mod{method=Method} = Mod,
- #uri_parts{doc=DocId,attachment=Attachment}=Parts) ->
- Db = open_db(Parts),
- case couch_db:open_doc(Db, DocId, []) of
- {ok, #doc{attachments=Attachments}} ->
- case proplists:get_value(Attachment, Attachments) of
- undefined ->
- throw({not_found, missing});
- {Type, Bin} ->
- ok = send_header(Mod, 200, resp_header(Mod, Type) ++
- [{"content-type", Type},
- {"content-length", integer_to_list(couch_doc:bin_size(Bin))}]),
- case Method of
- "GET" ->
- couch_doc:bin_foldl(Bin,
- fun(BinSegment, []) ->
- ok = send_chunk(Mod, BinSegment),
- {ok, []}
- end,
- []);
- "HEAD" ->
- ok
- end,
- ok = send_final_chunk(Mod),
- {ok, 200}
- end;
- Error ->
- throw(Error)
- end.
-
-
-send_json(Mod, Code, JsonData) ->
- send_json(Mod, Code, JsonData, []).
-
-send_json(#mod{method=Method}=Mod, Code, JsonData, Headers) ->
- case Method of
- "HEAD" ->
- ok = send_header(Mod, Code, resp_json_header(Mod, [no_body]) ++ Headers);
- _ ->
- ok = send_header(Mod, Code, resp_json_header(Mod) ++ Headers),
- ok = send_chunk(Mod, lists:flatten([cjson:encode(JsonData) | "\n"])),
- ok = send_final_chunk(Mod)
- end,
- {ok, Code}.
-
-
-send_ok(Mod, Code) ->
- send_ok(Mod, Code, []).
-
-send_ok(Mod, Code, AdditionalProps) ->
- send_ok(Mod, Code, AdditionalProps, []).
-
-send_ok(Mod, Code, AdditionalProps, AdditionalHeaders) ->
- send_json(Mod, Code, {obj, [{ok, true}|AdditionalProps]}, AdditionalHeaders).
-
-
-make_view_fold_fun(Mod, QueryArgs) ->
- #query_args{
- end_key=EndKey,
- end_docid=EndDocId,
- direction=Dir,
- count=Count
- } = QueryArgs,
-
- PassedEndFun =
- case Dir of
- fwd ->
- fun(ViewKey, ViewId) ->
- couch_view:less_json({EndKey,EndDocId}, {ViewKey,ViewId})
- end;
- rev->
- fun(ViewKey, ViewId) ->
- couch_view:less_json({ViewKey, ViewId}, {EndKey,EndDocId})
- end
- end,
-
- NegCountFun =
- fun(Id, Key, Value, Offset, TotalViewCount, {AccCount, AccSkip, HeaderSent, AccRevRows}) ->
- PassedEnd = PassedEndFun(Key, Id),
- case {PassedEnd, AccCount, AccSkip, HeaderSent} of
- {true,_,_,_} ->
- % The stop key has been passed, stop looping.
- {stop, {AccCount, AccSkip, HeaderSent, AccRevRows}};
- {_,0,_,_} ->
- {stop, {0, 0, HeaderSent, AccRevRows}}; % we've done "count" rows, stop foldling
- {_,_,AccSkip,_} when AccSkip > 0 ->
- {ok, {AccCount, AccSkip - 1, HeaderSent, AccRevRows}};
- {_,AccCount,_,header_sent} ->
- JsonObj = {obj, [{"key",Key},{"id",Id},{"value",Value}]},
- {ok, {AccCount + 1, 0, header_sent, [cjson:encode(JsonObj), "," | AccRevRows]}};
- {_,_,_,header_not_sent} ->
- ok = send_header(Mod, 200, resp_json_header(Mod)),
- Offset2= TotalViewCount - Offset -
- lists:min([TotalViewCount - Offset, - AccCount]),
- JsonBegin = io_lib:format("{\"total_rows\":~w,\"offset\":~w,\"rows\":[",
- [TotalViewCount, Offset2]),
- JsonObj = {obj, [{"key",Key},{"id",Id},{"value",Value}]},
- ok = send_chunk(Mod, lists:flatten(JsonBegin)),
- {ok, {AccCount + 1, 0, header_sent, [cjson:encode(JsonObj) | AccRevRows]}}
- end
- end,
-
- PosCountFun =
- fun(Id, Key, Value, Offset, TotalViewCount, {AccCount, AccSkip, HeaderSent, AccRevRows}) ->
- PassedEnd = PassedEndFun(Key, Id),
- case {PassedEnd, AccCount, AccSkip, HeaderSent} of
- {true,_,_,_} ->
- % The stop key has been passed, stop looping.
- {stop, {AccCount, AccSkip, HeaderSent, AccRevRows}};
- {_,0,_,_} ->
- {stop, {0, 0, HeaderSent, AccRevRows}}; % we've done "count" rows, stop foldling
- {_,_,AccSkip,_} when AccSkip > 0 ->
- {ok, {AccCount, AccSkip - 1, HeaderSent, AccRevRows}};
- {_,AccCount,_,header_sent} when (AccCount > 0) ->
- JsonObj = {obj, [{"key",Key},{"id",Id},{"value",Value}]},
- ok = send_chunk(Mod, "," ++ lists:flatten(cjson:encode(JsonObj))),
- {ok, {AccCount - 1, 0, header_sent, AccRevRows}};
- {_,_,_,header_not_sent} ->
- ok = send_header(Mod, 200, resp_json_header(Mod)),
- JsonBegin = io_lib:format("{\"total_rows\":~w,\"offset\":~w,\"rows\":[",
- [TotalViewCount, Offset]),
- JsonObj = {obj, [{"key",Key},{"id",Id},{"value",Value}]},
- ok = send_chunk(Mod, lists:flatten(JsonBegin ++ cjson:encode(JsonObj))),
- {ok, {AccCount - 1, 0, header_sent, AccRevRows}}
- end
- end,
- case Count > 0 of
- true -> PosCountFun;
- false -> NegCountFun
- end.
-
-finish_view_fold(Mod, FoldResult) ->
- case FoldResult of
- {ok, TotalRows, {_, _, header_not_sent, _}} ->
- % nothing found in the view, nothing has been returned
- % send empty view
- ok = send_header(Mod, 200, resp_json_header(Mod)),
- JsonEmptyView = lists:flatten(
- io_lib:format("{\"total_rows\":~w,\"rows\":[]}\n",
- [TotalRows])),
- ok = send_chunk(Mod, JsonEmptyView),
- ok = send_final_chunk(Mod),
- {ok, 200};
- {ok, _TotalRows, {_, _, header_sent, AccRevRows}} ->
- % end the view
- ok = send_chunk(Mod, lists:flatten(AccRevRows) ++ "]}\n"),
- ok = send_final_chunk(Mod),
- {ok, 200};
- Error ->
- throw(Error)
- end.
-
-
-send_temp_view(#mod{entity_body=Body,parsed_header=Headers}=Mod,
- #uri_parts{db=DbName, querystr=QueryStr}) ->
- #query_args{
- start_key=StartKey,
- count=Count,
- skip=SkipCount,
- direction=Dir,
- start_docid=StartDocId} = QueryArgs = view_parse_query(QueryStr),
- Type0 = proplists:get_value("content-type", Headers, "text/javascript"),
- % remove the charset ("...;charset=foo") if its there
- {ok, [Type|_]} = regexp:split(Type0, ";"),
- View = {temp, DbName, Type, Body},
- Start = {StartKey, StartDocId},
- FoldlFun = make_view_fold_fun(Mod, QueryArgs),
- FoldAccInit = {Count, SkipCount, header_not_sent, []},
- FoldResult = couch_view:fold(View, Start, Dir, FoldlFun, FoldAccInit),
- finish_view_fold(Mod, FoldResult).
-
-
-send_view(Mod, #uri_parts{db=DbName, doc=DesignDocId, view=ViewId, querystr=QueryStr}) ->
- #query_args{
- start_key=StartKey,
- count=Count,
- skip=SkipCount,
- direction=Dir,
- start_docid=StartDocId} = QueryArgs = view_parse_query(QueryStr),
- View = {DbName, DesignDocId, ViewId},
- Start = {StartKey, StartDocId},
- FoldlFun = make_view_fold_fun(Mod, QueryArgs),
- FoldAccInit = {Count, SkipCount, header_not_sent, []},
- Result = couch_view:fold(View, Start, Dir, FoldlFun, FoldAccInit),
- finish_view_fold(Mod, Result).
-
-
-send_all_docs(Mod, #uri_parts{querystr=QueryStr}=Parts) ->
- Db = open_db(Parts),
- #query_args{
- start_key=StartKey,
- start_docid=StartDocId,
- count=Count,
- skip=SkipCount,
- direction=Dir} = QueryArgs = view_parse_query(QueryStr),
- {ok, Info} = couch_db:get_db_info(Db),
- TotalRowCount = proplists:get_value(doc_count, Info),
-
- StartId =
- if is_list(StartKey) -> StartKey;
- true -> StartDocId
- end,
-
- FoldlFun = make_view_fold_fun(Mod, QueryArgs),
- AdapterFun =
- fun(#full_doc_info{id=Id}=FullDocInfo, Offset, Acc) ->
- case couch_doc:to_doc_info(FullDocInfo) of
- #doc_info{deleted=false, rev=Rev} ->
- FoldlFun(Id, Id, {obj, [{rev, Rev}]}, Offset, TotalRowCount, Acc);
- #doc_info{deleted=true} ->
- {ok, Acc}
- end
- end,
- {ok, FoldResult} = couch_db:enum_docs(Db, StartId, Dir, AdapterFun,
- {Count, SkipCount, header_not_sent, []}),
- finish_view_fold(Mod, {ok, TotalRowCount, FoldResult}).
-
-send_all_docs_by_seq(Mod, #uri_parts{querystr=QueryStr}=Parts) ->
- Db = open_db(Parts),
- QueryArgs = view_parse_query(QueryStr),
- #query_args{
- start_key=StartKey,
- count=Count,
- skip=SkipCount,
- direction=Dir} = QueryArgs,
-
- {ok, Info} = couch_db:get_db_info(Db),
- TotalRowCount = proplists:get_value(doc_count, Info),
-
- FoldlFun = make_view_fold_fun(Mod, QueryArgs),
-
- StartKey2 =
- case StartKey of
- nil -> 0;
- <<>> -> 100000000000;
- StartKey when is_integer(StartKey) -> StartKey
- end,
- {ok, FoldResult} =
- couch_db:enum_docs_since(Db, StartKey2, Dir,
- fun(DocInfo, Offset, Acc) ->
- #doc_info{
- id=Id,
- rev=Rev,
- update_seq=UpdateSeq,
- deleted=Deleted,
- conflict_revs=ConflictRevs,
- deleted_conflict_revs=DelConflictRevs} = DocInfo,
- Json =
- {obj,
- [{"rev", Rev}] ++
- case ConflictRevs of
- [] -> [];
- _ -> [{"conflicts", list_to_tuple(ConflictRevs)}]
- end ++
- case DelConflictRevs of
- [] -> [];
- _ -> [{"deleted_conflicts", list_to_tuple(DelConflictRevs)}]
- end ++
- case Deleted of
- true -> [{"deleted", true}];
- false -> []
- end
- },
- FoldlFun(Id, UpdateSeq, Json, Offset, TotalRowCount, Acc)
- end, {Count, SkipCount, header_not_sent, []}),
- finish_view_fold(Mod, {ok, TotalRowCount, FoldResult}).
-
-
-
-send_all_dbs(Mod, _Parts)->
- {ok, DbNames} = couch_server:all_databases(),
- ok = send_header(Mod, 200, resp_json_header(Mod)),
- ok = send_chunk(Mod, lists:flatten(cjson:encode(list_to_tuple(DbNames)))),
- ok = send_final_chunk(Mod),
- {ok, 200}.
-
-send_error(Mod, Error) ->
- {Json, Code} = error_to_json(Error),
- couch_log:info("HTTP Error (code ~w): ~p", [Code, Json]),
- send_json(Mod, Code, Json).
-
-
-
-% convert an error response into a json object and http error code.
-error_to_json(Error) ->
- {HttpCode, Atom, Reason} = error_to_json0(Error),
- Reason1 =
- case (catch io_lib:format("~s", [Reason])) of
- Reason0 when is_list(Reason0) ->
- lists:flatten(Reason0);
- _ ->
- lists:flatten(io_lib:format("~p", [Reason])) % else term to text
- end,
- Json =
- {obj,
- [{error, atom_to_list(Atom)},
- {reason, Reason1}]},
- {Json, HttpCode}.
-
-error_to_json0(not_found) ->
- {404, not_found, "missing"};
-error_to_json0({missing_rev, Msg}) ->
- {412, missing_rev, Msg};
-error_to_json0({not_found, Reason}) ->
- {404, not_found, Reason};
-error_to_json0({database_already_exists, Reason}) ->
- {409, database_already_exists, Reason};
-error_to_json0(conflict) ->
- {412, conflict, "Update conflict"};
-error_to_json0({doc_validation, Msg}) ->
- {406, doc_validation, Msg};
-error_to_json0({Id, Reason}) when is_atom(Id) ->
- {500, Id, Reason};
-error_to_json0(Error) ->
- {500, error, Error}.
-
-%%
-%% Configuration
-%%
-
-%% load
-
-load("Foo Bar", []) ->
- {ok, [], {script_alias, {"foo", "bar"}}}.