717 lines
25 KiB
Rust
717 lines
25 KiB
Rust
use std::sync::Arc;
|
|
|
|
use axum::{
|
|
extract::Extension,
|
|
http::{StatusCode, Uri},
|
|
middleware::{self, Next},
|
|
response::IntoResponse,
|
|
routing::{get, post},
|
|
Json, Router,
|
|
};
|
|
use tower_http::cors::{Any, CorsLayer};
|
|
use tower_http::services::{ServeDir, ServeFile};
|
|
|
|
use super::{auth, handlers, ws};
|
|
use crate::app_state::AppState;
|
|
|
|
pub fn build_router(state: Arc<AppState>, token: String, static_dir: std::path::PathBuf) -> Router {
|
|
let cors = CorsLayer::new()
|
|
.allow_origin(Any)
|
|
.allow_methods(Any)
|
|
.allow_headers(Any);
|
|
|
|
let token_for_ws = token.clone();
|
|
|
|
let api = Router::new()
|
|
.route("/health", post(health_check))
|
|
// ─── Conversations ───
|
|
.route(
|
|
"/list_conversations",
|
|
post(handlers::conversations::list_conversations),
|
|
)
|
|
.route(
|
|
"/get_conversation",
|
|
post(handlers::conversations::get_conversation),
|
|
)
|
|
.route(
|
|
"/list_all_conversations",
|
|
post(handlers::conversations::list_all_conversations),
|
|
)
|
|
.route(
|
|
"/get_folder_conversation",
|
|
post(handlers::conversations::get_folder_conversation),
|
|
)
|
|
.route(
|
|
"/list_opened_tabs",
|
|
post(handlers::conversations::list_opened_tabs),
|
|
)
|
|
.route(
|
|
"/save_opened_tabs",
|
|
post(handlers::conversations::save_opened_tabs),
|
|
)
|
|
.route(
|
|
"/import_local_conversations",
|
|
post(handlers::conversations::import_local_conversations),
|
|
)
|
|
.route("/list_folders", post(handlers::conversations::list_folders))
|
|
.route("/get_stats", post(handlers::conversations::get_stats))
|
|
.route(
|
|
"/get_sidebar_data",
|
|
post(handlers::conversations::get_sidebar_data),
|
|
)
|
|
.route(
|
|
"/create_conversation",
|
|
post(handlers::conversations::create_conversation),
|
|
)
|
|
.route(
|
|
"/update_conversation_status",
|
|
post(handlers::conversations::update_conversation_status),
|
|
)
|
|
.route(
|
|
"/update_conversation_title",
|
|
post(handlers::conversations::update_conversation_title),
|
|
)
|
|
.route(
|
|
"/delete_conversation",
|
|
post(handlers::conversations::delete_conversation),
|
|
)
|
|
.route(
|
|
"/update_conversation_external_id",
|
|
post(handlers::conversations::update_conversation_external_id),
|
|
)
|
|
// ─── Folders ───
|
|
.route(
|
|
"/load_folder_history",
|
|
post(handlers::folders::load_folder_history),
|
|
)
|
|
.route(
|
|
"/list_open_folders",
|
|
post(handlers::folders::list_open_folders),
|
|
)
|
|
.route(
|
|
"/list_open_folder_details",
|
|
post(handlers::folders::list_open_folder_details),
|
|
)
|
|
.route(
|
|
"/list_all_folder_details",
|
|
post(handlers::folders::list_all_folder_details),
|
|
)
|
|
.route("/get_folder", post(handlers::folders::get_folder))
|
|
.route("/open_folder", post(handlers::folders::open_folder))
|
|
.route(
|
|
"/open_folder_by_id",
|
|
post(handlers::folders::open_folder_by_id),
|
|
)
|
|
.route(
|
|
"/remove_folder_from_workspace",
|
|
post(handlers::folders::remove_folder_from_workspace),
|
|
)
|
|
.route(
|
|
"/add_folder_to_history",
|
|
post(handlers::folders::add_folder_to_history),
|
|
)
|
|
.route(
|
|
"/set_folder_parent_branch",
|
|
post(handlers::folders::set_folder_parent_branch),
|
|
)
|
|
.route(
|
|
"/remove_folder_from_history",
|
|
post(handlers::folders::remove_folder_from_history),
|
|
)
|
|
.route(
|
|
"/create_folder_directory",
|
|
post(handlers::folders::create_folder_directory),
|
|
)
|
|
.route("/get_git_branch", post(handlers::folders::get_git_branch))
|
|
.route(
|
|
"/get_home_directory",
|
|
post(handlers::folders::get_home_directory),
|
|
)
|
|
.route(
|
|
"/list_directory_entries",
|
|
post(handlers::folders::list_directory_entries),
|
|
)
|
|
.route("/get_file_tree", post(handlers::folders::get_file_tree))
|
|
.route(
|
|
"/start_workspace_state_stream",
|
|
post(handlers::workspace_state::start_workspace_state_stream),
|
|
)
|
|
.route(
|
|
"/stop_workspace_state_stream",
|
|
post(handlers::workspace_state::stop_workspace_state_stream),
|
|
)
|
|
.route(
|
|
"/get_workspace_snapshot",
|
|
post(handlers::workspace_state::get_workspace_snapshot),
|
|
)
|
|
// ─── Window navigation ───
|
|
.route(
|
|
"/open_settings_window",
|
|
post(handlers::folders::open_settings_window),
|
|
)
|
|
.route(
|
|
"/open_commit_window",
|
|
post(handlers::folders::open_commit_window),
|
|
)
|
|
.route(
|
|
"/open_merge_window",
|
|
post(handlers::folders::open_merge_window),
|
|
)
|
|
.route(
|
|
"/open_stash_window",
|
|
post(handlers::folders::open_stash_window),
|
|
)
|
|
.route(
|
|
"/open_push_window",
|
|
post(handlers::folders::open_push_window),
|
|
)
|
|
// ─── Git (pure) ───
|
|
.route("/git_status", post(handlers::git::git_status))
|
|
.route("/git_init", post(handlers::git::git_init))
|
|
.route("/git_log", post(handlers::git::git_log))
|
|
.route(
|
|
"/git_list_all_branches",
|
|
post(handlers::git::git_list_all_branches),
|
|
)
|
|
.route("/git_list_branches", post(handlers::git::git_list_branches))
|
|
.route(
|
|
"/git_commit_branches",
|
|
post(handlers::git::git_commit_branches),
|
|
)
|
|
.route("/git_show_file", post(handlers::git::git_show_file))
|
|
.route("/git_diff", post(handlers::git::git_diff))
|
|
.route(
|
|
"/git_diff_with_branch",
|
|
post(handlers::git::git_diff_with_branch),
|
|
)
|
|
.route("/git_show_diff", post(handlers::git::git_show_diff))
|
|
.route("/git_list_remotes", post(handlers::git::git_list_remotes))
|
|
.route("/git_add_remote", post(handlers::git::git_add_remote))
|
|
.route("/git_remove_remote", post(handlers::git::git_remove_remote))
|
|
.route(
|
|
"/git_set_remote_url",
|
|
post(handlers::git::git_set_remote_url),
|
|
)
|
|
.route("/git_new_branch", post(handlers::git::git_new_branch))
|
|
.route("/git_checkout", post(handlers::git::git_checkout))
|
|
.route("/git_reset", post(handlers::git::git_reset))
|
|
.route("/git_merge", post(handlers::git::git_merge))
|
|
.route("/git_rebase", post(handlers::git::git_rebase))
|
|
.route("/git_worktree_add", post(handlers::git::git_worktree_add))
|
|
.route("/git_push_info", post(handlers::git::git_push_info))
|
|
.route(
|
|
"/git_start_pull_merge",
|
|
post(handlers::git::git_start_pull_merge),
|
|
)
|
|
.route(
|
|
"/git_has_merge_head",
|
|
post(handlers::git::git_has_merge_head),
|
|
)
|
|
.route("/git_is_tracked", post(handlers::git::git_is_tracked))
|
|
.route("/git_rollback_file", post(handlers::git::git_rollback_file))
|
|
.route("/git_add_files", post(handlers::git::git_add_files))
|
|
.route(
|
|
"/git_list_conflicts",
|
|
post(handlers::git::git_list_conflicts),
|
|
)
|
|
.route(
|
|
"/git_conflict_file_versions",
|
|
post(handlers::git::git_conflict_file_versions),
|
|
)
|
|
.route(
|
|
"/git_resolve_conflict",
|
|
post(handlers::git::git_resolve_conflict),
|
|
)
|
|
.route(
|
|
"/git_abort_operation",
|
|
post(handlers::git::git_abort_operation),
|
|
)
|
|
.route(
|
|
"/git_continue_operation",
|
|
post(handlers::git::git_continue_operation),
|
|
)
|
|
.route("/git_stash_push", post(handlers::git::git_stash_push))
|
|
.route("/git_stash_pop", post(handlers::git::git_stash_pop))
|
|
.route("/git_stash_list", post(handlers::git::git_stash_list))
|
|
.route("/git_stash_apply", post(handlers::git::git_stash_apply))
|
|
.route("/git_stash_drop", post(handlers::git::git_stash_drop))
|
|
.route("/git_stash_clear", post(handlers::git::git_stash_clear))
|
|
.route("/git_stash_show", post(handlers::git::git_stash_show))
|
|
// ─── Git (remote) ───
|
|
.route("/git_pull", post(handlers::git::git_pull))
|
|
.route("/git_push", post(handlers::git::git_push))
|
|
.route("/git_fetch", post(handlers::git::git_fetch))
|
|
.route("/git_commit", post(handlers::git::git_commit))
|
|
.route("/git_fetch_remote", post(handlers::git::git_fetch_remote))
|
|
.route("/git_delete_branch", post(handlers::git::git_delete_branch))
|
|
.route(
|
|
"/git_delete_remote_branch",
|
|
post(handlers::git::git_delete_remote_branch),
|
|
)
|
|
.route("/clone_repository", post(handlers::git::clone_repository))
|
|
// ─── Files ───
|
|
.route(
|
|
"/read_file_preview",
|
|
post(handlers::files::read_file_preview),
|
|
)
|
|
.route("/read_file_base64", post(handlers::files::read_file_base64))
|
|
.route(
|
|
"/read_file_for_edit",
|
|
post(handlers::files::read_file_for_edit),
|
|
)
|
|
.route(
|
|
"/save_file_content",
|
|
post(handlers::files::save_file_content),
|
|
)
|
|
.route("/save_file_copy", post(handlers::files::save_file_copy))
|
|
.route(
|
|
"/rename_file_tree_entry",
|
|
post(handlers::files::rename_file_tree_entry),
|
|
)
|
|
.route(
|
|
"/delete_file_tree_entry",
|
|
post(handlers::files::delete_file_tree_entry),
|
|
)
|
|
.route(
|
|
"/create_file_tree_entry",
|
|
post(handlers::files::create_file_tree_entry),
|
|
)
|
|
// ─── Folder commands ───
|
|
.route(
|
|
"/list_folder_commands",
|
|
post(handlers::folder_commands::list_folder_commands),
|
|
)
|
|
.route(
|
|
"/create_folder_command",
|
|
post(handlers::folder_commands::create_folder_command),
|
|
)
|
|
.route(
|
|
"/update_folder_command",
|
|
post(handlers::folder_commands::update_folder_command),
|
|
)
|
|
.route(
|
|
"/delete_folder_command",
|
|
post(handlers::folder_commands::delete_folder_command),
|
|
)
|
|
.route(
|
|
"/reorder_folder_commands",
|
|
post(handlers::folder_commands::reorder_folder_commands),
|
|
)
|
|
.route(
|
|
"/bootstrap_folder_commands_from_package_json",
|
|
post(handlers::folder_commands::bootstrap_folder_commands_from_package_json),
|
|
)
|
|
// ─── MCP ───
|
|
.route("/mcp_scan_local", post(handlers::mcp::mcp_scan_local))
|
|
.route(
|
|
"/mcp_list_marketplaces",
|
|
post(handlers::mcp::mcp_list_marketplaces),
|
|
)
|
|
.route(
|
|
"/mcp_search_marketplace",
|
|
post(handlers::mcp::mcp_search_marketplace),
|
|
)
|
|
.route(
|
|
"/mcp_get_marketplace_server_detail",
|
|
post(handlers::mcp::mcp_get_marketplace_server_detail),
|
|
)
|
|
.route(
|
|
"/mcp_install_from_marketplace",
|
|
post(handlers::mcp::mcp_install_from_marketplace),
|
|
)
|
|
.route(
|
|
"/mcp_upsert_local_server",
|
|
post(handlers::mcp::mcp_upsert_local_server),
|
|
)
|
|
.route(
|
|
"/mcp_set_server_apps",
|
|
post(handlers::mcp::mcp_set_server_apps),
|
|
)
|
|
.route("/mcp_remove_server", post(handlers::mcp::mcp_remove_server))
|
|
// ─── Version control settings ───
|
|
.route("/detect_git", post(handlers::version_control::detect_git))
|
|
.route(
|
|
"/test_git_path",
|
|
post(handlers::version_control::test_git_path),
|
|
)
|
|
.route(
|
|
"/get_git_settings",
|
|
post(handlers::version_control::get_git_settings),
|
|
)
|
|
.route(
|
|
"/update_git_settings",
|
|
post(handlers::version_control::update_git_settings),
|
|
)
|
|
.route(
|
|
"/get_github_accounts",
|
|
post(handlers::version_control::get_github_accounts),
|
|
)
|
|
.route(
|
|
"/update_github_accounts",
|
|
post(handlers::version_control::update_github_accounts),
|
|
)
|
|
.route(
|
|
"/validate_github_token",
|
|
post(handlers::version_control::validate_github_token),
|
|
)
|
|
.route(
|
|
"/save_account_token",
|
|
post(handlers::version_control::save_account_token),
|
|
)
|
|
.route(
|
|
"/get_account_token",
|
|
post(handlers::version_control::get_account_token),
|
|
)
|
|
.route(
|
|
"/delete_account_token",
|
|
post(handlers::version_control::delete_account_token),
|
|
)
|
|
// ─── System settings ───
|
|
.route(
|
|
"/get_system_proxy_settings",
|
|
post(handlers::system_settings::get_system_proxy_settings),
|
|
)
|
|
.route(
|
|
"/get_system_language_settings",
|
|
post(handlers::system_settings::get_system_language_settings),
|
|
)
|
|
.route(
|
|
"/update_system_proxy_settings",
|
|
post(handlers::system_settings::update_system_proxy_settings),
|
|
)
|
|
.route(
|
|
"/update_system_language_settings",
|
|
post(handlers::system_settings::update_system_language_settings),
|
|
)
|
|
// ─── ACP ───
|
|
.route(
|
|
"/acp_get_agent_status",
|
|
post(handlers::acp::acp_get_agent_status),
|
|
)
|
|
.route("/acp_list_agents", post(handlers::acp::acp_list_agents))
|
|
.route("/acp_connect", post(handlers::acp::acp_connect))
|
|
.route("/acp_disconnect", post(handlers::acp::acp_disconnect))
|
|
.route("/acp_prompt", post(handlers::acp::acp_prompt))
|
|
.route("/acp_preflight", post(handlers::acp::acp_preflight))
|
|
.route("/acp_set_mode", post(handlers::acp::acp_set_mode))
|
|
.route(
|
|
"/acp_set_config_option",
|
|
post(handlers::acp::acp_set_config_option),
|
|
)
|
|
.route("/acp_cancel", post(handlers::acp::acp_cancel))
|
|
.route("/acp_fork", post(handlers::acp::acp_fork))
|
|
.route(
|
|
"/acp_respond_permission",
|
|
post(handlers::acp::acp_respond_permission),
|
|
)
|
|
.route(
|
|
"/acp_list_connections",
|
|
post(handlers::acp::acp_list_connections),
|
|
)
|
|
.route(
|
|
"/acp_clear_binary_cache",
|
|
post(handlers::acp::acp_clear_binary_cache),
|
|
)
|
|
.route(
|
|
"/acp_update_agent_preferences",
|
|
post(handlers::acp::acp_update_agent_preferences),
|
|
)
|
|
.route(
|
|
"/acp_update_agent_env",
|
|
post(handlers::acp::acp_update_agent_env),
|
|
)
|
|
.route(
|
|
"/acp_update_agent_config",
|
|
post(handlers::acp::acp_update_agent_config),
|
|
)
|
|
.route(
|
|
"/acp_download_agent_binary",
|
|
post(handlers::acp::acp_download_agent_binary),
|
|
)
|
|
.route(
|
|
"/acp_detect_agent_local_version",
|
|
post(handlers::acp::acp_detect_agent_local_version),
|
|
)
|
|
.route(
|
|
"/acp_prepare_npx_agent",
|
|
post(handlers::acp::acp_prepare_npx_agent),
|
|
)
|
|
.route(
|
|
"/acp_uninstall_agent",
|
|
post(handlers::acp::acp_uninstall_agent),
|
|
)
|
|
.route(
|
|
"/acp_reorder_agents",
|
|
post(handlers::acp::acp_reorder_agents),
|
|
)
|
|
.route(
|
|
"/acp_list_agent_skills",
|
|
post(handlers::acp::acp_list_agent_skills),
|
|
)
|
|
.route(
|
|
"/acp_read_agent_skill",
|
|
post(handlers::acp::acp_read_agent_skill),
|
|
)
|
|
.route(
|
|
"/acp_save_agent_skill",
|
|
post(handlers::acp::acp_save_agent_skill),
|
|
)
|
|
.route(
|
|
"/acp_delete_agent_skill",
|
|
post(handlers::acp::acp_delete_agent_skill),
|
|
)
|
|
.route(
|
|
"/opencode_list_plugins",
|
|
post(handlers::acp::opencode_list_plugins),
|
|
)
|
|
.route(
|
|
"/opencode_install_plugins",
|
|
post(handlers::acp::opencode_install_plugins),
|
|
)
|
|
.route(
|
|
"/opencode_uninstall_plugin",
|
|
post(handlers::acp::opencode_uninstall_plugin),
|
|
)
|
|
.route(
|
|
"/codex_request_device_code",
|
|
post(handlers::acp::codex_request_device_code),
|
|
)
|
|
.route(
|
|
"/codex_poll_device_code",
|
|
post(handlers::acp::codex_poll_device_code),
|
|
)
|
|
// ─── Experts ───
|
|
.route("/experts_list", post(handlers::experts::experts_list))
|
|
.route(
|
|
"/experts_list_for_agent",
|
|
post(handlers::experts::experts_list_for_agent),
|
|
)
|
|
.route(
|
|
"/experts_get_install_status",
|
|
post(handlers::experts::experts_get_install_status),
|
|
)
|
|
.route(
|
|
"/experts_link_to_agent",
|
|
post(handlers::experts::experts_link_to_agent),
|
|
)
|
|
.route(
|
|
"/experts_unlink_from_agent",
|
|
post(handlers::experts::experts_unlink_from_agent),
|
|
)
|
|
.route(
|
|
"/experts_read_content",
|
|
post(handlers::experts::experts_read_content),
|
|
)
|
|
.route(
|
|
"/experts_open_central_dir",
|
|
post(handlers::experts::experts_open_central_dir),
|
|
)
|
|
// ─── Project boot ───
|
|
.route(
|
|
"/detect_package_manager",
|
|
post(handlers::project_boot::detect_package_manager),
|
|
)
|
|
.route(
|
|
"/create_shadcn_project",
|
|
post(handlers::project_boot::create_shadcn_project),
|
|
)
|
|
// ─── Web Server ───
|
|
.route(
|
|
"/get_web_server_status",
|
|
post(handlers::web_server::get_web_server_status),
|
|
)
|
|
.route(
|
|
"/start_web_server",
|
|
post(handlers::web_server::start_web_server),
|
|
)
|
|
.route(
|
|
"/stop_web_server",
|
|
post(handlers::web_server::stop_web_server),
|
|
)
|
|
.route(
|
|
"/check_app_update",
|
|
post(handlers::web_server::check_app_update),
|
|
)
|
|
// ─── Chat Channels ───
|
|
.route(
|
|
"/list_chat_channels",
|
|
post(handlers::chat_channel::list_chat_channels),
|
|
)
|
|
.route(
|
|
"/create_chat_channel",
|
|
post(handlers::chat_channel::create_chat_channel),
|
|
)
|
|
.route(
|
|
"/update_chat_channel",
|
|
post(handlers::chat_channel::update_chat_channel),
|
|
)
|
|
.route(
|
|
"/delete_chat_channel",
|
|
post(handlers::chat_channel::delete_chat_channel),
|
|
)
|
|
.route(
|
|
"/save_chat_channel_token",
|
|
post(handlers::chat_channel::save_chat_channel_token),
|
|
)
|
|
.route(
|
|
"/get_chat_channel_has_token",
|
|
post(handlers::chat_channel::get_chat_channel_has_token),
|
|
)
|
|
.route(
|
|
"/delete_chat_channel_token",
|
|
post(handlers::chat_channel::delete_chat_channel_token),
|
|
)
|
|
.route(
|
|
"/connect_chat_channel",
|
|
post(handlers::chat_channel::connect_chat_channel),
|
|
)
|
|
.route(
|
|
"/disconnect_chat_channel",
|
|
post(handlers::chat_channel::disconnect_chat_channel),
|
|
)
|
|
.route(
|
|
"/test_chat_channel",
|
|
post(handlers::chat_channel::test_chat_channel),
|
|
)
|
|
.route(
|
|
"/get_chat_channel_status",
|
|
post(handlers::chat_channel::get_chat_channel_status),
|
|
)
|
|
.route(
|
|
"/list_chat_channel_messages",
|
|
post(handlers::chat_channel::list_chat_channel_messages),
|
|
)
|
|
.route(
|
|
"/get_chat_command_prefix",
|
|
post(handlers::chat_channel::get_chat_command_prefix),
|
|
)
|
|
.route(
|
|
"/set_chat_command_prefix",
|
|
post(handlers::chat_channel::set_chat_command_prefix),
|
|
)
|
|
.route(
|
|
"/get_chat_event_filter",
|
|
post(handlers::chat_channel::get_chat_event_filter),
|
|
)
|
|
.route(
|
|
"/set_chat_event_filter",
|
|
post(handlers::chat_channel::set_chat_event_filter),
|
|
)
|
|
.route(
|
|
"/get_chat_message_language",
|
|
post(handlers::chat_channel::get_chat_message_language),
|
|
)
|
|
.route(
|
|
"/set_chat_message_language",
|
|
post(handlers::chat_channel::set_chat_message_language),
|
|
)
|
|
.route(
|
|
"/weixin_get_qrcode",
|
|
post(handlers::chat_channel::weixin_get_qrcode),
|
|
)
|
|
.route(
|
|
"/weixin_check_qrcode",
|
|
post(handlers::chat_channel::weixin_check_qrcode),
|
|
)
|
|
// ─── Model Providers ───
|
|
.route(
|
|
"/list_model_providers",
|
|
post(handlers::model_provider::list_model_providers),
|
|
)
|
|
.route(
|
|
"/create_model_provider",
|
|
post(handlers::model_provider::create_model_provider),
|
|
)
|
|
.route(
|
|
"/update_model_provider",
|
|
post(handlers::model_provider::update_model_provider),
|
|
)
|
|
.route(
|
|
"/delete_model_provider",
|
|
post(handlers::model_provider::delete_model_provider),
|
|
)
|
|
// ─── Terminal ───
|
|
.route("/terminal_spawn", post(handlers::terminal::terminal_spawn))
|
|
.route("/terminal_write", post(handlers::terminal::terminal_write))
|
|
.route(
|
|
"/terminal_resize",
|
|
post(handlers::terminal::terminal_resize),
|
|
)
|
|
.route("/terminal_kill", post(handlers::terminal::terminal_kill))
|
|
.route("/terminal_list", post(handlers::terminal::terminal_list))
|
|
// Catch-all
|
|
.fallback(api_not_found)
|
|
.layer(middleware::from_fn(move |req, next| {
|
|
auth::require_token(req, next, token.clone())
|
|
}));
|
|
|
|
// WebSocket route (auth via query param)
|
|
let ws_route = Router::new()
|
|
.route("/ws/events", get(ws::ws_handler))
|
|
.layer(middleware::from_fn(move |req, next| {
|
|
auth::require_token(req, next, token_for_ws.clone())
|
|
}));
|
|
|
|
// Static file serving.
|
|
// Next.js static export produces "folder.html" for "/folder" route.
|
|
// We use a middleware to rewrite "/folder" → "/folder.html" before ServeDir.
|
|
let fallback =
|
|
ServeDir::new(&static_dir).fallback(ServeFile::new(static_dir.join("index.html")));
|
|
|
|
let static_dir_for_mw = static_dir.clone();
|
|
let html_rewrite = middleware::from_fn(move |req: axum::extract::Request, next: Next| {
|
|
let dir = static_dir_for_mw.clone();
|
|
async move {
|
|
let path = req.uri().path();
|
|
// If path has no extension (not a file) and a .html version exists, rewrite
|
|
if path != "/"
|
|
&& !path.contains('.')
|
|
&& !path.starts_with("/api")
|
|
&& !path.starts_with("/ws")
|
|
{
|
|
let html_path = format!("{}.html", path.trim_end_matches('/'));
|
|
let html_file = dir.join(html_path.trim_start_matches('/'));
|
|
if html_file.exists() {
|
|
// Rebuild URI with .html suffix preserving query string
|
|
let new_path = if let Some(q) = req.uri().query() {
|
|
format!("{}?{}", html_path, q)
|
|
} else {
|
|
html_path
|
|
};
|
|
if let Ok(new_uri) = new_path.parse::<Uri>() {
|
|
let (mut parts, body) = req.into_parts();
|
|
parts.uri = new_uri;
|
|
let req = axum::extract::Request::from_parts(parts, body);
|
|
return next.run(req).await;
|
|
}
|
|
}
|
|
}
|
|
next.run(req).await
|
|
}
|
|
});
|
|
|
|
Router::new()
|
|
.nest("/api", api)
|
|
.merge(ws_route)
|
|
.fallback_service(fallback)
|
|
.layer(html_rewrite)
|
|
.layer(cors)
|
|
.layer(Extension(state))
|
|
}
|
|
|
|
async fn health_check() -> impl IntoResponse {
|
|
Json(serde_json::json!({ "status": "ok" }))
|
|
}
|
|
|
|
async fn api_not_found(uri: axum::http::Uri) -> impl IntoResponse {
|
|
let command = uri.path().trim_start_matches('/');
|
|
eprintln!("[WEB] Unimplemented API endpoint: {}", command);
|
|
(
|
|
StatusCode::NOT_IMPLEMENTED,
|
|
Json(serde_json::json!({
|
|
"code": "not_implemented",
|
|
"message": format!("API endpoint '{}' is not available in web mode", command),
|
|
})),
|
|
)
|
|
}
|