From f4731e6c18c75761d35fceddd409c381b96c5af9 Mon Sep 17 00:00:00 2001
From: Xiaoran <lilian.zhou26@outlook.com>
Date: Mon, 18 Mar 2024 17:15:56 -0700
Subject: [PATCH] Add function add_schema_metadata for local repository

---
 .DS_Store                      | Bin 0 -> 6148 bytes
 oxen/Cargo.lock                |   4 +---
 oxen/python/oxen/local_repo.py |   7 +++++++
 oxen/src/py_local_repo.rs      |  21 +++++++++++++++++++++
 4 files changed, 29 insertions(+), 3 deletions(-)
 create mode 100644 .DS_Store

diff --git a/.DS_Store b/.DS_Store
new file mode 100644
index 0000000000000000000000000000000000000000..3f716695dc148592a705d963205a8ffccaec1bd4
GIT binary patch
literal 6148
zcmeH~F>b>!3`IX%4*|M(?5HIN=naG*JwY!Jq{$Y<Ll7XNj-F47OP$Px5qtvV6Dbq6
z|6rK_Y;!$)0V9AF-HEk_nHlo|6K*)-e7O9c_S1CoBJHgMp3+Cm_H$d10#ZN<NC7Dz
z1s0@09^<Qz1wE4<MG8oPWhmg^heCJOWb2Gi2Sbbi<iK(m*D*_w#S3IjwoX=PmeYe}
ztHl`N^=KzcURRT?v$w-?_^`aQ`4mI5-VQ5FXjTIXQa}pK6nO9X=;!}8{n`9KYf&l%
zq`-?QV8ixjyWvaa+4}H$Uf*Ta*Nslb<qUs*0+{$w{7Dbve(?oaldY2#ntlXA1_deb
HuL|4&H+d4j

literal 0
HcmV?d00001

diff --git a/oxen/Cargo.lock b/oxen/Cargo.lock
index 46dfda6..0c19bd5 100644
--- a/oxen/Cargo.lock
+++ b/oxen/Cargo.lock
@@ -2517,9 +2517,7 @@ checksum = "f7012b1bbb0719e1097c47611d3898568c546d597c2e74d66f6087edd5233ff4"
 
 [[package]]
 name = "liboxen"
-version = "0.12.2"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "766b27bcc2c9a8074dd24ef8e9d8c5365358227ec073674429949c0f5972cac8"
+version = "0.12.3"
 dependencies = [
  "actix-files",
  "actix-web",
diff --git a/oxen/python/oxen/local_repo.py b/oxen/python/oxen/local_repo.py
index afdc83f..c5bef47 100644
--- a/oxen/python/oxen/local_repo.py
+++ b/oxen/python/oxen/local_repo.py
@@ -112,6 +112,13 @@ def add(self, path: str):
 
         self._repo.add(path)
 
+    def add_schema_metadata(self, path: str, column_name: str, metadata: str):
+        """
+        Add schema to the local repository
+        """
+        self._repo.add_schema_metadata(path, column_name, metadata)
+
+
     def rm(self, path: str, recursive=False, staged=False, remote=False):
         """
         Remove a file or directory from being tracked.
diff --git a/oxen/src/py_local_repo.rs b/oxen/src/py_local_repo.rs
index c5a0192..6c7f9cc 100644
--- a/oxen/src/py_local_repo.rs
+++ b/oxen/src/py_local_repo.rs
@@ -71,6 +71,27 @@ impl PyLocalRepo {
         Ok(())
     }
 
+    pub fn add_schema_metadata(&self, path: &str, column: &str, metadata: &str) -> Result<(), PyOxenError> {
+        let repo = LocalRepository::from_dir(&self.path)?;
+
+        // make sure metadata is valid json, return oxen error if not
+        let metadata: serde_json::Value = serde_json::from_str(metadata.as_ref()).map_err(|e| {
+            OxenError::basic_str(format!(
+                "Metadata must be valid JSON: ''\n{}",
+                // metadata.as_ref(),
+                e
+            ))
+        })?;
+    
+        for (path, schema) in
+            command::schemas::add_column_metadata(&repo, path, column, &metadata)?
+        {
+            println!("{:?}\n{}", path, schema.verbose_str());
+        }
+    
+        Ok(())
+    }
+
     pub fn rm(&self, path: PathBuf, recursive: bool, staged: bool, remote: bool) -> Result<(), PyOxenError> {
         let repo = LocalRepository::from_dir(&self.path)?;
         let rm_opts = RmOpts {