From fa4381e35c4e4c6fbf4eabeaa0c31a252a59d726 Mon Sep 17 00:00:00 2001
From: Jana Lemke <dev@jana-lemke.de>
Date: Wed, 24 May 2023 11:11:12 +0200
Subject: [PATCH] feat: Add current semester to semesters if not specified

Fixes: #1
---
 src/courses.rs | 5 +++++
 src/main.rs    | 5 ++++-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/courses.rs b/src/courses.rs
index 0b86321..8bc0e99 100644
--- a/src/courses.rs
+++ b/src/courses.rs
@@ -66,6 +66,11 @@ impl Course {
         &self.semesters
     }
 
+    /// Get a reference to the course's semesters.
+    pub fn semesters_mut(&mut self) -> &mut Vec<String> {
+        &mut self.semesters
+    }
+
     /// Check if course is part of current semester
     pub fn is_current(&self) -> bool {
         self.semesters()
diff --git a/src/main.rs b/src/main.rs
index dfb0dd3..e55427c 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -66,7 +66,10 @@ fn main() -> Result<()> {
     Ok(())
 }
 
-fn add(course: Course, debug: bool) -> Result<()> {
+fn add(mut course: Course, debug: bool) -> Result<()> {
+    if course.semesters().is_empty() {
+        course.semesters_mut().push(util::get_current_semester())
+    }
     let toml = toml::to_string(&course)?;
     println!("{}", toml);
     let target_dir = [