From 72e7bff7123e3639bf7da0866c633bdc123d1af4 Mon Sep 17 00:00:00 2001
From: Alexander Sulfrian <alexander@sulfrian.net>
Date: Fri, 7 Jun 2013 13:43:09 +0200
Subject: Client/Tools/VCS: add support for symlinks

---
 src/lib/Bcfg2/Client/Tools/VCS.py | 21 +++++++++++++++++----
 1 file changed, 17 insertions(+), 4 deletions(-)

(limited to 'src/lib')

diff --git a/src/lib/Bcfg2/Client/Tools/VCS.py b/src/lib/Bcfg2/Client/Tools/VCS.py
index 4a30d9d90..947cff11f 100644
--- a/src/lib/Bcfg2/Client/Tools/VCS.py
+++ b/src/lib/Bcfg2/Client/Tools/VCS.py
@@ -6,6 +6,7 @@
 #   * integrate properly with reports
 missing = []
 
+import errno
 import os
 import shutil
 import sys
@@ -102,10 +103,22 @@ class VCS(Bcfg2.Client.Tools.Tool):
             full_path = os.path.join(destname, fname)
             dulwich.file.ensure_dir_exists(os.path.dirname(full_path))
 
-            file = open(full_path, 'wb')
-            file.write(destr[sha].as_raw_string())
-            file.close()
-            os.chmod(full_path, mode)
+            if dulwich.objects.S_ISGITLINK(mode):
+                src_path = destr[sha].as_raw_string()
+                try:
+                    os.symlink(src_path, full_path)
+                except OSError:
+                    e = sys.exc_info()[1]
+                    if e.errno == errno.EEXIST:
+                        os.unlink(full_path)
+                        os.symlink(src_path, full_path)
+                    else:
+                        raise
+            else:
+                file = open(full_path, 'wb')
+                file.write(destr[sha].as_raw_string())
+                file.close()
+                os.chmod(full_path, mode)
 
         return True
         # FIXME: figure out how to write the git index properly
-- 
cgit v1.2.3-1-g7c22