Skip to main content

fallow_extract/cache/
conversion.rs

1//! Conversion between [`ModuleInfo`](crate::ModuleInfo) and [`CachedModule`].
2
3use oxc_span::Span;
4
5use crate::ExportName;
6
7use super::types::*;
8
9/// Reconstruct a [`ModuleInfo`](crate::ModuleInfo) from a [`CachedModule`].
10pub fn cached_to_module(
11    cached: &CachedModule,
12    file_id: fallow_types::discover::FileId,
13) -> crate::ModuleInfo {
14    use crate::*;
15
16    let exports = cached
17        .exports
18        .iter()
19        .map(|e| ExportInfo {
20            name: if e.is_default {
21                ExportName::Default
22            } else {
23                ExportName::Named(e.name.clone())
24            },
25            local_name: e.local_name.clone(),
26            is_type_only: e.is_type_only,
27            is_public: e.is_public,
28            span: Span::new(e.span_start, e.span_end),
29            members: e
30                .members
31                .iter()
32                .map(|m| MemberInfo {
33                    name: m.name.clone(),
34                    kind: m.kind.clone(),
35                    span: Span::new(m.span_start, m.span_end),
36                    has_decorator: m.has_decorator,
37                })
38                .collect(),
39        })
40        .collect();
41
42    let imports = cached
43        .imports
44        .iter()
45        .map(|i| ImportInfo {
46            source: i.source.clone(),
47            imported_name: match i.kind {
48                IMPORT_KIND_DEFAULT => ImportedName::Default,
49                IMPORT_KIND_NAMESPACE => ImportedName::Namespace,
50                IMPORT_KIND_SIDE_EFFECT => ImportedName::SideEffect,
51                // IMPORT_KIND_NAMED (0) and any unknown value default to Named
52                _ => ImportedName::Named(i.imported_name.clone()),
53            },
54            local_name: i.local_name.clone(),
55            is_type_only: i.is_type_only,
56            span: Span::new(i.span_start, i.span_end),
57        })
58        .collect();
59
60    let re_exports = cached
61        .re_exports
62        .iter()
63        .map(|r| ReExportInfo {
64            source: r.source.clone(),
65            imported_name: r.imported_name.clone(),
66            exported_name: r.exported_name.clone(),
67            is_type_only: r.is_type_only,
68        })
69        .collect();
70
71    let dynamic_imports = cached
72        .dynamic_imports
73        .iter()
74        .map(|d| DynamicImportInfo {
75            source: d.source.clone(),
76            span: Span::new(d.span_start, d.span_end),
77            destructured_names: d.destructured_names.clone(),
78            local_name: d.local_name.clone(),
79        })
80        .collect();
81
82    let require_calls = cached
83        .require_calls
84        .iter()
85        .map(|r| RequireCallInfo {
86            source: r.source.clone(),
87            span: Span::new(r.span_start, r.span_end),
88            destructured_names: r.destructured_names.clone(),
89            local_name: r.local_name.clone(),
90        })
91        .collect();
92
93    let dynamic_import_patterns = cached
94        .dynamic_import_patterns
95        .iter()
96        .map(|p| crate::DynamicImportPattern {
97            prefix: p.prefix.clone(),
98            suffix: p.suffix.clone(),
99            span: Span::new(p.span_start, p.span_end),
100        })
101        .collect();
102
103    let suppressions = cached
104        .suppressions
105        .iter()
106        .map(|s| crate::suppress::Suppression {
107            line: s.line,
108            kind: if s.kind == 0 {
109                None
110            } else {
111                crate::suppress::IssueKind::from_discriminant(s.kind)
112            },
113        })
114        .collect();
115
116    ModuleInfo {
117        file_id,
118        exports,
119        imports,
120        re_exports,
121        dynamic_imports,
122        dynamic_import_patterns,
123        require_calls,
124        member_accesses: cached.member_accesses.clone(),
125        whole_object_uses: cached.whole_object_uses.clone(),
126        has_cjs_exports: cached.has_cjs_exports,
127        content_hash: cached.content_hash,
128        suppressions,
129        unused_import_bindings: cached.unused_import_bindings.clone(),
130        line_offsets: cached.line_offsets.clone(),
131    }
132}
133
134/// Convert a [`ModuleInfo`](crate::ModuleInfo) to a [`CachedModule`] for storage.
135///
136/// `mtime_secs` and `file_size` come from `std::fs::metadata()` at parse time
137/// and enable fast cache validation on subsequent runs (skip file read when
138/// mtime+size match).
139pub fn module_to_cached(
140    module: &crate::ModuleInfo,
141    mtime_secs: u64,
142    file_size: u64,
143) -> CachedModule {
144    CachedModule {
145        content_hash: module.content_hash,
146        mtime_secs,
147        file_size,
148        exports: module
149            .exports
150            .iter()
151            .map(|e| CachedExport {
152                name: match &e.name {
153                    ExportName::Named(n) => n.clone(),
154                    ExportName::Default => String::new(),
155                },
156                is_default: matches!(e.name, ExportName::Default),
157                is_type_only: e.is_type_only,
158                is_public: e.is_public,
159                local_name: e.local_name.clone(),
160                span_start: e.span.start,
161                span_end: e.span.end,
162                members: e
163                    .members
164                    .iter()
165                    .map(|m| CachedMember {
166                        name: m.name.clone(),
167                        kind: m.kind.clone(),
168                        span_start: m.span.start,
169                        span_end: m.span.end,
170                        has_decorator: m.has_decorator,
171                    })
172                    .collect(),
173            })
174            .collect(),
175        imports: module
176            .imports
177            .iter()
178            .map(|i| {
179                let (kind, imported_name) = match &i.imported_name {
180                    crate::ImportedName::Named(n) => (IMPORT_KIND_NAMED, n.clone()),
181                    crate::ImportedName::Default => (IMPORT_KIND_DEFAULT, String::new()),
182                    crate::ImportedName::Namespace => (IMPORT_KIND_NAMESPACE, String::new()),
183                    crate::ImportedName::SideEffect => (IMPORT_KIND_SIDE_EFFECT, String::new()),
184                };
185                CachedImport {
186                    source: i.source.clone(),
187                    imported_name,
188                    local_name: i.local_name.clone(),
189                    is_type_only: i.is_type_only,
190                    kind,
191                    span_start: i.span.start,
192                    span_end: i.span.end,
193                }
194            })
195            .collect(),
196        re_exports: module
197            .re_exports
198            .iter()
199            .map(|r| CachedReExport {
200                source: r.source.clone(),
201                imported_name: r.imported_name.clone(),
202                exported_name: r.exported_name.clone(),
203                is_type_only: r.is_type_only,
204            })
205            .collect(),
206        dynamic_imports: module
207            .dynamic_imports
208            .iter()
209            .map(|d| CachedDynamicImport {
210                source: d.source.clone(),
211                span_start: d.span.start,
212                span_end: d.span.end,
213                destructured_names: d.destructured_names.clone(),
214                local_name: d.local_name.clone(),
215            })
216            .collect(),
217        require_calls: module
218            .require_calls
219            .iter()
220            .map(|r| CachedRequireCall {
221                source: r.source.clone(),
222                span_start: r.span.start,
223                span_end: r.span.end,
224                destructured_names: r.destructured_names.clone(),
225                local_name: r.local_name.clone(),
226            })
227            .collect(),
228        member_accesses: module.member_accesses.clone(),
229        whole_object_uses: module.whole_object_uses.clone(),
230        dynamic_import_patterns: module
231            .dynamic_import_patterns
232            .iter()
233            .map(|p| CachedDynamicImportPattern {
234                prefix: p.prefix.clone(),
235                suffix: p.suffix.clone(),
236                span_start: p.span.start,
237                span_end: p.span.end,
238            })
239            .collect(),
240        has_cjs_exports: module.has_cjs_exports,
241        unused_import_bindings: module.unused_import_bindings.clone(),
242        suppressions: module
243            .suppressions
244            .iter()
245            .map(|s| CachedSuppression {
246                line: s.line,
247                kind: s.kind.map_or(0, |k| k.to_discriminant()),
248            })
249            .collect(),
250        line_offsets: module.line_offsets.clone(),
251    }
252}