Update index.js

index.js

@@ -338,27 +338,45 @@ app.post('/process', upload.single('htmlFile'), (req, res) => {
   }
 });
 
-// Add this function to handle script execution
+// Helper function to validate script structure
+function validateScript(scriptContent) {
+  // Check if script contains function declaration
+  if (!scriptContent.includes('function extract(')) {
+    throw new Error('Script must contain a function named "extract"');
+  }
+}
+
+// Modified function to execute the extract function
 function executeCheerioScript(html, scriptContent) {
   try {
-    const $ = cheerio.load(html, {
-      decodeEntities: false,
-      xmlMode: false,
-      lowerCaseTags: true
-    });
+    validateScript(scriptContent);
+    
+    // Create a safe context for the script
+    const context = {
+      cheerio,
+      input: html
+    };
 
-    // Create a safe function from the script content
-    const extractorFunction = new Function('$', `
+    // Create the function from script content
+    const extractorFunction = new Function('input', 'cheerio', `
       ${scriptContent}
+      return extract(input, cheerio);
     `);
 
-    // Execute the script with Cheerio instance
-    const result = extractorFunction($);
-    return {
-      success: true,
-      data: result,
-      error: null
-    };
+    // Execute the script with provided parameters
+    const result = extractorFunction(html, cheerio);
+
+    // Validate result structure
+    if (!result || typeof result !== 'object') {
+      throw new Error('Extract function must return an object');
+    }
+
+    if (!('success' in result && 'data' in result && 'error' in result)) {
+      throw new Error('Return object must contain success, data, and error fields');
+    }
+
+    return result;
+
   } catch (err) {
     return {
       success: false,
@@ -368,7 +386,7 @@ function executeCheerioScript(html, scriptContent) {
   }
 }
 
-// cherio extract endpoint
+// Updated endpoint
 app.post('/extract', upload.single('htmlFile'), (req, res) => {
   try {
     const startTime = Date.now();
@@ -405,7 +423,6 @@ app.post('/extract', upload.single('htmlFile'), (req, res) => {
   }
 });
 
-
 const PORT = process.env.PORT || 3000;
 app.listen(PORT, () => {
   console.log(`Server running on http://localhost:${PORT}`);