updated logic

Author: ashishexee

src/extension.ts

@@ -116,43 +116,14 @@ export function activate(context: vscode.ExtensionContext) {
                 }
 
                 const boilerplate = language === 'python'
-                    ? `import os
-
-def run_test_case(test_case_number, function):
-    # Determine the base directory dynamically\n
-    base_directory = os.path.dirname(os.path.dirname(__file__))\n
-    file_path = os.path.join(base_directory, 'test_cases', f"input_{test_case_number}.txt")\n
-
-    try:\n
-        # Read the input file\n
-        with open(file_path, "r") as file:\n
-            content = file.read().strip().splitlines()  # Read file and split lines\n
-
-        # Dynamically parse all lines as arguments\n
-        args = [eval(line.strip()) for line in content]  # Parse each line in the file\n
-
-        # Call the provided function with all parsed arguments\n
-        result = function(*args)  # Unpack the arguments dynamically\n
-        print(result)  # Output the result\n
-
-    except FileNotFoundError:\n
-        print(f"Error: File not found at {file_path}. Check the file path and try again.")\n
-    except ValueError as e:\n
-        print(f"Error: {e}")\n
-    except SyntaxError as e:\n
-        print(f"Error: Check your input file format. {e}")\n
-    except Exception as e:\n
-        print(f"Unexpected Error: {e}")\n
-
-
-# WRITE YOUR FUNCTION HERE 
-#-------------------------------------------------------------------------------\n
-
-#-------------------------------------------------------------------------------\n
-
-#run_test_case(ENTER THE TESTCASE NUMBER, ENTER THE FUNCTION NAME)\n
-# EXAMPLE:\n
-run_test_case()  # Replace with the relevant test case number and function name\n`
+                    ? `import os; run_test_case=lambda n,f: print((lambda p: (lambda a: f(*a))(eval(l.strip()) for l in open(p).read().strip().splitlines()) if os.path.isfile(p) else f"Error: File not found at {p}. Check the file path and try again.")((lambda d: os.path.join(d, 'test_cases', f"input_{n}.txt"))(os.path.dirname(os.path.dirname(file)))))
+#WRITE YOUR CODE LOGIC HERE
+
+#---------------------------------------------------------------------------------------
+# Example usage
+# Provide the test case number and function name
+#run_test_case(TEST_CASE_NUMBER, FUNCTION_NAME);
+run_test_case()`
 
 
                     : `#include <bits/stdc++.h>

src/fetchTestCases.ts

@@ -3,15 +3,20 @@ import * as fs from 'fs';
 import * as path from 'path';
 import * as vscode from 'vscode';
 
-const RETRY_DELAY = 100; // 0.1 seconds
+interface TestCase {
+    input: string | null;
+    output: string | null;
+}
+
+const RETRY_DELAY = 3000; // 0.1 seconds
 const MAX_RETRIES = 5; // Maximum retry attempts
 
 export async function fetchTestCases(url: string): Promise<void> {
     const browser = await puppeteer.launch({ headless: false }); // Run in visible mode for debugging
     const page = await browser.newPage();
 
     let attempts = 0;
-    let testCases = [];
+    let testCases: TestCase[] = [];
 
     const baseDirectory = path.join(__dirname, 'dist', 'test_cases');
 
@@ -83,7 +88,6 @@ export async function fetchTestCases(url: string): Promise<void> {
         console.log('Failed to fetch test cases after maximum retries.');
     }
 }
-
 async function fetchTestCasesWithRetry(page: puppeteer.Page, url: string): Promise<any[]> {
     // Load cookies if available
     await loadCookies(page);
@@ -96,24 +100,50 @@ async function fetchTestCasesWithRetry(page: puppeteer.Page, url: string): Promi
     }
 
     console.log('Waiting for test cases to load...');
+    let testCases: TestCase[] = [];
     try {
         await page.waitForSelector('pre', { timeout: 20000 }); // Wait up to 20 seconds for the <pre> tags
+        // Extract test cases
+        testCases = await page.$$eval('pre', (elements) => {
+            return elements.map(pre => {
+                const inputMatch = pre.innerText.match(/Input:\s*(.+)/);
+                const outputMatch = pre.innerText.match(/Output:\s*(.+)/);
+                return {
+                    input: inputMatch ? inputMatch[1].trim() : null,
+                    output: outputMatch ? outputMatch[1].trim() : null,
+                };
+            });
+        });
     } catch (error) {
-        console.error('Failed to find <pre> tags in time:', error);
-        return []; // Return empty array if fetching fails
+        console.error('Failed to extract test cases using <pre> selector:', error);
     }
 
-    // Extract test cases
-    const testCases = await page.$$eval('pre', (elements) => {
-        return elements.map(pre => {
-            const inputMatch = pre.innerText.match(/Input:\s*(.+)/);
-            const outputMatch = pre.innerText.match(/Output:\s*(.+)/);
-            return {
-                input: inputMatch ? inputMatch[1].trim() : null,
-                output: outputMatch ? outputMatch[1].trim() : null,
-            };
-        });
-    });
+    if (testCases.length === 0) {
+        try {
+            // Fallback to the .example-block structure
+            console.log('Trying to extract test cases using .example-block structure...');
+            await page.waitForSelector('.example-block', { timeout: 5000 }); // Adjust timeout as needed
+            testCases = await page.$$eval('.example-block', (blocks) => {
+                return blocks.map(block => {
+                    const inputElement = block.querySelector('strong:contains("Input:") + span.example-io');
+                    const outputElement = block.querySelector('strong:contains("Output:") + span.example-io');
+
+                    return {
+                        input: inputElement ? (inputElement as HTMLElement).innerText.trim() : null,
+                        output: outputElement ? (outputElement as HTMLElement).innerText.trim() : null,
+                    };
+                });
+            });
+
+            if (testCases.length > 0) {
+                console.log('Test cases successfully extracted using .example-block structure.');
+            } else {
+                console.log('No test cases found using .example-block structure.');
+            }
+        } catch (error) {
+            console.error('Failed to extract test cases using .example-block structure:', error);
+        }
+    }
 
     return testCases;
 }