fix: Remote parsing in --generate (#268)

Closes #266

Co-authored-by: Dylan Anthony <[email protected]>

Author: dbanty

src/config.rs

@@ -135,8 +135,14 @@ pub(crate) fn generate() -> Result<()> {
     let release_steps = match git::get_first_remote() {
         Some(remote) if remote.contains("github.com") => {
             let parts = remote.split('/').collect::<Vec<_>>();
-            let owner = parts[parts.len() - 2].to_string();
-            let repo = parts[parts.len() - 1].to_string();
+            let owner = parts[parts.len() - 2];
+            let owner = owner
+                .strip_prefix("[email protected]:")
+                .unwrap_or(owner)
+                .to_string();
+
+            let repo = parts[parts.len() - 1];
+            let repo = repo.strip_suffix(".git").unwrap_or(repo).to_string();
             github = Some(GitHub { owner, repo });
             vec![
                 Step::Command {

tests/generate.rs

@@ -108,14 +108,16 @@ fn generate_packages_changelog(#[case] has_changelog: bool, #[case] target_file:
 }
 
 /// Run `--generate` on a repo with a GitHub remote.
-#[test]
-fn generate_github() {
+#[rstest]
+#[case::ssh("[email protected]:knope-dev/knope.git")]
+#[case::https("https://github.com/knope-dev/knope.git")]
+fn generate_github(#[case] remote: &str) {
     // Arrange
     let temp_dir = tempfile::tempdir().unwrap();
     let temp_path = temp_dir.path();
     let source_path = Path::new("tests/generate/github");
     init(temp_path);
-    add_remote(temp_path, "github.com/knope-dev/knope");
+    add_remote(temp_path, remote);
 
     // Act
     let assert = Command::new(cargo_bin!("knope"))