feat: Add logging for extractor archive size and improve CodeQL installation error handling

Author: GeekMasher

Dockerfile

@@ -31,4 +31,7 @@ RUN apt-get update && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
 
+# Install the CodeQL extension for GitHub CLI
+RUN gh extensions install github/gh-codeql
+
 ENTRYPOINT [ "codeql-extractor-action" ]

src/extractors.rs

@@ -86,6 +86,15 @@ pub async fn fetch_extractor(
         }
     };
 
+    // Get and log the size of the extractor archive
+    if let Ok(metadata) = std::fs::metadata(&extractor_archive) {
+        let size_bytes = metadata.len();
+        let size_mb = size_bytes as f64 / 1_048_576.0; // Convert to MB (1 MB = 1,048,576 bytes)
+        log::info!("Extractor archive size: {:.2} MB ({} bytes)", size_mb, size_bytes);
+    } else {
+        log::warn!("Unable to get size information for the extractor archive");
+    }
+
     if attest {
         log::info!("Attesting asset {extractor_tarball:?}");
 

src/main.rs

@@ -41,10 +41,18 @@ async fn main() -> Result<()> {
     if !codeql.is_installed().await {
         let codeql_version = action.codeql_version();
         log::info!("CodeQL not installed, installing `{codeql_version}`...");
-        codeql
-            .install(&octocrab, codeql_version)
-            .await
-            .context("Failed to install CodeQL")?;
+
+        if let Err(error) = codeql.install(&octocrab, codeql_version).await {
+            log::warn!("Failed to install CodeQL: {error:?}");
+            log::info!("Attempting to install CodeQL using GitHub CLI...");
+
+            tokio::process::Command::new("gh")
+                .args(&["codeql", "set-version", codeql_version.into()])
+                .status()
+                .await
+                .context("Failed to execute `gh codeql install` command")?;
+        }
+
         log::info!("CodeQL installed");
     } else {
         log::info!("CodeQL already installed");