wat_service 0.8.0

use crate::{
    binder::{SymbolKey, SymbolTable},
    document::Document,
    helpers,
};
use rowan::TextRange;
use rustc_hash::FxHashMap;
use wat_syntax::{SyntaxKind, SyntaxNodePtr};

#[salsa::tracked(returns(ref))]
pub(crate) fn get_exports<'db>(
    db: &'db dyn salsa::Database,
    document: Document,
) -> FxHashMap<SyntaxNodePtr, Vec<Export>> {
    let symbol_table = SymbolTable::of(db, document);
    document
        .root_tree(db)
        .children()
        .map(|module| {
            let mut exports = Vec::new();
            module.children().for_each(|module_field| {
                if module_field.kind() == SyntaxKind::MODULE_FIELD_EXPORT {
                    if let Some(name) =
                        module_field.first_child_by_kind(&|kind| kind == SyntaxKind::NAME)
                        && let Some(def_key) = helpers::ast::extract_index_from_export(
                            &module_field,
                        )
                        .and_then(|index| symbol_table.resolved.get(&SymbolKey::new(&index)))
                    {
                        exports.push(Export {
                            def_key: *def_key,
                            name: name.to_string(),
                            range: name.text_range(),
                        });
                    }
                } else {
                    exports.extend(
                        module_field
                            .children()
                            .filter(|child| child.kind() == SyntaxKind::EXPORT)
                            .filter_map(|export| {
                                export.first_child_by_kind(&|kind| kind == SyntaxKind::NAME)
                            })
                            .map(|name| Export {
                                def_key: SymbolKey::new(&module_field),
                                name: name.to_string(),
                                range: name.text_range(),
                            }),
                    );
                }
            });
            (SyntaxNodePtr::new(&module), exports)
        })
        .collect()
}

#[derive(Clone, Debug, PartialEq, Eq)]
pub(crate) struct Export {
    pub def_key: SymbolKey,
    /// Export name contains double quotes.
    pub name: String,
    pub range: TextRange,
}