From bc095200f1afcb604f49f5fd411baf9bd45751c0 Mon Sep 17 00:00:00 2001
From: Alexander Clouter <alex@coremem.com>
Date: Wed, 22 Feb 2023 08:26:46 +0000
Subject: [PATCH] autorun support

disabled by default, and configurable via trailets
---
 nbgitpuller/pull.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/nbgitpuller/pull.py b/nbgitpuller/pull.py
index f5a7dab1..ae0d960a 100644
--- a/nbgitpuller/pull.py
+++ b/nbgitpuller/pull.py
@@ -1,4 +1,5 @@
 import os
+import re
 import subprocess
 import logging
 import time
@@ -7,6 +8,7 @@
 from traitlets import Integer, default
 from traitlets.config import Configurable
 from functools import partial
+from .config import NbGitPullerFeatures
 
 
 def execute_cmd(cmd, **kwargs):
@@ -80,6 +82,9 @@ def __init__(self, git_url, repo_dir, **kwargs):
         elif not self.branch_exists(self.branch_name):
             raise ValueError(f"Branch: {self.branch_name} -- not found in repo: {self.git_url}")
 
+        self._features = NbGitPullerFeatures(parent=kwargs.get("parent"))
+        self._autorun = any(( re.match(pattern, git_url) for pattern in self._features.autorun_allow ))
+
         self.repo_dir = repo_dir
         newargs = {k: v for k, v in kwargs.items() if v is not None}
         super(GitPuller, self).__init__(**newargs)
@@ -143,6 +148,20 @@ def pull(self):
         else:
             yield from self.update()
 
+    def autorun(self, operation):
+        """
+        Search for and execute the autorun script.
+        """
+        if not self._autorun:
+            return
+
+        script = next(( s for s in self._features.autorun_script if os.path.exists(os.path.join(self.repo_dir, s)) ), None)
+        if not script:
+            return
+
+        logging.info(f'Running "{script} {operation}')
+        yield from execute_cmd([ script, operation ], cwd=self.repo_dir, shell=True)
+
     def initialize_repo(self):
         """
         Clones repository
@@ -154,6 +173,7 @@ def initialize_repo(self):
         clone_args.extend(['--branch', self.branch_name])
         clone_args.extend(["--", self.git_url, self.repo_dir])
         yield from execute_cmd(clone_args)
+        self._autorun('init')
         logging.info('Repo {} initialized'.format(self.repo_dir))
 
     def reset_deleted_files(self):
@@ -343,6 +363,7 @@ def update(self):
         yield from self.ensure_lock()
         yield from self.merge()
 
+        self._autorun('update')
 
 def main():
     """