#![allow(clippy::too_many_arguments)]

use crate::docs::ModuleDocumentation;
use crate::module::{
    CheckedModule, ConstrainedModule, EntryPoint, Expectations, ExposedToHost,
    FoundSpecializationsModule, LateSpecializationsModule, LoadedModule, ModuleHeader,
    ModuleTiming, MonomorphizedModule, ParsedModule, ToplevelExpects, TypeCheckedModule,
};
use crate::module_cache::ModuleCache;
use bumpalo::{collections::CollectIn, Bump};
use crossbeam::channel::{bounded, Sender};
use crossbeam::deque::{Injector, Worker};
use crossbeam::thread;
use parking_lot::Mutex;
use roc_builtins::roc::module_source;
use roc_can::abilities::{AbilitiesStore, PendingAbilitiesStore, ResolvedImpl};
use roc_can::constraint::{Constraint as ConstraintSoa, Constraints, TypeOrVar};
use roc_can::env::FxMode;
use roc_can::expr::{DbgLookup, Declarations, ExpectLookup, PendingDerives};
use roc_can::module::{
    canonicalize_module_defs, ExposedByModule, ExposedForModule, ExposedModuleTypes, Module,
    ModuleParams, ResolvedImplementations, TypeState,
};
use roc_collections::soa::slice_extend_new;
use roc_collections::{default_hasher, BumpMap, MutMap, MutSet, VecMap, VecSet};
use roc_constrain::module::constrain_module;
use roc_debug_flags::dbg_do;
#[cfg(debug_assertions)]
use roc_debug_flags::{
    ROC_CHECK_MONO_IR, ROC_PRINT_IR_AFTER_DROP_SPECIALIZATION, ROC_PRINT_IR_AFTER_REFCOUNT,
    ROC_PRINT_IR_AFTER_RESET_REUSE, ROC_PRINT_IR_AFTER_SPECIALIZATION, ROC_PRINT_IR_AFTER_TRMC,
    ROC_PRINT_LOAD_LOG,
};
use roc_derive::SharedDerivedModule;
use roc_error_macros::internal_error;
use roc_late_solve::{AbilitiesView, WorldAbilities};
use roc_module::ident::{Ident, IdentSuffix, ModuleName, QualifiedModuleName};
use roc_module::symbol::{
    IdentIds, IdentIdsByModule, Interns, ModuleId, ModuleIds, PQModuleName, PackageModuleIds,
    PackageQualified, Symbol,
};
use roc_mono::ir::{
    CapturedSymbols, ExternalSpecializations, GlueLayouts, HostExposedLambdaSets, PartialProc,
    Proc, ProcLayout, Procs, ProcsBase, UpdateModeIds, UsageTrackingMap,
};
use roc_mono::layout::{
    GlobalLayoutInterner, LambdaName, Layout, LayoutCache, LayoutProblem, Niche, STLayoutInterner,
};
use roc_mono::reset_reuse;
use roc_mono::{drop_specialization, inc_dec};
use roc_packaging::cache::RocCacheDir;
use roc_parse::ast::{self, CommentOrNewline, ExtractSpaces, Spaced, ValueDef};
use roc_parse::header::parse_module_defs;
use roc_parse::header::{
    self, AppHeader, ExposedName, HeaderType, ImportsKeywordItem, PackageEntry, PackageHeader,
    PlatformHeader, To,
};
use roc_parse::parser::{FileError, SourceError, SyntaxError};
use roc_problem::Severity;
use roc_region::all::{LineInfo, Loc, Region};
use roc_reporting::error::r#type::suggest;
#[cfg(not(target_family = "wasm"))]
use roc_reporting::report::to_https_problem_report_string;
use roc_reporting::report::{to_file_problem_report_string, Palette, RenderTarget};
use roc_solve::module::{extract_module_owned_implementations, SolveConfig, Solved, SolvedModule};
use roc_solve::FunctionKind;
use roc_solve_problem::TypeError;
use roc_target::Target;
use roc_types::subs::{CopiedImport, ExposedTypesStorageSubs, Subs, VarStore, Variable};
use roc_types::types::{Alias, Types};
use roc_worker::ChannelProblem;
use std::collections::hash_map::Entry::{Occupied, Vacant};
use std::collections::HashMap;
use std::io;
use std::ops::ControlFlow;
use std::path::{Path, PathBuf};
use std::str::from_utf8_unchecked;
use std::sync::Arc;
use std::{env, fs};
#[cfg(not(target_family = "wasm"))]
use {
    roc_packaging::cache::{self},
    roc_packaging::https::{PackageMetadata, Problem},
};

pub use roc_work::Phase;
use roc_work::{DepCycle, Dependencies};

#[cfg(target_family = "wasm")]
use crate::wasm_instant::{Duration, Instant};
#[cfg(not(target_family = "wasm"))]
use std::time::{Duration, Instant};

/// Filename extension for normal Roc modules
const ROC_FILE_EXTENSION: &str = "roc";

/// The . in between module names like Foo.Bar.Baz
const MODULE_SEPARATOR: char = '.';

/// Default name for the main module
const DEFAULT_MAIN_NAME: &str = "main.roc";

const EXPANDED_STACK_SIZE: usize = 8 * 1024 * 1024;

macro_rules! log {
    ($($arg:tt)*) => (dbg_do!(ROC_PRINT_LOAD_LOG, println!($($arg)*)))
}

#[derive(Debug)]
pub struct LoadConfig {
    pub target: Target,
    pub render: RenderTarget,
    pub palette: Palette,
    pub threading: Threading,
    pub exec_mode: ExecutionMode,
    pub function_kind: FunctionKind,
}

#[derive(Debug, Clone, Copy)]
pub enum ExecutionMode {
    Check,
    Executable,
    /// Like [`ExecutionMode::Executable`], but stops in the presence of type errors.
    ExecutableIfCheck,
    /// Test is like [`ExecutionMode::ExecutableIfCheck`], but rather than producing a proper
    /// executable, run tests.
    Test,
}

impl ExecutionMode {
    fn goal_phase(&self) -> Phase {
        use ExecutionMode::*;

        match self {
            Executable => Phase::MakeSpecializations,
            Check | ExecutableIfCheck | Test => Phase::SolveTypes,
        }
    }

    fn build_if_checks(&self) -> bool {
        matches!(self, Self::ExecutableIfCheck | Self::Test)
    }
}

type SharedIdentIdsByModule = Arc<Mutex<roc_module::symbol::IdentIdsByModule>>;

fn start_phase<'a>(
    module_id: ModuleId,
    phase: Phase,
    arena: &'a Bump,
    state: &mut State<'a>,
) -> Vec<BuildTask<'a>> {
    // we blindly assume all dependencies are met

    use roc_work::PrepareStartPhase::*;

    match state.dependencies.prepare_start_phase(module_id, phase) {
        Continue => {
            // fall through
        }
        Done => {
            // no more work to do
            return vec![];
        }
        Recurse(new) => {
            return new
                .into_iter()
                .flat_map(|(module_id, phase)| start_phase(module_id, phase, arena, state))
                .collect()
        }
    }

    let task = {
        match phase {
            Phase::LoadHeader => {
                let opt_dep_name = state.module_cache.module_names.get(&module_id);

                match opt_dep_name {
                    None => {
                        panic!("Module {module_id:?} is not in module_cache.module_names")
                    }
                    Some(dep_name) => {
                        let module_name = dep_name.clone();

                        BuildTask::LoadModule {
                            module_name,
                            // Provide mutexes of ModuleIds and IdentIds by module,
                            // so other modules can populate them as they load.
                            module_ids: Arc::clone(&state.arc_modules),
                            shorthands: Arc::clone(&state.arc_shorthands),
                            ident_ids_by_module: Arc::clone(&state.ident_ids_by_module),
                        }
                    }
                }
            }
            Phase::Parse => {
                // parse the file
                let header = state.module_cache.headers.remove(&module_id).unwrap();

                BuildTask::Parse {
                    header,
                    arc_shorthands: Arc::clone(&state.arc_shorthands),
                    module_ids: Arc::clone(&state.arc_modules),
                    ident_ids_by_module: Arc::clone(&state.ident_ids_by_module),
                    root_type: state.root_type.clone(),
                }
            }
            Phase::CanonicalizeAndConstrain => {
                // canonicalize the file
                let parsed = state.module_cache.parsed.remove(&module_id).unwrap();

                let deps_by_name = &parsed.deps_by_name;
                let num_deps = deps_by_name.len();
                let mut dep_idents: IdentIdsByModule = IdentIds::exposed_builtins(num_deps);

                let State {
                    ident_ids_by_module,
                    ..
                } = &state;

                {
                    let ident_ids_by_module = (*ident_ids_by_module).lock();

                    // Populate dep_idents with each of their IdentIds,
                    // which we'll need during canonicalization to translate
                    // identifier strings into IdentIds, which we need to build Symbols.
                    // We only include the modules we care about (the ones we import).
                    //
                    // At the end of this loop, dep_idents contains all the information to
                    // resolve a symbol from another module: if it's in here, that means
                    // we have both imported the module and the ident was exported by that module.
                    for dep_id in deps_by_name.values() {
                        // We already verified that these are all present,
                        // so unwrapping should always succeed here.
                        let idents = ident_ids_by_module.get(dep_id).unwrap();

                        dep_idents.insert(*dep_id, idents.clone());
                    }
                }

                // Clone the module_ids we'll need for canonicalization.
                // This should be small, and cloning it should be quick.
                // We release the lock as soon as we're done cloning, so we don't have
                // to lock the global module_ids while canonicalizing any given module.
                let qualified_module_ids = Arc::clone(&state.arc_modules);
                let qualified_module_ids = { (*qualified_module_ids).lock().clone() };

                let exposed_symbols = state
                    .exposed_symbols_by_module
                    .get(&module_id)
                    .expect("Could not find listener ID in exposed_symbols_by_module")
                    .clone();

                let mut aliases = MutMap::default();
                let mut abilities_store = PendingAbilitiesStore::default();
                let mut imported_module_params = VecMap::default();

                for imported in parsed.available_modules.keys() {
                    match state.module_cache.aliases.get(imported) {
                        None => unreachable!(
                            r"imported module {:?} did not register its aliases, so {:?} cannot use them",
                            imported, parsed.module_id,
                        ),
                        Some(new) => {
                            aliases.extend(new.iter().filter_map(|(s, (exposed, a))| {
                                // only pass this on if it's exposed, or the alias is a transitive import
                                if *exposed || s.module_id() != *imported {
                                    Some((*s, a.clone()))
                                } else {
                                    None
                                }
                            }));
                        }
                    }

                    match state.module_cache.pending_abilities.get(imported) {
                        None => unreachable!(
                            r"imported module {:?} did not register its abilities, so {:?} cannot use them",
                            imported, parsed.module_id,
                        ),
                        Some(import_store) => {
                            let exposed_symbols = state
                                .exposed_symbols_by_module
                                .get(imported)
                                .unwrap_or_else(|| {
                                    internal_error!(
                                        "Could not find exposed symbols of imported {:?}",
                                        imported
                                    )
                                });

                            // Add the declared abilities from the modules we import;
                            // we may not know all their types yet since type-solving happens in
                            // parallel, but we'll fill that in during type-checking our module.
                            abilities_store
                                .union(import_store.closure_from_imported(exposed_symbols));
                        }
                    }

                    if let Some(params) = state.module_cache.module_params.get(imported) {
                        imported_module_params.insert(*imported, params.clone());
                    }
                }

                let skip_constraint_gen = {
                    // Give this its own scope to make sure that the Guard from the lock() is dropped
                    // immediately after contains_key returns
                    state.cached_types.lock().contains_key(&module_id)
                };

                BuildTask::CanonicalizeAndConstrain {
                    parsed,
                    dep_idents,
                    exposed_symbols,
                    qualified_module_ids,
                    aliases,
                    abilities_store,
                    skip_constraint_gen,
                    exposed_module_ids: state.exposed_modules,
                    exec_mode: state.exec_mode,
                    imported_module_params,
                    fx_mode: state.fx_mode,
                }
            }

            Phase::SolveTypes => {
                let constrained = state.module_cache.constrained.remove(&module_id).unwrap();

                let ConstrainedModule {
                    module,
                    ident_ids,
                    module_timing,
                    constraints,
                    constraint,
                    var_store,
                    available_modules,
                    declarations,
                    dep_idents,
                    pending_derives,
                    types,
                    ..
                } = constrained;

                let derived_module = SharedDerivedModule::clone(&state.derived_module);

                #[cfg(debug_assertions)]
                let checkmate = if roc_checkmate::is_checkmate_enabled() {
                    Some(roc_checkmate::Collector::new())
                } else {
                    None
                };

                let is_host_exposed = state.root_id == module.module_id;

                BuildTask::solve_module(
                    module,
                    ident_ids,
                    module_timing,
                    types,
                    constraints,
                    constraint,
                    state.function_kind,
                    pending_derives,
                    var_store,
                    available_modules,
                    &state.exposed_types,
                    dep_idents,
                    declarations,
                    state.cached_types.clone(),
                    derived_module,
                    state.exec_mode,
                    is_host_exposed,
                    //
                    #[cfg(debug_assertions)]
                    checkmate,
                )
            }
            Phase::FindSpecializations => {
                let typechecked = state.module_cache.typechecked.remove(&module_id).unwrap();

                let TypeCheckedModule {
                    layout_cache,
                    module_id,
                    module_timing,
                    solved_subs,
                    decls,
                    ident_ids,
                    abilities_store,
                    expectations,
                    //
                    #[cfg(debug_assertions)]
                        checkmate: _,
                } = typechecked;

                let our_exposed_types = state
                    .exposed_types
                    .get(&module_id)
                    .unwrap_or_else(|| internal_error!("Exposed types for {:?} missing", module_id))
                    .clone();

                state.world_abilities.insert(
                    module_id,
                    abilities_store.clone(),
                    our_exposed_types.exposed_types_storage_subs,
                );

                let mut imported_module_thunks = bumpalo::collections::Vec::new_in(arena);

                if let Some(imports) = state.module_cache.imports.get(&module_id) {
                    for imported in imports.iter() {
                        imported_module_thunks.extend(
                            state.module_cache.top_level_thunks[imported]
                                .iter()
                                .copied(),
                        );
                    }
                }

                let derived_module = SharedDerivedModule::clone(&state.derived_module);

                let build_expects =
                    matches!(state.exec_mode, ExecutionMode::Test) && expectations.is_some();

                BuildTask::BuildPendingSpecializations {
                    layout_cache,
                    module_id,
                    module_timing,
                    solved_subs,
                    imported_module_thunks: imported_module_thunks.into_bump_slice(),
                    decls,
                    ident_ids,
                    exposed_to_host: state.exposed_to_host.clone(),
                    world_abilities: state.world_abilities.clone_ref(),
                    // TODO: awful, how can we get rid of the clone?
                    exposed_by_module: state.exposed_types.clone(),
                    derived_module,
                    expectations,
                    build_expects,
                }
            }
            Phase::MakeSpecializations => {
                let mut specializations_we_must_make = state
                    .module_cache
                    .external_specializations_requested
                    .remove(&module_id)
                    .unwrap_or_default();

                if module_id == ModuleId::DERIVED_GEN {
                    // The derived gen module must also fulfill also specializations asked of the
                    // derived synth module.
                    let derived_synth_specializations = state
                        .module_cache
                        .external_specializations_requested
                        .remove(&ModuleId::DERIVED_SYNTH)
                        .unwrap_or_default();
                    specializations_we_must_make.extend(derived_synth_specializations)
                }

                let (
                    mut ident_ids,
                    mut subs,
                    expectations,
                    mut procs_base,
                    layout_cache,
                    mut module_timing,
                ) = if state.make_specializations_pass.current_pass() == 1
                    && module_id == ModuleId::DERIVED_GEN
                {
                    // This is the first time the derived module is introduced into the load
                    // graph. It has no abilities of its own or anything, just generate fresh
                    // information for it.
                    (
                        IdentIds::default(),
                        Subs::default(),
                        None, // no expectations for derived module
                        ProcsBase::default(),
                        LayoutCache::new(state.layout_interner.fork(), state.target),
                        ModuleTiming::new(Instant::now()),
                    )
                } else if state.make_specializations_pass.current_pass() == 1 {
                    let found_specializations = state
                        .module_cache
                        .found_specializations
                        .remove(&module_id)
                        .unwrap();

                    let FoundSpecializationsModule {
                        ident_ids,
                        subs,
                        procs_base,
                        layout_cache,
                        module_timing,
                        expectations,
                    } = found_specializations;

                    (
                        ident_ids,
                        subs,
                        expectations,
                        procs_base,
                        layout_cache,
                        module_timing,
                    )
                } else {
                    let LateSpecializationsModule {
                        ident_ids,
                        subs,
                        expectations,
                        module_timing,
                        layout_cache,
                        procs_base,
                    } = state
                        .module_cache
                        .late_specializations
                        .remove(&module_id)
                        .unwrap();

                    (
                        ident_ids,
                        subs,
                        expectations,
                        procs_base,
                        layout_cache,
                        module_timing,
                    )
                };

                if module_id == ModuleId::DERIVED_GEN {
                    load_derived_partial_procs(
                        module_id,
                        arena,
                        &mut subs,
                        &mut ident_ids,
                        &state.derived_module,
                        &mut module_timing,
                        state.target,
                        &state.exposed_types,
                        &mut procs_base,
                        &mut state.world_abilities,
                    );
                }

                let derived_module = SharedDerivedModule::clone(&state.derived_module);

                BuildTask::MakeSpecializations {
                    module_id,
                    ident_ids,
                    subs,
                    procs_base,
                    layout_cache,
                    specializations_we_must_make,
                    module_timing,
                    world_abilities: state.world_abilities.clone_ref(),
                    // TODO: awful, how can we get rid of the clone?
                    exposed_by_module: state.exposed_types.clone(),
                    derived_module,
                    expectations,
                }
            }
        }
    };

    vec![task]
}

/// Values used to render expect output
pub struct ExpectMetadata<'a> {
    pub interns: Interns,
    pub layout_interner: STLayoutInterner<'a>,
    pub expectations: VecMap<ModuleId, Expectations>,
}

type LocExpects = VecMap<Region, Vec<ExpectLookup>>;
type LocDbgs = VecMap<Symbol, DbgLookup>;

/// A message sent out _from_ a worker thread,
/// representing a result of work done, or a request for further work
#[derive(Debug)]
enum Msg<'a> {
    Many(Vec<Msg<'a>>),
    Header(ModuleHeader<'a>),
    Parsed(ParsedModule<'a>),
    CanonicalizedAndConstrained(CanAndCon),
    SolvedTypes {
        module_id: ModuleId,
        ident_ids: IdentIds,
        solved_module: SolvedModule,
        solved_subs: Solved<Subs>,
        decls: Declarations,
        dep_idents: IdentIdsByModule,
        module_timing: ModuleTiming,
        abilities_store: AbilitiesStore,
        loc_expects: LocExpects,
        loc_dbgs: LocDbgs,

        #[cfg(debug_assertions)]
        checkmate: Option<roc_checkmate::Collector>,
    },
    FinishedAllTypeChecking {
        solved_subs: Solved<Subs>,
        exposed_vars_by_symbol: Vec<(Symbol, Variable)>,
        exposed_aliases_by_symbol: MutMap<Symbol, (bool, Alias)>,
        exposed_types_storage: ExposedTypesStorageSubs,
        resolved_implementations: ResolvedImplementations,
        dep_idents: IdentIdsByModule,
        documentation: VecMap<ModuleId, ModuleDocumentation>,
        abilities_store: AbilitiesStore,

        #[cfg(debug_assertions)]
        checkmate: Option<roc_checkmate::Collector>,
    },
    FoundSpecializations {
        module_id: ModuleId,
        ident_ids: IdentIds,
        layout_cache: LayoutCache<'a>,
        procs_base: ProcsBase<'a>,
        solved_subs: Solved<Subs>,
        module_timing: ModuleTiming,
        toplevel_expects: ToplevelExpects,
        expectations: Option<Expectations>,
    },
    MadeSpecializations {
        module_id: ModuleId,
        ident_ids: IdentIds,
        layout_cache: LayoutCache<'a>,
        external_specializations_requested: BumpMap<ModuleId, ExternalSpecializations<'a>>,
        procs_base: ProcsBase<'a>,
        procedures: MutMap<(Symbol, ProcLayout<'a>), Proc<'a>>,
        host_exposed_lambda_sets: HostExposedLambdaSets<'a>,
        update_mode_ids: UpdateModeIds,
        module_timing: ModuleTiming,
        subs: Subs,
        expectations: Option<Expectations>,
    },

    /// The task is to only typecheck AND monomorphize modules
    /// all modules are now monomorphized, we are done
    FinishedAllSpecialization {
        subs: Subs,
        /// The layout interner after all passes in mono are done.
        /// DO NOT use the one on state; that is left in an empty state after specialization is complete!
        layout_interner: STLayoutInterner<'a>,
        exposed_to_host: ExposedToHost,
        module_expectations: VecMap<ModuleId, Expectations>,
    },

    FailedToParse(FileError<'a, SyntaxError<'a>>),
    FailedToReadFile {
        filename: PathBuf,
        error: io::ErrorKind,
    },

    FailedToLoad(LoadingProblem<'a>),
    IncorrectModuleName(FileError<'a, IncorrectModuleName<'a>>),
}

#[derive(Debug)]
struct CanAndCon {
    constrained_module: ConstrainedModule,
    canonicalization_problems: Vec<roc_problem::can::Problem>,
    module_docs: Option<ModuleDocumentation>,
}

#[derive(Debug)]
enum PlatformPath<'a> {
    NotSpecified,
    Valid(To<'a>),
    RootIsModule,
    RootIsHosted,
    RootIsPlatformModule,
}

#[derive(Debug)]
struct PlatformData<'a> {
    module_id: ModuleId,
    provides: &'a [Loc<ExposedName<'a>>],
    is_prebuilt: bool,
}

#[derive(Debug, Clone, Copy)]
enum MakeSpecializationsPass {
    Pass(u8),
}

impl MakeSpecializationsPass {
    fn inc(&mut self) {
        match self {
            &mut Self::Pass(n) => {
                *self = Self::Pass(
                    n.checked_add(1)
                        .expect("way too many specialization passes!"),
                )
            }
        }
    }

    fn current_pass(&self) -> u8 {
        match self {
            MakeSpecializationsPass::Pass(n) => *n,
        }
    }
}

#[derive(Debug)]
struct State<'a> {
    pub root_id: ModuleId,
    pub root_subs: Option<Subs>,
    pub root_path: PathBuf,
    pub root_type: RootType,
    pub cache_dir: PathBuf,
    /// If the root is an app module, the shorthand specified in its header's `to` field
    pub opt_platform_shorthand: Option<&'a str>,
    pub platform_data: Option<PlatformData<'a>>,
    pub exposed_types: ExposedByModule,
    pub platform_path: PlatformPath<'a>,
    pub target: Target,
    pub(self) function_kind: FunctionKind,
    pub fx_mode: FxMode,

    /// Note: only packages and platforms actually expose any modules;
    /// for all others, this will be empty.
    pub exposed_modules: &'a [ModuleId],

    pub module_cache: ModuleCache<'a>,
    pub dependencies: Dependencies<'a>,
    pub procedures: MutMap<(Symbol, ProcLayout<'a>), Proc<'a>>,
    pub host_exposed_lambda_sets: HostExposedLambdaSets<'a>,
    pub toplevel_expects: MutMap<ModuleId, ToplevelExpects>,
    pub exposed_to_host: ExposedToHost,

    /// This is the "final" list of IdentIds, after canonicalization and constraint gen
    /// have completed for a given module.
    pub constrained_ident_ids: IdentIdsByModule,

    /// From now on, these will be used by multiple threads; time to make an Arc<Mutex<_>>!
    pub arc_modules: Arc<Mutex<PackageModuleIds<'a>>>,
    pub arc_shorthands: Arc<Mutex<MutMap<&'a str, ShorthandPath>>>,
    pub derived_module: SharedDerivedModule,

    pub ident_ids_by_module: SharedIdentIdsByModule,

    pub declarations_by_id: MutMap<ModuleId, Declarations>,

    pub exposed_symbols_by_module: MutMap<ModuleId, VecSet<Symbol>>,

    pub timings: MutMap<ModuleId, ModuleTiming>,

    // Each thread gets its own layout cache. When one "pending specializations"
    // pass completes, it returns its layout cache so another thread can use it.
    // We don't bother trying to union them all together to maximize cache hits,
    // since the unioning process could potentially take longer than the savings.
    // (Granted, this has not been attempted or measured!)
    pub layout_caches: std::vec::Vec<LayoutCache<'a>>,

    pub render: RenderTarget,
    pub palette: Palette,
    pub exec_mode: ExecutionMode,

    /// All abilities across all modules.
    pub world_abilities: WorldAbilities,

    make_specializations_pass: MakeSpecializationsPass,

    // cached types (used for builtin modules, could include packages in the future too)
    cached_types: CachedTypeState,

    layout_interner: GlobalLayoutInterner<'a>,
}

type CachedTypeState = Arc<Mutex<MutMap<ModuleId, TypeState>>>;

impl<'a> State<'a> {
    fn goal_phase(&self) -> Phase {
        self.exec_mode.goal_phase()
    }

    fn new(
        root_id: ModuleId,
        root_path: PathBuf,
        root_type: RootType,
        opt_platform_shorthand: Option<&'a str>,
        target: Target,
        function_kind: FunctionKind,
        exposed_types: ExposedByModule,
        arc_modules: Arc<Mutex<PackageModuleIds<'a>>>,
        ident_ids_by_module: SharedIdentIdsByModule,
        arc_shorthands: Arc<Mutex<MutMap<&'a str, ShorthandPath>>>,
        cached_types: MutMap<ModuleId, TypeState>,
        render: RenderTarget,
        palette: Palette,
        number_of_workers: usize,
        exec_mode: ExecutionMode,
    ) -> Self {
        let cache_dir = roc_packaging::cache::roc_cache_packages_dir();
        let dependencies = Dependencies::new(exec_mode.goal_phase());

        Self {
            root_id,
            root_path,
            root_subs: None,
            root_type,
            opt_platform_shorthand,
            cache_dir,
            target,
            function_kind,
            fx_mode: FxMode::Task,
            platform_data: None,
            platform_path: PlatformPath::NotSpecified,
            module_cache: ModuleCache::default(),
            dependencies,
            procedures: MutMap::default(),
            host_exposed_lambda_sets: std::vec::Vec::new(),
            toplevel_expects: MutMap::default(),
            exposed_to_host: ExposedToHost::default(),
            exposed_modules: &[],
            exposed_types,
            arc_modules,
            arc_shorthands,
            derived_module: Default::default(),
            constrained_ident_ids: IdentIds::exposed_builtins(0),
            ident_ids_by_module,
            declarations_by_id: MutMap::default(),
            exposed_symbols_by_module: MutMap::default(),
            timings: MutMap::default(),
            layout_caches: std::vec::Vec::with_capacity(number_of_workers),
            cached_types: Arc::new(Mutex::new(cached_types)),
            render,
            palette,
            exec_mode,
            make_specializations_pass: MakeSpecializationsPass::Pass(1),
            world_abilities: Default::default(),
            layout_interner: GlobalLayoutInterner::with_capacity(128, target),
        }
    }
}

fn report_timing(
    buf: &mut impl std::fmt::Write,
    label: &str,
    duration: Duration,
) -> std::fmt::Result {
    writeln!(
        buf,
        "        {:9.3} ms   {}",
        duration.as_secs_f64() * 1000.0,
        label,
    )
}

impl std::fmt::Display for ModuleTiming {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        let module_timing = self;

        report_timing(f, "Read .roc file from disk", module_timing.read_roc_file)?;
        report_timing(f, "Parse header", module_timing.parse_header)?;
        report_timing(f, "Parse body", module_timing.parse_body)?;
        report_timing(f, "Canonicalize", module_timing.canonicalize)?;
        report_timing(f, "Constrain", module_timing.constrain)?;
        report_timing(f, "Solve", module_timing.solve)?;
        report_timing(
            f,
            "Find Specializations",
            module_timing.find_specializations,
        )?;
        let multiple_make_specializations_passes = module_timing.make_specializations.len() > 1;
        for (i, pass_time) in module_timing.make_specializations.iter().enumerate() {
            let suffix = if multiple_make_specializations_passes {
                format!(" (Pass {i})")
            } else {
                String::new()
            };
            report_timing(f, &format!("Make Specializations{suffix}"), *pass_time)?;
        }
        report_timing(f, "Other", module_timing.other())?;
        f.write_str("\n")?;
        report_timing(f, "Total", module_timing.total())?;

        Ok(())
    }
}

/// A message sent _to_ a worker thread, describing the work to be done
#[derive(Debug)]
#[allow(dead_code)]
enum BuildTask<'a> {
    LoadModule {
        module_name: PQModuleName<'a>,
        module_ids: Arc<Mutex<PackageModuleIds<'a>>>,
        shorthands: Arc<Mutex<MutMap<&'a str, ShorthandPath>>>,
        ident_ids_by_module: SharedIdentIdsByModule,
    },
    Parse {
        header: ModuleHeader<'a>,
        arc_shorthands: Arc<Mutex<MutMap<&'a str, ShorthandPath>>>,
        module_ids: Arc<Mutex<PackageModuleIds<'a>>>,
        ident_ids_by_module: SharedIdentIdsByModule,
        root_type: RootType,
    },
    CanonicalizeAndConstrain {
        parsed: ParsedModule<'a>,
        qualified_module_ids: PackageModuleIds<'a>,
        dep_idents: IdentIdsByModule,
        exposed_symbols: VecSet<Symbol>,
        aliases: MutMap<Symbol, Alias>,
        abilities_store: PendingAbilitiesStore,
        exposed_module_ids: &'a [ModuleId],
        skip_constraint_gen: bool,
        exec_mode: ExecutionMode,
        imported_module_params: VecMap<ModuleId, ModuleParams>,
        fx_mode: FxMode,
    },
    Solve {
        module: Module,
        ident_ids: IdentIds,
        exposed_for_module: ExposedForModule,
        module_timing: ModuleTiming,
        types: Types,
        constraints: Constraints,
        constraint: ConstraintSoa,
        function_kind: FunctionKind,
        pending_derives: PendingDerives,
        var_store: VarStore,
        declarations: Declarations,
        dep_idents: IdentIdsByModule,
        cached_subs: CachedTypeState,
        derived_module: SharedDerivedModule,
        exec_mode: ExecutionMode,
        is_host_exposed: bool,

        #[cfg(debug_assertions)]
        checkmate: Option<roc_checkmate::Collector>,
    },
    BuildPendingSpecializations {
        module_timing: ModuleTiming,
        layout_cache: LayoutCache<'a>,
        solved_subs: Solved<Subs>,
        imported_module_thunks: &'a [Symbol],
        module_id: ModuleId,
        ident_ids: IdentIds,
        decls: Declarations,
        exposed_to_host: ExposedToHost,
        exposed_by_module: ExposedByModule,
        world_abilities: WorldAbilities,
        derived_module: SharedDerivedModule,
        expectations: Option<Expectations>,
        build_expects: bool,
    },
    MakeSpecializations {
        module_id: ModuleId,
        ident_ids: IdentIds,
        subs: Subs,
        procs_base: ProcsBase<'a>,
        layout_cache: LayoutCache<'a>,
        specializations_we_must_make: Vec<ExternalSpecializations<'a>>,
        module_timing: ModuleTiming,
        exposed_by_module: ExposedByModule,
        world_abilities: WorldAbilities,
        derived_module: SharedDerivedModule,
        expectations: Option<Expectations>,
    },
}

#[derive(Debug)]
pub struct IncorrectModuleName<'a> {
    pub module_id: ModuleId,
    pub found: Loc<PQModuleName<'a>>,
    pub expected: PQModuleName<'a>,
}

#[derive(Debug)]
pub enum LoadingProblem<'a> {
    FileProblem {
        filename: PathBuf,
        error: io::ErrorKind,
    },
    ParsingFailed(FileError<'a, SyntaxError<'a>>),
    UnexpectedHeader(String),
    MultiplePlatformPackages {
        filename: PathBuf,
        module_id: ModuleId,
        source: &'a [u8],
        region: Region,
    },
    NoPlatformPackage {
        filename: PathBuf,
        module_id: ModuleId,
        source: &'a [u8],
        region: Region,
    },
    UnrecognizedPackageShorthand {
        filename: PathBuf,
        module_id: ModuleId,
        source: &'a str,
        region: Region,
        shorthand: &'a str,
        available: AvailableShorthands<'a>,
    },

    ErrJoiningWorkerThreads,
    TriedToImportAppModule,

    /// a formatted report
    FormattedReport(String),

    ImportCycle(PathBuf, Vec<ModuleId>),
    IncorrectModuleName(FileError<'a, IncorrectModuleName<'a>>),
    CouldNotFindCacheDir,
    ChannelProblem(ChannelProblem),
}

#[derive(Debug)]
pub enum AvailableShorthands<'a> {
    FromRoot(Vec<&'a str>),
    FromMain(PathBuf, Vec<&'a str>),
    UnknownMain,
}

impl<'a> AvailableShorthands<'a> {
    fn new(root_type: RootType, shorthands: Vec<&'a str>) -> Self {
        match root_type {
            RootType::Main => AvailableShorthands::FromRoot(shorthands),
            RootType::Module { main_path } => match main_path {
                None => AvailableShorthands::UnknownMain,
                Some(main_path) => AvailableShorthands::FromMain(main_path, shorthands),
            },
        }
    }
}

pub enum Phases {
    /// Parse, canonicalize, check types
    TypeCheck,
    /// Parse, canonicalize, check types, monomorphize
    Monomorphize,
}

type MsgSender<'a> = Sender<Msg<'a>>;

/// Add a task to the queue, and notify all the listeners.
fn enqueue_task<'a>(
    injector: &Injector<BuildTask<'a>>,
    listeners: &[Sender<()>],
    task: BuildTask<'a>,
) -> Result<(), LoadingProblem<'a>> {
    injector.push(task);

    for listener in listeners {
        listener
            .send(())
            .map_err(|_| LoadingProblem::ChannelProblem(ChannelProblem::FailedToEnqueueTask))?;
    }

    Ok(())
}

pub fn load_and_typecheck_str<'a>(
    arena: &'a Bump,
    filename: PathBuf,
    source: &'a str,
    src_dir: PathBuf,
    opt_main_path: Option<PathBuf>,
    exposed_types: ExposedByModule,
    target: Target,
    function_kind: FunctionKind,
    render: RenderTarget,
    palette: Palette,
    roc_cache_dir: RocCacheDir<'_>,
    threading: Threading,
) -> Result<LoadedModule, LoadingProblem<'a>> {
    use LoadResult::*;

    let load_start = LoadStart::from_str(
        arena,
        filename,
        opt_main_path,
        source,
        roc_cache_dir,
        src_dir,
    )?;

    // this function is used specifically in the case
    // where we want to regenerate the cached data
    let cached_subs = MutMap::default();

    let load_config = LoadConfig {
        target,
        render,
        palette,
        threading,
        exec_mode: ExecutionMode::Check,
        function_kind,
    };

    match load(
        arena,
        load_start,
        exposed_types,
        cached_subs,
        roc_cache_dir,
        load_config,
    )? {
        Monomorphized(_) => unreachable!(),
        TypeChecked(module) => Ok(module),
    }
}

#[derive(Clone, Copy)]
pub enum PrintTarget {
    ColorTerminal,
    Generic,
}

pub struct LoadStart<'a> {
    arc_modules: Arc<Mutex<PackageModuleIds<'a>>>,
    ident_ids_by_module: SharedIdentIdsByModule,
    arc_shorthands: Arc<Mutex<MutMap<&'a str, ShorthandPath>>>,
    root_id: ModuleId,
    root_path: PathBuf,
    root_msg: Msg<'a>,
    root_type: RootType,
    opt_platform_shorthand: Option<&'a str>,
    src_dir: PathBuf,
}

#[derive(Debug, Clone)]
enum RootType {
    Main,
    Module { main_path: Option<PathBuf> },
}

impl<'a> LoadStart<'a> {
    pub fn from_path(
        arena: &'a Bump,
        filename: PathBuf,
        opt_main_path: Option<PathBuf>,
        render: RenderTarget,
        roc_cache_dir: RocCacheDir<'_>,
        palette: Palette,
    ) -> Result<Self, LoadingProblem<'a>> {
        let arc_modules = Arc::new(Mutex::new(PackageModuleIds::default()));
        let arc_shorthands = Arc::new(Mutex::new(MutMap::default()));
        let root_exposed_ident_ids = IdentIds::exposed_builtins(0);
        let ident_ids_by_module = Arc::new(Mutex::new(root_exposed_ident_ids));
        let mut src_dir = filename.parent().unwrap().to_path_buf();

        // Load the root module synchronously; we can't proceed until we have its id.
        let root_start_time = Instant::now();

        let load_result = load_filename(
            arena,
            filename.clone(),
            true,
            None,
            None,
            Arc::clone(&arc_modules),
            Arc::clone(&ident_ids_by_module),
            roc_cache_dir,
            root_start_time,
        );

        let load_result = match load_result {
            Ok(header_output) => handle_root_type(
                arena,
                Arc::clone(&arc_shorthands),
                Arc::clone(&arc_modules),
                Arc::clone(&ident_ids_by_module),
                roc_cache_dir,
                header_output,
                opt_main_path,
                &mut src_dir,
            ),
            Err(problem) => Err(problem),
        };

        let (header_output, root_type) = match load_result {
            Ok(header_output) => header_output,

            Err(problem) => {
                let module_ids = Arc::try_unwrap(arc_modules)
                    .unwrap_or_else(|_| {
                        panic!("There were still outstanding Arc references to module_ids")
                    })
                    .into_inner()
                    .into_module_ids();

                let report = report_loading_problem(problem, module_ids, render, palette);

                // TODO try to gracefully recover and continue
                // instead of changing the control flow to exit.
                return Err(LoadingProblem::FormattedReport(report));
            }
        };

        Ok(LoadStart {
            arc_modules,
            ident_ids_by_module,
            arc_shorthands,
            src_dir,
            root_id: header_output.module_id,
            root_path: filename,
            root_msg: header_output.msg,
            root_type,
            opt_platform_shorthand: header_output.opt_platform_shorthand,
        })
    }

    pub fn from_str(
        arena: &'a Bump,
        filename: PathBuf,
        opt_main_path: Option<PathBuf>,
        src: &'a str,
        roc_cache_dir: RocCacheDir<'_>,
        mut src_dir: PathBuf,
    ) -> Result<Self, LoadingProblem<'a>> {
        let arc_modules = Arc::new(Mutex::new(PackageModuleIds::default()));
        let arc_shorthands = Arc::new(Mutex::new(MutMap::default()));
        let root_exposed_ident_ids = IdentIds::exposed_builtins(0);
        let ident_ids_by_module = Arc::new(Mutex::new(root_exposed_ident_ids));

        // Load the root module synchronously; we can't proceed until we have its id.
        let root_start_time = Instant::now();

        let header_output = load_from_str(
            arena,
            filename.clone(),
            src,
            Arc::clone(&arc_modules),
            Arc::clone(&ident_ids_by_module),
            roc_cache_dir,
            root_start_time,
        )?;

        let (header_output, root_type) = handle_root_type(
            arena,
            Arc::clone(&arc_shorthands),
            Arc::clone(&arc_modules),
            Arc::clone(&ident_ids_by_module),
            roc_cache_dir,
            header_output,
            opt_main_path,
            &mut src_dir,
        )?;

        let HeaderOutput {
            module_id: root_id,
            msg: root_msg,
            opt_platform_shorthand: opt_platform_id,
        } = header_output;

        Ok(LoadStart {
            arc_modules,
            arc_shorthands,
            src_dir,
            ident_ids_by_module,
            root_id,
            root_path: filename,
            root_msg,
            root_type,
            opt_platform_shorthand: opt_platform_id,
        })
    }
}

fn handle_root_type<'a>(
    arena: &'a Bump,
    arc_shorthands: Arc<Mutex<MutMap<&'a str, ShorthandPath>>>,
    arc_modules: Arc<Mutex<PackageModuleIds<'a>>>,
    ident_ids_by_module: SharedIdentIdsByModule,
    roc_cache_dir: RocCacheDir<'_>,
    mut header_output: HeaderOutput<'a>,
    opt_main_path: Option<PathBuf>,
    src_dir: &mut PathBuf,
) -> Result<(HeaderOutput<'a>, RootType), LoadingProblem<'a>> {
    if let Msg::Header(ModuleHeader {
        module_id: header_id,
        header_type,
        ..
    }) = &header_output.msg
    {
        debug_assert_eq!(*header_id, header_output.module_id);

        use HeaderType::*;

        match header_type {
            Module { .. } | Builtin { .. } | Hosted { .. } => {
                let main_path = opt_main_path.or_else(|| find_main_roc_recursively(src_dir));

                let cache_dir = roc_cache_dir.as_persistent_path();

                if let (Some(main_path), Some(cache_dir)) = (main_path.clone(), cache_dir) {
                    let mut messages = Vec::with_capacity(4);
                    messages.push(header_output.msg);

                    load_packages_from_main(
                        arena,
                        src_dir.clone(),
                        main_path,
                        &mut messages,
                        Arc::clone(&arc_modules),
                        Arc::clone(&ident_ids_by_module),
                        Arc::clone(&arc_shorthands),
                        cache_dir,
                    )?;

                    header_output.msg = Msg::Many(messages);
                }

                Ok((header_output, RootType::Module { main_path }))
            }
            App { .. } | Package { .. } | Platform { .. } => Ok((header_output, RootType::Main)),
        }
    } else {
        Ok((header_output, RootType::Main))
    }
}

fn find_main_roc_recursively(src_dir: &mut PathBuf) -> Option<PathBuf> {
    let original_src_dir = src_dir.clone();

    loop {
        match src_dir.join(DEFAULT_MAIN_NAME).canonicalize() {
            Ok(main_roc) => break Some(main_roc),
            Err(_) => {
                if !src_dir.pop() {
                    // reached the root, no main.roc found
                    *src_dir = original_src_dir;
                    break None;
                }
            }
        }
    }
}

fn load_packages_from_main<'a>(
    arena: &'a Bump,
    src_dir: PathBuf,
    filename: PathBuf,
    messages: &mut Vec<Msg<'a>>,
    module_ids: Arc<Mutex<PackageModuleIds<'a>>>,
    ident_ids_by_module: SharedIdentIdsByModule,
    arc_shorthands: Arc<Mutex<MutMap<&'a str, ShorthandPath>>>,
    cache_dir: &Path,
) -> Result<(), LoadingProblem<'a>> {
    let src_bytes = fs::read(&filename).map_err(|err| LoadingProblem::FileProblem {
        filename: filename.clone(),
        error: err.kind(),
    })?;

    let parse_state = roc_parse::state::State::new(arena.alloc(src_bytes));

    let (parsed_header, _) =
        roc_parse::header::parse_header(arena, parse_state.clone()).map_err(|fail| {
            LoadingProblem::ParsingFailed(
                fail.map_problem(SyntaxError::Header)
                    .into_file_error(filename.clone()),
            )
        })?;

    use ast::Header::*;

    let packages = match parsed_header.item {
        App(AppHeader { packages, .. }) | Package(PackageHeader { packages, .. }) => {
            unspace(arena, packages.value.items)
        }
        Platform(PlatformHeader { packages, .. }) => unspace(arena, packages.item.items),
        Module(_) | Hosted(_) => todo!("expected {} to be an app or package", filename.display()),
    };

    load_packages(
        packages,
        messages,
        RocCacheDir::Persistent(cache_dir),
        src_dir.clone(),
        arena,
        None,
        module_ids,
        ident_ids_by_module,
        filename.clone(),
    );

    let package_entries = packages
        .iter()
        .map(|Loc { value: pkg, .. }| (pkg.shorthand, pkg.package_name.value))
        .collect::<MutMap<_, _>>();

    let mut shorthands = arc_shorthands.lock();

    register_package_shorthands(
        &mut shorthands,
        &package_entries,
        &filename,
        &src_dir,
        cache_dir,
    )
}

pub enum LoadResult<'a> {
    TypeChecked(LoadedModule),
    Monomorphized(MonomorphizedModule<'a>),
}

#[derive(Copy, Clone, Debug, PartialEq, Eq)]
pub enum Threading {
    Single,
    AllAvailable,
    AtMost(usize),
}

/// The loading process works like this, starting from the given filename (e.g. "main.roc"):
///
/// 1. Open the file.
/// 2. Parse the module's header.
/// 3. For each of its imports, send a message on the channel to the coordinator thread, which
///    will repeat this process to load that module - starting with step 1.
/// 4. Add everything we were able to import unqualified to the module's default scope.
/// 5. Parse the module's defs.
/// 6. Canonicalize the module.
/// 7. Before type checking, block on waiting for type checking to complete on all imports.
///    (Since Roc doesn't allow cyclic dependencies, this cannot deadlock.)
/// 8. Type check the module and create type annotations for its top-level declarations.
/// 9. Report the completed type annotation to the coordinator thread, so other modules
///    that are waiting in step 7 can unblock.
///
/// The loaded_modules argument specifies which modules have already been loaded.
/// It typically contains *at least* the standard modules, but is empty when loading
/// the standard modules themselves.
///
/// If we're just type-checking everything (e.g. running `roc check` at the command line),
/// we can stop there. However, if we're generating code, then there are additional steps.
///
/// 10. After reporting the completed type annotation, we have all the information necessary
///     to monomorphize. However, since we want to monomorphize in parallel without
///     duplicating work, we do monomorphization in two steps. First, we go through and
///     determine all the specializations this module *wants*. We compute the hashes
///     and report them to the coordinator thread, along with the mono::expr::Expr values of
///     the current function's body. At this point, we have not yet begun to assemble Procs;
///     all we've done is send a list of requested specializations to the coordinator.
/// 11. The coordinator works through the specialization requests in parallel, adding them
///     to a global map once they're finished. Performing one specialization may result
///     in requests for others; these are added to the queue and worked through as normal.
///     This process continues until *both* all modules have reported that they've finished
///     adding specialization requests to the queue, *and* the queue is empty (including
///     of any requests that were added in the course of completing other requests). Now
///     we have a map of specializations, and everything was assembled in parallel with
///     no unique specialization ever getting assembled twice (meaning no wasted effort).
///
///     a. Note that this might mean that we have to specialize certain modules multiple times.
///        When might this happen? Well, abilities can introduce implicit edges in the dependency
///        graph, and even cycles. For example, suppose module Ab provides "ability1" and a function
///        "f" that uses "ability1", and module App implements "ability1" and calls "f" with the
///        implementing type. Then the specialization of "Ab#f" depends on the specialization of
///        "ability1" back in the App module.
/// 12. Now that we have our final map of specializations, we can proceed to code gen!
///     As long as the specializations are stored in a per-ModuleId map, we can also
///     parallelize this code gen. (e.g. in dev builds, building separate LLVM modules
///     and then linking them together, and possibly caching them by the hash of their
///     specializations, so if none of their specializations changed, we don't even need
///     to rebuild the module and can link in the cached one directly.)
pub fn load<'a>(
    arena: &'a Bump,
    load_start: LoadStart<'a>,
    exposed_types: ExposedByModule,
    cached_types: MutMap<ModuleId, TypeState>,
    roc_cache_dir: RocCacheDir<'_>,
    load_config: LoadConfig,
) -> Result<LoadResult<'a>, LoadingProblem<'a>> {
    enum Threads {
        Single,

        #[allow(dead_code)]
        Many(usize),
    }

    let threads = {
        if cfg!(target_family = "wasm") {
            // When compiling to wasm, we cannot spawn extra threads
            // so we have a single-threaded implementation
            Threads::Single
        } else {
            match std::thread::available_parallelism().map(|v| v.get()) {
                Err(_) => Threads::Single,
                Ok(0) => unreachable!("NonZeroUsize"),
                Ok(1) => Threads::Single,
                Ok(reported) => match load_config.threading {
                    Threading::Single => Threads::Single,
                    Threading::AllAvailable => Threads::Many(reported),
                    Threading::AtMost(at_most) => Threads::Many(Ord::min(reported, at_most)),
                },
            }
        }
    };

    match threads {
        Threads::Single => load_single_threaded(
            arena,
            load_start,
            exposed_types,
            load_config.target,
            load_config.function_kind,
            cached_types,
            load_config.render,
            load_config.palette,
            load_config.exec_mode,
            roc_cache_dir,
        ),
        Threads::Many(threads) => load_multi_threaded(
            arena,
            load_start,
            exposed_types,
            load_config.target,
            load_config.function_kind,
            cached_types,
            load_config.render,
            load_config.palette,
            threads,
            load_config.exec_mode,
            roc_cache_dir,
        ),
    }
}

/// Load using only a single thread; used when compiling to webassembly
pub fn load_single_threaded<'a>(
    arena: &'a Bump,
    load_start: LoadStart<'a>,
    exposed_types: ExposedByModule,
    target: Target,
    function_kind: FunctionKind,
    cached_types: MutMap<ModuleId, TypeState>,
    render: RenderTarget,
    palette: Palette,
    exec_mode: ExecutionMode,
    roc_cache_dir: RocCacheDir<'_>,
) -> Result<LoadResult<'a>, LoadingProblem<'a>> {
    let LoadStart {
        arc_modules,
        ident_ids_by_module,
        arc_shorthands,
        root_id,
        root_path,
        root_msg,
        root_type,
        src_dir,
        opt_platform_shorthand,
        ..
    } = load_start;

    let (msg_tx, msg_rx) = bounded(1024);

    msg_tx
        .send(root_msg)
        .map_err(|_| LoadingProblem::ChannelProblem(ChannelProblem::FailedToSendRootMsg))?;

    let number_of_workers = 1;
    let mut state = State::new(
        root_id,
        root_path,
        root_type,
        opt_platform_shorthand,
        target,
        function_kind,
        exposed_types,
        arc_modules,
        ident_ids_by_module,
        arc_shorthands,
        cached_types,
        render,
        palette,
        number_of_workers,
        exec_mode,
    );

    // We'll add tasks to this, and then worker threads will take tasks from it.
    let injector = Injector::new();

    let (worker_wakup_tx, worker_wakup_rx) = bounded(1024);
    let worker_waker = worker_wakup_tx;
    let worker_wakers = [worker_waker];

    let worker = Worker::new_fifo();
    let stealer = worker.stealer();
    let stealers = &[stealer];

    // now we just manually interleave stepping the state "thread" and the worker "thread"
    loop {
        match state_thread_step(arena, state, &worker_wakers, &injector, &msg_tx, &msg_rx) {
            Ok(ControlFlow::Break(done)) => return Ok(done),
            Ok(ControlFlow::Continue(new_state)) => {
                state = new_state;
            }
            Err(e) => return Err(e),
        }

        // then check if the worker can step
        let control_flow =
            roc_worker::worker_task_step(&worker, &injector, stealers, &worker_wakup_rx, |task| {
                run_task(task, arena, &src_dir, msg_tx.clone(), roc_cache_dir, target)
            });

        match control_flow {
            Ok(ControlFlow::Break(())) => panic!("the worker should not break!"),
            Ok(ControlFlow::Continue(())) => {
                // progress was made
            }
            Err(e) => return Err(LoadingProblem::ChannelProblem(e)),
        }
    }
}

fn state_thread_step<'a>(
    arena: &'a Bump,
    state: State<'a>,
    worker_wakers: &[Sender<()>],
    injector: &Injector<BuildTask<'a>>,
    msg_tx: &crossbeam::channel::Sender<Msg<'a>>,
    msg_rx: &crossbeam::channel::Receiver<Msg<'a>>,
) -> Result<ControlFlow<LoadResult<'a>, State<'a>>, LoadingProblem<'a>> {
    match msg_rx.try_recv() {
        Ok(msg) => {
            match msg {
                Msg::FinishedAllTypeChecking {
                    solved_subs,
                    exposed_vars_by_symbol,
                    exposed_aliases_by_symbol,
                    exposed_types_storage,
                    resolved_implementations,
                    dep_idents,
                    documentation,
                    abilities_store,

                    #[cfg(debug_assertions)]
                    checkmate,
                } => {
                    // We're done! There should be no more messages pending.
                    debug_assert!(msg_rx.is_empty());

                    let exposed_aliases_by_symbol = exposed_aliases_by_symbol
                        .into_iter()
                        .map(|(k, (_, v))| (k, v))
                        .collect();

                    let typechecked = finish(
                        state,
                        solved_subs,
                        exposed_aliases_by_symbol,
                        exposed_vars_by_symbol,
                        exposed_types_storage,
                        resolved_implementations,
                        dep_idents,
                        documentation,
                        abilities_store,
                        //
                        #[cfg(debug_assertions)]
                        checkmate,
                    );

                    Ok(ControlFlow::Break(LoadResult::TypeChecked(typechecked)))
                }
                Msg::FinishedAllSpecialization {
                    subs,
                    layout_interner,
                    exposed_to_host,
                    module_expectations,
                } => {
                    // We're done! There should be no more messages pending.
                    debug_assert!(msg_rx.is_empty());

                    let monomorphized = finish_specialization(
                        arena,
                        state,
                        subs,
                        layout_interner,
                        exposed_to_host,
                        module_expectations,
                    )?;

                    Ok(ControlFlow::Break(LoadResult::Monomorphized(monomorphized)))
                }
                Msg::FailedToReadFile { filename, error } => {
                    let buf = to_file_problem_report_string(filename, error, true);
                    Err(LoadingProblem::FormattedReport(buf))
                }

                Msg::FailedToParse(problem) => {
                    let module_ids = (*state.arc_modules).lock().clone().into_module_ids();
                    let buf = to_parse_problem_report(
                        problem,
                        module_ids,
                        state.constrained_ident_ids,
                        state.render,
                        state.palette,
                    );
                    Err(LoadingProblem::FormattedReport(buf))
                }
                Msg::IncorrectModuleName(FileError {
                    problem: SourceError { problem, bytes },
                    filename,
                }) => {
                    let module_ids = (*state.arc_modules).lock().clone().into_module_ids();
                    let buf = to_incorrect_module_name_report(
                        module_ids,
                        state.constrained_ident_ids,
                        problem,
                        filename,
                        bytes,
                        state.render,
                    );
                    Err(LoadingProblem::FormattedReport(buf))
                }
                msg => {
                    // This is where most of the main thread's work gets done.
                    // Everything up to this point has been setting up the threading
                    // system which lets this logic work efficiently.
                    let arc_modules = state.arc_modules.clone();

                    let render = state.render;
                    let palette = state.palette;

                    let res_state =
                        update(state, msg, msg_tx.clone(), injector, worker_wakers, arena);

                    match res_state {
                        Ok(new_state) => Ok(ControlFlow::Continue(new_state)),
                        Err(LoadingProblem::ParsingFailed(problem)) => {
                            let module_ids = Arc::try_unwrap(arc_modules)
                                .unwrap_or_else(|_| {
                                    panic!(
                                        r"There were still outstanding Arc references to module_ids"
                                    )
                                })
                                .into_inner()
                                .into_module_ids();

                            // if parsing failed, this module did not add anything to IdentIds
                            let root_exposed_ident_ids = IdentIds::exposed_builtins(0);
                            let buf = to_parse_problem_report(
                                problem,
                                module_ids,
                                root_exposed_ident_ids,
                                render,
                                palette,
                            );
                            Err(LoadingProblem::FormattedReport(buf))
                        }
                        Err(LoadingProblem::ImportCycle(filename, cycle)) => {
                            let module_ids = arc_modules.lock().clone().into_module_ids();

                            let root_exposed_ident_ids = IdentIds::exposed_builtins(0);
                            let buf = to_import_cycle_report(
                                module_ids,
                                root_exposed_ident_ids,
                                cycle,
                                filename,
                                render,
                            );
                            return Err(LoadingProblem::FormattedReport(buf));
                        }
                        Err(LoadingProblem::IncorrectModuleName(FileError {
                            problem: SourceError { problem, bytes },
                            filename,
                        })) => {
                            let module_ids = arc_modules.lock().clone().into_module_ids();

                            let root_exposed_ident_ids = IdentIds::exposed_builtins(0);
                            let buf = to_incorrect_module_name_report(
                                module_ids,
                                root_exposed_ident_ids,
                                problem,
                                filename,
                                bytes,
                                render,
                            );
                            return Err(LoadingProblem::FormattedReport(buf));
                        }
                        Err(LoadingProblem::UnrecognizedPackageShorthand {
                            filename,
                            module_id,
                            source,
                            region,
                            shorthand,
                            available,
                        }) => {
                            let module_ids = arc_modules.lock().clone().into_module_ids();

                            let root_exposed_ident_ids = IdentIds::exposed_builtins(0);
                            let buf = to_unrecognized_package_shorthand_report(
                                module_ids,
                                root_exposed_ident_ids,
                                module_id,
                                filename,
                                region,
                                source,
                                shorthand,
                                available,
                                render,
                            );
                            return Err(LoadingProblem::FormattedReport(buf));
                        }
                        Err(e) => Err(e),
                    }
                }
            }
        }
        Err(err) => match err {
            crossbeam::channel::TryRecvError::Empty => Ok(ControlFlow::Continue(state)),
            crossbeam::channel::TryRecvError::Disconnected => Err(LoadingProblem::ChannelProblem(
                ChannelProblem::ChannelDisconnected,
            )),
        },
    }
}

pub fn report_loading_problem(
    problem: LoadingProblem<'_>,
    module_ids: ModuleIds,
    render: RenderTarget,
    palette: Palette,
) -> String {
    match problem {
        LoadingProblem::ParsingFailed(problem) => {
            // if parsing failed, this module did not add anything to IdentIds
            let root_exposed_ident_ids = IdentIds::exposed_builtins(0);

            to_parse_problem_report(problem, module_ids, root_exposed_ident_ids, render, palette)
        }
        LoadingProblem::ImportCycle(filename, cycle) => {
            let root_exposed_ident_ids = IdentIds::exposed_builtins(0);

            to_import_cycle_report(module_ids, root_exposed_ident_ids, cycle, filename, render)
        }
        LoadingProblem::IncorrectModuleName(FileError {
            problem: SourceError { problem, bytes },
            filename,
        }) => {
            let root_exposed_ident_ids = IdentIds::exposed_builtins(0);

            to_incorrect_module_name_report(
                module_ids,
                root_exposed_ident_ids,
                problem,
                filename,
                bytes,
                render,
            )
        }
        LoadingProblem::FormattedReport(report) => report,
        LoadingProblem::FileProblem { filename, error } => {
            to_file_problem_report_string(filename, error, true)
        }
        LoadingProblem::NoPlatformPackage {
            filename,
            module_id,
            source,
            region,
        } => to_no_platform_package_report(
            module_ids,
            IdentIds::exposed_builtins(0),
            module_id,
            filename,
            region,
            source,
            render,
        ),
        LoadingProblem::MultiplePlatformPackages {
            filename,
            module_id,
            source,
            region,
        } => to_multiple_platform_packages_report(
            module_ids,
            IdentIds::exposed_builtins(0),
            module_id,
            filename,
            region,
            source,
            render,
        ),
        LoadingProblem::UnrecognizedPackageShorthand {
            filename,
            module_id,
            region,
            source,
            shorthand,
            available,
        } => to_unrecognized_package_shorthand_report(
            module_ids,
            IdentIds::exposed_builtins(0),
            module_id,
            filename,
            region,
            source,
            shorthand,
            available,
            render,
        ),
        err => todo!("Loading error: {:?}", err),
    }
}

fn load_multi_threaded<'a>(
    arena: &'a Bump,
    load_start: LoadStart<'a>,
    exposed_types: ExposedByModule,
    target: Target,
    function_kind: FunctionKind,
    cached_types: MutMap<ModuleId, TypeState>,
    render: RenderTarget,
    palette: Palette,
    available_threads: usize,
    exec_mode: ExecutionMode,
    roc_cache_dir: RocCacheDir<'_>,
) -> Result<LoadResult<'a>, LoadingProblem<'a>> {
    let LoadStart {
        arc_modules,
        ident_ids_by_module,
        arc_shorthands,
        root_id,
        root_path,
        root_msg,
        root_type,
        src_dir,
        opt_platform_shorthand,
        ..
    } = load_start;

    let (msg_tx, msg_rx) = bounded(1024);
    msg_tx
        .send(root_msg)
        .map_err(|_| LoadingProblem::ChannelProblem(ChannelProblem::FailedToSendRootMsg))?;

    // Reserve one CPU for the main thread, and let all the others be eligible
    // to spawn workers.
    let available_workers = available_threads - 1;

    let num_workers = match env::var("ROC_NUM_WORKERS") {
        Ok(env_str) => env_str
            .parse::<usize>()
            .unwrap_or(available_workers)
            .min(available_workers),
        Err(_) => available_workers,
    };

    assert!(
        num_workers >= 1,
        "`load_multi_threaded` needs at least one worker"
    );

    let mut state = State::new(
        root_id,
        root_path,
        root_type,
        opt_platform_shorthand,
        target,
        function_kind,
        exposed_types,
        arc_modules,
        ident_ids_by_module,
        arc_shorthands,
        cached_types,
        render,
        palette,
        num_workers,
        exec_mode,
    );

    // an arena for every worker, stored in an arena-allocated bumpalo vec to make the lifetimes work
    let arenas = std::iter::repeat_with(Bump::new).take(num_workers);
    let worker_arenas = arena.alloc(bumpalo::collections::Vec::from_iter_in(arenas, arena));

    // We'll add tasks to this, and then worker threads will take tasks from it.
    let injector = Injector::new();

    // We need to allocate worker *queues* on the main thread and then move them
    // into the worker threads, because those workers' stealers need to be
    // shared between all threads, and this coordination work is much easier
    // on the main thread.
    let mut worker_queues = bumpalo::collections::Vec::with_capacity_in(num_workers, arena);
    let mut stealers = bumpalo::collections::Vec::with_capacity_in(num_workers, arena);

    for _ in 0..num_workers {
        let worker = Worker::new_fifo();

        stealers.push(worker.stealer());
        worker_queues.push(worker);
    }

    // Get a reference to the completed stealers, so we can send that
    // reference to each worker. (Slices are Sync, but bumpalo Vecs are not.)
    let stealers = stealers.into_bump_slice();
    let it = worker_arenas.iter_mut();

    {
        let thread_result = thread::scope(|thread_scope| {
            // Careful! It's important that worker listeners aren't allocated in the arena,
            // since they need to be correctly dropped if we have a panic in this thread::scope code.
            // Making sure they're owned means they'll be dropped correctly on either normal exit
            // of this thread::scope block or on panicking. When they're dropped, the worker threads
            // will correctly exit their message processing loops.
            // If these were allocated in the arena, we might panic without shutting down the worker threads,
            // causing the thread::scope block to hang while it waits for the worker threads to exit.
            let mut worker_wakers = Vec::with_capacity(num_workers);

            for worker_arena in it {
                let msg_tx = msg_tx.clone();
                let worker = worker_queues.pop().unwrap();

                let (worker_wakup_tx, worker_wakup_rx) = bounded(1024);
                worker_wakers.push(worker_wakup_tx);

                // We only want to move a *reference* to the main task queue's
                // injector in the thread, not the injector itself
                // (since other threads need to reference it too). Same with src_dir.
                let injector = &injector;
                let src_dir = &src_dir;

                // Record this thread's handle so the main thread can join it later.
                let res_join_handle = thread_scope
                    .builder()
                    .stack_size(EXPANDED_STACK_SIZE)
                    .spawn(move |_| {
                        // will process messages until we run out
                        roc_worker::worker_task(
                            worker,
                            injector,
                            stealers,
                            worker_wakup_rx,
                            |task| {
                                run_task(
                                    task,
                                    worker_arena,
                                    src_dir,
                                    msg_tx.clone(),
                                    roc_cache_dir,
                                    target,
                                )
                            },
                        )
                    });

                res_join_handle.unwrap_or_else(|_| {
                    panic!("Join handle panicked!");
                });
            }

            // We've now distributed one worker queue to each thread.
            // There should be no queues left to distribute!
            debug_assert!(worker_queues.is_empty());
            drop(worker_queues);

            // Grab a reference to these Senders outside the loop, so we can share
            // it across each iteration of the loop.
            let msg_tx = msg_tx.clone();

            // The root module will have already queued up messages to process,
            // and processing those messages will in turn queue up more messages.
            loop {
                match state_thread_step(arena, state, &worker_wakers, &injector, &msg_tx, &msg_rx) {
                    Ok(ControlFlow::Break(load_result)) => {
                        return Ok(load_result);
                    }
                    Ok(ControlFlow::Continue(new_state)) => {
                        state = new_state;
                        continue;
                    }
                    Err(e) => {
                        return Err(e);
                    }
                }
            }
        });

        thread_result.unwrap_or_else(|_| {
            // This most likely means a panic occurred in one of the threads.
            // Therefore, print all the error info we've accumulated, and note afterwards
            // that there was a compiler crash.
            //
            // Unfortunately, this often has no information to report if there's a panic in mono.
            // Consequently, the following ends up being more misleading than helpful.
            //
            // roc_reporting::cli::report_problems(
            //     &sources_recorded,
            //     &mut interns_recorded,
            //     &mut can_problems_recorded,
            //     &mut type_problems_recorded,
            // )
            // .print_error_warning_count(Duration::default()); // TODO determine total elapsed time and use it here

            Err(LoadingProblem::FormattedReport(
                concat!(
                    "\n\nThere was an unrecoverable error in the Roc compiler. The `roc check` ",
                    "command can sometimes give a more helpful error report than other commands.\n\n"
                )
                .to_string(),
            ))
        })
    }
}

fn start_tasks<'a>(
    arena: &'a Bump,
    state: &mut State<'a>,
    work: MutSet<(ModuleId, Phase)>,
    injector: &Injector<BuildTask<'a>>,
    worker_wakers: &[Sender<()>],
) -> Result<(), LoadingProblem<'a>> {
    for (module_id, phase) in work {
        let tasks = start_phase(module_id, phase, arena, state);

        for task in tasks {
            enqueue_task(injector, worker_wakers, task)?
        }
    }

    Ok(())
}

macro_rules! debug_print_ir {
    ($state:expr, $interner:expr, $flag:path) => {
        dbg_do!($flag, {
            let procs_string = $state
                .procedures
                .values()
                .map(|proc| proc.to_pretty($interner, 200, true))
                .collect::<Vec<_>>();

            let result = procs_string.join("\n");

            eprintln!("{}", result);
        })
    };
}

macro_rules! debug_check_ir {
    ($state:expr, $arena:expr, $interner:expr, $flag:path) => {
        dbg_do!($flag, {
            use roc_mono::debug::{check_procs, format_problems};

            let interns = Interns {
                module_ids: $state.arc_modules.lock().clone().into_module_ids(),
                all_ident_ids: $state.constrained_ident_ids.clone(),
            };

            let procedures = &$state.procedures;

            let problems = check_procs($arena, &mut $interner, procedures);
            if !problems.is_empty() {
                let formatted = format_problems(&interns, &$interner, problems);
                eprintln!("IR PROBLEMS FOUND:\n{formatted}");
            }
        })
    };
}

fn extend_module_with_builtin_import(module: &mut ParsedModule, module_id: ModuleId) {
    module
        .package_qualified_available_modules
        .insert(PackageQualified::Unqualified(module_id));

    module.available_modules.insert(module_id, Region::zero());

    let types = Symbol::builtin_types_in_scope(module_id)
        .iter()
        .map(|(name, info)| (Ident::from(*name), *info));
    module.initial_scope.extend(types);
}

fn update<'a>(
    mut state: State<'a>,
    msg: Msg<'a>,
    msg_tx: MsgSender<'a>,
    injector: &Injector<BuildTask<'a>>,
    worker_wakers: &[Sender<()>],
    arena: &'a Bump,
) -> Result<State<'a>, LoadingProblem<'a>> {
    use self::Msg::*;

    match msg {
        Many(messages) => {
            // enqueue all these message
            for msg in messages {
                msg_tx.send(msg).map_err(|_| {
                    LoadingProblem::ChannelProblem(ChannelProblem::FailedToSendManyMsg)
                })?;
            }

            Ok(state)
        }
        Header(header) => {
            log!("loaded header for {:?}", header.module_id);
            let home = header.module_id;
            let mut work = MutSet::default();

            // Only lock shorthands if this header has packages
            if !header.packages.is_empty() {
                let mut shorthands = state.arc_shorthands.lock();

                let mut parent_dir = header.module_path.clone();
                parent_dir.pop();

                register_package_shorthands(
                    &mut shorthands,
                    &header.packages,
                    &header.module_path,
                    &parent_dir,
                    &state.cache_dir,
                )?;
            }

            use HeaderType::*;

            match header.header_type {
                App { to_platform, .. } => {
                    state.platform_path = PlatformPath::Valid(to_platform);
                }
                Package {
                    config_shorthand,
                    exposes_ids,
                    ..
                } => {
                    if header.is_root_module {
                        state.exposed_modules = exposes_ids;
                    }

                    work.extend(state.dependencies.notify_package(config_shorthand));
                }
                Platform {
                    config_shorthand,
                    provides,
                    exposes_ids,
                    requires,
                    ..
                } => {
                    work.extend(state.dependencies.notify_package(config_shorthand));

                    let is_prebuilt = if header.is_root_module {
                        debug_assert!(matches!(state.platform_path, PlatformPath::NotSpecified));
                        state.platform_path = PlatformPath::RootIsPlatformModule;

                        // If the root module is this platform, then the platform is the very
                        // thing we're rebuilding!
                        false
                    } else {
                        // platforms from HTTPS URLs are always prebuilt
                        matches!(
                            state.arc_shorthands.lock().get(config_shorthand),
                            Some(ShorthandPath::FromHttpsUrl { .. })
                        )
                    };

                    // If we're building an app module, and this was the platform
                    // specified in its header's `to` field, record it as our platform.
                    if state.opt_platform_shorthand == Some(config_shorthand) {
                        debug_assert!(state.platform_data.is_none());

                        state.platform_data = Some(PlatformData {
                            module_id: header.module_id,
                            provides,
                            is_prebuilt,
                        });
                    }

                    if header.is_root_module {
                        state.exposed_modules = exposes_ids;
                    }

                    if requires.iter().any(|requires| {
                        IdentSuffix::from_name(requires.value.ident.value).is_bang()
                    }) {
                        state.fx_mode = FxMode::PurityInference;
                    }
                }
                Builtin { .. } | Module { .. } => {
                    if header.is_root_module {
                        debug_assert!(matches!(state.platform_path, PlatformPath::NotSpecified));
                        state.platform_path = PlatformPath::RootIsModule;
                    }
                }
                Hosted { exposes, .. } => {
                    if header.is_root_module {
                        debug_assert!(matches!(state.platform_path, PlatformPath::NotSpecified));
                        state.platform_path = PlatformPath::RootIsHosted;
                    }

                    if exposes
                        .iter()
                        .any(|exposed| exposed.value.is_effectful_fn())
                    {
                        state.fx_mode = FxMode::PurityInference;
                    }
                }
            }

            state.module_cache.headers.insert(header.module_id, header);

            work.extend(state.dependencies.notify(home, Phase::LoadHeader));
            work.insert((home, Phase::Parse));

            start_tasks(arena, &mut state, work, injector, worker_wakers)?;

            Ok(state)
        }
        Parsed(parsed) => {
            let module_id = parsed.module_id;

            // store an ID to name mapping, so we know the file to read when fetching dependencies' headers
            for (name, id) in parsed.deps_by_name.iter() {
                state.module_cache.module_names.insert(*id, name.clone());
            }

            // This was a dependency. Write it down and keep processing messages.
            let mut exposed_symbols: VecSet<Symbol> = VecSet::with_capacity(parsed.exposes.len());

            // TODO can we avoid this loop by storing them as a Set in Header to begin with?
            for symbol in parsed.exposes.iter() {
                exposed_symbols.insert(*symbol);
            }

            // NOTE we currently re-parse the headers when a module is imported twice.
            // We need a proper solution that marks a phase as in-progress so it's not repeated
            // debug_assert!(!state.exposed_symbols_by_module.contains_key(&home));

            state
                .exposed_symbols_by_module
                .insert(module_id, exposed_symbols);

            // add the prelude
            let mut parsed = parsed;

            if !module_id.is_builtin() {
                let parsed = &mut parsed;

                extend_module_with_builtin_import(parsed, ModuleId::NUM);
                extend_module_with_builtin_import(parsed, ModuleId::BOOL);
                extend_module_with_builtin_import(parsed, ModuleId::STR);
                extend_module_with_builtin_import(parsed, ModuleId::LIST);
                extend_module_with_builtin_import(parsed, ModuleId::RESULT);
                extend_module_with_builtin_import(parsed, ModuleId::DICT);
                extend_module_with_builtin_import(parsed, ModuleId::SET);
                extend_module_with_builtin_import(parsed, ModuleId::BOX);
                extend_module_with_builtin_import(parsed, ModuleId::ENCODE);
                extend_module_with_builtin_import(parsed, ModuleId::DECODE);
                extend_module_with_builtin_import(parsed, ModuleId::HASH);
                extend_module_with_builtin_import(parsed, ModuleId::INSPECT);
                extend_module_with_builtin_import(parsed, ModuleId::TASK);
            }

            state
                .module_cache
                .imports
                .entry(module_id)
                .or_default()
                .extend(
                    parsed
                        .package_qualified_available_modules
                        .iter()
                        .map(|x| *x.as_inner()),
                );

            let added_deps_result = state.dependencies.add_module(
                module_id,
                &parsed.package_qualified_available_modules,
                state.exec_mode.goal_phase(),
            );

            let work = match added_deps_result {
                Ok(work) => work,
                Err(DepCycle { cycle }) => {
                    return Err(LoadingProblem::ImportCycle(
                        parsed.module_path.clone(),
                        cycle,
                    ));
                }
            };

            start_tasks(arena, &mut state, work, injector, worker_wakers)?;

            state
                .module_cache
                .sources
                .insert(parsed.module_id, (parsed.module_path.clone(), parsed.src));

            state.module_cache.parsed.insert(module_id, parsed);

            let work = state.dependencies.notify(module_id, Phase::Parse);

            start_tasks(arena, &mut state, work, injector, worker_wakers)?;

            Ok(state)
        }

        CanonicalizedAndConstrained(CanAndCon {
            constrained_module,
            canonicalization_problems,
            module_docs,
        }) => {
            let module_id = constrained_module.module.module_id;
            log!("generated constraints for {:?}", module_id);
            state
                .module_cache
                .can_problems
                .insert(module_id, canonicalization_problems);

            if let Some(docs) = module_docs {
                state.module_cache.documentation.insert(module_id, docs);
            }

            state
                .module_cache
                .exposed_imports
                .insert(module_id, constrained_module.module.exposed_imports.clone());

            state
                .module_cache
                .aliases
                .insert(module_id, constrained_module.module.aliases.clone());

            state
                .module_cache
                .pending_abilities
                .insert(module_id, constrained_module.module.abilities_store.clone());

            if let Some(module_params) = constrained_module.module.module_params.clone() {
                state
                    .module_cache
                    .module_params
                    .insert(module_id, module_params);
            }

            state
                .module_cache
                .constrained
                .insert(module_id, constrained_module);

            let work = state
                .dependencies
                .notify(module_id, Phase::CanonicalizeAndConstrain);

            start_tasks(arena, &mut state, work, injector, worker_wakers)?;

            Ok(state)
        }
        SolvedTypes {
            module_id,
            ident_ids,
            solved_module,
            solved_subs,
            decls,
            dep_idents,
            mut module_timing,
            abilities_store,
            loc_expects,
            loc_dbgs,

            #[cfg(debug_assertions)]
            checkmate,
        } => {
            log!("solved types for {:?}", module_id);
            module_timing.end_time = Instant::now();

            state
                .module_cache
                .type_problems
                .insert(module_id, solved_module.problems);
            state
                .module_cache
                .exposes
                .insert(module_id, solved_module.exposed_vars_by_symbol.clone());

            let should_include_expects = (!loc_expects.is_empty() || !loc_dbgs.is_empty()) && {
                let modules = state.arc_modules.lock();
                modules
                    .package_eq(module_id, state.root_id)
                    .expect("root or this module is not yet known - that's a bug!")
            };

            let opt_expectations = if should_include_expects {
                let (path, _) = state.module_cache.sources.get(&module_id).unwrap();

                Some(Expectations {
                    expectations: loc_expects,
                    dbgs: loc_dbgs,
                    subs: solved_subs.clone().into_inner(),
                    path: path.to_owned(),
                    ident_ids: ident_ids.clone(),
                })
            } else {
                None
            };

            let work = state.dependencies.notify(module_id, Phase::SolveTypes);

            // if there is a platform, the `platform` module provides host-exposed,
            // otherwise the App module exposes host-exposed
            let is_host_exposed = match state.platform_data {
                None => module_id == state.root_id,
                Some(ref platform_data) => module_id == platform_data.module_id,
            };

            let add_to_host_exposed = is_host_exposed &&
                // During testing, we don't need to expose anything to the host.
                !matches!(state.exec_mode, ExecutionMode::Test);

            if add_to_host_exposed {
                state.exposed_to_host.top_level_values.extend(
                    solved_module
                        .exposed_vars_by_symbol
                        .iter()
                        .filter_map(|(k, v)| {
                            if abilities_store.is_specialization_name(*k) {
                                None
                            } else {
                                Some((*k, *v))
                            }
                        }),
                );

                state
                    .exposed_to_host
                    .closure_types
                    .extend(solved_module.aliases.keys().copied());
            }

            let finish_type_checking = is_host_exposed &&
                (state.goal_phase() == Phase::SolveTypes)
                // If we're running in check-and-then-build mode, only exit now there are errors.
                && (!state.exec_mode.build_if_checks() || state.module_cache.has_errors());

            if finish_type_checking {
                debug_assert!(work.is_empty());
                debug_assert!(state.dependencies.solved_all());

                state.timings.insert(module_id, module_timing);

                if state.exec_mode.build_if_checks() {
                    // We there may outstanding modules in the typecheked cache whose ident IDs
                    // aren't registered; transfer all of their idents over to the state, since
                    // we're now done and ready to report errors.
                    for (
                        module_id,
                        TypeCheckedModule {
                            ident_ids,
                            module_timing,
                            ..
                        },
                    ) in state.module_cache.typechecked.drain()
                    {
                        state.constrained_ident_ids.insert(module_id, ident_ids);
                        state.timings.insert(module_id, module_timing);
                    }
                }

                let documentation = {
                    let mut empty = VecMap::default();
                    std::mem::swap(&mut empty, &mut state.module_cache.documentation);

                    empty
                };

                msg_tx
                    .send(Msg::FinishedAllTypeChecking {
                        solved_subs,
                        exposed_vars_by_symbol: solved_module.exposed_vars_by_symbol,
                        exposed_aliases_by_symbol: solved_module.aliases,
                        exposed_types_storage: solved_module.exposed_types,
                        resolved_implementations: solved_module.solved_implementations,
                        dep_idents,
                        documentation,
                        abilities_store,

                        #[cfg(debug_assertions)]
                        checkmate,
                    })
                    .map_err(|_| {
                        LoadingProblem::ChannelProblem(
                            ChannelProblem::FailedToSendFinishedTypeCheckingMsg,
                        )
                    })?;

                // bookkeeping
                state.declarations_by_id.insert(module_id, decls);
                state.constrained_ident_ids.insert(module_id, ident_ids);

                // As far as type-checking goes, once we've solved
                // the originally requested module, we're all done!
                return Ok(state);
            } else {
                state.exposed_types.insert(
                    module_id,
                    ExposedModuleTypes {
                        exposed_types_storage_subs: solved_module.exposed_types,
                        resolved_implementations: solved_module.solved_implementations,
                    },
                );

                if state.goal_phase() > Phase::SolveTypes || state.exec_mode.build_if_checks() {
                    let layout_cache = state.layout_caches.pop().unwrap_or_else(|| {
                        LayoutCache::new(state.layout_interner.fork(), state.target)
                    });

                    let typechecked = TypeCheckedModule {
                        module_id,
                        layout_cache,
                        module_timing,
                        solved_subs,
                        decls,
                        ident_ids,
                        abilities_store,
                        expectations: opt_expectations,

                        #[cfg(debug_assertions)]
                        checkmate,
                    };

                    state
                        .module_cache
                        .typechecked
                        .insert(module_id, typechecked);
                } else {
                    state.module_cache.checked.insert(
                        module_id,
                        CheckedModule {
                            solved_subs,
                            decls,
                            abilities_store,
                        },
                    );
                    state.constrained_ident_ids.insert(module_id, ident_ids);
                    state.timings.insert(module_id, module_timing);
                }

                let work = if is_host_exposed && state.exec_mode.build_if_checks() {
                    debug_assert!(
                        work.is_empty(),
                        "work left over after host exposed is checked"
                    );

                    // Load the find + make specializations portion of the dependency graph.
                    state
                        .dependencies
                        .load_find_and_make_specializations_after_check()
                } else {
                    work
                };

                start_tasks(arena, &mut state, work, injector, worker_wakers)?;
            }

            Ok(state)
        }
        FoundSpecializations {
            module_id,
            procs_base,
            solved_subs,
            ident_ids,
            layout_cache,
            module_timing,
            toplevel_expects,
            expectations,
        } => {
            log!("found specializations for {:?}", module_id);

            let subs = solved_subs.into_inner();

            if !toplevel_expects.pure.is_empty() {
                state.toplevel_expects.insert(module_id, toplevel_expects);
            }

            state
                .module_cache
                .top_level_thunks
                .entry(module_id)
                .or_default()
                .extend(procs_base.module_thunks.iter().copied());

            let found_specializations_module = FoundSpecializationsModule {
                ident_ids,
                layout_cache,
                procs_base,
                subs,
                module_timing,
                expectations,
            };

            state
                .module_cache
                .found_specializations
                .insert(module_id, found_specializations_module);

            let work = state
                .dependencies
                .notify(module_id, Phase::FindSpecializations);

            start_tasks(arena, &mut state, work, injector, worker_wakers)?;

            Ok(state)
        }
        MadeSpecializations {
            module_id,
            ident_ids,
            mut update_mode_ids,
            subs,
            procs_base,
            procedures,
            host_exposed_lambda_sets,
            external_specializations_requested,
            module_timing,
            layout_cache,
            expectations,
            ..
        } => {
            debug_assert!(
                state.goal_phase() == Phase::MakeSpecializations
                    || state.exec_mode.build_if_checks()
            );

            log!("made specializations for {:?}", module_id);

            // in the future, layouts will be in SoA form and we'll want to hold on to this data
            let _ = layout_cache;

            state.procedures.extend(procedures);
            state
                .host_exposed_lambda_sets
                .extend(host_exposed_lambda_sets);
            state.module_cache.late_specializations.insert(
                module_id,
                LateSpecializationsModule {
                    ident_ids,
                    module_timing,
                    subs,
                    layout_cache,
                    procs_base,
                    expectations,
                },
            );

            let work = state
                .dependencies
                .notify(module_id, Phase::MakeSpecializations);

            for (module_id, requested) in external_specializations_requested {
                let existing = match state
                    .module_cache
                    .external_specializations_requested
                    .entry(module_id)
                {
                    Vacant(entry) => entry.insert(vec![]),
                    Occupied(entry) => entry.into_mut(),
                };

                existing.push(requested);
            }

            enum NextStep {
                Done,
                RelaunchPhase,
                MakingInPhase,
            }

            let all_work_done = work.is_empty() && state.dependencies.solved_all();
            let next_step = if all_work_done {
                if state
                    .module_cache
                    .external_specializations_requested
                    .is_empty()
                {
                    NextStep::Done
                } else {
                    NextStep::RelaunchPhase
                }
            } else {
                NextStep::MakingInPhase
            };

            match next_step {
                NextStep::Done => {
                    // We are all done with specializations across all modules.
                    // Insert post-specialization operations and report our completion.

                    if !state
                        .module_cache
                        .external_specializations_requested
                        .is_empty()
                    {
                        internal_error!(
                            "No more work left, but external specializations left over: {:?}",
                            state.module_cache.external_specializations_requested
                        );
                    }

                    let mut module_expectations =
                        VecMap::with_capacity(state.module_cache.module_names.len());

                    // Flush late-specialization module information to the top-level of the state
                    // where it will be visible to others, since we don't need late specialization
                    // anymore.
                    for (
                        module_id,
                        LateSpecializationsModule {
                            ident_ids,
                            subs,
                            module_timing,
                            layout_cache: _layout_cache,
                            procs_base: _,
                            expectations,
                        },
                    ) in state.module_cache.late_specializations.drain()
                    {
                        state.constrained_ident_ids.insert(module_id, ident_ids);
                        if module_id == state.root_id {
                            state.root_subs = Some(subs);
                        }
                        state.timings.insert(module_id, module_timing);
                        if let Some(expectations) = expectations {
                            module_expectations.insert(module_id, expectations);
                        }

                        #[cfg(debug_assertions)]
                        {
                            log_layout_stats(module_id, &_layout_cache);
                        }
                    }

                    let layout_interner = {
                        let mut taken = GlobalLayoutInterner::with_capacity(0, state.target);
                        std::mem::swap(&mut state.layout_interner, &mut taken);
                        taken
                    };

                    #[allow(unused_mut)]
                    let mut layout_interner = layout_interner
                        .unwrap()
                        .expect("outstanding references to global layout interener, but we just drained all layout caches");

                    log!("specializations complete from {:?}", module_id);

                    debug_print_ir!(state, &layout_interner, ROC_PRINT_IR_AFTER_SPECIALIZATION);
                    debug_check_ir!(state, arena, layout_interner, ROC_CHECK_MONO_IR);

                    let ident_ids = state.constrained_ident_ids.get_mut(&module_id).unwrap();

                    roc_mono::tail_recursion::apply_trmc(
                        arena,
                        &mut layout_interner,
                        module_id,
                        ident_ids,
                        &mut state.procedures,
                    );

                    debug_print_ir!(state, &layout_interner, ROC_PRINT_IR_AFTER_TRMC);

                    inc_dec::insert_inc_dec_operations(
                        arena,
                        &layout_interner,
                        &mut state.procedures,
                    );

                    debug_print_ir!(state, &layout_interner, ROC_PRINT_IR_AFTER_REFCOUNT);

                    drop_specialization::specialize_drops(
                        arena,
                        &mut layout_interner,
                        module_id,
                        ident_ids,
                        &mut state.procedures,
                    );

                    debug_print_ir!(
                        state,
                        &layout_interner,
                        ROC_PRINT_IR_AFTER_DROP_SPECIALIZATION
                    );

                    reset_reuse::insert_reset_reuse_operations(
                        arena,
                        &layout_interner,
                        module_id,
                        state.target,
                        ident_ids,
                        &mut update_mode_ids,
                        &mut state.procedures,
                    );

                    debug_print_ir!(state, &layout_interner, ROC_PRINT_IR_AFTER_RESET_REUSE);

                    // This is not safe with the new non-recursive RC updates that we do for tag unions
                    //
                    // Proc::optimize_refcount_operations(
                    //     arena,
                    //     module_id,
                    //     &mut ident_ids,
                    //     &mut state.procedures,
                    // );

                    // use the subs of the root module;
                    // this is used in the repl to find the type of `main`
                    let subs = state.root_subs.clone().unwrap();

                    msg_tx
                        .send(Msg::FinishedAllSpecialization {
                            subs,
                            layout_interner,
                            exposed_to_host: state.exposed_to_host.clone(),
                            module_expectations,
                        })
                        .map_err(|_| {
                            LoadingProblem::ChannelProblem(
                                ChannelProblem::FailedToSendFinishedSpecializationsMsg,
                            )
                        })?;

                    Ok(state)
                }

                NextStep::RelaunchPhase => {
                    // We passed through the dependency graph of modules to be specialized, but
                    // there are still specializations left over. Restart the make specializations
                    // phase in reverse topological order.
                    //
                    // This happens due to abilities. In detail, consider
                    //
                    //   # Default module
                    //   module [default, getDefault]
                    //
                    //   Default implements default : {} -> a where a implements Default
                    //
                    //   getDefault = \{} -> default {}
                    //
                    //   # App module
                    //   app "test" provides [main] imports [Default.{default, getDefault}]
                    //
                    //   Foo := {}
                    //
                    //   default = \{} -> @Foo {}
                    //
                    //   main =
                    //     f : Foo
                    //     f = getDefault {}
                    //     f
                    //
                    // The syntactic make specializations graph (based on imports) will be
                    // App -> Default, and in a pass will build the specializations `App#main` and
                    // `Default#getDefault for Foo`. But now notice that `Default#getDefault` will
                    // have gained an implicit dependency on the specialized `default` for `Foo`,
                    // `App#Foo#default`. So for abilities, the syntactic import graph is not
                    // enough to express the entire dependency graph.
                    //
                    // The simplest way to resolve these leftover, possibly circular
                    // specializations is to relaunch the make-specializations phase in the import
                    // order until there are no more specializations left to be made. This is a bit
                    // unfortunate in that we may look again into modules that don't need any
                    // specializations made, but there are also some nice properties:
                    //
                    // - no more specializations will be made than needed
                    // - the number of phase relaunches scales linearly with the largest number of
                    //   "bouncing back and forth" between ability calls, which is likely to be
                    //   small in practice
                    // - the phases will always terminate. suppose they didn't; then there must be
                    //   an infinite chain of calls all of which have different layouts. In Roc
                    //   this can only be true if the calls are all mutually recursive, and
                    //   furthermore are polymorphically recursive. But polymorphic recursion is
                    //   illegal in Roc, will have been enforced during type inference.

                    if state
                        .module_cache
                        .external_specializations_requested
                        .is_empty()
                    {
                        internal_error!(
                            "No specializations left over, but we were told to loop making specializations"
                        );
                    }

                    log!(
                        "re-launching make-specializations: pass {}",
                        state.make_specializations_pass.current_pass() + 1
                    );

                    state.make_specializations_pass.inc();

                    let work = state.dependencies.reload_make_specialization_pass();

                    start_tasks(arena, &mut state, work, injector, worker_wakers)?;

                    Ok(state)
                }

                NextStep::MakingInPhase => {
                    start_tasks(arena, &mut state, work, injector, worker_wakers)?;

                    Ok(state)
                }
            }
        }
        Msg::FailedToLoad(problem) => {
            // TODO report the error and continue instead of erroring out
            Err(problem)
        }
        Msg::FinishedAllTypeChecking { .. } => {
            unreachable!();
        }
        Msg::FinishedAllSpecialization { .. } => {
            unreachable!();
        }
        Msg::FailedToParse(_) => {
            unreachable!();
        }
        Msg::FailedToReadFile { .. } => {
            unreachable!();
        }
        Msg::IncorrectModuleName(..) => {
            internal_error!();
        }
    }
}

fn register_package_shorthands<'a>(
    shorthands: &mut MutMap<&'a str, ShorthandPath>,
    package_entries: &MutMap<&'a str, header::PackageName<'a>>,
    #[allow(unused_variables)] // for wasm
    module_path: &Path,
    src_dir: &Path,
    #[allow(unused_variables)] // for wasm
    cache_dir: &Path,
) -> Result<(), LoadingProblem<'a>> {
    for (shorthand, package_name) in package_entries.iter() {
        let package_str = package_name.as_str();
        let shorthand_path = if package_str.starts_with("https://") {
            #[cfg(not(target_family = "wasm"))]
            {
                let url = package_str;
                match PackageMetadata::try_from(url) {
                    Ok(url_metadata) => {
                        // This was a valid URL
                        let root_module_dir = cache_dir
                            .join(url_metadata.cache_subdir)
                            .join(url_metadata.content_hash);
                        let root_module = root_module_dir.join(
                            url_metadata
                                .root_module_filename
                                .unwrap_or(DEFAULT_MAIN_NAME),
                        );

                        ShorthandPath::FromHttpsUrl {
                            root_module_dir,
                            root_module,
                        }
                    }
                    Err(url_err) => {
                        let buf = to_https_problem_report_string(
                            url,
                            Problem::InvalidUrl(url_err),
                            module_path.to_path_buf(),
                        );
                        return Err(LoadingProblem::FormattedReport(buf));
                    }
                }
            }

            #[cfg(target_family = "wasm")]
            {
                panic!("Specifying packages via URLs is currently unsupported in wasm.");
            }
        } else {
            // This wasn't a URL, so it must be a filesystem path.
            let root_module: PathBuf = src_dir.join(package_str);
            let root_module_dir = root_module.parent().unwrap_or_else(|| {
                if root_module.is_file() {
                    // Files must have parents!
                    internal_error!("Somehow I got a file path to a real file on the filesystem that has no parent!");
                } else {
                    // TODO make this a nice report
                    todo!(
                        "platform module {:?} was not a file.",
                        package_str
                    )
                }
            }).into();

            ShorthandPath::RelativeToSrc {
                root_module_dir,
                root_module,
            }
        };

        log!(
            "New package shorthand: {:?} => {:?}",
            shorthand,
            shorthand_path
        );

        shorthands.insert(shorthand, shorthand_path);
    }

    Ok(())
}

#[cfg(debug_assertions)]
fn log_layout_stats(module_id: ModuleId, layout_cache: &LayoutCache) {
    let (cache_stats, raw_function_cache_stats) = layout_cache.statistics();
    roc_tracing::info!(
        module = ?module_id,
        insertions = cache_stats.insertions,
        hits = cache_stats.hits,
        misses = cache_stats.misses,
        non_insertable = cache_stats.non_insertable,
        non_reusable = cache_stats.non_reusable,
        "cache stats"
    );
    roc_tracing::info!(
        module = ?module_id,
        insertions = raw_function_cache_stats.insertions,
        hits = raw_function_cache_stats.hits,
        misses = raw_function_cache_stats.misses,
        non_insertable = raw_function_cache_stats.non_insertable,
        non_reusable = raw_function_cache_stats.non_reusable,
        "raw function cache stats"
    );
}

fn finish_specialization<'a>(
    arena: &'a Bump,
    state: State<'a>,
    subs: Subs,
    layout_interner: STLayoutInterner<'a>,
    exposed_to_host: ExposedToHost,
    module_expectations: VecMap<ModuleId, Expectations>,
) -> Result<MonomorphizedModule<'a>, LoadingProblem<'a>> {
    if false {
        println!(
            "total Type clones: {} ",
            roc_types::types::get_type_clone_count()
        );
    }
    let module_ids = Arc::try_unwrap(state.arc_modules)
        .unwrap_or_else(|_| panic!("There were still outstanding Arc references to module_ids"))
        .into_inner()
        .into_module_ids();

    let mut all_ident_ids = state.constrained_ident_ids;

    // Associate the ident IDs from the derived synth module
    let (_, derived_synth_ident_ids) = Arc::try_unwrap(state.derived_module)
        .unwrap_or_else(|_| internal_error!("Outstanding references to the derived module"))
        .into_inner()
        .unwrap()
        .decompose();
    ModuleId::DERIVED_SYNTH.register_debug_idents(&derived_synth_ident_ids);
    all_ident_ids.insert(ModuleId::DERIVED_SYNTH, derived_synth_ident_ids);

    let mut interns = Interns {
        module_ids,
        all_ident_ids,
    };

    let entry_point = {
        let interns: &mut Interns = &mut interns;
        match state.exec_mode {
            ExecutionMode::Test => Ok(EntryPoint::Test),
            ExecutionMode::Executable | ExecutionMode::ExecutableIfCheck => {
                use PlatformPath::*;

                let platform_path = match &state.platform_path {
                    Valid(To::ExistingPackage(shorthand)) => {
                        match state.arc_shorthands.lock().get(shorthand) {
                            Some(shorthand_path) => shorthand_path.root_module().to_path_buf(),
                            None => unreachable!(),
                        }
                    }
                    Valid(To::NewPackage(p_or_p)) => PathBuf::from(p_or_p.as_str()),
                    other => {
                        let buf = report_cannot_run(state.root_id, state.root_path, other);
                        return Err(LoadingProblem::FormattedReport(buf));
                    }
                };

                let exposed_symbols_and_layouts = match state.platform_data {
                    None => {
                        let src = &state.exposed_to_host.top_level_values;
                        let mut buf = bumpalo::collections::Vec::with_capacity_in(src.len(), arena);

                        for &symbol in src.keys() {
                            let proc_layout =
                                proc_layout_for(state.procedures.keys().copied(), symbol);

                            buf.push(("", symbol, proc_layout));
                        }

                        buf.into_bump_slice()
                    }
                    Some(PlatformData {
                        module_id,
                        provides,
                        ..
                    }) => {
                        let ident_ids = interns.all_ident_ids.get_mut(&module_id).unwrap();
                        let mut buf =
                            bumpalo::collections::Vec::with_capacity_in(provides.len(), arena);

                        for loc_name in provides {
                            let fn_name = loc_name.value.as_str();
                            let ident_id = ident_ids.get_or_insert(fn_name);
                            let symbol = Symbol::new(module_id, ident_id);
                            let proc_layout =
                                proc_layout_for(state.procedures.keys().copied(), symbol);

                            buf.push((fn_name, symbol, proc_layout));
                        }

                        buf.into_bump_slice()
                    }
                };

                Ok(EntryPoint::Executable {
                    exposed_to_host: exposed_symbols_and_layouts,
                    platform_path,
                })
            }
            ExecutionMode::Check => unreachable!(),
        }
    }?;

    let State {
        toplevel_expects,
        procedures,
        host_exposed_lambda_sets,
        module_cache,
        platform_data,
        ..
    } = state;

    let ModuleCache {
        type_problems,
        can_problems,
        sources,
        ..
    } = module_cache;

    let sources: MutMap<ModuleId, (PathBuf, Box<str>)> = sources
        .into_iter()
        .map(|(id, (path, src))| (id, (path, src.into())))
        .collect();

    let module_id = state.root_id;
    let needs_prebuilt_host = match platform_data {
        Some(data) => data.is_prebuilt,
        // If there's no platform data (e.g. because we're building a module)
        // then there's no prebuilt platform either!
        None => false,
    };

    Ok(MonomorphizedModule {
        can_problems,
        type_problems,
        expectations: module_expectations,
        exposed_to_host,
        module_id,
        subs,
        interns,
        layout_interner,
        procedures,
        host_exposed_lambda_sets,
        entry_point,
        sources,
        timings: state.timings,
        toplevel_expects,
        glue_layouts: GlueLayouts { getters: vec![] },
        needs_prebuilt_host,
    })
}

fn proc_layout_for<'a>(
    mut proc_symbols: impl Iterator<Item = (Symbol, ProcLayout<'a>)>,
    symbol: Symbol,
) -> ProcLayout<'a> {
    match proc_symbols.find(|(s, _)| *s == symbol) {
        Some((_, layout)) => layout,
        None => {
            // the entry point is not specialized. This can happen if the repl output
            // is a function value
            roc_mono::ir::ProcLayout {
                arguments: &[],
                result: Layout::UNIT,
                niche: Niche::NONE,
            }
        }
    }
}

#[allow(clippy::too_many_arguments)]
fn finish(
    mut state: State,
    solved: Solved<Subs>,
    exposed_aliases_by_symbol: MutMap<Symbol, Alias>,
    exposed_vars_by_symbol: Vec<(Symbol, Variable)>,
    exposed_types_storage: ExposedTypesStorageSubs,
    resolved_implementations: ResolvedImplementations,
    dep_idents: IdentIdsByModule,
    documentation: VecMap<ModuleId, ModuleDocumentation>,
    abilities_store: AbilitiesStore,
    //
    #[cfg(debug_assertions)] checkmate: Option<roc_checkmate::Collector>,
) -> LoadedModule {
    let module_ids = Arc::try_unwrap(state.arc_modules)
        .unwrap_or_else(|_| panic!("There were still outstanding Arc references to module_ids"))
        .into_inner()
        .into_module_ids();

    // Associate the ident IDs from the derived synth module
    let (_, derived_synth_ident_ids) = Arc::try_unwrap(state.derived_module)
        .unwrap_or_else(|_| internal_error!("Outstanding references to the derived module"))
        .into_inner()
        .unwrap()
        .decompose();
    ModuleId::DERIVED_SYNTH.register_debug_idents(&derived_synth_ident_ids);
    state
        .constrained_ident_ids
        .insert(ModuleId::DERIVED_SYNTH, derived_synth_ident_ids);

    let interns = Interns {
        module_ids,
        all_ident_ids: state.constrained_ident_ids,
    };

    let sources = state
        .module_cache
        .sources
        .into_iter()
        .map(|(id, (path, src))| (id, (path, src.into())))
        .collect();

    let exposed_values = exposed_vars_by_symbol.iter().map(|x| x.0).collect();

    let declarations_by_id = state.declarations_by_id;

    roc_checkmate::dump_checkmate!(checkmate);

    LoadedModule {
        module_id: state.root_id,
        filename: state.root_path,
        interns,
        solved,
        can_problems: state.module_cache.can_problems,
        type_problems: state.module_cache.type_problems,
        declarations_by_id,
        typechecked: state.module_cache.checked,
        dep_idents,
        exposed_aliases: exposed_aliases_by_symbol,
        exposed_values,
        exposed_to_host: exposed_vars_by_symbol.into_iter().collect(),
        exposed_types_storage,
        exposed_modules: state.exposed_modules.into(),
        resolved_implementations,
        sources,
        timings: state.timings,
        docs_by_module: documentation,
        abilities_store,
        exposed_imports: state.module_cache.exposed_imports,
        imports: state.module_cache.imports,
        exposes: state.module_cache.exposes,
    }
}

/// Load a `package` or `platform` module from disk
fn load_package_from_disk<'a>(
    arena: &'a Bump,
    filename: &Path,
    shorthand: &'a str,
    roc_cache_dir: RocCacheDir,
    app_module_id: Option<ModuleId>,
    module_ids: Arc<Mutex<PackageModuleIds<'a>>>,
    ident_ids_by_module: SharedIdentIdsByModule,
) -> Result<Msg<'a>, LoadingProblem<'a>> {
    let module_start_time = Instant::now();
    let file_io_start = module_start_time;
    let read_result = fs::read(filename);
    let file_io_duration = file_io_start.elapsed();

    match read_result {
        Ok(bytes_vec) => {
            let parse_start = Instant::now();
            let bytes = arena.alloc(bytes_vec);
            let parse_state = roc_parse::state::State::new(bytes);
            let parsed = roc_parse::header::parse_header(arena, parse_state.clone());
            let parse_header_duration = parse_start.elapsed();

            // Insert the first entries for this module's timings
            let mut pkg_module_timing = ModuleTiming::new(module_start_time);

            pkg_module_timing.read_roc_file = file_io_duration;
            pkg_module_timing.parse_header = parse_header_duration;

            match parsed {
                Ok((
                    ast::SpacesBefore {
                        item: ast::Header::Module(header),
                        ..
                    },
                    _parse_state,
                )) => Err(LoadingProblem::UnexpectedHeader(format!(
                    "expected platform/package module, got Module with header\n{header:?}"
                ))),
                Ok((
                    ast::SpacesBefore {
                        item: ast::Header::Hosted(header),
                        ..
                    },
                    _parse_state,
                )) => Err(LoadingProblem::UnexpectedHeader(format!(
                    "expected platform/package module, got Hosted module with header\n{header:?}"
                ))),
                Ok((
                    ast::SpacesBefore {
                        item: ast::Header::App(header),
                        ..
                    },
                    _parse_state,
                )) => Err(LoadingProblem::UnexpectedHeader(format!(
                    "expected platform/package module, got App with header\n{header:?}"
                ))),
                Ok((
                    ast::SpacesBefore {
                        item: ast::Header::Package(header),
                        before: comments,
                    },
                    parser_state,
                )) => {
                    let mut parent_dir = filename.to_path_buf();
                    parent_dir.pop();

                    let packages = unspace(arena, header.packages.value.items);

                    let (_, _, header) = build_package_header(
                        arena,
                        Some(shorthand),
                        false, // cannot be the root if loaded as a package
                        filename.to_path_buf(),
                        parser_state,
                        module_ids.clone(),
                        ident_ids_by_module.clone(),
                        &header,
                        comments,
                        pkg_module_timing,
                    )?;

                    let filename = header.module_path.clone();
                    let mut messages = Vec::with_capacity(packages.len() + 1);
                    messages.push(Msg::Header(header));

                    load_packages(
                        packages,
                        &mut messages,
                        roc_cache_dir,
                        parent_dir,
                        arena,
                        None,
                        module_ids,
                        ident_ids_by_module,
                        filename,
                    );

                    Ok(Msg::Many(messages))
                }
                Ok((
                    ast::SpacesBefore {
                        item: ast::Header::Platform(header),
                        before: comments,
                    },
                    parser_state,
                )) => {
                    let mut parent_dir = filename.to_path_buf();
                    parent_dir.pop();

                    let packages = unspace(arena, header.packages.item.items);

                    let exposes_ids = get_exposes_ids(
                        header.exposes.item.items,
                        arena,
                        &module_ids,
                        &ident_ids_by_module,
                    );

                    // make a `platform` module that ultimately exposes `main` to the host
                    let (_, _, header) = build_platform_header(
                        arena,
                        Some(shorthand),
                        false, // cannot be the root if loaded as a package
                        app_module_id,
                        filename.to_path_buf(),
                        parser_state,
                        module_ids.clone(),
                        exposes_ids.into_bump_slice(),
                        &header,
                        comments,
                        pkg_module_timing,
                    )?;

                    let filename = header.module_path.clone();
                    let mut messages = Vec::with_capacity(packages.len() + 1);
                    messages.push(Msg::Header(header));

                    load_packages(
                        packages,
                        &mut messages,
                        roc_cache_dir,
                        parent_dir,
                        arena,
                        None,
                        module_ids,
                        ident_ids_by_module,
                        filename,
                    );

                    Ok(Msg::Many(messages))
                }
                Err(fail) => Err(LoadingProblem::ParsingFailed(
                    fail.map_problem(SyntaxError::Header)
                        .into_file_error(filename.to_path_buf()),
                )),
            }
        }

        Err(err) => Err(LoadingProblem::FileProblem {
            filename: filename.to_path_buf(),
            error: err.kind(),
        }),
    }
}

fn get_exposes_ids<'a>(
    entries: &'a [Loc<Spaced<'a, roc_parse::header::ModuleName<'a>>>],
    arena: &'a Bump,
    module_ids: &Arc<Mutex<PackageModuleIds<'a>>>,
    ident_ids_by_module: &Arc<Mutex<IdentIdsByModule>>,
) -> bumpalo::collections::Vec<'a, ModuleId> {
    let mut exposes_ids = bumpalo::collections::Vec::with_capacity_in(entries.len(), arena);

    // Lock just long enough to perform the minimal operations necessary.
    let mut module_ids = (**module_ids).lock();
    let mut ident_ids_by_module = (**ident_ids_by_module).lock();

    // TODO can we "iterate unspaced" instead of calling unspace here?
    for entry in unspace(arena, entries) {
        let module_id =
            module_ids.get_or_insert(&PQModuleName::Unqualified(entry.value.as_str().into()));

        // Ensure this module has an entry in the ident_ids_by_module map.
        ident_ids_by_module.get_or_insert(module_id);

        exposes_ids.push(module_id);
    }

    exposes_ids
}

fn load_builtin_module_help<'a>(
    arena: &'a Bump,
    filename: &str,
    src_bytes: &'a str,
) -> (HeaderInfo<'a>, roc_parse::state::State<'a>) {
    let is_root_module = false;
    let opt_shorthand = None;
    let filename = PathBuf::from(filename);
    let parse_state = roc_parse::state::State::new(src_bytes.as_bytes());
    let parsed = roc_parse::header::parse_header(arena, parse_state.clone());

    match parsed {
        Ok((
            ast::SpacesBefore {
                item: ast::Header::Module(header),
                before: comments,
            },
            parse_state,
        )) => {
            let name_stem = arena.alloc_str(filename.file_stem().unwrap().to_str().unwrap());

            let info = HeaderInfo {
                filename,
                is_root_module,
                opt_shorthand,
                packages: &[],
                header_type: HeaderType::Builtin {
                    name: header::ModuleName::new(name_stem),
                    exposes: unspace(arena, header.exposes.items),
                    opt_params: header.params,
                },
                module_comments: comments,
                header_imports: header.interface_imports,
            };

            (info, parse_state)
        }
        Ok(_) => panic!("invalid header format for builtin module"),
        Err(e) => panic!("Hit a parse error in the header of {filename:?}:\n{e:?}"),
    }
}

fn load_builtin_module<'a>(
    arena: &'a Bump,
    module_ids: Arc<Mutex<PackageModuleIds<'a>>>,
    module_timing: ModuleTiming,
    module_id: ModuleId,
    module_name: &str,
) -> Result<(ModuleId, Msg<'a>), LoadingProblem<'a>> {
    let src_bytes = module_source(module_id);

    let (info, parse_state) = load_builtin_module_help(arena, module_name, src_bytes);

    let (module_id, _, header) = build_header(info, parse_state, module_ids, module_timing)?;
    Ok((module_id, Msg::Header(header)))
}

/// Load a module by its module name, rather than by its filename
fn load_module<'a>(
    arena: &'a Bump,
    src_dir: &Path,
    module_name: PQModuleName<'a>,
    module_ids: Arc<Mutex<PackageModuleIds<'a>>>,
    arc_shorthands: Arc<Mutex<MutMap<&'a str, ShorthandPath>>>,
    roc_cache_dir: RocCacheDir<'_>,
    ident_ids_by_module: SharedIdentIdsByModule,
) -> Result<HeaderOutput<'a>, LoadingProblem<'a>> {
    let module_start_time = Instant::now();

    let parse_start = Instant::now();
    let parse_header_duration = parse_start.elapsed();

    // Insert the first entries for this module's timings
    let mut module_timing = ModuleTiming::new(module_start_time);

    module_timing.read_roc_file = Default::default();
    module_timing.parse_header = parse_header_duration;

    macro_rules! load_builtins {
        ($($name:literal, $module_id:path)*) => {
            match module_name.unqualified().map(|name| name.as_str()) {
            $(
                Some($name) => {
                    let (module_id, msg) = load_builtin_module(
                        arena,
                        module_ids,
                        module_timing,
                        $module_id,
                        concat!($name, ".roc")
                    )?;

                    return Ok(HeaderOutput { module_id, msg, opt_platform_shorthand: None });
                }
            )*
                _ => { /* fall through */ }
            }}
    }

    load_builtins! {
        "Result", ModuleId::RESULT
        "List", ModuleId::LIST
        "Str", ModuleId::STR
        "Dict", ModuleId::DICT
        "Set", ModuleId::SET
        "Num", ModuleId::NUM
        "Bool", ModuleId::BOOL
        "Box", ModuleId::BOX
        "Encode", ModuleId::ENCODE
        "Decode", ModuleId::DECODE
        "Hash", ModuleId::HASH
        "Inspect", ModuleId::INSPECT
        "Task", ModuleId::TASK
    }

    let (filename, opt_shorthand) = module_name_to_path(src_dir, &module_name, arc_shorthands);

    load_filename(
        arena,
        filename,
        false,
        opt_shorthand,
        Some(module_name),
        module_ids,
        ident_ids_by_module,
        roc_cache_dir,
        module_start_time,
    )
}

#[derive(Debug)]
enum ShorthandPath {
    /// e.g. "/home/rtfeldman/.cache/roc/0.1.0/oUkxSOI9zFGtSoIaMB40QPdrXphr1p1780eiui2iO9Mz"
    #[allow(dead_code)]
    // wasm warns FromHttpsUrl is unused, but errors if it is removed ¯\_(ツ)_/¯
    FromHttpsUrl {
        /// e.g. "/home/rtfeldman/.cache/roc/0.1.0/oUkxSOI9zFGtSoIaMB40QPdrXphr1p1780eiui2iO9Mz"
        root_module_dir: PathBuf,
        /// e.g. "/home/rtfeldman/.cache/roc/0.1.0/oUkxSOI9zFGtSoIaMB40QPdrXphr1p1780eiui2iO9Mz/main.roc"
        root_module: PathBuf,
    },
    RelativeToSrc {
        /// e.g. "/home/rtfeldman/my-roc-code/examples/cli/cli-platform/"
        root_module_dir: PathBuf,
        /// e.g. "/home/rtfeldman/my-roc-code/examples/cli/cli-platform/main.roc"
        root_module: PathBuf,
    },
}

impl ShorthandPath {
    pub fn root_module(&self) -> &Path {
        match self {
            ShorthandPath::FromHttpsUrl { root_module, .. }
            | ShorthandPath::RelativeToSrc { root_module, .. } => root_module.as_path(),
        }
    }

    pub fn root_module_dir(&self) -> &Path {
        match self {
            ShorthandPath::FromHttpsUrl {
                root_module_dir, ..
            }
            | ShorthandPath::RelativeToSrc {
                root_module_dir, ..
            } => root_module_dir.as_path(),
        }
    }
}

fn module_name_to_path<'a>(
    src_dir: &Path,
    module_name: &PQModuleName<'a>,
    arc_shorthands: Arc<Mutex<MutMap<&'a str, ShorthandPath>>>,
) -> (PathBuf, Option<&'a str>) {
    let mut filename;
    let opt_shorthand;

    match module_name {
        PQModuleName::Unqualified(name) => {
            filename = src_dir.to_path_buf();

            opt_shorthand = None;
            // Convert dots in module name to directories
            for part in name.split(MODULE_SEPARATOR) {
                filename.push(part);
            }
        }
        PQModuleName::Qualified(shorthand, name) => {
            opt_shorthand = Some(*shorthand);
            let shorthands = arc_shorthands.lock();
            filename = shorthands
                .get(shorthand)
                .expect("All shorthands should have been validated by now.")
                .root_module_dir()
                .to_path_buf();

            // Convert dots in module name to directories
            for part in name.split(MODULE_SEPARATOR) {
                filename.push(part);
            }
        }
    }

    // End with .roc
    filename.set_extension(ROC_FILE_EXTENSION);

    (filename, opt_shorthand)
}

#[derive(Debug)]
struct HeaderOutput<'a> {
    module_id: ModuleId,
    msg: Msg<'a>,
    /// Only comes up if we're parsing an app module
    opt_platform_shorthand: Option<&'a str>,
}

pub enum RocFileErr {
    InvalidExtension,
    NotDotRocAndNoHashbangOnFirstLine,
    InvalidUtf8,
}

fn ensure_roc_file(filename: &Path, src_bytes: &[u8]) -> Result<(), RocFileErr> {
    match filename.extension() {
        Some(ext) => {
            if ext == ROC_FILE_EXTENSION {
                Ok(())
            } else {
                Err(RocFileErr::InvalidExtension)
            }
        }
        None => {
            let index = src_bytes
                .iter()
                .position(|a| *a == b'\n')
                .unwrap_or(src_bytes.len());
            let first_line_bytes = &src_bytes[0..index];

            if let Ok(first_line) = core::str::from_utf8(first_line_bytes) {
                if !(first_line.starts_with("#!") && first_line.contains("roc")) {
                    Err(RocFileErr::NotDotRocAndNoHashbangOnFirstLine)
                } else {
                    Ok(())
                }
            } else {
                Err(RocFileErr::InvalidUtf8)
            }
        }
    }
}

fn parse_header<'a>(
    arena: &'a Bump,
    read_file_duration: Duration,
    filename: PathBuf,
    is_root_module: bool,
    opt_shorthand: Option<&'a str>,
    opt_expected_module_name: Option<PackageQualified<'a, ModuleName>>,
    module_ids: Arc<Mutex<PackageModuleIds<'a>>>,
    ident_ids_by_module: SharedIdentIdsByModule,
    src_bytes: &'a [u8],
    roc_cache_dir: RocCacheDir<'_>,
    start_time: Instant,
) -> Result<HeaderOutput<'a>, LoadingProblem<'a>> {
    let parse_start = Instant::now();
    let parse_state = roc_parse::state::State::new(src_bytes);
    let parsed = roc_parse::header::parse_header(arena, parse_state.clone());
    let parse_header_duration = parse_start.elapsed();

    if let Err(problem) = ensure_roc_file(&filename, src_bytes) {
        let problem = match problem {
            // TODO we should print separate error messages for these
            RocFileErr::InvalidExtension | RocFileErr::NotDotRocAndNoHashbangOnFirstLine => {
                LoadingProblem::FileProblem {
                    filename,
                    error: io::ErrorKind::Unsupported,
                }
            }
            RocFileErr::InvalidUtf8 => LoadingProblem::FileProblem {
                filename,
                error: io::ErrorKind::InvalidData,
            },
        };

        return Err(problem);
    };

    // Insert the first entries for this module's timings
    let mut module_timing = ModuleTiming::new(start_time);

    module_timing.read_roc_file = read_file_duration;
    module_timing.parse_header = parse_header_duration;

    match parsed {
        Ok((
            ast::SpacesBefore {
                item: ast::Header::Module(header),
                before: comments,
            },
            parse_state,
        )) => {
            let module_name = match opt_expected_module_name {
                Some(pq_name) => arena.alloc_str(pq_name.as_inner().as_str()),
                None => {
                    // [modules-revamp] [privacy-changes] TODO: Support test/check on nested modules
                    arena.alloc_str(filename.file_stem().unwrap().to_str().unwrap())
                }
            };

            let info = HeaderInfo {
                filename,
                is_root_module,
                opt_shorthand,
                packages: &[],
                header_type: HeaderType::Module {
                    name: roc_parse::header::ModuleName::new(module_name),
                    exposes: unspace(arena, header.exposes.items),
                    opt_params: header.params,
                },
                module_comments: comments,
                header_imports: header.interface_imports,
            };

            let (module_id, _, header) =
                build_header(info, parse_state.clone(), module_ids, module_timing)?;

            Ok(HeaderOutput {
                module_id,
                msg: Msg::Header(header),
                opt_platform_shorthand: None,
            })
        }
        Ok((
            ast::SpacesBefore {
                item: ast::Header::Hosted(header),
                before: comments,
            },
            parse_state,
        )) => {
            let info = HeaderInfo {
                filename,
                is_root_module,
                opt_shorthand,
                packages: &[],
                header_type: HeaderType::Hosted {
                    name: header.name.value,
                    exposes: unspace(arena, header.exposes.item.items),
                },
                module_comments: comments,
                header_imports: Some(header.imports),
            };

            let (module_id, _, header) =
                build_header(info, parse_state, module_ids, module_timing)?;

            Ok(HeaderOutput {
                module_id,
                msg: Msg::Header(header),
                opt_platform_shorthand: None,
            })
        }
        Ok((
            ast::SpacesBefore {
                item: ast::Header::App(header),
                before: comments,
            },
            parse_state,
        )) => {
            let mut app_file_dir = filename.clone();
            app_file_dir.pop();

            let packages = unspace(arena, header.packages.value.items);

            let mut platform_shorthand = None;

            for package in packages.iter() {
                if package.value.platform_marker.is_some() {
                    if platform_shorthand.is_some() {
                        let mut module_ids = (*module_ids).lock();

                        let module_id = module_ids.get_or_insert(
                            arena.alloc(PQModuleName::Unqualified(ModuleName::APP.into())),
                        );

                        return Err(LoadingProblem::MultiplePlatformPackages {
                            module_id,
                            filename,
                            source: src_bytes,
                            region: header.packages.region,
                        });
                    }

                    platform_shorthand = Some(package.value.shorthand);
                }
            }

            let to_platform = match platform_shorthand {
                Some(shorthand) => To::ExistingPackage(shorthand),
                None => {
                    if let Some(new_pkg) = header.old_provides_to_new_package {
                        // This is a piece of old syntax that's used by the REPL and a number of
                        // tests where an ephemeral platform package is specified that is never
                        // loaded.
                        // We can remove this when we have upgraded everything to the new syntax.
                        To::NewPackage(new_pkg)
                    } else {
                        let mut module_ids = (*module_ids).lock();

                        let module_id = module_ids.get_or_insert(
                            arena.alloc(PQModuleName::Unqualified(ModuleName::APP.into())),
                        );

                        return Err(LoadingProblem::NoPlatformPackage {
                            module_id,
                            filename,
                            region: header.packages.region,
                            source: src_bytes,
                        });
                    }
                }
            };

            let mut provides = bumpalo::collections::Vec::new_in(arena);

            provides.extend(unspace(arena, header.provides.items));

            let info = HeaderInfo {
                filename,
                is_root_module,
                opt_shorthand,
                packages,
                header_type: HeaderType::App {
                    provides: provides.into_bump_slice(),
                    to_platform,
                },
                module_comments: comments,
                header_imports: header.old_imports,
            };

            let (module_id, _, resolved_header) =
                build_header(info, parse_state, module_ids.clone(), module_timing)?;

            let filename = resolved_header.module_path.clone();
            let mut messages = Vec::with_capacity(packages.len() + 1);

            // It's important that the app header is first in the list!
            messages.push(Msg::Header(resolved_header));

            load_packages(
                packages,
                &mut messages,
                roc_cache_dir,
                app_file_dir,
                arena,
                Some(module_id),
                module_ids,
                ident_ids_by_module,
                filename,
            );

            Ok(HeaderOutput {
                module_id,
                msg: Msg::Many(messages),
                opt_platform_shorthand: platform_shorthand,
            })
        }
        Ok((
            ast::SpacesBefore {
                item: ast::Header::Package(header),
                before: comments,
            },
            parse_state,
        )) => {
            let (module_id, _, header) = build_package_header(
                arena,
                None,
                is_root_module,
                filename,
                parse_state,
                module_ids,
                ident_ids_by_module,
                &header,
                comments,
                module_timing,
            )?;

            Ok(HeaderOutput {
                module_id,
                msg: Msg::Header(header),
                opt_platform_shorthand: None,
            })
        }

        Ok((
            ast::SpacesBefore {
                item: ast::Header::Platform(header),
                before: comments,
            },
            parse_state,
        )) => {
            let exposes_ids = get_exposes_ids(
                header.exposes.item.items,
                arena,
                &module_ids,
                &ident_ids_by_module,
            );

            let (module_id, _, header) = build_platform_header(
                arena,
                None,
                is_root_module,
                None,
                filename,
                parse_state,
                module_ids,
                exposes_ids.into_bump_slice(),
                &header,
                comments,
                module_timing,
            )?;

            Ok(HeaderOutput {
                module_id,
                msg: Msg::Header(header),
                opt_platform_shorthand: None,
            })
        }
        Err(fail) => Err(LoadingProblem::ParsingFailed(
            fail.map_problem(SyntaxError::Header)
                .into_file_error(filename),
        )),
    }
}

fn load_packages<'a>(
    packages: &[Loc<PackageEntry<'a>>],
    load_messages: &mut Vec<Msg<'a>>,
    roc_cache_dir: RocCacheDir,
    cwd: PathBuf,
    arena: &'a Bump,
    app_module_id: Option<ModuleId>,
    module_ids: Arc<Mutex<PackageModuleIds<'a>>>,
    ident_ids_by_module: SharedIdentIdsByModule,
    #[allow(unused_variables)] // for wasm
    filename: PathBuf,
) {
    // Load all the packages
    for Loc { value: entry, .. } in packages.iter() {
        let PackageEntry {
            shorthand,
            package_name:
                Loc {
                    value: package_name,
                    ..
                },
            ..
        } = entry;

        let src = package_name.to_str();

        // find the `package` or `platform` module on disk,
        // downloading it into a cache dir first if necessary.
        let root_module_path = if src.starts_with("https://") {
            #[cfg(not(target_family = "wasm"))]
            {
                // If this is a HTTPS package, synchronously download it
                // to the cache before proceeding.

                // TODO we should do this async; however, with the current
                // architecture of file.rs (which doesn't use async/await),
                // this would be very difficult!
                match cache::install_package(roc_cache_dir, src) {
                    Ok((package_dir, opt_root_module)) => {
                        // You can optionally specify the root module using the URL fragment,
                        // e.g. #foo.roc
                        // (defaults to main.roc)
                        match opt_root_module {
                            Some(root_module) => package_dir.join(root_module),
                            None => package_dir.join(DEFAULT_MAIN_NAME),
                        }
                    }
                    Err(problem) => {
                        let buf = to_https_problem_report_string(src, problem, filename);

                        load_messages.push(Msg::FailedToLoad(LoadingProblem::FormattedReport(buf)));
                        return;
                    }
                }
            }

            #[cfg(target_family = "wasm")]
            {
                panic!("Specifying packages via URLs is currently unsupported in wasm.");
            }
        } else {
            cwd.join(src)
        };

        match load_package_from_disk(
            arena,
            &root_module_path,
            shorthand,
            roc_cache_dir,
            app_module_id,
            module_ids.clone(),
            ident_ids_by_module.clone(),
        ) {
            Ok(msg) => {
                load_messages.push(msg);
            }
            Err(problem) => {
                load_messages.push(Msg::FailedToLoad(problem));
            }
        }
    }
}

/// Load a module by its filename
fn load_filename<'a>(
    arena: &'a Bump,
    filename: PathBuf,
    is_root_module: bool,
    opt_shorthand: Option<&'a str>,
    opt_expected_module_name: Option<PackageQualified<'a, ModuleName>>,
    module_ids: Arc<Mutex<PackageModuleIds<'a>>>,
    ident_ids_by_module: SharedIdentIdsByModule,
    roc_cache_dir: RocCacheDir<'_>,
    module_start_time: Instant,
) -> Result<HeaderOutput<'a>, LoadingProblem<'a>> {
    let file_io_start = Instant::now();
    let file = fs::read(&filename);
    let file_io_duration = file_io_start.elapsed();

    match file {
        Ok(bytes) => parse_header(
            arena,
            file_io_duration,
            filename,
            is_root_module,
            opt_shorthand,
            opt_expected_module_name,
            module_ids,
            ident_ids_by_module,
            arena.alloc(bytes),
            roc_cache_dir,
            module_start_time,
        ),
        Err(err) => Err(LoadingProblem::FileProblem {
            filename,
            error: err.kind(),
        }),
    }
}

/// Load a module from a str
/// the `filename` is never read, but used for the module name
fn load_from_str<'a>(
    arena: &'a Bump,
    filename: PathBuf,
    src: &'a str,
    module_ids: Arc<Mutex<PackageModuleIds<'a>>>,
    ident_ids_by_module: SharedIdentIdsByModule,
    roc_cache_dir: RocCacheDir<'_>,
    module_start_time: Instant,
) -> Result<HeaderOutput<'a>, LoadingProblem<'a>> {
    let file_io_start = Instant::now();
    let file_io_duration = file_io_start.elapsed();

    parse_header(
        arena,
        file_io_duration,
        filename,
        false,
        None,
        None,
        module_ids,
        ident_ids_by_module,
        src.as_bytes(),
        roc_cache_dir,
        module_start_time,
    )
}

#[derive(Debug)]
struct HeaderInfo<'a> {
    filename: PathBuf,
    is_root_module: bool,
    opt_shorthand: Option<&'a str>,
    packages: &'a [Loc<PackageEntry<'a>>],
    header_type: HeaderType<'a>,
    module_comments: &'a [CommentOrNewline<'a>],
    header_imports: Option<ImportsKeywordItem<'a>>,
}

fn build_header<'a>(
    info: HeaderInfo<'a>,
    parse_state: roc_parse::state::State<'a>,
    module_ids: Arc<Mutex<PackageModuleIds<'a>>>,
    module_timing: ModuleTiming,
) -> Result<(ModuleId, PQModuleName<'a>, ModuleHeader<'a>), LoadingProblem<'a>> {
    let HeaderInfo {
        filename,
        is_root_module,
        opt_shorthand,
        packages,
        header_type,
        module_comments: header_comments,
        header_imports,
    } = info;

    let declared_name: ModuleName = match &header_type {
        HeaderType::App { .. } => ModuleName::APP.into(),
        HeaderType::Platform { .. } => {
            // Platform modules do not have names. This is important because otherwise
            // those names end up in host-generated symbols, and they may contain
            // characters that hosts might not allow in their function names.
            String::new().into()
        }
        HeaderType::Package { .. } => {
            // Package modules do not have names.
            String::new().into()
        }
        HeaderType::Module { name, .. }
        | HeaderType::Builtin { name, .. }
        | HeaderType::Hosted { name, .. } => name.as_str().into(),
    };

    let (name, home) = {
        // Lock just long enough to perform the minimal operations necessary.
        let mut module_ids = (*module_ids).lock();

        let name = match opt_shorthand {
            Some(shorthand) => PQModuleName::Qualified(shorthand, declared_name),
            None => PQModuleName::Unqualified(declared_name),
        };
        let home = module_ids.get_or_insert(&name);

        (name, home)
    };

    let package_entries = packages
        .iter()
        .map(|Loc { value: pkg, .. }| (pkg.shorthand, pkg.package_name.value))
        .collect::<MutMap<_, _>>();

    Ok((
        home,
        name,
        ModuleHeader {
            module_id: home,
            module_path: filename,
            is_root_module,
            packages: package_entries,
            parse_state,
            header_type: header_type.to_maybe_builtin(home),
            header_comments,
            module_timing,
            opt_shorthand,
            header_imports,
        },
    ))
}

impl<'a> BuildTask<'a> {
    // TODO trim down these arguments - possibly by moving Constraint into Module
    fn solve_module(
        module: Module,
        ident_ids: IdentIds,
        module_timing: ModuleTiming,
        types: Types,
        constraints: Constraints,
        constraint: ConstraintSoa,
        function_kind: FunctionKind,
        pending_derives: PendingDerives,
        var_store: VarStore,
        imported_modules: MutMap<ModuleId, Region>,
        exposed_types: &ExposedByModule,
        dep_idents: IdentIdsByModule,
        declarations: Declarations,
        cached_subs: CachedTypeState,
        derived_module: SharedDerivedModule,
        exec_mode: ExecutionMode,
        is_host_exposed: bool,

        #[cfg(debug_assertions)] checkmate: Option<roc_checkmate::Collector>,
    ) -> Self {
        let exposed_by_module = exposed_types.retain_modules(imported_modules.keys());

        let exposed_for_module =
            ExposedForModule::new(module.referenced_values.iter(), exposed_by_module);

        Self::Solve {
            module,
            ident_ids,
            exposed_for_module,
            types,
            constraints,
            constraint,
            function_kind,
            pending_derives,
            var_store,
            declarations,
            dep_idents,
            module_timing,
            cached_subs,
            derived_module,
            exec_mode,
            is_host_exposed,

            #[cfg(debug_assertions)]
            checkmate,
        }
    }
}

fn synth_import(subs: &mut Subs, content: roc_types::subs::Content) -> Variable {
    use roc_types::subs::{Descriptor, Mark, OptVariable, Rank};
    subs.fresh(Descriptor {
        content,
        rank: Rank::import(),
        mark: Mark::NONE,
        copy: OptVariable::NONE,
    })
}

fn synth_list_len_type(subs: &mut Subs) -> Variable {
    use roc_types::subs::{Content, FlatType, LambdaSet, OptVariable, SubsSlice, UnionLabels};

    // List.len : List a -> U64
    let a = synth_import(subs, Content::FlexVar(None));
    let a_slice = slice_extend_new(&mut subs.variables, [a]);
    let list_a = synth_import(
        subs,
        Content::Structure(FlatType::Apply(Symbol::LIST_LIST, a_slice)),
    );
    let fn_var = synth_import(subs, Content::Error);
    let solved_list_len = UnionLabels::insert_into_subs(subs, [(Symbol::LIST_LEN_U64, [])]);
    let clos_list_len = synth_import(
        subs,
        Content::LambdaSet(LambdaSet {
            solved: solved_list_len,
            recursion_var: OptVariable::NONE,
            unspecialized: SubsSlice::default(),
            ambient_function: fn_var,
        }),
    );
    let fn_args_slice = slice_extend_new(&mut subs.variables, [list_a]);
    subs.set_content(
        fn_var,
        Content::Structure(FlatType::Func(
            fn_args_slice,
            clos_list_len,
            Variable::U64,
            Variable::PURE,
        )),
    );
    fn_var
}

pub fn add_imports(
    my_module: ModuleId,
    constraints: &mut Constraints,
    subs: &mut Subs,
    mut pending_abilities: PendingAbilitiesStore,
    exposed_for_module: &ExposedForModule,
    def_types: &mut Vec<(Symbol, Loc<TypeOrVar>)>,
    imported_rigid_vars: &mut Vec<Variable>,
    imported_flex_vars: &mut Vec<Variable>,
) -> (Vec<Variable>, VecMap<ModuleId, Variable>, AbilitiesStore) {
    let mut import_variables = Vec::new();

    let mut cached_symbol_vars = VecMap::default();

    let imported_values = exposed_for_module.imported_values.iter().copied();

    for symbol in imported_values {
        import_variable_for_symbol(
            subs,
            constraints,
            def_types,
            &mut import_variables,
            imported_rigid_vars,
            imported_flex_vars,
            &mut cached_symbol_vars,
            &exposed_for_module.exposed_by_module,
            symbol,
            OnSymbolNotFound::AssertIsBuiltin,
        );
    }

    // Patch used symbols from circular dependencies.
    if my_module == ModuleId::NUM {
        // Num needs List.len, but List imports Num.
        let list_len_type_var = synth_list_len_type(subs);
        let list_len_type_index = constraints.push_variable(list_len_type_var);
        def_types.push((Symbol::LIST_LEN_U64, Loc::at_zero(list_len_type_index)));
        import_variables.push(list_len_type_var);
    }

    // Fill in the implementation information of the abilities from the modules we import, which we
    // now know because all imported modules should be solved by now.
    //
    // TODO: see if we can reduce the amount of specializations we need to import.
    // One idea is to just always assume external modules fulfill their specialization obligations
    // and save lambda set resolution for mono.
    for (_, module_types) in exposed_for_module.exposed_by_module.iter_all() {
        for (impl_key, resolved_impl) in module_types.resolved_implementations.iter() {
            pending_abilities.import_implementation(*impl_key, resolved_impl);
        }
    }

    struct Ctx<'a> {
        subs: &'a mut Subs,
        exposed_by_module: &'a ExposedByModule,
        imported_variables: &'a mut Vec<Variable>,
        imported_rigids: &'a mut Vec<Variable>,
        imported_flex: &'a mut Vec<Variable>,
    }

    let abilities_store = pending_abilities.resolve_for_module(
        my_module,
        &mut Ctx {
            subs,
            exposed_by_module: &exposed_for_module.exposed_by_module,
            imported_variables: &mut import_variables,
            imported_rigids: imported_rigid_vars,
            imported_flex: imported_flex_vars,
        },
        |ctx, symbol| match cached_symbol_vars.get(&symbol).copied() {
            Some(var) => var,
            None => {
                import_variable_for_symbol(
                    ctx.subs,
                    constraints,
                    def_types,
                    ctx.imported_variables,
                    ctx.imported_rigids,
                    ctx.imported_flex,
                    &mut cached_symbol_vars,
                    &exposed_for_module.exposed_by_module,
                    symbol,
                    OnSymbolNotFound::AbilityMemberMustBeAvailable,
                );

                *cached_symbol_vars.get(&symbol).unwrap()
            }
        },
        |ctx, module, lset_var| match ctx.exposed_by_module.get(&module) {
            Some(ExposedModuleTypes {
                exposed_types_storage_subs: exposed_types,
                resolved_implementations: _,
            }) => {
                let var = exposed_types
                    .stored_specialization_lambda_set_vars
                    .get(&lset_var)
                    .expect("Lambda set var from other module not available");

                let copied_import = exposed_types
                    .storage_subs
                    .export_variable_to(ctx.subs, *var);

                #[allow(clippy::let_and_return)]
                let copied_import_var = extend_imports_data_with_copied_import(
                    copied_import,
                    ctx.imported_variables,
                    ctx.imported_rigids,
                    ctx.imported_flex,
                );

                copied_import_var
            }
            None => internal_error!("Imported module {:?} is not available", module),
        },
    );

    let mut imported_param_vars = VecMap::default();

    for (module_id, _) in exposed_for_module.exposed_by_module.iter_all() {
        let ExposedModuleTypes {
            exposed_types_storage_subs: exposed_types,
            ..
        } = exposed_for_module.exposed_by_module.get(module_id).unwrap();

        if let Some(stored_aprams_var) = exposed_types.stored_params_var {
            let copied_import = exposed_types
                .storage_subs
                .export_variable_to(subs, stored_aprams_var);

            let copied_import_var = extend_imports_data_with_copied_import(
                copied_import,
                &mut import_variables,
                imported_rigid_vars,
                imported_flex_vars,
            );

            imported_param_vars.insert(*module_id, copied_import_var);
        }
    }

    (import_variables, imported_param_vars, abilities_store)
}

enum OnSymbolNotFound {
    AssertIsBuiltin,
    AbilityMemberMustBeAvailable,
}

fn extend_imports_data_with_copied_import(
    copied_import: CopiedImport,
    imported_variables: &mut Vec<Variable>,
    imported_rigids: &mut Vec<Variable>,
    imported_flex: &mut Vec<Variable>,
) -> Variable {
    // not a typo; rigids are turned into flex during type inference, but when imported we must
    // consider them rigid variables
    imported_rigids.extend(copied_import.rigid);
    imported_flex.extend(copied_import.flex);

    // Rigid vars bound to abilities are also treated like rigids.
    imported_rigids.extend(copied_import.rigid_able);
    imported_flex.extend(copied_import.flex_able);

    imported_variables.extend(copied_import.registered);

    copied_import.variable
}

fn import_variable_for_symbol(
    subs: &mut Subs,
    constraints: &mut Constraints,
    def_types: &mut Vec<(Symbol, Loc<TypeOrVar>)>,
    imported_variables: &mut Vec<Variable>,
    imported_rigids: &mut Vec<Variable>,
    imported_flex: &mut Vec<Variable>,
    cached_symbol_vars: &mut VecMap<Symbol, Variable>,
    exposed_by_module: &ExposedByModule,
    symbol: Symbol,
    on_symbol_not_found: OnSymbolNotFound,
) {
    let module_id = symbol.module_id();
    match exposed_by_module.get(&module_id) {
        Some(ExposedModuleTypes {
            exposed_types_storage_subs: exposed_types,
            resolved_implementations: _,
        }) => {
            let variable = match exposed_types
                .stored_vars_by_symbol
                .iter()
                .find(|(s, _)| **s == symbol)
            {
                None => {
                    use OnSymbolNotFound::*;
                    match on_symbol_not_found {
                        AssertIsBuiltin => {
                            // Today we define builtins in each module that uses them
                            // so even though they have a different module name from
                            // the surrounding module, they are not technically imported
                            debug_assert!(symbol.is_builtin(), "The symbol {:?} was not found and was assumed to be a builtin, but it wasn't a builtin.", symbol);
                            return;
                        }
                        AbilityMemberMustBeAvailable => {
                            internal_error!("Import ability member {:?} not available", symbol);
                        }
                    }
                }
                Some((_, x)) => *x,
            };

            let copied_import = exposed_types
                .storage_subs
                .export_variable_to(subs, variable);

            let copied_import_var = extend_imports_data_with_copied_import(
                copied_import,
                imported_variables,
                imported_rigids,
                imported_flex,
            );

            let copied_import_index = constraints.push_variable(copied_import_var);

            def_types.push((symbol, Loc::at_zero(copied_import_index)));

            cached_symbol_vars.insert(symbol, copied_import_var);
        }
        None => {
            internal_error!("Imported module {:?} is not available", module_id)
        }
    }
}

struct SolveResult {
    solved: Solved<Subs>,
    solved_implementations: ResolvedImplementations,
    exposed_vars_by_symbol: Vec<(Symbol, Variable)>,
    problems: Vec<TypeError>,
    abilities_store: AbilitiesStore,
    imported_modules_with_params: Vec<ModuleId>,

    #[cfg(debug_assertions)]
    checkmate: Option<roc_checkmate::Collector>,
}

#[allow(clippy::complexity)]
fn run_solve_solve(
    exposed_for_module: ExposedForModule,
    mut types: Types,
    mut constraints: Constraints,
    constraint: ConstraintSoa,
    function_kind: FunctionKind,
    pending_derives: PendingDerives,
    var_store: VarStore,
    module: Module,
    derived_module: SharedDerivedModule,
    is_host_exposed: bool,

    #[cfg(debug_assertions)] checkmate: Option<roc_checkmate::Collector>,
) -> SolveResult {
    let Module {
        exposed_symbols,
        aliases,
        rigid_variables,
        abilities_store: pending_abilities,
        module_params,
        ..
    } = module;

    let mut imported_rigid_vars: Vec<Variable> = Vec::new();
    let mut imported_flex_vars: Vec<Variable> = Vec::new();
    let mut def_types: Vec<(Symbol, Loc<TypeOrVar>)> = Vec::new();

    let mut subs = Subs::new_from_varstore(var_store);

    let (import_variables, imported_param_vars, abilities_store) = add_imports(
        module.module_id,
        &mut constraints,
        &mut subs,
        pending_abilities,
        &exposed_for_module,
        &mut def_types,
        &mut imported_rigid_vars,
        &mut imported_flex_vars,
    );

    let imported_modules_with_params = imported_param_vars
        .keys()
        .copied()
        .collect::<Vec<ModuleId>>();

    let actual_constraint = constraints.let_import_constraint(
        imported_rigid_vars,
        imported_flex_vars,
        def_types,
        constraint,
        &import_variables,
    );

    let mut solve_aliases = roc_solve::Aliases::with_capacity(aliases.len());
    for (name, (_, alias)) in aliases.iter() {
        solve_aliases.insert(&mut types, *name, alias.clone());
    }

    let (solve_output, solved_implementations, exposed_vars_by_symbol) = {
        let module_id = module.module_id;

        let host_exposed_idents = if is_host_exposed {
            Some(&exposed_symbols)
        } else {
            None
        };

        let solve_config = SolveConfig {
            home: module_id,
            types,
            constraints: &constraints,
            root_constraint: actual_constraint,
            function_kind,
            pending_derives,
            exposed_by_module: &exposed_for_module.exposed_by_module,
            derived_module,
            #[cfg(debug_assertions)]
            checkmate,
            module_params,
            module_params_vars: imported_param_vars,
            host_exposed_symbols: host_exposed_idents,
        };

        let solve_output = roc_solve::module::run_solve(
            solve_config,
            rigid_variables,
            subs,
            solve_aliases,
            abilities_store,
        );

        let solved_implementations =
            extract_module_owned_implementations(module_id, &solve_output.resolved_abilities_store);

        let is_specialization_symbol = |sym| {
            solved_implementations
                .values()
                .any(|resolved_impl| match resolved_impl {
                    ResolvedImpl::Impl(specialization) => specialization.symbol == sym,
                    ResolvedImpl::Error => false,
                })
        };

        // Expose anything that is explicitly exposed by the header, or is a specialization of an
        // ability.
        let exposed_vars_by_symbol: Vec<_> = solve_output
            .scope
            .vars_by_symbol()
            .filter(|(k, _)| {
                exposed_symbols.contains(k)
                    || is_specialization_symbol(*k)
                    || k.is_exposed_for_builtin_derivers()
            })
            .collect();

        (solve_output, solved_implementations, exposed_vars_by_symbol)
    };

    let roc_solve::module::SolveOutput {
        subs,
        scope: _,
        errors,
        resolved_abilities_store,

        #[cfg(debug_assertions)]
        checkmate,
    } = solve_output;

    SolveResult {
        solved: subs,
        solved_implementations,
        exposed_vars_by_symbol,
        problems: errors,
        abilities_store: resolved_abilities_store,
        imported_modules_with_params,

        #[cfg(debug_assertions)]
        checkmate,
    }
}

fn run_solve<'a>(
    module: Module,
    ident_ids: IdentIds,
    mut module_timing: ModuleTiming,
    exposed_for_module: ExposedForModule,
    types: Types,
    constraints: Constraints,
    constraint: ConstraintSoa,
    function_kind: FunctionKind,
    pending_derives: PendingDerives,
    var_store: VarStore,
    decls: Declarations,
    dep_idents: IdentIdsByModule,
    cached_types: CachedTypeState,
    derived_module: SharedDerivedModule,
    exec_mode: ExecutionMode,
    is_host_exposed: bool,

    #[cfg(debug_assertions)] checkmate: Option<roc_checkmate::Collector>,
) -> Msg<'a> {
    let solve_start = Instant::now();

    let module_id = module.module_id;

    // TODO remove when we write builtins in roc
    let aliases = module.aliases.clone();

    let opt_params_var = module.module_params.as_ref().map(|params| params.whole_var);
    let home_has_params = opt_params_var.is_some();

    let mut module = module;
    let loc_expects = std::mem::take(&mut module.loc_expects);
    let loc_dbgs = std::mem::take(&mut module.loc_dbgs);
    let module = module;

    let solve_result = {
        if module_id.is_builtin() {
            match cached_types.lock().remove(&module_id) {
                None => run_solve_solve(
                    exposed_for_module,
                    types,
                    constraints,
                    constraint,
                    function_kind,
                    pending_derives,
                    var_store,
                    module,
                    derived_module,
                    is_host_exposed,
                    //
                    #[cfg(debug_assertions)]
                    checkmate,
                ),
                Some(TypeState {
                    subs,
                    exposed_vars_by_symbol,
                    abilities,
                    solved_implementations,
                }) => SolveResult {
                    solved: Solved(subs),
                    solved_implementations,
                    exposed_vars_by_symbol,
                    problems: vec![],
                    abilities_store: abilities,
                    imported_modules_with_params: vec![],

                    #[cfg(debug_assertions)]
                    checkmate: None,
                },
            }
        } else {
            run_solve_solve(
                exposed_for_module,
                types,
                constraints,
                constraint,
                function_kind,
                pending_derives,
                var_store,
                module,
                derived_module,
                is_host_exposed,
                //
                #[cfg(debug_assertions)]
                checkmate,
            )
        }
    };

    let SolveResult {
        solved: mut solved_subs,
        solved_implementations,
        exposed_vars_by_symbol,
        mut problems,
        abilities_store,
        imported_modules_with_params,

        #[cfg(debug_assertions)]
        checkmate,
    } = solve_result;

    let exposed_types = roc_solve::module::exposed_types_storage_subs(
        module_id,
        &mut solved_subs,
        &exposed_vars_by_symbol,
        opt_params_var,
        &solved_implementations,
        &abilities_store,
    );

    match exec_mode {
        ExecutionMode::Check => {
            // Params are not lowered in check mode
        }
        ExecutionMode::Executable | ExecutionMode::ExecutableIfCheck | ExecutionMode::Test => {
            roc_lower_params::type_error::remove_module_param_arguments(
                &mut problems,
                home_has_params,
                &decls.symbols,
                imported_modules_with_params,
            );
        }
    }

    let solved_module = SolvedModule {
        exposed_vars_by_symbol,
        problems,
        aliases,
        solved_implementations,
        exposed_types,
    };

    // Record the final timings
    let solve_end = Instant::now();
    module_timing.solve = solve_end.duration_since(solve_start);

    // Send the subs to the main thread for processing,
    Msg::SolvedTypes {
        module_id,
        solved_subs,
        ident_ids,
        decls,
        dep_idents,
        solved_module,
        module_timing,
        abilities_store,
        loc_expects,
        loc_dbgs,

        #[cfg(debug_assertions)]
        checkmate,
    }
}

fn unspace<'a, T: Copy>(arena: &'a Bump, items: &[Loc<Spaced<'a, T>>]) -> &'a [Loc<T>] {
    bumpalo::collections::Vec::from_iter_in(
        items
            .iter()
            .map(|item| Loc::at(item.region, item.value.extract_spaces().item)),
        arena,
    )
    .into_bump_slice()
}

fn build_package_header<'a>(
    arena: &'a Bump,
    opt_shorthand: Option<&'a str>,
    is_root_module: bool,
    filename: PathBuf,
    parse_state: roc_parse::state::State<'a>,
    module_ids: Arc<Mutex<PackageModuleIds<'a>>>,
    ident_ids_by_module: SharedIdentIdsByModule,
    header: &PackageHeader<'a>,
    comments: &'a [CommentOrNewline<'a>],
    module_timing: ModuleTiming,
) -> Result<(ModuleId, PQModuleName<'a>, ModuleHeader<'a>), LoadingProblem<'a>> {
    let exposes = bumpalo::collections::Vec::from_iter_in(
        unspace(arena, header.exposes.items).iter().copied(),
        arena,
    );
    let packages = unspace(arena, header.packages.value.items);
    let exposes_ids = get_exposes_ids(
        header.exposes.items,
        arena,
        &module_ids,
        &ident_ids_by_module,
    );
    let header_type = HeaderType::Package {
        // A config_shorthand of "" should be fine
        config_shorthand: opt_shorthand.unwrap_or_default(),
        exposes: exposes.into_bump_slice(),
        exposes_ids: exposes_ids.into_bump_slice(),
    };

    let info = HeaderInfo {
        filename,
        is_root_module,
        opt_shorthand,
        packages,
        header_type,
        module_comments: comments,
        header_imports: None,
    };

    build_header(info, parse_state, module_ids, module_timing)
}

fn build_platform_header<'a>(
    arena: &'a Bump,
    opt_shorthand: Option<&'a str>,
    is_root_module: bool,
    opt_app_module_id: Option<ModuleId>,
    filename: PathBuf,
    parse_state: roc_parse::state::State<'a>,
    module_ids: Arc<Mutex<PackageModuleIds<'a>>>,
    exposes_ids: &'a [ModuleId],
    header: &PlatformHeader<'a>,
    comments: &'a [CommentOrNewline<'a>],
    module_timing: ModuleTiming,
) -> Result<(ModuleId, PQModuleName<'a>, ModuleHeader<'a>), LoadingProblem<'a>> {
    let requires = header
        .requires
        .item
        .signatures
        .map_items(arena, |item| {
            Loc::at(item.region, item.extract_spaces().item)
        })
        .items;
    let provides = bumpalo::collections::Vec::from_iter_in(
        unspace(arena, header.provides.item.items).iter().copied(),
        arena,
    );
    let packages = unspace(arena, header.packages.item.items);
    let exposes = bumpalo::collections::Vec::from_iter_in(
        unspace(arena, header.exposes.item.items).iter().copied(),
        arena,
    );
    let requires_types = unspace(arena, header.requires.item.rigids.items);

    let header_type = HeaderType::Platform {
        // A config_shorthand of "" should be fine
        config_shorthand: opt_shorthand.unwrap_or_default(),
        exposes_ids,
        opt_app_module_id,
        provides: provides.into_bump_slice(),
        exposes: exposes.into_bump_slice(),
        requires,
        requires_types,
    };

    let info = HeaderInfo {
        filename,
        is_root_module,
        opt_shorthand,
        packages,
        header_type,
        module_comments: comments,
        header_imports: Some(header.imports),
    };

    build_header(info, parse_state, module_ids, module_timing)
}

#[allow(clippy::unnecessary_wraps)]
fn canonicalize_and_constrain<'a>(
    arena: &'a Bump,
    qualified_module_ids: &'a PackageModuleIds<'a>,
    dep_idents: IdentIdsByModule,
    exposed_symbols: VecSet<Symbol>,
    aliases: MutMap<Symbol, Alias>,
    imported_abilities_state: PendingAbilitiesStore,
    parsed: ParsedModule<'a>,
    skip_constraint_gen: bool,
    exposed_module_ids: &[ModuleId],
    exec_mode: ExecutionMode,
    imported_module_params: VecMap<ModuleId, ModuleParams>,
    fx_mode: FxMode,
) -> CanAndCon {
    let canonicalize_start = Instant::now();

    let ParsedModule {
        module_id,
        module_path,
        src,
        header_type,
        exposed_ident_ids,
        parsed_defs,
        initial_scope,
        available_modules,
        mut module_timing,
        symbols_from_requires,
        opt_shorthand,
        ..
    } = parsed;

    // _before has an underscore because it's unused in --release builds
    let _before = roc_types::types::get_type_clone_count();

    let parsed_defs_for_docs = parsed_defs.clone();
    let parsed_defs = arena.alloc(parsed_defs);

    let mut var_store = VarStore::default();

    let fx_mode = if module_id.is_builtin() {
        // Allow builtins to expose effectful functions
        // even if platform is `Task`-based
        FxMode::PurityInference
    } else {
        fx_mode
    };

    let mut module_output = canonicalize_module_defs(
        arena,
        parsed_defs,
        &header_type,
        module_id,
        &*arena.alloc(module_path.to_string_lossy()),
        src,
        qualified_module_ids,
        exposed_ident_ids,
        &dep_idents,
        aliases,
        imported_abilities_state,
        initial_scope,
        exposed_symbols,
        &symbols_from_requires,
        &mut var_store,
        opt_shorthand,
        fx_mode,
    );

    let mut types = Types::new();

    // _after has an underscore because it's unused in --release builds
    let _after = roc_types::types::get_type_clone_count();

    log!(
        "canonicalize of {:?} cloned Type {} times ({} -> {})",
        module_id,
        _after - _before,
        _before,
        _after
    );

    let canonicalize_end = Instant::now();

    module_timing.canonicalize = canonicalize_end.duration_since(canonicalize_start);

    // Generate documentation information
    // TODO: store timing information?
    let module_docs = {
        let module_name = header_type.get_name();
        module_name.map(|module_name| {
            let mut scope = module_output.scope.clone();
            scope.add_docs_imports();
            crate::docs::generate_module_docs(
                scope,
                module_id,
                arena.alloc(qualified_module_ids.clone().into_module_ids()),
                module_name.into(),
                &parsed_defs_for_docs,
                exposed_module_ids,
                module_output.exposed_symbols.clone(),
                parsed.header_comments,
            )
        })
    };

    // _before has an underscore because it's unused in --release builds
    let _before = roc_types::types::get_type_clone_count();

    match exec_mode {
        ExecutionMode::Check => {
            // No need to lower params for `roc check` and lang server
            // If we did, we'd have to update the language server to exclude the extra arguments
        }
        ExecutionMode::Executable | ExecutionMode::ExecutableIfCheck | ExecutionMode::Test => {
            // We need to lower params only if the current module has any or imports at least one with params
            if module_output.module_params.is_some() || !imported_module_params.is_empty() {
                roc_lower_params::lower::lower(
                    module_id,
                    &module_output.module_params,
                    imported_module_params,
                    &mut module_output.declarations,
                    &mut module_output.scope.locals.ident_ids,
                    &mut var_store,
                );
            }
        }
    }

    let mut constraints = Constraints::new();

    let constraint = if skip_constraint_gen {
        roc_can::constraint::Constraint::True
    } else {
        constrain_module(
            &mut types,
            &mut constraints,
            module_output.symbols_from_requires,
            &module_output.scope.abilities_store,
            &module_output.declarations,
            &module_output.module_params,
            module_id,
        )
    };

    // _after has an underscore because it's unused in --release builds
    let _after = roc_types::types::get_type_clone_count();

    log!(
        "constraint gen of {:?} cloned Type {} times ({} -> {})",
        module_id,
        _after - _before,
        _before,
        _after
    );

    // scope has imported aliases, but misses aliases from inner scopes
    // module_output.aliases does have those aliases, so we combine them
    let mut aliases: MutMap<Symbol, (bool, Alias)> = module_output
        .aliases
        .into_iter()
        .map(|(k, v)| (k, (true, v)))
        .collect();

    for (name, alias) in module_output.scope.aliases {
        match aliases.entry(name) {
            Occupied(_) => {
                // do nothing
            }
            Vacant(vacant) => {
                let should_include_builtin = matches!(
                    name.module_id(),
                    ModuleId::ENCODE
                        | ModuleId::DECODE
                        | ModuleId::DICT
                        | ModuleId::SET
                        | ModuleId::HASH
                        | ModuleId::INSPECT
                        | ModuleId::TASK
                );

                if !name.is_builtin() || should_include_builtin {
                    vacant.insert((false, alias));
                }
            }
        }
    }

    let module = Module {
        module_id,
        exposed_imports: module_output.exposed_imports,
        exposed_symbols: module_output.exposed_symbols,
        referenced_values: module_output.referenced_values,
        aliases,
        rigid_variables: module_output.rigid_variables,
        abilities_store: module_output.scope.abilities_store,
        loc_expects: module_output.loc_expects,
        loc_dbgs: module_output.loc_dbgs,
        module_params: module_output.module_params,
    };

    let constrained_module = ConstrainedModule {
        module,
        declarations: module_output.declarations,
        available_modules,
        var_store,
        constraints,
        constraint,
        ident_ids: module_output.scope.locals.ident_ids,
        dep_idents,
        module_timing,
        types,
        pending_derives: module_output.pending_derives,
    };

    CanAndCon {
        constrained_module,
        canonicalization_problems: module_output.problems,
        module_docs,
    }
}

fn parse<'a>(
    arena: &'a Bump,
    header: ModuleHeader<'a>,
    arc_shorthands: Arc<Mutex<MutMap<&'a str, ShorthandPath>>>,
    module_ids: Arc<Mutex<PackageModuleIds<'a>>>,
    ident_ids_by_module: SharedIdentIdsByModule,
    root_type: RootType,
) -> Result<Msg<'a>, LoadingProblem<'a>> {
    let mut module_timing = header.module_timing;
    let parse_start = Instant::now();
    let source = header.parse_state.original_bytes();
    let parse_state = header.parse_state;

    let header_import_defs =
        roc_parse::ast::Header::header_imports_to_defs(arena, header.header_imports);

    let parsed_defs = match parse_module_defs(arena, parse_state.clone(), header_import_defs) {
        Ok(success) => success,
        Err(fail) => {
            return Err(LoadingProblem::ParsingFailed(
                fail.into_file_error(header.module_path, &parse_state),
            ));
        }
    };

    // SAFETY: By this point we've already incrementally verified that there
    // are no UTF-8 errors in these bytes. If there had been any UTF-8 errors,
    // we'd have bailed out before now.
    let src = unsafe { from_utf8_unchecked(source) };

    // Record the parse end time once, to avoid checking the time a second time
    // immediately afterward (for the beginning of canonicalization).
    let parse_end = Instant::now();

    module_timing.parse_body = parse_end.duration_since(parse_start);

    let mut available_modules: MutMap<ModuleId, Region> = MutMap::default();
    let exposed_values = header.header_type.exposed_or_provided_values();
    let num_exposes = exposed_values.len();
    let mut deps_by_name: MutMap<PQModuleName, ModuleId> =
        HashMap::with_capacity_and_hasher(num_exposes, default_hasher());

    match &header.header_type {
        HeaderType::Platform {
            opt_app_module_id, ..
        } => {
            // Add standard imports, if there is an app module.
            // (There might not be, e.g. when running `roc check myplatform.roc` or
            // when generating bindings.)
            if let Some(app_module_id) = opt_app_module_id {
                available_modules.insert(*app_module_id, Region::zero());
                deps_by_name.insert(
                    PQModuleName::Unqualified(ModuleName::APP.into()),
                    *app_module_id,
                );
            }
        }
        HeaderType::App { .. }
        | HeaderType::Package { .. }
        | HeaderType::Module { .. }
        | HeaderType::Builtin { .. }
        | HeaderType::Hosted { .. } => {}
    };

    let mut imported: Vec<(QualifiedModuleName, Region)> = vec![];
    let mut used_shorthands = VecMap::default();

    for (def, region) in ast::RecursiveValueDefIter::new(&parsed_defs) {
        if let ValueDef::ModuleImport(import) = def {
            imported.push((import.name.value.into(), *region));

            if let Some(shorthand) = import.name.value.package {
                used_shorthands.insert(shorthand, import.name.region);
            }
        }
    }

    // THEORY: a significant number of modules do not import from packages
    // so we can avoid locking the shorthands in those cases
    if !used_shorthands.is_empty() {
        let shorthands = arc_shorthands.lock();

        for (shorthand, region) in used_shorthands {
            if !shorthands.contains_key(shorthand) {
                let available =
                    AvailableShorthands::new(root_type, shorthands.keys().copied().collect());

                return Err(LoadingProblem::UnrecognizedPackageShorthand {
                    filename: header.module_path,
                    module_id: header.module_id,
                    source: src,
                    region,
                    shorthand,
                    available,
                });
            }
        }
    }

    let mut exposed: Vec<Symbol> = Vec::with_capacity(num_exposes);

    // Make sure the module_ids has ModuleIds for all our deps,
    // then record those ModuleIds in can_module_ids for later.
    {
        // Lock just long enough to perform the minimal operations necessary.
        let mut module_ids = module_ids.lock();

        // For each of our imports, add an entry to deps_by_name
        //
        // e.g. for `import pf.Foo exposing [bar]`, add `Foo` to deps_by_name
        //
        // Also build a list of imported_values_to_expose (like `bar` above.)
        for (qualified_module_name, region) in imported.into_iter() {
            let pq_module_name = qualified_module_name.into_pq_module_name(header.opt_shorthand);
            let module_id = module_ids.get_or_insert(&pq_module_name);

            available_modules.insert(module_id, region);
            deps_by_name.insert(pq_module_name, module_id);
        }
    }

    let mut scope: MutMap<Ident, (Symbol, Region)> = HashMap::with_hasher(default_hasher());
    let symbols_from_requires;

    let ident_ids = {
        // Lock just long enough to perform the minimal operations necessary.
        let mut ident_ids_by_module = (*ident_ids_by_module).lock();

        // Ensure this module has an entry in the ident_ids_by_module map.
        ident_ids_by_module.get_or_insert(header.module_id);

        symbols_from_requires = if let HeaderType::Platform {
            requires,
            requires_types,
            opt_app_module_id,
            ..
        } = header.header_type
        {
            // If we don't have an app module id (e.g. because we're doing
            // `roc check myplatform.roc` or because we're generating glue code),
            // insert the `requires` symbols into the platform module's IdentIds.
            //
            // Otherwise, get them from the app module's IdentIds, because it
            // should already have a symbol for each `requires` entry, and we
            // want to make sure we're referencing the same symbols!
            let module_id = opt_app_module_id.unwrap_or(header.module_id);
            let mut symbols_from_requires = Vec::with_capacity(requires.len());
            let ident_ids = ident_ids_by_module.get_or_insert(module_id);

            for Loc {
                value: entry,
                region: _,
            } in requires
            {
                let ident: Ident = entry.ident.value.into();
                let ident_id = ident_ids.get_or_insert(entry.ident.value);
                let symbol = Symbol::new(module_id, ident_id);

                // Since this value is exposed, add it to our module's default scope.
                debug_assert!(!scope.contains_key(&ident.clone()));

                scope.insert(ident, (symbol, entry.ident.region));
                symbols_from_requires.push((Loc::at(entry.ident.region, symbol), entry.ann));
            }

            for entry in requires_types {
                let str_entry: &str = entry.value.into();
                let ident: Ident = str_entry.into();
                let ident_id = ident_ids.get_or_insert(str_entry);
                let symbol = Symbol::new(module_id, ident_id);

                // Since this value is exposed, add it to our module's default scope.
                debug_assert!(!scope.contains_key(&ident));
                scope.insert(ident, (symbol, entry.region));
            }

            symbols_from_requires
        } else {
            Vec::new()
        };

        let ident_ids = ident_ids_by_module.get_mut(&header.module_id).unwrap();

        for loc_exposed in exposed_values.iter() {
            // Use get_or_insert here because the ident_ids may already
            // created an IdentId for this, when it was imported exposed
            // in a dependent module.
            //
            // For example, if module A implements [B.{ foo }], then
            // when we get here for B, `foo` will already have
            // an IdentId. We must reuse that!
            let ident_id = ident_ids.get_or_insert(loc_exposed.value.as_str());
            let symbol = Symbol::new(header.module_id, ident_id);

            exposed.push(symbol);
        }

        // Generate IdentIds entries for all values this module provides,
        // and treat them as `exposes` values for later purposes.
        // This way, when we encounter them in Defs later, they already
        // have an IdentIds entry.
        //
        // We must *not* add them to scope yet, or else the Defs will
        // incorrectly think they're shadowing them!
        if let HeaderType::Platform { provides, .. } = header.header_type {
            exposed.reserve(provides.len());

            for loc_name in provides.iter() {
                // Use get_or_insert here because the ident_ids may already
                // created an IdentId for this, when it was imported exposed
                // in a dependent module.
                //
                // For example, if module A implements [B.{ foo }], then
                // when we get here for B, `foo` will already have
                // an IdentId. We must reuse that!
                let ident_id = ident_ids.get_or_insert(loc_name.value.as_str());
                let symbol = Symbol::new(header.module_id, ident_id);

                exposed.push(symbol);
            }
        }

        if cfg!(debug_assertions) {
            header.module_id.register_debug_idents(ident_ids);
        }

        ident_ids.clone()
    };

    // This module depends on all the modules it exposes. We need to load those in order
    // to do things like report errors for them, generate docs for them, etc.
    if header.is_root_module {
        if let HeaderType::Platform {
            exposes,
            exposes_ids,
            ..
        }
        | HeaderType::Package {
            exposes,
            exposes_ids,
            ..
        } = header.header_type
        {
            for (loc_module_name, module_id) in exposes.iter().zip(exposes_ids.iter().copied()) {
                let module_name_str = loc_module_name.value.as_str();
                let pq_module_name = PackageQualified::Unqualified(module_name_str.into());

                // We should never change an entry here. Either we should have no entry,
                // or if we do have one, it should be unchanged by this insertion.
                debug_assert_eq!(
                    &module_id,
                    deps_by_name.get(&pq_module_name).unwrap_or(&module_id),
                    "Already had a deps_by_name entry for {:?}, but it was {:?} rather than the expected {:?}",
                    pq_module_name,
                    deps_by_name.get(&pq_module_name).unwrap(),
                    module_id,
                );
                deps_by_name.insert(pq_module_name, module_id);
            }
        }
    }

    // Send the deps to the coordinator thread for processing,
    // then continue on to parsing and canonicalizing defs.
    //
    // We always need to send these, even if deps is empty,
    // because the coordinator thread needs to receive this message
    // to decrement its "pending" count.
    let package_qualified_available_modules = deps_by_name
        .iter()
        .map(|(pq_module_name, module_id)| pq_module_name.map_module(|_| *module_id))
        .collect();

    let ModuleHeader {
        module_id,
        module_path,
        header_type,
        header_comments: header_docs,
        ..
    } = header;

    let parsed = ParsedModule {
        module_id,
        module_path,
        src,
        module_timing,
        deps_by_name,
        available_modules,
        packages: header.packages,
        package_qualified_available_modules,
        exposed_ident_ids: ident_ids,
        initial_scope: scope,
        exposes: exposed,
        parsed_defs,
        symbols_from_requires,
        header_type,
        header_comments: header_docs,
        opt_shorthand: header.opt_shorthand,
    };

    Ok(Msg::Parsed(parsed))
}

fn make_specializations<'a>(
    arena: &'a Bump,
    home: ModuleId,
    mut ident_ids: IdentIds,
    mut subs: Subs,
    procs_base: ProcsBase<'a>,
    mut layout_cache: LayoutCache<'a>,
    specializations_we_must_make: Vec<ExternalSpecializations<'a>>,
    mut module_timing: ModuleTiming,
    target: Target,
    world_abilities: WorldAbilities,
    exposed_by_module: &ExposedByModule,
    derived_module: SharedDerivedModule,
    mut expectations: Option<Expectations>,
) -> Msg<'a> {
    let make_specializations_start = Instant::now();
    let mut update_mode_ids = UpdateModeIds::new();

    // do the thing
    let mut mono_env = roc_mono::ir::Env {
        arena,
        subs: &mut subs,
        expectation_subs: expectations.as_mut().map(|e| &mut e.subs),
        home,
        ident_ids: &mut ident_ids,
        target,
        update_mode_ids: &mut update_mode_ids,
        // call_specialization_counter=0 is reserved
        call_specialization_counter: 1,
        abilities: AbilitiesView::World(&world_abilities),
        exposed_by_module,
        derived_module: &derived_module,
        struct_indexing: UsageTrackingMap::default(),
    };

    let mut procs = Procs::new_in(arena);

    let host_exposed_symbols: bumpalo::collections::Vec<_> =
        procs_base.get_host_exposed_symbols().collect_in(arena);

    for (symbol, partial_proc) in procs_base.partial_procs.into_iter() {
        procs.partial_procs.insert(symbol, partial_proc);
    }

    procs.host_exposed_symbols = host_exposed_symbols.into_bump_slice();
    procs.module_thunks = procs_base.module_thunks;
    procs.runtime_errors = procs_base.runtime_errors;
    procs.imported_module_thunks = procs_base.imported_module_thunks;

    // TODO: for now this final specialization pass is sequential,
    // with no parallelization at all. We should try to parallelize
    // this, but doing so will require a redesign of Procs.
    procs = roc_mono::ir::specialize_all(
        &mut mono_env,
        procs,
        specializations_we_must_make,
        procs_base.host_specializations,
        &mut layout_cache,
    );

    let external_specializations_requested = procs.externals_we_need.clone();
    let (procedures, host_exposed_lambda_sets, restored_procs_base) =
        procs.get_specialized_procs_without_rc();

    // Turn `Bytes.Decode.IdentId(238)` into `Bytes.Decode.238`, we rely on this in mono tests
    mono_env.home.register_debug_idents(mono_env.ident_ids);

    let make_specializations_end = Instant::now();
    module_timing
        .make_specializations
        .push(make_specializations_end.duration_since(make_specializations_start));

    Msg::MadeSpecializations {
        module_id: home,
        ident_ids,
        layout_cache,
        procs_base: restored_procs_base,
        procedures,
        host_exposed_lambda_sets,
        update_mode_ids,
        subs,
        expectations,
        external_specializations_requested,
        module_timing,
    }
}

fn build_pending_specializations<'a>(
    arena: &'a Bump,
    solved_subs: Solved<Subs>,
    imported_module_thunks: &'a [Symbol],
    home: ModuleId,
    mut ident_ids: IdentIds,
    declarations: Declarations,
    mut module_timing: ModuleTiming,
    mut layout_cache: LayoutCache<'a>,
    target: Target,
    exposed_to_host: ExposedToHost,
    exposed_by_module: &ExposedByModule,
    world_abilities: WorldAbilities,
    derived_module: SharedDerivedModule,
    mut expectations: Option<Expectations>,
    build_expects: bool,
) -> Msg<'a> {
    let find_specializations_start = Instant::now();

    let mut module_thunks = bumpalo::collections::Vec::new_in(arena);
    let mut toplevel_expects = ToplevelExpects::default();

    let mut procs_base = ProcsBase {
        partial_procs: BumpMap::default(),
        module_thunks: &[],
        host_specializations: roc_mono::ir::HostSpecializations::new(),
        runtime_errors: BumpMap::default(),
        imported_module_thunks,
    };

    let mut update_mode_ids = UpdateModeIds::new();
    let mut subs = solved_subs.into_inner();
    let mut mono_env = roc_mono::ir::Env {
        arena,
        subs: &mut subs,
        expectation_subs: expectations.as_mut().map(|e| &mut e.subs),
        home,
        ident_ids: &mut ident_ids,
        target,
        update_mode_ids: &mut update_mode_ids,
        // call_specialization_counter=0 is reserved
        call_specialization_counter: 1,
        // NB: for getting pending specializations the module view is enough because we only need
        // to know the types and abilities in our modules. Only for building *all* specializations
        // do we need a global view.
        abilities: AbilitiesView::World(&world_abilities),
        exposed_by_module,
        derived_module: &derived_module,
        struct_indexing: UsageTrackingMap::default(),
    };

    let layout_cache_snapshot = layout_cache.snapshot();

    let get_host_annotation = |declarations: &Declarations, index: usize| {
        declarations
            .host_exposed_annotations
            .get(&index)
            .map(|(v, _)| v)
            .copied()
    };

    // Add modules' decls to Procs
    for index in 0..declarations.len() {
        use roc_can::expr::DeclarationTag::*;

        let symbol = declarations.symbols[index].value;
        let expr_var = declarations.variables[index];

        let is_host_exposed = exposed_to_host.top_level_values.contains_key(&symbol);

        let body = declarations.expressions[index].clone();

        let tag = declarations.declarations[index];
        match tag {
            Value => {
                // If this is an exposed symbol, we need to
                // register it as such. Otherwise, since it
                // never gets called by Roc code, it will never
                // get specialized!
                if is_host_exposed {
                    let layout_result =
                        layout_cache.raw_from_var(mono_env.arena, expr_var, mono_env.subs);

                    // cannot specialize when e.g. main's type contains type variables
                    if let Err(e) = layout_result {
                        match e {
                            LayoutProblem::Erroneous => {
                                let message = "top level function has erroneous type";
                                procs_base.runtime_errors.insert(symbol, message);
                                continue;
                            }
                            LayoutProblem::UnresolvedTypeVar(v) => {
                                let message = format!(
                                    "top level function has unresolved type variable {v:?}"
                                );
                                procs_base
                                    .runtime_errors
                                    .insert(symbol, mono_env.arena.alloc(message));
                                continue;
                            }
                        }
                    }

                    procs_base.host_specializations.insert_host_exposed(
                        mono_env.subs,
                        LambdaName::no_niche(symbol),
                        get_host_annotation(&declarations, index),
                        expr_var,
                    );
                }

                match body.value {
                    roc_can::expr::Expr::RecordAccessor(accessor_data) => {
                        let fresh_record_symbol = mono_env.unique_symbol();
                        let closure_data = accessor_data.to_closure_data(fresh_record_symbol);
                        register_toplevel_function_into_procs_base(
                            &mut mono_env,
                            &mut procs_base,
                            closure_data.name,
                            expr_var,
                            closure_data.arguments,
                            closure_data.return_type,
                            *closure_data.loc_body,
                            false,
                        );
                    }
                    _ => {
                        // mark this symbols as a top-level thunk before any other work on the procs
                        module_thunks.push(symbol);

                        let proc = PartialProc {
                            annotation: expr_var,
                            // This is a 0-arity thunk, so it has no arguments.
                            pattern_symbols: &[],
                            // This is a top-level definition, so it cannot capture anything
                            captured_symbols: CapturedSymbols::None,
                            body: body.value,
                            body_var: expr_var,
                            // This is a 0-arity thunk, so it cannot be recursive
                            is_self_recursive: false,
                        };

                        procs_base.partial_procs.insert(symbol, proc);
                    }
                }
            }
            Function(f_index) | Recursive(f_index) | TailRecursive(f_index) => {
                let function_def = &declarations.function_bodies[f_index.index()].value;
                // this is a top-level definition, it should not capture anything
                debug_assert!(
                    function_def.captured_symbols.is_empty(),
                    "{:?}",
                    (symbol, symbol.module_id(), &function_def.captured_symbols)
                );

                // If this is an exposed symbol, we need to
                // register it as such. Otherwise, since it
                // never gets called by Roc code, it will never
                // get specialized!
                if is_host_exposed {
                    let layout_result =
                        layout_cache.raw_from_var(mono_env.arena, expr_var, mono_env.subs);

                    // cannot specialize when e.g. main's type contains type variables
                    if let Err(e) = layout_result {
                        match e {
                            LayoutProblem::Erroneous => {
                                let message = "top level function has erroneous type";
                                procs_base.runtime_errors.insert(symbol, message);
                                continue;
                            }
                            LayoutProblem::UnresolvedTypeVar(v) => {
                                let message = format!(
                                    "top level function has unresolved type variable {v:?}"
                                );
                                procs_base
                                    .runtime_errors
                                    .insert(symbol, mono_env.arena.alloc(message));
                                continue;
                            }
                        }
                    }

                    procs_base.host_specializations.insert_host_exposed(
                        mono_env.subs,
                        LambdaName::no_niche(symbol),
                        get_host_annotation(&declarations, index),
                        expr_var,
                    );
                }

                let is_recursive = matches!(tag, Recursive(_) | TailRecursive(_));

                register_toplevel_function_into_procs_base(
                    &mut mono_env,
                    &mut procs_base,
                    symbol,
                    expr_var,
                    function_def.arguments.clone(),
                    function_def.return_type,
                    body,
                    is_recursive,
                );
            }
            Destructure(d_index) => {
                let loc_pattern = &declarations.destructs[d_index.index()].loc_pattern;

                use roc_can::pattern::Pattern;
                let symbol = match &loc_pattern.value {
                    Pattern::Identifier(_) => {
                        debug_assert!(false, "identifier ended up in Destructure {symbol:?}");
                        symbol
                    }
                    Pattern::AbilityMemberSpecialization { ident, specializes } => {
                        debug_assert!(
                            false,
                            "ability member ended up in Destructure {ident:?} specializes {specializes:?}"
                        );
                        symbol
                    }
                    Pattern::Shadowed(_, _, shadowed) => {
                        // this seems to work for now
                        *shadowed
                    }
                    _ => todo!("top-level destrucuture patterns are not implemented"),
                };

                // mark this symbols as a top-level thunk before any other work on the procs
                module_thunks.push(symbol);

                // If this is an exposed symbol, we need to
                // register it as such. Otherwise, since it
                // never gets called by Roc code, it will never
                // get specialized!
                if is_host_exposed {
                    let layout_result =
                        layout_cache.raw_from_var(mono_env.arena, expr_var, mono_env.subs);

                    // cannot specialize when e.g. main's type contains type variables
                    if let Err(e) = layout_result {
                        match e {
                            LayoutProblem::Erroneous => {
                                let message = "top level function has erroneous type";
                                procs_base.runtime_errors.insert(symbol, message);
                                continue;
                            }
                            LayoutProblem::UnresolvedTypeVar(v) => {
                                let message = format!(
                                    "top level function has unresolved type variable {v:?}"
                                );
                                procs_base
                                    .runtime_errors
                                    .insert(symbol, mono_env.arena.alloc(message));
                                continue;
                            }
                        }
                    }

                    procs_base.host_specializations.insert_host_exposed(
                        mono_env.subs,
                        LambdaName::no_niche(symbol),
                        get_host_annotation(&declarations, index),
                        expr_var,
                    );
                }

                let proc = PartialProc {
                    annotation: expr_var,
                    // This is a 0-arity thunk, so it has no arguments.
                    pattern_symbols: &[],
                    // This is a top-level definition, so it cannot capture anything
                    captured_symbols: CapturedSymbols::None,
                    body: body.value,
                    body_var: expr_var,
                    // This is a 0-arity thunk, so it cannot be recursive
                    is_self_recursive: false,
                };

                procs_base.partial_procs.insert(symbol, proc);
            }
            MutualRecursion { .. } => {
                // the declarations of this group will be treaded individually by later iterations
            }
            Expectation => {
                // skip expectations if we're not going to run them
                if !build_expects {
                    continue;
                }

                // mark this symbol as a top-level thunk before any other work on the procs
                module_thunks.push(symbol);

                let expr_var = Variable::EMPTY_RECORD;

                let is_host_exposed = true;

                // If this is an exposed symbol, we need to
                // register it as such. Otherwise, since it
                // never gets called by Roc code, it will never
                // get specialized!
                if is_host_exposed {
                    let layout_result =
                        layout_cache.raw_from_var(mono_env.arena, expr_var, mono_env.subs);

                    // cannot specialize when e.g. main's type contains type variables
                    if let Err(e) = layout_result {
                        match e {
                            LayoutProblem::Erroneous => {
                                let message = "top level function has erroneous type";
                                procs_base.runtime_errors.insert(symbol, message);
                                continue;
                            }
                            LayoutProblem::UnresolvedTypeVar(v) => {
                                let message = format!(
                                    "top level function has unresolved type variable {v:?}"
                                );
                                procs_base
                                    .runtime_errors
                                    .insert(symbol, mono_env.arena.alloc(message));
                                continue;
                            }
                        }
                    }

                    procs_base.host_specializations.insert_host_exposed(
                        mono_env.subs,
                        LambdaName::no_niche(symbol),
                        None,
                        expr_var,
                    );
                }

                let body = roc_can::expr::toplevel_expect_to_inline_expect_pure(body);

                let proc = PartialProc {
                    annotation: expr_var,
                    // This is a 0-arity thunk, so it has no arguments.
                    pattern_symbols: &[],
                    // This is a top-level definition, so it cannot capture anything
                    captured_symbols: CapturedSymbols::None,
                    body: body.value,
                    body_var: expr_var,
                    // This is a 0-arity thunk, so it cannot be recursive
                    is_self_recursive: false,
                };

                // extend the region of the expect expression with the region of the preceding
                // comment, so it is shown in failure/panic messages
                let name_region = declarations.symbols[index].region;
                let expr_region = declarations.expressions[index].region;
                let region = Region::span_across(&name_region, &expr_region);

                toplevel_expects.pure.insert(symbol, region);
                procs_base.partial_procs.insert(symbol, proc);
            }
        }
    }

    layout_cache.rollback_to(layout_cache_snapshot);

    procs_base.module_thunks = module_thunks.into_bump_slice();

    let find_specializations_end = Instant::now();
    module_timing.find_specializations =
        find_specializations_end.duration_since(find_specializations_start);

    Msg::FoundSpecializations {
        module_id: home,
        solved_subs: Solved(subs),
        ident_ids,
        layout_cache,
        procs_base,
        module_timing,
        toplevel_expects,
        expectations,
    }
}

fn register_toplevel_function_into_procs_base<'a>(
    mono_env: &mut roc_mono::ir::Env<'a, '_>,
    procs_base: &mut ProcsBase<'a>,
    symbol: Symbol,
    expr_var: Variable,
    arguments: Vec<(
        Variable,
        roc_can::expr::AnnotatedMark,
        Loc<roc_can::pattern::Pattern>,
    )>,
    return_type: Variable,
    body: Loc<roc_can::expr::Expr>,
    is_recursive: bool,
) {
    let partial_proc = PartialProc::from_named_function(
        mono_env,
        expr_var,
        arguments,
        body,
        CapturedSymbols::None,
        is_recursive,
        return_type,
    );

    procs_base.partial_procs.insert(symbol, partial_proc);
}

/// Loads derived ability members up for specialization into the Derived module, prior to making
/// their specializations.
// TODO: right now, this runs sequentially, and no other modules are mono'd in parallel to the
// derived module.
fn load_derived_partial_procs<'a>(
    home: ModuleId,
    arena: &'a Bump,
    subs: &mut Subs,
    ident_ids: &mut IdentIds,
    derived_module: &SharedDerivedModule,
    module_timing: &mut ModuleTiming,
    target: Target,
    exposed_by_module: &ExposedByModule,
    procs_base: &mut ProcsBase<'a>,
    world_abilities: &mut WorldAbilities,
) {
    debug_assert_eq!(home, ModuleId::DERIVED_GEN);

    let load_derived_procs_start = Instant::now();

    let mut new_module_thunks = bumpalo::collections::Vec::new_in(arena);

    let mut update_mode_ids = UpdateModeIds::new();

    let derives_to_add = {
        let mut derived_module = derived_module.lock().unwrap();

        derived_module.iter_load_for_gen_module(subs, |symbol| {
            !procs_base.partial_procs.contains_key(&symbol)
        })
    };

    // TODO: we can be even lazier here if we move `add_def_to_module` to happen in mono. Also, the
    // timings would be more accurate.
    for (derived_symbol, (derived_expr, derived_expr_var)) in derives_to_add.into_iter() {
        let mut mono_env = roc_mono::ir::Env {
            arena,
            subs,
            // There are no derived expectations.
            expectation_subs: None,
            home,
            ident_ids,
            target,
            update_mode_ids: &mut update_mode_ids,
            // call_specialization_counter=0 is reserved
            call_specialization_counter: 1,
            // NB: for getting pending specializations the module view is enough because we only need
            // to know the types and abilities in our modules. Only for building *all* specializations
            // do we need a global view.
            abilities: AbilitiesView::World(world_abilities),
            exposed_by_module,
            derived_module,
            struct_indexing: UsageTrackingMap::default(),
        };

        let partial_proc = match derived_expr {
            roc_can::expr::Expr::Closure(roc_can::expr::ClosureData {
                function_type,
                arguments,
                loc_body,
                captured_symbols,
                return_type,
                recursive,
                ..
            }) => {
                debug_assert!(captured_symbols.is_empty());
                PartialProc::from_named_function(
                    &mut mono_env,
                    function_type,
                    arguments.clone(),
                    *loc_body,
                    CapturedSymbols::None,
                    recursive.is_recursive(),
                    return_type,
                )
            }
            _ => {
                // mark this symbols as a top-level thunk before any other work on the procs
                new_module_thunks.push(derived_symbol);

                PartialProc {
                    annotation: derived_expr_var,
                    // This is a 0-arity thunk, so it has no arguments.
                    pattern_symbols: &[],
                    // This is a top-level definition, so it cannot capture anything
                    captured_symbols: CapturedSymbols::None,
                    body: derived_expr,
                    body_var: derived_expr_var,
                    // This is a 0-arity thunk, so it cannot be recursive
                    is_self_recursive: false,
                }
            }
        };

        procs_base
            .partial_procs
            .insert(derived_symbol, partial_proc);
    }

    if !new_module_thunks.is_empty() {
        new_module_thunks.extend(procs_base.module_thunks);
        procs_base.module_thunks = new_module_thunks.into_bump_slice();
    }

    let load_derived_procs_end = Instant::now();

    module_timing.find_specializations =
        load_derived_procs_end.duration_since(load_derived_procs_start);
}

fn run_task<'a>(
    task: BuildTask<'a>,
    arena: &'a Bump,
    src_dir: &Path,
    msg_tx: MsgSender<'a>,
    roc_cache_dir: RocCacheDir<'_>,
    target: Target,
) -> Result<(), ChannelProblem> {
    use BuildTask::*;

    let msg_result = match task {
        LoadModule {
            module_name,
            module_ids,
            shorthands,
            ident_ids_by_module,
        } => load_module(
            arena,
            src_dir,
            module_name,
            module_ids,
            shorthands,
            roc_cache_dir,
            ident_ids_by_module,
        )
        .map(|HeaderOutput { msg, .. }| msg),
        Parse {
            header,
            arc_shorthands,
            module_ids,
            ident_ids_by_module,
            root_type,
        } => parse(
            arena,
            header,
            arc_shorthands,
            module_ids,
            ident_ids_by_module,
            root_type,
        ),
        CanonicalizeAndConstrain {
            parsed,
            qualified_module_ids,
            dep_idents,
            exposed_symbols,
            aliases,
            abilities_store,
            skip_constraint_gen,
            exposed_module_ids,
            exec_mode,
            imported_module_params,
            fx_mode,
        } => {
            let can_and_con = canonicalize_and_constrain(
                arena,
                &qualified_module_ids,
                dep_idents,
                exposed_symbols,
                aliases,
                abilities_store,
                parsed,
                skip_constraint_gen,
                exposed_module_ids,
                exec_mode,
                imported_module_params,
                fx_mode,
            );

            Ok(Msg::CanonicalizedAndConstrained(can_and_con))
        }
        Solve {
            module,
            module_timing,
            exposed_for_module,
            types,
            constraints,
            constraint,
            function_kind,
            pending_derives,
            var_store,
            ident_ids,
            declarations,
            dep_idents,
            cached_subs,
            derived_module,
            exec_mode,
            is_host_exposed,

            #[cfg(debug_assertions)]
            checkmate,
        } => Ok(run_solve(
            module,
            ident_ids,
            module_timing,
            exposed_for_module,
            types,
            constraints,
            constraint,
            function_kind,
            pending_derives,
            var_store,
            declarations,
            dep_idents,
            cached_subs,
            derived_module,
            exec_mode,
            is_host_exposed,
            //
            #[cfg(debug_assertions)]
            checkmate,
        )),
        BuildPendingSpecializations {
            module_id,
            ident_ids,
            decls,
            module_timing,
            layout_cache,
            solved_subs,
            imported_module_thunks,
            exposed_to_host,
            world_abilities,
            exposed_by_module,
            derived_module,
            expectations,
            build_expects,
        } => Ok(build_pending_specializations(
            arena,
            solved_subs,
            imported_module_thunks,
            module_id,
            ident_ids,
            decls,
            module_timing,
            layout_cache,
            target,
            exposed_to_host,
            &exposed_by_module,
            world_abilities,
            derived_module,
            expectations,
            build_expects,
        )),
        MakeSpecializations {
            module_id,
            ident_ids,
            subs,
            procs_base,
            layout_cache,
            specializations_we_must_make,
            module_timing,
            world_abilities,
            exposed_by_module,
            derived_module,
            expectations,
        } => Ok(make_specializations(
            arena,
            module_id,
            ident_ids,
            subs,
            procs_base,
            layout_cache,
            specializations_we_must_make,
            module_timing,
            target,
            world_abilities,
            &exposed_by_module,
            derived_module,
            expectations,
        )),
    };

    match msg_result {
        Ok(msg) => {
            msg_tx
                .send(msg)
                .map_err(|_| ChannelProblem::FailedToSendTaskMsg)?;

            Ok(())
        }
        Err(loading_problem) => {
            let result = match loading_problem {
                LoadingProblem::ChannelProblem(problem) => {
                    return Err(problem);
                }
                LoadingProblem::ParsingFailed(problem) => msg_tx.send(Msg::FailedToParse(problem)),
                LoadingProblem::FileProblem { filename, error } => {
                    msg_tx.send(Msg::FailedToReadFile { filename, error })
                }
                LoadingProblem::IncorrectModuleName(err) => {
                    msg_tx.send(Msg::IncorrectModuleName(err))
                }
                err => msg_tx.send(Msg::FailedToLoad(err)),
            };

            match result {
                Ok(()) => Ok(()),
                Err(_) => Err(ChannelProblem::FailedToSendTaskMsg),
            }
        }
    }
}

fn to_import_cycle_report(
    module_ids: ModuleIds,
    all_ident_ids: IdentIdsByModule,
    import_cycle: Vec<ModuleId>,
    filename: PathBuf,
    render: RenderTarget,
) -> String {
    use roc_reporting::report::{Report, RocDocAllocator, DEFAULT_PALETTE};
    use ven_pretty::DocAllocator;

    // import_cycle looks like CycleModule, Import1, ..., ImportN, CycleModule
    // In a self-referential case, it just looks like CycleModule, CycleModule.
    debug_assert!(import_cycle.len() >= 2);
    let source_of_cycle = import_cycle.first().unwrap();

    // We won't be printing any lines for this report, so this is okay.
    // TODO: it would be nice to show how each module imports another in the cycle.
    let src_lines = &[];

    let interns = Interns {
        module_ids,
        all_ident_ids,
    };
    let alloc = RocDocAllocator::new(src_lines, *source_of_cycle, &interns);

    let doc = alloc.stack([
        alloc.concat([
            alloc.reflow("I can't compile "),
            alloc.module(*source_of_cycle),
            alloc.reflow(
                " because it depends on itself through the following chain of module imports:",
            ),
        ]),
        roc_reporting::report::cycle(
            &alloc,
            4,
            alloc.module(*source_of_cycle),
            import_cycle
                .into_iter()
                .skip(1)
                .map(|module| alloc.module(module))
                .collect(),
        ),
        alloc.reflow("Cyclic dependencies are not allowed in Roc! Can you restructure a module in this import chain so that it doesn't have to depend on itself?")
    ]);

    let report = Report {
        filename,
        doc,
        title: "IMPORT CYCLE".to_string(),
        severity: Severity::RuntimeError,
    };

    let mut buf = String::new();
    let palette = DEFAULT_PALETTE;
    report.render(render, &mut buf, &alloc, &palette);
    buf
}

fn to_incorrect_module_name_report<'a>(
    module_ids: ModuleIds,
    all_ident_ids: IdentIdsByModule,
    problem: IncorrectModuleName<'a>,
    filename: PathBuf,
    src: &'a [u8],
    render: RenderTarget,
) -> String {
    use roc_reporting::report::{Report, RocDocAllocator, DEFAULT_PALETTE};
    use ven_pretty::DocAllocator;

    let IncorrectModuleName {
        module_id,
        found,
        expected,
    } = problem;

    let severity = Severity::RuntimeError;

    // SAFETY: if the module was not UTF-8, that would be reported as a parsing problem, rather
    // than an incorrect module name problem (the latter can happen only after parsing).
    let src = unsafe { from_utf8_unchecked(src) };
    let src_lines = src.lines().collect::<Vec<_>>();
    let lines = LineInfo::new(src);

    let interns = Interns {
        module_ids,
        all_ident_ids,
    };
    let alloc = RocDocAllocator::new(&src_lines, module_id, &interns);

    let doc = alloc.stack([
        alloc.reflow("This module has a different name than I expected:"),
        alloc.region(lines.convert_region(found.region), severity),
        alloc.reflow("Based on the nesting and use of this module, I expect it to have name"),
        alloc.pq_module_name(expected).indent(4),
    ]);

    let report = Report {
        filename,
        doc,
        title: "INCORRECT MODULE NAME".to_string(),
        severity,
    };

    let mut buf = String::new();
    let palette = DEFAULT_PALETTE;
    report.render(render, &mut buf, &alloc, &palette);
    buf
}

fn to_no_platform_package_report(
    module_ids: ModuleIds,
    all_ident_ids: IdentIdsByModule,
    module_id: ModuleId,
    filename: PathBuf,
    region: Region,
    src: &[u8],
    render: RenderTarget,
) -> String {
    use roc_reporting::report::{Report, RocDocAllocator, DEFAULT_PALETTE};
    use ven_pretty::DocAllocator;
    let severity = Severity::RuntimeError;

    // SAFETY: if the module was not UTF-8, that would be reported as a parsing problem, rather
    // than an incorrect module name problem (the latter can happen only after parsing).
    let src = unsafe { from_utf8_unchecked(src) };
    let src_lines = src.lines().collect::<Vec<_>>();
    let lines = LineInfo::new(src);

    let interns = Interns {
        module_ids,
        all_ident_ids,
    };
    let alloc = RocDocAllocator::new(&src_lines, module_id, &interns);

    let doc = alloc.stack([
        alloc.reflow("This app does not specify a platform:"),
        alloc.region(lines.convert_region(region),severity),
        alloc.reflow("Make sure you have exactly one package specified as `platform`:"),
        alloc
            .parser_suggestion("    app [main] {\n        pf: platform \"…path or URL to platform…\"\n            ^^^^^^^^\n    }"),
        alloc.reflow("Tip: See an example in the tutorial:\n\n<https://www.roc-lang.org/tutorial#building-an-application>"),

    ]);

    let report = Report {
        filename,
        doc,
        title: "UNSPECIFIED PLATFORM".to_string(),
        severity,
    };

    let mut buf = String::new();
    let palette = DEFAULT_PALETTE;
    report.render(render, &mut buf, &alloc, &palette);
    buf
}

fn to_multiple_platform_packages_report(
    module_ids: ModuleIds,
    all_ident_ids: IdentIdsByModule,
    module_id: ModuleId,
    filename: PathBuf,
    region: Region,
    src: &[u8],
    render: RenderTarget,
) -> String {
    use roc_reporting::report::{Report, RocDocAllocator, DEFAULT_PALETTE};
    use ven_pretty::DocAllocator;
    let severity = Severity::RuntimeError;

    // SAFETY: if the module was not UTF-8, that would be reported as a parsing problem, rather
    // than an incorrect module name problem (the latter can happen only after parsing).
    let src = unsafe { from_utf8_unchecked(src) };
    let src_lines = src.lines().collect::<Vec<_>>();
    let lines = LineInfo::new(src);

    let interns = Interns {
        module_ids,
        all_ident_ids,
    };
    let alloc = RocDocAllocator::new(&src_lines, module_id, &interns);

    let doc = alloc.stack([
        alloc.reflow("This app specifies multiple packages as `platform`:"),
        alloc.region(lines.convert_region(region), severity),
        alloc.reflow("Roc apps must specify exactly one platform."),
    ]);

    let report = Report {
        filename,
        doc,
        title: "MULTIPLE PLATFORMS".to_string(),
        severity,
    };

    let mut buf = String::new();
    let palette = DEFAULT_PALETTE;
    report.render(render, &mut buf, &alloc, &palette);
    buf
}

fn to_unrecognized_package_shorthand_report(
    module_ids: ModuleIds,
    all_ident_ids: IdentIdsByModule,
    module_id: ModuleId,
    filename: PathBuf,
    region: Region,
    src: &str,
    shorthand: &str,
    available: AvailableShorthands,
    render: RenderTarget,
) -> String {
    use roc_reporting::report::{Report, RocDocAllocator, DEFAULT_PALETTE};
    use ven_pretty::DocAllocator;

    let src_lines = src.lines().collect::<Vec<_>>();
    let lines = LineInfo::new(src);

    let interns = Interns {
        module_ids,
        all_ident_ids,
    };
    let alloc = RocDocAllocator::new(&src_lines, module_id, &interns);

    let help = match available {
        AvailableShorthands::FromRoot(options) => {
            let mut suggestions = suggest::sort(shorthand, options);
            suggestions.truncate(4);

            if suggestions.is_empty() {
                alloc.reflow("A lowercase name indicates a package shorthand, but no packages have been specified.")
            } else {
                alloc.stack([
                    alloc.reflow("A lowercase name indicates a package shorthand, but I don't recognize this one. Did you mean one of these?"),
                    alloc
                        .vcat(suggestions.into_iter().map(|v| alloc.shorthand(v)))
                        .indent(4),
                ])
            }
        }
        AvailableShorthands::FromMain(main_path, options) => {
            let mut suggestions = suggest::sort(shorthand, options);
            suggestions.truncate(4);

            let suggestions = if suggestions.is_empty() {
                alloc.reflow("A lowercase name indicates a package shorthand, but no packages have been specified.")
            } else {
                alloc.stack([
                    alloc.reflow("A lowercase name indicates a package shorthand, but I don't recognize this one. Did you mean one of these?"),
                    alloc
                        .vcat(suggestions.into_iter().map(|v| alloc.shorthand(v)))
                        .indent(4),
                ])
            };

            alloc.stack([
                suggestions,
                alloc.note("I'm using the following module to resolve package shorthands:"),
                alloc.file_path(&main_path).indent(4),
                alloc.concat([
                    alloc.reflow("You can specify a different one with the "),
                    alloc.keyword("--main"),
                    alloc.reflow(" flag.")
                ]),
            ])
        }
        AvailableShorthands::UnknownMain => {
            alloc.stack([
                alloc.reflow("A lowercase name indicates a package shorthand, but I don't know which packages are available."),
                alloc.concat([
                    alloc.reflow("When checking a module directly, I look for a `"),
                    alloc.reflow(DEFAULT_MAIN_NAME),
                    alloc.reflow("` app or package to resolve shorthands from."),
                ]),
                alloc.concat([
                    alloc.reflow("You can create it, or specify an existing one with the "),
                   alloc.keyword("--main"),
                    alloc.reflow(" flag."),
                ])
            ])
        }
    };

    let severity = Severity::RuntimeError;

    let doc = alloc.stack([
        alloc.concat([
            alloc.reflow("This module is trying to import from `"),
            alloc.shorthand(shorthand),
            alloc.reflow("`:"),
        ]),
        alloc.region(lines.convert_region(region), severity),
        help,
    ]);

    let report = Report {
        filename,
        doc,
        title: "UNRECOGNIZED PACKAGE".to_string(),
        severity,
    };

    let mut buf = String::new();
    let palette = DEFAULT_PALETTE;
    report.render(render, &mut buf, &alloc, &palette);
    buf
}

fn to_parse_problem_report<'a>(
    problem: FileError<'a, SyntaxError<'a>>,
    mut module_ids: ModuleIds,
    all_ident_ids: IdentIdsByModule,
    render: RenderTarget,
    palette: Palette,
) -> String {
    use roc_reporting::report::{parse_problem, RocDocAllocator};

    // TODO this is not in fact safe
    let src = unsafe { from_utf8_unchecked(problem.problem.bytes) };
    let src_lines = src.lines().collect::<Vec<_>>();
    // let mut src_lines: Vec<&str> = problem.prefix.lines().collect();
    // src_lines.extend(src.lines().skip(1));

    let module_id = module_ids.get_or_insert(&"find module name somehow?".into());

    let interns = Interns {
        module_ids,
        all_ident_ids,
    };

    // Report parsing and canonicalization problems
    let alloc = RocDocAllocator::new(&src_lines, module_id, &interns);

    let starting_line = 0;

    let lines = LineInfo::new(src);

    let report = parse_problem(
        &alloc,
        &lines,
        problem.filename.clone(),
        starting_line,
        problem,
    );

    let mut buf = String::new();

    report.render(render, &mut buf, &alloc, &palette);

    buf
}

fn report_cannot_run(
    module_id: ModuleId,
    filename: PathBuf,
    platform_path: &PlatformPath,
) -> String {
    use roc_reporting::report::{Report, RocDocAllocator, DEFAULT_PALETTE};
    use ven_pretty::DocAllocator;
    use PlatformPath::*;

    // Report parsing and canonicalization problems
    let interns = Interns::default();
    let alloc = RocDocAllocator::new(&[], module_id, &interns);

    let report = {
        match platform_path {
            Valid(_) => unreachable!(),
            NotSpecified => {
                let doc = alloc.stack([
                    alloc.reflow("I could not find a platform based on your input file."),
                    alloc.reflow(r"Does the module header have an entry that looks like this?"),
                    alloc
                        .parser_suggestion("app [main] { pf: platform \"…path or URL to platform…\" }")
                        .indent(4),
                    alloc.reflow("Tip: The following part of the tutorial has an example of specifying a platform:\n\n<https://www.roc-lang.org/tutorial#building-an-application>"),
                ]);

                Report {
                    filename,
                    doc,
                    title: "NO PLATFORM".to_string(),
                    severity: Severity::RuntimeError,
                }
            }
            RootIsModule => {
                let doc = alloc.stack([
                    alloc.reflow(
                        r"The input file is a module, but only `app` modules can be run.",
                    ),
                    alloc.reflow(r"Tip: You can use `roc check` or `roc test` to verify a module like this one."),
                ]);

                Report {
                    filename,
                    doc,
                    title: "NO PLATFORM".to_string(),
                    severity: Severity::RuntimeError,
                }
            }
            RootIsHosted => {
                let doc = alloc.stack([
                    alloc.reflow(
                        r"The input file is a `hosted` module, but only `app` modules can be run.",
                    ),
                    alloc.reflow(r"Tip: You can use `roc check` or `roc test` to verify a hosted module like this one."),
                ]);

                Report {
                    filename,
                    doc,
                    title: "NO PLATFORM".to_string(),
                    severity: Severity::RuntimeError,
                }
            }
            RootIsPlatformModule => {
                let doc = alloc.stack([
                    alloc.reflow(
                        r"The input file is a `platform` module, but only `app` modules can be run.",
                    ),
                    alloc.reflow(r"Tip: You can use `roc check` or `roc test` to verify a platform module like this one."),
                ]);

                Report {
                    filename,
                    doc,
                    title: "NO PLATFORM".to_string(),
                    severity: Severity::RuntimeError,
                }
            }
        }
    };

    let palette = DEFAULT_PALETTE;
    let mut buf = String::new();
    report.render_color_terminal(&mut buf, &alloc, &palette);

    buf
}