From 7683b0d762cf0df489ad4bc006e8db2358e81cf4 Mon Sep 17 00:00:00 2001
From: Nadrieril
Date: Tue, 28 Jan 2020 21:50:04 +0000
Subject: Implement assert & merge and fix more bugs

---
 dhall/src/semantics/phase/normalize.rs | 20 +++++++++++++++++---
 1 file changed, 17 insertions(+), 3 deletions(-)

(limited to 'dhall/src/semantics/phase/normalize.rs')

diff --git a/dhall/src/semantics/phase/normalize.rs b/dhall/src/semantics/phase/normalize.rs
index a11cb75..532dae3 100644
--- a/dhall/src/semantics/phase/normalize.rs
+++ b/dhall/src/semantics/phase/normalize.rs
@@ -855,17 +855,30 @@ pub(crate) enum NzEnvItem {
 #[derive(Debug, Clone)]
 pub(crate) struct NzEnv {
     items: Vec<NzEnvItem>,
+    vars: QuoteEnv,
 }
 
 impl NzEnv {
     pub fn new() -> Self {
-        NzEnv { items: Vec::new() }
+        NzEnv {
+            items: Vec::new(),
+            vars: QuoteEnv::new(),
+        }
     }
     pub fn construct(items: Vec<NzEnvItem>) -> Self {
-        NzEnv { items }
+        let vars = QuoteEnv::construct(
+            items
+                .iter()
+                .filter(|i| match i {
+                    NzEnvItem::Kept(_) => true,
+                    NzEnvItem::Replaced(_) => false,
+                })
+                .count(),
+        );
+        NzEnv { items, vars }
     }
     pub fn as_quoteenv(&self) -> QuoteEnv {
-        QuoteEnv::construct(self.items.len())
+        self.vars
     }
     pub fn to_alpha_tyenv(&self) -> TyEnv {
         TyEnv::from_nzenv_alpha(self)
@@ -874,6 +887,7 @@ impl NzEnv {
     pub fn insert_type(&self, t: Value) -> Self {
         let mut env = self.clone();
         env.items.push(NzEnvItem::Kept(t));
+        env.vars = env.vars.insert();
         env
     }
     pub fn insert_value(&self, e: Value) -> Self {
-- 
cgit v1.2.3