Check user group when creating shared repos

1 files changed, 9 insertions, 4 deletions

diff --git a/src/git.rs b/src/git.rs
index c61bec1..dcdb71b 100644
--- a/src/git.rs
+++ b/src/git.rs
@@ -7,7 +7,7 @@ use std::{
     path::{Path, PathBuf},
     process::Command,
 };
-use user_info::get_username;
+use user_info::{get_user_groups, get_username};
 
 pub struct GitInitResult {
     pub path: PathBuf,
@@ -18,8 +18,13 @@ fn personal_git_dir() -> Result<PathBuf, ShackleError> {
     Ok(Path::new("git").join(username))
 }
 
-fn group_git_dir(group: &str) -> PathBuf {
-    Path::new("git").join(group)
+fn group_git_dir(group: &str) -> Result<PathBuf, ShackleError> {
+    let groups = get_user_groups();
+    if !groups.iter().any(|g| g == group) {
+        Err(ShackleError::InvalidGroup)
+    } else {
+        Ok(Path::new("git").join(group))
+    }
 }
 
 fn is_valid_personal_git_repo(path: &PathBuf) -> Result<bool, ShackleError> {
@@ -42,7 +47,7 @@ fn is_valid_personal_git_repo(path: &PathBuf) -> Result<bool, ShackleError> {
 
 pub fn init(repo_name: &str, group: &Option<String>) -> Result<GitInitResult, ShackleError> {
     fn init_group(repo_name: &str, group: &str) -> Result<GitInitResult, ShackleError> {
-        let path = group_git_dir(group).join(repo_name).with_extension("git");
+        let path = group_git_dir(group)?.join(repo_name).with_extension("git");
 
         Repository::init_opts(
             &path,