From 6d8c7dcf0d30e7d0b883d845addf67e10d98763c Mon Sep 17 00:00:00 2001
From: Jason <jason@crabnebula.dev>
Date: Wed, 18 Oct 2023 23:32:34 +0800
Subject: [PATCH] feat(macos): add team_id option for apple notarization Port
 of tauri-apps/tauri#7775

Co-authored-by: Trey Smith <trey@fulcrum-acoustic.com>
Co-authored-by: Lucas Nogueira <lucas@tauri.app>
---
 .changes/mac-notarytool-team-id.md    |  5 ++
 crates/packager/src/codesign/macos.rs | 86 ++++++++++++---------------
 2 files changed, 43 insertions(+), 48 deletions(-)
 create mode 100644 .changes/mac-notarytool-team-id.md

diff --git a/.changes/mac-notarytool-team-id.md b/.changes/mac-notarytool-team-id.md
new file mode 100644
index 00000000..5882be9c
--- /dev/null
+++ b/.changes/mac-notarytool-team-id.md
@@ -0,0 +1,5 @@
+---
+"cargo-packager": minor
+---
+
+Read the `APPLE_TEAM_ID` environment variable for macOS notarization arguments.
\ No newline at end of file
diff --git a/crates/packager/src/codesign/macos.rs b/crates/packager/src/codesign/macos.rs
index 0d959e0b..31c5e679 100644
--- a/crates/packager/src/codesign/macos.rs
+++ b/crates/packager/src/codesign/macos.rs
@@ -338,13 +338,14 @@ fn staple_app(app_bundle_path: PathBuf) -> crate::Result<()> {
 #[derive(Debug)]
 pub enum NotarizeAuth {
     AppleId {
-        apple_id: String,
-        password: String,
+        apple_id: OsString,
+        password: OsString,
+        team_id: Option<OsString>,
     },
     ApiKey {
-        key: String,
+        key: OsString,
         key_path: PathBuf,
-        issuer: String,
+        issuer: OsString,
     },
 }
 
@@ -355,11 +356,20 @@ pub trait NotarytoolCmdExt {
 impl NotarytoolCmdExt for Command {
     fn notarytool_args(&mut self, auth: &NotarizeAuth) -> &mut Self {
         match auth {
-            NotarizeAuth::AppleId { apple_id, password } => self
-                .arg("--apple-id")
-                .arg(apple_id)
-                .arg("--password")
-                .arg(password),
+            NotarizeAuth::AppleId {
+                apple_id,
+                password,
+                team_id,
+            } => {
+                self.arg("--username")
+                    .arg(apple_id)
+                    .arg("--password")
+                    .arg(password);
+                if let Some(team_id) = team_id {
+                    self.arg("--team-id").arg(team_id);
+                }
+                self
+            }
             NotarizeAuth::ApiKey {
                 key,
                 key_path,
@@ -380,50 +390,28 @@ pub fn notarize_auth() -> crate::Result<NotarizeAuth> {
     match (
         std::env::var_os("APPLE_ID"),
         std::env::var_os("APPLE_PASSWORD"),
+        std::env::var_os("APPLE_TEAM_ID"),
     ) {
-        (Some(apple_id), Some(apple_password)) => {
-            let apple_id = apple_id
-                .to_str()
-                .expect("failed to convert APPLE_ID to string")
-                .to_string();
-            let password = apple_password
-                .to_str()
-                .expect("failed to convert APPLE_PASSWORD to string")
-                .to_string();
-            Ok(NotarizeAuth::AppleId { apple_id, password })
-        }
+        (Some(apple_id), Some(password), team_id) => Ok(NotarizeAuth::AppleId {
+            apple_id,
+            password,
+            team_id,
+        }),
         _ => {
             match (
                 std::env::var_os("APPLE_API_KEY"),
                 std::env::var_os("APPLE_API_ISSUER"),
                 std::env::var("APPLE_API_KEY_PATH"),
             ) {
-                (Some(api_key), Some(api_issuer), Ok(key_path)) => {
-                    let key = api_key
-                        .to_str()
-                        .expect("failed to convert APPLE_API_KEY to string")
-                        .to_string();
-                    let issuer = api_issuer
-                        .to_str()
-                        .expect("failed to convert APPLE_API_ISSUER to string")
-                        .to_string();
-                    Ok(NotarizeAuth::ApiKey {
-                        key,
-                        key_path: key_path.into(),
-                        issuer,
-                    })
-                }
-                (Some(api_key), Some(api_issuer), Err(_)) => {
-                    let key = api_key
-                        .to_str()
-                        .expect("failed to convert APPLE_API_KEY to string")
-                        .to_string();
-                    let issuer = api_issuer
-                        .to_str()
-                        .expect("failed to convert APPLE_API_ISSUER to string")
-                        .to_string();
-
-                    let api_key_file_name = format!("AuthKey_{key}.p8");
+                (Some(key), Some(issuer), Ok(key_path)) => Ok(NotarizeAuth::ApiKey {
+                    key,
+                    key_path: key_path.into(),
+                    issuer,
+                }),
+                (Some(key), Some(issuer), Err(_)) => {
+                    let mut api_key_file_name = OsString::from("AuthKey_");
+                    api_key_file_name.push(&key);
+                    api_key_file_name.push(".p8");
                     let mut key_path = None;
 
                     let mut search_paths = vec!["./private_keys".into()];
@@ -448,7 +436,9 @@ pub fn notarize_auth() -> crate::Result<NotarizeAuth> {
                         })
                     } else {
                         Err(Error::ApiKeyMissing {
-                            filename: api_key_file_name,
+                            filename: api_key_file_name
+                                .into_string()
+                                .expect("failed to convert api_key_file_name to string"),
                         })
                     }
                 }
@@ -458,7 +448,7 @@ pub fn notarize_auth() -> crate::Result<NotarizeAuth> {
     }
 }
 
-fn find_api_key(folder: PathBuf, file_name: &str) -> Option<PathBuf> {
+fn find_api_key(folder: PathBuf, file_name: &OsString) -> Option<PathBuf> {
     let path = folder.join(file_name);
     if path.exists() {
         Some(path)