publish script improvements

Author: Taniya Mathur

lib/idp_common_pkg/tests/unit/test_publish.py

@@ -1044,6 +1044,7 @@ def test_run_minimal_success_flow(self):
             patch.object(publisher, "check_parameters") as mock_check_params,
             patch.object(publisher, "setup_environment") as mock_setup_env,
             patch.object(publisher, "check_prerequisites") as mock_check_prereq,
+            patch.object(publisher, "ensure_aws_sam_directory") as mock_ensure_sam,
             patch.object(publisher, "setup_artifacts_bucket") as mock_setup_bucket,
             patch.object(publisher, "clean_temp_files"),
             patch.object(publisher, "clean_lib"),
@@ -1072,6 +1073,7 @@ def test_run_minimal_success_flow(self):
         mock_check_params.assert_called_once()
         mock_setup_env.assert_called_once()
         mock_check_prereq.assert_called_once()
+        mock_ensure_sam.assert_called_once()
         mock_setup_bucket.assert_called_once()
         mock_build_patterns.assert_called_once()
         mock_build_options.assert_called_once()
@@ -1121,6 +1123,7 @@ def test_run_pattern_build_failure(self):
             patch.object(publisher, "check_parameters"),
             patch.object(publisher, "setup_environment"),
             patch.object(publisher, "check_prerequisites"),
+            patch.object(publisher, "ensure_aws_sam_directory"),
             patch.object(publisher, "setup_artifacts_bucket"),
             patch.object(publisher, "clean_temp_files"),
             patch.object(publisher, "clean_lib"),

publish.py

@@ -35,7 +35,6 @@
     TextColumn,
     TimeElapsedColumn,
 )
-from rich.table import Table
 
 
 class IDPPublisher:
@@ -269,6 +268,22 @@ def check_prerequisites(self):
             )
             sys.exit(1)
 
+    def ensure_aws_sam_directory(self):
+        """Ensure .aws-sam directory exists"""
+        aws_sam_dir = ".aws-sam"
+
+        if not os.path.exists(aws_sam_dir):
+            self.console.print(
+                "[yellow].aws-sam directory not found. Creating it...[/yellow]"
+            )
+            os.makedirs(aws_sam_dir, exist_ok=True)
+            self.console.print(
+                "[green]✅ Successfully created .aws-sam directory[/green]"
+            )
+        else:
+            if self.verbose:
+                self.console.print("[dim].aws-sam directory already exists[/dim]")
+
     def version_compare(self, version1, version2):
         """Compare two version strings. Returns -1 if v1 < v2, 0 if equal, 1 if v1 > v2"""
 
@@ -480,6 +495,17 @@ def get_stored_checksum(self, directory):
 
     def needs_rebuild(self, *paths):
         """Check if any of the paths have changed since last build"""
+        # Special case for ./lib directory - use .lib_checksum in root and get_directory_checksum
+        if len(paths) == 1 and paths[0] == "./lib":
+            current_checksum = self.get_directory_checksum("./lib")
+            checksum_file = ".lib_checksum"
+            if os.path.exists(checksum_file):
+                with open(checksum_file, "r") as f:
+                    stored_checksum = f.read().strip()
+                return current_checksum != stored_checksum
+            return True
+
+        # For all other cases, use get_checksum
         current_checksum = self.get_checksum(*paths)
 
         # For single directory, check its stored checksum
@@ -1186,30 +1212,7 @@ def print_outputs(self):
         encoded_template_url = quote(template_url, safe=":/?#[]@!$&'()*+,;=")
         launch_url = f"https://{self.region}.console.aws.amazon.com/cloudformation/home?region={self.region}#/stacks/create/review?templateURL={encoded_template_url}&stackName=IDP"
 
-        # First, display URLs in plain text to avoid Rich formatting issues
-        self.console.print("\n[bold green]Deployment Outputs[/bold green]")
-        self.console.print("[cyan]Template URL (use to update existing stack):[/cyan]")
-        self.console.print(f"{template_url}")
-        self.console.print(
-            "\n[cyan]1-Click Launch URL (use to launch new stack):[/cyan]"
-        )
-        self.console.print(f"{launch_url}")
-
-        # Also display in a table with clickable links
-        table = Table(title="[bold green]Quick Links[/bold green]", show_lines=True)
-        table.add_column("Link", style="cyan", no_wrap=False, width=60)
-        table.add_column("Description", style="yellow", width=40)
-
-        # Create clickable links using Rich's link syntax
-        template_link = f"[link={template_url}]Template URL[/link]"
-        launch_link = f"[link={launch_url}]1-Click Launch[/link]"
-
-        table.add_row(template_link, "Go to cloudformation and update existing stack")
-        table.add_row(launch_link, "Use to launch new stack")
-
-        self.console.print(table)
-
-        # Additional information for troubleshooting
+        # Display deployment information first
         self.console.print("\n[bold cyan]Deployment Information:[/bold cyan]")
         self.console.print(f"  • Region: [yellow]{self.region}[/yellow]")
         self.console.print(f"  • Bucket: [yellow]{self.bucket}[/yellow]")
@@ -1220,6 +1223,15 @@ def print_outputs(self):
             f"  • Public Access: [yellow]{'Yes' if self.public else 'No'}[/yellow]"
         )
 
+        # Then display URLs
+        self.console.print("\n[bold green]Deployment Outputs[/bold green]")
+        self.console.print("[cyan]Template URL (use to update existing stack):[/cyan]")
+        self.console.print(f"{template_url}")
+        self.console.print(
+            "\n[cyan]1-Click Launch URL (use to launch new stack):[/cyan]"
+        )
+        self.console.print(f"{launch_url}")
+
     def run(self, args):
         """Main execution method"""
         try:
@@ -1232,6 +1244,9 @@ def run(self, args):
             # Check prerequisites
             self.check_prerequisites()
 
+            # Ensure .aws-sam directory exists
+            self.ensure_aws_sam_directory()
+
             # Set up S3 bucket
             self.setup_artifacts_bucket()