@jshookmcp/jshook 0.1.5 → 0.1.6

package/dist/server/domains/workflow/handlers.impl.workflow-api.js

@@ -119,8 +119,7 @@ export class WorkflowHandlersApi extends WorkflowHandlersBase {
   }
 
   var results = {};
-  for (var i = 0; i < paths.length; i++) {
-    var path = paths[i];
+  async function probePath(path) {
     try {
       var opts = {method: method, headers: headers};
       if (bodyTemplate && (method === 'POST' || method === 'PUT' || method === 'PATCH')) {
@@ -137,11 +136,22 @@ export class WorkflowHandlersApi extends WorkflowHandlersBase {
           snippet = '[HTML/XML response suppressed]';
         }
       }
-      results[path] = {status: resp.status, contentType: ct.split(';')[0].trim(), snippet: snippet};
+      return [path, {status: resp.status, contentType: ct.split(';')[0].trim(), snippet: snippet}];
     } catch(e) {
-      results[path] = {status: -1, error: e.message};
+      return [path, {status: -1, error: e instanceof Error ? e.message : String(e)}];
     }
   }
+
+  var nextIndex = 0;
+  var maxConcurrency = Math.min(paths.length, 6);
+  await Promise.all(Array.from({ length: maxConcurrency }, async function() {
+    while (nextIndex < paths.length) {
+      var currentIndex = nextIndex++;
+      var currentPath = paths[currentIndex];
+      var entry = await probePath(currentPath);
+      results[entry[0]] = entry[1];
+    }
+  }));
   return {probed: paths.length, method: method, baseUrl: baseUrl, results: results};
 })()`;
         try {

package/dist/server/domains/workflow/handlers.impl.workflow-base.js

@@ -164,69 +164,69 @@ export class WorkflowHandlersBase {
     initBuiltinScripts() {
         this.scriptRegistry.set('auth_extract', {
             description: 'Extract auth tokens from localStorage and cookies',
-            code: `(function(){
-  var keys=['token','active_token','access_token','jwt','auth_token','userRole','id_token','refresh_token'];
-  var r={};
-  for(var i=0;i<keys.length;i++){var v=localStorage.getItem(keys[i]);if(v)r[keys[i]]=v;}
-  r._cookies=document.cookie;
-  return r;
+            code: `(function(){
+  var keys=['token','active_token','access_token','jwt','auth_token','userRole','id_token','refresh_token'];
+  var r={};
+  for(var i=0;i<keys.length;i++){var v=localStorage.getItem(keys[i]);if(v)r[keys[i]]=v;}
+  r._cookies=document.cookie;
+  return r;
 })()`,
         });
         this.scriptRegistry.set('bundle_search', {
             description: 'Fetch a remote JS bundle and search it with regex patterns. params: { url: string, patterns: string[] }',
-            code: `(async function(){
-  var p=typeof __params__!=='undefined'?__params__:{};
-  if(!p.url)return{error:'params.url required'};
-  var resp=await fetch(p.url);
-  var text=await resp.text();
-  var patterns=p.patterns||[];
-  var results={};
-  for(var i=0;i<patterns.length;i++){
-    var re=new RegExp(patterns[i],'g');
-    var matches=[];var m;
-    while((m=re.exec(text))!==null){
-      var s=Math.max(0,m.index-80),e=Math.min(text.length,m.index+m[0].length+80);
-      matches.push({match:m[0],ctx:text.slice(s,e)});
-      if(matches.length>=10)break;
-    }
-    results[patterns[i]]=matches;
-  }
-  return{size:text.length,results:results};
+            code: `(async function(){
+  var p=typeof __params__!=='undefined'?__params__:{};
+  if(!p.url)return{error:'params.url required'};
+  var resp=await fetch(p.url);
+  var text=await resp.text();
+  var patterns=p.patterns||[];
+  var results={};
+  for(var i=0;i<patterns.length;i++){
+    var re=new RegExp(patterns[i],'g');
+    var matches=[];var m;
+    while((m=re.exec(text))!==null){
+      var s=Math.max(0,m.index-80),e=Math.min(text.length,m.index+m[0].length+80);
+      matches.push({match:m[0],ctx:text.slice(s,e)});
+      if(matches.length>=10)break;
+    }
+    results[patterns[i]]=matches;
+  }
+  return{size:text.length,results:results};
 })()`,
         });
         this.scriptRegistry.set('react_fill_form', {
             description: 'Fill React controlled form inputs using native setter trick. params: { fields: { "selector": "value" } }',
-            code: `(function(){
-  var p=typeof __params__!=='undefined'?__params__:{};
-  var fields=p.fields||{};
-  var ns=Object.getOwnPropertyDescriptor(window.HTMLInputElement.prototype,'value').set;
-  var r={};
-  var entries=Object.entries(fields);
-  for(var i=0;i<entries.length;i++){
-    var sel=entries[i][0],val=entries[i][1];
-    var el=document.querySelector(sel);
-    if(!el){r[sel]='not found';continue;}
-    ns.call(el,val);
-    el.dispatchEvent(new Event('input',{bubbles:true}));
-    el.dispatchEvent(new Event('change',{bubbles:true}));
-    r[sel]='filled';
-  }
-  return r;
+            code: `(function(){
+  var p=typeof __params__!=='undefined'?__params__:{};
+  var fields=p.fields||{};
+  var ns=Object.getOwnPropertyDescriptor(window.HTMLInputElement.prototype,'value').set;
+  var r={};
+  var entries=Object.entries(fields);
+  for(var i=0;i<entries.length;i++){
+    var sel=entries[i][0],val=entries[i][1];
+    var el=document.querySelector(sel);
+    if(!el){r[sel]='not found';continue;}
+    ns.call(el,val);
+    el.dispatchEvent(new Event('input',{bubbles:true}));
+    el.dispatchEvent(new Event('change',{bubbles:true}));
+    r[sel]='filled';
+  }
+  return r;
 })()`,
         });
         this.scriptRegistry.set('dom_find_upgrade_buttons', {
             description: 'Scan the current page for upgrade/subscription/tier-related UI elements',
-            code: `(function(){
-  var kw=['upgrade','plus','pro','premium','subscribe','plan','tier','vip','membership'];
-  var r=[];
-  document.querySelectorAll('button,a,[role=button],[class*=upgrade],[class*=premium],[class*=plus]').forEach(function(el){
-    var t=(el.textContent||'').toLowerCase().trim();
-    var c=(el.className||'').toLowerCase();
-    if(kw.some(function(k){return t.includes(k)||c.includes(k);})){
-      r.push({tag:el.tagName,text:t.slice(0,120),cls:c.slice(0,100),href:el.href||null,id:el.id||null});
-    }
-  });
-  return r;
+            code: `(function(){
+  var kw=['upgrade','plus','pro','premium','subscribe','plan','tier','vip','membership'];
+  var r=[];
+  document.querySelectorAll('button,a,[role=button],[class*=upgrade],[class*=premium],[class*=plus]').forEach(function(el){
+    var t=(el.textContent||'').toLowerCase().trim();
+    var c=(el.className||'').toLowerCase();
+    if(kw.some(function(k){return t.includes(k)||c.includes(k);})){
+      r.push({tag:el.tagName,text:t.slice(0,120),cls:c.slice(0,100),href:el.href||null,id:el.id||null});
+    }
+  });
+  return r;
 })()`,
         });
     }

package/dist/server/registry/discovery.js

@@ -1,6 +1,6 @@
 import { readdir, stat } from 'node:fs/promises';
-import { join } from 'node:path';
-import { fileURLToPath, pathToFileURL } from 'node:url';
+import { dirname, join, relative, sep } from 'node:path';
+import { fileURLToPath } from 'node:url';
 import { logger } from '../../utils/logger.js';
 function isDomainManifest(value) {
     if (!value || typeof value !== 'object')
@@ -35,24 +35,29 @@ async function discoverManifestPaths() {
         logger.error('[discovery] Cannot read domains directory:', err);
         return [];
     }
-    const paths = [];
-    for (const entry of entries) {
-        if (!entry.isDirectory())
-            continue;
+    const directories = entries.filter(e => e.isDirectory());
+    const resolved = await Promise.all(directories.map(async (entry) => {
         for (const ext of ['manifest.js', 'manifest.ts']) {
             const manifestPath = join(domainsDir, entry.name, ext);
             try {
                 const s = await stat(manifestPath);
-                if (s.isFile()) {
-                    paths.push(manifestPath);
-                    break;
-                }
+                if (s.isFile())
+                    return manifestPath;
             }
             catch {
             }
         }
+        return null;
+    }));
+    return resolved.filter((value) => value !== null);
+}
+function toImportSpecifier(absPath) {
+    const currentDir = dirname(fileURLToPath(import.meta.url));
+    const relPath = relative(currentDir, absPath).split(sep).join('/');
+    if (relPath.startsWith('.')) {
+        return relPath;
     }
-    return paths;
+    return `./${relPath}`;
 }
 export async function discoverDomainManifests() {
     const files = await discoverManifestPaths();
@@ -61,7 +66,7 @@ export async function discoverDomainManifests() {
     const seenDepKeys = new Set();
     for (const absPath of files) {
         try {
-            const mod = await import(pathToFileURL(absPath).href);
+            const mod = await import(toImportSpecifier(absPath));
             const manifest = extractManifest(mod);
             if (!manifest) {
                 logger.warn('[discovery] Skipping ' + absPath + ': no valid DomainManifest export');

package/dist/server/registry/index.js

@@ -3,6 +3,8 @@ import { logger } from '../../utils/logger.js';
 let _manifests = null;
 let _registrations = null;
 let _initPromise = null;
+let _domainsView = null;
+let _toolNamesView = null;
 async function init() {
     if (_manifests !== null)
         return;
@@ -26,6 +28,8 @@ async function init() {
             }
         }
         _registrations = [...uniqueByToolName.values()];
+        _domainsView = new Set(_manifests.map(m => m.domain));
+        _toolNamesView = new Set(_registrations.map(r => r.tool.name));
     })();
     await _initPromise;
 }
@@ -49,10 +53,14 @@ export function getAllRegistrations() {
     return getRegistrations();
 }
 export function getAllDomains() {
-    return new Set(getManifests().map(m => m.domain));
+    if (!_domainsView)
+        throw new Error('[registry] Not initialised - call initRegistry() first.');
+    return _domainsView;
 }
 export function getAllToolNames() {
-    return new Set(getRegistrations().map(r => r.tool.name));
+    if (!_toolNamesView)
+        throw new Error('[registry] Not initialised - call initRegistry() first.');
+    return _toolNamesView;
 }
 export function buildToolGroups() {
     const groups = {};

package/dist/utils/TokenBudgetManager.d.ts

@@ -45,6 +45,7 @@ export declare class TokenBudgetManager {
     recordToolCall(toolName: string, request: unknown, response: unknown): void;
     private isRecord;
     private hasDetailedSummarySize;
+    private tryEstimateMcpEnvelope;
     private calculateSize;
     private normalizeForSizeEstimate;
     private estimateTokens;

package/dist/utils/TokenBudgetManager.js

@@ -74,11 +74,33 @@ export class TokenBudgetManager {
         const { size } = summary;
         return typeof size === 'number' && Number.isFinite(size) && size > 0;
     }
+    tryEstimateMcpEnvelope(data) {
+        if (!this.isRecord(data))
+            return null;
+        const content = data['content'];
+        if (!Array.isArray(content) || content.length === 0)
+            return null;
+        const first = content[0];
+        if (!this.isRecord(first))
+            return null;
+        if (first['type'] !== 'text' || typeof first['text'] !== 'string')
+            return null;
+        const text = first['text'];
+        const truncated = text.length > this.MAX_ESTIMATION_STRING_LENGTH
+            ? `${text.slice(0, this.MAX_ESTIMATION_STRING_LENGTH)}...[truncated:${text.length}]`
+            : text;
+        const overhead = 42 + (data['isError'] === true ? 14 : 0);
+        const textBytes = Buffer.byteLength(truncated, 'utf8');
+        return Math.min(overhead + textBytes, this.MAX_ESTIMATION_BYTES);
+    }
     calculateSize(data) {
         try {
             if (this.hasDetailedSummarySize(data)) {
                 return Math.min(data.summary.size, this.MAX_ESTIMATION_BYTES);
             }
+            const mcpSize = this.tryEstimateMcpEnvelope(data);
+            if (mcpSize !== null)
+                return mcpSize;
             const normalized = this.normalizeForSizeEstimate(data, 0, new WeakSet());
             const serialized = JSON.stringify(normalized);
             if (!serialized) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@jshookmcp/jshook",
-  "version": "0.1.5",
+  "version": "0.1.6",
   "packageManager": "pnpm@10.28.2",
   "description": "MCP server with 244 built-in tools (236 domain tools across 16 domains) for AI-assisted JavaScript analysis and security analysis — browser automation, CDP debugging, network monitoring, JS hooks, code analysis, and workflow orchestration",
   "mcpName": "io.github.vmoranv/jshookmcp",
@@ -42,11 +42,13 @@
     "lint": "eslint src --ext .ts",
     "format": "prettier --write \"src/**/*.ts\"",
     "test": "vitest run",
+    "test:coverage": "cross-env ENABLE_INJECTION_TOOLS=true vitest run --coverage",
     "prepack": "pnpm run build",
     "audit:tools": "node scripts/audit-tools.mjs",
     "check": "pnpm run lint && pnpm run typecheck && pnpm run test",
     "package": "pnpm pack",
     "prepublishOnly": "pnpm run check",
+    "postinstall": "lefthook install",
     "install:full": "pnpm install && pnpm exec camoufox-js fetch"
   },
   "keywords": [
@@ -104,8 +106,10 @@
     "@types/node": "^25.3.0",
     "@typescript-eslint/eslint-plugin": "^8.56.1",
     "@typescript-eslint/parser": "^8.56.1",
+    "cross-env": "^10.1.0",
     "eslint": "^10.0.2",
     "estree-walker": "^2.0.2",
+    "lefthook": "^2.1.3",
     "nodemon": "^3.1.14",
     "prettier": "^3.8.1",
     "tsc-alias": "^1.8.16",

package/src/native/scripts/linux/enum-windows.sh

@@ -1,12 +1,12 @@
-#!/bin/bash
-# enum-windows.sh
-# Linux window enumeration script (placeholder)
-
-TARGET_PID=$1
-
-# Use xdotool or wmctrl for window enumeration
-# This is a placeholder for future implementation
-
-if command -v xdotool &> /dev/null; then
-    xdotool search --pid "$TARGET_PID" --name "" get-window-name
-fi
+#!/bin/bash
+# enum-windows.sh
+# Linux window enumeration script (placeholder)
+
+TARGET_PID=$1
+
+# Use xdotool or wmctrl for window enumeration
+# This is a placeholder for future implementation
+
+if command -v xdotool &> /dev/null; then
+    xdotool search --pid "$TARGET_PID" --name "" get-window-name
+fi

package/src/native/scripts/macos/enum-windows.applescript

@@ -1,22 +1,22 @@
--- enum-windows.applescript
--- macOS window enumeration script (placeholder)
-
-param TargetPid
-
--- AppleScript implementation for window enumeration
--- This is a placeholder for future implementation
-
-tell application "System Events"
-    set windowList to {}
-    repeat with proc in (every process whose unix id is TargetPid)
-        repeat with win in windows of proc
-            set end of windowList to {¬
-                title: name of win, ¬
-                position: position of win, ¬
-                size: size of win ¬
-            }
-        end repeat
-    end repeat
-end tell
-
-return windowList
+-- enum-windows.applescript
+-- macOS window enumeration script (placeholder)
+
+param TargetPid
+
+-- AppleScript implementation for window enumeration
+-- This is a placeholder for future implementation
+
+tell application "System Events"
+    set windowList to {}
+    repeat with proc in (every process whose unix id is TargetPid)
+        repeat with win in windows of proc
+            set end of windowList to {¬
+                title: name of win, ¬
+                position: position of win, ¬
+                size: size of win ¬
+            }
+        end repeat
+    end repeat
+end tell
+
+return windowList

package/src/native/scripts/windows/enum-windows-by-class.ps1

@@ -1,51 +1,51 @@
-param(
-    [string]$ClassPattern
-)
-
-Add-Type @"
-  using System;
-  using System.Runtime.InteropServices;
-  public class Win32 {
-    [DllImport("user32.dll")] public static extern IntPtr FindWindowEx(IntPtr parent, IntPtr childAfter, string className, string title);
-    [DllImport("user32.dll")] public static extern int GetWindowThreadProcessId(IntPtr hWnd, out int pid);
-    [DllImport("user32.dll")] public static extern int GetWindowText(IntPtr hWnd, System.Text.StringBuilder text, int count);
-    [DllImport("user32.dll")] public static extern int GetClassName(IntPtr hWnd, System.Text.StringBuilder className, int maxCount);
-  }
-"@
-
-$windows = @()
-$hwnd = [IntPtr]::Zero
-while ($true) {
-  $hwnd = [Win32]::FindWindowEx([IntPtr]::Zero, $hwnd, $null, $null)
-  if ($hwnd -eq [IntPtr]::Zero) { break }
-
-  $className = New-Object System.Text.StringBuilder 256
-  [Win32]::GetClassName($hwnd, $className, 256) | Out-Null
-  $classNameStr = $className.ToString()
-
-  # Support wildcard pattern matching
-  $isMatch = $false
-  if ($ClassPattern -eq $classNameStr) {
-    $isMatch = $true
-  } elseif ($ClassPattern.Contains('*')) {
-    # Convert wildcard pattern to regex
-    $regexPattern = [regex]::Escape($ClassPattern).Replace('\*', '.*')
-    if ($classNameStr -match $regexPattern) {
-      $isMatch = $true
-    }
-  }
-
-  if ($isMatch) {
-    $windowPid = 0
-    [Win32]::GetWindowThreadProcessId($hwnd, [ref]$windowPid) | Out-Null
-    $title = New-Object System.Text.StringBuilder 256
-    [Win32]::GetWindowText($hwnd, $title, 256) | Out-Null
-    $windows += @{
-      Handle = $hwnd.ToString()
-      Title = $title.ToString()
-      ClassName = $classNameStr
-      ProcessId = $windowPid
-    }
-  }
-}
-$windows | ConvertTo-Json -Compress
+param(
+    [string]$ClassPattern
+)
+
+Add-Type @"
+  using System;
+  using System.Runtime.InteropServices;
+  public class Win32 {
+    [DllImport("user32.dll")] public static extern IntPtr FindWindowEx(IntPtr parent, IntPtr childAfter, string className, string title);
+    [DllImport("user32.dll")] public static extern int GetWindowThreadProcessId(IntPtr hWnd, out int pid);
+    [DllImport("user32.dll")] public static extern int GetWindowText(IntPtr hWnd, System.Text.StringBuilder text, int count);
+    [DllImport("user32.dll")] public static extern int GetClassName(IntPtr hWnd, System.Text.StringBuilder className, int maxCount);
+  }
+"@
+
+$windows = @()
+$hwnd = [IntPtr]::Zero
+while ($true) {
+  $hwnd = [Win32]::FindWindowEx([IntPtr]::Zero, $hwnd, $null, $null)
+  if ($hwnd -eq [IntPtr]::Zero) { break }
+
+  $className = New-Object System.Text.StringBuilder 256
+  [Win32]::GetClassName($hwnd, $className, 256) | Out-Null
+  $classNameStr = $className.ToString()
+
+  # Support wildcard pattern matching
+  $isMatch = $false
+  if ($ClassPattern -eq $classNameStr) {
+    $isMatch = $true
+  } elseif ($ClassPattern.Contains('*')) {
+    # Convert wildcard pattern to regex
+    $regexPattern = [regex]::Escape($ClassPattern).Replace('\*', '.*')
+    if ($classNameStr -match $regexPattern) {
+      $isMatch = $true
+    }
+  }
+
+  if ($isMatch) {
+    $windowPid = 0
+    [Win32]::GetWindowThreadProcessId($hwnd, [ref]$windowPid) | Out-Null
+    $title = New-Object System.Text.StringBuilder 256
+    [Win32]::GetWindowText($hwnd, $title, 256) | Out-Null
+    $windows += @{
+      Handle = $hwnd.ToString()
+      Title = $title.ToString()
+      ClassName = $classNameStr
+      ProcessId = $windowPid
+    }
+  }
+}
+$windows | ConvertTo-Json -Compress

package/src/native/scripts/windows/enum-windows.ps1

@@ -1,44 +1,44 @@
-param(
-    [int]$TargetPid
-)
-
-Add-Type @"
-  using System;
-  using System.Runtime.InteropServices;
-  public class Win32 {
-    [DllImport("user32.dll")] public static extern IntPtr FindWindowEx(IntPtr parent, IntPtr childAfter, string className, string title);
-    [DllImport("user32.dll")] public static extern int GetWindowThreadProcessId(IntPtr hWnd, out int pid);
-    [DllImport("user32.dll")] public static extern int GetWindowText(IntPtr hWnd, System.Text.StringBuilder text, int count);
-    [DllImport("user32.dll")] public static extern int GetClassName(IntPtr hWnd, System.Text.StringBuilder className, int maxCount);
-    [DllImport("user32.dll")] public static extern bool GetWindowRect(IntPtr hWnd, out RECT rect);
-    [StructLayout(LayoutKind.Sequential)] public struct RECT { public int Left, Top, Right, Bottom; }
-  }
-"@
-
-$windows = @()
-$hwnd = [IntPtr]::Zero
-while ($true) {
-  $hwnd = [Win32]::FindWindowEx([IntPtr]::Zero, $hwnd, $null, $null)
-  if ($hwnd -eq [IntPtr]::Zero) { break }
-  $windowPid = 0
-  [Win32]::GetWindowThreadProcessId($hwnd, [ref]$windowPid) | Out-Null
-  if ($windowPid -eq $TargetPid) {
-    $title = New-Object System.Text.StringBuilder 256
-    $className = New-Object System.Text.StringBuilder 256
-    [Win32]::GetWindowText($hwnd, $title, 256) | Out-Null
-    [Win32]::GetClassName($hwnd, $className, 256) | Out-Null
-    $rect = New-Object Win32+RECT
-    [Win32]::GetWindowRect($hwnd, [ref]$rect) | Out-Null
-    $windows += @{
-      Handle = $hwnd.ToString()
-      Title = $title.ToString()
-      ClassName = $className.ToString()
-      ProcessId = $windowPid
-      Left = $rect.Left
-      Top = $rect.Top
-      Right = $rect.Right
-      Bottom = $rect.Bottom
-    }
-  }
-}
-$windows | ConvertTo-Json -Compress
+param(
+    [int]$TargetPid
+)
+
+Add-Type @"
+  using System;
+  using System.Runtime.InteropServices;
+  public class Win32 {
+    [DllImport("user32.dll")] public static extern IntPtr FindWindowEx(IntPtr parent, IntPtr childAfter, string className, string title);
+    [DllImport("user32.dll")] public static extern int GetWindowThreadProcessId(IntPtr hWnd, out int pid);
+    [DllImport("user32.dll")] public static extern int GetWindowText(IntPtr hWnd, System.Text.StringBuilder text, int count);
+    [DllImport("user32.dll")] public static extern int GetClassName(IntPtr hWnd, System.Text.StringBuilder className, int maxCount);
+    [DllImport("user32.dll")] public static extern bool GetWindowRect(IntPtr hWnd, out RECT rect);
+    [StructLayout(LayoutKind.Sequential)] public struct RECT { public int Left, Top, Right, Bottom; }
+  }
+"@
+
+$windows = @()
+$hwnd = [IntPtr]::Zero
+while ($true) {
+  $hwnd = [Win32]::FindWindowEx([IntPtr]::Zero, $hwnd, $null, $null)
+  if ($hwnd -eq [IntPtr]::Zero) { break }
+  $windowPid = 0
+  [Win32]::GetWindowThreadProcessId($hwnd, [ref]$windowPid) | Out-Null
+  if ($windowPid -eq $TargetPid) {
+    $title = New-Object System.Text.StringBuilder 256
+    $className = New-Object System.Text.StringBuilder 256
+    [Win32]::GetWindowText($hwnd, $title, 256) | Out-Null
+    [Win32]::GetClassName($hwnd, $className, 256) | Out-Null
+    $rect = New-Object Win32+RECT
+    [Win32]::GetWindowRect($hwnd, [ref]$rect) | Out-Null
+    $windows += @{
+      Handle = $hwnd.ToString()
+      Title = $title.ToString()
+      ClassName = $className.ToString()
+      ProcessId = $windowPid
+      Left = $rect.Left
+      Top = $rect.Top
+      Right = $rect.Right
+      Bottom = $rect.Bottom
+    }
+  }
+}
+$windows | ConvertTo-Json -Compress

package/src/native/scripts/windows/inject-dll.ps1

@@ -1,21 +1,21 @@
-param(
-    [int]$TargetPid,
-    [string]$DllPath
-)
-
-Add-Type @"
-  using System;
-  using System.Runtime.InteropServices;
-  public class Injector {
-    [DllImport("kernel32.dll")] public static extern IntPtr OpenProcess(int access, bool inherit, int pid);
-    [DllImport("kernel32.dll")] public static extern IntPtr VirtualAllocEx(IntPtr hProcess, IntPtr addr, int size, int alloc, int protect);
-    [DllImport("kernel32.dll")] public static extern bool WriteProcessMemory(IntPtr hProcess, IntPtr addr, byte[] buffer, int size, out int written);
-    [DllImport("kernel32.dll")] public static extern IntPtr CreateRemoteThread(IntPtr hProcess, IntPtr attr, int stack, IntPtr start, IntPtr param, int flags, out int threadId);
-    [DllImport("kernel32.dll")] public static extern IntPtr GetModuleHandle(string name);
-    [DllImport("kernel32.dll")] public static extern IntPtr GetProcAddress(IntPtr hModule, string name);
-    [DllImport("kernel32.dll")] public static extern bool CloseHandle(IntPtr handle);
-  }
-"@
-
-# Injection requires elevated privileges and is disabled for safety
-Write-Output "DLL injection is disabled for safety in this implementation. PID: $TargetPid, DLL: $DllPath"
+param(
+    [int]$TargetPid,
+    [string]$DllPath
+)
+
+Add-Type @"
+  using System;
+  using System.Runtime.InteropServices;
+  public class Injector {
+    [DllImport("kernel32.dll")] public static extern IntPtr OpenProcess(int access, bool inherit, int pid);
+    [DllImport("kernel32.dll")] public static extern IntPtr VirtualAllocEx(IntPtr hProcess, IntPtr addr, int size, int alloc, int protect);
+    [DllImport("kernel32.dll")] public static extern bool WriteProcessMemory(IntPtr hProcess, IntPtr addr, byte[] buffer, int size, out int written);
+    [DllImport("kernel32.dll")] public static extern IntPtr CreateRemoteThread(IntPtr hProcess, IntPtr attr, int stack, IntPtr start, IntPtr param, int flags, out int threadId);
+    [DllImport("kernel32.dll")] public static extern IntPtr GetModuleHandle(string name);
+    [DllImport("kernel32.dll")] public static extern IntPtr GetProcAddress(IntPtr hModule, string name);
+    [DllImport("kernel32.dll")] public static extern bool CloseHandle(IntPtr handle);
+  }
+"@
+
+# Injection requires elevated privileges and is disabled for safety
+Write-Output "DLL injection is disabled for safety in this implementation. PID: $TargetPid, DLL: $DllPath"