% 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(chttpd_misc). -export([handle_welcome_req/2,handle_favicon_req/2,handle_utils_dir_req/2, handle_all_dbs_req/1,handle_replicate_req/1,handle_restart_req/1, handle_uuids_req/1,handle_config_req/1,handle_log_req/1, handle_task_status_req/1,handle_sleep_req/1,handle_welcome_req/1, handle_utils_dir_req/1, handle_favicon_req/1, handle_system_req/1]). -include_lib("couch/include/couch_db.hrl"). -import(chttpd, [send_json/2,send_json/3,send_json/4,send_method_not_allowed/2, start_json_response/2,send_chunk/2,end_json_response/1, start_chunked_response/3, send_error/4]). % httpd global handlers handle_welcome_req(Req) -> handle_welcome_req(Req, <<"Welcome">>). handle_welcome_req(#httpd{method='GET'}=Req, WelcomeMessage) -> send_json(Req, {[ {couchdb, WelcomeMessage}, {version, list_to_binary(couch:version())}, {bigcouch, get_version()} ]}); handle_welcome_req(Req, _) -> send_method_not_allowed(Req, "GET,HEAD"). get_version() -> Releases = release_handler:which_releases(), Version = case [V || {"bigcouch", V, _, current} <- Releases] of [] -> case [V || {"bigcouch", V, _, permanent} <- Releases] of [] -> "dev"; [Permanent] -> Permanent end; [Current] -> Current end, list_to_binary(Version). handle_favicon_req(Req) -> handle_favicon_req(Req, couch_config:get("chttpd", "docroot")). handle_favicon_req(#httpd{method='GET'}=Req, DocumentRoot) -> chttpd:serve_file(Req, "favicon.ico", DocumentRoot); handle_favicon_req(Req, _) -> send_method_not_allowed(Req, "GET,HEAD"). handle_utils_dir_req(Req) -> handle_utils_dir_req(Req, couch_config:get("chttpd", "docroot")). handle_utils_dir_req(#httpd{method='GET'}=Req, DocumentRoot) -> "/" ++ UrlPath = chttpd:path(Req), case chttpd:partition(UrlPath) of {_ActionKey, "/", RelativePath} -> % GET /_utils/path or GET /_utils/ chttpd:serve_file(Req, RelativePath, DocumentRoot); {_ActionKey, "", _RelativePath} -> % GET /_utils RedirectPath = chttpd:path(Req) ++ "/", chttpd:send_redirect(Req, RedirectPath) end; handle_utils_dir_req(Req, _) -> send_method_not_allowed(Req, "GET,HEAD"). handle_sleep_req(#httpd{method='GET'}=Req) -> Time = list_to_integer(chttpd:qs_value(Req, "time")), receive snicklefart -> ok after Time -> ok end, send_json(Req, {[{ok, true}]}); handle_sleep_req(Req) -> send_method_not_allowed(Req, "GET,HEAD"). handle_all_dbs_req(#httpd{method='GET'}=Req) -> {ok, DbNames} = fabric:all_dbs(), send_json(Req, DbNames); handle_all_dbs_req(Req) -> send_method_not_allowed(Req, "GET,HEAD"). handle_task_status_req(#httpd{method='GET'}=Req) -> {Replies, _BadNodes} = gen_server:multi_call(couch_task_status, all), Response = lists:flatmap(fun({Node, Tasks}) -> [{[{node,Node} | Task]} || Task <- Tasks] end, Replies), send_json(Req, lists:sort(Response)); handle_task_status_req(Req) -> send_method_not_allowed(Req, "GET,HEAD"). handle_replicate_req(#httpd{method='POST', user_ctx=Ctx} = Req) -> PostBody = chttpd:json_body_obj(Req), try couch_rep:replicate(PostBody, Ctx) of {ok, {continuous, RepId}} -> send_json(Req, 202, {[{ok, true}, {<<"_local_id">>, RepId}]}); {ok, {cancelled, RepId}} -> send_json(Req, 200, {[{ok, true}, {<<"_local_id">>, RepId}]}); {ok, {JsonResults}} -> send_json(Req, {[{ok, true} | JsonResults]}); {error, {Type, Details}} -> send_json(Req, 500, {[{error, Type}, {reason, Details}]}); {error, not_found} -> send_json(Req, 404, {[{error, not_found}]}); {error, Reason} -> send_json(Req, 500, {[{error, Reason}]}) catch throw:{db_not_found, Msg} -> send_json(Req, 404, {[{error, db_not_found}, {reason, Msg}]}); throw:{node_not_connected, Msg} -> send_json(Req, 404, {[{error, node_not_connected}, {reason, Msg}]}) end; handle_replicate_req(Req) -> send_method_not_allowed(Req, "POST"). handle_restart_req(#httpd{method='POST'}=Req) -> couch_server_sup:restart_core_server(), send_json(Req, 200, {[{ok, true}]}); handle_restart_req(Req) -> send_method_not_allowed(Req, "POST"). handle_uuids_req(Req) -> couch_httpd_misc_handlers:handle_uuids_req(Req). % Config request handler % GET /_config/ % GET /_config handle_config_req(#httpd{method='GET', path_parts=[_]}=Req) -> Grouped = lists:foldl(fun({{Section, Key}, Value}, Acc) -> case dict:is_key(Section, Acc) of true -> dict:append(Section, {list_to_binary(Key), list_to_binary(Value)}, Acc); false -> dict:store(Section, [{list_to_binary(Key), list_to_binary(Value)}], Acc) end end, dict:new(), couch_config:all()), KVs = dict:fold(fun(Section, Values, Acc) -> [{list_to_binary(Section), {Values}} | Acc] end, [], Grouped), send_json(Req, 200, {KVs}); % GET /_config/Section handle_config_req(#httpd{method='GET', path_parts=[_,Section]}=Req) -> KVs = [{list_to_binary(Key), list_to_binary(Value)} || {Key, Value} <- couch_config:get(Section)], send_json(Req, 200, {KVs}); % PUT /_config/Section/Key % "value" handle_config_req(#httpd{method='PUT', path_parts=[_, Section, Key]}=Req) -> Value = chttpd:json_body(Req), Persist = chttpd:header_value(Req, "X-Couch-Persist") /= "false", OldValue = couch_config:get(Section, Key, ""), ok = couch_config:set(Section, Key, ?b2l(Value), Persist), send_json(Req, 200, list_to_binary(OldValue)); % GET /_config/Section/Key handle_config_req(#httpd{method='GET', path_parts=[_, Section, Key]}=Req) -> case couch_config:get(Section, Key, null) of null -> throw({not_found, unknown_config_value}); Value -> send_json(Req, 200, list_to_binary(Value)) end; % DELETE /_config/Section/Key handle_config_req(#httpd{method='DELETE',path_parts=[_,Section,Key]}=Req) -> Persist = chttpd:header_value(Req, "X-Couch-Persist") /= "false", case couch_config:get(Section, Key, null) of null -> throw({not_found, unknown_config_value}); OldValue -> couch_config:delete(Section, Key, Persist), send_json(Req, 200, list_to_binary(OldValue)) end; handle_config_req(Req) -> send_method_not_allowed(Req, "GET,PUT,DELETE"). % httpd log handlers handle_log_req(#httpd{method='GET'}=Req) -> Bytes = list_to_integer(chttpd:qs_value(Req, "bytes", "1000")), Offset = list_to_integer(chttpd:qs_value(Req, "offset", "0")), Chunk = couch_log:read(Bytes, Offset), {ok, Resp} = start_chunked_response(Req, 200, [ % send a plaintext response {"Content-Type", "text/plain; charset=utf-8"}, {"Content-Length", integer_to_list(length(Chunk))} ]), send_chunk(Resp, Chunk), send_chunk(Resp, ""); handle_log_req(Req) -> send_method_not_allowed(Req, "GET"). % Note: this resource is exposed on the backdoor interface, but it's in chttpd % because it's not couch trunk handle_system_req(Req) -> Other = erlang:memory(system) - lists:sum([X || {_,X} <- erlang:memory([atom, code, binary, ets])]), Memory = [{other, Other} | erlang:memory([atom, atom_used, processes, processes_used, binary, code, ets])], send_json(Req, {[ {memory, {Memory}}, {run_queue, statistics(run_queue)}, {process_count, erlang:system_info(process_count)}, {process_limit, erlang:system_info(process_limit)} ]}).