From 968f211df6ea379b168c6782d28781e9e7e61fd7 Mon Sep 17 00:00:00 2001
From: Theo Chatzimichos <tampakrap@opensuse.org>
Date: Nov 12 2017 16:39:14 +0000
Subject: bin/get_formulas.sh: add argument to support fetching our gitlab forks


---

diff --git a/bin/get_formulas.py b/bin/get_formulas.py
index add9bd3..905f4cd 100755
--- a/bin/get_formulas.py
+++ b/bin/get_formulas.py
@@ -20,6 +20,8 @@ def clone_or_pull(DEST, SYMLINK=False):
             subprocess.Popen(['git', 'pull'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
         else:
             subprocess.Popen(['git', 'clone', url, FULL_PATH], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        subprocess.Popen(['git', 'remote', 'add', 'opensuse', opensuse_fork_url], stdout=subprocess.PIPE, stderr=subprocess.PIPE, cwd=FULL_PATH)
+        subprocess.Popen(['git', 'fetch', 'opensuse'], stdout=subprocess.PIPE, stderr=subprocess.PIPE, cwd=FULL_PATH)
 
     def use_pygit2_to_clone_or_pull_repo():
         import pygit2
@@ -34,18 +36,31 @@ def clone_or_pull(DEST, SYMLINK=False):
         namespace = data.get('namespace', 'saltstack-formulas')
         prefix = data.get('prefix', '')
         url = 'https://github.com/%s/%s%s-formula' % (namespace, prefix, formula)
+        opensuse_fork_url = 'https://gitlab.infra.opensuse.org/saltstack-formulas/%s-formula' % formula
         FULL_PATH = '%s/%s-formula' % (DEST, formula)
         use_git_to_clone_or_pull_repo()
         if SYMLINK:
             os.symlink('%s/%s' % (FULL_PATH, formula), '/srv/salt/%s' % formula)
 
 
+def enable_remote(REMOTE, DEST):
+    def use_git_to_enable_remote():
+        # pygit2 is not available for python3 in Leap, use plain git instead
+
+        import subprocess
+
+        for formula in FORMULAS.keys():
+            FULL_PATH = '%s/%s-formula' % (DEST, formula)
+            subprocess.Popen(['git', 'checkout', '-B', 'master', '%s/master' % REMOTE], stdout=subprocess.PIPE, stderr=subprocess.PIPE, cwd=FULL_PATH)
+
+
 with open('FORMULAS.yaml', 'r') as f:
     FORMULAS = yaml.load(f)
 
-parser = argparse.ArgumentParser(description='Loads the formulas from FORMULAS.yaml and optionally clones them in a specified destination. Optionally it can also create a symlink from the cloned path to /srv/salt, useful for the CI worker.')
-parser.add_argument('-c', '--clone', action='store_true', help='Clone the formulas to the destination specified with "--destination". If the repository is already cloned, then it will be pulled/fetched.')
+parser = argparse.ArgumentParser(description='Loads the formulas from FORMULAS.yaml and optionally clones them in a specified destination. Optionally it can also create a symlink from the cloned path to /srv/salt, useful for the CI worker. The internal gitlab fork will also be added as secondary remote.')
+parser.add_argument('-c', '--clone', action='store_true', help='Clone the formulas to the destination specified with "--destination". The gitlab fork will also be added as remote. If the repository is already cloned, then both remotes will be pulled/fetched.')
 parser.add_argument('-s', '--symlink', action='store_true', help='Creates symlink from the specified destination to /srv/salt.')
+parser.add_argument('-r', '--remote', nargs=1, help='Enable the specified remote. Available remotes: origin, opensuse. Default: origin')
 requiredArgs = parser.add_argument_group('required arguments')
 requiredArgs.add_argument('-d', '--destination', nargs=1, required=True, help='Destination absolute path of the cloned (or to-be-cloned) repositories of the formulas.')
 args = parser.parse_args()
@@ -54,5 +69,12 @@ if not os.path.isabs(args.destination[0]):
     parser.print_help()
     sys.exit(1)
 
+if args.remote and args.remote[0] not in ['origin', 'opensuse']:
+    parser.print_help()
+    sys.exit(1)
+
 if args.clone:
     clone_or_pull(args.destination[0], args.symlink)
+
+if args.remote:
+    enable_remote(args.remote[0], args.destination[0])