@W-18129346 - get available permsets based on unused licenses

Author: vsbharath

cumulusci/tasks/preflight/licenses.py

@@ -1,13 +1,28 @@
 from cumulusci.tasks.salesforce import BaseSalesforceApiTask
 
 
-class GetAvailableLicenses(BaseSalesforceApiTask):
+class BaseUserLicenseAwareTask(BaseSalesforceApiTask):
+    def get_available_user_licenses(self):
+        """Fetch active user licenses with availability."""
+        query = """
+            SELECT Id, LicenseDefinitionKey, TotalLicenses, UsedLicenses 
+            FROM UserLicense 
+            WHERE Status = 'Active'
+        """
+        return {
+            lic["Id"]: lic
+            for lic in self.sf.query(query)["records"]
+            if lic["TotalLicenses"] > lic["UsedLicenses"]
+        }
+
+    def _log_list(self, title, items):
+        self.logger.info(f"{title} ({len(items)}):\n" + "\n".join(f"- {item}" for item in items))
+
+class GetAvailableLicenses(BaseUserLicenseAwareTask):
     def _run_task(self):
         self.return_values = [
             result["LicenseDefinitionKey"]
-            for result in self.sf.query("SELECT LicenseDefinitionKey FROM UserLicense")[
-                "records"
-            ]
+            for result in self.get_available_user_licenses().values()
         ]
         licenses = "\n".join(self.return_values)
         self.logger.info(f"Found licenses:\n{licenses}")
@@ -35,11 +50,16 @@ def _run_task(self):
         self.logger.info(f"Found permission licenses sets assigned:\n{permsets}")
 
 
-class GetAvailablePermissionSets(BaseSalesforceApiTask):
+class GetAvailablePermissionSets(BaseUserLicenseAwareTask):
     def _run_task(self):
-        self.return_values = [
-            result["Name"]
-            for result in self.sf.query_all("SELECT Name FROM PermissionSet")["records"]
+        license_data = self.get_available_user_licenses()
+        permsets = self.sf.query_all("SELECT LicenseId, Name FROM PermissionSet")["records"]
+        available_permsets = [
+            ps["Name"]
+            for ps in permsets
+            if not ps["LicenseId"] or ps["LicenseId"] in license_data
         ]
+
+        self.return_values = available_permsets
         permsets = "\n".join(self.return_values)
         self.logger.info(f"Found Permission Sets:\n{permsets}")