From 3c2566fc11ed3c575e0379f42deb5935c1d8f779 Mon Sep 17 00:00:00 2001
From: Nathan Sobo <nathan@zed.dev>
Date: Tue, 6 Sep 2022 08:56:49 -0600
Subject: [PATCH] Copy WebRTC.framework when building livekit crate

We determine the location of the target executable directory in a somewhat hacky way, but it seems reasonably stable.

Co-Authored-By: Antonio Scandurra <me@as-cii.com>
---
 crates/capture/build.rs  |  7 +---
 crates/live_kit/build.rs | 88 +++++++++++++++++++++++++++-------------
 2 files changed, 61 insertions(+), 34 deletions(-)

diff --git a/crates/capture/build.rs b/crates/capture/build.rs
index e2ffda4c53..c637f2ce9b 100644
--- a/crates/capture/build.rs
+++ b/crates/capture/build.rs
@@ -1,12 +1,7 @@
 use std::{env, path::PathBuf, process::Command};
 
 fn main() {
-    println!(
-        "cargo:rustc-link-search=framework={}",
-        "crates/live_kit/LiveKitBridge/.build/arm64-apple-macosx/debug"
-    );
-
-    // Find frameworks as a sibling of the executable at runtime
+    // Find WebRTC.framework as a sibling of the executable when running outside of an application bundle
     println!("cargo:rustc-link-arg=-Wl,-rpath,@executable_path");
 
     println!("cargo:rustc-link-lib=framework=ScreenCaptureKit");
diff --git a/crates/live_kit/build.rs b/crates/live_kit/build.rs
index dab126d12c..ca9c2d5cb9 100644
--- a/crates/live_kit/build.rs
+++ b/crates/live_kit/build.rs
@@ -29,53 +29,70 @@ pub struct SwiftTarget {
 const MACOS_TARGET_VERSION: &str = "12";
 
 fn main() {
-    build_bridge();
-    link_swift_stdlib();
+    let swift_target = get_swift_target();
+
+    build_bridge(&swift_target);
+    link_swift_stdlib(&swift_target);
+    link_webrtc_framework(&swift_target);
 }
 
-fn build_bridge() {
-    let profile = env::var("PROFILE").unwrap();
-    let package_name = "LiveKitBridge";
-    let package_root = env::current_dir().unwrap().join(package_name);
+fn build_bridge(swift_target: &SwiftTarget) {
+    let swift_package_root = swift_package_root();
     if !Command::new("swift")
-        .args(&["build", "-c", &profile])
-        .current_dir(&package_root)
+        .args(&["build", "-c", &env::var("PROFILE").unwrap()])
+        .current_dir(&swift_package_root)
         .status()
         .unwrap()
         .success()
     {
         panic!(
             "Failed to compile swift package in {}",
-            package_root.display()
+            swift_package_root.display()
         );
     }
 
-    let swift_target_info = get_swift_target();
-    let swift_out_dir_path = format!(
-        "{}/.build/{}/{}",
-        package_root.display(),
-        swift_target_info.target.unversioned_triple,
-        profile
-    );
-
-    println!("cargo:rustc-link-search=native={}", swift_out_dir_path);
     println!(
-        "cargo:rustc-link-search=framework={}",
-        "/Users/nathan/src/zed/crates/live_kit/frameworks"
+        "cargo:rustc-link-search=native={}",
+        swift_target.out_dir_path().display()
     );
-    println!("cargo:rustc-link-lib=static={}", package_name);
-    println!("cargo:rustc-link-lib=framework=WebRTC");
+    println!("cargo:rustc-link-lib=static={}", SWIFT_PACKAGE_NAME);
 }
 
-fn link_swift_stdlib() {
-    let target = get_swift_target();
-    if target.target.libraries_require_rpath {
+fn link_swift_stdlib(swift_target: &SwiftTarget) {
+    if swift_target.target.libraries_require_rpath {
         panic!("Libraries require RPath! Change minimum MacOS value to fix.")
     }
 
-    target.paths.runtime_library_paths.iter().for_each(|path| {
-        println!("cargo:rustc-link-search=native={}", path);
-    });
+    swift_target
+        .paths
+        .runtime_library_paths
+        .iter()
+        .for_each(|path| {
+            println!("cargo:rustc-link-search=native={}", path);
+        });
+}
+
+fn link_webrtc_framework(swift_target: &SwiftTarget) {
+    let swift_out_dir_path = swift_target.out_dir_path();
+    println!("cargo:rustc-link-lib=framework=WebRTC");
+    println!(
+        "cargo:rustc-link-search=framework={}",
+        swift_out_dir_path.display()
+    );
+
+    let source_path = swift_out_dir_path.join("WebRTC.framework");
+    let target_path = PathBuf::from(env::var("OUT_DIR").unwrap()).join("../../../WebRTC.framework");
+    assert!(
+        Command::new("cp")
+            .arg("-r")
+            .args(&[&source_path, &target_path])
+            .status()
+            .unwrap()
+            .success(),
+        "could not copy WebRTC.framework from {:?} to {:?}",
+        source_path,
+        target_path
+    );
 }
 
 fn get_swift_target() -> SwiftTarget {
@@ -93,3 +110,18 @@ fn get_swift_target() -> SwiftTarget {
 
     serde_json::from_slice(&swift_target_info_str).unwrap()
 }
+
+const SWIFT_PACKAGE_NAME: &'static str = "LiveKitBridge";
+
+fn swift_package_root() -> PathBuf {
+    env::current_dir().unwrap().join(SWIFT_PACKAGE_NAME)
+}
+
+impl SwiftTarget {
+    fn out_dir_path(&self) -> PathBuf {
+        swift_package_root()
+            .join(".build")
+            .join(&self.target.unversioned_triple)
+            .join(env::var("PROFILE").unwrap())
+    }
+}