From a48f308f05a116f8ca05c88f3e17eb597de45542 Mon Sep 17 00:00:00 2001
From: dachengx <dx2227@columbia.edu>
Date: Sun, 15 Sep 2024 14:49:07 -0500
Subject: [PATCH 1/2] Add a class to tarball the editable user-installed from
 git repo

---
 utilix/tarball.py | 100 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 100 insertions(+)
 create mode 100644 utilix/tarball.py

diff --git a/utilix/tarball.py b/utilix/tarball.py
new file mode 100644
index 0000000..9d14c4b
--- /dev/null
+++ b/utilix/tarball.py
@@ -0,0 +1,100 @@
+import os
+import site
+import tarfile
+import importlib
+from git import Repo, InvalidGitRepositoryError
+
+
+def filter_tarinfo(tarinfo, git_ignored_files):
+    """Custom filter for tarfile to exclude Git-ignored files."""
+
+    # Exclude Git-ignored files
+    if any(f in tarinfo.name for f in git_ignored_files):
+        return None
+
+    # Include the file
+    return tarinfo
+
+
+class Tarball:
+    def __init__(self, destination, package_name):
+        """Class to create a tarball of package.
+
+        :param destination: the destination folder of the tarball
+        :param package_name: the name of package
+
+        """
+        self.destination = destination
+        self.package_name = package_name
+
+    @property
+    def tarball_name(self):
+        return f"{self.package_name}.tar.gz"
+
+    @property
+    def tarball_path(self):
+        return os.path.join(self.destination, self.tarball_name)
+
+    def create_tarball(self):
+        """Create the tarball of package."""
+
+        if os.path.exists(self.tarball_path):
+            raise RuntimeError(f"{self.tarball_path} already exists!")
+
+        # Get the origin of the package source
+        repo = Tarball.get_installed_git_repo(self.package_name)
+        package_origin = repo.working_dir
+        if not package_origin:
+            raise RuntimeError(
+                f"Can not make tarball of package {self.package_name},"
+                "because it is not in user-installed editable mode."
+            )
+
+        # List ignored files
+        git_ignored_files = repo.git.ls_files(
+            "--others", "--ignored", "--exclude-standard"
+        ).splitlines()
+
+        # Define the output tarball filename
+        with tarfile.open(self.tarball_path, "w:gz") as tar:
+            tar.add(
+                package_origin,
+                arcname=os.path.basename(package_origin),
+                recursive=True,
+                filter=lambda tarinfo: filter_tarinfo(tarinfo, git_ignored_files),
+            )
+
+    @staticmethod
+    def get_installed_git_repo(package_name):
+        """If a package is in editable user-installed mode, we can get its git working directory.
+
+        The editable user-installed is usually installed by `pip install -e ./ --user`.
+
+        """
+        # Find the package's location
+        mod = importlib.import_module(package_name)
+        # Try initialize git repository
+        try:
+            repo = Repo(mod.__path__[0], search_parent_directories=True)
+            return repo
+        except InvalidGitRepositoryError:
+            return
+
+    @staticmethod
+    def is_user_installed(package_name):
+        """Test if a package is in user-installed mode.
+
+        The user-installed is usually installed by `pip install ./ --user`.
+
+        """
+        # Find the package's location using importlib
+        spec = importlib.util.find_spec(package_name)
+        if spec is None:
+            raise ModuleNotFoundError(f"Package {package_name} is not installed.")
+        package_location = spec.origin
+
+        # Get the user-specific package directory
+        user_site_packages = site.getusersitepackages()
+
+        # Check if the package is installed in the user-specific directory
+        return package_location.startswith(user_site_packages)

From d4adacafcc93880659f29ca82d6032c5cd68cbda Mon Sep 17 00:00:00 2001
From: dachengx <dx2227@columbia.edu>
Date: Sun, 15 Sep 2024 14:54:08 -0500
Subject: [PATCH 2/2] Minor change

---
 utilix/tarball.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utilix/tarball.py b/utilix/tarball.py
index 9d14c4b..30b52e4 100644
--- a/utilix/tarball.py
+++ b/utilix/tarball.py
@@ -18,7 +18,7 @@ def filter_tarinfo(tarinfo, git_ignored_files):
 
 class Tarball:
     def __init__(self, destination, package_name):
-        """Class to create a tarball of package.
+        """Class to tarball the editable user-installed from git repo.
 
         :param destination: the destination folder of the tarball
         :param package_name: the name of package