indexof

SteffenDE · SteffenDE · commit b2bc5470231a · 2026-07-20T19:05:44.000+02:00
diff --git a/src/toolbar.ts b/src/toolbar.ts
@@ -18,7 +18,7 @@ export function injectToolbarHtml(
   if (config.toolbar === false) return html;
   if (toolbarAlreadyInjected(html)) return html;
 
-  const closingHeadIndex = html.toLowerCase().lastIndexOf('</head>');
+  const closingHeadIndex = html.toLowerCase().indexOf('</head>');
   if (closingHeadIndex === -1) return html;
 
   return `${html.slice(0, closingHeadIndex)}${toolbarHtml(config, getLocalPort)}${html.slice(closingHeadIndex)}`;
diff --git a/test/toolbar.test.ts b/test/toolbar.test.ts
@@ -0,0 +1,23 @@
+import { describe, expect, it } from 'vitest';
+import { injectToolbarHtml } from '../src/toolbar';
+
+describe('toolbar HTML injection', () => {
+  it('should inject before the first closing head tag', () => {
+    const html =
+      '<html><head><title>App</title></head><body><script>const marker = "</head>";</script></body></html>';
+
+    const injected = injectToolbarHtml(
+      html,
+      { clientUrl: 'http://localhost:4000' },
+      () => 5173,
+    );
+
+    const toolbarIndex = injected.indexOf('/tc/toolbar.js');
+    const firstClosingHeadIndex = injected.indexOf('</head>');
+    const scriptMarkerIndex = injected.indexOf('const marker = "</head>";');
+
+    expect(toolbarIndex).toBeGreaterThan(-1);
+    expect(toolbarIndex).toBeLessThan(firstClosingHeadIndex);
+    expect(firstClosingHeadIndex).toBeLessThan(scriptMarkerIndex);
+  });
+});
