From b01918538088e4a2dc89fa0d4456026ca53b0060 Mon Sep 17 00:00:00 2001
From: IamTheFij <ian@iamthefij.com>
Date: Sun, 21 Jul 2024 09:42:46 -0700
Subject: [PATCH] Use branch zip for downloading forks

---
 unhacs/packages.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/unhacs/packages.py b/unhacs/packages.py
index 28f530e..c297950 100644
--- a/unhacs/packages.py
+++ b/unhacs/packages.py
@@ -13,9 +13,10 @@ from zipfile import ZipFile
 import requests
 import yaml
 
+from unhacs.git import get_branch_zip
 from unhacs.git import get_latest_sha
-from unhacs.git import get_tag_zip
 from unhacs.git import get_repo_tags
+from unhacs.git import get_tag_zip
 
 DEFAULT_HASS_CONFIG_PATH: Path = Path(".")
 DEFAULT_PACKAGE_FILE = Path("unhacs.yaml")
@@ -265,8 +266,10 @@ class Package:
 
     def install_fork_component(self, hass_config_path: Path):
         """Installs the integration from hass fork."""
+        # TODO: Replace asserts with errors
         assert self.fork_component
-        zipball_url = get_tag_zip(self.url, self.version)
+        assert self.branch_name
+        zipball_url = get_branch_zip(self.url, self.branch_name)
         response = requests.get(zipball_url)
         response.raise_for_status()