From 6aa609144d4b8dde995749ad2de4aaa8d69f16f4 Mon Sep 17 00:00:00 2001
From: YuKun Liu <41265098+mrxiaozhuox@users.noreply.github.com>
Date: Mon, 6 Jun 2022 08:46:16 +0800
Subject: [PATCH] feat: add `profile` flag

---
 src/builder.rs | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/builder.rs b/src/builder.rs
index 31bb96f46..57738b1bc 100644
--- a/src/builder.rs
+++ b/src/builder.rs
@@ -51,7 +51,8 @@ pub fn build(config: &CrateConfig) -> Result<()> {
 
     if config.custom_profile.is_some() {
         let custom_profile = config.custom_profile.as_ref().unwrap();
-        cmd.arg(format!("--profile {}", custom_profile));
+        cmd.arg("--profile");
+        cmd.arg(custom_profile);
     }
 
     match executable {
@@ -194,6 +195,12 @@ pub fn build_desktop(config: &CrateConfig, is_serve: bool) -> Result<()> {
         cmd.arg("--release");
     }
 
+    if config.custom_profile.is_some() {
+        let custom_profile = config.custom_profile.as_ref().unwrap();
+        cmd.arg("--profile");
+        cmd.arg(custom_profile);
+    }
+
     match &config.executable {
         crate::ExecutableType::Binary(name) => cmd.arg("--bin").arg(name),
         crate::ExecutableType::Lib(name) => cmd.arg("--lib").arg(name),