Merge pull request #82 from lambdaclass/check

Add code checking skeleton

Cargo.toml

@@ -1,7 +1,7 @@
 [workspace]
 resolver = "2"
 
-members = [ "crates/concrete","crates/concrete_ast", "crates/concrete_codegen_mlir", "crates/concrete_driver", "crates/concrete_parser", "crates/concrete_session", "crates/concrete_type_checker"]
+members = [ "crates/concrete","crates/concrete_ast", "crates/concrete_codegen_mlir", "crates/concrete_driver", "crates/concrete_parser", "crates/concrete_session", "crates/concrete_check"]
 
 [profile.release]
 lto = true

crates/concrete_ast/src/common.rs

@@ -1,3 +1,5 @@
+use std::ops::Range;
+
 #[derive(Debug, Clone, Copy, Hash, PartialEq, Eq, PartialOrd, Ord)]
 pub struct Span {
     pub from: usize,
@@ -10,6 +12,12 @@ impl Span {
     }
 }
 
+impl From<Span> for Range<usize> {
+    fn from(val: Span) -> Self {
+        val.from..val.to
+    }
+}
+
 #[derive(Clone, Debug, Eq, Hash, PartialEq, PartialOrd, Ord)]
 pub struct DocString {
     contents: String,

crates/concrete_ast/src/functions.rs

@@ -1,5 +1,5 @@
 use crate::{
-    common::{DocString, GenericParam, Ident},
+    common::{DocString, GenericParam, Ident, Span},
     statements::Statement,
     types::TypeSpec,
 };
@@ -17,6 +17,7 @@ pub struct FunctionDecl {
 pub struct FunctionDef {
     pub decl: FunctionDecl,
     pub body: Vec<Statement>,
+    pub span: Span,
 }
 
 #[derive(Clone, Debug, Eq, Hash, PartialEq)]

crates/concrete_ast/src/imports.rs

@@ -1,7 +1,8 @@
-use crate::common::Ident;
+use crate::common::{Ident, Span};
 
 #[derive(Clone, Debug, Eq, Hash, PartialEq)]
 pub struct ImportStmt {
     pub module: Vec<Ident>,
     pub symbols: Vec<Ident>,
+    pub span: Span,
 }

crates/concrete_ast/src/modules.rs

@@ -1,5 +1,5 @@
 use crate::{
-    common::{DocString, Ident},
+    common::{DocString, Ident, Span},
     constants::ConstantDef,
     functions::FunctionDef,
     imports::ImportStmt,
@@ -13,6 +13,7 @@ pub struct Module {
     pub imports: Vec<ImportStmt>,
     pub name: Ident,
     pub contents: Vec<ModuleDefItem>,
+    pub span: Span,
 }
 
 #[derive(Clone, Debug, Eq, PartialEq)]

crates/concrete_ast/src/types.rs

@@ -35,6 +35,14 @@ impl TypeSpec {
             TypeSpec::Array { is_ref, .. } => *is_ref,
         }
     }
+
+    pub fn get_name(&self) -> String {
+        match self {
+            TypeSpec::Simple { name, .. } => name.name.clone(),
+            TypeSpec::Generic { name, .. } => name.name.clone(),
+            TypeSpec::Array { of_type, .. } => format!("[{}]", of_type.get_name()),
+        }
+    }
 }
 
 #[derive(Clone, Debug, Eq, Hash, PartialEq)]

crates/concrete_check/Cargo.toml

@@ -0,0 +1,13 @@
+[package]
+name = "concrete_check"
+version = "0.1.0"
+edition = "2021"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+ariadne = { version = "0.4.0", features = ["auto-color"] }
+concrete_ast = { version = "0.1.0", path = "../concrete_ast" }
+concrete_session = { version = "0.1.0", path = "../concrete_session" }
+itertools = "0.12.0"
+thiserror = "1.0.56"