From fc48fbbde18aad4a3545ba39ca35e9d90de2875a Mon Sep 17 00:00:00 2001
From: Evan Pratten <ewpratten@gmail.com>
Date: Sun, 23 Jan 2022 14:08:18 -0500
Subject: [PATCH] Add profile setting

---
 src/commands/profile.rs | 29 +++++++++++++++++++++++++++--
 src/config.rs           | 13 +++++++++++++
 2 files changed, 40 insertions(+), 2 deletions(-)

diff --git a/src/commands/profile.rs b/src/commands/profile.rs
index 4630c1e..c2081e4 100644
--- a/src/commands/profile.rs
+++ b/src/commands/profile.rs
@@ -1,4 +1,29 @@
+use colored::Colorize;
+
+use crate::config::{get_device_profiles, get_profile_script};
 
 pub fn set_profile(tablet_name: &str, profile_name: &str) {
-    todo!();
-}
\ No newline at end of file
+    // Get all valid profiles for the tablet
+    let tablet_profiles = get_device_profiles(tablet_name).unwrap();
+
+    // Check if the profile is valid
+    if !tablet_profiles.contains(&profile_name.to_string()) {
+        println!(
+            "{} is not a valid profile for {}",
+            profile_name, tablet_name
+        );
+        return;
+    }
+
+    // Set the profile
+    println!(
+        "Setting tablet {} to use profile {}",
+        tablet_name, profile_name.green()
+    );
+    let profile_script = get_profile_script(tablet_name, profile_name).unwrap();
+    let _output = std::process::Command::new("bash")
+        .arg("-exc")
+        .arg(profile_script)
+        .spawn()
+        .unwrap();
+}
diff --git a/src/config.rs b/src/config.rs
index 7a9f78f..7bd8f81 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -84,3 +84,16 @@ pub fn get_device_profiles(config_name: &str) -> Result<Vec<String>, std::io::Er
             .collect::<Vec<_>>()
     });
 }
+
+/// Gets the path to the script for a profile for a tablet from its config name and profile name
+pub fn get_profile_script(
+    config_name: &str,
+    profile_name: &str,
+) -> Result<PathBuf, std::io::Error> {
+    let config_dir = get_config_dir()?;
+    return Ok(config_dir
+        .join("devices")
+        .join(config_name)
+        .join("profiles")
+        .join(format!("{}.sh", profile_name)));
+}