Ensure the correct source URL is stored with an install.

Fixes #259

Author: zooba

src/manage/indexutils.py

@@ -225,11 +225,11 @@ def find_to_install(self, tag, *, loose_company=False, prefer_prerelease=False):
         tag_list = [tag] if tag else []
         LOGGER.debug("Finding %s to install", tag_list)
         for i in self.find_all(tag_list, loose_company=loose_company, with_prerelease=prefer_prerelease):
-            return i
+            return {**i, "source": self.source_url}
         if not loose_company:
             return self.find_to_install(tag, loose_company=True, prefer_prerelease=prefer_prerelease)
         if not prefer_prerelease:
             for i in self.find_all(tag_list, loose_company=loose_company, with_prerelease=True):
-                return i
+                return {**i, "source": self.source_url}
         LOGGER.debug("No install found for %s", tag_list)
         raise LookupError(tag)

src/manage/install_command.py

@@ -609,8 +609,9 @@ def _install_one(cmd, source, install, *, target=None):
             install["shortcuts"] = shortcuts
 
         install["url"] = sanitise_url(install["url"])
-        if source != cmd.fallback_source:
-            install["source"] = sanitise_url(source)
+        # If there's a non-empty and non-default source, sanitise it
+        if install.get("source") and install["source"] != cmd.fallback_source:
+            install["source"] = sanitise_url(install["source"])
 
         LOGGER.debug("Write __install__.json to %s", dest)
         with open(dest / "__install__.json", "w", encoding="utf-8") as f: