Update create_frameworks.sh

Author: shoumikhin

scripts/create_frameworks.sh

@@ -13,14 +13,15 @@ function usage() {
     echo ""
     echo "Usage: $0 --directory=<dir> --framework=<lib> [--output=<output>]"
     echo "  --directory: Directory containing the libs"
-    echo "  --framework: Framework to create in the format 'target:lib1,lib2:headers'"
+    echo "  --framework: Framework to create in the format 'target:lib1,lib2:headers:swiftmodule'"
     echo "               'target' is the name of the target library."
     echo "               'lib1,lib2' is a comma-separated list of input libraries."
     echo "               'headers' is an optional path to a directory with headers."
+    echo "               ':swiftmodule' is an optional module name to embed its .swiftmodule folder"
     echo "  --output: Optional output directory. Defaults to the current directory."
     echo ""
     echo "Example:"
-    echo "$0 --directory=ios-arm64 --directory=ios-arm64-simulator --framework=\"mylib:lib1.a,lib2.a:include\" --output=output/dir"
+    echo "$0 --directory=ios-arm64 --directory=ios-arm64-simulator --framework=\"mylib:lib1.a,lib2.a:include:MyModule\" --output=output/dir"
     exit 1
 }
 
@@ -58,6 +59,8 @@ create_xcframework() {
     libraries_list=$(echo "$1" | cut -d: -f2 | tr ',' '\n')
     local headers_directory
     headers_directory=$(echo "$1" | cut -d: -f3)
+    local swift_module
+    swift_module=$(echo "$1" | cut -d: -f4)
     local dir
     local libraries=()
     local merged_libs=()
@@ -117,8 +120,36 @@ create_xcframework() {
 
     echo -e "\nCreating XCFramework ${xcframework}"
 
+    # Create the new .xcframework.
     xcodebuild -create-xcframework "${libraries[@]}" -output "${xcframework}"
 
+    # Copy the .swiftmodule files into the .xcframework if applicable.
+    if [[ -n "$swift_module" ]]; then
+        echo -e "\nCopying Swift module ${swift_module}.swiftmodule into ${xcframework}"
+        for dir in "${directories[@]}"; do
+            local module_source_dir="${dir}/${swift_module}.swiftmodule"
+            if [ ! -d "$module_source_dir" ]; then
+                echo "Swiftmodule directory ${module_source_dir} does not exist"
+                exit 1
+            fi
+            local swiftmodule_file
+            swiftmodule_file=$(find "$module_source_dir" -maxdepth 1 -type f -name '*.swiftmodule' | head -n1)
+            if [[ -z "$swiftmodule_file" ]]; then
+                echo "No .swiftmodule file found in ${module_source_dir}"
+                exit 1
+            fi
+
+            local dir_suffix
+            dir_suffix=$(echo "$dir" | cut -d'/' -f1 | tr '[:upper:]' '[:lower:]' | sed 's/[\/\.~]/_/g')
+            for slice_path in "${xcframework}/${dir_suffix}-"*; do
+                if [ -d "${slice_path}/Headers" ]; then
+                    echo " - Copying ${swiftmodule_file##*/} to ${slice_path}/Headers/${swift_module}.swiftmodule"
+                    cp "${swiftmodule_file}" "${slice_path}/Headers/${swift_module}.swiftmodule"
+                fi
+            done
+        done
+    fi
+
     echo -e "\nDeleting intermediate libraries:"
     for merged_lib in "${merged_libs[@]}"; do
         if [[ -f "${merged_lib}" ]]; then