Support native stdlib

src/cli/cmd_build_and_check.rs

@@ -11,14 +11,16 @@ use crate::{
         r1cs::{snarkjs::SnarkjsExporter, R1CS},
         Backend, BackendField, BackendKind,
     },
-    cli::packages::path_to_package,
+    cli::packages::{path_to_package, path_to_stdlib},
     compiler::{compile, generate_witness, typecheck_next_file, Sources},
     inputs::{parse_inputs, JsonInputs},
+    stdlib::init_stdlib_dep,
     type_checker::TypeChecker,
 };
 
 use super::packages::{
-    get_deps_of_package, is_lib, validate_package_and_get_manifest, DependencyGraph, UserRepo,
+    download_stdlib, get_deps_of_package, is_lib,
+    validate_package_and_get_manifest, DependencyGraph, UserRepo,
 };
 
 const COMPILED_DIR: &str = "compiled";
@@ -137,6 +139,26 @@ pub fn cmd_check(args: CmdCheck) -> miette::Result<()> {
     Ok(())
 }
 
+fn add_stdlib<B: Backend>(
+    sources: &mut Sources,
+    tast: &mut TypeChecker<B>,
+    node_id: usize,
+) -> miette::Result<usize> {
+    let mut node_id = node_id;
+
+    // check if the release folder exists, otherwise download the latest release
+    // todo: check the latest version and compare it with the current version, to decide if download is needed
+    let stdlib_dir = path_to_stdlib();
+
+    if !stdlib_dir.exists() {
+        download_stdlib()?;
+    }
+
+    node_id = init_stdlib_dep(sources, tast, node_id, stdlib_dir.as_ref());
+
+    Ok(node_id)
+}
+
 fn produce_all_asts<B: Backend>(path: &PathBuf) -> miette::Result<(Sources, TypeChecker<B>)> {
     // find manifest
     let manifest = validate_package_and_get_manifest(&path, false)?;
@@ -161,6 +183,9 @@ fn produce_all_asts<B: Backend>(path: &PathBuf) -> miette::Result<(Sources, Type
 
     let mut tast = TypeChecker::new();
 
+    // adding stdlib
+    add_stdlib(&mut sources, &mut tast, node_id)?;
+
     for dep in dep_graph.from_leaves_to_roots() {
         let path = path_to_package(&dep);
 

src/cli/mod.rs

@@ -15,3 +15,6 @@ pub const NONAME_DIRECTORY: &str = ".noname";
 
 /// The directory under [NONAME_DIRECTORY] containing all package-related files.
 pub const PACKAGE_DIRECTORY: &str = "packages";
+
+/// The directory under [NONAME_DIRECTORY] containing all the latest noname release.
+pub const RELEASE_DIRECTORY: &str = "release";

src/cli/packages.rs

@@ -9,7 +9,7 @@ use serde::{Deserialize, Serialize};
 
 use super::{
     manifest::{read_manifest, Manifest},
-    NONAME_DIRECTORY, PACKAGE_DIRECTORY,
+    NONAME_DIRECTORY, PACKAGE_DIRECTORY, RELEASE_DIRECTORY,
 };
 
 /// A dependency is a Github `user/repo` pair.
@@ -241,6 +241,16 @@ pub(crate) fn path_to_package(dep: &UserRepo) -> PathBuf {
     package_dir.join(&dep.user).join(&dep.repo)
 }
 
+pub(crate) fn path_to_stdlib() -> PathBuf {
+    let home_dir: PathBuf = dirs::home_dir()
+        .expect("could not find home directory of current user")
+        .try_into()
+        .expect("invalid UTF8 path");
+    let noname_dir = home_dir.join(NONAME_DIRECTORY);
+
+    noname_dir.join(RELEASE_DIRECTORY).join("src/stdlib/native")
+}
+
 /// download package from github
 pub fn download_from_github(dep: &UserRepo) -> Result<()> {
     let url = format!(
@@ -264,6 +274,44 @@ pub fn download_from_github(dep: &UserRepo) -> Result<()> {
     Ok(())
 }
 
+pub fn download_stdlib() -> Result<()> {
+    // Hardcoded repository details and target branch
+    let repo_owner = "katat";
+    let repo_name = "noname";
+    let target_branch = "release";
+    let repo_url = format!(
+        "https://github.com/{owner}/{repo}.git",
+        owner = repo_owner,
+        repo = repo_name
+    );
+
+    let home_dir: PathBuf = dirs::home_dir()
+        .expect("could not find home directory of current user")
+        .try_into()
+        .expect("invalid UTF8 path");
+    let noname_dir = home_dir.join(NONAME_DIRECTORY);
+    let release_dir = noname_dir.join("release");
+
+    // Clone the repository and checkout the specified branch to the temporary directory
+    let output = process::Command::new("git")
+        .arg("clone")
+        .arg("--branch")
+        .arg(target_branch)
+        .arg("--single-branch")
+        .arg(repo_url)
+        .arg(release_dir)
+        .output()
+        .expect("failed to execute git clone command");
+
+    if !output.status.success() {
+        miette::bail!(format!(
+            "Could not clone branch `{target_branch}` of repository `{repo_owner}/{repo_name}`."
+        ));
+    }
+
+    Ok(())
+}
+
 pub fn is_lib(path: &PathBuf) -> bool {
     path.join("src").join("lib.no").exists()
 }

src/stdlib/mod.rs

@@ -87,6 +87,7 @@ pub fn init_stdlib_dep<B: Backend>(
     sources: &mut Sources,
     tast: &mut TypeChecker<B>,
     node_id: usize,
+    path_prefix: &str,
 ) -> usize {
     // list the stdlib dependency in order
     let libs = vec!["bits", "int", "comparator"];
@@ -95,7 +96,7 @@ pub fn init_stdlib_dep<B: Backend>(
 
     for lib in libs {
         let module = UserRepo::new(&format!("std/{}", lib));
-        let prefix_stdlib = Path::new("src/stdlib/native/");
+        let prefix_stdlib = Path::new(path_prefix);
         let code = std::fs::read_to_string(prefix_stdlib.join(format!("{lib}.no"))).unwrap();
         node_id =
             typecheck_next_file(tast, Some(module), sources, lib.to_string(), code, 0).unwrap();

src/tests/examples.rs

@@ -38,7 +38,7 @@ fn test_file(
             let mut sources = Sources::new();
             let mut tast = TypeChecker::new();
             let mut node_id = 0;
-            node_id = init_stdlib_dep(&mut sources, &mut tast, node_id);
+            node_id = init_stdlib_dep(&mut sources, &mut tast, node_id, "src/stdlib/native/");
             let this_module = None;
             let _node_id = typecheck_next_file(
                 &mut tast,
@@ -102,7 +102,7 @@ fn test_file(
             let mut sources = Sources::new();
             let mut tast = TypeChecker::new();
             let mut node_id = 0;
-            node_id = init_stdlib_dep(&mut sources, &mut tast, node_id);
+            node_id = init_stdlib_dep(&mut sources, &mut tast, node_id, "src/stdlib/native/");
             let this_module = None;
             let _node_id = typecheck_next_file(
                 &mut tast,

src/tests/stdlib/mod.rs

@@ -36,7 +36,7 @@ fn test_stdlib(
     let mut sources = Sources::new();
     let mut tast = TypeChecker::new();
     let mut node_id = 0;
-    node_id = init_stdlib_dep(&mut sources, &mut tast, node_id);
+    node_id = init_stdlib_dep(&mut sources, &mut tast, node_id, "src/stdlib/native/");
 
     let this_module = None;
     let _node_id = typecheck_next_file(