/* file LocalEnv.hpp
 *
 * author: Roland Conybeare, Jun 2024
 */

#pragma once

#include "Environment.hpp"

namespace xo {
    namespace ast {
        /** @brief LocalEnv
         *
         *  @class Local environment for a lambda.
         *  Lists the Variables corresponding to this lambda's formal
         *  parameters,  but also links to @ref Environment for
         *  innermost enclosing @ref Lambda.
         **/
        class LocalEnv : public Environment {
        public:
            using TypeDescr = xo::reflect::TypeDescr;

        public:
            /** named ctor idiom.  Create instance with local variables per @p argv **/
            static ref::rp<LocalEnv> make(const std::vector<ref::rp<Variable>> & argv) {
                return new LocalEnv(argv);
            }

            const std::vector<ref::rp<Variable>> & argv() const { return argv_; }
            int n_arg() const { return argv_.size(); }
            TypeDescr fn_arg(uint32_t i) const { return argv_[i]->valuetype(); }

            /** single-assign this environment's parent **/
            void assign_parent(ref::brw<Environment> p) {
                assert(parent_env_.get() == nullptr);
                parent_env_ = p.get();
            }

            // ----- Environment -----

            virtual ref::brw<Expression> lookup_var(const std::string & target) const override {
                for (const auto & arg : argv_) {
                    if (arg->name() == target)
                        return arg;
                }

                /* here: target not found in local vars,
                 * delegate to innermost ancestor
                 */
                return parent_env_->lookup_var(target);
            }

        private:
            LocalEnv(const std::vector<ref::rp<Variable>> & argv)
                : argv_(argv) {}

        private:
            /** formal argument names **/
            std::vector<ref::rp<Variable>> argv_;

            /** parent environment.  Free variable in this lambda's
             *  body,  will be resolved by referring them to @ref parent_env_.
             **/
            ref::rp<Environment> parent_env_;
        };
    } /*namespace ast*/
} /*namespace xo*/


/* end LocalEnv.hpp */