From 25fc538888f16086ac19fe4f9ecc35b516e31d3a Mon Sep 17 00:00:00 2001
From: Nicholas Tillman <ntillman@barracuda.com>
Date: Fri, 20 Jul 2018 14:25:07 -0700
Subject: [PATCH] Add support for building with Git LFS.

---
 src/tito/builder/__init__.py |  1 +
 src/tito/builder/main.py     | 37 ++++++++++++++++++++++++++++++++++++
 2 files changed, 38 insertions(+)

diff --git a/src/tito/builder/__init__.py b/src/tito/builder/__init__.py
index 4d89864..b092c1d 100644
--- a/src/tito/builder/__init__.py
+++ b/src/tito/builder/__init__.py
@@ -12,6 +12,7 @@ from tito.builder.main import \
     MockBuilder, \
     BrewDownloadBuilder, \
     GitAnnexBuilder, \
+    GitLfsBuilder, \
     MeadBuilder
 
 from tito.builder.fetch import FetchBuilder
diff --git a/src/tito/builder/main.py b/src/tito/builder/main.py
index 7c3fe7c..daa2d83 100644
--- a/src/tito/builder/main.py
+++ b/src/tito/builder/main.py
@@ -1282,6 +1282,43 @@ class GitAnnexBuilder(NoTgzBuilder):
         return compare_version(version, '5.20131213') >= 0
 
 
+class GitLfsBuilder(NoTgzBuilder):
+    """
+    Builder class for Git LFS support.
+    """
+
+    def _setup_sources(self):
+        super(GitLfsBuilder, self)._setup_sources()
+
+        self.old_cwd = os.getcwd()
+        os.chdir(os.path.join(self.old_cwd, self.relative_project_dir))
+        debug("learning %s" % self.relative_project_dir)
+        (status, output) = getstatusoutput("which git-lfs")
+        if status != 0:
+            msg = "Please run '%s' as root." % self.package_manager.install(["git-lfs"])
+            error_out('%s' % msg)
+
+        gitlfs_files = run_command("git-lfs ls-files").splitlines()
+        run_command("git-lfs fetch")
+        debug("  Lfs files: %s" % gitlfs_files)
+        for gfile in gitlfs_files:
+            m = re.search('(?<=' + self.relative_project_dir + ')(.*)', gfile)
+            if m:
+                filterfile = m.groups(0)
+                mstr = filterfile[0]
+                debug("Copying file %s " % mstr)
+                debug("To %s " % self.rpmbuild_gitcopy)
+                os.remove(os.path.join(self.rpmbuild_gitcopy, mstr))
+                shutil.copy(mstr, self.rpmbuild_gitcopy)
+
+        os.chdir(self.old_cwd)
+
+    def cleanup(self):
+        if hasattr(self, 'old_cwd'):
+            os.chdir(self.old_cwd)
+        super(GitLfsBuilder, self).cleanup()
+
+
 def package_manager():
     if os.path.isfile("/usr/bin/dnf"):
         return Dnf()