From fc9de6126996556c4cf672e29f0d1905466802dd Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Mon, 15 Apr 2019 22:40:20 -0700
Subject: [PATCH] Added copy_source()

---
 .bin/Scripts/build-ufd        |  8 ++++++++
 .bin/Scripts/functions/ufd.py | 11 +++++++++++
 2 files changed, 19 insertions(+)

diff --git a/.bin/Scripts/build-ufd b/.bin/Scripts/build-ufd
index 7571eae3..fbe1bdff 100755
--- a/.bin/Scripts/build-ufd
+++ b/.bin/Scripts/build-ufd
@@ -65,6 +65,14 @@ if __name__ == '__main__':
       )
 
     # Copy sources
+    for s_label, s_path in sources.items():
+      try_and_print(
+        message='Copying {} files...'.format(s_label),
+        function=copy_source,
+        source=s_path,
+        items=ITEMS[s_label],
+        overwrite=True,
+        )
 
     # Update boot entries
 
diff --git a/.bin/Scripts/functions/ufd.py b/.bin/Scripts/functions/ufd.py
index 96ce2d0a..344c273d 100644
--- a/.bin/Scripts/functions/ufd.py
+++ b/.bin/Scripts/functions/ufd.py
@@ -58,6 +58,17 @@ def confirm_selections(args):
   print_standard(' ')
 
 
+def copy_source(source, items, overwrite=False):
+  """Mount source and copy items to /mnt/UFD."""
+  os.makedirs('/mnt/Source', exist_ok=True)
+  mount(source, '/mnt/Source')
+  for i_source, i_dest in items:
+    i_source = '/mnt/Source{}'.format(i_source)
+    i_dest = '/mnt/UFD{}'.format(i_dest)
+    recursive_copy(i_source, i_dest, overwrite=overwrite)
+  unmount('/mnt/Source')
+
+
 def find_path(path):
   """Find path case-insensitively, returns pathlib.Path obj."""
   path_obj = pathlib.Path(path).resolve()