Spaces:
Sleeping
Sleeping
File size: 2,979 Bytes
7dc28be | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 | import type { FastMCP } from 'fastmcp';
import { UserError } from 'fastmcp';
import { z } from 'zod';
import { getSheetsClient } from '../../clients.js';
import * as SheetsHelpers from '../../googleSheetsApiHelpers.js';
export function register(server: FastMCP) {
server.addTool({
name: 'updateTableRange',
description:
"Modifies a table's dimensions (add/remove rows and columns) by updating its range. The new range must include the original table range.",
parameters: z.strictObject({
spreadsheetId: z
.string()
.describe(
'The spreadsheet ID — the long string between /d/ and /edit in a Google Sheets URL.'
),
tableIdentifier: z
.string()
.describe('The table name or table ID to update. Use listTables to see available tables.'),
range: z
.string()
.describe(
'New A1 notation range for the table (e.g., "Sheet1!A1:F15"). Must include the original table range.'
),
}),
execute: async (args, { log }) => {
const sheets = await getSheetsClient();
log.info(`Updating table range for "${args.tableIdentifier}": ${args.range}`);
try {
// Resolve the table to get its current info
const { table, sheetName } = await SheetsHelpers.resolveTableIdentifier(
sheets,
args.spreadsheetId,
args.tableIdentifier
);
// Parse the new range
const { a1Range } = SheetsHelpers.parseRange(args.range);
const sheetId = await SheetsHelpers.resolveSheetId(
sheets,
args.spreadsheetId,
sheetName || undefined
);
const newRange = SheetsHelpers.parseA1ToGridRange(a1Range, sheetId);
// Update the table range
const updatedTable = await SheetsHelpers.updateTableRangeHelper(
sheets,
args.spreadsheetId,
table.tableId || '',
newRange
);
return JSON.stringify(
{
tableId: updatedTable.tableId,
name: updatedTable.name,
oldRange: table.range
? `${SheetsHelpers.rowColToA1(
table.range.startRowIndex || 0,
table.range.startColumnIndex || 0
)}:${SheetsHelpers.rowColToA1(
(table.range.endRowIndex || 1) - 1,
(table.range.endColumnIndex || 1) - 1
)}`
: 'Unknown',
newRange: args.range,
columnCount: updatedTable.columnProperties?.length || 0,
message: `Table "${updatedTable.name}" range updated successfully.`,
},
null,
2
);
} catch (error: any) {
log.error(`Error updating table range: ${error.message || error}`);
if (error instanceof UserError) throw error;
throw new UserError(`Failed to update table range: ${error.message || 'Unknown error'}`);
}
},
});
}
|