Load caps from env

Co-authored-by: SrinivasanTarget <[email protected]>

Author: saikrishna321

capabilities.json

@@ -0,0 +1,10 @@
+{
+  "android": {
+    "appium:app": "/app/path"
+  },
+  "ios": {
+    "appium:deviceName": "iPhone 16 Pro",
+    "appium:platformVersion": "18.2",
+    "appium:udid": ""
+  }
+}

src/resources/java/template.ts

@@ -6,7 +6,7 @@ export default function javaTemplatesResource(server: any) {
     uri: 'generate://code-with-locators',
     name: 'Generate Code With Locators',
     description: `Generate code for the current page with locators which was retrieved from generate_locators tool. 
-    which follows the page factory pattern.`,
+    which follows the page factory pattern. Dont not strictly create test cases, just generate the code for locators.`,
     mimeType: 'text/plain',
     async load() {
       // Return all content directly without templates object

src/tools/create-session.ts

@@ -2,6 +2,8 @@
  * Tool to create a new mobile session (Android or iOS)
  */
 import { z } from 'zod';
+import fs from 'fs';
+import path from 'path';
 import { AndroidUiautomator2Driver } from 'appium-uiautomator2-driver';
 import { XCUITestDriver } from 'appium-xcuitest-driver';
 import { setSession, getDriver, getSessionId } from './sessionStore.js';
@@ -10,10 +12,16 @@ import { setSession, getDriver, getSessionId } from './sessionStore.js';
 interface Capabilities {
   platformName: string;
   'appium:automationName': string;
-  'appium:deviceName': string;
+  'appium:deviceName'?: string;
   [key: string]: any;
 }
 
+// Define capabilities config type
+interface CapabilitiesConfig {
+  android: Record<string, any>;
+  ios: Record<string, any>;
+}
+
 export default function createSession(server: any): void {
   server.addTool({
     name: 'create_session',
@@ -42,35 +50,63 @@ export default function createSession(server: any): void {
 
         let defaultCapabilities: Capabilities;
         let driver: any;
-
+        let finalCapabilities: Capabilities;
+        
+        // Load capabilities from config file
+        let configCapabilities: CapabilitiesConfig = { android: {}, ios: {} };
+        const configPath = process.env.CAPABILITIES_CONFIG;
+        
+        if (configPath && fs.existsSync(configPath)) {
+          try {
+            const configContent = fs.readFileSync(configPath, 'utf8');
+            configCapabilities = JSON.parse(configContent);
+          } catch (error) {
+            console.warn(`Failed to parse capabilities config: ${error}`);
+          }
+        }
+        
         if (platform === 'android') {
           defaultCapabilities = {
             platformName: 'Android',
             'appium:automationName': 'UiAutomator2',
             'appium:deviceName': 'Android Device',
           };
+          
+          // Get platform-specific capabilities from config
+          const androidCaps = configCapabilities.android || {};
+
+          // Merge custom capabilities with defaults and config capabilities
+          finalCapabilities = {
+            ...defaultCapabilities,
+            ...androidCaps,
+            ...customCapabilities,
+          };
+          
           driver = new AndroidUiautomator2Driver();
         } else if (platform === 'ios') {
           defaultCapabilities = {
             platformName: 'iOS',
             'appium:automationName': 'XCUITest',
-            'appium:deviceName': 'iPhone 16 Pro',
-            'appium:platformVersion': '18.2',
-            'appium:udid': 'E9F10A3E-58E6-4506-B273-B22AF836014E',
+            'appium:deviceName': 'iPhone Simulator',
+          };
+          
+          // Get platform-specific capabilities from config
+          const iosCaps = configCapabilities.ios || {};
+              
+          // Merge custom capabilities with defaults and config capabilities
+          finalCapabilities = {
+            ...defaultCapabilities,
+            ...iosCaps,
+            ...customCapabilities,
           };
+          
           driver = new XCUITestDriver();
         } else {
           throw new Error(
             `Unsupported platform: ${platform}. Please choose 'android' or 'ios'.`
           );
         }
 
-        // Merge custom capabilities with defaults
-        const finalCapabilities = {
-          ...defaultCapabilities,
-          ...customCapabilities,
-        };
-
         console.log(
           `Creating new ${platform.toUpperCase()} session with capabilities:`,
           JSON.stringify(finalCapabilities, null, 2)