allow rewriting repo and owner (#110)

Co-authored-by: Leonardo Cavallucci <[email protected]>

Author: non7top

changes/00.feature

@@ -0,0 +1 @@
+Allow rewriting repo and owner

giturlparse/result.py

@@ -38,6 +38,12 @@ def valid(self):
     ##
     # Alias properties
     ##
+    def _update_url(self):
+        protocol = getattr(self, "protocol", None)
+        if protocol:
+            self.url = self.format(protocol)
+            self._parsed["url"] = self.format(protocol)
+
     @property
     def host(self):
         return self.domain
@@ -50,6 +56,21 @@ def resource(self):
     def name(self):
         return self.repo
 
+    @name.setter
+    def name(self, new_name):
+        self.repo = new_name
+        self._parsed["repo"] = new_name
+        self._update_url()
+
+    @property
+    def owner(self):
+        return self._parsed["owner"]
+
+    @owner.setter
+    def owner(self, new_owner):
+        self._parsed["owner"] = new_owner
+        self._update_url()
+
     @property
     def user(self):
         if hasattr(self, "_user"):

giturlparse/tests/test_rewrite.py

@@ -71,6 +71,20 @@
 )
 
 
+REWRITE_COMPONENTS = {
+    "name": {
+        "source": "[email protected]:Org/Repo.git",
+        "value": "New-repo",
+        "expected": "[email protected]:Org/New-repo.git",
+    },
+    "owner": {
+        "source": "[email protected]:Org/Repo.git",
+        "value": "New-Org",
+        "expected": "[email protected]:New-Org/Repo.git",
+    },
+}
+
+
 class UrlRewriteTestCase(unittest.TestCase):
     def _test_rewrite(self, source, protocol, expected):
         parsed = parse(source)
@@ -81,6 +95,16 @@ def test_rewrites(self):
         for data in REWRITE_URLS:
             self._test_rewrite(*data)
 
+    def _test_rewrite_components(self, field, data):
+        parsed = parse(data["source"])
+        setattr(parsed, field, data["value"])
+        self.assertTrue(parsed.valid, "Invalid Url: %s" % data["source"])
+        return self.assertEqual(parsed.url, data["expected"])
+
+    def test_rewrite_components(self):
+        for field, data in REWRITE_COMPONENTS.items():
+            self._test_rewrite_components(field, data)
+
 
 # Test Suite
 suite = unittest.TestLoader().loadTestsFromTestCase(UrlRewriteTestCase)