Improve import handling in parser

5 files changed, 44 insertions, 6 deletions

diff --git a/dhall/src/binary.rs b/dhall/src/binary.rs
index 8c5b902..1fa075e 100644
--- a/dhall/src/binary.rs
+++ b/dhall/src/binary.rs
@@ -1,6 +1,7 @@
 use dhall_core::*;
 use itertools::*;
 use serde_cbor::value::value as cbor;
+use std::path::PathBuf;
 use std::rc::Rc;
 
 type ParsedExpr = Rc<Expr<X, Import>>;
@@ -175,6 +176,20 @@ fn cbor_value_to_dhall(data: &cbor::Value) -> Result<ParsedExpr, DecodeError> {
                         .collect::<Result<_, _>>()?,
                 )))
             }
+            [U64(24), Null, U64(0), U64(3), rest..] => {
+                let mut path = PathBuf::new();
+                for s in rest {
+                    match s {
+                        String(s) => path.push(s),
+                        _ => Err(DecodeError::WrongFormatError)?,
+                    }
+                }
+                Embed(Import {
+                    mode: ImportMode::Code,
+                    hash: None,
+                    location: ImportLocation::Local(FilePrefix::Here, path),
+                })
+            }
             [U64(25), bindings..] => {
                 let mut tuples = bindings.iter().tuples();
                 let bindings = (&mut tuples)
diff --git a/dhall/src/imports.rs b/dhall/src/imports.rs
index 2435663..9f60ee7 100644
--- a/dhall/src/imports.rs
+++ b/dhall/src/imports.rs
@@ -84,3 +84,12 @@ pub fn load_dhall_file(
     };
     Ok(expr)
 }
+
+pub fn load_dhall_file_no_resolve_imports(
+    f: &Path,
+) -> Result<ParsedExpr, DhallError> {
+    let mut buffer = String::new();
+    File::open(f)?.read_to_string(&mut buffer)?;
+    let expr = parse_expr(&*buffer)?;
+    Ok(expr)
+}
diff --git a/dhall/src/lib.rs b/dhall/src/lib.rs
index d8ca955..dcc1ff3 100644
--- a/dhall/src/lib.rs
+++ b/dhall/src/lib.rs
@@ -10,7 +10,16 @@
 mod normalize;
 pub use crate::normalize::*;
 pub mod binary;
+pub mod dhall_type;
 pub mod imports;
 pub mod typecheck;
 
-pub use crate::imports::{load_dhall_file, DhallError};
+pub use crate::imports::*;
+
+pub struct DhallExpr(dhall_core::DhallExpr);
+
+impl DhallExpr {
+    pub fn normalize(self) -> Self {
+        DhallExpr(crate::normalize::normalize(self.0))
+    }
+}
diff --git a/dhall/tests/common/mod.rs b/dhall/tests/common/mod.rs
index ffc6d06..6b893e0 100644
--- a/dhall/tests/common/mod.rs
+++ b/dhall/tests/common/mod.rs
@@ -46,6 +46,12 @@ pub fn read_dhall_file<'i>(file_path: &str) -> Result<Expr<X, X>, DhallError> {
     load_dhall_file(&PathBuf::from(file_path), true)
 }
 
+pub fn read_dhall_file_no_resolve_imports<'i>(
+    file_path: &str,
+) -> Result<ParsedExpr, DhallError> {
+    load_dhall_file_no_resolve_imports(&PathBuf::from(file_path))
+}
+
 pub fn run_test(base_path: &str, feature: Feature) {
     use self::Feature::*;
     let base_path_prefix = match feature {
@@ -61,7 +67,7 @@ pub fn run_test(base_path: &str, feature: Feature) {
         ParserSuccess => {
             let expr_file_path = base_path.clone() + "A.dhall";
             let expected_file_path = base_path + "B.dhallb";
-            let expr = read_dhall_file(&expr_file_path)
+            let expr = read_dhall_file_no_resolve_imports(&expr_file_path)
                 .map_err(|e| println!("{}", e))
                 .unwrap();
 
@@ -71,18 +77,17 @@ pub fn run_test(base_path: &str, feature: Feature) {
             let mut data = Vec::new();
             file.read_to_end(&mut data).unwrap();
             let expected = dhall::binary::decode(&data).unwrap();
-            let expected = dhall::imports::panic_imports(&expected);
 
             assert_eq_pretty!(expr, expected);
 
             // Round-trip pretty-printer
             let expr = parse_expr(&expr.to_string()).unwrap();
-            let expr = dhall::imports::panic_imports(&expr);
             assert_eq!(expr, expected);
         }
         ParserFailure => {
             let file_path = base_path + ".dhall";
-            let err = read_dhall_file(&file_path).unwrap_err();
+            let err =
+                read_dhall_file_no_resolve_imports(&file_path).unwrap_err();
             match err {
                 DhallError::ParseError(_) => {}
                 e => panic!("Expected parse error, got: {:?}", e),
diff --git a/dhall/tests/parser.rs b/dhall/tests/parser.rs
index 4fa0ff7..79a059d 100644
--- a/dhall/tests/parser.rs
+++ b/dhall/tests/parser.rs
@@ -17,7 +17,7 @@ parser_success!(spec_parser_success_annotations, "annotations");
 // parser_success!(spec_parser_success_asText, "asText");
 parser_success!(spec_parser_success_blockComment, "blockComment");
 parser_success!(spec_parser_success_builtins, "builtins");
-// parser_success!(spec_parser_success_collectionImportType, "collectionImportType");
+parser_success!(spec_parser_success_collectionImportType, "collectionImportType");
 parser_success!(spec_parser_success_double, "double");
 parser_success!(spec_parser_success_doubleQuotedString, "doubleQuotedString");
 // parser_success!(spec_parser_success_environmentVariables, "environmentVariables");