% 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_metrics_req/1]). -export([increment_update_seq_req/2]). -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())}, {cloudant_build, get_version()} ]}); handle_welcome_req(Req, _) -> send_method_not_allowed(Req, "GET,HEAD"). get_version() -> Releases = release_handler:which_releases(), Version = case [V || {"dbcore", V, _, current} <- Releases] of [] -> case [V || {"dbcore", 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) -> Customer = cloudant_util:customer_name(Req), {ok, DbNames} = fabric:all_dbs(Customer), 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), % TODO filter by customer send_json(Req, Response); handle_task_status_req(Req) -> send_method_not_allowed(Req, "GET,HEAD"). handle_replicate_req(#httpd{method='POST'}=Req) -> PostBody = get(post_body), try showroom_api:replicate_db(PostBody, Req#httpd.user_ctx) of {ok, {continuous, RepId}} -> send_json(Req, 202, {[{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, 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 db handlers increment_update_seq_req(#httpd{method='POST'}=Req, Db) -> {ok, NewSeq} = ?COUCH:increment_update_seq(Db), send_json(Req, {[{ok, true}, {update_seq, NewSeq} ]}); increment_update_seq_req(Req, _Db) -> send_method_not_allowed(Req, "POST"). % 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"). handle_metrics_req(#httpd{method='GET', path_parts=[_]}=Req) -> {_, L} = lists:unzip(gen_event:which_handlers(couch_metrics_event_manager)), send_json(Req, L); handle_metrics_req(#httpd{method='GET', path_parts=[_, Id]}=Req) -> case chttpd:qs_value(Req, "slice") of undefined -> Data = couch_metrics_event:get_global_metrics(?b2a(Id)); SliceStr -> Slice = list_to_integer(SliceStr), Data = couch_metrics_event:get_global_metrics(?b2a(Id), Slice) end, send_json(Req, Data); handle_metrics_req(Req) -> send_method_not_allowed(Req, "GET,HEAD").