cairo-lang-defs 2.18.0

use std::hash::{Hash, Hasher};
use std::ops::{Deref, DerefMut};
use std::path::PathBuf;
use std::sync::Arc;

use cairo_lang_diagnostics::{DiagnosticNote, Severity};
use cairo_lang_filesystem::db::FilesGroup;
use cairo_lang_filesystem::flag::{Flag, FlagsGroup};
use cairo_lang_filesystem::ids::{
    CodeMapping, CrateId, CrateLongId, FileId, FileKind, FileLongId, SmolStrId, SpanInFile,
    VirtualFile,
};
use cairo_lang_filesystem::span::{TextSpan, TextWidth};
use cairo_lang_syntax::node::ast::{
    FunctionWithBodyPtr, GenericParamPtr, ItemConstantPtr, ItemEnumPtr, ItemExternFunctionPtr,
    ItemExternTypePtr, ItemImplAliasPtr, ItemImplPtr, ItemInlineMacroPtr, ItemMacroDeclarationPtr,
    ItemModulePtr, ItemStructPtr, ItemTraitPtr, ItemTypeAliasPtr, UsePathLeafPtr, UsePathStarPtr,
};
use cairo_lang_syntax::node::green::{GreenNode, GreenNodeDetails};
use cairo_lang_syntax::node::ids::{GreenId, SyntaxStablePtrId};
use cairo_lang_syntax::node::kind::SyntaxKind;
use cairo_lang_syntax::node::{SyntaxNode, SyntaxNodeId, TypedSyntaxNode, ast};
use cairo_lang_utils::Intern;
use cairo_lang_utils::ordered_hash_map::OrderedHashMap;
use salsa::Database;
use serde::{Deserialize, Serialize};

use crate::db::{
    ModuleData, ModuleDataCacheAndLoadingData, ModuleFilesData, ModuleNamedItemsData,
    ModuleTypesData, ModuleUnnamedItemsData,
};
use crate::ids::{
    ConstantId, ConstantLongId, EnumId, EnumLongId, ExternFunctionId, ExternFunctionLongId,
    ExternTypeId, ExternTypeLongId, FreeFunctionId, FreeFunctionLongId, GenericParamId,
    GenericParamLongId, GlobalUseId, GlobalUseLongId, ImplAliasId, ImplAliasLongId, ImplDefId,
    ImplDefLongId, LanguageElementId, MacroCallId, MacroCallLongId, MacroDeclarationId,
    MacroDeclarationLongId, ModuleId, ModuleItemId, ModuleTypeAliasId, ModuleTypeAliasLongId,
    PluginGeneratedFileId, PluginGeneratedFileLongId, StructId, StructLongId, SubmoduleId,
    SubmoduleLongId, TraitId, TraitLongId, UseId, UseLongId,
};
use crate::plugin::{DynGeneratedFileAuxData, PluginDiagnostic};

/// Metadata for a cached crate.
#[derive(Serialize, Deserialize)]
pub struct CachedCrateMetadata {
    /// Hash of the settings the crate was compiled with.
    pub settings: Option<u64>,
    /// The version of the compiler that compiled the crate.
    pub compiler_version: String,
    /// The global flags the crate was compiled with.
    pub global_flags: OrderedHashMap<String, Flag>,
}

impl CachedCrateMetadata {
    /// Creates a new [CachedCrateMetadata] from the input crate with the current settings.
    pub fn new(crate_id: CrateId<'_>, db: &dyn Database) -> Self {
        let settings = db.crate_config(crate_id).map(|config| &config.settings).map(|v| {
            let mut hasher = xxhash_rust::xxh3::Xxh3::default();
            v.hash(&mut hasher);
            hasher.finish()
        });
        let compiler_version = env!("CARGO_PKG_VERSION").to_string();
        let global_flags =
            db.flags().iter().map(|(flag_id, flag)| (flag_id.long(db).0.clone(), *flag)).collect();
        Self { settings, compiler_version, global_flags }
    }
}

/// Validates that the metadata of the cached crate is valid.
fn validate_metadata(crate_id: CrateId<'_>, metadata: &CachedCrateMetadata, db: &dyn Database) {
    let current_metadata = CachedCrateMetadata::new(crate_id, db);

    if current_metadata.compiler_version != metadata.compiler_version {
        panic!("Cached crate was compiled with a different compiler version.");
    }
    if current_metadata.settings != metadata.settings {
        panic!("Cached crate was compiled with different settings.");
    }

    if !current_metadata.global_flags.eq_unordered(&metadata.global_flags) {
        panic!("Cached crate was compiled with different global flags.");
    }
}

type DefCache<'db> = (CachedCrateMetadata, CrateDefCache<'db>, DefCacheLookups);

/// Load the cached lowering of a crate if it has a cache file configuration.
pub fn load_cached_crate_modules<'db>(
    db: &'db dyn Database,
    crate_id: CrateId<'db>,
) -> Option<ModuleDataCacheAndLoadingData<'db>> {
    let blob_id = db.crate_config(crate_id)?.cache_file?;
    let Some(content) = db.blob_content(blob_id) else {
        return Default::default();
    };

    let size = usize::from_be_bytes(content[..8].try_into().unwrap());

    let content = &content[8..size + 8];

    let (metadata, module_data, defs_lookups): DefCache<'_> = postcard::from_bytes(content)
        .unwrap_or_else(|e| {
            panic!(
                "failed to deserialize modules cache for crate `{}`: {e}",
                crate_id.long(db).name().long(db),
            )
        });

    validate_metadata(crate_id, &metadata, db);

    let mut ctx = DefCacheLoadingContext::new(db, defs_lookups, crate_id);
    Some((
        module_data
            .0
            .into_iter()
            .map(|(module_id, module_data)| {
                let module_id = module_id.embed(&mut ctx);

                let module_data = module_data.embed(&mut ctx);
                (module_id, module_data)
            })
            .collect::<OrderedHashMap<_, _>>()
            .into(),
        ctx.data.into(),
    ))
}

#[derive(Serialize, Deserialize)]
pub struct CrateDefCache<'db>(Vec<(ModuleIdCached, ModuleDataCached<'db>)>);

impl<'db> CrateDefCache<'db> {
    pub fn new(modules: Vec<(ModuleIdCached, ModuleDataCached<'db>)>) -> Self {
        Self(modules)
    }
}

/// Context for loading cache into the database.
struct DefCacheLoadingContext<'db> {
    /// The variable ids of the flat lowered that is currently being loaded.
    db: &'db dyn Database,

    /// data for loading the entire cache into the database.
    data: DefCacheLoadingData<'db>,
}

impl<'db> DefCacheLoadingContext<'db> {
    pub fn new(
        db: &'db dyn Database,
        lookups: DefCacheLookups,
        self_crate_id: CrateId<'db>,
    ) -> Self {
        let mut res = Self { db, data: DefCacheLoadingData::new(lookups, self_crate_id) };
        res.embed_lookups();
        res
    }
    fn embed_lookups(&mut self) {
        for id in 0..self.lookups.green_ids_lookup.len() {
            let id = GreenIdCached(id);
            let embed = id.embed(self);
            self.reverse_green_ids.insert(embed, id);
        }
        for id in 0..self.lookups.crate_ids_lookup.len() {
            CrateIdCached::Other(id).embed(self);
        }
        for id in 0..self.lookups.file_ids_lookup.len() {
            FileIdCached(id).embed(self);
        }

        for id in 0..self.lookups.syntax_node_lookup.len() {
            SyntaxNodeCached(id).embed(self);
        }

        for id in 0..self.lookups.submodule_ids_lookup.len() {
            SubmoduleIdCached(id).embed(self);
        }
        for id in 0..self.lookups.constant_ids_lookup.len() {
            ConstantIdCached(id).embed(self);
        }
        for id in 0..self.lookups.use_ids_lookup.len() {
            UseIdCached(id).embed(self);
        }
        for id in 0..self.lookups.free_function_ids_lookup.len() {
            FreeFunctionIdCached(id).embed(self);
        }
        for id in 0..self.lookups.struct_ids_lookup.len() {
            StructIdCached(id).embed(self);
        }
        for id in 0..self.lookups.enum_ids_lookup.len() {
            EnumIdCached(id).embed(self);
        }
        for id in 0..self.lookups.type_alias_ids_lookup.len() {
            ModuleTypeAliasIdCached(id).embed(self);
        }
        for id in 0..self.lookups.impl_alias_ids_lookup.len() {
            ImplAliasIdCached(id).embed(self);
        }
        for id in 0..self.lookups.trait_ids_lookup.len() {
            TraitIdCached(id).embed(self);
        }
        for id in 0..self.lookups.impl_def_ids_lookup.len() {
            ImplDefIdCached(id).embed(self);
        }
        for id in 0..self.lookups.extern_type_ids_lookup.len() {
            ExternTypeIdCached(id).embed(self);
        }
        for id in 0..self.lookups.extern_function_ids_lookup.len() {
            ExternFunctionIdCached(id).embed(self);
        }
        for id in 0..self.lookups.global_use_ids_lookup.len() {
            GlobalUseIdCached(id).embed(self);
        }
    }
}

impl<'db> Deref for DefCacheLoadingContext<'db> {
    type Target = DefCacheLoadingData<'db>;

    fn deref(&self) -> &Self::Target {
        &self.data
    }
}
impl DerefMut for DefCacheLoadingContext<'_> {
    fn deref_mut(&mut self) -> &mut Self::Target {
        &mut self.data
    }
}

/// Data for loading cache into the database.
#[derive(PartialEq, Eq, salsa::Update)]
pub struct DefCacheLoadingData<'db> {
    green_ids: OrderedHashMap<GreenIdCached, GreenId<'db>>,
    reverse_green_ids: OrderedHashMap<GreenId<'db>, GreenIdCached>,
    crate_ids: OrderedHashMap<CrateIdCached, CrateId<'db>>,
    reverse_syntax_node_lookup: OrderedHashMap<SyntaxNodeIdCached, SyntaxNodeCached>,
    syntax_nodes: OrderedHashMap<SyntaxNodeCached, SyntaxNode<'db>>,
    submodule_ids: OrderedHashMap<SubmoduleIdCached, SubmoduleId<'db>>,
    constant_ids: OrderedHashMap<ConstantIdCached, ConstantId<'db>>,
    use_ids: OrderedHashMap<UseIdCached, UseId<'db>>,
    free_function_ids: OrderedHashMap<FreeFunctionIdCached, FreeFunctionId<'db>>,
    struct_ids: OrderedHashMap<StructIdCached, StructId<'db>>,
    enum_ids: OrderedHashMap<EnumIdCached, EnumId<'db>>,
    type_alias_ids: OrderedHashMap<ModuleTypeAliasIdCached, ModuleTypeAliasId<'db>>,
    impl_alias_ids: OrderedHashMap<ImplAliasIdCached, ImplAliasId<'db>>,
    trait_ids: OrderedHashMap<TraitIdCached, TraitId<'db>>,
    impl_def_ids: OrderedHashMap<ImplDefIdCached, ImplDefId<'db>>,
    extern_type_ids: OrderedHashMap<ExternTypeIdCached, ExternTypeId<'db>>,
    extern_function_ids: OrderedHashMap<ExternFunctionIdCached, ExternFunctionId<'db>>,
    macro_declaration_ids: OrderedHashMap<MacroDeclarationIdCached, MacroDeclarationId<'db>>,
    macro_call_ids: OrderedHashMap<MacroCallIdCached, MacroCallId<'db>>,
    global_use_ids: OrderedHashMap<GlobalUseIdCached, GlobalUseId<'db>>,

    file_ids: OrderedHashMap<FileIdCached, FileId<'db>>,
    self_crate_id: CrateId<'db>,
    lookups: DefCacheLookups,
}

impl<'db> DefCacheLoadingData<'db> {
    fn new(lookups: DefCacheLookups, self_crate_id: CrateId<'db>) -> Self {
        Self {
            green_ids: OrderedHashMap::default(),
            reverse_green_ids: OrderedHashMap::default(),
            reverse_syntax_node_lookup: lookups
                .syntax_node_lookup
                .iter()
                .enumerate()
                .map(|(index, syntax_node)| (syntax_node.clone(), SyntaxNodeCached(index)))
                .collect(),
            syntax_nodes: OrderedHashMap::default(),
            crate_ids: OrderedHashMap::default(),
            submodule_ids: OrderedHashMap::default(),
            constant_ids: OrderedHashMap::default(),
            use_ids: OrderedHashMap::default(),
            free_function_ids: OrderedHashMap::default(),
            struct_ids: OrderedHashMap::default(),
            enum_ids: OrderedHashMap::default(),
            type_alias_ids: OrderedHashMap::default(),
            impl_alias_ids: OrderedHashMap::default(),
            trait_ids: OrderedHashMap::default(),
            impl_def_ids: OrderedHashMap::default(),
            extern_type_ids: OrderedHashMap::default(),
            extern_function_ids: OrderedHashMap::default(),
            macro_declaration_ids: OrderedHashMap::default(),
            macro_call_ids: OrderedHashMap::default(),
            global_use_ids: OrderedHashMap::default(),

            file_ids: OrderedHashMap::default(),
            self_crate_id,
            lookups,
        }
    }
}

impl<'db> std::fmt::Debug for DefCacheLoadingData<'db> {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("DefCacheLoadingData").finish()
    }
}

impl<'db> Deref for DefCacheLoadingData<'db> {
    type Target = DefCacheLookups;

    fn deref(&self) -> &Self::Target {
        &self.lookups
    }
}
impl<'db> DerefMut for DefCacheLoadingData<'db> {
    fn deref_mut(&mut self) -> &mut Self::Target {
        &mut self.lookups
    }
}

/// Context for saving cache from the database.
pub struct DefCacheSavingContext<'db> {
    db: &'db dyn Database,
    data: DefCacheSavingData<'db>,
    self_crate_id: CrateId<'db>,
}
impl<'db> Deref for DefCacheSavingContext<'db> {
    type Target = DefCacheSavingData<'db>;

    fn deref(&self) -> &Self::Target {
        &self.data
    }
}
impl DerefMut for DefCacheSavingContext<'_> {
    fn deref_mut(&mut self) -> &mut Self::Target {
        &mut self.data
    }
}
impl<'db> DefCacheSavingContext<'db> {
    pub fn new(db: &'db dyn Database, self_crate_id: CrateId<'db>) -> Self {
        Self { db, data: DefCacheSavingData::default(), self_crate_id }
    }
}

/// Data for saving cache from the database.
#[derive(Default)]
pub struct DefCacheSavingData<'db> {
    green_ids: OrderedHashMap<GreenId<'db>, GreenIdCached>,
    crate_ids: OrderedHashMap<CrateId<'db>, CrateIdCached>,
    submodule_ids: OrderedHashMap<SubmoduleId<'db>, SubmoduleIdCached>,
    constant_ids: OrderedHashMap<ConstantId<'db>, ConstantIdCached>,
    use_ids: OrderedHashMap<UseId<'db>, UseIdCached>,
    free_function_ids: OrderedHashMap<FreeFunctionId<'db>, FreeFunctionIdCached>,
    struct_ids: OrderedHashMap<StructId<'db>, StructIdCached>,
    enum_ids: OrderedHashMap<EnumId<'db>, EnumIdCached>,
    type_alias_ids: OrderedHashMap<ModuleTypeAliasId<'db>, ModuleTypeAliasIdCached>,
    impl_alias_ids: OrderedHashMap<ImplAliasId<'db>, ImplAliasIdCached>,
    trait_ids: OrderedHashMap<TraitId<'db>, TraitIdCached>,
    impl_def_ids: OrderedHashMap<ImplDefId<'db>, ImplDefIdCached>,
    extern_type_ids: OrderedHashMap<ExternTypeId<'db>, ExternTypeIdCached>,
    extern_function_ids: OrderedHashMap<ExternFunctionId<'db>, ExternFunctionIdCached>,
    global_use_ids: OrderedHashMap<GlobalUseId<'db>, GlobalUseIdCached>,
    macro_declaration_ids: OrderedHashMap<MacroDeclarationId<'db>, MacroDeclarationIdCached>,
    macro_call_ids: OrderedHashMap<MacroCallId<'db>, MacroCallIdCached>,

    syntax_nodes: OrderedHashMap<SyntaxNode<'db>, SyntaxNodeCached>,
    file_ids: OrderedHashMap<FileId<'db>, FileIdCached>,

    pub lookups: DefCacheLookups,
}

impl<'db> Deref for DefCacheSavingData<'db> {
    type Target = DefCacheLookups;

    fn deref(&self) -> &Self::Target {
        &self.lookups
    }
}
impl<'db> DerefMut for DefCacheSavingData<'db> {
    fn deref_mut(&mut self) -> &mut Self::Target {
        &mut self.lookups
    }
}

#[derive(Serialize, Deserialize)]
pub struct ModuleDataCached<'db> {
    items: Vec<ModuleItemIdCached>,

    constants: OrderedHashMap<ConstantIdCached, TypeSyntaxNodeCached<'db, ast::ItemConstant<'db>>>,
    submodules: OrderedHashMap<SubmoduleIdCached, TypeSyntaxNodeCached<'db, ast::ItemModule<'db>>>,
    uses: OrderedHashMap<UseIdCached, TypeSyntaxNodeCached<'db, ast::UsePathLeaf<'db>>>,
    free_functions:
        OrderedHashMap<FreeFunctionIdCached, TypeSyntaxNodeCached<'db, ast::FunctionWithBody<'db>>>,
    structs: OrderedHashMap<StructIdCached, TypeSyntaxNodeCached<'db, ast::ItemStruct<'db>>>,
    enums: OrderedHashMap<EnumIdCached, TypeSyntaxNodeCached<'db, ast::ItemEnum<'db>>>,
    type_aliases:
        OrderedHashMap<ModuleTypeAliasIdCached, TypeSyntaxNodeCached<'db, ast::ItemTypeAlias<'db>>>,
    impl_aliases:
        OrderedHashMap<ImplAliasIdCached, TypeSyntaxNodeCached<'db, ast::ItemImplAlias<'db>>>,
    traits: OrderedHashMap<TraitIdCached, TypeSyntaxNodeCached<'db, ast::ItemTrait<'db>>>,
    impls: OrderedHashMap<ImplDefIdCached, TypeSyntaxNodeCached<'db, ast::ItemImpl<'db>>>,
    extern_types:
        OrderedHashMap<ExternTypeIdCached, TypeSyntaxNodeCached<'db, ast::ItemExternType<'db>>>,
    extern_functions: OrderedHashMap<
        ExternFunctionIdCached,
        TypeSyntaxNodeCached<'db, ast::ItemExternFunction<'db>>,
    >,
    macro_declarations: OrderedHashMap<
        MacroDeclarationIdCached,
        TypeSyntaxNodeCached<'db, ast::ItemMacroDeclaration<'db>>,
    >,
    global_uses:
        OrderedHashMap<GlobalUseIdCached, TypeSyntaxNodeCached<'db, ast::UsePathStar<'db>>>,
    macro_calls:
        OrderedHashMap<MacroCallIdCached, TypeSyntaxNodeCached<'db, ast::ItemInlineMacro<'db>>>,

    files: Vec<FileIdCached>,

    generated_file_aux_data: OrderedHashMap<FileIdCached, Option<DynGeneratedFileAuxData>>,
    plugin_diagnostics: Vec<(ModuleIdCached, PluginDiagnosticCached)>,
    diagnostics_notes: PluginFileDiagnosticNotesCached,
}
impl<'db> ModuleDataCached<'db> {
    pub fn new(
        db: &'db dyn Database,
        module_data: ModuleData<'db>,
        ctx: &mut DefCacheSavingContext<'db>,
    ) -> Self {
        Self {
            items: module_data
                .items(db)
                .iter()
                .map(|id| ModuleItemIdCached::new(*id, ctx))
                .collect(),
            constants: module_data
                .constants(db)
                .iter()
                .map(|(id, node)| {
                    (ConstantIdCached::new(*id, ctx), TypeSyntaxNodeCached::new(node.clone(), ctx))
                })
                .collect(),

            submodules: module_data
                .submodules(db)
                .iter()
                .map(|(id, node)| {
                    (SubmoduleIdCached::new(*id, ctx), TypeSyntaxNodeCached::new(node.clone(), ctx))
                })
                .collect(),

            uses: module_data
                .uses(db)
                .iter()
                .map(|(id, node)| {
                    (UseIdCached::new(*id, ctx), TypeSyntaxNodeCached::new(node.clone(), ctx))
                })
                .collect(),
            free_functions: module_data
                .free_functions(db)
                .iter()
                .map(|(id, node)| {
                    (
                        FreeFunctionIdCached::new(*id, ctx),
                        TypeSyntaxNodeCached::new(node.clone(), ctx),
                    )
                })
                .collect(),
            structs: module_data
                .structs(db)
                .iter()
                .map(|(id, node)| {
                    (StructIdCached::new(*id, ctx), TypeSyntaxNodeCached::new(node.clone(), ctx))
                })
                .collect(),
            enums: module_data
                .enums(db)
                .iter()
                .map(|(id, node)| {
                    (EnumIdCached::new(*id, ctx), TypeSyntaxNodeCached::new(node.clone(), ctx))
                })
                .collect(),
            type_aliases: module_data
                .type_aliases(db)
                .iter()
                .map(|(id, node)| {
                    (
                        ModuleTypeAliasIdCached::new(*id, ctx),
                        TypeSyntaxNodeCached::new(node.clone(), ctx),
                    )
                })
                .collect(),
            impl_aliases: module_data
                .impl_aliases(db)
                .iter()
                .map(|(id, node)| {
                    (ImplAliasIdCached::new(*id, ctx), TypeSyntaxNodeCached::new(node.clone(), ctx))
                })
                .collect(),
            traits: module_data
                .traits(db)
                .iter()
                .map(|(id, node)| {
                    (TraitIdCached::new(*id, ctx), TypeSyntaxNodeCached::new(node.clone(), ctx))
                })
                .collect(),
            impls: module_data
                .impls(db)
                .iter()
                .map(|(id, node)| {
                    (ImplDefIdCached::new(*id, ctx), TypeSyntaxNodeCached::new(node.clone(), ctx))
                })
                .collect(),
            extern_types: module_data
                .extern_types(db)
                .iter()
                .map(|(id, node)| {
                    (
                        ExternTypeIdCached::new(*id, ctx),
                        TypeSyntaxNodeCached::new(node.clone(), ctx),
                    )
                })
                .collect(),
            extern_functions: module_data
                .extern_functions(db)
                .iter()
                .map(|(id, node)| {
                    (
                        ExternFunctionIdCached::new(*id, ctx),
                        TypeSyntaxNodeCached::new(node.clone(), ctx),
                    )
                })
                .collect(),

            macro_declarations: module_data
                .macro_declarations(db)
                .iter()
                .map(|(id, node)| {
                    (
                        MacroDeclarationIdCached::new(*id, ctx),
                        TypeSyntaxNodeCached::new(node.clone(), ctx),
                    )
                })
                .collect(),
            global_uses: module_data
                .global_uses(db)
                .iter()
                .map(|(id, node)| {
                    (GlobalUseIdCached::new(*id, ctx), TypeSyntaxNodeCached::new(node.clone(), ctx))
                })
                .collect(),
            macro_calls: module_data
                .macro_calls(db)
                .iter()
                .map(|(id, node)| {
                    (MacroCallIdCached::new(*id, ctx), TypeSyntaxNodeCached::new(node.clone(), ctx))
                })
                .collect(),
            files: module_data.files(db).iter().map(|id| FileIdCached::new(*id, ctx)).collect(),
            generated_file_aux_data: module_data
                .generated_file_aux_data(db)
                .iter()
                .map(|(file, aux_data)| (FileIdCached::new(*file, ctx), aux_data.clone()))
                .collect(),

            plugin_diagnostics: module_data
                .plugin_diagnostics(db)
                .iter()
                .map(|(file_id, diagnostic)| {
                    (
                        ModuleIdCached::new(*file_id, ctx),
                        PluginDiagnosticCached::new(diagnostic, ctx),
                    )
                })
                .collect(),
            diagnostics_notes: module_data
                .diagnostics_notes(db)
                .iter()
                .map(|(file_id, note)| {
                    (FileIdCached::new(*file_id, ctx), DiagnosticNoteCached::new(note.clone(), ctx))
                })
                .collect(),
        }
    }
    fn embed(self, ctx: &mut DefCacheLoadingContext<'db>) -> ModuleData<'db> {
        ModuleData::new(
            ctx.db,
            self.items.iter().map(|id| id.embed(ctx)).collect(),
            ModuleTypesData::new(
                ctx.db,
                self.structs.iter().map(|(id, node)| (id.embed(ctx), node.embed(ctx))).collect(),
                self.enums.iter().map(|(id, node)| (id.embed(ctx), node.embed(ctx))).collect(),
                self.type_aliases
                    .iter()
                    .map(|(id, node)| (id.embed(ctx), node.embed(ctx)))
                    .collect(),
                self.extern_types
                    .iter()
                    .map(|(id, node)| (id.embed(ctx), node.embed(ctx)))
                    .collect(),
            ),
            ModuleNamedItemsData::new(
                ctx.db,
                self.constants.iter().map(|(id, node)| (id.embed(ctx), node.embed(ctx))).collect(),
                self.submodules.iter().map(|(id, node)| (id.embed(ctx), node.embed(ctx))).collect(),
                self.uses.iter().map(|(id, node)| (id.embed(ctx), node.embed(ctx))).collect(),
                self.free_functions
                    .iter()
                    .map(|(id, node)| (id.embed(ctx), node.embed(ctx)))
                    .collect(),
                self.impl_aliases
                    .iter()
                    .map(|(id, node)| (id.embed(ctx), node.embed(ctx)))
                    .collect(),
                self.traits.iter().map(|(id, node)| (id.embed(ctx), node.embed(ctx))).collect(),
                self.impls.iter().map(|(id, node)| (id.embed(ctx), node.embed(ctx))).collect(),
                self.extern_functions
                    .iter()
                    .map(|(id, node)| (id.embed(ctx), node.embed(ctx)))
                    .collect(),
                self.macro_declarations
                    .iter()
                    .map(|(id, node)| (id.embed(ctx), node.embed(ctx)))
                    .collect(),
            ),
            ModuleUnnamedItemsData::new(
                ctx.db,
                self.global_uses
                    .iter()
                    .map(|(id, node)| (id.embed(ctx), node.embed(ctx)))
                    .collect(),
                self.macro_calls
                    .iter()
                    .map(|(id, node)| (id.embed(ctx), node.embed(ctx)))
                    .collect(),
            ),
            ModuleFilesData::new(
                ctx.db,
                self.files.iter().map(|id| id.embed(ctx)).collect(),
                self.generated_file_aux_data
                    .into_iter()
                    .map(|(file, aux_data)| (file.embed(ctx), aux_data))
                    .collect(),
                self.plugin_diagnostics
                    .into_iter()
                    .map(|(file_id, diagnostic)| (file_id.embed(ctx), diagnostic.embed(ctx)))
                    .collect(),
                self.diagnostics_notes
                    .into_iter()
                    .map(|(file_id, note)| (file_id.embed(ctx), note.embed(ctx)))
                    .collect(),
            ),
        )
    }
}

/// Saved interned items for the cache.
#[derive(Serialize, Deserialize, Default, PartialEq, Eq, salsa::Update)]
pub struct DefCacheLookups {
    green_ids_lookup: Vec<GreenNodeCached>,
    crate_ids_lookup: Vec<CrateCached>,
    submodule_ids_lookup: Vec<SubmoduleCached>,
    constant_ids_lookup: Vec<ConstantCached>,
    use_ids_lookup: Vec<UseCached>,
    free_function_ids_lookup: Vec<FreeFunctionCached>,
    struct_ids_lookup: Vec<StructCached>,
    enum_ids_lookup: Vec<EnumCached>,
    type_alias_ids_lookup: Vec<ModuleTypeAliasCached>,
    impl_alias_ids_lookup: Vec<ImplAliasCached>,
    trait_ids_lookup: Vec<TraitCached>,
    impl_def_ids_lookup: Vec<ImplDefCached>,
    extern_type_ids_lookup: Vec<ExternTypeCached>,
    extern_function_ids_lookup: Vec<ExternFunctionCached>,
    global_use_ids_lookup: Vec<GlobalUseCached>,
    macro_declaration_ids_lookup: Vec<MacroDeclarationCached>,
    macro_call_ids_lookup: Vec<MacroCallCached>,

    syntax_node_lookup: Vec<SyntaxNodeIdCached>,
    file_ids_lookup: Vec<FileCached>,
}

#[derive(Serialize, Deserialize, Clone)]
struct TypeSyntaxNodeCached<'db, T: TypedSyntaxNode<'db>> {
    syntax_node: SyntaxNodeCached,
    _marker: std::marker::PhantomData<&'db T>,
}
impl<'db, T: TypedSyntaxNode<'db>> TypeSyntaxNodeCached<'db, T> {
    fn new(syntax_node: T, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        Self {
            syntax_node: SyntaxNodeCached::new(syntax_node.as_syntax_node(), ctx),
            _marker: std::marker::PhantomData,
        }
    }
    fn embed(&self, ctx: &mut DefCacheLoadingContext<'db>) -> T {
        T::from_syntax_node(ctx.db, self.syntax_node.embed(ctx))
    }
}

#[derive(Serialize, Deserialize, Clone, PartialEq, Eq)]
pub struct GenericParamCached {
    language_element: LanguageElementCached,
}
impl GenericParamCached {
    pub fn new<'db>(
        generic_param_id: GenericParamId<'db>,
        ctx: &mut DefCacheSavingContext<'db>,
    ) -> Self {
        Self { language_element: LanguageElementCached::new(generic_param_id, ctx) }
    }

    pub fn get_embedded<'db>(
        self,
        data: &Arc<DefCacheLoadingData<'db>>,
        db: &'db dyn Database,
    ) -> GenericParamId<'db> {
        let (module_id, stable_ptr) = self.language_element.get_embedded(data);
        GenericParamLongId(module_id, GenericParamPtr(stable_ptr)).intern(db)
    }
}

#[derive(Serialize, Deserialize, Clone, Hash, Eq, PartialEq)]
pub enum ModuleIdCached {
    CrateRoot(CrateIdCached),
    Submodule(SubmoduleIdCached),
    MacroCall { id: MacroCallIdCached, generated_file_id: FileIdCached, is_expose: bool },
}
impl ModuleIdCached {
    pub fn new<'db>(module_id: ModuleId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        match module_id {
            ModuleId::CrateRoot(crate_id) => {
                ModuleIdCached::CrateRoot(CrateIdCached::new(crate_id, ctx))
            }
            ModuleId::Submodule(submodule_id) => {
                ModuleIdCached::Submodule(SubmoduleIdCached::new(submodule_id, ctx))
            }
            ModuleId::MacroCall { id, generated_file_id, is_expose } => ModuleIdCached::MacroCall {
                id: MacroCallIdCached::new(id, ctx),
                generated_file_id: FileIdCached::new(generated_file_id, ctx),
                is_expose,
            },
        }
    }
    fn embed<'db>(&self, ctx: &mut DefCacheLoadingContext<'db>) -> ModuleId<'db> {
        match self {
            ModuleIdCached::CrateRoot(crate_id) => ModuleId::CrateRoot(crate_id.embed(ctx)),
            ModuleIdCached::Submodule(submodule_id) => ModuleId::Submodule(submodule_id.embed(ctx)),
            ModuleIdCached::MacroCall { id, generated_file_id, is_expose } => ModuleId::MacroCall {
                id: id.embed(ctx),
                generated_file_id: generated_file_id.embed(ctx),
                is_expose: *is_expose,
            },
        }
    }
    pub fn get_embedded<'db>(self, data: &Arc<DefCacheLoadingData<'db>>) -> ModuleId<'db> {
        match self {
            ModuleIdCached::CrateRoot(crate_id) => ModuleId::CrateRoot(crate_id.get_embedded(data)),
            ModuleIdCached::Submodule(submodule_id) => {
                ModuleId::Submodule(submodule_id.get_embedded(data))
            }
            ModuleIdCached::MacroCall { id, generated_file_id, is_expose } => ModuleId::MacroCall {
                id: id.get_embedded(data),
                generated_file_id: generated_file_id.get_embedded(data),
                is_expose,
            },
        }
    }
}

#[derive(Serialize, Deserialize, Clone, PartialEq, Eq)]
enum CrateCached {
    Real { name: String, discriminator: Option<String> },
    Virtual { name: String, file_id: FileIdCached, settings: String },
}
impl CrateCached {
    fn new<'db>(crate_id: CrateLongId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        match crate_id {
            CrateLongId::Real { name, discriminator } => {
                CrateCached::Real { name: name.to_string(ctx.db), discriminator }
            }
            CrateLongId::Virtual { name, file_id, settings, cache_file: _ } => {
                CrateCached::Virtual {
                    name: name.to_string(ctx.db),
                    file_id: FileIdCached::new(file_id, ctx),
                    settings,
                }
            }
        }
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> CrateLongId<'db> {
        match self {
            CrateCached::Real { name, discriminator } => {
                CrateLongId::Real { name: SmolStrId::from(ctx.db, name), discriminator }
            }
            CrateCached::Virtual { name, file_id, settings } => {
                CrateLongId::Virtual {
                    name: SmolStrId::from(ctx.db, name),
                    file_id: file_id.embed(ctx),
                    settings,
                    cache_file: None, // todo: if two virtual crates are supported
                }
            }
        }
    }
}
#[derive(Serialize, Deserialize, Copy, Clone, PartialEq, Eq, Hash, salsa::Update)]
pub enum CrateIdCached {
    SelfCrate,
    Other(usize),
}
impl CrateIdCached {
    fn new<'db>(crate_id: CrateId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        if crate_id == ctx.self_crate_id {
            return CrateIdCached::SelfCrate;
        }
        if let Some(id) = ctx.crate_ids.get(&crate_id) {
            return *id;
        }
        let crate_long_id = CrateCached::new(crate_id.long(ctx.db).clone(), ctx);
        let id = CrateIdCached::Other(ctx.crate_ids_lookup.len());
        ctx.crate_ids_lookup.push(crate_long_id);
        ctx.crate_ids.insert(crate_id, id);
        id
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> CrateId<'db> {
        let CrateIdCached::Other(id) = self else {
            return ctx.self_crate_id;
        };

        if let Some(crate_id) = ctx.crate_ids.get(&self) {
            return *crate_id;
        }
        let crate_long_id = ctx.crate_ids_lookup[id].clone();
        let crate_id = crate_long_id.embed(ctx).intern(ctx.db);
        ctx.data.crate_ids.insert(self, crate_id);
        crate_id
    }
    fn get_embedded<'db>(self, data: &Arc<DefCacheLoadingData<'db>>) -> CrateId<'db> {
        let CrateIdCached::Other(_) = self else {
            return data.self_crate_id;
        };
        data.crate_ids[&self]
    }
}

#[derive(Serialize, Deserialize, Copy, Clone)]
pub enum ModuleItemIdCached {
    Constant(ConstantIdCached),
    Submodule(SubmoduleIdCached),
    Use(UseIdCached),
    FreeFunction(FreeFunctionIdCached),
    Struct(StructIdCached),
    Enum(EnumIdCached),
    TypeAlias(ModuleTypeAliasIdCached),
    ImplAlias(ImplAliasIdCached),
    Trait(TraitIdCached),
    Impl(ImplDefIdCached),
    ExternType(ExternTypeIdCached),
    ExternFunction(ExternFunctionIdCached),
    MacroDeclaration(MacroDeclarationIdCached),
}

impl ModuleItemIdCached {
    pub fn new<'db>(
        module_item_id: ModuleItemId<'db>,
        ctx: &mut DefCacheSavingContext<'db>,
    ) -> Self {
        match module_item_id {
            ModuleItemId::Constant(constant_id) => {
                ModuleItemIdCached::Constant(ConstantIdCached::new(constant_id, ctx))
            }
            ModuleItemId::Submodule(submodule_id) => {
                ModuleItemIdCached::Submodule(SubmoduleIdCached::new(submodule_id, ctx))
            }
            ModuleItemId::Use(use_id) => ModuleItemIdCached::Use(UseIdCached::new(use_id, ctx)),
            ModuleItemId::FreeFunction(free_function_id) => {
                ModuleItemIdCached::FreeFunction(FreeFunctionIdCached::new(free_function_id, ctx))
            }
            ModuleItemId::Struct(struct_id) => {
                ModuleItemIdCached::Struct(StructIdCached::new(struct_id, ctx))
            }
            ModuleItemId::Enum(enum_id) => {
                ModuleItemIdCached::Enum(EnumIdCached::new(enum_id, ctx))
            }
            ModuleItemId::TypeAlias(type_alias_id) => {
                ModuleItemIdCached::TypeAlias(ModuleTypeAliasIdCached::new(type_alias_id, ctx))
            }
            ModuleItemId::ImplAlias(impl_alias_id) => {
                ModuleItemIdCached::ImplAlias(ImplAliasIdCached::new(impl_alias_id, ctx))
            }
            ModuleItemId::Trait(trait_id) => {
                ModuleItemIdCached::Trait(TraitIdCached::new(trait_id, ctx))
            }
            ModuleItemId::Impl(impl_def_id) => {
                ModuleItemIdCached::Impl(ImplDefIdCached::new(impl_def_id, ctx))
            }
            ModuleItemId::ExternType(extern_type_id) => {
                ModuleItemIdCached::ExternType(ExternTypeIdCached::new(extern_type_id, ctx))
            }
            ModuleItemId::ExternFunction(extern_function_id) => ModuleItemIdCached::ExternFunction(
                ExternFunctionIdCached::new(extern_function_id, ctx),
            ),
            ModuleItemId::MacroDeclaration(macro_declaration_id) => {
                ModuleItemIdCached::MacroDeclaration(MacroDeclarationIdCached::new(
                    macro_declaration_id,
                    ctx,
                ))
            }
        }
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> ModuleItemId<'db> {
        match self {
            ModuleItemIdCached::Constant(constant_id) => {
                ModuleItemId::Constant(constant_id.embed(ctx))
            }
            ModuleItemIdCached::Submodule(submodule_id) => {
                ModuleItemId::Submodule(submodule_id.embed(ctx))
            }
            ModuleItemIdCached::Use(use_id) => ModuleItemId::Use(use_id.embed(ctx)),
            ModuleItemIdCached::FreeFunction(free_function_id) => {
                ModuleItemId::FreeFunction(free_function_id.embed(ctx))
            }
            ModuleItemIdCached::Struct(struct_id) => ModuleItemId::Struct(struct_id.embed(ctx)),
            ModuleItemIdCached::Enum(enum_id) => ModuleItemId::Enum(enum_id.embed(ctx)),
            ModuleItemIdCached::TypeAlias(type_alias_id) => {
                ModuleItemId::TypeAlias(type_alias_id.embed(ctx))
            }
            ModuleItemIdCached::ImplAlias(impl_alias_id) => {
                ModuleItemId::ImplAlias(impl_alias_id.embed(ctx))
            }
            ModuleItemIdCached::Trait(trait_id) => ModuleItemId::Trait(trait_id.embed(ctx)),
            ModuleItemIdCached::Impl(impl_def_id) => ModuleItemId::Impl(impl_def_id.embed(ctx)),
            ModuleItemIdCached::ExternType(extern_type_id) => {
                ModuleItemId::ExternType(extern_type_id.embed(ctx))
            }
            ModuleItemIdCached::ExternFunction(extern_function_id) => {
                ModuleItemId::ExternFunction(extern_function_id.embed(ctx))
            }
            ModuleItemIdCached::MacroDeclaration(macro_declaration_id) => {
                ModuleItemId::MacroDeclaration(macro_declaration_id.embed(ctx))
            }
        }
    }
    pub fn get_embedded<'db>(&self, data: &Arc<DefCacheLoadingData<'db>>) -> ModuleItemId<'db> {
        match self {
            ModuleItemIdCached::Constant(id) => ModuleItemId::Constant(id.get_embedded(data)),
            ModuleItemIdCached::Submodule(id) => ModuleItemId::Submodule(id.get_embedded(data)),
            ModuleItemIdCached::Use(id) => ModuleItemId::Use(id.get_embedded(data)),
            ModuleItemIdCached::FreeFunction(id) => {
                ModuleItemId::FreeFunction(id.get_embedded(data))
            }
            ModuleItemIdCached::Struct(id) => ModuleItemId::Struct(id.get_embedded(data)),
            ModuleItemIdCached::Enum(id) => ModuleItemId::Enum(id.get_embedded(data)),
            ModuleItemIdCached::TypeAlias(id) => ModuleItemId::TypeAlias(id.get_embedded(data)),
            ModuleItemIdCached::ImplAlias(id) => ModuleItemId::ImplAlias(id.get_embedded(data)),
            ModuleItemIdCached::Trait(id) => ModuleItemId::Trait(id.get_embedded(data)),
            ModuleItemIdCached::Impl(id) => ModuleItemId::Impl(id.get_embedded(data)),
            ModuleItemIdCached::ExternType(id) => ModuleItemId::ExternType(id.get_embedded(data)),
            ModuleItemIdCached::ExternFunction(id) => {
                ModuleItemId::ExternFunction(id.get_embedded(data))
            }
            ModuleItemIdCached::MacroDeclaration(id) => {
                ModuleItemId::MacroDeclaration(id.get_embedded(data))
            }
        }
    }
}

#[derive(Serialize, Deserialize, Clone, PartialEq, Eq)]
struct ConstantCached {
    language_element: LanguageElementCached,
}
impl ConstantCached {
    fn new<'db>(constant_id: ConstantId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        Self { language_element: LanguageElementCached::new(constant_id, ctx) }
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> ConstantLongId<'db> {
        let (module_id, stable_ptr) = self.language_element.embed(ctx);

        ConstantLongId(module_id, ItemConstantPtr(stable_ptr))
    }
}

#[derive(Serialize, Deserialize, Copy, Clone, PartialEq, Eq, Hash, salsa::Update)]
pub struct ConstantIdCached(usize);

impl ConstantIdCached {
    fn new<'db>(id: ConstantId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        if let Some(cached_id) = ctx.constant_ids.get(&id) {
            return *cached_id;
        }
        let cached = ConstantCached::new(id, ctx);
        let cached_id = ConstantIdCached(ctx.constant_ids_lookup.len());
        ctx.constant_ids_lookup.push(cached);
        ctx.constant_ids.insert(id, cached_id);
        cached_id
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> ConstantId<'db> {
        if let Some(id) = ctx.constant_ids.get(&self) {
            return *id;
        }
        let cached = ctx.constant_ids_lookup[self.0].clone();
        let id = cached.embed(ctx).intern(ctx.db);
        ctx.constant_ids.insert(self, id);
        id
    }
    pub fn get_embedded<'db>(self, data: &Arc<DefCacheLoadingData<'db>>) -> ConstantId<'db> {
        data.constant_ids[&self]
    }
}

#[derive(Serialize, Deserialize, Clone, PartialEq, Eq)]
struct SubmoduleCached {
    language_element: LanguageElementCached,
}
impl SubmoduleCached {
    fn new<'db>(submodule_id: SubmoduleId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        Self { language_element: LanguageElementCached::new(submodule_id, ctx) }
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> SubmoduleLongId<'db> {
        let (module_id, stable_ptr) = self.language_element.embed(ctx);

        SubmoduleLongId(module_id, ItemModulePtr(stable_ptr))
    }
}

#[derive(Serialize, Deserialize, Copy, Clone, PartialEq, Eq, Hash, salsa::Update)]
pub struct SubmoduleIdCached(usize);

impl SubmoduleIdCached {
    fn new<'db>(id: SubmoduleId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        if let Some(cached_id) = ctx.submodule_ids.get(&id) {
            return *cached_id;
        }
        let cached = SubmoduleCached::new(id, ctx);
        let cached_id = SubmoduleIdCached(ctx.submodule_ids_lookup.len());
        ctx.submodule_ids_lookup.push(cached);
        ctx.submodule_ids.insert(id, cached_id);
        cached_id
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> SubmoduleId<'db> {
        if let Some(id) = ctx.submodule_ids.get(&self) {
            return *id;
        }
        let cached = ctx.submodule_ids_lookup[self.0].clone();
        let id = cached.embed(ctx).intern(ctx.db);
        ctx.submodule_ids.insert(self, id);
        id
    }
    fn get_embedded<'db>(self, data: &Arc<DefCacheLoadingData<'db>>) -> SubmoduleId<'db> {
        data.submodule_ids[&self]
    }
}

#[derive(Serialize, Deserialize, Clone, PartialEq, Eq)]
struct UseCached {
    language_element: LanguageElementCached,
}
impl UseCached {
    fn new<'db>(use_id: UseId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        Self { language_element: LanguageElementCached::new(use_id, ctx) }
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> UseLongId<'db> {
        let (module_id, stable_ptr) = self.language_element.embed(ctx);

        UseLongId(module_id, UsePathLeafPtr(stable_ptr))
    }
}

#[derive(Serialize, Deserialize, Copy, Clone, PartialEq, Eq, Hash, salsa::Update)]
pub struct UseIdCached(usize);

impl UseIdCached {
    fn new<'db>(id: UseId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        if let Some(cached_id) = ctx.use_ids.get(&id) {
            return *cached_id;
        }
        let cached = UseCached::new(id, ctx);
        let cached_id = UseIdCached(ctx.use_ids_lookup.len());
        ctx.use_ids_lookup.push(cached);
        ctx.use_ids.insert(id, cached_id);
        cached_id
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> UseId<'db> {
        if let Some(id) = ctx.use_ids.get(&self) {
            return *id;
        }
        let cached = ctx.use_ids_lookup[self.0].clone();
        let id = cached.embed(ctx).intern(ctx.db);
        ctx.use_ids.insert(self, id);
        id
    }
    pub fn get_embedded<'db>(self, data: &Arc<DefCacheLoadingData<'db>>) -> UseId<'db> {
        data.use_ids[&self]
    }
}

#[derive(Serialize, Deserialize, Clone, PartialEq, Eq)]
struct FreeFunctionCached {
    language_element: LanguageElementCached,
}
impl FreeFunctionCached {
    fn new<'db>(
        free_function_id: FreeFunctionId<'db>,
        ctx: &mut DefCacheSavingContext<'db>,
    ) -> Self {
        Self { language_element: LanguageElementCached::new(free_function_id, ctx) }
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> FreeFunctionLongId<'db> {
        let (module_id, stable_ptr) = self.language_element.embed(ctx);

        FreeFunctionLongId(module_id, FunctionWithBodyPtr(stable_ptr))
    }
}

#[derive(Serialize, Deserialize, Copy, Clone, PartialEq, Eq, Hash, salsa::Update)]
pub struct FreeFunctionIdCached(usize);

impl FreeFunctionIdCached {
    fn new<'db>(id: FreeFunctionId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        if let Some(cached_id) = ctx.free_function_ids.get(&id) {
            return *cached_id;
        }
        let cached = FreeFunctionCached::new(id, ctx);
        let cached_id = FreeFunctionIdCached(ctx.free_function_ids_lookup.len());
        ctx.free_function_ids_lookup.push(cached);
        ctx.free_function_ids.insert(id, cached_id);
        cached_id
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> FreeFunctionId<'db> {
        if let Some(id) = ctx.free_function_ids.get(&self) {
            return *id;
        }
        let cached = ctx.free_function_ids_lookup[self.0].clone();
        let id = cached.embed(ctx).intern(ctx.db);
        ctx.free_function_ids.insert(self, id);
        id
    }
    pub fn get_embedded<'db>(self, data: &Arc<DefCacheLoadingData<'db>>) -> FreeFunctionId<'db> {
        data.free_function_ids[&self]
    }
}

#[derive(Serialize, Deserialize, Clone, PartialEq, Eq)]
struct StructCached {
    language_element: LanguageElementCached,
}
impl StructCached {
    fn new<'db>(struct_id: StructId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        Self { language_element: LanguageElementCached::new(struct_id, ctx) }
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> StructLongId<'db> {
        let (module_id, stable_ptr) = self.language_element.embed(ctx);

        StructLongId(module_id, ItemStructPtr(stable_ptr))
    }
}

#[derive(Serialize, Deserialize, Copy, Clone, PartialEq, Eq, Hash, salsa::Update)]
pub struct StructIdCached(usize);

impl StructIdCached {
    fn new<'db>(id: StructId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        if let Some(cached_id) = ctx.struct_ids.get(&id) {
            return *cached_id;
        }
        let cached = StructCached::new(id, ctx);
        let cached_id = StructIdCached(ctx.struct_ids_lookup.len());
        ctx.struct_ids_lookup.push(cached);
        ctx.struct_ids.insert(id, cached_id);
        cached_id
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> StructId<'db> {
        if let Some(id) = ctx.struct_ids.get(&self) {
            return *id;
        }
        let cached = ctx.struct_ids_lookup[self.0].clone();
        let id = cached.embed(ctx).intern(ctx.db);
        ctx.struct_ids.insert(self, id);
        id
    }
    pub fn get_embedded<'db>(self, data: &Arc<DefCacheLoadingData<'db>>) -> StructId<'db> {
        data.struct_ids[&self]
    }
}

#[derive(Serialize, Deserialize, Clone, PartialEq, Eq)]
struct EnumCached {
    language_element: LanguageElementCached,
}
impl EnumCached {
    fn new<'db>(enum_id: EnumId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        Self { language_element: LanguageElementCached::new(enum_id, ctx) }
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> EnumLongId<'db> {
        let (module_id, stable_ptr) = self.language_element.embed(ctx);

        EnumLongId(module_id, ItemEnumPtr(stable_ptr))
    }
}

#[derive(Serialize, Deserialize, Copy, Clone, PartialEq, Eq, Hash, salsa::Update)]
pub struct EnumIdCached(usize);

impl EnumIdCached {
    fn new<'db>(id: EnumId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        if let Some(cached_id) = ctx.enum_ids.get(&id) {
            return *cached_id;
        }
        let cached = EnumCached::new(id, ctx);
        let cached_id = EnumIdCached(ctx.enum_ids_lookup.len());
        ctx.enum_ids_lookup.push(cached);
        ctx.enum_ids.insert(id, cached_id);
        cached_id
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> EnumId<'db> {
        if let Some(id) = ctx.enum_ids.get(&self) {
            return *id;
        }
        let cached = ctx.enum_ids_lookup[self.0].clone();
        let id = cached.embed(ctx).intern(ctx.db);
        ctx.enum_ids.insert(self, id);
        id
    }
    pub fn get_embedded<'db>(self, data: &Arc<DefCacheLoadingData<'db>>) -> EnumId<'db> {
        data.enum_ids[&self]
    }
}

#[derive(Serialize, Deserialize, Clone, PartialEq, Eq)]
struct ModuleTypeAliasCached {
    language_element: LanguageElementCached,
}
impl ModuleTypeAliasCached {
    fn new<'db>(
        type_alias_id: ModuleTypeAliasId<'db>,
        ctx: &mut DefCacheSavingContext<'db>,
    ) -> Self {
        Self { language_element: LanguageElementCached::new(type_alias_id, ctx) }
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> ModuleTypeAliasLongId<'db> {
        let (module_id, stable_ptr) = self.language_element.embed(ctx);

        ModuleTypeAliasLongId(module_id, ItemTypeAliasPtr(stable_ptr))
    }
}

#[derive(Serialize, Deserialize, Copy, Clone, PartialEq, Eq, Hash, salsa::Update)]
pub struct ModuleTypeAliasIdCached(usize);

impl ModuleTypeAliasIdCached {
    fn new<'db>(id: ModuleTypeAliasId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        if let Some(cached_id) = ctx.type_alias_ids.get(&id) {
            return *cached_id;
        }
        let cached = ModuleTypeAliasCached::new(id, ctx);
        let cached_id = ModuleTypeAliasIdCached(ctx.type_alias_ids_lookup.len());
        ctx.type_alias_ids_lookup.push(cached);
        ctx.type_alias_ids.insert(id, cached_id);
        cached_id
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> ModuleTypeAliasId<'db> {
        if let Some(id) = ctx.type_alias_ids.get(&self) {
            return *id;
        }
        let cached = ctx.type_alias_ids_lookup[self.0].clone();
        let id = cached.embed(ctx).intern(ctx.db);
        ctx.type_alias_ids.insert(self, id);
        id
    }
    pub fn get_embedded<'db>(self, data: &Arc<DefCacheLoadingData<'db>>) -> ModuleTypeAliasId<'db> {
        data.type_alias_ids[&self]
    }
}

#[derive(Serialize, Deserialize, Clone, PartialEq, Eq)]
struct ImplAliasCached {
    language_element: LanguageElementCached,
}
impl ImplAliasCached {
    fn new<'db>(impl_alias_id: ImplAliasId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        Self { language_element: LanguageElementCached::new(impl_alias_id, ctx) }
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> ImplAliasLongId<'db> {
        let (module_id, stable_ptr) = self.language_element.embed(ctx);

        ImplAliasLongId(module_id, ItemImplAliasPtr(stable_ptr))
    }
}

#[derive(Serialize, Deserialize, Copy, Clone, PartialEq, Eq, Hash, salsa::Update)]
pub struct ImplAliasIdCached(usize);

impl ImplAliasIdCached {
    pub fn new<'db>(id: ImplAliasId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        if let Some(cached_id) = ctx.impl_alias_ids.get(&id) {
            return *cached_id;
        }
        let cached = ImplAliasCached::new(id, ctx);
        let cached_id = ImplAliasIdCached(ctx.impl_alias_ids_lookup.len());
        ctx.impl_alias_ids_lookup.push(cached);
        ctx.impl_alias_ids.insert(id, cached_id);
        cached_id
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> ImplAliasId<'db> {
        if let Some(id) = ctx.impl_alias_ids.get(&self) {
            return *id;
        }
        let cached = ctx.impl_alias_ids_lookup[self.0].clone();
        let id = cached.embed(ctx).intern(ctx.db);
        ctx.impl_alias_ids.insert(self, id);
        id
    }
    pub fn get_embedded<'db>(self, data: &Arc<DefCacheLoadingData<'db>>) -> ImplAliasId<'db> {
        data.impl_alias_ids[&self]
    }
}

#[derive(Serialize, Deserialize, Clone, PartialEq, Eq)]
struct TraitCached {
    language_element: LanguageElementCached,
}
impl TraitCached {
    fn new<'db>(trait_id: TraitId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        Self { language_element: LanguageElementCached::new(trait_id, ctx) }
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> TraitLongId<'db> {
        let (module_id, stable_ptr) = self.language_element.embed(ctx);

        TraitLongId(module_id, ItemTraitPtr(stable_ptr))
    }
}

#[derive(Serialize, Deserialize, Copy, Clone, PartialEq, Eq, Hash, salsa::Update)]
pub struct TraitIdCached(usize);

impl TraitIdCached {
    fn new<'db>(id: TraitId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        if let Some(cached_id) = ctx.trait_ids.get(&id) {
            return *cached_id;
        }
        let cached = TraitCached::new(id, ctx);
        let cached_id = TraitIdCached(ctx.trait_ids_lookup.len());
        ctx.trait_ids_lookup.push(cached);
        ctx.trait_ids.insert(id, cached_id);
        cached_id
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> TraitId<'db> {
        if let Some(id) = ctx.trait_ids.get(&self) {
            return *id;
        }
        let cached = ctx.trait_ids_lookup[self.0].clone();
        let id = cached.embed(ctx).intern(ctx.db);
        ctx.trait_ids.insert(self, id);
        id
    }
    pub fn get_embedded<'db>(self, data: &Arc<DefCacheLoadingData<'db>>) -> TraitId<'db> {
        data.trait_ids[&self]
    }
}

#[derive(Serialize, Deserialize, Clone, PartialEq, Eq)]
struct ImplDefCached {
    language_element: LanguageElementCached,
}
impl ImplDefCached {
    fn new<'db>(impl_def_id: ImplDefId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        Self { language_element: LanguageElementCached::new(impl_def_id, ctx) }
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> ImplDefLongId<'db> {
        let (module_id, stable_ptr) = self.language_element.embed(ctx);

        ImplDefLongId(module_id, ItemImplPtr(stable_ptr))
    }
}

#[derive(Serialize, Deserialize, Copy, Clone, PartialEq, Eq, Hash, salsa::Update)]
pub struct ImplDefIdCached(usize);

impl ImplDefIdCached {
    pub fn new<'db>(id: ImplDefId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        if let Some(cached_id) = ctx.impl_def_ids.get(&id) {
            return *cached_id;
        }
        let cached = ImplDefCached::new(id, ctx);
        let cached_id = ImplDefIdCached(ctx.impl_def_ids_lookup.len());
        ctx.impl_def_ids_lookup.push(cached);
        ctx.impl_def_ids.insert(id, cached_id);
        cached_id
    }

    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> ImplDefId<'db> {
        if let Some(id) = ctx.impl_def_ids.get(&self) {
            return *id;
        }
        let cached = ctx.impl_def_ids_lookup[self.0].clone();
        let id = cached.embed(ctx).intern(ctx.db);
        ctx.impl_def_ids.insert(self, id);
        id
    }

    pub fn get_embedded<'db>(self, data: &Arc<DefCacheLoadingData<'db>>) -> ImplDefId<'db> {
        data.impl_def_ids[&self]
    }
}

#[derive(Serialize, Deserialize, Clone, PartialEq, Eq)]
struct ExternTypeCached {
    language_element: LanguageElementCached,
}
impl ExternTypeCached {
    fn new<'db>(extern_type_id: ExternTypeId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        Self { language_element: LanguageElementCached::new(extern_type_id, ctx) }
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> ExternTypeLongId<'db> {
        let (module_id, stable_ptr) = self.language_element.embed(ctx);

        ExternTypeLongId(module_id, ItemExternTypePtr(stable_ptr))
    }
}

#[derive(Serialize, Deserialize, Copy, Clone, PartialEq, Eq, Hash, salsa::Update)]
pub struct ExternTypeIdCached(usize);

impl ExternTypeIdCached {
    fn new<'db>(id: ExternTypeId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        if let Some(cached_id) = ctx.extern_type_ids.get(&id) {
            return *cached_id;
        }
        let cached = ExternTypeCached::new(id, ctx);
        let cached_id = ExternTypeIdCached(ctx.extern_type_ids_lookup.len());
        ctx.extern_type_ids_lookup.push(cached);
        ctx.extern_type_ids.insert(id, cached_id);
        cached_id
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> ExternTypeId<'db> {
        if let Some(id) = ctx.extern_type_ids.get(&self) {
            return *id;
        }
        let cached = ctx.extern_type_ids_lookup[self.0].clone();
        let id = cached.embed(ctx).intern(ctx.db);
        ctx.extern_type_ids.insert(self, id);
        id
    }
    pub fn get_embedded<'db>(self, data: &Arc<DefCacheLoadingData<'db>>) -> ExternTypeId<'db> {
        data.extern_type_ids[&self]
    }
}

#[derive(Serialize, Deserialize, Clone, PartialEq, Eq)]
struct ExternFunctionCached {
    language_element: LanguageElementCached,
}
impl ExternFunctionCached {
    fn new<'db>(
        extern_function_id: ExternFunctionId<'db>,
        ctx: &mut DefCacheSavingContext<'db>,
    ) -> Self {
        Self { language_element: LanguageElementCached::new(extern_function_id, ctx) }
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> ExternFunctionLongId<'db> {
        let (module_id, stable_ptr) = self.language_element.embed(ctx);

        ExternFunctionLongId(module_id, ItemExternFunctionPtr(stable_ptr))
    }
}

#[derive(Serialize, Deserialize, Copy, Clone, PartialEq, Eq, Hash, salsa::Update)]
pub struct ExternFunctionIdCached(usize);

impl ExternFunctionIdCached {
    fn new<'db>(id: ExternFunctionId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        if let Some(cached_id) = ctx.extern_function_ids.get(&id) {
            return *cached_id;
        }
        let cached = ExternFunctionCached::new(id, ctx);
        let cached_id = ExternFunctionIdCached(ctx.extern_function_ids_lookup.len());
        ctx.extern_function_ids_lookup.push(cached);
        ctx.extern_function_ids.insert(id, cached_id);
        cached_id
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> ExternFunctionId<'db> {
        if let Some(id) = ctx.extern_function_ids.get(&self) {
            return *id;
        }
        let cached = ctx.extern_function_ids_lookup[self.0].clone();
        let id = cached.embed(ctx).intern(ctx.db);
        ctx.extern_function_ids.insert(self, id);
        id
    }
    pub fn get_embedded<'db>(self, data: &Arc<DefCacheLoadingData<'db>>) -> ExternFunctionId<'db> {
        data.extern_function_ids[&self]
    }
}
#[derive(Serialize, Deserialize, Clone, PartialEq, Eq)]
struct MacroDeclarationCached {
    language_element: LanguageElementCached,
}

impl MacroDeclarationCached {
    fn new<'db>(
        macro_declaration_id: MacroDeclarationId<'db>,
        ctx: &mut DefCacheSavingContext<'db>,
    ) -> Self {
        Self { language_element: LanguageElementCached::new(macro_declaration_id, ctx) }
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> MacroDeclarationLongId<'db> {
        let (module_id, stable_ptr) = self.language_element.embed(ctx);

        MacroDeclarationLongId(module_id, ItemMacroDeclarationPtr(stable_ptr))
    }
}

#[derive(Serialize, Deserialize, Copy, Clone, PartialEq, Eq, Hash, salsa::Update)]
pub struct MacroDeclarationIdCached(usize);

impl MacroDeclarationIdCached {
    fn new<'db>(id: MacroDeclarationId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        if let Some(cached_id) = ctx.macro_declaration_ids.get(&id) {
            return *cached_id;
        }
        let cached = MacroDeclarationCached::new(id, ctx);
        let cached_id = MacroDeclarationIdCached(ctx.macro_declaration_ids_lookup.len());
        ctx.macro_declaration_ids_lookup.push(cached);
        ctx.macro_declaration_ids.insert(id, cached_id);
        cached_id
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> MacroDeclarationId<'db> {
        if let Some(id) = ctx.macro_declaration_ids.get(&self) {
            return *id;
        }
        let cached = ctx.macro_declaration_ids_lookup[self.0].clone();
        let id = cached.embed(ctx).intern(ctx.db);
        ctx.macro_declaration_ids.insert(self, id);
        id
    }
    pub fn get_embedded<'db>(
        self,
        data: &Arc<DefCacheLoadingData<'db>>,
    ) -> MacroDeclarationId<'db> {
        data.macro_declaration_ids[&self]
    }
}

#[derive(Serialize, Deserialize, Clone, PartialEq, Eq)]
struct MacroCallCached {
    language_element: LanguageElementCached,
}
impl MacroCallCached {
    fn new<'db>(macro_call: MacroCallId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        Self { language_element: LanguageElementCached::new(macro_call, ctx) }
    }

    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> MacroCallLongId<'db> {
        let (module_id, stable_ptr) = self.language_element.embed(ctx);
        MacroCallLongId(module_id, ItemInlineMacroPtr(stable_ptr))
    }
}

#[derive(Serialize, Deserialize, Copy, Clone, PartialEq, Eq, Hash, salsa::Update)]
pub struct MacroCallIdCached(usize);
impl MacroCallIdCached {
    fn new<'db>(id: MacroCallId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        if let Some(cached_id) = ctx.macro_call_ids.get(&id) {
            return *cached_id;
        }
        let cached = MacroCallCached::new(id, ctx);
        let id_cached = MacroCallIdCached(ctx.macro_call_ids_lookup.len());
        ctx.macro_call_ids_lookup.push(cached);
        ctx.macro_call_ids.insert(id, id_cached);
        id_cached
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> MacroCallId<'db> {
        if let Some(id) = ctx.macro_call_ids.get(&self) {
            return *id;
        }
        let cached = ctx.macro_call_ids_lookup[self.0].clone();
        let id = cached.embed(ctx).intern(ctx.db);
        ctx.macro_call_ids.insert(self, id);
        id
    }
    fn get_embedded<'db>(&self, data: &Arc<DefCacheLoadingData<'db>>) -> MacroCallId<'db> {
        data.macro_call_ids[self]
    }
}

#[derive(Serialize, Deserialize, Clone, PartialEq, Eq)]
struct GlobalUseCached {
    language_element: LanguageElementCached,
}
impl GlobalUseCached {
    fn new<'db>(global_use_id: GlobalUseId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        Self { language_element: LanguageElementCached::new(global_use_id, ctx) }
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> GlobalUseLongId<'db> {
        let (module_id, stable_ptr) = self.language_element.embed(ctx);

        GlobalUseLongId(module_id, UsePathStarPtr(stable_ptr))
    }
}

#[derive(Serialize, Deserialize, Copy, Clone, PartialEq, Eq, Hash, salsa::Update)]
pub struct GlobalUseIdCached(usize);

impl GlobalUseIdCached {
    pub fn new<'db>(id: GlobalUseId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        if let Some(cached_id) = ctx.global_use_ids.get(&id) {
            return *cached_id;
        }
        let cached = GlobalUseCached::new(id, ctx);
        let cached_id = GlobalUseIdCached(ctx.global_use_ids_lookup.len());
        ctx.global_use_ids_lookup.push(cached);
        ctx.global_use_ids.insert(id, cached_id);
        cached_id
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> GlobalUseId<'db> {
        if let Some(id) = ctx.global_use_ids.get(&self) {
            return *id;
        }
        let cached = ctx.global_use_ids_lookup[self.0].clone();
        let id = cached.embed(ctx).intern(ctx.db);
        ctx.global_use_ids.insert(self, id);
        id
    }
    pub fn get_embedded<'db>(&self, data: &Arc<DefCacheLoadingData<'db>>) -> GlobalUseId<'db> {
        data.global_use_ids[self]
    }
}

#[derive(Serialize, Deserialize, Clone, Copy, Hash, Eq, PartialEq, salsa::Update, Debug)]
struct SyntaxNodeCached(usize);

#[derive(Serialize, Deserialize, Clone, Hash, Eq, PartialEq, Debug)]
enum SyntaxNodeIdCached {
    Root(FileIdCached, GreenIdCached),
    Child {
        parent: SyntaxNodeCached,
        kind: SyntaxKind,
        key_fields: Vec<GreenIdCached>,
        index: usize,
    },
}

impl SyntaxNodeIdCached {
    fn new<'db>(
        ctx: &mut DefCacheSavingContext<'db>,
        id: &SyntaxNodeId<'db>,
        syntax_node: SyntaxNode<'db>,
    ) -> Self {
        match id {
            SyntaxNodeId::Root(file_id) => {
                let green = syntax_node.green_node(ctx.db).clone().intern(ctx.db);
                Self::Root(FileIdCached::new(*file_id, ctx), GreenIdCached::new(green, ctx))
            }
            SyntaxNodeId::Child { parent, key_fields, index } => Self::Child {
                parent: SyntaxNodeCached::new(*parent, ctx),
                kind: syntax_node.kind(ctx.db),
                key_fields: key_fields.into_iter().map(|id| GreenIdCached::new(*id, ctx)).collect(),
                index: *index,
            },
        }
    }

    fn from_raw(
        ctx: &mut DefCacheLoadingContext<'_>,
        parent: SyntaxNodeCached,
        kind: SyntaxKind,
        index: usize,
        key_fields: &[GreenId<'_>],
    ) -> Option<Self> {
        let mut cached_key_fields = Vec::with_capacity(key_fields.len());
        for id in key_fields {
            let Some(cached_id) = ctx.reverse_green_ids.get(id) else {
                // Green ID not found in cache, this child can't be reconstructed
                return None;
            };
            cached_key_fields.push(*cached_id);
        }
        Some(Self::Child { parent, kind, key_fields: cached_key_fields, index })
    }
}

impl SyntaxNodeCached {
    fn new<'db>(syntax_node: SyntaxNode<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        if let Some(res) = ctx.syntax_nodes.get(&syntax_node) {
            return *res;
        }
        let db = ctx.db;
        // Cache the green node and all its children to ensure they're available during load
        let green = syntax_node.green_node(db);
        GreenIdCached::new(green.clone().intern(db), ctx);

        let id = SyntaxNodeIdCached::new(ctx, syntax_node.raw_id(db), syntax_node);
        let cached_node = SyntaxNodeCached(ctx.syntax_node_lookup.len());
        ctx.syntax_node_lookup.push(id);
        ctx.syntax_nodes.insert(syntax_node, cached_node);
        cached_node
    }
    fn embed<'db>(&self, ctx: &mut DefCacheLoadingContext<'db>) -> SyntaxNode<'db> {
        if let Some(node) = ctx.syntax_nodes.get(self) {
            return *node;
        }
        let id_cached = ctx.syntax_node_lookup[self.0].clone();
        match &id_cached {
            SyntaxNodeIdCached::Root(file_id, green_id) => {
                let fid = file_id.embed(ctx);
                let green = green_id.embed(ctx);
                let syntax = cairo_lang_syntax::node::SyntaxNode::new_root(ctx.db, fid, green);
                ctx.syntax_nodes.insert(*self, syntax);
            }
            SyntaxNodeIdCached::Child { parent, .. } => {
                let parent_node = parent.embed(ctx);
                let children = parent_node.get_children(ctx.db);
                // For each child, create the cached syntax node ID and insert into syntax_nodes map
                for child in children {
                    let SyntaxNodeId::Child { index, key_fields, .. } = child.raw_id(ctx.db) else {
                        panic!("Unexpected SyntaxNodeId type root when creating child");
                    };
                    let kind = child.kind(ctx.db);
                    let Some(child_id) =
                        SyntaxNodeIdCached::from_raw(ctx, *parent, kind, *index, key_fields)
                    else {
                        continue;
                    };
                    // Check if this child was cached during save
                    let Some(&child_cached) = ctx.reverse_syntax_node_lookup.get(&child_id) else {
                        continue;
                    };
                    ctx.syntax_nodes.insert(child_cached, *child);
                }
            }
        };
        *ctx.syntax_nodes
            .get(self)
            .unwrap_or_else(|| panic!("Failed to find syntax node {:?} after embedding", self.0))
    }
    fn get_embedded<'db>(&self, data: &Arc<DefCacheLoadingData<'db>>) -> SyntaxNode<'db> {
        data.syntax_nodes[self]
    }
}

#[derive(Serialize, Deserialize, Clone, Hash, Eq, PartialEq)]
pub struct LanguageElementCached {
    module_id: ModuleIdCached,
    stable_ptr: SyntaxStablePtrIdCached,
}
impl LanguageElementCached {
    pub fn new<'db, T: LanguageElementId<'db> + 'db>(
        language_element: T,
        ctx: &mut DefCacheSavingContext<'db>,
    ) -> Self {
        Self {
            module_id: ModuleIdCached::new(language_element.parent_module(ctx.db), ctx),
            stable_ptr: SyntaxStablePtrIdCached::new(
                language_element.untyped_stable_ptr(ctx.db),
                ctx,
            ),
        }
    }
    fn embed<'db>(
        self,
        ctx: &mut DefCacheLoadingContext<'db>,
    ) -> (ModuleId<'db>, SyntaxStablePtrId<'db>) {
        let module_id = self.module_id.embed(ctx);
        let stable_ptr = self.stable_ptr.embed(ctx);
        (module_id, stable_ptr)
    }
    pub fn get_embedded<'db>(
        self,
        data: &Arc<DefCacheLoadingData<'db>>,
    ) -> (ModuleId<'db>, SyntaxStablePtrId<'db>) {
        let module_id = self.module_id.get_embedded(data);
        let stable_ptr = self.stable_ptr.get_embedded(data);
        (module_id, stable_ptr)
    }
}

#[derive(Serialize, Deserialize, Clone, PartialEq, Eq)]
enum GreenNodeDetailsCached {
    Token(String),
    Node { children: Vec<GreenIdCached>, width: TextWidth },
}

impl GreenNodeDetailsCached {
    fn new<'db>(
        green_node_details: &GreenNodeDetails<'db>,
        ctx: &mut DefCacheSavingContext<'db>,
    ) -> GreenNodeDetailsCached {
        match green_node_details {
            GreenNodeDetails::Token(token) => {
                GreenNodeDetailsCached::Token(token.long(ctx.db).to_string())
            }
            GreenNodeDetails::Node { children, width } => GreenNodeDetailsCached::Node {
                children: children.iter().map(|child| GreenIdCached::new(*child, ctx)).collect(),
                width: *width,
            },
        }
    }
    fn embed<'db>(&self, ctx: &mut DefCacheLoadingContext<'db>) -> GreenNodeDetails<'db> {
        match self {
            GreenNodeDetailsCached::Token(token) => {
                GreenNodeDetails::Token(SmolStrId::from(ctx.db, token))
            }
            GreenNodeDetailsCached::Node { children, width } => GreenNodeDetails::Node {
                children: children.iter().map(|child| child.embed(ctx)).collect(),
                width: *width,
            },
        }
    }
}

#[derive(Serialize, Deserialize, Clone, PartialEq, Eq)]
struct GreenNodeCached {
    kind: SyntaxKind,
    details: GreenNodeDetailsCached,
}
impl GreenNodeCached {
    fn new<'db>(green_node: &GreenNode<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        Self {
            kind: green_node.kind,
            details: GreenNodeDetailsCached::new(&green_node.details, ctx),
        }
    }
    fn embed<'db>(&self, ctx: &mut DefCacheLoadingContext<'db>) -> GreenNode<'db> {
        GreenNode { kind: self.kind, details: self.details.embed(ctx) }
    }
}

#[derive(Serialize, Deserialize, Clone, Copy, Eq, Hash, PartialEq, salsa::Update, Debug)]
struct GreenIdCached(usize);

impl GreenIdCached {
    fn new<'db>(id: GreenId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        if let Some(cached_id) = ctx.green_ids.get(&id) {
            return *cached_id;
        }
        let cached = GreenNodeCached::new(id.long(ctx.db), ctx);
        let cached_id = GreenIdCached(ctx.green_ids_lookup.len());
        ctx.green_ids_lookup.push(cached);
        ctx.green_ids.insert(id, cached_id);
        cached_id
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> GreenId<'db> {
        if let Some(id) = ctx.green_ids.get(&self) {
            return *id;
        }
        let cached = ctx.green_ids_lookup[self.0].clone();
        let id = cached.embed(ctx).intern(ctx.db);
        ctx.green_ids.insert(self, id);
        id
    }
}
#[derive(Serialize, Deserialize, Clone, PartialEq, Eq)]
enum FileCached {
    OnDisk(PathBuf),
    Virtual(VirtualFileCached),
    External(PluginGeneratedFileCached),
}

impl FileCached {
    fn new<'db>(file: &FileLongId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        match file {
            FileLongId::OnDisk(path) => FileCached::OnDisk(path.clone()),
            FileLongId::Virtual(virtual_file) => {
                FileCached::Virtual(VirtualFileCached::new(virtual_file, ctx))
            }
            FileLongId::External(external_file) => {
                FileCached::External(PluginGeneratedFileCached::new(
                    PluginGeneratedFileId::from_intern_id(*external_file),
                    ctx,
                ))
            }
        }
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> FileLongId<'db> {
        match self {
            FileCached::OnDisk(path) => FileLongId::OnDisk(path),
            FileCached::Virtual(virtual_file) => FileLongId::Virtual(virtual_file.embed(ctx)),
            FileCached::External(external_file) => {
                FileLongId::External(external_file.embed(ctx).as_intern_id())
            }
        }
    }
}

#[derive(Serialize, Deserialize, Clone, Copy, Eq, Hash, PartialEq, salsa::Update, Debug)]
pub struct FileIdCached(usize);
impl FileIdCached {
    fn new<'db>(id: FileId<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        if let Some(cached_id) = ctx.file_ids.get(&id) {
            return *cached_id;
        }
        let cached = FileCached::new(id.long(ctx.db), ctx);
        let cached_id = FileIdCached(ctx.file_ids_lookup.len());
        ctx.file_ids_lookup.push(cached);
        ctx.file_ids.insert(id, cached_id);
        cached_id
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> FileId<'db> {
        if let Some(id) = ctx.file_ids.get(&self) {
            return *id;
        }
        let cached = ctx.file_ids_lookup[self.0].clone();
        let id = cached.embed(ctx).intern(ctx.db);
        ctx.file_ids.insert(self, id);
        id
    }
    fn get_embedded<'db>(&self, data: &Arc<DefCacheLoadingData<'db>>) -> FileId<'db> {
        data.file_ids[self]
    }
}

#[derive(Serialize, Deserialize, Clone, PartialEq, Eq)]
struct VirtualFileCached {
    parent: Option<SpanInFileCached>,
    name: String,
    content: String,
    code_mappings: Vec<CodeMapping>,
    kind: FileKind,
    original_item_removed: bool,
}

impl VirtualFileCached {
    fn new<'db>(virtual_file: &VirtualFile<'db>, ctx: &mut DefCacheSavingContext<'db>) -> Self {
        Self {
            parent: virtual_file.parent.map(|parent| SpanInFileCached::new(&parent, ctx)),
            name: virtual_file.name.to_string(ctx.db),
            content: virtual_file.content.to_string(ctx.db),
            code_mappings: virtual_file.code_mappings.to_vec(),
            kind: virtual_file.kind,
            original_item_removed: virtual_file.original_item_removed,
        }
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> VirtualFile<'db> {
        VirtualFile {
            parent: self.parent.map(|parent| parent.embed(ctx)),
            name: SmolStrId::from(ctx.db, self.name),
            content: SmolStrId::from(ctx.db, self.content),
            code_mappings: self.code_mappings.into(),
            kind: self.kind,
            original_item_removed: self.original_item_removed,
        }
    }
}

#[derive(Serialize, Deserialize, Clone, Copy, Eq, Hash, PartialEq, salsa::Update)]
pub struct SyntaxStablePtrIdCached(SyntaxNodeCached);

impl SyntaxStablePtrIdCached {
    pub fn new<'db>(
        stable_ptr: SyntaxStablePtrId<'db>,
        ctx: &mut DefCacheSavingContext<'db>,
    ) -> Self {
        Self(SyntaxNodeCached::new(stable_ptr.0, ctx))
    }

    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> SyntaxStablePtrId<'db> {
        SyntaxStablePtrId(self.0.embed(ctx))
    }

    pub fn get_embedded<'db>(
        &self,
        data: &Arc<DefCacheLoadingData<'db>>,
    ) -> SyntaxStablePtrId<'db> {
        SyntaxStablePtrId(self.0.get_embedded(data))
    }
}

#[derive(Serialize, Deserialize, Clone, PartialEq, Eq)]
struct PluginGeneratedFileCached {
    module_id: ModuleIdCached,
    stable_ptr: SyntaxStablePtrIdCached,
    name: String,
}

impl PluginGeneratedFileCached {
    fn new<'db>(
        plugin_generated_file: PluginGeneratedFileId<'db>,
        ctx: &mut DefCacheSavingContext<'db>,
    ) -> Self {
        let long_id = plugin_generated_file.long(ctx.db);
        Self {
            module_id: ModuleIdCached::new(long_id.module_id, ctx),
            stable_ptr: SyntaxStablePtrIdCached::new(long_id.stable_ptr, ctx),
            name: long_id.name.clone(),
        }
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> PluginGeneratedFileId<'db> {
        let module_id = self.module_id.embed(ctx);
        let stable_ptr = self.stable_ptr.embed(ctx);
        let long_id = PluginGeneratedFileLongId { module_id, stable_ptr, name: self.name };
        long_id.intern(ctx.db)
    }
}

#[derive(Serialize, Deserialize, Clone)]
enum SeverityCached {
    Error,
    Warning,
}

#[derive(Serialize, Deserialize, Clone)]
struct PluginDiagnosticCached {
    stable_ptr: SyntaxStablePtrIdCached,
    message: String,
    severity: SeverityCached,
    inner_span: Option<(TextWidth, TextWidth)>,
}

impl PluginDiagnosticCached {
    fn new<'db>(
        diagnostic: &PluginDiagnostic<'db>,
        ctx: &mut DefCacheSavingContext<'db>,
    ) -> PluginDiagnosticCached {
        PluginDiagnosticCached {
            stable_ptr: SyntaxStablePtrIdCached::new(diagnostic.stable_ptr, ctx),
            message: diagnostic.message.clone(),
            severity: match diagnostic.severity {
                Severity::Error => SeverityCached::Error,
                Severity::Warning => SeverityCached::Warning,
            },
            inner_span: diagnostic.inner_span,
        }
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> PluginDiagnostic<'db> {
        PluginDiagnostic {
            stable_ptr: self.stable_ptr.embed(ctx),
            message: self.message,
            severity: match self.severity {
                SeverityCached::Error => Severity::Error,
                SeverityCached::Warning => Severity::Warning,
            },
            inner_span: self.inner_span,
            error_code: None,
        }
    }
}

type PluginFileDiagnosticNotesCached = OrderedHashMap<FileIdCached, DiagnosticNoteCached>;

#[derive(Serialize, Deserialize, Clone)]
struct DiagnosticNoteCached {
    text: String,
    location: Option<SpanInFileCached>,
}

impl DiagnosticNoteCached {
    fn new<'db>(
        note: DiagnosticNote<'db>,
        ctx: &mut DefCacheSavingContext<'db>,
    ) -> DiagnosticNoteCached {
        DiagnosticNoteCached {
            text: note.text.clone(),
            location: note.location.map(|location| SpanInFileCached::new(&location, ctx)),
        }
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> DiagnosticNote<'db> {
        DiagnosticNote {
            text: self.text,
            location: self.location.map(|location| location.embed(ctx)),
        }
    }
}

#[derive(Serialize, Deserialize, Clone, PartialEq, Eq)]
struct SpanInFileCached {
    file_id: FileIdCached,
    span: TextSpan,
}

impl SpanInFileCached {
    fn new<'db>(
        location: &SpanInFile<'db>,
        ctx: &mut DefCacheSavingContext<'db>,
    ) -> SpanInFileCached {
        SpanInFileCached { file_id: FileIdCached::new(location.file_id, ctx), span: location.span }
    }
    fn embed<'db>(self, ctx: &mut DefCacheLoadingContext<'db>) -> SpanInFile<'db> {
        SpanInFile { file_id: self.file_id.embed(ctx), span: self.span }
    }
}