@curiousvlxd/linkedin-badge-renderer 0.1.14 → 0.1.16

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@curiousvlxd/linkedin-badge-renderer",
-  "version": "0.1.14",
+  "version": "0.1.16",
   "type": "module",
   "bin": {
     "linkedin-badge": "src/cli.mjs"

package/src/renderer/formats/pdf.mjs

@@ -1,17 +1,23 @@
 import { withBrowser, createContext, warmUp, waitFonts } from "../playwright.mjs";
-import { buildLinkedinWrapperHtml, extractBadgeFromIframe, buildRenderHtml, measureRoot } from "../linkedin.mjs";
+import {
+  buildLinkedinWrapperHtml,
+  extractBadgeFromIframe,
+  buildRenderHtml,
+  measureRoot,
+} from "../linkedin.mjs";
 
 export async function renderBadgeArtifacts({ handle, theme, size, locale, orientation, pad }) {
   return await withBrowser(async (browser) => {
-    const context = await createContext(browser);
+    const context = await createContext(browser, theme);
     const page = await context.newPage();
 
     try {
       await warmUp(context);
 
-      await page.setContent(buildLinkedinWrapperHtml({ handle, theme, size, locale, orientation }), {
-        waitUntil: "domcontentloaded",
-      });
+      await page.setContent(
+        buildLinkedinWrapperHtml({ handle, theme, size, locale, orientation }),
+        { waitUntil: "domcontentloaded" }
+      );
 
       const iframeLocator = page.locator(".LI-profile-badge iframe");
       await iframeLocator.waitFor({ state: "attached", timeout: 30000 });
@@ -24,7 +30,20 @@ export async function renderBadgeArtifacts({ handle, theme, size, locale, orient
       if (!frame) throw new Error("LinkedIn badge iframe frame is null");
 
       await frame.waitForLoadState("load").catch(() => {});
-      await frame.waitForTimeout(700);
+
+      await frame
+        .waitForSelector(".profile-badge__content, .profile-badge__name, .profile-badge__cta-btn", {
+          timeout: 30000,
+        })
+        .catch(() => {});
+
+      await frame
+        .waitForFunction(() => {
+          const imgs = Array.from(document.images || []);
+          if (!imgs.length) return false;
+          return imgs.every((i) => i.complete && i.naturalWidth > 0);
+        }, { timeout: 30000 })
+        .catch(() => {});
 
       const { badgeInner, cssHrefs } = await extractBadgeFromIframe(frame);
       const renderHtml = buildRenderHtml({ badgeInner, cssHrefs, pad });

package/src/renderer/playwright.mjs

@@ -9,20 +9,14 @@ export async function withBrowser(fn) {
   }
 }
 
-export async function createContext(browser) {
+export async function createContext(browser, theme = "light") {
   return await browser.newContext({
     userAgent:
       "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/122.0.0.0 Safari/537.36",
-
     locale: "en-US",
-
-    viewport: {
-      width: 1920,
-      height: 1080,
-    },
-
+    viewport: { width: 1920, height: 1080 },
     deviceScaleFactor: 1.5,
-    colorScheme: "dark",
+    colorScheme: theme === "dark" ? "dark" : "light",
     reducedMotion: "no-preference",
   });
 }