use std::collections::BTreeMap;

use axum::extract::State;
use ruma::{
	RoomVersionId,
	api::client::discovery::{
		get_capabilities,
		get_capabilities::v3::{
			Capabilities, GetLoginTokenCapability, ProfileFieldsCapability, RoomVersionStability,
			RoomVersionsCapability, ThirdPartyIdChangesCapability,
		},
	},
};
use serde_json::json;
use tuwunel_core::{Result, Server};

use crate::Ruma;

/// # `GET /_matrix/client/v3/capabilities`
///
/// Get information on the supported feature set and other relevant capabilities
/// of this server.
pub(crate) async fn get_capabilities_route(
	State(services): State<crate::State>,
	_body: Ruma<get_capabilities::v3::Request>,
) -> Result<get_capabilities::v3::Response> {
	let available: BTreeMap<RoomVersionId, RoomVersionStability> =
		Server::available_room_versions().collect();

	let mut capabilities = Capabilities::default();
	capabilities.room_versions = RoomVersionsCapability {
		available,
		default: services
			.server
			.config
			.default_room_version
			.clone(),
	};

	// we do not implement 3PID stuff
	capabilities.thirdparty_id_changes = ThirdPartyIdChangesCapability { enabled: false };

	capabilities.get_login_token = GetLoginTokenCapability {
		enabled: services.server.config.login_via_existing_session,
	};

	capabilities.profile_fields = ProfileFieldsCapability::new(true).into();

	capabilities.set(
		"org.matrix.msc4267.forget_forced_upon_leave",
		json!({"enabled": services.config.forget_forced_upon_leave}),
	)?;

	Ok(get_capabilities::v3::Response { capabilities })
}