use std::borrow::Cow;

use lsp_types::DocumentSymbolRequest;
use lsp_types::{DocumentSymbol, DocumentSymbolParams, Uri};
use ruff_db::files::File;
use ty_ide::{HierarchicalSymbols, SymbolId, SymbolInfo, document_symbols};
use ty_project::{ProjectDatabase, SemanticDb as _};

use crate::Db;
use crate::document::{PositionEncoding, ToRangeExt};
use crate::server::api::symbols::{convert_symbol_kind, convert_to_lsp_symbol_information};
use crate::server::api::traits::{
    BackgroundDocumentRequestHandler, RequestHandler, RetriableRequestHandler,
};
use crate::session::DocumentSnapshot;
use crate::session::client::Client;

pub(crate) struct DocumentSymbolRequestHandler;

impl RequestHandler for DocumentSymbolRequestHandler {
    type RequestType = DocumentSymbolRequest;
}

impl BackgroundDocumentRequestHandler for DocumentSymbolRequestHandler {
    fn document_uri(params: &DocumentSymbolParams) -> Cow<'_, Uri> {
        Cow::Borrowed(&params.text_document.uri)
    }

    fn run_with_snapshot(
        db: &ProjectDatabase,
        snapshot: &DocumentSnapshot,
        _client: &Client,
        _params: DocumentSymbolParams,
    ) -> crate::server::Result<Option<lsp_types::DocumentSymbolResponse>> {
        if snapshot
            .workspace_settings()
            .is_language_services_disabled()
        {
            return Ok(None);
        }

        let Some(file) = snapshot.to_notebook_or_file(db) else {
            return Ok(None);
        };

        // Check if the client supports hierarchical document symbols
        let supports_hierarchical = snapshot
            .resolved_client_capabilities()
            .supports_hierarchical_document_symbols();

        let symbols = document_symbols(db, db.program_file(file));
        if symbols.is_empty() {
            return Ok(None);
        }

        if supports_hierarchical {
            let symbols = symbols.to_hierarchical();
            let lsp_symbols = symbols
                .iter()
                .filter_map(|(id, symbol)| {
                    convert_to_lsp_document_symbol(
                        db,
                        file,
                        &symbols,
                        id,
                        symbol,
                        snapshot.encoding(),
                    )
                })
                .collect();

            Ok(Some(lsp_types::DocumentSymbolResponse::DocumentSymbolList(
                lsp_symbols,
            )))
        } else {
            // Return flattened symbols as SymbolInformation
            let lsp_symbols = symbols
                .iter()
                .filter_map(|(_, symbol)| {
                    convert_to_lsp_symbol_information(db, file, symbol, snapshot.encoding())
                })
                .collect();

            Ok(Some(
                lsp_types::DocumentSymbolResponse::SymbolInformationList(lsp_symbols),
            ))
        }
    }
}

impl RetriableRequestHandler for DocumentSymbolRequestHandler {}

fn convert_to_lsp_document_symbol(
    db: &dyn Db,
    file: File,
    symbols: &HierarchicalSymbols,
    id: SymbolId,
    symbol: SymbolInfo<'_>,
    encoding: PositionEncoding,
) -> Option<DocumentSymbol> {
    let symbol_kind = convert_symbol_kind(symbol.kind);

    Some(DocumentSymbol {
        name: symbol.name.into_owned(),
        detail: None,
        kind: symbol_kind,
        tags: None,
        #[allow(deprecated)]
        deprecated: None,
        range: symbol
            .full_range
            .to_lsp_range(db, file, encoding)?
            .local_range(),
        selection_range: symbol
            .name_range
            .to_lsp_range(db, file, encoding)?
            .local_range(),
        children: Some(
            symbols
                .children(id)
                .filter_map(|(child_id, child)| {
                    convert_to_lsp_document_symbol(db, file, symbols, child_id, child, encoding)
                })
                .collect(),
        ),
    })
}
