Implement OSDK functionalities and opt-in OSDK for asterinas

osdk/src/utils.rs

@@ -1,13 +1,16 @@
 // SPDX-License-Identifier: MPL-2.0
 
-use std::{ffi::OsStr, path::Path, process::Command};
+use std::{ffi::OsStr, path::{Path, PathBuf}, process::Command};
 
 use crate::{error::Errno, error_msg};
 
 // FIXME: Crates belonging to Asterinas require a different dependency format. The dependency
 // should be specified using a relative path instead of a URL.
+// TODO: The dependency should be corrected when this branch is merged.
 pub const ASTER_FRAME_DEP: &str =
-    "aster-frame = { git = \"https://github.com/asterinas/asterinas\", rev = \"f2f991b\" }";
+    "aster-frame = { git = \"https://github.com/junyang-zh/asterinas\", branch = \"osdk\" }";
+pub const KTEST_DEP: &str =
+    "ktest = { git = \"https://github.com/junyang-zh/asterinas\", branch = \"osdk\" }";
 
 fn cargo() -> Command {
     Command::new("cargo")
@@ -24,10 +27,13 @@ pub fn cargo_new_lib(crate_name: &str) {
     }
 }
 
+/// Get the Cargo metadata parsed from the standard output
+/// of the invocation of Cargo. Return `None` if the command
+/// fails or the `current_dir` is not in a Cargo workspace.
 pub fn get_cargo_metadata<S1: AsRef<Path>, S2: AsRef<OsStr>>(
     current_dir: Option<S1>,
     cargo_args: Option<&[S2]>,
-) -> serde_json::Value {
+) -> Option<serde_json::Value> {
     let mut command = cargo();
     command.args(["metadata", "--no-deps", "--format-version", "1"]);
 
@@ -42,13 +48,48 @@ pub fn get_cargo_metadata<S1: AsRef<Path>, S2: AsRef<OsStr>>(
     let output = command.output().unwrap();
 
     if !output.status.success() {
-        let stderr = String::from_utf8_lossy(&output.stderr);
-        eprintln!("{}", &stderr);
-
-        error_msg!("Failed to get metadata for newly created crate");
-        std::process::exit(Errno::GetMetadata as _);
+        return None;
     }
 
     let stdout = String::from_utf8_lossy(&output.stdout);
-    serde_json::from_str(&stdout).unwrap()
+    Some(serde_json::from_str(&stdout).unwrap())
+}
+
+pub fn get_target_directory() -> PathBuf {
+    let metadata = get_cargo_metadata(None::<&str>, None::<&[&str]>).unwrap();
+    metadata
+        .get("target_directory")
+        .unwrap()
+        .as_str()
+        .unwrap()
+        .into()
+}
+
+pub struct CrateInfo {
+    pub name: String,
+    pub version: String,
+    pub path: String,
+}
+
+pub fn get_current_crate_info() -> CrateInfo {
+    let metadata = get_cargo_metadata(None::<&str>, None::<&[&str]>).unwrap();
+    let default_members = metadata.get("workspace_default_members").unwrap();
+    assert_eq!(default_members.as_array().unwrap().len(), 1);
+    // The default member string here is in the form of "<crate_name> <crate_version> (path+file://<crate_path>)"
+    let default_member = default_members[0]
+        .as_str()
+        .unwrap()
+        .split(" ")
+        .collect::<Vec<&str>>();
+    let name = default_member[0].to_string();
+    let version = default_member[1].to_string();
+    let path = default_member[2]
+        .trim_start_matches("(path+file://")
+        .trim_end_matches(")")
+        .to_string();
+    CrateInfo {
+        name,
+        version,
+        path,
+    }
 }