/*
 * MRustC - Rust Compiler
 * - By John Hodge (Mutabah/thePowersGang)
 *
 * hir_expand/lifetime_infer.cpp
 * - Infer and check lifetime annotations
 */
#include <hir/visitor.hpp>
#include <hir/expr.hpp>
#include <hir_typeck/static.hpp>
#include <algorithm>
#include <hir/expr_state.hpp>
#include "main_bindings.hpp"


namespace {
    struct LifetimeInferState
    {
        struct LifetimeBound {
            ::HIR::LifetimeRef  lft;
            const ::HIR::TypeRef* ty;
            ::HIR::LifetimeRef  valid_for;
            LifetimeBound(::HIR::LifetimeRef lft, ::HIR::LifetimeRef valid_for)
                : lft(lft)
                , ty(nullptr)
                , valid_for(valid_for)
            {
            }
            LifetimeBound(const ::HIR::TypeRef& ty, ::HIR::LifetimeRef valid_for)
                : lft()
                , ty(&ty)
                , valid_for(valid_for)
            {
            }

            ::std::ostream& fmt(::std::ostream& os) const {
                if(ty) {
                    os << *ty;
                }
                else {
                    os << lft;
                }
                os << " : " << valid_for;
                return os;
            }
            bool operator==(const LifetimeBound& x) const {
                if( static_cast<bool>(ty) != static_cast<bool>(x.ty) )
                    return false;
                if( ty && *ty != *x.ty )
                    return false;
                if( !ty && lft != x.lft )
                    return false;
                if( valid_for != x.valid_for )
                    return false;
                return true;
            }
            bool operator!=(const LifetimeBound& x) const {
                return !operator==(x);
            }
        };
        TAGGED_UNION(LocalLifetimeData, Composite,
        (Composite, std::vector<HIR::LifetimeRef>),
        (PatternBinding, struct {
            const HIR::ExprNode*    borrow_point;
            const HIR::ExprNode*    value;
            const HIR::PatternBinding* pat;
            }),
        (Node, struct {
            const HIR::ExprNode* borrow_point;
            const HIR::ExprNode* value;
            })
        );
        struct LocalLifetime
        {
            Span    borrow_span;
            LocalLifetimeData   data;
        };
        struct IvarLifetime
        {
            Span    sp;
            /// The final assigned lifetime (not an ivar)
            HIR::LifetimeRef    known;

            /// Lifetimes that become this ivar
            std::vector<HIR::LifetimeRef>   sources;
            /// Lifetimes this ivar becomes
            std::vector<HIR::LifetimeRef>   destinations;
            /// Lifetimes that must be unified with this ivar
            std::vector<HIR::LifetimeRef>   equals;

            IvarLifetime(const Span& sp): sp(sp) {}
            bool is_known() const {
                return this->known.binding != HIR::LifetimeRef::UNKNOWN;
            }
        };

        const StaticTraitResolve& m_resolve;
        bool    m_in_constant_context;
        std::vector<LifetimeBound>  m_bounds;
        /// @brief Locally started/located lifetimes (i.e. borrows of locals/temporaries)
        std::vector<LocalLifetime>  m_locals;
        std::vector<IvarLifetime>   m_ivars;
        /// Assignment values to Type-Alias Impl-Trait, so they can be assigned after all is done
        /// - The `PathParams` is used to map between this context and the TAIT
        std::map<const HIR::TypeData_ErasedType_AliasInner*,std::pair<HIR::PathParams,HIR::TypeRef>>  m_tait_values;

        LifetimeInferState(const StaticTraitResolve& resolve, bool in_constant_context)
            : m_resolve(resolve)
            , m_in_constant_context(in_constant_context)
        {}

        void add_lifetime_bound(const HIR::LifetimeRef& test, const HIR::LifetimeRef& valid_for) {
            DEBUG("Add bound: " << test << ": " << valid_for);
            m_bounds.push_back(LifetimeInferState::LifetimeBound(test, valid_for));
        }
        void add_type_lifetime_bound(const HIR::TypeRef& ty, const HIR::LifetimeRef& valid_for) {
            DEBUG("Add bound: " << ty << ": " << valid_for);
            m_bounds.push_back(LifetimeInferState::LifetimeBound(ty, valid_for));
            if( const auto* te = ty.data().opt_Tuple() ) {
                for(const auto& t : *te) {
                    add_type_lifetime_bound(t, valid_for);
                }
            }
            else if( const auto* te = ty.data().opt_Slice() ) {
                add_type_lifetime_bound(te->inner, valid_for);
            }
            else if( const auto* te = ty.data().opt_Array() ) {
                add_type_lifetime_bound(te->inner, valid_for);
            }
            else if( const auto* te = ty.data().opt_Borrow() ) {
                add_lifetime_bound(te->lifetime, valid_for);
                add_type_lifetime_bound(te->inner, valid_for);
            }
            else if( const auto* te = ty.data().opt_Path() ) {
                if( /*const auto* pe =*/ te->path.m_data.opt_Generic() ) {
                    // TODO: Look up variance of the lifetime params
                    // 
                    // TODO: Get the variance info of each parameter (instead of assuming "variant")
                    const auto& p = te->path.m_data.as_Generic();
                    if( !p.m_params.m_lifetimes.empty() ) {
                        DEBUG("TODO: Check variance " << ty);
                        for(const auto& l : p.m_params.m_lifetimes) {
                            add_lifetime_bound(l, valid_for);
                        }
                    }
                    // TODO: Add a type bound for associated types?
                }
            }
        }

    private:
        HIR::LifetimeRef allocate_local(const Span& sp, LocalLifetimeData v) {
            auto idx = m_locals.size();
            m_locals.push_back(LocalLifetime { sp, std::move(v) });
            assert(idx < (HIR::LifetimeRef::MAX_LOCAL - 0x1'0000));
            return HIR::LifetimeRef( 0x1'0000 + idx );
        }
    public:
        HIR::LifetimeRef allocate_local(const Span& sp, std::vector<HIR::LifetimeRef> sources) {
            auto rv = allocate_local(sp, LocalLifetimeData::make_Composite(std::move(sources)));
            DEBUG(rv << " = {" << opt_local(sp, rv)->data.as_Composite() << "} (span=" << sp << ")");
            return rv;
        }
        HIR::LifetimeRef allocate_local(const Span& sp, const HIR::ExprNode& borrow_point, const HIR::ExprNode& value, const HIR::PatternBinding& pb) {
            auto rv = allocate_local(sp, LocalLifetimeData::make_PatternBinding({ &borrow_point, &value, &pb }));
            DEBUG(rv << " = " << pb << " (span=" << sp << ")");
            return rv;
        }
        HIR::LifetimeRef allocate_local(const HIR::ExprNode& borrow_point, const HIR::ExprNode& value) {
            auto rv = allocate_local(borrow_point.span(), LocalLifetimeData::Data_Node { &borrow_point, &value });
            DEBUG(rv << " = " << value.span() << " (span=" << borrow_point.span() << ")");
            return rv;
        }
        bool is_local(const HIR::LifetimeRef& lft) const {
            return 0x1'0000 <= lft.binding && lft.binding < HIR::LifetimeRef::MAX_LOCAL;
        }
        const LocalLifetime* opt_local(const Span& sp, const HIR::LifetimeRef& lft) const {
            if( is_local(lft) ) {
                auto idx = lft.binding - 0x1'0000;
                ASSERT_BUG(sp, idx < m_locals.size(), "Local lifetime index out of range - " << lft);
                return &m_locals[idx];
            }
            else {
                return nullptr;
            }
        }

        HIR::LifetimeRef allocate_ivar(const Span& sp) {
            auto idx = m_ivars.size();
            m_ivars.push_back(IvarLifetime(sp));
            assert(idx < (0x10'0000 -  HIR::LifetimeRef::MAX_LOCAL));
            auto rv = HIR::LifetimeRef( HIR::LifetimeRef::MAX_LOCAL + idx );
            DEBUG("Allocate " << rv);
            return rv;
        }
        bool is_ivar(const HIR::LifetimeRef& lft) const {
            return HIR::LifetimeRef::MAX_LOCAL <= lft.binding;
        }
        IvarLifetime* opt_ivar(const Span& sp, const HIR::LifetimeRef& lft) {
            if( is_ivar(lft) ) {
                auto idx = lft.binding - HIR::LifetimeRef::MAX_LOCAL;
                ASSERT_BUG(sp, idx < m_ivars.size(), "IVar lifetime index out of range - " << lft);
                return &m_ivars[idx];
            }
            else {
                return nullptr;
            }
        }
        const IvarLifetime* opt_ivar(const Span& sp, const HIR::LifetimeRef& lft) const {
            if( HIR::LifetimeRef::MAX_LOCAL <= lft.binding ) {
                auto idx = lft.binding - HIR::LifetimeRef::MAX_LOCAL;
                ASSERT_BUG(sp, idx < m_ivars.size(), "IVar lifetime index out of range - " << lft);
                return &m_ivars[idx];
            }
            else {
                return nullptr;
            }
        }



        HIR::LifetimeRef get_lft_for_iv(const IvarLifetime& iv) const {
            return HIR::LifetimeRef( HIR::LifetimeRef::MAX_LOCAL + (&iv - m_ivars.data()) );
        }
        HIR::LifetimeRef get_final_lft(Span sp, HIR::LifetimeRef lft) const {
            while( const auto* s = opt_ivar(sp, lft) ) {
                // TODO: Detect and panic on infinite loops
                if( !s->is_known() ) {
                    return lft;
                }
                sp = s->sp;
                lft = s->known;
            }
            return lft;
        }
        void dump() {
            for(const auto& l : m_locals) {
                auto lft = HIR::LifetimeRef( 0x1'0000 + (&l - m_locals.data()) );
                TU_MATCH_HDRA( (l.data), { )
                TU_ARMA(Composite, le) {
                    DEBUG(lft << " := {" << le << "}");
                    }
                TU_ARMA(PatternBinding, le) {
                    DEBUG(lft << " := PB " << le.pat << " (" << le.value << " @ " << le.borrow_point << ")");
                    }
                TU_ARMA(Node, le) {
                    DEBUG(lft << " := Node " << le.value << " @ " << le.borrow_point);
                    }
                }
            }
            for(const auto& iv : m_ivars) {
                auto lft = get_lft_for_iv(iv);
                if( iv.is_known() ) {
                    DEBUG(lft << " = " << iv.known << " to=[" << iv.destinations << "]");
                }
                else {
                    DEBUG(lft << " -- to=[" << iv.destinations << "], from=[" << iv.sources << "]");
                }
            }
        }

        bool iterate_lft_bounds(const HIR::LifetimeRef& target, std::function<bool(const HIR::LifetimeRef&)> cb) const
        {
            for(const auto& b : m_bounds) {
                if(!b.ty) {
                    if( b.lft == target ) {
                        if( cb(b.valid_for) )
                            return true;
                    }
                }
            }
            return false;
        }

        bool iterate_type_lifetime_bounds(const HIR::TypeRef& ty, std::function<bool(const HIR::LifetimeRef&)> cb) const
        {
            for(const auto& b : m_bounds) {
                if(b.ty) {
                    if( *b.ty == ty ) {
                        if( cb(b.valid_for) )
                            return true;
                    }
                }
            }
            return false;
        }

        /// Check that `rhs` is valid for `lhs` (assignment ordering)
        /// Stores the root RHS lifetimes that failed bounds in `fails`
        bool check_lifetimes(const Span& sp, const HIR::LifetimeRef& lhs, const HIR::LifetimeRef& rhs, std::vector<HIR::LifetimeRef>& fails) const
        {
            //TRACE_FUNCTION_F(lhs << " = " << rhs);
            assert( !opt_ivar(sp, lhs) );
            assert( !opt_ivar(sp, rhs) );

            // Identical lifetimes: valid
            if( lhs == rhs ) {
                return true;
            }
            // 'static outlives everything
            if( rhs.binding == HIR::LifetimeRef::STATIC ) {
                return true;
            }

            // Source composite local: Valid if all entries are valid
            if( const auto* r_l = opt_local(sp, rhs) ) {
                if( const auto* r_le = r_l->data.opt_Composite() ) {
                    DEBUG("RHS composite: all");
                    bool rv = true;
                    for(const auto& inner : *r_le) {
                        rv &= this->check_lifetimes(sp, lhs, inner, fails);
                    }
                    return rv;
                }
            }
            // Destination composite local: Valid if any entry is valid
            if( const auto* l_l = opt_local(sp, lhs) ) {
                if( const auto* l_le = l_l->data.opt_Composite() ) {
                    DEBUG("LHS composite: any");
                    std::vector<HIR::LifetimeRef>   tmp;
                    for(const auto& inner : *l_le) {
                        if( this->check_lifetimes(sp, inner, rhs, tmp) )
                            return true;
                    }
                    fails.push_back(rhs);
                    return false;
                }
            }

            if( lhs.is_param() ) {
                if( rhs.is_param() ) {
                    // Check for an outlives relationship
                    bool rv = iterate_lft_bounds(rhs, [&](const HIR::LifetimeRef& valid_for)->bool {
                        if( valid_for == lhs )
                        //if( check_lifetimes(sp, lhs, valid_for, fails) )
                        {
                            return true;
                        }
                        return false;
                        });
                    if( !rv ) {
                        fails.push_back(rhs);
                    }
                    return rv;
                }
                else if( opt_local(sp, rhs) ) {
                    DEBUG("param from local");
                    fails.push_back(rhs);
                    return false;
                }
                else {
                    BUG(sp, "Unknown lft - " << rhs);
                }
            }
            else if( lhs.binding == HIR::LifetimeRef::STATIC ) {
                if( rhs.is_param() ) {
                    // Check for an outlives relationship (this param must be bounded to `'static`)
                    bool rv = iterate_lft_bounds(rhs, [&](const HIR::LifetimeRef& valid_for)->bool {
                        if(valid_for == lhs ) {
                        //if( check_lifetimes(sp, lhs, valid_for, fails) ) {
                            return true;
                        }
                        return false;
                        });
                    if( !rv ) {
                        fails.push_back(rhs);
                    }
                    return rv;
                }
                else if( opt_local(sp, rhs) ) {
                    DEBUG("static from local");
                    fails.push_back(rhs);
                    return false;
                }
                else {
                    BUG(sp, "Unknown lft - " << rhs);
                }
            }
            else if( const auto* lhs_l = opt_local(sp, lhs) ) {
                assert(!lhs_l->data.is_Composite());

                if( rhs.is_param() ) {
                    // Invalid
                    fails.push_back(rhs);
                    return false;
                }
                else if( /*const auto* rhs_l =*/ opt_local(sp, rhs) ) {
                    // Only valid if the source is identical? Or if the source's scope is longer than the destination
                    fails.push_back(rhs);
                    return false;
                }
                else {
                    BUG(sp, "Unknown lft - " << rhs);
                }
            }
            else {
                BUG(sp, "Unknown lifetime " << lhs);
            }
        }
        // NOTE: lhs/rhs is about assignments, not lifetime bounds
        void ensure_outlives(const Span& sp, const HIR::LifetimeRef& lhs, const HIR::LifetimeRef& rhs) const
        {
            DEBUG(lhs << " = " << rhs);
            std::vector<HIR::LifetimeRef>   failed_bounds;
            if( !check_lifetimes(sp, lhs, rhs, failed_bounds) )
            {
                ASSERT_BUG(sp, !failed_bounds.empty(), "");
                for(const HIR::LifetimeRef& b : failed_bounds)
                {
                    if(const auto* l = this->opt_local(sp, b)) {
                        TU_MATCH_HDRA( (l->data), { )
                        TU_ARMA(Composite, le) {
                            NOTE(sp, "Composite - " << le);
                            }
                        TU_ARMA(PatternBinding, le) {
                            NOTE(l->borrow_span, "Pattern binding @ " << le.pat);
                            NOTE(le.value->span(), " borrow of value here");
                            NOTE(le.borrow_point->span(), " borrowed here");
                            }
                        TU_ARMA(Node, le) {
                            NOTE(le.value->span(), "Borrow of value here");
                            NOTE(le.borrow_point->span(), " borrowed here");
                            }
                        }
                    }
                }
                if( lhs.binding == HIR::LifetimeRef::STATIC && m_in_constant_context ) {
                    // If we're in a constant context, allow a failed assignment to static
                    WARNING(sp, W0000, "IGNORE (in constant context) : Lifetime bound " << rhs << ": " << lhs << " failed - [" << failed_bounds << "]");
                    return ;
                }
                ERROR(sp, E0000, "Lifetime bound " << rhs << ": " << lhs << " failed - [" << failed_bounds << "]");
            }
        }
    };

    class ExprVisitor_Enumerate;

    struct LocalTraitResolve: public StaticTraitResolve {
        ExprVisitor_Enumerate* m_lifetime_state = nullptr;
        LocalTraitResolve(const HIR::Crate& crate): StaticTraitResolve(crate) {}
        bool replace_equalities(::HIR::TypeRef& input) const override;
    };

    static bool is_opaque(const ::HIR::TypeRef& ty) {
        if(ty.data().is_Generic())
            return true;
        if(ty.data().is_ErasedType())
            return true;
        if(ty.data().is_Path() && ty.data().as_Path().binding.is_Opaque())
            return true;
        if(ty.data().is_Path() && ty.data().as_Path().binding.is_Unbound())
            return true;
        return false;
    }

    class ExprVisitor_Enumerate: public HIR::ExprVisitorDef
    {
        friend struct LocalTraitResolve;
        const StaticTraitResolve&   m_resolve;
        const ::HIR::Function::args_t&  m_args;
        const HIR::TypeRef& m_real_ret_type;
        HIR::TypeRef m_ret_ty;
        bool remove_locals;

        LifetimeInferState& m_state;
        const std::vector<HIR::TypeRef>*    m_binding_types_ptr;

        std::vector<const HIR::ExprNode_Loop*>  m_loops;
        std::vector<const HIR::TypeRef*>    m_returns;
        std::vector<const HIR::TypeRef*>    m_yields;

    public:
        ExprVisitor_Enumerate(const StaticTraitResolve& resolve, const ::HIR::Function::args_t& args, const HIR::TypeRef& ret_ty, bool remove_locals, LifetimeInferState& state)
            : m_resolve(resolve)
            , m_args(args)
            , m_real_ret_type(ret_ty)
            , remove_locals(remove_locals)
            , m_state(state)
        {
        }

        void visit_root(HIR::ExprPtr& ep) {

            for(auto& ty : ep.m_erased_types) {
                this->visit_type(ty);
            }

            // Translate the return type's erased types
            const auto& sp = ep->span();
            DEBUG("m_real_ret_type = " << m_real_ret_type);
            m_ret_ty = clone_ty_with(sp, m_real_ret_type, [&](const HIR::TypeRef& tpl, HIR::TypeRef& rv)->bool {
                if( const auto* e = tpl.data().opt_ErasedType() )
                {
                    if( const auto* ee = e->m_inner.opt_Fcn() )
                    {
                        ASSERT_BUG(sp, ee->m_index < ep.m_erased_types.size(),
                            "Erased type index OOB - " << ee->m_origin << " " << ee->m_index << " >= " << ep.m_erased_types.size());
                        rv = ep.m_erased_types[ee->m_index].clone();
                        return true;
                    }
                }
                return false;
                });
            DEBUG("m_ret_ty = " << m_ret_ty);

            for(auto& ty : ep.m_bindings) {
                this->visit_type(ty);
                DEBUG("_" << (&ty - ep.m_bindings.data()) << ": " << ty);
            }
            this->m_binding_types_ptr = &ep.m_bindings;

            this->visit_type(ep->m_res_type);
            ep->visit(*this);
            this->equate_types(ep->m_span, m_ret_ty, ep->m_res_type);

            m_binding_types_ptr = nullptr;
        }

    private:
        const HIR::TypeRef& get_local_var_ty(const Span& sp, size_t binding) const {
            ASSERT_BUG(sp, m_binding_types_ptr, "get_loca_var_ty: m_binding_types_ptr is null");
            ASSERT_BUG(sp, binding < m_binding_types_ptr->size(), "get_local_var_ty: #" << binding << " out of range (" << m_binding_types_ptr->size() << " bindings)");
            return m_binding_types_ptr->at(binding);
        }

        void sanity_check_lft(const Span& sp, const HIR::LifetimeRef& lft) const {
            ASSERT_BUG(sp, lft != HIR::LifetimeRef() && lft.binding != HIR::LifetimeRef::INFER, "Unspecified lifetime - " << lft);
            //ASSERT_BUG(sp, !lft.is_hrl(), "Encountered HRL - " << lft);
            if( lft.is_param() ) {
                if( lft.as_param().group() == 0 ) {
                    ASSERT_BUG(sp, m_resolve.m_impl_generics && lft.as_param().idx() < m_resolve.m_impl_generics->m_lifetimes.size(), "Unexpected generic: " << lft);
                }
                if( lft.as_param().group() == 1 ) {
                    ASSERT_BUG(sp, m_resolve.m_item_generics && lft.as_param().idx() < m_resolve.m_item_generics->m_lifetimes.size(), "Unexpected generic: " << lft);
                }
            }
        }

        void equate_lifetimes(const Span& sp, const HIR::LifetimeRef& lhs, const HIR::LifetimeRef& rhs) {
            DEBUG(lhs << " := " << rhs);
            sanity_check_lft(sp, lhs);
            sanity_check_lft(sp, rhs);

            if(lhs == rhs) {
                return ;
            }

            ASSERT_BUG(sp, !lhs.is_hrl(), "Encountered HRL - " << lhs);
            ASSERT_BUG(sp, !rhs.is_hrl(), "Encountered HRL - " << rhs);

            if( auto* iv = m_state.opt_ivar(sp, lhs) ) {
                iv->sources.push_back(rhs);
            }
            if( auto* iv = m_state.opt_ivar(sp, rhs) ) {
                iv->destinations.push_back(lhs);
            }
            if( !m_state.opt_ivar(sp, lhs) && !m_state.opt_ivar(sp, rhs) ) {
                m_state.ensure_outlives(sp, lhs, rhs);
            }
        }

        void equate_node_types(const HIR::ExprNode& lhs, const HIR::ExprNodeP& rhs) {
            assert(rhs);
            this->equate_types(rhs->span(), lhs.m_res_type, rhs->m_res_type);
        }
        void equate_pps(const Span& sp, const HIR::PathParams& lhs, const HIR::PathParams& rhs) {
            ASSERT_BUG(sp, lhs.m_lifetimes.size() == rhs.m_lifetimes.size(), "equate_pps: " << lhs << " = " << rhs);
            for(size_t i = 0; i < lhs.m_lifetimes.size(); i ++) {
                equate_lifetimes(sp, lhs.m_lifetimes[i], rhs.m_lifetimes[i]);
            }
            ASSERT_BUG(sp, lhs.m_types.size() == rhs.m_types.size(), "equate_pps: " << lhs << " = " << rhs);
            for(size_t i = 0; i < lhs.m_types.size(); i ++) {
                equate_types(sp, lhs.m_types[i], rhs.m_types[i]);
            }
            // Ignore values
        }
        void equate_traitpath(const Span& sp, const HIR::TraitPath& lhs, const HIR::TraitPath& rhs) {
            //if( le.m_hrtbs && re.m_hrtbs ) {
            //}
            //else if( le.m_hrtbs || re.m_hrtbs ) {
            //    TODO(sp, lhs << " == " << rhs);
            //}
            auto pp_l = lhs.m_hrtbs ? lhs.m_hrtbs->make_empty_params(true) : HIR::PathParams();
            auto pp_r = rhs.m_hrtbs ? rhs.m_hrtbs->make_empty_params(true) : HIR::PathParams();
            visit_path_params(pp_l);
            visit_path_params(pp_r);
            auto ms_l = MonomorphHrlsOnly(pp_l);
            auto ms_r = MonomorphHrlsOnly(pp_r);
            equate_pps(sp, ms_l.monomorph_path_params(sp, lhs.m_path.m_params, false), ms_r.monomorph_path_params(sp, rhs.m_path.m_params, false));
            ASSERT_BUG(sp, lhs.m_type_bounds.size() == rhs.m_type_bounds.size(), "");
            ASSERT_BUG(sp, lhs.m_trait_bounds.size() == rhs.m_trait_bounds.size(), "");
            for(const auto& l : lhs.m_type_bounds)
            {
                const auto& r = rhs.m_type_bounds.at(l.first);
                equate_types(sp, ms_l.monomorph_type(sp, l.second.type, false), ms_r.monomorph_type(sp, r.type, false));
            }
            for(const auto& l : lhs.m_trait_bounds)
            {
                const auto& r = rhs.m_trait_bounds.at(l.first);
                for(size_t i = 0; i < l.second.traits.size(); i++)
                {
                    equate_traitpath(sp, ms_l.monomorph_traitpath(sp, l.second.traits[i], false), ms_r.monomorph_traitpath(sp, r.traits[i], false));
                }
            }
        }

        void equate_paths(const Span& sp, const HIR::Path& lhs, const HIR::Path& rhs) {
            TU_MATCH_HDRA( (lhs.m_data, rhs.m_data), { )
            TU_ARMA(Generic, le, re) {
                equate_pps(sp, le.m_params, re.m_params);
                }
            TU_ARMA(UfcsKnown, le, re) {
                equate_types(sp, le.type, re.type);
                equate_pps(sp, le.params, re.params);
                equate_pps(sp, le.trait.m_params, re.trait.m_params);
                }
            TU_ARMA(UfcsInherent, le, re) {
                equate_types(sp, le.type, re.type);
                equate_pps(sp, le.params, re.params);
                equate_pps(sp, le.impl_params, re.impl_params);
                }
            TU_ARMA(UfcsUnknown, le, re) {
                BUG(sp, lhs << " := " << rhs << ": UfcsUnknown");
                }
            }
        }

        void equate_types(const Span& sp, const HIR::TypeRef& lhs, const HIR::TypeRef& rhs) {
            // TODO: Only print when at the top-level
            TRACE_FUNCTION_F(lhs << " == " << rhs);
            // Match the types, letting lifetimes equate
            if(rhs.data().is_Diverge())
                return ;

#if 0
            // HACK: Handle equality between expanded and unexpanded closures.
            if( lhs.data().is_Closure() && rhs.data().is_Path() ) {
                const auto& le = lhs.data().as_Closure();
                const auto& re = rhs.data().as_Path();
                //le.node->m_obj_path_base
                return ;
            }
#endif

            if( TU_TEST1(lhs.data(), ErasedType, .m_inner.is_Alias()) ) {
                const auto& le = lhs.data().as_ErasedType().m_inner.as_Alias();
                const auto* ee = le.inner.get();
                if( TU_TEST2(rhs.data(), ErasedType, .m_inner, Alias, .inner.get() == ee) ) {
                }
                else if( TU_TEST1(rhs.data(), ErasedType, .m_inner.is_Alias()) ) {
                }
                else {
                    DEBUG("LHS is TAIT");
                    // TODO: Only expand if valid module?
                    auto it = m_state.m_tait_values.insert(std::make_pair(ee, std::make_pair(
                        le.params.clone(), rhs.clone()
                        )));
                    equate_types(sp, it.first->second.second, rhs);
                    return;
                }
            }
            else if( TU_TEST1(rhs.data(), ErasedType, .m_inner.is_Alias()) ) {
                const auto& re = rhs.data().as_ErasedType().m_inner.as_Alias();
                const auto* ee = re.inner.get();
                // TODO: Only expand if valid module?
                auto it = m_state.m_tait_values.insert(std::make_pair(ee, std::make_pair(
                    re.params.clone(), lhs.clone()
                    )));
                equate_types(sp, lhs, it.first->second.second);
                return;
            }

            ASSERT_BUG(sp, lhs.data().tag() == rhs.data().tag(), "Mismatched types: " << lhs << " != " << rhs);
            TU_MATCH_HDRA( (lhs.data(), rhs.data()), { )
            TU_ARMA(Infer, l, r) BUG(sp, "");
            TU_ARMA(Primitive, l, r) {}
            TU_ARMA(Generic, l, r) {}
            TU_ARMA(Diverge, l, r) {}
            TU_ARMA(Path, l, r) {
                equate_paths(sp, l.path, r.path);
                }
            TU_ARMA(TraitObject, l, r) {
                this->equate_lifetimes(sp, l.m_lifetime, r.m_lifetime);
                this->equate_traitpath(sp, l.m_trait, r.m_trait);
                ASSERT_BUG(sp, l.m_markers.size() == r.m_markers.size(), "");
                for(size_t i = 0; i < l.m_markers.size(); i ++)
                    this->equate_pps(sp, l.m_markers[i].m_params, r.m_markers[i].m_params);
                }
            TU_ARMA(ErasedType, l, r) {
                for(size_t i = 0; i < l.m_lifetime_bounds.size(); i++)
                    this->equate_lifetimes(sp, l.m_lifetime_bounds[i], r.m_lifetime_bounds[i]);
                ASSERT_BUG(sp, l.m_traits.size() == r.m_traits.size(), "");
                for(size_t i = 0; i < l.m_traits.size(); i ++)
                    this->equate_traitpath(sp, l.m_traits[i], r.m_traits[i]);
                this->equate_pps(sp, l.m_use, r.m_use);
                TU_MATCH_HDRA( (l.m_inner, r.m_inner),{ )
                TU_ARMA(Known, l_ie, r_ie) {
                    this->equate_types(sp, l_ie, r_ie);
                    }
                TU_ARMA(Fcn, l_ie, r_ie) {
                    this->equate_paths(sp, l_ie.m_origin, r_ie.m_origin);
                    }
                TU_ARMA(Alias, l_ie, r_ie) {
                    this->equate_pps(sp, l_ie.params, r_ie.params);
                    }
                }
                }
            TU_ARMA(Array, l, r) {
                this->equate_types(sp, l.inner, r.inner);
                }
            TU_ARMA(Slice, l, r) {
                this->equate_types(sp, l.inner, r.inner);
                }
            TU_ARMA(Tuple, l, r) {
                ASSERT_BUG(sp, l.size() == r.size(), "");
                for(size_t i = 0; i < l.size(); i ++)
                    this->equate_types(sp, l[i], r[i]);
                }
            TU_ARMA(Borrow, l, r) {
                this->equate_lifetimes(sp, l.lifetime, r.lifetime);
                this->equate_types(sp, l.inner, r.inner);
                }
            TU_ARMA(Pointer, l, r) {
                this->equate_types(sp, l.inner, r.inner);
                }
            TU_ARMA(NamedFunction, l, r) {
                equate_paths(sp, l.path, r.path);
                }
            TU_ARMA(Function, l, r) {
                // Handling required?
                }
            TU_ARMA(NodeType, l, r) {
                ASSERT_BUG(sp, l == r, "Mismatched types: " << lhs << " != " << rhs);
                }
            }
        }
        std::vector<HIR::LifetimeRef>   m_pattern_lifetime_stack;
        void equate_pattern_binding(const Span& sp, const HIR::PatternBinding& b, const HIR::TypeRef& src_ty, const HIR::ExprNode& root_node)
        {
            const HIR::TypeRef*   cur_ty = &src_ty;
            if( b.m_implicit_deref_count ) {
                TODO(sp, "Apply implicit derefs (binding) " << *cur_ty << " deref " << b.m_implicit_deref_count);
            }
            auto get_lft = [&]()->HIR::LifetimeRef {
                if(m_pattern_lifetime_stack.empty()) {
                    return get_borrow_lifetime(root_node, [&](const HIR::ExprNode& value){ return m_state.allocate_local(sp, root_node, value, b); });
                }
                else {
                    return m_pattern_lifetime_stack.back();
                }
            };
            const auto& slot = get_local_var_ty(sp, b.m_slot);
            switch(b.m_type)
            {
            case ::HIR::PatternBinding::Type::Move:
                this->equate_types(sp, slot, *cur_ty);
                break;
            case ::HIR::PatternBinding::Type::MutRef:
                this->equate_types(sp, slot, HIR::TypeRef::new_borrow(HIR::BorrowType::Unique, cur_ty->clone(), get_lft()));
                break;
            case ::HIR::PatternBinding::Type::Ref:
                this->equate_types(sp, slot, HIR::TypeRef::new_borrow(HIR::BorrowType::Shared, cur_ty->clone(), get_lft()));
                break;
            }
        }
        void equate_pattern(const Span& sp, const HIR::Pattern& pat, const HIR::TypeRef& src_ty, const HIR::ExprNode& root_node)
        {
            // Match pattern into the type (including bindings)
            if(src_ty.data().is_Diverge())
                return ;
            for(const auto& b : pat.m_bindings)
            {
                equate_pattern_binding(sp, b, src_ty, root_node);
            }
            const HIR::TypeRef*   cur_ty = &src_ty;
            size_t start_lifetime_stack_height = m_pattern_lifetime_stack.size();
            for(auto i = pat.m_implicit_deref_count; i --; )
            {
                if( const auto* tep = cur_ty->data().opt_Borrow() ) {
                    m_pattern_lifetime_stack.push_back(tep->lifetime);
                    cur_ty = &tep->inner;
                }
                else {
                    TODO(sp, "Apply implicit derefs - " << *cur_ty);
                }
            }
            TU_MATCH_HDRA( (pat.m_data),  { )
            TU_ARMA(Any, _) {}
            TU_ARMA(Slice, pe) {
                const auto& ity = cur_ty->data().is_Array() ? cur_ty->data().as_Array().inner : cur_ty->data().as_Slice().inner;
                for(const auto& subpat : pe.sub_patterns) {
                    equate_pattern(sp, subpat, ity, root_node); 
                }
                }
            TU_ARMA(SplitSlice, pe) {
                const auto& ity = cur_ty->data().is_Array() ? cur_ty->data().as_Array().inner : cur_ty->data().as_Slice().inner;
                for(const auto& subpat : pe.leading) {
                    equate_pattern(sp, subpat, ity, root_node); 
                }
                if(pe.extra_bind.is_valid()) {
                    equate_pattern_binding(sp, pe.extra_bind, *cur_ty, root_node);
                }
                for(const auto& subpat : pe.trailing) {
                    equate_pattern(sp, subpat, ity, root_node); 
                }
                }
            TU_ARMA(Tuple, pe) {
                const auto& te = cur_ty->data().as_Tuple();
                ASSERT_BUG(sp, pe.sub_patterns.size() == te.size(), "");
                for(size_t i = 0; i < te.size(); i ++)
                {
                    equate_pattern(sp, pe.sub_patterns[i], te[i], root_node);
                }
                }
            TU_ARMA(SplitTuple, pe) {
                const auto& te = cur_ty->data().as_Tuple();
                ASSERT_BUG(sp, pe.leading.size() + pe.trailing.size() <= te.size(), "");
                for(size_t i = 0; i < pe.leading.size(); i ++)
                {
                    equate_pattern(sp, pe.leading[i], te[i], root_node);
                }
                for(size_t i = 0; i < pe.trailing.size(); i ++)
                {
                    equate_pattern(sp, pe.trailing[i], te[te.size() - pe.trailing.size() + i], root_node);
                }
                }
            TU_ARMA(Box, pe) {
                const auto& gp = cur_ty->data().as_Path().path.m_data.as_Generic();
                ASSERT_BUG(sp, gp.m_path == m_resolve.m_lang_Box, *cur_ty);
                const HIR::TypeRef& inner_ty = gp.m_params.m_types.at(0);
                equate_pattern(sp, *pe.sub, inner_ty, root_node);
                }
            TU_ARMA(Ref, pe) {
                const auto& te = cur_ty->data().as_Borrow();
                m_pattern_lifetime_stack.push_back(te.lifetime);
                equate_pattern(sp, *pe.sub, te.inner, root_node);
                m_pattern_lifetime_stack.pop_back();
                }

            TU_ARMA(Or, pe) {
                for(auto& subpat : pe) {
                    equate_pattern(sp, subpat, *cur_ty, root_node);
                }
                }
            
            TU_ARMA(PathTuple, pe) {
                const HIR::Struct::Data::Data_Tuple*    flds = nullptr;
                //HIR::TypeRef    ty;
                TU_MATCH_HDRA( (pe.binding), { )
                TU_ARMA(Unbound, pbe) {}
                TU_ARMA(Enum, pbe) {
                    const auto& sub_ty = pbe.ptr->m_data.as_Data().at(pbe.var_idx).type;
                    const auto& str = *sub_ty.data().as_Path().binding.as_Struct();
                    flds = &str.m_data.as_Tuple();
                    // Assume that the enum and the inner struct have the same parameter set
                    // TODO: Equate type?
                    }
                TU_ARMA(Struct, pbe) {
                    flds = &pbe->m_data.as_Tuple();
                    // TODO: Equate type?
                    }
                }
                assert(flds);
                auto ms = MonomorphStatePtr(nullptr, &pe.path.m_data.as_Generic().m_params, nullptr);
                ASSERT_BUG(sp, pe.leading.size() + pe.trailing.size() <= flds->size(), "");
                size_t trailing_start = flds->size() - pe.trailing.size();
                for(size_t i = 0; i < flds->size(); i++)
                {
                    if( i < pe.leading.size() ) {
                        auto exp_ty = m_resolve.monomorph_expand(sp, (*flds)[i].ent, ms);
                        equate_pattern(sp, pe.leading[i], exp_ty, root_node);
                    }
                    else if( i > trailing_start ) {
                        auto exp_ty = m_resolve.monomorph_expand(sp, (*flds)[i].ent, ms);
                        equate_pattern(sp, pe.trailing[i - trailing_start], exp_ty, root_node);
                    }
                    else {
                    }
                }
                }
            TU_ARMA(PathNamed, pe) {
                //this->equate_types(sp, 
                //TODO(sp, "PathNamed patterns w/ " << *cur_ty);
                }

            // No need to check lifetimes of values (they have zero impact)
            TU_ARMA(Value, pe) {
                }
            TU_ARMA(Range, pe) {
                }
            TU_ARMA(PathValue, pe) {
                // Might need to equate types?
                }
            }
            assert(m_pattern_lifetime_stack.size() >= start_lifetime_stack_height);
            m_pattern_lifetime_stack.resize(start_lifetime_stack_height);
        }

        struct Monomorph_AddLifetimes: public Monomorphiser {
            ExprVisitor_Enumerate& parent;

            ::HIR::TypeRef get_type(const Span& sp, const ::HIR::GenericRef& g) const override {
                return HIR::TypeRef(g.name, g.binding);
            }
            ::HIR::ConstGeneric get_value(const Span& sp, const ::HIR::GenericRef& g) const override {
                return g;
            }
            ::HIR::LifetimeRef get_lifetime(const Span& sp, const ::HIR::GenericRef& g) const override {
                // Placeholder.
                if( g.group() == ::HIR::GENERIC_Placeholder ) {
                    // HACK: Ideally - placeholders won't be here, but just in case...
                    return parent.m_state.allocate_ivar(sp);
                }
                return HIR::LifetimeRef(g.binding);
            }

            ::HIR::LifetimeRef monomorph_lifetime(const Span& sp, const ::HIR::LifetimeRef& tpl) const override {
                if( tpl.binding == HIR::LifetimeRef::UNKNOWN || tpl.binding == HIR::LifetimeRef::INFER) {
                    return parent.m_state.allocate_ivar(sp);
                }
                else if( false && tpl.is_hrl() ) {
                    return parent.m_state.allocate_ivar(sp);
                }
                else {
                    if(parent.remove_locals && (parent.m_state.is_local(tpl) || parent.m_state.is_ivar(tpl)) ) {
                        return parent.m_state.allocate_ivar(sp);
                    }
                    ASSERT_BUG(sp, !parent.m_state.is_ivar(tpl), "Found ivar while adding lifetimes - " << tpl);
                    ASSERT_BUG(sp, !parent.m_state.is_local(tpl), "Found local while adding lifetimes - " << tpl);
                    return Monomorphiser::monomorph_lifetime(sp, tpl);
                }
            }

            Monomorph_AddLifetimes(ExprVisitor_Enumerate& parent): parent(parent) {}
        };
        Monomorph_AddLifetimes get_monomorph_add() {
            return Monomorph_AddLifetimes(*this);
        }
        void visit_path(::HIR::Visitor::PathContext pc, HIR::Path& p) override {
            Span    sp;
            TRACE_FUNCTION_FR(p, p);
            //if( auto* pe = p.m_data.opt_UfcsKnown() ) {
            //    if( pe->trait.m_hrls && !pe->trait.m_hrls->is_empty() ) {
            //        // Remove this level of HRL
            //        auto pp = pe->trait.m_hrls->make_empty_params(true);
            //        pe->trait.m_hrls.reset();
            //        pe->trait = MonomorphHrlsOnly(pp).monomorph_genericpath(sp, pe->trait);
            //    }
            //}
            p = get_monomorph_add().monomorph_path(sp, p, false);
        }
        void visit_path_params(HIR::PathParams& pps) override {
            TRACE_FUNCTION_FR(pps, pps);
            pps = get_monomorph_add().monomorph_path_params(Span(), pps, false);
        }
        void visit_type(HIR::TypeRef& ty) override {
            TRACE_FUNCTION_FR(ty, ty);
            ty = get_monomorph_add().monomorph_type(Span(), ty, false);
        }


        /// <summary>
        /// Obtain the root lifetime for a borrow operation
        /// </summary>
        /// <param name="node">Root node</param>
        /// <param name="cb">Function to get a local lifetime from the root value</param>
        /// <returns>Lifetime reference (a local, or the first dereferenced borrow)</returns>
        HIR::LifetimeRef get_borrow_lifetime(const ::HIR::ExprNode& node, std::function<HIR::LifetimeRef(const ::HIR::ExprNode&)> cb) const
        {
            // Determine a suitable lifetime for this value
            // - Deref? Grab lifetime of deref-ed value
            // - Static - 'static
            // - Local variable (really anything else) - allocate local
            struct V: public HIR::ExprVisitor {
                const ExprVisitor_Enumerate& m_parent;
                std::function<HIR::LifetimeRef(const ::HIR::ExprNode&)>& m_cb;
                HIR::LifetimeRef    m_res;

                V(const ExprVisitor_Enumerate& parent, std::function<HIR::LifetimeRef(const ::HIR::ExprNode&)>& cb)
                    : m_parent(parent)
                    , m_cb(cb)
                {
                }

                void local(const HIR::ExprNode& cur) {
                    m_res = m_cb(cur);
                }

                #define NV(nt)   void visit(HIR::nt& node) override { local(node); }
                NV(ExprNode_Block)
                NV(ExprNode_ConstBlock)
                NV(ExprNode_Asm)
                NV(ExprNode_Asm2)
                NV(ExprNode_Return)
                NV(ExprNode_Yield)
                NV(ExprNode_AWait)
                NV(ExprNode_Let)
                NV(ExprNode_Loop)
                NV(ExprNode_LoopControl)
                NV(ExprNode_Match)
                NV(ExprNode_If)

                NV(ExprNode_Assign)
                NV(ExprNode_BinOp)
                NV(ExprNode_UniOp)
                NV(ExprNode_Borrow)
                NV(ExprNode_RawBorrow)
                NV(ExprNode_Cast)
                //NV(ExprNode_Unsize)

                NV(ExprNode_Emplace)

                NV(ExprNode_TupleVariant);
                NV(ExprNode_CallPath);
                NV(ExprNode_CallValue);
                NV(ExprNode_CallMethod);

                NV(ExprNode_Literal);
                NV(ExprNode_UnitVariant);
                NV(ExprNode_Variable);
                NV(ExprNode_ConstParam);

                NV(ExprNode_StructLiteral);
                NV(ExprNode_Tuple);
                NV(ExprNode_ArrayList);
                NV(ExprNode_ArraySized);

                NV(ExprNode_Closure);
                NV(ExprNode_Generator);
                NV(ExprNode_GeneratorWrapper);
                NV(ExprNode_AsyncBlock);
                #undef NV

                void visit(HIR::ExprNode_Unsize& node) override {
                    // If the inner type is an array, then this will eventually become a `deref(unsize(borrow(...)))`
                    if( node.m_value->m_res_type.data().is_Array() ) {
                        node.m_value->visit(*this);
                    }
                    else {
                        local(node);
                    }
                }

                void visit(HIR::ExprNode_PathValue& node) override {
                    // If this is a static, return 'static
                    MonomorphState  ms;
                    auto v = m_parent.m_resolve.get_value(node.span(), node.m_path, ms, /*signature_only*/true);
                    DEBUG("get_borrow_lifetime: " << node.m_path << " : " << v.tag_str());
                    if( v.is_Static() ) {
                        m_res = HIR::LifetimeRef::new_static();
                    }
                    else {
                        // Should this be possible? Maybe if there's generics going on.
                        local(node);
                    }
                }
                void visit(HIR::ExprNode_Deref& node) override {
                    TU_MATCH_HDRA( (node.m_value->m_res_type.data()), {)
                    default:
                        // For deref impls, just propagate through (the signature is `fn deref(&self) -> &Target`)
                        node.m_value->visit(*this);
                    TU_ARMA(Pointer, te) {
                        m_res = HIR::LifetimeRef::new_static();
                        }
                    TU_ARMA(Borrow, te) {
                        // TODO: Is anything needed for mut reborrow chains?
                        // `&mut **foo` - this will use the outermost-most lifetime (i.e. the inner borrow)
                        // - But, it should use the outermost for mut?
                        m_res = te.lifetime;
                        }
                    }
                }
                void visit(HIR::ExprNode_Index& node) override {
                    // Indexing (like field access) propagates inwards
                    node.m_value->visit(*this);
                }
                void visit(HIR::ExprNode_Field& node) override {
                    // Field access propagates inwards
                    node.m_value->visit(*this);
                }

            } v { *this, cb };
            const_cast<HIR::ExprNode&>(node).visit(v);
            ASSERT_BUG(node.span(), v.m_res != HIR::LifetimeRef(), "");
            return v.m_res;
        }

    public:
        void visit_node_ptr(::std::unique_ptr< ::HIR::ExprNode>& node_ptr) override {
            assert(node_ptr);
            visit_type(node_ptr->m_res_type);
            DEBUG("RES: " << node_ptr->m_res_type);
            node_ptr->visit(*this);
        }

        void visit(::HIR::ExprNode_Block& node) override {
            HIR::ExprVisitorDef::visit(node);
            if( node.m_value_node ) {
                equate_node_types(node, node.m_value_node);
            }
        }
        void visit(::HIR::ExprNode_Asm& node) override {
            HIR::ExprVisitorDef::visit(node);
        }
        void visit(::HIR::ExprNode_Asm2& node) override {
            HIR::ExprVisitorDef::visit(node);
        }
        void visit(::HIR::ExprNode_Return& node) override {
            HIR::ExprVisitorDef::visit(node);
            if( node.m_value ) {
                equate_types(node.m_value->span(), m_returns.empty() ? m_ret_ty : *m_returns.back(), node.m_value->m_res_type);
            }
        }
        void visit(::HIR::ExprNode_Yield& node) override {
            HIR::ExprVisitorDef::visit(node);
            if( node.m_value ) {
                ASSERT_BUG(node.span(), !m_yields.empty(), "Yield without a generator block?");
                equate_types(node.m_value->span(), *m_yields.back(), node.m_value->m_res_type);
            }
        }
        void visit(::HIR::ExprNode_AWait& node) override {
            HIR::ExprVisitorDef::visit(node);
            auto t = ::HIR::TypeRef::new_path(::HIR::Path(node.m_value->m_res_type.clone(), m_resolve.m_lang_Future, "Output"), {});
            m_resolve.expand_associated_types(node.span(), t);
            equate_types(node.span(), node.m_res_type, t);
        }
        void visit(::HIR::ExprNode_Let& node) override {
            HIR::ExprVisitorDef::visit(node);
            if( node.m_value ) {
                equate_pattern(node.span(), node.m_pattern, node.m_value->m_res_type, *node.m_value);
            }
        }
        void visit(::HIR::ExprNode_Loop& node) override {
            m_loops.push_back(&node);
            HIR::ExprVisitorDef::visit(node);
            m_loops.pop_back(/*&node*/);
        }
        void visit(::HIR::ExprNode_LoopControl& node) override {
            HIR::ExprVisitorDef::visit(node);
            if( node.m_value ) {
                auto it = ::std::find(this->m_loops.rbegin(), this->m_loops.rend(), node.m_target_node);
                ASSERT_BUG(node.span(), it != this->m_loops.rend(), "Loop target node not found in the loop stack");
                equate_types(node.m_value->span(), node.m_target_node->m_res_type, node.m_value->m_res_type);
            }
        }
        void visit(::HIR::ExprNode_Match& node) override {
            HIR::ExprVisitorDef::visit(node);
            for(auto& arm : node.m_arms) {
                for(auto& pat : arm.m_patterns)
                    equate_pattern(node.span(), pat, node.m_value->m_res_type, *node.m_value);
                equate_node_types(node, arm.m_code);
            }
        }
        void visit(::HIR::ExprNode_If& node) override {
            HIR::ExprVisitorDef::visit(node);
            // Equate both arms to the output
            if( node.m_false ) {
                equate_node_types(node, node.m_true );
                equate_node_types(node, node.m_false);
            }
        }

        void visit(::HIR::ExprNode_Assign& node) override {
            HIR::ExprVisitorDef::visit(node);
            if( node.m_op == HIR::ExprNode_Assign::Op::None ) {
                equate_types(node.span(), node.m_slot->m_res_type, node.m_value->m_res_type);
            }
        }
        void visit(::HIR::ExprNode_BinOp& node) override {
            HIR::ExprVisitorDef::visit(node);
            // No lifetimes involved.
        }
        void visit(::HIR::ExprNode_UniOp& node) override {
            HIR::ExprVisitorDef::visit(node);
            // No lifetimes involved.
        }

        void visit(::HIR::ExprNode_Borrow& node) override {
            HIR::ExprVisitorDef::visit(node);
            const auto& val_node = *node.m_value;
            auto lft = node.m_is_valid_static_borrow_constant
                ? ::HIR::LifetimeRef::new_static()
                : get_borrow_lifetime(val_node, [&](const HIR::ExprNode& value)->HIR::LifetimeRef{
                    DEBUG("LOCAL: " << typeid(val_node).name() << " " << typeid(value).name());
                    return m_state.allocate_local(node, value);
                    });
            equate_types(node.span(), node.m_res_type, HIR::TypeRef::new_borrow(node.m_type, val_node.m_res_type.clone(), lft));
        }
        void visit(::HIR::ExprNode_RawBorrow& node) override {
            HIR::ExprVisitorDef::visit(node);
            // No lifetime!
        }
        void visit(::HIR::ExprNode_Cast& node) override {
            const auto& sp = node.span();
            HIR::ExprVisitorDef::visit(node);
            DEBUG("Cast:");
            const auto& dst = node.m_dst_type;
            const auto& src = node.m_value->m_res_type;
            if( dst == src ) {
                this->equate_types(node.span(), dst, src);
            }
            else if( const auto* de = dst.data().opt_Function() ) {
                auto pp = de->hrls.make_empty_params(true);
                visit_path_params(pp);
                auto ms_d = MonomorphHrlsOnly(pp);
                if( src.data().is_Function() || src.data().is_NamedFunction() ) {
                    ::HIR::TypeRef  tmp;
                    const auto* se = src.data().opt_Function();
                    if( !se ) {
                        tmp = ::HIR::TypeRef( src.data().as_NamedFunction().decay(node.span()) );
                        DEBUG("tmp = " << tmp);
                        DEBUG("dst = " << dst);
                        se = tmp.data().opt_Function();
                    }

                    // TODO: Check that the HRLs are compatible
                    //ASSERT_BUG(node.span(), se->hrls.m_lifetimes.size() == de->hrls.m_lifetimes.size(),
                    //    "HRL count mismatch in casting " << src << " to " << dst);
                    auto pp_s = se->hrls.make_empty_params(true);
                    visit_path_params(pp_s);
                    auto ms_s = MonomorphHrlsOnly(pp_s);
                    ASSERT_BUG(node.span(), de->m_arg_types.size() == se->m_arg_types.size(), "");
                    for(size_t i = 0; i < de->m_arg_types.size(); i ++) {
                        this->equate_types(node.span(), ms_d.monomorph_type(sp, de->m_arg_types[i]), ms_s.monomorph_type(sp, se->m_arg_types[i]));
                    }
                    this->equate_types(node.span(), ms_d.monomorph_type(sp, de->m_rettype), ms_s.monomorph_type(sp, se->m_rettype));
                }
                else if( const auto* node_pp = TU_OPT1(src.data(), NodeType, .opt_Closure()) ) {
                    const HIR::ExprNode_Closure& cnode = **node_pp;
                    ASSERT_BUG(node.span(), de->m_arg_types.size() == cnode.m_args.size(), "");
                    for(size_t i = 0; i < de->m_arg_types.size(); i ++) {
                        this->equate_types(node.span(), ms_d.monomorph_type(sp, de->m_arg_types[i]), cnode.m_args[i].second);
                    }
                    this->equate_types(node.span(), ms_d.monomorph_type(sp, de->m_rettype), cnode.m_return);
                }
                else {
                    TODO(node.span(), "Propagate lifetimes through cast - " << dst << " := " << src);
                }
            }
            else if( dst.data().is_Primitive() || src.data().is_Primitive() ) {
                // Nothing to do
            }
            else if( dst.data().is_Pointer() || src.data().is_Pointer() ) {
                // TODO: What about trait objects? Should they maintain their internal liftimes?
            }
            else {
                TODO(node.span(), "Propagate lifetimes through cast - " << dst << " := " << src);
            }
            this->equate_types(node.span(), node.m_res_type, dst);
        }

        /// Extract lifetimes from a type and equate with the target lifetime.
        void equate_type_lifetimes(const Span& sp, const HIR::LifetimeRef& dst_lft, const HIR::TypeRef& ty)
        {
            struct V: public HIR::Visitor {
                ExprVisitor_Enumerate& parent;
                const Span& sp;
                const HIR::LifetimeRef& dst_lft;
                std::vector<HIR::PathParams>    m_hrls;

                V( ExprVisitor_Enumerate& parent, const Span& sp, const HIR::LifetimeRef& dst_lft)
                    : parent(parent)
                    , sp(sp)
                    , dst_lft(dst_lft)
                {
                }

                void equate_lifetime(const HIR::LifetimeRef& src) {
                    if( src.is_hrl() ) {
                        ASSERT_BUG(sp, m_hrls.size() > 0, "Encountered HRL with no HRL in the stack - " << dst_lft << " = " << src);
                        //parent.equate_lifetimes(sp, dst_lft, m_hrls.back().m_lifetimes[src.binding & 0xFF]);
                    }
                    else {
                        parent.equate_lifetimes(sp, dst_lft, src);
                    }
                }

                void visit_trait_path(HIR::TraitPath& tp) override {
                    // Handle HRLs
                    if( tp.m_hrtbs ) {
                        m_hrls.push_back(tp.m_hrtbs->make_empty_params(true));
                    }
                    HIR::Visitor::visit_trait_path(tp);
                    if( tp.m_hrtbs ) {
                        m_hrls.pop_back();
                    }
                }
                void visit_type(HIR::TypeRef& t) override {
                    if(t.data().is_Borrow())
                        equate_lifetime(t.data().as_Borrow().lifetime);

                    bool hrl_pushed = false;
                    auto push_hrls = [&](const HIR::GenericParams& hrls_def) {
                        hrl_pushed = true;
                        m_hrls.push_back(hrls_def.make_empty_params(true));
                        //for(auto& l : m_hrls.back().m_lifetimes)
                        //    l = parent.m_state.allocate_ivar(sp);
                    };

                    if(is_opaque(t)) {
                        // Iterate type lifetime bounds
                        parent.m_state.iterate_type_lifetime_bounds(t, [&](const HIR::LifetimeRef& lft)->bool {
                            equate_lifetime(lft);
                            return false;
                            });
                        if( t.data().is_Generic() || t.data().is_Path() ) {
                            // TODO: If the above didn't return anything, then assign a "only this function" liftime
                        }
                        else if( const auto* e = t.data().opt_ErasedType() ) {
                            for(const auto& lft : e->m_lifetime_bounds)
                                equate_lifetime(lft);
                            //if( e->m_trait.m_hrls ) {
                            //    push_hrls(*e->m_trait.m_hrls);
                            //}
                        }
                        else {
                            TODO(sp, "Get lifetime (from bounds) for opaque type - " << t);
                        }
                    }

                    if( const auto* e = t.data().opt_TraitObject() ) {
                        // Get the lifetime
                        equate_lifetime(e->m_lifetime);
                    }
                    if( const auto* e = t.data().opt_Function() ) {
                        push_hrls(e->hrls);
                    }
                    if( const auto* e = t.data().opt_ErasedType() ) {
                        for(const auto& l : e->m_lifetime_bounds) {
                            equate_lifetime(l);
                        }
                    }
                    if( t.data().is_Path() && t.data().as_Path().path.m_data.is_Generic() ) {
                        for(const auto& l : t.data().as_Path().path.m_data.as_Generic().m_params.m_lifetimes)
                            equate_lifetime(l);
                    }
                    HIR::Visitor::visit_type(t);
                    if( hrl_pushed ) {
                        m_hrls.pop_back();
                    }
                }
            } v { *this, sp, dst_lft };
            DEBUG(ty);
            v.visit_type(const_cast<HIR::TypeRef&>(ty));
        }
        void unsize_types(const Span& sp, const ::HIR::TypeRef& dst, const ::HIR::TypeRef& src) {
            if(dst == src) {
                this->equate_types(sp, dst, src);
            }
            else if( dst.data().is_Slice() && src.data().is_Array() ) {
                this->equate_types(sp,
                    dst.data().as_Slice().inner,
                    src.data().as_Array().inner
                    );
            }
            else if( dst.data().is_TraitObject() ) {
                const auto& dst_lft = dst.data().as_TraitObject().m_lifetime;
                equate_type_lifetimes(sp, dst_lft, src);
            }
            // If either side is an opaque type (generic/erased/opaque) then just look for an impl
            else if( is_opaque(dst) || is_opaque(src) ) {
                HIR::PathParams pp;
                pp.m_types.push_back(dst.clone());
                m_resolve.find_impl(sp, m_resolve.m_lang_Unsize, pp, src, [&](ImplRef ir, bool is_fuzzed)->bool {
                    equate_types(sp, ir.get_impl_type(), src);
                    equate_types(sp, dst, ir.get_trait_params().m_types.at(0));
                    return false;
                    });
            }
            // If the inner is a path, it must be a struct with an `Unsize` impl somewhere
            else if( dst.data().is_Path() ) {
                ASSERT_BUG(sp, src.data().is_Path(), "");
                ASSERT_BUG(sp, dst.data().as_Path().binding == src.data().as_Path().binding, "");
                ASSERT_BUG(sp, dst.data().as_Path().binding.is_Struct(), "");
                const ::HIR::Struct& str = *dst.data().as_Path().binding.as_Struct();
                const auto& dst_p = dst.data().as_Path().path.m_data.as_Generic();
                const auto& src_p = src.data().as_Path().path.m_data.as_Generic();
                auto param = str.m_struct_markings.unsized_param;

                return unsize_types(sp, dst_p.m_params.m_types.at(param), src_p.m_params.m_types.at(param));
            }
            // TODO: Look for an Unsize impl?
            else {
                TODO(sp, "Propagate lifetimes through unsize - " << dst << " := " << src);
            }
        }
        void coerce_unsize_types(const Span& sp, const ::HIR::TypeRef& dst, const ::HIR::TypeRef& src) {
            if(dst == src) {
                this->equate_types(sp, dst, src);
            }
            else if( auto* dst_te = dst.data().opt_Borrow() ) {
                auto& src_te = src.data().as_Borrow();
                // Must be from a borrow
                this->equate_lifetimes(sp, dst_te->lifetime, src_te.lifetime);

                unsize_types(sp, dst_te->inner, src_te.inner);
            }
            else if( const auto* dst_te = dst.data().opt_Pointer() ) {
                if( const auto* src_te = src.data().opt_Borrow() ) {
                    unsize_types(sp, dst_te->inner, src_te->inner);
                }
                else if( const auto* src_te = src.data().opt_Pointer() ) {
                    unsize_types(sp, dst_te->inner, src_te->inner);
                }
                else {
                    TODO(sp, "Propagate lifetimes through unsize - " << dst << " := " << src);
                }
            }
            else if( dst.data().is_Slice() && src.data().is_Array() ) {
                this->equate_types(sp, dst.data().as_Slice().inner, src.data().as_Array().inner);
            }
            // If either side is an opaque type (generic/erased/opaque) then just look for an impl
            else if( is_opaque(dst) || is_opaque(src) ) {
                // TODO: Look for the `CoerceUnsize` impl? Could it provide a lifetime rule?
                TODO(sp, "Propagate lifetimes through unsize (generic) - " << dst << " := " << src);
            }
            // If the type is a path, it must be a struct with an `CoerceUnsize` impl somewhere
            else if( dst.data().is_Path() ) {
                ASSERT_BUG(sp, src.data().is_Path(), "");
                ASSERT_BUG(sp, dst.data().as_Path().binding == src.data().as_Path().binding, "");
                ASSERT_BUG(sp, dst.data().as_Path().binding.is_Struct(), "");
                const ::HIR::Struct& str = *dst.data().as_Path().binding.as_Struct();
                const auto& dst_p = dst.data().as_Path().path.m_data.as_Generic();
                const auto& src_p = src.data().as_Path().path.m_data.as_Generic();
                auto param = str.m_struct_markings.coerce_param;
                switch( str.m_struct_markings.coerce_unsized )
                {
                case HIR::StructMarkings::Coerce::None:
                    BUG(sp, "");
                case HIR::StructMarkings::Coerce::Passthrough:
                    return coerce_unsize_types(sp, dst_p.m_params.m_types.at(param), src_p.m_params.m_types.at(param));
                case HIR::StructMarkings::Coerce::Pointer:
                    return unsize_types(sp, dst_p.m_params.m_types.at(param), src_p.m_params.m_types.at(param));
                }
            }
            else {
                TODO(sp, "Propagate lifetimes through unsize - " << dst << " := " << src);
            }
        }
        void visit(::HIR::ExprNode_Unsize& node) override {
            HIR::ExprVisitorDef::visit(node);
            this->equate_types(node.span(), node.m_res_type, node.m_dst_type);

            this->coerce_unsize_types(node.span(), node.m_dst_type, node.m_value->m_res_type);
        }
        void visit(::HIR::ExprNode_Index& node) override {
            HIR::ExprVisitorDef::visit(node);
            auto ty = HIR::TypeRef::new_path(HIR::Path(
                node.m_value->m_res_type.clone(),
                HIR::GenericPath(m_resolve.m_crate.get_lang_item_path(node.span(), "index"), { node.m_index->m_res_type.clone() }),
                RcString::new_interned("Output")
                ), {});
            m_resolve.expand_associated_types(node.span(), ty);
            this->equate_types(node.span(), node.m_res_type, ty);
        }
        void visit(::HIR::ExprNode_Deref& node) override {
            HIR::ExprVisitorDef::visit(node);
            TU_MATCH_HDRA( (node.m_value->m_res_type.data()), {)
            default: {
                auto ty = HIR::TypeRef::new_path(HIR::Path(
                    node.m_value->m_res_type.clone(),
                    HIR::GenericPath(m_resolve.m_crate.get_lang_item_path(node.span(), "deref"), {}),
                    RcString::new_interned("Target")
                    ), {});
                m_resolve.expand_associated_types(node.span(), ty);
                this->equate_types(node.span(), node.m_res_type, ty);
                }
            TU_ARMA(Pointer, te) {
                this->equate_types(node.span(), node.m_res_type, te.inner);
                }
            TU_ARMA(Borrow, te) {
                this->equate_types(node.span(), node.m_res_type, te.inner);
                }
            }
        }
        void visit(::HIR::ExprNode_Field& node) override {
            HIR::ExprVisitorDef::visit(node);
            TRACE_FUNCTION_FR("_Field " << node.m_field, "_Field " << node.m_field);
            const auto& sp = node.span();
            const auto& str_ty = node.m_value->m_res_type;

            bool is_index = ( '0' <= node.m_field.c_str()[0] && node.m_field.c_str()[0] <= '9' );
            unsigned index = is_index ? std::strtol(node.m_field.c_str(), nullptr, 10) : ~0u;
            // TODO: De-duplicate this logic (shared by this AND the check pass)

            if( const auto* te = str_ty.data().opt_Tuple() )
            {
                ASSERT_BUG(sp, is_index, "Non-index _Field on tuple");
                this->equate_types(sp, node.m_res_type, te->at(index));
            }
            else if( str_ty.data().is_NodeType() )
            {
                BUG(sp, "Magic type type being accessed too early");
            }
            else
            {
                ASSERT_BUG(sp, str_ty.data().is_Path(), "Value type of _Field isn't Path - " << str_ty);
                const auto& ty_e = str_ty.data().as_Path();
                const HIR::TypeRef* fld_ty_ptr = nullptr;
                if( const auto* strpp = ty_e.binding.opt_Struct() )
                {
                    const HIR::Struct& str = **strpp;
                    ASSERT_BUG(sp, is_index == str.m_data.is_Tuple(), "");
                    if( is_index ) {
                        const auto& flds = str.m_data.as_Tuple();
                        ASSERT_BUG(sp, index < flds.size(), "");
                        fld_ty_ptr = &flds[index].ent;
                    }
                    else {
                        const auto& flds = str.m_data.as_Named();
                        auto it = std::find_if(flds.begin(), flds.end(), [&](const HIR::StructField& f) { return f.name == node.m_field; });
                        ASSERT_BUG(sp, it != flds.end(), "");
                        fld_ty_ptr = &it->ty;
                    }
                }
                else if( const auto* unnpp = ty_e.binding.opt_Union() )
                {
                    const HIR::Union& unn = **unnpp;
                    const auto& flds = unn.m_variants;
                    auto it = std::find_if(flds.begin(), flds.end(), [&](const HIR::StructField& f) { return f.name == node.m_field; });
                    ASSERT_BUG(sp, it != flds.end(), "");
                    fld_ty_ptr = &it->ty;
                }
                else
                {
                    BUG(sp, "Value type of _Field isn't a Struct or Union - " << str_ty);
                }
                assert(fld_ty_ptr);

                auto ms = MonomorphStatePtr(&str_ty, &ty_e.path.m_data.as_Generic().m_params, nullptr);
                auto fld_ty = m_resolve.monomorph_expand(sp, *fld_ty_ptr, ms);
                DEBUG("Field type " << node.m_field << ": " << *fld_ty_ptr << " -> " << fld_ty);
                this->equate_types(sp, node.m_res_type, fld_ty);
            }
        }

        void visit(::HIR::ExprNode_CallPath& node) override {
            HIR::ExprVisitorDef::visit(node);
            DEBUG("CallPath:");

            // Equate arguments and returns (monomorphised)
            MonomorphState  ms;
            auto v = m_resolve.get_value(node.span(), node.m_path, ms, /*signature_only*/true);
            if( const auto* pe = node.m_path.m_data.opt_UfcsInherent() ) {
                this->equate_pps(node.span(), pe->impl_params, *ms.pp_impl);
            }
            const auto& fcn = *v.as_Function();

            for(size_t i = 0; i < fcn.m_args.size(); i ++)
            {
                auto arg_ty = m_resolve.monomorph_expand(node.m_args[i]->span(), fcn.m_args[i].second, ms);
                DEBUG("arg_ty " << i << " = " << arg_ty);
                this->equate_types(node.m_args[i]->span(), arg_ty, node.m_args[i]->m_res_type);
                this->equate_types(node.m_args[i]->span(), node.m_cache.m_arg_types[i], arg_ty);
            }
            auto ret_mono = m_resolve.monomorph_expand(node.span(), fcn.m_return, ms);
            DEBUG("ret_mono = " << ret_mono);
            this->equate_types(node.span(), node.m_cache.m_arg_types.back(), ret_mono);
            this->equate_types(node.span(), node.m_res_type, node.m_cache.m_arg_types.back());
        }
        void visit(::HIR::ExprNode_CallValue& node) override {
            HIR::ExprVisitorDef::visit(node);
            TRACE_FUNCTION_FR("_CallValue: m_value->m_res_type=" << node.m_value->m_res_type, "_CallValue");
            // TODO: Equate arguments and returns (after monomorphising away the HRLs)
            const auto& val_ty = node.m_value->m_res_type;
            if( const auto* tep = val_ty.data().opt_Function() ) {
                ::HIR::PathParams   hrl_params = tep->hrls.make_empty_params(true);
                this->visit_path_params(hrl_params);
                auto ms = MonomorphHrlsOnly(hrl_params);

                ASSERT_BUG(node.span(), tep->m_arg_types.size() <= node.m_args.size(), "");
                for(size_t i = 0; i < tep->m_arg_types.size(); i ++) {
                    this->equate_types(node.m_args[i]->span(), ms.monomorph_type(node.span(), tep->m_arg_types[i], false), node.m_args[i]->m_res_type);
                }
                this->equate_types(node.span(), node.m_res_type, ms.monomorph_type(node.span(), tep->m_rettype, false));
            }
            else if( const auto* tep_r = val_ty.data().opt_NamedFunction() ) {
                auto ft = tep_r->decay(node.span());
                for(auto& t : ft.m_arg_types) {
                    this->m_resolve.expand_associated_types(node.span(), t);
                }
                this->m_resolve.expand_associated_types(node.span(), ft.m_rettype);
                const auto* tep = &ft;
                ::HIR::PathParams   hrl_params = tep->hrls.make_empty_params(true);
                this->visit_path_params(hrl_params);
                auto ms = MonomorphHrlsOnly(hrl_params);

                ASSERT_BUG(node.span(), tep->m_arg_types.size() <= node.m_args.size(), "");
                for(size_t i = 0; i < tep->m_arg_types.size(); i ++) {
                    this->equate_types(node.m_args[i]->span(), ms.monomorph_type(node.span(), tep->m_arg_types[i], false), node.m_args[i]->m_res_type);
                }
                this->equate_types(node.span(), node.m_res_type, ms.monomorph_type(node.span(), tep->m_rettype, false));
            }
            else if( const auto* node_pp = TU_OPT1(val_ty.data(), NodeType, .opt_Closure()) ) {
                if( (*node_pp)->m_obj_path.m_path != HIR::SimplePath() ) {
                    TODO(node.span(), "Handle CallValue (expanded) - " << val_ty);
                }
                else {
                    ASSERT_BUG(node.span(), (*node_pp)->m_args.size() == node.m_args.size(), "");
                    for(size_t i = 0; i < node.m_args.size(); i ++) {
                        this->equate_types(node.m_args[i]->span(), (*node_pp)->m_args[i].second, node.m_args[i]->m_res_type);
                    }
                    this->equate_types(node.span(), node.m_res_type, (*node_pp)->m_return);
                }
            }
            else if( val_ty.data().is_Path() || val_ty.data().is_Generic() || val_ty.data().is_ErasedType() || val_ty.data().is_TraitObject() ) {
                // Look up the trait impl and check the generics on it
                // - If it's a bound, then handle the HRLs
                // - If an impl ref, just as normal.
                auto trait = m_resolve.m_crate.get_lang_item_path(node.span(), "fn_once");
                ::std::vector< ::HIR::TypeRef>  tup_ents;
                for(const auto& arg : node.m_args) {
                    tup_ents.push_back( arg->m_res_type.clone() );
                }
                ::HIR::PathParams   params;
                params.m_types.push_back( ::HIR::TypeRef( mv$(tup_ents) ) );

                bool found = m_resolve.find_impl(node.span(), trait, &params, val_ty, [&](ImplRef impl_ref, bool fuzzy)->bool{
                    ASSERT_BUG(node.span(), !fuzzy, "Fuzzy match in check pass");

                    HIR::PathParams*    hrls = nullptr;
                    TU_MATCH_HDRA((impl_ref.m_data), {)
                    TU_ARMA(TraitImpl, e) {}
                    TU_ARMA(Bounded, e) {
                        hrls = &e.hrls;
                        }
                    TU_ARMA(BoundedPtr, e) {
                        hrls = &e.hrls;
                        }
                    }
                    if(hrls) {
                        for(auto& lft : hrls->m_lifetimes) {
                            if( lft.binding == HIR::LifetimeRef::UNKNOWN || lft.binding == HIR::LifetimeRef::INFER) {
                                lft = m_state.allocate_ivar(node.span());
                            }
                        }
                    }
                    equate_pps(node.span(), impl_ref.get_trait_params(), params);
                    auto res_ty = impl_ref.get_type("Output", {});
                    m_resolve.expand_associated_types(node.span(), res_ty);
                    equate_types(node.span(), node.m_res_type, res_ty);

                    return true;
                    });
                if( !found ) {
                    ERROR(node.span(), E0000, "Unable to find a matching impl of " << trait << " for " << val_ty);
                }
            }
            else {
                TODO(node.span(), "Handle CallValue - " << val_ty);
            }
        }
        void visit(::HIR::ExprNode_CallMethod& node) override {
            DEBUG("_CallMethod: " << node.m_method_path);
            HIR::ExprVisitorDef::visit(node);

            TRACE_FUNCTION_FR("_CallMethod: " << node.m_method_path, "_CallMethod");

            // Equate arguments and returns (monomorphised)
            MonomorphState  ms;
            auto v = m_resolve.get_value(node.span(), node.m_method_path, ms, /*signature_only*/true);
            if( const auto* pe = node.m_method_path.m_data.opt_UfcsInherent() ) {
                this->equate_pps(node.span(), pe->impl_params, *ms.pp_impl);
            }
            HIR::PathParams pp_hrls;
            if( const auto* pe = node.m_method_path.m_data.opt_UfcsKnown() ) {
                if( const auto* te = pe->type.data().opt_TraitObject() ) {
                    if(te->m_trait.m_hrtbs) {
                        pp_hrls = te->m_trait.m_hrtbs->make_empty_params(true);
                        this->visit_path_params(pp_hrls);
                    }
                }
                if( pe->hrtbs ) {
                    TODO(node.span(), "Handle HRTBs on trait");
                }
            }
            MonomorphHrlsOnly   ms2(pp_hrls);
            const auto& fcn = *v.as_Function();

            for(size_t i = 0; i < fcn.m_args.size(); i ++)
            {
                const auto& n = (i == 0 ? node.m_value : node.m_args[i-1]);
                auto arg_ty = m_resolve.monomorph_expand(n->span(), fcn.m_args[i].second, ms);
                arg_ty = ms2.monomorph_type(n->span(), arg_ty);
                DEBUG("ARG " << arg_ty);
                this->equate_types(n->span(), arg_ty, n->m_res_type);
                //this->equate_types(n->span(), node.m_cache.m_arg_types[i], arg_ty);
                node.m_cache.m_arg_types[i] = std::move(arg_ty);
            }
            DEBUG("Raw Ret " << fcn.m_return);
            auto ret_mono = m_resolve.monomorph_expand(node.span(), fcn.m_return, ms);
            ret_mono = ms2.monomorph_type(node.span(), ret_mono);
            DEBUG("RET " << ret_mono);
            this->equate_types(node.span(), node.m_res_type, ret_mono);
            //this->equate_types(node.span(), node.m_cache.m_arg_types.back(), ret_mono);
            node.m_cache.m_arg_types.back() = std::move(ret_mono);
        }

        void visit(::HIR::ExprNode_Literal& node) override {
            if( node.m_data.is_String() || node.m_data.is_ByteString() ) {
                const auto& be = node.m_res_type.data().as_Borrow();
                this->equate_lifetimes(node.span(), be.lifetime, ::HIR::LifetimeRef::new_static());
            }
        }
        void visit(::HIR::ExprNode_UnitVariant& node) override {
            HIR::ExprVisitorDef::visit(node);
            
            // Just assign the type through (for the path params)
            this->equate_pps(node.span(),
                node.m_res_type.data().as_Path().path.m_data.as_Generic().m_params,
                node.m_path.m_params
                );
        }
        void visit(::HIR::ExprNode_PathValue& node) override {
            const Span& sp = node.span();
            HIR::ExprVisitorDef::visit(node);
            MonomorphState  ms;
            auto v = m_resolve.get_value(sp, node.m_path, ms, /*signature_only*/true);
            if( const auto* pe = node.m_path.m_data.opt_UfcsInherent() ) {
                ms.pp_impl = &pe->impl_params;
            }

            HIR::TypeRef    ty;
            TU_MATCH_HDRA( (v), { )
            default:
                TODO(node.span(), "PathValue - " << node.m_path << " - " << v.tag_str());
            TU_ARMA(EnumConstructor, ve) {
                ty = ::HIR::TypeRef(::HIR::TypeData::make_NamedFunction({
                    node.m_path.clone(),
                    ::HIR::TypeData_NamedFunction_Ty::make_EnumConstructor({ ve.e, ve.v })
                    }));
                }
            TU_ARMA(StructConstructor, ve) {
                ty = ::HIR::TypeRef(::HIR::TypeData::make_NamedFunction({ node.m_path.clone(), ve.s }));
                }
            TU_ARMA(Constant, ve) {
                ty = m_resolve.monomorph_expand(sp, ve->m_type, ms);
                }
            TU_ARMA(Static, ve) {
                ty = m_resolve.monomorph_expand(sp, ve->m_type, ms);
                }
            TU_ARMA(Function, ve) {
                ty = ::HIR::TypeRef(::HIR::TypeData::make_NamedFunction({ node.m_path.clone(), ve }));
                }
            }
            this->equate_types(sp, node.m_res_type, ty);
        }
        void visit(::HIR::ExprNode_Variable& node) override {
            HIR::ExprVisitorDef::visit(node);
            this->equate_types(node.span(), node.m_res_type, get_local_var_ty(node.span(), node.m_slot));
        }
        void visit(::HIR::ExprNode_ConstParam& node) override {
            HIR::ExprVisitorDef::visit(node);
            // Nothing to do?
        }

        void visit(::HIR::ExprNode_Emplace& node) override {
            HIR::ExprVisitorDef::visit(node);
            if(TARGETVER_MOST_1_19)
            {
                switch( node.m_type )
                {
                case ::HIR::ExprNode_Emplace::Type::Noop:
                    assert( !node.m_place );
                    this->equate_types(node.span(), node.m_res_type, node.m_value->m_res_type);
                    break;
                case ::HIR::ExprNode_Emplace::Type::Boxer: {
                    // NOTE: `m_place` is `()` - so just ignore it.
                    //assert( !node.m_place );
                    const auto& data_ty = node.m_value->m_res_type;
                    const auto& box_ty = node.m_res_type;

                    // TODO: Full trait magic?
                    // - `<box_ty as Boxed>::finalize( < <box_ty as Boxer>::Place as BoxPlace<data_ty> >::make_place() )`

                    const auto& box_path = box_ty.data().as_Path().path.m_data.as_Generic();
                    this->equate_types(node.span(), box_path.m_params.m_types.at(0), data_ty);
                    } break;
                case ::HIR::ExprNode_Emplace::Type::Placer: {
                    const auto& data_ty = node.m_value->m_res_type;
                    const auto& placer_ty = node.m_place->m_res_type;
                    // Where P = `placer_ty` and D = `data_ty`
                    // Result type is <<P as Placer<D>>::Place as InPlace<D>>::Owner
                    const auto& lang_Placer = m_resolve.m_crate.get_lang_item_path(node.span(), "placer_trait");
                    const auto& lang_InPlace = m_resolve.m_crate.get_lang_item_path(node.span(), "in_place_trait");
                    // - 
                    auto place_ty = ::HIR::TypeRef::new_path( ::HIR::Path(placer_ty.clone(), ::HIR::GenericPath(lang_Placer, ::HIR::PathParams(data_ty.clone())), "Place"), {} );
                    auto owner_ty = ::HIR::TypeRef::new_path( ::HIR::Path(std::move(place_ty), ::HIR::GenericPath(lang_InPlace, ::HIR::PathParams(data_ty.clone())), "Owner"), {} );
                    m_resolve.expand_associated_types(node.span(), owner_ty);
                    this->equate_types(node.span(), node.m_res_type, owner_ty);
                    } break;
                }
            }
            else
            {
                assert( node.m_type == ::HIR::ExprNode_Emplace::Type::Boxer );
                const auto& data_ty = node.m_value->m_res_type;
                const auto& box_ty = node.m_res_type;
                const auto& box_path = box_ty.data().as_Path().path.m_data.as_Generic();
                this->equate_types(node.span(), box_path.m_params.m_types.at(0), data_ty);
            }
        }
        void visit(::HIR::ExprNode_TupleVariant& node) override {
            const Span& sp = node.span();
            HIR::ExprVisitorDef::visit(node);
            
            // Just assign the type through (for the path params)
            this->equate_pps(node.span(),
                node.m_res_type.data().as_Path().path.m_data.as_Generic().m_params,
                node.m_path.m_params
                );

            const auto& ty = node.m_res_type;

            const ::HIR::t_tuple_fields* fields_ptr = nullptr;
            ASSERT_BUG(sp, ty.data().is_Path(), "Result type of _TupleVariant isn't Path");
            TU_MATCH_HDRA( (ty.data().as_Path().binding), {)
            TU_ARMA(Unbound, e) {
                BUG(sp, "Unbound type in _TupleVariant - " << ty);
                }
            TU_ARMA(Opaque, e) {
                BUG(sp, "Opaque type binding in _TupleVariant - " << ty);
                }
            TU_ARMA(Enum, e) {
                const auto& var_name = node.m_path.m_path.components().back();
                const auto& enm = *e;
                size_t idx = enm.find_variant(var_name);
                const auto& var_ty = enm.m_data.as_Data()[idx].type;
                const auto& str = *var_ty.data().as_Path().binding.as_Struct();
                ASSERT_BUG(sp, str.m_data.is_Tuple(), "Pointed variant of TupleVariant (" << node.m_path << ") isn't a Tuple");
                fields_ptr = &str.m_data.as_Tuple();
                }
            TU_ARMA(Union, e) {
                BUG(sp, "Union in TupleVariant");
                }
            TU_ARMA(ExternType, e) {
                BUG(sp, "ExternType in TupleVariant");
                }
            TU_ARMA(Struct, e) {
                ASSERT_BUG(sp, e->m_data.is_Tuple(), "Pointed struct in TupleVariant (" << node.m_path << ") isn't a Tuple");
                fields_ptr = &e->m_data.as_Tuple();
                }
            }
            assert(fields_ptr);
            const ::HIR::t_tuple_fields& fields = *fields_ptr;
            ASSERT_BUG(sp, fields.size() == node.m_args.size(), "");
            auto ms = MonomorphStatePtr(&ty, &ty.data().as_Path().path.m_data.as_Generic().m_params, nullptr);

            // Bind fields with type params (coercable)
            for( unsigned int i = 0; i < node.m_args.size(); i ++ )
            {
                const auto& des_ty_r = fields[i].ent;
                auto des_ty = m_resolve.monomorph_expand(node.m_args[i]->span(), des_ty_r, ms);
                this->equate_types(node.m_args[i]->span(), des_ty, node.m_args[i]->m_res_type);
            }
        }
        void visit(::HIR::ExprNode_StructLiteral& node) override {
            HIR::ExprVisitorDef::visit(node);
            
            // Just assign the type through (for the path params)
            this->equate_pps(node.span(),
                node.m_res_type.data().as_Path().path.m_data.as_Generic().m_params,
                node.m_real_path.m_params
                );
            
            const Span& sp = node.span();
            const auto& ty_path = node.m_real_path;
            const auto& ty = node.m_res_type;
            ASSERT_BUG(sp, ty.data().is_Path(), "Result type of _StructLiteral isn't Path");

            const ::HIR::t_struct_fields* fields_ptr = nullptr;
            TU_MATCH_HDRA( (ty.data().as_Path().binding), {)
            TU_ARMA(Unbound, e) {}
            TU_ARMA(Opaque, e) {}
            TU_ARMA(Enum, e) {
                const auto& var_name = ty_path.m_path.components().back();
                const auto& enm = *e;
                auto idx = enm.find_variant(var_name);
                ASSERT_BUG(sp, idx != SIZE_MAX, "");
                ASSERT_BUG(sp, enm.m_data.is_Data(), "");
                const auto& var = enm.m_data.as_Data()[idx];

                const auto& str = *var.type.data().as_Path().binding.as_Struct();
                ASSERT_BUG(sp, var.is_struct, "Struct literal for enum on non-struct variant");
                fields_ptr = &str.m_data.as_Named();

                ASSERT_BUG(sp, !node.m_base_value, "Struct literal for enum with a base value");
                }
            TU_ARMA(Union, e) {
                fields_ptr = &e->m_variants;
                ASSERT_BUG(sp, node.m_values.size() > 0, "Union with no values");
                ASSERT_BUG(sp, node.m_values.size() == 1, "Union with multiple values");
                ASSERT_BUG(sp, !node.m_base_value, "Union can't have a base value");
                }
            TU_ARMA(ExternType, e) {
                BUG(sp, "ExternType in StructLiteral");
                }
            TU_ARMA(Struct, e) {
                if( e->m_data.is_Unit() )
                {
                    ASSERT_BUG(sp, node.m_values.size() == 0, "Values provided for unit-like struct");
                    ASSERT_BUG(sp, ! node.m_base_value, "Values provided for unit-like struct");
                    return ;
                }

                ASSERT_BUG(sp, e->m_data.is_Named(), "StructLiteral not pointing to a braced struct, instead " << e->m_data.tag_str() << " - " << ty);
                fields_ptr = &e->m_data.as_Named();
                }
            }
            ASSERT_BUG(node.span(), fields_ptr, "Didn't get field for path in _StructLiteral - " << ty);
            const ::HIR::t_struct_fields& fields = *fields_ptr;
            for(const auto& fld : fields) {
                DEBUG(fld.name << ": " << fld.ty);
            }
            auto ms = MonomorphStatePtr(&ty, &ty_path.m_params, nullptr);

            if( node.m_base_value ) {
                #if 0
                this->equate_types( node.m_base_value->span(), node.m_res_type, node.m_base_value->m_res_type );
                #else
                const auto& ty_base = node.m_base_value->m_res_type;
                const auto& ty_path_base = ty_base.data().as_Path().path.m_data.as_Generic();
                auto ms_base = MonomorphStatePtr(&ty_base, &ty_path_base.m_params, nullptr);
                for(const auto& fld : fields) {
                    auto it = ::std::find_if(node.m_values.begin(), node.m_values.end(), [&](const auto& v)->bool{ return v.first == fld.name; });
                    if( it == node.m_values.end() ) {
                        auto dst_ty = m_resolve.monomorph_expand(node.span(), fld.ty, ms);
                        auto src_ty = m_resolve.monomorph_expand(node.span(), fld.ty, ms_base);
                        DEBUG("BASE ." << fld.name << ": " << dst_ty << " = " << src_ty);
                        this->equate_types( node.m_base_value->span(), dst_ty, src_ty );
                    }
                }
                #endif
            }

            // Bind fields with type params (coercable)
            for(auto& val : node.m_values)
            {
                const auto& name = val.first;
                auto it = ::std::find_if(fields.begin(), fields.end(), [&](const HIR::StructField& v)->bool{ return v.name == name; });
                assert(it != fields.end());
                const auto& des_ty_r = it->ty;

                DEBUG(name << " : " << des_ty_r);
                auto des_ty = m_resolve.monomorph_expand(val.second->span(), des_ty_r, ms);
                DEBUG("." << name << " : " << des_ty);
                this->equate_types(val.second->span(), des_ty, val.second->m_res_type);
            }
        }
        void visit(::HIR::ExprNode_Tuple& node) override {
            HIR::ExprVisitorDef::visit(node);
            // Destrucure output type
            auto& te = node.m_res_type.data().as_Tuple();
            for(size_t i = 0; i < te.size(); i ++)
            {
                auto& vnode = *node.m_vals[i];
                this->equate_types(vnode.span(), te[i], vnode.m_res_type);
            }
        }
        void visit(::HIR::ExprNode_ArrayList& node) override {
            HIR::ExprVisitorDef::visit(node);
            const auto& inner = node.m_res_type.data().as_Array().inner;
            for(auto& v : node.m_vals) {
                this->equate_types(v->span(), inner, v->m_res_type);
            }
        }
        void visit(::HIR::ExprNode_ArraySized& node) override {
            HIR::ExprVisitorDef::visit(node);
            const auto& inner = node.m_res_type.data().as_Array().inner;
            this->equate_types(node.m_val->span(), inner, node.m_val->m_res_type);
        }

        void visit(::HIR::ExprNode_Closure& node) override {
            const Span& sp = node.span();
            m_returns.push_back(&node.m_return);
            DEBUG("Closure " << &node);
            HIR::ExprVisitorDef::visit(node);
            if( !node.m_code ) {
            }
            else if( !node.m_is_move ) {
                if( node.m_capture_lifetime == HIR::LifetimeRef() || this->remove_locals ) {
                    node.m_capture_lifetime = m_state.allocate_ivar(sp);
                    DEBUG("m_capture_lifetime = " << node.m_capture_lifetime);
                }

                auto l = m_state.allocate_local(node, /**/node);
                for( const auto& cap : node.m_avu_cache.captured_vars ) {
                    if( cap.usage != ::HIR::ValueUsage::Move ) {
                        HIR::TypeRef    tmp_ty;
                        const auto* cap_ty_p = &m_binding_types_ptr->at(cap.root_slot);
                        HIR::TypeRef    deref_source;
                        for(const auto& n : cap.fields) {
                            if( n == RcString() ) {
                                deref_source = cap_ty_p->clone();
                            }
                            tmp_ty = m_resolve.get_field_type(sp, *cap_ty_p, n);
                            cap_ty_p = &tmp_ty;
                        }

                        if( deref_source.data().is_Infer() ) {
                            auto wrapped_ty = HIR::TypeRef::new_borrow(
                                cap.usage == HIR::ValueUsage::Mutate ? ::HIR::BorrowType::Unique : ::HIR::BorrowType::Shared,
                                cap_ty_p->clone(), l
                                );
                            DEBUG("Capture " << cap << " -> " << wrapped_ty);
                            equate_type_lifetimes(node.span(), node.m_capture_lifetime, wrapped_ty);
                        }
                        else {
                            DEBUG("Capture " << cap << " from" << deref_source);
                            equate_type_lifetimes(node.span(), node.m_capture_lifetime, deref_source);
                        }
                    }
                }
            }
            else {
                node.m_capture_lifetime = ::HIR::LifetimeRef::new_static();
            }
            m_returns.pop_back();
        }
        void visit(::HIR::ExprNode_Generator& node) override {
            const Span& sp = node.span();

            m_returns.push_back(&node.m_return);
            m_yields.push_back(&node.m_yield_ty);

            if( node.m_is_move ) {
                node.m_capture_lifetime = ::HIR::LifetimeRef::new_static();
            }
            else {
                if( node.m_capture_lifetime == HIR::LifetimeRef() || this->remove_locals ) {
                    node.m_capture_lifetime = m_state.allocate_ivar(sp);
                    DEBUG("m_capture_lifetime = " << node.m_capture_lifetime);
                }
            }
            
            // TODO: Yield?
            HIR::ExprVisitorDef::visit(node);
            m_yields.pop_back();
            m_returns.pop_back();
        }
        void visit(::HIR::ExprNode_GeneratorWrapper& node) override {
            BUG(node.span(), "Encountered ExprNode_GeneratorWrapper too early");
        }
        void visit(::HIR::ExprNode_AsyncBlock& node) override {
            const Span& sp = node.span();

            if( node.m_is_move ) {
                node.m_capture_lifetime = ::HIR::LifetimeRef::new_static();
            }
            else {
                if( node.m_capture_lifetime == HIR::LifetimeRef() || this->remove_locals ) {
                    node.m_capture_lifetime = m_state.allocate_ivar(sp);
                    DEBUG("m_capture_lifetime = " << node.m_capture_lifetime);
                }
            }

            m_returns.push_back(&node.m_code->m_res_type);
            HIR::ExprVisitorDef::visit(node);
            m_returns.pop_back();
        }
    };

    /// <summary>
    /// Visitor to compact (de-duplicate) types
    /// </summary>
    struct ExprVisitor_CompactTypes: HIR::ExprVisitorDef
    {
        std::map<std::string, HIR::TypeRef>  types;

        void visit_root(HIR::ExprPtr& ep)
        {
            ep->visit(*this);
            for(auto& b : ep.m_bindings) {
                visit_type(b);
            }
            for(auto& b : ep.m_erased_types) {
                visit_type(b);
            }
        }
        void visit_type(HIR::TypeRef& ty) override
        {
            HIR::ExprVisitorDef::visit_type(ty);

            // Use string comparison to ensure that lifetimes are checked
            auto s = FMT(ty);
            if(s[0] == '{') {
                auto p = s.find('}');
                s = s.substr(p+1);
            }

            auto it = types.find(s);
            if( it != types.end() ) {
                ty = HIR::TypeRef(it->second);
            }
            else {
                types.insert(std::make_pair(s, ty));
            }
        };
    };


    void HIR_Expand_LifetimeInfer_ExprInner(LocalTraitResolve& resolve, const ::HIR::Function::args_t& args, const HIR::TypeRef& ret_ty, HIR::ExprPtr& ep,
        bool remove_locals, bool is_const_context)
    {
        LifetimeInferState  state { resolve, is_const_context };

        // Before running algorithm, dump the HIR (just as a reference for debugging)
        DEBUG("\n" << FMT_CB(os, HIR_DumpExpr(os, ep)));

        // Build up a simplified list of in-scope liftime rules (bounds)
        {
            TRACE_FUNCTION_FR("Enumerating bounds", "Enumerating bounds");
            // - Add all bounds from `GenericParams`
            auto push_bounds = [&state](const HIR::GenericParams& params) {
                for(const auto& b : params.m_bounds) {
                    if( const auto* be = b.opt_TypeLifetime() ) {
                        state.add_type_lifetime_bound(be->type, be->valid_for);
                    }
                    else if( const auto* be = b.opt_Lifetime() ) {
                        state.add_lifetime_bound(be->test, be->valid_for);
                    }
                    else {
                    }
                }
            };
            push_bounds(resolve.impl_generics());
            push_bounds(resolve.item_generics());
            // Visit types and add implicit bounds
            // -See rustc-1.29.0/src/vendor/regex-syntax/src/error.rs:167: `fn from_formatter<'e, E: fmt::Display>( \n fmter: &'p Formatter<'e, E>, `
            struct TypeVisitor: public HIR::Visitor {
                LifetimeInferState&     m_state;
                std::vector<const HIR::LifetimeRef*>  m_stack;
                TypeVisitor(LifetimeInferState& state): m_state(state) {}

                void maybe_push_lifetime_bound(const HIR::LifetimeRef& test, const HIR::LifetimeRef& valid_for) {
                    // Only check parameters
                    if( !test.is_param() ) {
                        return;
                    }
                    // but not HRLs
                    if( test.is_hrl() ) {
                        return ;
                    }
                    // if the bound is a param, it can't be a HRL.
                    if( valid_for.is_hrl() ) {
                        return ;
                    }

                    auto b = LifetimeInferState::LifetimeBound(test, valid_for);
                    DEBUG("> Infer liftime bound: " << b);
                    if( std::find(m_state.m_bounds.begin(), m_state.m_bounds.end(), b) == m_state.m_bounds.end() ) {
                         m_state.m_bounds.push_back(b);
                    }
                }
                /// Add a bound for this lifetime (requiring that it outlives the top of the stack)
                void check_lifetime_variant(const HIR::LifetimeRef& lft) {
                    if( !m_stack.empty() && m_stack.back() ) {
                        auto& valid_for = *m_stack.back();
                        maybe_push_lifetime_bound(lft, valid_for);
                    }
                }

                void visit_type(HIR::TypeRef& ty) override {
                    auto ssize = m_stack.size();
                    if(const auto* te = ty.data().opt_Borrow()) {
                        check_lifetime_variant(te->lifetime);
                        m_stack.push_back(&te->lifetime);
                    }
                    if(const auto* te = ty.data().opt_TraitObject()) {
                        check_lifetime_variant(te->m_lifetime);
                    }
                    if( /*const auto* te =*/ ty.data().opt_Path() ) {
                        if( !m_stack.empty() && m_stack.back() ) {
                            m_state.add_type_lifetime_bound(ty, *m_stack.back());
                        }
                    }
                    if(ty.data().is_Generic()) {
                        // TODO: Add a type bound?
                    }
                    if( ty.data().is_Function() ) {
                        m_stack.push_back(nullptr);
                    }
                    HIR::Visitor::visit_type(ty);
                    m_stack.erase(m_stack.begin() + ssize, m_stack.end());
                }
            } tv(state);
            for(const auto& a : args) {
                DEBUG("Visit ARG " << a.second);
                tv.visit_type(const_cast<HIR::TypeRef&>(a.second));
            }

            // Build up chains
            // - Iterate bounds, see if there's a bound for the `valid_for` of each. If there is, make another for the chain.
            for(bool added = true; added; )
            {
                added = false;
                for(size_t i = 0; i < state.m_bounds.size(); i ++) {
                     for(size_t j = 0; j < state.m_bounds.size(); j ++) {
                         // `'a: 'b` and `'b: 'c` then add `'a: 'c`
                         if( !state.m_bounds[i].ty && state.m_bounds[i].valid_for == state.m_bounds[j].lft ) {
                             auto b = state.m_bounds[i];
                             b.valid_for = state.m_bounds[j].valid_for;

                             if( std::find(state.m_bounds.begin(), state.m_bounds.end(), b) == state.m_bounds.end() ) {
                                 DEBUG("Add chained bound: " << b << " (from " << state.m_bounds[i] << " and " << state.m_bounds[j] << ")");
                                 state.m_bounds.push_back(b);
                                 added = true;
                                 break;
                             }
                         }
                     }
                     if(added) {
                         break;
                     }
                }
            }

            DEBUG(state.m_bounds.size() << " lifetime bounds");
            for(const auto& b : state.m_bounds) {
                DEBUG("BOUND - " << b);
            }
        }

        // Enumerate lifetimes and relationships
        {
            TRACE_FUNCTION_FR("Enumerating lifetimes", "Enumerating lifetimes");
            // TODO: Also do lifetime equality of the return type and the ATY version
            ExprVisitor_Enumerate   ev(resolve, args, ret_ty, remove_locals, state);
            resolve.m_lifetime_state = &ev;
            ev.visit_root(ep);
            resolve.m_lifetime_state = nullptr;
        }

        // Shortcut for when there's nothing to infer
        if( state.m_ivars.size() == 0 ) {
            ExprVisitor_CompactTypes().visit_root(ep);
            return ;
        }

        // Before running algorithm, dump the HIR (just as a reference for debugging)
        DEBUG("\n" << FMT_CB(os, HIR_DumpExpr(os, ep)));

        // If there were inferred liftimes present

        for(size_t cur_iter = 0, remaining_iters = 1000; remaining_iters--; cur_iter ++ )
        {
            TRACE_FUNCTION_FR("=== Iter " << cur_iter << " ===", "=== Iter " << cur_iter << " ===");
            state.dump();

            bool change = false;
            auto set = [&](LifetimeInferState::IvarLifetime& iv, const HIR::LifetimeRef& lft, const char* log_reason) {
                assert(!iv.is_known());
                auto real_lft = state.get_final_lft(iv.sp, lft);
                DEBUG(state.get_lft_for_iv(iv) << " := " << real_lft << "[" << lft << "] (" << log_reason << ")");
                ASSERT_BUG(iv.sp, real_lft != state.get_lft_for_iv(iv), real_lft);
                iv.known = real_lft;
                change = true;
                };


            // Run through the ivar list, looking for ones with only one source/destination
            for(auto& iv : state.m_ivars) {
                if( !iv.is_known() ) {
                    // Zero sources - set to `'static`
                    if( iv.sources.size() == 0 ) {
                        set(iv, HIR::LifetimeRef::new_static(), "No source");
                        continue ;
                    }

                    if( iv.sources.size() == 1 ) {
                        auto lft = state.get_final_lft(iv.sp, iv.sources[0]);
                        //auto lft = iv.sources[0];
                        if( !state.opt_ivar(iv.sp, lft) ) {
                            set(iv, lft, "Only source");
                            continue ;
                        }
                    }
                    if( iv.destinations.size() == 1 ) {
                        // Don't let ivars propagate upwards
                        auto lft = state.get_final_lft(iv.sp, iv.destinations[0]);
                        if( !state.opt_ivar(iv.sp, lft) ) {
                            set(iv, lft, "Only destination");
                            continue ;
                        }
                    }

                    // If all sources are known, then unify
                    if( std::all_of(iv.sources.begin(), iv.sources.end(), [&](const HIR::LifetimeRef& lft) { return state.opt_ivar(iv.sp, lft) == nullptr; }) )
                    {
                        // De-duplicate
                        std::vector<HIR::LifetimeRef>   dedup_sources;
                        for(const auto& s : iv.sources) {
                            auto lft = state.get_final_lft(iv.sp, s);
                            if( std::find(dedup_sources.begin(), dedup_sources.end(), lft) == dedup_sources.end() )
                                dedup_sources.push_back(lft);
                        }

                        // Remove lifetimes that are source-compatible to another option
                        // - E.g. `'static` with anything else can be safely removed.
                        // - Or if `'a: 'b` and both are in the list then `'a` can be removed (it's longer than `'b`)
                        for(bool any_removed = true; any_removed; )
                        {
                            any_removed = false;
                            for(auto it = dedup_sources.begin(); it != dedup_sources.end(); ) {
                                bool remove = false;
                                for(const auto& lft : dedup_sources) {
                                    if( *it == lft ) {
                                    }
                                    else {
                                        std::vector<HIR::LifetimeRef>   _unused_fails;
                                        if( state.check_lifetimes(Span(), lft, *it, _unused_fails) ) {
                                            // `it2` is shorter than `it`, so `it` isn't needed
                                            remove = true;
                                        }
                                    }
                                }
                                if( remove ) {
                                    it = dedup_sources.erase(it);
                                    any_removed = true;
                                }
                                else {
                                    ++ it;
                                }
                            }
                        }

                        if( dedup_sources.size() == 1 ) {
                            set(iv, dedup_sources[0], "Only source (after dedup)");
                        }
                        else {
                            // Create a composite local lifetime
                            auto lft = state.allocate_local(iv.sp, std::move(dedup_sources));
                            set(iv, lft, "Composite");
                        }
                        continue ;
                    }
                }
            }

            // If all ivars are known, break
            if( std::all_of(state.m_ivars.begin(), state.m_ivars.end(), [](const LifetimeInferState::IvarLifetime& iv){ return iv.is_known(); }) ) {
                break;
            }
            // If there was no change, bail
            if( !change || remaining_iters == 0 ) {
                state.dump();
                BUG(ep->span(), "Lifetime inferrence stalled");
            }
            // Compact the lifetime list (resolve all known lifetimes to their actual value)
            for(auto& iv : state.m_ivars) {
                for(auto& l : iv.sources)
                    l = state.get_final_lft(iv.sp, l);
                for(auto& l : iv.destinations)
                    l = state.get_final_lft(iv.sp, l);
            }
        }

        {
            TRACE_FUNCTION_FR("COMPACT", "COMPACT");
            state.dump();

            // Compact (replace any mentioned ivars by the their known lifetime)
            for(auto& iv : state.m_ivars)
            {
                ASSERT_BUG(iv.sp, iv.is_known(), "Unresolved lifetime?");
                iv.known = state.get_final_lft(iv.sp, iv.known);
                ASSERT_BUG(iv.sp, !state.opt_ivar(iv.sp, iv.known), "Lifetime resolved to an ivar - " << state.get_lft_for_iv(iv) << " = " << iv.known);
                for(auto& l : iv.destinations)
                    l = state.get_final_lft(iv.sp, l);
            }
        }

        // Validate (check that everything roughly satisfies lifetime requirements)
        {
            TRACE_FUNCTION_FR("VALIDATE", "VALIDATE");
            for(auto& iv : state.m_ivars)
            {
                DEBUG("Validate IVar: " << state.get_lft_for_iv(iv));
                for(const auto& d : iv.destinations)
                    state.ensure_outlives(iv.sp, d, iv.known);
            }

            DEBUG("Validate return: " << ret_ty);
            clone_ty_with(ep->m_span, ret_ty, [&](const HIR::TypeRef& tpl, HIR::TypeRef& rv)->bool {
                if( tpl.data().is_ErasedType() && tpl.data().as_ErasedType().m_inner.is_Fcn() )
                {
                    const auto* e = &tpl.data().as_ErasedType().m_inner.as_Fcn();
                    ASSERT_BUG(ep->m_span, e->m_index < ep.m_erased_types.size(),
                        "Erased type index OOB - " << e->m_origin << " " << e->m_index << " >= " << ep.m_erased_types.size());

                    struct V: public HIR::Visitor {
                        LifetimeInferState& state;
                        const Span& sp;
                        const ::std::vector<HIR::TypeRef>& bindings;
                        const ThinVector<HIR::LifetimeRef>& dst_lfts;
                        std::vector<HIR::PathParams>    m_hrls;

                        V( LifetimeInferState& state, const Span& sp, const ::std::vector<HIR::TypeRef>& bindings, const ThinVector<HIR::LifetimeRef>& dst_lfts)
                            : state(state)
                            , sp(sp)
                            , bindings(bindings)
                            , dst_lfts(dst_lfts)
                        {
                        }

                        void equate_lifetime(const HIR::LifetimeRef& src) {
                            const auto& src_real = state.get_final_lft(sp, src);
                            if( src_real.is_hrl() ) {
                                ASSERT_BUG(sp, m_hrls.size() > 0, "Encountered HRL with no HRL in the stack");
                                //parent.equate_lifetimes(sp, dst_lft, m_hrls.back().m_lifetimes[src.binding & 0xFF]);
                            }
                            else if( src_real.binding == HIR::LifetimeRef::STATIC ) {
                                // Ignore `'static`, it's always valid
                            }
                            else {
                                for(const auto& dst_lft : dst_lfts) {
                                    std::vector<HIR::LifetimeRef>   fails;
                                    if( state.check_lifetimes(sp, dst_lft, src_real, fails) )
                                        return ;
                                    DEBUG(dst_lft << " = " << src_real << " (" << src << ") failed: " << fails);
                                }
                                ERROR(sp, E0000, "None of erased lifetime lifetime bounds [" << dst_lfts << "]: " << src_real << " (" << src << ") passed");
                            }
                        }

                        void visit_capture(
                            const ::HIR::LifetimeRef& capture_lifetime, HIR::ValueUsage usage,
                            unsigned root_slot, const std::vector<RcString>& fields
                        )
                        {
                            ::HIR::TypeRef  tmp_ty;

                            ::HIR::LifetimeRef  borrow_source = capture_lifetime;
                            const auto* cap_ty_p = &this->bindings[root_slot];
                            for(const auto& n : fields) {
                                if( const auto* p = cap_ty_p->data().opt_Borrow() ) {
                                    borrow_source = p->lifetime;
                                }
                                tmp_ty = state.m_resolve.get_field_type(sp, *cap_ty_p, n);
                                state.m_resolve.expand_associated_types(sp, tmp_ty);
                                cap_ty_p = &tmp_ty;
                            }

                            DEBUG("Capture: #" << root_slot << "{" << fields << "}: " << usage << " " << capture_lifetime << " " << *cap_ty_p);
                            if( usage != HIR::ValueUsage::Move ) {
                                equate_lifetime(borrow_source);
                            }
                            else {
                                visit_type(const_cast<HIR::TypeRef&>(*cap_ty_p));
                            }
                        }

                        void visit_type(HIR::TypeRef& t) override {
                            if(t.data().is_Borrow()) {
                                DEBUG("Borrow " << t);
                                equate_lifetime(t.data().as_Borrow().lifetime);
                            }

                            bool hrl_pushed = false;
                            auto push_hrls = [&](const HIR::GenericParams& hrls_def) {
                                hrl_pushed = true;
                                m_hrls.push_back(hrls_def.make_empty_params(true));
                            };

                            if(is_opaque(t)) {
                                // Iterate type lifetime bounds
                                state.iterate_type_lifetime_bounds(t, [&](const HIR::LifetimeRef& lft)->bool {
                                    DEBUG("Generic opaque bound: " << t << ": " << lft);
                                    equate_lifetime(lft);
                                    return false;
                                    });
                                if( t.data().is_Generic() || t.data().is_Path() ) {
                                    // TODO: If the above didn't return anything, then assign a "only this function" liftime
                                }
                                else if( const auto* e = t.data().opt_ErasedType() ) {
                                    for(const auto& lft : e->m_lifetime_bounds) {
                                        DEBUG("Erased " << t);
                                        equate_lifetime(lft);
                                    }
                                    for(size_t i = 1; i < e->m_traits.size(); i ++) {
                                        assert(!e->m_traits[i].m_hrtbs);
                                    }
                                    if( e->m_traits[0].m_hrtbs ) {
                                        push_hrls(*e->m_traits[0].m_hrtbs);
                                    }
                                }
                                else {
                                    TODO(sp, "Get lifetime (from bounds) for opaque type - " << t);
                                }
                            }

                            if( const auto* e = t.data().opt_TraitObject() ) {
                                DEBUG("Trait object " << t);
                                // Get the lifetime
                                equate_lifetime(e->m_lifetime);
                                // Handle HRLs
                                if( e->m_trait.m_hrtbs ) {
                                    push_hrls(*e->m_trait.m_hrtbs);
                                }
                            }
                            if( const auto* e = t.data().opt_NodeType() ) {
                                DEBUG("Node Type: " << t);
                                TU_MATCH_HDRA((*e), {)
                                TU_ARMA(Async, p) {
                                    DEBUG("- " << p->m_capture_lifetime);
                                    equate_lifetime(p->m_capture_lifetime);
                                    for(const auto& cap : p->m_avu_cache.captured_vars) {
                                        visit_capture(p->m_capture_lifetime, cap.second, cap.first, {});
                                    }
                                    }
                                TU_ARMA(Closure, p) {
                                    DEBUG("- " << p->m_capture_lifetime);
                                    equate_lifetime(p->m_capture_lifetime);
                                    for(const auto& cap : p->m_avu_cache.captured_vars) {
                                        visit_capture(p->m_capture_lifetime, cap.usage, cap.root_slot, cap.fields);
                                    }
                                    }
                                TU_ARMA(Generator, p) {
                                    DEBUG("- " << p->m_capture_lifetime);
                                    equate_lifetime(p->m_capture_lifetime);
                                    for(const auto& cap : p->m_avu_cache.captured_vars) {
                                        visit_capture(p->m_capture_lifetime, cap.second, cap.first, {});
                                    }
                                    }
                                }

                            }
                            if( const auto* e = t.data().opt_Function() ) {
                                push_hrls(e->hrls);
                            }
                            if( t.data().is_Path() && t.data().as_Path().path.m_data.is_Generic() ) {
                                DEBUG("Generic path " << t);
                                for(const auto& l : t.data().as_Path().path.m_data.as_Generic().m_params.m_lifetimes)
                                    equate_lifetime(l);
                            }
                            HIR::Visitor::visit_type(t);
                            if( hrl_pushed ) {
                                m_hrls.pop_back();
                            }
                        }
                    } v { state, ep->m_span, ep.m_bindings, tpl.data().as_ErasedType().m_use.m_lifetimes };

                    DEBUG("Checking erased: " << ep.m_erased_types[e->m_index]);
                    DEBUG("vs " << tpl);
                    v.visit_type(ep.m_erased_types[e->m_index]);

                    rv = HIR::TypeRef::new_unit();
                    return true;
                }
                return false;
                });
        }


        // ---
        // Visit lifetimes in tree, updating all of them
        // ---
        struct Monomorph_CommitLifetimes: public Monomorphiser
        {
            LifetimeInferState& state;

            Monomorph_CommitLifetimes(LifetimeInferState& state): state(state) {}

            ::HIR::TypeRef get_type(const Span& sp, const ::HIR::GenericRef& g) const override {
                return HIR::TypeRef(g.name, g.binding);
            }
            ::HIR::ConstGeneric get_value(const Span& sp, const ::HIR::GenericRef& g) const override {
                return g;
            }
            ::HIR::LifetimeRef get_lifetime(const Span& sp, const ::HIR::GenericRef& g) const override {
                return HIR::LifetimeRef(g.binding);
            }

            ::HIR::LifetimeRef monomorph_lifetime(const Span& sp, const ::HIR::LifetimeRef& tpl) const override {
                if( tpl.binding == HIR::LifetimeRef::UNKNOWN || tpl.binding == HIR::LifetimeRef::INFER ) {
                    BUG(sp, "Unexpected unknown lifetime");
                }
                else if( const auto* iv = state.opt_ivar(sp, tpl) ) {
                    return clean_up_local(sp, iv->known);
                }
                else {
                    return clean_up_local(sp, tpl);
                }
            }

        private:
            HIR::LifetimeRef clean_up_local(const Span& sp, const HIR::LifetimeRef& tpl) const {
                ASSERT_BUG(sp, tpl.binding < ::HIR::LifetimeRef::MAX_LOCAL, "Found invalid lifetime - " << tpl);
                // If this is a composite, check if there is a clear shortest option
                // - If there is, use that instead
                // - Otherwise, keep as a composite
                //   > Ensure that the composite is recorded? (or just let the downstream code re-derive)
                if( const auto* l = state.opt_local(sp, tpl) ) {
                    if( /*const auto* c =*/ l->data.opt_Composite() ) {
                        ::std::vector<HIR::LifetimeRef> flat;
                        flatten_composite(flat, sp, tpl);
                        DEBUG(tpl << " flat=" << flat);
                        if( std::all_of(flat.begin(), flat.end(), [&](const HIR::LifetimeRef& l) { return state.opt_local(sp, l) == nullptr; }) )
                        {
                            ::HIR::LifetimeRef  min = ::HIR::LifetimeRef::new_static();
                            for(const auto& e : flat) {
                                ::std::vector<HIR::LifetimeRef> tmp;
                                // Skip actually equal options
                                if( e == min )
                                    continue ;
                                bool e_min = state.check_lifetimes(sp, e, min, tmp);
                                bool min_e = state.check_lifetimes(sp, min, e, tmp);
                                // If they're both equal, or they're not compatible (not sorting) then we can't get a minimum - have to keep as a template
                                if( e_min == min_e ) {
                                    min = ::HIR::LifetimeRef();
                                    break;
                                }
                                // Is `e` smaller than `min` - then update `min`
                                if( e_min ) {
                                    min = e;
                                }
                            }
                            if( min != ::HIR::LifetimeRef() ) {
                                DEBUG("-> " << min);
                                return min;
                            }
                        }
                        // TODO: Register and return a flattened composite?
                        // - Just return the original one (which will be just an unnamed local)
                        return tpl;
                    }
                    else {
                        return tpl;
                    }
                }
                return tpl;
            }
            void flatten_composite(::std::vector<HIR::LifetimeRef>& dst, const Span& sp, const HIR::LifetimeRef& tpl) const {
                if( const auto* l = state.opt_local(sp, tpl) ) {
                    if( const auto* c = l->data.opt_Composite() ) {
                        for(const auto& r : *c) {
                            flatten_composite(dst, sp, r);
                        }
                        return ;
                    }
                }
                dst.push_back(tpl);
            }
        } ms(state);
        {
            struct Visitor_CommitLifetimes: HIR::ExprVisitorDef
            {
                Monomorph_CommitLifetimes&   ms;
                Visitor_CommitLifetimes(Monomorph_CommitLifetimes& ms): ms(ms) {}

                void visit_root(HIR::ExprPtr& ep) {
                    for(auto& ty : ep.m_bindings) {
                        this->visit_type(ty);
                    }
                    for(auto& b : ep.m_erased_types) {
                        this->visit_type(b);
                    }

                    this->visit_type(ep->m_res_type);
                    ep->visit(*this);
                }

                void visit_node_ptr(::std::unique_ptr< ::HIR::ExprNode>& node_ptr) override {
                    assert(node_ptr);
                    visit_type(node_ptr->m_res_type);
                    HIR::ExprVisitorDef::visit_node_ptr(node_ptr);
                }
                void visit_type(HIR::TypeRef& ty) override {
                    TRACE_FUNCTION_FR(ty, ty);
                    ty = ms.monomorph_type(Span(), ty, false);
                }
                void visit_path_params(HIR::PathParams& pp) override {
                    pp = ms.monomorph_path_params(Span(), pp, false);
                }
                void visit(HIR::ExprNode_Closure& node) override {
                    if( !node.m_code ) {
                    }
                    else if( !node.m_is_move ) {
                        node.m_capture_lifetime = ms.monomorph_lifetime(node.span(), node.m_capture_lifetime);
                    }
                    HIR::ExprVisitorDef::visit(node);
                }
            } visitor(ms);
            visitor.visit_root(ep);
        }
        // Commit lifetimes in TAITs
        {
            for(auto& ent : state.m_tait_values)
            {
                const Span& sp = ep->span();    // TODO: Get a better liftime (first usage?)
                auto& tait_inner = *const_cast<HIR::TypeData_ErasedType_AliasInner*>(ent.first);
                HIR::PathParams& pp = ent.second.first;
                HIR::TypeRef& ty = ent.second.second;

                // Commit lifetimes
                pp = ms.monomorph_path_params(Span(), pp, false);
                ty = ms.monomorph_type(Span(), ty, false);
                DEBUG("TAIT " << tait_inner.path << " pp=" << pp << " ty=" << ty);

                // Do a reverse monomorph of `ty` to convert from local context into alias context
                if( !pp.m_values.empty() ) {
                    TODO(sp, "Handle const-generics in TAIT");
                }
                struct ReverseMonomorph
                    : public Monomorphiser
                {
                    /// Definition of the generics (on the TAIT inner)
                    const HIR::GenericParams& m_generics;
                    /// Local path params (input to the TAIT)
                    const HIR::PathParams& m_params;
                    ReverseMonomorph(const HIR::GenericParams& generics, const HIR::PathParams& params)
                        : m_generics(generics)
                        , m_params(params)
                    {}

                    // No generics should be reached, as all of them should have been covered by the params.
                    ::HIR::TypeRef get_type(const Span& sp, const ::HIR::GenericRef& g) const override {
                        TODO(sp, "Should this see any generics? Probably not: Type " << g);
                    }
                    ::HIR::ConstGeneric get_value(const Span& sp, const ::HIR::GenericRef& g) const override {
                        TODO(sp, "Should this see any generics? Probably not: Value " << g);
                    }
                    ::HIR::LifetimeRef get_lifetime(const Span& sp, const ::HIR::GenericRef& g) const override {
                        TODO(sp, "Should this see any generics? Probably not: Lifetime " << g);
                    }

                    HIR::TypeRef monomorph_type(const Span& sp, const HIR::TypeRef& tpl, bool allow_infer) const override {
                        for(size_t i = 0; i < m_params.m_types.size(); i++ ) {
                            if( tpl == m_params.m_types[i] ) {
                                return HIR::TypeRef(m_generics.m_types.at(i).m_name, i);
                            }
                        }
                        return Monomorphiser::monomorph_type(sp, tpl, allow_infer);
                    }
                    HIR::LifetimeRef monomorph_lifetime(const Span& sp, const HIR::LifetimeRef& tpl) const override {
                        for(size_t i = 0; i < m_params.m_lifetimes.size(); i++ ) {
                            if( tpl == m_params.m_lifetimes[i] ) {
                                return HIR::LifetimeRef(i);
                            }
                        }
                        return Monomorphiser::monomorph_lifetime(sp, tpl);
                    }
                } rm { tait_inner.generics, pp };

                HIR::TypeRef transformed_type = rm.monomorph_type(sp, ty, /*allow_infer=*/false);
                // Ensure that these types are compatible, by checking equality (which ignores liftimes)
                ASSERT_BUG(sp, tait_inner.type == transformed_type, "Lifetime expanded TAIT type did not match existing!:\n"
                    << "- Was: " << tait_inner.type
                    << "- Now: " << transformed_type
                    );
                DEBUG("Set TAIT " << tait_inner.path << " = " << transformed_type);
                tait_inner.type = std::move(transformed_type);
            }
        }
        {
            for(const auto& t : ep.m_erased_types) {
                struct MonomorphCheckLft: public MonomorphiserNop {
                    const HIR::TypeRef& tpl;
                    MonomorphCheckLft(const HIR::TypeRef& tpl): tpl(tpl) {}
                    HIR::LifetimeRef monomorph_lifetime(const Span& sp, const HIR::LifetimeRef& lft) const override {
                        //ASSERT_BUG(sp, lft.binding <= ::HIR::LifetimeRef::STATIC, "Found local/ivar lifetime - " << lft << "\n in " << tpl);
                        return lft;
                    }
                };
                DEBUG("Check for locals: " << t);
                MonomorphCheckLft(t).monomorph_type(ep->span(), t);
            }
        }

        // Run type de-duplication over the tree again
        ExprVisitor_CompactTypes().visit_root(ep);
        DEBUG("\n" << FMT_CB(os, HIR_DumpExpr(os, ep)));
        ep.m_state->stage = ::HIR::ExprState::Stage::Lifetimes;
    }

    class OuterVisitor:
        public ::HIR::Visitor
    {
        LocalTraitResolve m_resolve;
        bool m_remove_locals;
    public:
        OuterVisitor(const ::HIR::Crate& crate, bool remove_locals)
            : m_resolve(crate)
            , m_remove_locals(remove_locals)
        {}

        void check(const HIR::TypeRef& ret_ty, const ::HIR::Function::args_t& args, HIR::ExprPtr& root, bool is_function=false)
        {
            if( root.m_mir ) {
                DEBUG("MIR present, skipping");
                return ;
            }
            HIR_Expand_LifetimeInfer_ExprInner(m_resolve, args, ret_ty, root, m_remove_locals, !is_function);
        }

        // NOTE: This is left here to ensure that any expressions that aren't handled by higher code cause a failure
        void visit_expr(::HIR::ExprPtr& exp) override {
            BUG(Span(), "visit_expr hit in OuterVisitor");
        }

        void visit_type(::HIR::TypeRef& ty) override
        {
            if(auto* e = ty.data_mut().opt_Array())
            {
                this->visit_type( e->inner );
                DEBUG("Array size " << ty);
                if( auto* se1 = e->size.opt_Unevaluated() ) {
                    if( auto* se = se1->opt_Unevaluated() ) {
                        check(::HIR::TypeRef(::HIR::CoreType::Usize), ::HIR::Function::args_t {}, *(**se).expr);
                    }
                }
            }
            else {
                ::HIR::Visitor::visit_type(ty);
            }
        }
        // ------
        // Code-containing items
        // ------
        void visit_function(::HIR::ItemPath p, ::HIR::Function& item) override {
            if( item.m_code )
            {
                DEBUG("Function code " << p);
                auto _ = this->m_resolve.set_item_generics(item.m_params);
                check(item.m_return, item.m_args, item.m_code, /*is_function=*/true);
            }
            else
            {
                DEBUG("Function code " << p << " (none)");
            }
        }
        void visit_static(::HIR::ItemPath p, ::HIR::Static& item) override {
            if( item.m_value )
            {
                check(item.m_type, ::HIR::Function::args_t(), item.m_value);
            }
        }
        void visit_constant(::HIR::ItemPath p, ::HIR::Constant& item) override {
            if( item.m_value )
            {
                check(item.m_type, ::HIR::Function::args_t(), item.m_value);
            }
            m_resolve.expand_associated_types(Span(), item.m_type);
        }
        void visit_enum(::HIR::ItemPath p, ::HIR::Enum& item) override {
            //auto _ = this->m_ms.set_item_generics(item.m_params);

            if( auto* e = item.m_data.opt_Value() )
            {
                auto enum_type = ::HIR::Enum::get_repr_type(item.m_tag_repr);
                for(auto& var : e->variants)
                {
                    DEBUG("Enum value " << p << " - " << var.name);

                    if( var.expr )
                    {
                        check(enum_type, ::HIR::Function::args_t(), var.expr);
                    }
                }
            }
        }

        void visit_trait(::HIR::ItemPath p, ::HIR::Trait& item) override
        {
            TRACE_FUNCTION_F("trait " << p);
            auto _ = this->m_resolve.set_impl_generics(MetadataType::TraitObject, item.m_params);
            ::HIR::Visitor::visit_trait(p, item);
        }
        void visit_type_impl(::HIR::TypeImpl& impl) override
        {
            TRACE_FUNCTION_F("impl " << impl.m_type);
            auto _ = this->m_resolve.set_impl_generics(impl.m_type, impl.m_params);
            ::HIR::Visitor::visit_type_impl(impl);
        }
        void visit_trait_impl(const ::HIR::SimplePath& trait_path, ::HIR::TraitImpl& impl) override
        {
            TRACE_FUNCTION_F("impl" << impl.m_params.fmt_args() << " " << trait_path << " for " << impl.m_type);
            auto _ = this->m_resolve.set_impl_generics(impl.m_type, impl.m_params);
            ::HIR::Visitor::visit_trait_impl(trait_path, impl);
        }
    };

    bool LocalTraitResolve::replace_equalities(::HIR::TypeRef& input) const {
        static Span sp;
        if( true ) { //m_lifetime_state ) {
            TRACE_FUNCTION_FR(input, input);
            // Lookup ignoring lifetimes...
            auto a = m_type_equalities.find(input);
            if( a != m_type_equalities.end() ) {
                DEBUG("From " << a->first);
                DEBUG("To " << a->second.ty);
                // Then equate, matching lifetimes through
                if( !a->second.hrbs.is_empty() ) {
                    // Match the HRLs, so they can propagate through to the output
                    struct Matcher: public HIR::MatchGenerics {
                        const Span& sp;
                        HIR::PathParams pp;
                        Matcher(const Span& sp, const HIR::GenericParams& def)
                            : sp(sp)
                        {
                            pp.m_lifetimes.resize( def.m_lifetimes.size() );
                        }
                        ::HIR::Compare match_ty(const ::HIR::GenericRef& g, const ::HIR::TypeRef& ty, HIR::t_cb_resolve_type resolve_cb) override {
                            assert(ty.data().is_Generic());
                            assert(ty.data().as_Generic().binding == g.binding);
                            return ::HIR::Compare::Equal;
                        }
                        HIR::Compare match_val(const HIR::GenericRef&, const HIR::ConstGeneric&) override {
                            return ::HIR::Compare::Equal;
                        }
                        HIR::Compare match_lft(const HIR::GenericRef& g, const HIR::LifetimeRef& lft) override {
                            if( g.group() != 3 ) {
                                if(g.binding != lft.binding) {
                                    DEBUG("TODO: Test " << g << " == " << lft);
                                }
                                return ::HIR::Compare::Equal;
                            }
                            else {
                                auto& slot = pp.m_lifetimes.at(g.idx());
                                if( slot != HIR::LifetimeRef() ) {
                                    // TODO: If already bound, then may want to do equalities
                                    // - OR, define new ivars for each entry in `hrbs` and do equalities there
                                    // - BUT, this function is called as part of trait lookup, and may not be a final decision.

                                    // For now, just assume equality - until that causes a lifetime infer error
                                }
                                else {
                                    slot = lft;
                                }
                                return ::HIR::Compare::Equal;
                            }
                        }
                    } m { sp, a->second.hrbs };
                    HIR::ResolvePlaceholdersNop  resolve_placeholders;
                    a->first.match_test_generics(sp, input, resolve_placeholders, m);
                    //TODO(sp, "Handle HRL match: for" << a->second.hrbs.fmt_args() << " " << input << " (" << a->first << ") => " << a->second.ty);
                    input = MonomorphHrlsOnly(m.pp).monomorph_type(sp, a->second.ty);
                }
                else {
                    input = a->second.ty.clone();
                }
                return true;
            }
            return false;
        }
        else /* */ {
            return StaticTraitResolve::replace_equalities(input);
        }
    }
}

void HIR_Expand_LifetimeInfer(::HIR::Crate& crate)
{
    OuterVisitor    ov(crate, false);
    ov.visit_crate( crate );
}
void HIR_Expand_LifetimeInfer_Validate(::HIR::Crate& crate)
{
    // TODO: When running, clear all local lifetimes (replace with empty - to be turned into ivars)
    OuterVisitor    ov(crate, true);
    ov.visit_crate( crate );
}

void HIR_Expand_LifetimeInfer_Expr(const ::HIR::Crate& crate, const ::HIR::ItemPath& ip, const HIR::Function::args_t& args, const HIR::TypeRef& ret_ty, ::HIR::ExprPtr& exp)
{
    TRACE_FUNCTION_F("ip=" << ip << " ret_ty=" << ret_ty << ", args=" << args);
    LocalTraitResolve  resolve { crate };
    resolve.set_both_generics_raw(exp.m_state->m_impl_generics, exp.m_state->m_item_generics);

    HIR_Expand_LifetimeInfer_ExprInner(resolve, args, ret_ty, exp, /*remove_locals*/false, /*is_const_context=*/true);
}
