forklaunch 1.11.2

Launch faster with forklaunch
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69

use anyhow::Result;
use oxc_allocator::Allocator;
use oxc_ast::ast::{Program, SourceType};

use crate::core::ast::{
    injections::{
        inject_into_import_statement::inject_specifier_into_import_statement,
        inject_into_registrations_ts::inject_into_registrations_config_injector,
    },
    parse_ast_program::parse_ast_program,
};

pub(crate) fn database_entity_manager_runtime_dependency<'a>(
    allocator: &'a Allocator,
    registrations_program: &mut Program<'a>,
    orm_token: &str,
    em_token: &str,
) -> Result<()> {
    let entity_manager_registration_text: &'static str = Box::leak(
        format!(
            "const configInjector = createConfigInjector(SchemaValidator(), {{
                {orm_token}: {{
                  lifetime: Lifetime.Singleton,
                  type: MikroORM,
                  factory: () => new MikroORM(mikroOrmOptionsConfig)
                }},
                {em_token}: {{
                  lifetime: Lifetime.Scoped,
                  type: EntityManager,
                  factory: (
                    {{ {orm_token} }},
                    context?: {{ entityManagerOptions?: ForkOptions; tenantId?: string }}
                  ) =>
                    wrapEmWithTenantContext(
                      {orm_token}.em.fork(context?.entityManagerOptions),
                      context?.tenantId
                    ),
                }},
            }})"
        )
        .into_boxed_str(),
    );

    let mut entity_manager_registration_program = parse_ast_program(
        allocator,
        entity_manager_registration_text,
        SourceType::ts(),
    );

    inject_into_registrations_config_injector(
        allocator,
        registrations_program,
        &mut entity_manager_registration_program,
        "runtimeDependencies",
    )?;

    // Ensure `wrapEmWithTenantContext` is imported. The factory body above
    // calls it; without this the registrations file won't compile after
    // the database infra is added.
    inject_specifier_into_import_statement(
        allocator,
        registrations_program,
        "wrapEmWithTenantContext",
        "@forklaunch/core/persistence",
    )
    .ok();

    Ok(())
}