test: Add unit tests for 'parse_git_url()'

src/lib.rs

@@ -37,6 +37,11 @@ impl GitView {
         self.populate_remote()?;
 
         // TODO: Figure out how to default to 'master' or 'main' if branch doesn't exist on remote
+        //
+        // Current theory on how to get it working:
+        // 1. Run "git checkout" to check if there's a remote branch for your current local one (there should be no output if remote branch doesn't exist)
+        // 2. Then run "git rev-parse --abbrev-ref <remote>/HEAD" and split on the first '/' to get the current default branch
+        //      - Although, I think that this command isn't foolproof, it might be the best option though without trying to use some command line parsers
 
         // Retrieve the remote reference
         let remote_ref = self.get_remote_reference()?;
@@ -203,6 +208,9 @@ impl GitView {
      *  - [user@]host.xz:path/to/repo.git/
      */
     fn parse_git_url(&self, git_url: &str) -> Result<(String, String, String), AppError> {
+        // rust-url cannot parse 'scp-like' urls -> https://github.com/servo/rust-url/issues/220
+        // Manually parse the url ourselves
+        if git_url.contains("://") {
             match Url::parse(git_url) {
                 Ok(url) => Ok((
                     url.scheme().to_string(),
@@ -210,6 +218,7 @@ impl GitView {
                         .map_or_else(|| "github.com", |host| host)
                         .to_string(),
                     url.path()
+                        .trim_start_matches('/')
                         .trim_end_matches('/')
                         .trim_end_matches(".git")
                         .to_string(),
@@ -219,6 +228,26 @@ impl GitView {
                     git_url
                 ))),
             }
+        } else {
+            match git_url.split_once(':') {
+                Some((domain, path)) => {
+                    let protocol = "https";
+                    let path = path.trim_end_matches('/').trim_end_matches(".git");
+                    let split_domain = match domain.split_once('@') {
+                        Some((_username, dom)) => {
+                            dom
+                        }
+                        None => domain,
+                    };
+
+                    Ok((protocol.to_string(), split_domain.to_string(), path.to_string()))
+                }
+                None => Err(AppError::InvalidGitUrl(format!(
+                    "Sorry, couldn't parse git url '{}'",
+                    git_url
+                ))),
+            }
+        }
     }
 }
 
@@ -238,6 +267,7 @@ fn stdout_to_string(bytes: &[u8]) -> Result<String, AppError> {
 #[cfg(test)]
 mod parse_git_url {
     use crate::{error::AppError, GitView};
+    use test_case::test_case;
 
     fn instantiate_handler() -> GitView {
         GitView::new(
@@ -248,32 +278,34 @@ mod parse_git_url {
         )
     }
 
-    #[test]
-    fn with_dot_git() -> Result<(), AppError> {
+    // http[s]://host.xz[:port]/path/to/repo.git/
+    #[test_case("https://github.com:8080/sgoudham/git-view.git" ; "with port")]
+    #[test_case("https://github.com/sgoudham/git-view.git" ; "normal")]
+    #[test_case("https://github.com/sgoudham/git-view.git/" ; "with trailing slash")]
+    fn https(git_url: &str) -> Result<(), AppError> {
         let handler = instantiate_handler();
 
-        let git_url_normal = "https://github.com/sgoudham/git-view.git";
-
-        let (protocol, domain, urlpath) = handler.parse_git_url(git_url_normal)?;
+        let (protocol, domain, urlpath) = handler.parse_git_url(git_url)?;
 
         assert_eq!(protocol, "https");
         assert_eq!(domain, "github.com");
-        assert_eq!(urlpath, "/sgoudham/git-view");
+        assert_eq!(urlpath, "sgoudham/git-view");
 
         Ok(())
     }
 
-    #[test]
-    fn with_dot_git_and_trailing_slash() -> Result<(), AppError> {
+    // [user@]host.xz:path/to/repo.git/
+    #[test_case("git@github.com:sgoudham/git-view.git" ; "with username")]
+    #[test_case("github.com:sgoudham/git-view.git" ; "normal")]
+    #[test_case("github.com:sgoudham/git-view.git/" ; "with trailing slash")]
+    fn ssh(git_url: &str) -> Result<(), AppError> {
         let handler = instantiate_handler();
 
-        let git_url_normal = "https://github.com/sgoudham/git-view.git/";
-
-        let (protocol, domain, urlpath) = handler.parse_git_url(git_url_normal)?;
+        let (protocol, domain, urlpath) = handler.parse_git_url(git_url)?;
 
         assert_eq!(protocol, "https");
         assert_eq!(domain, "github.com");
-        assert_eq!(urlpath, "/sgoudham/git-view");
+        assert_eq!(urlpath, "sgoudham/git-view");
 
         Ok(())
     }