Default to current folder when run without arguments

Fixes #12.
perlun · Jul 7, 2017 · f46f4ac · f46f4ac
1 parent 867d222
commit f46f4ac
Showing 1 changed file with 22 additions and 16 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -14,22 +14,11 @@ use std::process::exit;
 fn main() {
     let args: Vec<_> = env::args().collect();
 
-    if args.len() == 2 {
-        let ref repository_path = args[1];
-        let git_tag_parser = GitTagParser {
-            repository_path: repository_path.clone()
-        };
-
-        let version_tag_pairs = git_tag_parser.get_version_tag_pairs();
-
-        for (from_tag, to_tag) in version_tag_pairs.into_iter() {
-            let generator = ChangelogGenerator {
-                repository_path: repository_path.clone(),
-                from_revision: from_tag,
-                to_revision: to_tag
-            };
-            generator.generate_changelog();
-        }
+    if args.len() == 1 {
+        generate_changelog_for_folder(&String::from("."));
+    }
+    else if args.len() == 2 {
+        generate_changelog_for_folder(&args[1]);
     } else if args.len() == 4 {
         let generator = ChangelogGenerator {
             repository_path: args[1].clone(),
@@ -46,3 +35,20 @@ fn main() {
         exit(1);
     }
 }
+
+fn generate_changelog_for_folder(repository_path: &String) {
+    let git_tag_parser = GitTagParser {
+        repository_path: repository_path.clone()
+    };
+
+    let version_tag_pairs = git_tag_parser.get_version_tag_pairs();
+
+    for (from_tag, to_tag) in version_tag_pairs.into_iter() {
+        let generator = ChangelogGenerator {
+            repository_path: repository_path.clone(),
+            from_revision: from_tag,
+            to_revision: to_tag
+        };
+        generator.generate_changelog();
+    }
+}