chore: fix pr comment

Author: JustasMonkev

docs/tool-reference.md

@@ -195,10 +195,8 @@
 
 **Parameters:**
 
-- **clearGeolocation** (boolean) _(optional)_: Set to true to clear the geolocation override.
 - **cpuThrottlingRate** (number) _(optional)_: Represents the CPU slowdown factor. Set the rate to 1 to disable throttling. If omitted, throttling remains unchanged.
-- **latitude** (number) _(optional)_: Latitude between -90 and 90 for geolocation emulation. Must be provided together with longitude.
-- **longitude** (number) _(optional)_: Longitude between -180 and 180 for geolocation emulation. Must be provided together with latitude.
+- **geolocation** (unknown) _(optional)_: Geolocation to [`emulate`](#emulate). Set to null to clear the geolocation override.
 - **networkConditions** (enum: "No emulation", "Offline", "Slow 3G", "Fast 3G", "Slow 4G", "Fast 4G") _(optional)_: Throttle network. Set to "No emulation" to disable. If omitted, conditions remain unchanged.
 
 ---

src/tools/emulation.ts

@@ -37,36 +37,28 @@ export const emulate = defineTool({
       .describe(
         'Represents the CPU slowdown factor. Set the rate to 1 to disable throttling. If omitted, throttling remains unchanged.',
       ),
-    latitude: zod
-      .number()
-      .min(-90)
-      .max(90)
-      .optional()
-      .describe(
-        'Latitude between -90 and 90 for geolocation emulation. Must be provided together with longitude.',
-      ),
-    longitude: zod
-      .number()
-      .min(-180)
-      .max(180)
+    geolocation: zod
+      .object({
+        latitude: zod
+          .number()
+          .min(-90)
+          .max(90)
+          .describe('Latitude between -90 and 90.'),
+        longitude: zod
+          .number()
+          .min(-180)
+          .max(180)
+          .describe('Longitude between -180 and 180.'),
+      })
+      .nullable()
       .optional()
       .describe(
-        'Longitude between -180 and 180 for geolocation emulation. Must be provided together with latitude.',
+        'Geolocation to emulate. Set to null to clear the geolocation override.',
       ),
-    clearGeolocation: zod
-      .boolean()
-      .optional()
-      .describe('Set to true to clear the geolocation override.'),
   },
   handler: async (request, _response, context) => {
     const page = context.getSelectedPage();
-    const {
-      networkConditions,
-      cpuThrottlingRate,
-      latitude,
-      longitude,
-      clearGeolocation,
-    } = request.params;
+    const {networkConditions, cpuThrottlingRate, geolocation} = request.params;
 
     if (networkConditions) {
       if (networkConditions === 'No emulation') {
@@ -95,17 +87,13 @@ export const emulate = defineTool({
       context.setCpuThrottlingRate(cpuThrottlingRate);
     }
 
-    if (clearGeolocation) {
-      await page.setGeolocation({latitude: 0, longitude: 0});
-      context.setGeolocation(null);
-    } else if (latitude !== undefined || longitude !== undefined) {
-      if (latitude !== undefined && longitude !== undefined) {
-        await page.setGeolocation({latitude, longitude});
-        context.setGeolocation({latitude, longitude});
+    if (geolocation !== undefined) {
+      if (geolocation === null) {
+        await page.setGeolocation({latitude: 0, longitude: 0});
+        context.setGeolocation(null);
       } else {
-        throw new Error(
-          'Both latitude and longitude must be provided together for geolocation emulation.',
-        );
+        await page.setGeolocation(geolocation);
+        context.setGeolocation(geolocation);
       }
     }
   },

tests/tools/emulation.test.ts

@@ -159,8 +159,10 @@ describe('emulation', () => {
         await emulate.handler(
           {
             params: {
-              latitude: 48.137154,
-              longitude: 11.576124,
+              geolocation: {
+                latitude: 48.137154,
+                longitude: 11.576124,
+              },
             },
           },
           response,
@@ -173,14 +175,16 @@ describe('emulation', () => {
       });
     });
 
-    it('clears geolocation override when clearGeolocation is true', async () => {
+    it('clears geolocation override when geolocation is set to null', async () => {
       await withMcpContext(async (response, context) => {
         // First set a geolocation
         await emulate.handler(
           {
             params: {
-              latitude: 48.137154,
-              longitude: 11.576124,
+              geolocation: {
+                latitude: 48.137154,
+                longitude: 11.576124,
+              },
             },
           },
           response,
@@ -189,11 +193,11 @@ describe('emulation', () => {
 
         assert.notStrictEqual(context.getGeolocation(), null);
 
-        // Then clear it using clearGeolocation
+        // Then clear it by setting geolocation to null
         await emulate.handler(
           {
             params: {
-              clearGeolocation: true,
+              geolocation: null,
             },
           },
           response,
@@ -204,35 +208,15 @@ describe('emulation', () => {
       });
     });
 
-    it('throws error when only latitude is provided', async () => {
-      await withMcpContext(async (response, context) => {
-        await assert.rejects(
-          async () => {
-            await emulate.handler(
-              {
-                params: {
-                  latitude: 48.137154,
-                },
-              },
-              response,
-              context,
-            );
-          },
-          {
-            message:
-              'Both latitude and longitude must be provided together for geolocation emulation.',
-          },
-        );
-      });
-    });
-
     it('reports correctly for the currently selected page', async () => {
       await withMcpContext(async (response, context) => {
         await emulate.handler(
           {
             params: {
-              latitude: 48.137154,
-              longitude: 11.576124,
+              geolocation: {
+                latitude: 48.137154,
+                longitude: 11.576124,
+              },
             },
           },
           response,