Spaces:
Sleeping
Sleeping
File size: 2,793 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 | 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: 'autoResizeColumns',
description:
'Auto-resizes columns in a spreadsheet to fit their content. Optionally restrict to a column range (e.g., "A:S"); defaults to all columns.',
parameters: z.strictObject({
spreadsheetId: z
.string()
.describe(
'The spreadsheet ID — the long string between /d/ and /edit in a Google Sheets URL.'
),
sheetName: z
.string()
.optional()
.describe('Name of the sheet/tab. Defaults to the first sheet if not provided.'),
columns: z
.string()
.optional()
.describe(
'Column range to resize in A1 notation (e.g., "A:S"). Omit to resize all columns.'
),
}),
execute: async (args, { log }) => {
const sheets = await getSheetsClient();
log.info(`Auto-resizing columns in spreadsheet ${args.spreadsheetId}`);
try {
const sheetId = await SheetsHelpers.resolveSheetId(
sheets,
args.spreadsheetId,
args.sheetName
);
const dimensionRange: any = {
sheetId,
dimension: 'COLUMNS',
};
if (args.columns) {
const colonIdx = args.columns.indexOf(':');
if (colonIdx !== -1) {
dimensionRange.startIndex = SheetsHelpers.colLettersToIndex(
args.columns.slice(0, colonIdx).trim()
);
dimensionRange.endIndex =
SheetsHelpers.colLettersToIndex(args.columns.slice(colonIdx + 1).trim()) + 1;
} else {
const idx = SheetsHelpers.colLettersToIndex(args.columns.trim());
dimensionRange.startIndex = idx;
dimensionRange.endIndex = idx + 1;
}
}
await sheets.spreadsheets.batchUpdate({
spreadsheetId: args.spreadsheetId,
requestBody: {
requests: [
{
autoResizeDimensions: {
dimensions: dimensionRange,
},
},
],
},
});
const rangeDesc = args.columns ? `columns ${args.columns}` : 'all columns';
return `Successfully auto-resized ${rangeDesc} to fit content.`;
} catch (error: any) {
log.error(`Error auto-resizing columns: ${error.message || error}`);
if (error instanceof UserError) throw error;
throw new UserError(`Failed to auto-resize columns: ${error.message || 'Unknown error'}`);
}
},
});
}
|