@shortcut/mcp 0.17.0 → 0.19.0

package/README.md

@@ -148,6 +148,10 @@ Or you can edit the local JSON file directly:
 - **stories-set-external-links** - Replace all external links on a story with a new set of links
 - **stories-get-by-external-link** - Find all stories that contain a specific external link
 
+### Labels
+- **labels-list** - List all labels in the Shortcut workspace.
+- **labels-create** - Create a new label in Shortcut.
+
 ### Epics
 
 - **epics-get-by-id** - Get a Shortcut epic by ID
@@ -188,6 +192,7 @@ Or you can edit the local JSON file directly:
 ### Documents
 
 - **documents-create** - Create a new document in Shortcut with HTML content
+- **documents-update** - Update content of an existing document by its ID
 - **documents-list** - List all documents in Shortcut
 - **documents-search** - Search for documents
 - **documents-get-by-id** - Retrieve a specific document in markdown format by its ID
@@ -227,6 +232,7 @@ The following values are accepted in addition to the full tool names listed abov
 - `stories`
 - `epics`
 - `iterations`
+- `labels`
 - `objectives`
 - `teams`
 - `workflows`
@@ -236,6 +242,8 @@ The following values are accepted in addition to the full tool names listed abov
 
 You can run the MCP server in read-only mode by setting the `SHORTCUT_READONLY` environment variable to `true`. This will disable all tools that modify data in Shortcut.
 
+Additionally, Shortcut now supports **read-only API tokens**, which you can use to ensure that the MCP server is limited to read-only operations at the API level. This provides an additional layer of security since the restriction is enforced by the Shortcut API itself, not just the MCP server. You can create a read-only token from your [Shortcut API tokens settings](https://app.shortcut.com/settings/account/api-tokens).
+
 Example:
 
 ```json

package/dist/index.js

@@ -1,8 +1,51 @@
 #!/usr/bin/env node
-import { CustomMcpServer, DocumentTools, EpicTools, IterationTools, ObjectiveTools, ShortcutClientWrapper, StoryTools, TeamTools, UserTools, WorkflowTools } from "./workflows-TjriXV16.js";
+import { a as ObjectiveTools, c as DocumentTools, d as ShortcutClientWrapper, i as StoryTools, l as BaseTools, n as UserTools, o as IterationTools, r as TeamTools, s as EpicTools, t as WorkflowTools, u as CustomMcpServer } from "./workflows-DYvKtRHR.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { ShortcutClient } from "@shortcut/client";
+import { z } from "zod";
 
+//#region src/tools/labels.ts
+/**
+* Tools for managing Shortcut labels.
+*/
+var LabelTools = class LabelTools extends BaseTools {
+	static create(client$1, server$1) {
+		const tools = new LabelTools(client$1);
+		server$1.addToolWithReadAccess("labels-list", "List all labels in the Shortcut workspace.", { includeArchived: z.boolean().optional().describe("Whether to include archived labels in the list.").default(false) }, async (params) => await tools.listLabels(params));
+		server$1.addToolWithWriteAccess("labels-create", "Create a new label in Shortcut.", {
+			name: z.string().min(1).max(128).describe("The name of the new label. Required."),
+			color: z.string().regex(/^#[a-fA-F0-9]{6}$/).optional().describe("The hex color to be displayed with the label (e.g., \"#ff0000\")."),
+			description: z.string().max(1024).optional().describe("A description of the label.")
+		}, async (params) => await tools.createLabel(params));
+		return tools;
+	}
+	formatLabel(label, { includeDescription = false, includeArchived = false } = {}) {
+		return {
+			id: label.id,
+			name: label.name,
+			app_url: label.app_url,
+			...includeDescription ? { description: label.description ?? null } : {},
+			...includeArchived ? { archived: label.archived } : {},
+			stats: Object.fromEntries(Object.entries(label.stats || {}).filter(([key, value]) => !key.match(/(unestimated|total)$/) && !!value))
+		};
+	}
+	async listLabels({ includeArchived = false }) {
+		const labels = await this.client.listLabels({ includeArchived });
+		if (!labels.length) return this.toResult("Result: No labels found.");
+		const formattedLabels = labels.map((label) => this.formatLabel(label, { includeArchived }));
+		return this.toResult(`Result (${labels.length} labels found):`, { labels: formattedLabels });
+	}
+	async createLabel({ name, color, description }) {
+		const label = await this.client.createLabel({
+			name,
+			color,
+			description
+		});
+		return this.toResult(`Label created with ID: ${label.id}.`, { label: this.formatLabel(label, { includeDescription: true }) });
+	}
+};
+
+//#endregion
 //#region src/server.ts
 let apiToken = process.env.SHORTCUT_API_TOKEN;
 let isReadonly = process.env.SHORTCUT_READONLY === "true";
@@ -29,6 +72,7 @@ ObjectiveTools.create(client, server);
 TeamTools.create(client, server);
 WorkflowTools.create(client, server);
 DocumentTools.create(client, server);
+LabelTools.create(client, server);
 async function startServer() {
 	try {
 		const transport = new StdioServerTransport();
@@ -40,4 +84,5 @@ async function startServer() {
 }
 startServer();
 
-//#endregion
+//#endregion
+export {  };

package/dist/server-http.js

@@ -1,4 +1,4 @@
-import { CustomMcpServer, DocumentTools, EpicTools, IterationTools, ObjectiveTools, ShortcutClientWrapper, StoryTools, TeamTools, UserTools, WorkflowTools } from "./workflows-TjriXV16.js";
+import { a as ObjectiveTools, c as DocumentTools, d as ShortcutClientWrapper, i as StoryTools, n as UserTools, o as IterationTools, r as TeamTools, s as EpicTools, t as WorkflowTools, u as CustomMcpServer } from "./workflows-DYvKtRHR.js";
 import { ShortcutClient } from "@shortcut/client";
 import { randomUUID } from "node:crypto";
 import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
@@ -52,15 +52,18 @@ const logger = pino({
 function loadConfig() {
 	let isReadonly = process.env.SHORTCUT_READONLY !== "false";
 	let enabledTools = parseToolsList(process.env.SHORTCUT_TOOLS || "");
+	let httpDebug = process.env.SHORTCUT_HTTP_DEBUG === "true";
 	if (process.argv.length >= 3) process.argv.slice(2).map((arg) => arg.split("=")).forEach(([name, value]) => {
 		if (name === "SHORTCUT_READONLY") isReadonly = value !== "false";
 		if (name === "SHORTCUT_TOOLS") enabledTools = parseToolsList(value);
+		if (name === "SHORTCUT_HTTP_DEBUG") httpDebug = value === "true";
 	});
 	return {
 		port: Number.parseInt(process.env.PORT || String(DEFAULT_PORT), 10),
 		isReadonly,
 		enabledTools,
-		sessionTimeoutMs: SESSION_TIMEOUT_MS
+		sessionTimeoutMs: SESSION_TIMEOUT_MS,
+		httpDebug
 	};
 }
 function parseToolsList(toolsStr) {
@@ -91,8 +94,7 @@ var SessionManager = class {
 		logger.info({ sessionId }, "Session initialized");
 	}
 	remove(sessionId) {
-		const session = this.sessions.get(sessionId);
-		if (session) {
+		if (this.sessions.get(sessionId)) {
 			this.sessions.delete(sessionId);
 			logger.info({ sessionId }, "Session removed");
 		}
@@ -105,10 +107,7 @@ var SessionManager = class {
 	cleanupStaleSessions() {
 		const now = Date.now();
 		const staleSessionIds = [];
-		for (const [sessionId, session] of this.sessions.entries()) {
-			const timeSinceLastAccess = now - session.lastAccessedAt.getTime();
-			if (timeSinceLastAccess > this.timeoutMs) staleSessionIds.push(sessionId);
-		}
+		for (const [sessionId, session] of this.sessions.entries()) if (now - session.lastAccessedAt.getTime() > this.timeoutMs) staleSessionIds.push(sessionId);
 		if (staleSessionIds.length > 0) {
 			logger.info({ count: staleSessionIds.length }, "Cleaning up stale sessions");
 			for (const sessionId of staleSessionIds) {
@@ -156,8 +155,7 @@ function extractApiToken(req) {
 */
 async function validateApiToken(token) {
 	try {
-		const client = new ShortcutClient(token);
-		await client.getCurrentMemberInfo();
+		await new ShortcutClient(token).getCurrentMemberInfo();
 		return true;
 	} catch (error) {
 		logger.debug({ error: error instanceof Error ? error.message : error }, "API token validation failed");
@@ -242,8 +240,7 @@ async function createTransport(apiToken, config, sessionManager) {
 			if (sid && sessionManager.has(sid)) sessionManager.remove(sid);
 		}
 	};
-	const server = createServerInstance(apiToken, config);
-	await server.connect(transport);
+	await createServerInstance(apiToken, config).connect(transport);
 	return transport;
 }
 async function handleMcpPost(req, res, sessionManager, config) {
@@ -266,8 +263,7 @@ async function handleMcpPost(req, res, sessionManager, config) {
 				sendUnauthorizedError(res, "API token does not match the session");
 				return;
 			}
-			const session = sessionManager.get(sessionId);
-			await session.transport.handleRequest(req, res, req.body);
+			await sessionManager.get(sessionId).transport.handleRequest(req, res, req.body);
 			return;
 		}
 		if (isInitializeRequest(req.body)) {
@@ -276,15 +272,13 @@ async function handleMcpPost(req, res, sessionManager, config) {
 				return;
 			}
 			reqLogger.info("Validating API token");
-			const isValid = await validateApiToken(apiToken);
-			if (!isValid) {
+			if (!await validateApiToken(apiToken)) {
 				reqLogger.warn("API token validation failed");
 				sendInvalidTokenError(res, requestId);
 				return;
 			}
 			reqLogger.info("API token validated, creating session");
-			const transport = await createTransport(apiToken, config, sessionManager);
-			await transport.handleRequest(req, res, req.body);
+			await (await createTransport(apiToken, config, sessionManager)).handleRequest(req, res, req.body);
 			return;
 		}
 		if (sessionId && !sessionManager.has(sessionId)) {
@@ -321,8 +315,7 @@ async function handleMcpGet(req, res, sessionManager) {
 	if (lastEventId) reqLogger.info({ lastEventId }, "Client reconnecting with Last-Event-ID");
 	else reqLogger.info("Establishing SSE stream");
 	try {
-		const session = sessionManager.get(sessionId);
-		await session.transport.handleRequest(req, res);
+		await sessionManager.get(sessionId).transport.handleRequest(req, res);
 	} catch (error) {
 		reqLogger.error({ error }, "Error handling MCP GET request");
 		if (!res.headersSent) res.status(500).send("Internal server error");
@@ -350,8 +343,7 @@ async function handleMcpDelete(req, res, sessionManager) {
 	}
 	reqLogger.info("Terminating session");
 	try {
-		const session = sessionManager.get(sessionId);
-		await session.transport.handleRequest(req, res);
+		await sessionManager.get(sessionId).transport.handleRequest(req, res);
 	} catch (error) {
 		reqLogger.error({ error }, "Error handling session termination");
 		if (!res.headersSent) res.status(500).send("Error processing session termination");
@@ -382,13 +374,43 @@ function loggingMiddleware(req, _res, next) {
 	}, "Incoming request");
 	next();
 }
+function httpDebugRequestMiddleware(req, _res, next) {
+	const headers = { ...req.headers };
+	delete headers[HEADERS.AUTHORIZATION];
+	delete headers[HEADERS.X_SHORTCUT_API_TOKEN];
+	delete headers.cookie;
+	logger.info(JSON.stringify({
+		event: "http_request",
+		method: req.method,
+		path: req.path,
+		url: req.originalUrl,
+		query: req.query,
+		headers,
+		body: req.body
+	}));
+	next();
+}
 async function startServer() {
 	const config = loadConfig();
 	const sessionManager = new SessionManager(config.sessionTimeoutMs);
 	const app = express();
 	app.use(express.json());
+	if (config.httpDebug) app.use(httpDebugRequestMiddleware);
 	app.use(corsMiddleware);
 	app.use(loggingMiddleware);
+	app.use((req, res, next) => {
+		const start = Date.now();
+		res.on("finish", () => {
+			if (res.statusCode >= 400) logger.info(JSON.stringify({
+				event: "http_request_failed",
+				method: req.method,
+				path: req.path,
+				status: res.statusCode,
+				ms: Date.now() - start
+			}));
+		});
+		next();
+	});
 	app.get("/health", (_req, res) => {
 		res.json({
 			status: "ok",
@@ -427,4 +449,5 @@ startServer().catch((error) => {
 	process.exit(1);
 });
 
-//#endregion
+//#endregion
+export {  };

package/dist/{workflows-TjriXV16.js → workflows-DYvKtRHR.js}

@@ -71,43 +71,37 @@ var ShortcutClientWrapper = class {
 	}
 	async loadMembers() {
 		if (this.userCache.isStale) {
-			const response = await this.client.listMembers({});
-			const members = response?.data ?? null;
+			const members = (await this.client.listMembers({}))?.data ?? null;
 			if (members) this.userCache.setMany(members.map((member) => [member.id, member]));
 		}
 	}
 	async loadTeams() {
 		if (this.teamCache.isStale) {
-			const response = await this.client.listGroups();
-			const groups = response?.data ?? null;
+			const groups = (await this.client.listGroups())?.data ?? null;
 			if (groups) this.teamCache.setMany(groups.map((group) => [group.id, group]));
 		}
 	}
 	async loadWorkflows() {
 		if (this.workflowCache.isStale) {
-			const response = await this.client.listWorkflows();
-			const workflows = response?.data ?? null;
+			const workflows = (await this.client.listWorkflows())?.data ?? null;
 			if (workflows) this.workflowCache.setMany(workflows.map((workflow) => [workflow.id, workflow]));
 		}
 	}
 	async loadCustomFields() {
 		if (this.customFieldCache.isStale) {
-			const response = await this.client.listCustomFields();
-			const customFields = response?.data ?? null;
+			const customFields = (await this.client.listCustomFields())?.data ?? null;
 			if (customFields) this.customFieldCache.setMany(customFields.map((customField) => [customField.id, customField]));
 		}
 	}
 	async getCurrentUser() {
 		if (this.currentUser) return this.currentUser;
-		const response = await this.client.getCurrentMemberInfo();
-		const user$1 = response?.data;
+		const user$1 = (await this.client.getCurrentMemberInfo())?.data;
 		if (!user$1) return null;
 		this.currentUser = user$1;
 		return user$1;
 	}
 	async getUser(userId) {
-		const response = await this.client.getMember(userId, {});
-		const user$1 = response?.data;
+		const user$1 = (await this.client.getMember(userId, {}))?.data;
 		if (!user$1) return null;
 		return user$1;
 	}
@@ -121,8 +115,7 @@ var ShortcutClientWrapper = class {
 	}
 	async listMembers() {
 		await this.loadMembers();
-		const members = Array.from(this.userCache.values());
-		return members;
+		return Array.from(this.userCache.values());
 	}
 	async getWorkflowMap(workflowIds) {
 		await this.loadWorkflows();
@@ -133,23 +126,20 @@ var ShortcutClientWrapper = class {
 		return Array.from(this.workflowCache.values());
 	}
 	async getWorkflow(workflowPublicId) {
-		const response = await this.client.getWorkflow(workflowPublicId);
-		const workflow = response?.data;
+		const workflow = (await this.client.getWorkflow(workflowPublicId))?.data;
 		if (!workflow) return null;
 		return workflow;
 	}
 	async getTeams() {
 		await this.loadTeams();
-		const teams = Array.from(this.teamCache.values());
-		return teams;
+		return Array.from(this.teamCache.values());
 	}
 	async getTeamMap(teamIds) {
 		await this.loadTeams();
 		return new Map(teamIds.map((id) => [id, this.teamCache.get(id)]).filter((team) => team[1] !== null));
 	}
 	async getTeam(teamPublicId) {
-		const response = await this.client.getGroup(teamPublicId);
-		const group = response?.data;
+		const group = (await this.client.getGroup(teamPublicId))?.data;
 		if (!group) return null;
 		return group;
 	}
@@ -166,26 +156,22 @@ var ShortcutClientWrapper = class {
 		return story;
 	}
 	async getStory(storyPublicId) {
-		const response = await this.client.getStory(storyPublicId);
-		const story = response?.data ?? null;
+		const story = (await this.client.getStory(storyPublicId))?.data ?? null;
 		if (!story) return null;
 		return story;
 	}
 	async getEpic(epicPublicId) {
-		const response = await this.client.getEpic(epicPublicId);
-		const epic = response?.data ?? null;
+		const epic = (await this.client.getEpic(epicPublicId))?.data ?? null;
 		if (!epic) return null;
 		return epic;
 	}
 	async getIteration(iterationPublicId) {
-		const response = await this.client.getIteration(iterationPublicId);
-		const iteration = response?.data ?? null;
+		const iteration = (await this.client.getIteration(iterationPublicId))?.data ?? null;
 		if (!iteration) return null;
 		return iteration;
 	}
 	async getMilestone(milestonePublicId) {
-		const response = await this.client.getMilestone(milestonePublicId);
-		const milestone = response?.data ?? null;
+		const milestone = (await this.client.getMilestone(milestonePublicId))?.data ?? null;
 		if (!milestone) return null;
 		return milestone;
 	}
@@ -232,11 +218,10 @@ var ShortcutClientWrapper = class {
 		};
 	}
 	async getActiveIteration(teamIds) {
-		const response = await this.client.listIterations();
-		const iterations = response?.data;
+		const iterations = (await this.client.listIterations())?.data;
 		if (!iterations) return /* @__PURE__ */ new Map();
 		const [today] = (/* @__PURE__ */ new Date()).toISOString().split("T");
-		const activeIterationByTeam = iterations.reduce((acc, iteration) => {
+		return iterations.reduce((acc, iteration) => {
 			if (iteration.status !== "started") return acc;
 			const [startDate] = new Date(iteration.start_date).toISOString().split("T");
 			const [endDate] = new Date(iteration.end_date).toISOString().split("T");
@@ -251,14 +236,12 @@ var ShortcutClientWrapper = class {
 			}
 			return acc;
 		}, /* @__PURE__ */ new Map());
-		return activeIterationByTeam;
 	}
 	async getUpcomingIteration(teamIds) {
-		const response = await this.client.listIterations();
-		const iterations = response?.data;
+		const iterations = (await this.client.listIterations())?.data;
 		if (!iterations) return /* @__PURE__ */ new Map();
 		const [today] = (/* @__PURE__ */ new Date()).toISOString().split("T");
-		const upcomingIterationByTeam = iterations.reduce((acc, iteration) => {
+		return iterations.reduce((acc, iteration) => {
 			if (iteration.status !== "unstarted") return acc;
 			const [startDate] = new Date(iteration.start_date).toISOString().split("T");
 			const [endDate] = new Date(iteration.end_date).toISOString().split("T");
@@ -273,7 +256,6 @@ var ShortcutClientWrapper = class {
 			}
 			return acc;
 		}, /* @__PURE__ */ new Map());
-		return upcomingIterationByTeam;
 	}
 	async searchEpics(query, nextToken) {
 		const response = await this.client.searchEpics({
@@ -318,8 +300,7 @@ var ShortcutClientWrapper = class {
 		};
 	}
 	async listIterationStories(iterationPublicId, includeDescription = false) {
-		const response = await this.client.listIterationStories(iterationPublicId, { includes_description: includeDescription });
-		const stories = response?.data;
+		const stories = (await this.client.listIterationStories(iterationPublicId, { includes_description: includeDescription }))?.data;
 		if (!stories) return {
 			stories: null,
 			total: null
@@ -397,13 +378,11 @@ var ShortcutClientWrapper = class {
 	async removeExternalLinkFromStory(storyPublicId, externalLink) {
 		const story = await this.getStory(storyPublicId);
 		if (!story) throw new Error(`Story ${storyPublicId} not found`);
-		const currentLinks = story.external_links || [];
-		const updatedLinks = currentLinks.filter((link) => link.toLowerCase() !== externalLink.toLowerCase());
+		const updatedLinks = (story.external_links || []).filter((link) => link.toLowerCase() !== externalLink.toLowerCase());
 		return await this.updateStory(storyPublicId, { external_links: updatedLinks });
 	}
 	async getStoriesByExternalLink(externalLink) {
-		const response = await this.client.getExternalLinkStories({ external_link: externalLink.toLowerCase() });
-		const stories = response?.data;
+		const stories = (await this.client.getExternalLinkStories({ external_link: externalLink.toLowerCase() }))?.data;
 		if (!stories) return {
 			stories: null,
 			total: null
@@ -422,6 +401,12 @@ var ShortcutClientWrapper = class {
 		if (!doc) throw new Error(`Failed to create the document: ${response.status}`);
 		return doc;
 	}
+	async updateDoc(docPublicId, params) {
+		const response = await this.client.updateDoc(docPublicId, params);
+		const doc = response?.data ?? null;
+		if (!doc) throw new Error(`Failed to update the document: ${response.status}`);
+		return doc;
+	}
 	async listDocs() {
 		const response = await this.client.listDocs();
 		if (response.status === 403) throw new Error("Docs feature disabled for this workspace.");
@@ -476,12 +461,23 @@ var ShortcutClientWrapper = class {
 		await this.loadCustomFields();
 		return Array.from(this.customFieldCache.values());
 	}
+	async listLabels({ includeArchived = false }) {
+		const allLabels = (await this.client.listLabels({ slim: false }))?.data ?? [];
+		if (includeArchived) return allLabels;
+		return allLabels.filter((label) => !label.archived);
+	}
+	async createLabel(params) {
+		const response = await this.client.createLabel(params);
+		const label = response?.data ?? null;
+		if (!label) throw new Error(`Failed to create the label: ${response.status}`);
+		return label;
+	}
 };
 
 //#endregion
 //#region package.json
 var name = "@shortcut/mcp";
-var version = "0.17.0";
+var version = "0.19.0";
 
 //#endregion
 //#region src/mcp/CustomMcpServer.ts
@@ -527,15 +523,14 @@ var BaseTools = class {
 	}
 	renameEntityProps(entity) {
 		if (!entity || typeof entity !== "object") return entity;
-		const renames = [
+		for (const [from, to] of [
 			["team_id", null],
 			["entity_type", null],
 			["group_id", "team_id"],
 			["group_ids", "team_ids"],
 			["milestone_id", "objective_id"],
 			["milestone_ids", "objective_ids"]
-		];
-		for (const [from, to] of renames) if (from in entity) {
+		]) if (from in entity) {
 			const value = entity[from];
 			delete entity[from];
 			if (to) entity = {
@@ -857,10 +852,15 @@ var BaseTools = class {
 var DocumentTools = class DocumentTools extends BaseTools {
 	static create(client, server) {
 		const tools = new DocumentTools(client);
-		server.addToolWithWriteAccess("documents-create", "Create a new document in Shortcut with a title and content. Returns the document's id, title, and app_url. Note: Use HTML markup for the content (e.g., <p>, <h1>, <ul>, <strong>) rather than Markdown.", {
+		server.addToolWithWriteAccess("documents-create", "Create a new document in Shortcut with a title and content. Returns the document's id, title, and app_url. Note: Use Markdown format for the content.", {
 			title: z.string().max(256).describe("The title for the new document (max 256 characters)"),
-			content: z.string().describe("The content for the new document in HTML format (e.g., <p>Hello</p>, <h1>Title</h1>, <ul><li>Item</li></ul>)")
+			content: z.string().describe("The content for the new document in Markdown format.")
 		}, async ({ title, content }) => await tools.createDocument(title, content));
+		server.addToolWithWriteAccess("documents-update", "Update the content and/or title of an existing document in Shortcut.", {
+			docId: z.string().describe("The ID of the document to retrieve"),
+			title: z.string().max(256).describe("The title for the document (max 256 characters)").optional(),
+			content: z.string().describe("The updated content for the document in Markdown format").optional()
+		}, async ({ docId, content, title }) => await tools.updateDocument(docId, title, content));
 		server.addToolWithReadAccess("documents-list", "List all documents in Shortcut.", async () => await tools.listDocuments());
 		server.addToolWithReadAccess("documents-search", "Find documents.", {
 			nextPageToken: z.string().optional().describe("If a next_page_token was returned from the search result, pass it in to get the next page of results. Should be combined with the original search parameters."),
@@ -881,7 +881,8 @@ var DocumentTools = class DocumentTools extends BaseTools {
 		try {
 			const doc = await this.client.createDoc({
 				title,
-				content
+				content,
+				content_format: "markdown"
 			});
 			return this.toResult("Document created successfully", {
 				id: doc.id,
@@ -893,6 +894,26 @@ var DocumentTools = class DocumentTools extends BaseTools {
 			return this.toResult(`Failed to create document: ${errorMessage}`);
 		}
 	}
+	async updateDocument(docId, title, content) {
+		try {
+			const doc = await this.client.getDocById(docId);
+			if (!doc) return this.toResult(`Document with ID ${docId} not found.`);
+			const result = await this.client.updateDoc(docId, {
+				title: title ?? doc.title ?? "",
+				content: content ?? doc.content_markdown ?? "",
+				content_format: "markdown"
+			});
+			return this.toResult("Document updated successfully", {
+				id: result.id,
+				title: result.title,
+				content: result.content_markdown,
+				app_url: result.app_url
+			});
+		} catch (error) {
+			const errorMessage = error instanceof Error ? error.message : "Unknown error";
+			return this.toResult(`Failed to update document: ${errorMessage}`);
+		}
+	}
 	async listDocuments() {
 		try {
 			const docs = await this.client.listDocs();
@@ -942,7 +963,7 @@ const getKey = (prop) => {
 	return mapKeyName(prop);
 };
 const buildSearchQuery = async (params, currentUser) => {
-	const query = Object.entries(params).map(([key, value]) => {
+	return Object.entries(params).map(([key, value]) => {
 		const q = getKey(key);
 		if (key === "owner" || key === "requester") {
 			if (value === "me") return `${q}:${currentUser?.mention_name || value}`;
@@ -953,7 +974,6 @@ const buildSearchQuery = async (params, currentUser) => {
 		if (typeof value === "string" && value.includes(" ")) return `${q}:"${value}"`;
 		return `${q}:${value}`;
 	}).join(" ");
-	return query;
 };
 
 //#endregion
@@ -1021,7 +1041,7 @@ var EpicTools = class EpicTools extends BaseTools {
 			updated: date(),
 			completed: date(),
 			due: date()
-		}, async ({ nextPageToken,...params }) => await tools.searchEpics(params, nextPageToken));
+		}, async ({ nextPageToken, ...params }) => await tools.searchEpics(params, nextPageToken));
 		server.addToolWithWriteAccess("epics-create", "Create a new Shortcut epic.", {
 			name: z.string().describe("The name of the epic"),
 			owner: z.string().optional().describe("The user ID of the owner of the epic"),
@@ -1031,8 +1051,7 @@ var EpicTools = class EpicTools extends BaseTools {
 		return tools;
 	}
 	async searchEpics(params, nextToken) {
-		const currentUser = await this.client.getCurrentUser();
-		const query = await buildSearchQuery(params, currentUser);
+		const query = await buildSearchQuery(params, await this.client.getCurrentUser());
 		const { epics, total, next_page_token } = await this.client.searchEpics(query, nextToken);
 		if (!epics) throw new Error(`Failed to search for epics matching your query: "${query}"`);
 		if (!epics.length) return this.toResult(`Result: No epics found.`);
@@ -1082,7 +1101,7 @@ var IterationTools = class IterationTools extends BaseTools {
 			updated: date(),
 			startDate: date(),
 			endDate: date()
-		}, async ({ nextPageToken,...params }) => await tools.searchIterations(params, nextPageToken));
+		}, async ({ nextPageToken, ...params }) => await tools.searchIterations(params, nextPageToken));
 		server.addToolWithWriteAccess("iterations-create", "Create a new Shortcut iteration", {
 			name: z.string().describe("The name of the iteration"),
 			startDate: z.string().describe("The start date of the iteration in YYYY-MM-DD format"),
@@ -1100,8 +1119,7 @@ var IterationTools = class IterationTools extends BaseTools {
 		return this.toResult(`Result (${stories.length} stories found):`, await this.entitiesWithRelatedEntities(stories, "stories"));
 	}
 	async searchIterations(params, nextToken) {
-		const currentUser = await this.client.getCurrentUser();
-		const query = await buildSearchQuery(params, currentUser);
+		const query = await buildSearchQuery(params, await this.client.getCurrentUser());
 		const { iterations, total, next_page_token } = await this.client.searchIterations(query, nextToken);
 		if (!iterations) throw new Error(`Failed to search for iterations matching your query: "${query}".`);
 		if (!iterations.length) return this.toResult(`Result: No iterations found.`);
@@ -1125,41 +1143,33 @@ var IterationTools = class IterationTools extends BaseTools {
 	}
 	async getActiveIterations(teamId) {
 		if (teamId) {
-			const team = await this.client.getTeam(teamId);
-			if (!team) throw new Error(`No team found matching id: "${teamId}"`);
-			const result = await this.client.getActiveIteration([teamId]);
-			const iterations = result.get(teamId);
+			if (!await this.client.getTeam(teamId)) throw new Error(`No team found matching id: "${teamId}"`);
+			const iterations = (await this.client.getActiveIteration([teamId])).get(teamId);
 			if (!iterations?.length) return this.toResult(`Result: No active iterations found for team.`);
 			if (iterations.length === 1) return this.toResult("The active iteration for the team is:", await this.entityWithRelatedEntities(iterations[0], "iteration"));
 			return this.toResult("The active iterations for the team are:", await this.entitiesWithRelatedEntities(iterations, "iterations"));
 		}
 		const currentUser = await this.client.getCurrentUser();
 		if (!currentUser) throw new Error("Failed to retrieve current user.");
-		const teams = await this.client.getTeams();
-		const teamIds = teams.filter((team) => team.member_ids.includes(currentUser.id)).map((team) => team.id);
+		const teamIds = (await this.client.getTeams()).filter((team) => team.member_ids.includes(currentUser.id)).map((team) => team.id);
 		if (!teamIds.length) throw new Error("Current user does not belong to any teams.");
-		const resultsByTeam = await this.client.getActiveIteration(teamIds);
-		const allActiveIterations = [...resultsByTeam.values()].flat();
+		const allActiveIterations = [...(await this.client.getActiveIteration(teamIds)).values()].flat();
 		if (!allActiveIterations.length) return this.toResult("Result: No active iterations found for any of your teams.");
 		return this.toResult(`You have ${allActiveIterations.length} active iterations for your teams:`, await this.entitiesWithRelatedEntities(allActiveIterations, "iterations"));
 	}
 	async getUpcomingIterations(teamId) {
 		if (teamId) {
-			const team = await this.client.getTeam(teamId);
-			if (!team) throw new Error(`No team found matching id: "${teamId}"`);
-			const result = await this.client.getUpcomingIteration([teamId]);
-			const iterations = result.get(teamId);
+			if (!await this.client.getTeam(teamId)) throw new Error(`No team found matching id: "${teamId}"`);
+			const iterations = (await this.client.getUpcomingIteration([teamId])).get(teamId);
 			if (!iterations?.length) return this.toResult(`Result: No upcoming iterations found for team.`);
 			if (iterations.length === 1) return this.toResult("The next upcoming iteration for the team is:", await this.entityWithRelatedEntities(iterations[0], "iteration"));
 			return this.toResult("The next upcoming iterations for the team are:", await this.entitiesWithRelatedEntities(iterations, "iterations"));
 		}
 		const currentUser = await this.client.getCurrentUser();
 		if (!currentUser) throw new Error("Failed to retrieve current user.");
-		const teams = await this.client.getTeams();
-		const teamIds = teams.filter((team) => team.member_ids.includes(currentUser.id)).map((team) => team.id);
+		const teamIds = (await this.client.getTeams()).filter((team) => team.member_ids.includes(currentUser.id)).map((team) => team.id);
 		if (!teamIds.length) throw new Error("Current user does not belong to any teams.");
-		const resultsByTeam = await this.client.getUpcomingIteration(teamIds);
-		const allUpcomingIterations = [...resultsByTeam.values()].flat();
+		const allUpcomingIterations = [...(await this.client.getUpcomingIteration(teamIds)).values()].flat();
 		if (!allUpcomingIterations.length) return this.toResult("Result: No upcoming iterations found for any of your teams.");
 		return this.toResult("The upcoming iterations for all your teams are:", await this.entitiesWithRelatedEntities(allUpcomingIterations, "iterations"));
 	}
@@ -1195,12 +1205,11 @@ var ObjectiveTools = class ObjectiveTools extends BaseTools {
 			created: date(),
 			updated: date(),
 			completed: date()
-		}, async ({ nextPageToken,...params }) => await tools.searchObjectives(params, nextPageToken));
+		}, async ({ nextPageToken, ...params }) => await tools.searchObjectives(params, nextPageToken));
 		return tools;
 	}
 	async searchObjectives(params, nextToken) {
-		const currentUser = await this.client.getCurrentUser();
-		const query = await buildSearchQuery(params, currentUser);
+		const query = await buildSearchQuery(params, await this.client.getCurrentUser());
 		const { milestones, total, next_page_token } = await this.client.searchMilestones(query, nextToken);
 		if (!milestones) throw new Error(`Failed to search for milestones matching your query: "${query}"`);
 		if (!milestones.length) return this.toResult(`Result: No milestones found.`);
@@ -1272,7 +1281,7 @@ var StoryTools = class StoryTools extends BaseTools {
 			updated: date(),
 			completed: date(),
 			due: date()
-		}, async ({ nextPageToken,...params }) => await tools.searchStories(params, nextPageToken));
+		}, async ({ nextPageToken, ...params }) => await tools.searchStories(params, nextPageToken));
 		server.addToolWithReadAccess("stories-get-branch-name", "Get a valid branch name for a specific story.", { storyPublicId: z.number().positive().describe("The public Id of the story") }, async ({ storyPublicId }) => await tools.getStoryBranchName(storyPublicId));
 		server.addToolWithWriteAccess("stories-create", `Create a new Shortcut story. 
 Name is required, and either a Team or Workflow must be specified:
@@ -1411,10 +1420,7 @@ The story will be added to the default state for the workflow.
 	}
 	async createStory({ name: name$1, description, type, owner, epic, iteration, team, workflow }) {
 		if (!workflow && !team) throw new Error("Team or Workflow has to be specified");
-		if (!workflow && team) {
-			const fullTeam = await this.client.getTeam(team);
-			workflow = fullTeam?.workflow_ids?.[0];
-		}
+		if (!workflow && team) workflow = (await this.client.getTeam(team))?.workflow_ids?.[0];
 		if (!workflow) throw new Error("Failed to find workflow for team");
 		const fullWorkflow = await this.client.getWorkflow(workflow);
 		if (!fullWorkflow) throw new Error("Failed to find workflow");
@@ -1453,23 +1459,19 @@ The story will be added to the default state for the workflow.
 	async addStoryAsSubTask({ parentStoryPublicId, subTaskPublicId }) {
 		if (!parentStoryPublicId) throw new Error("ID of parent story is required");
 		if (!subTaskPublicId) throw new Error("ID of sub-task story is required");
-		const subTask = await this.client.getStory(subTaskPublicId);
-		if (!subTask) throw new Error(`Failed to retrieve story with public ID: ${subTaskPublicId}`);
-		const parentStory = await this.client.getStory(parentStoryPublicId);
-		if (!parentStory) throw new Error(`Failed to retrieve parent story with public ID: ${parentStoryPublicId}`);
+		if (!await this.client.getStory(subTaskPublicId)) throw new Error(`Failed to retrieve story with public ID: ${subTaskPublicId}`);
+		if (!await this.client.getStory(parentStoryPublicId)) throw new Error(`Failed to retrieve parent story with public ID: ${parentStoryPublicId}`);
 		await this.client.updateStory(subTaskPublicId, { parent_story_id: parentStoryPublicId });
 		return this.toResult(`Added story sc-${subTaskPublicId} as a sub-task of sc-${parentStoryPublicId}`);
 	}
 	async removeSubTaskFromParent({ subTaskPublicId }) {
 		if (!subTaskPublicId) throw new Error("ID of sub-task story is required");
-		const subTask = await this.client.getStory(subTaskPublicId);
-		if (!subTask) throw new Error(`Failed to retrieve story with public ID: ${subTaskPublicId}`);
+		if (!await this.client.getStory(subTaskPublicId)) throw new Error(`Failed to retrieve story with public ID: ${subTaskPublicId}`);
 		await this.client.updateStory(subTaskPublicId, { parent_story_id: null });
 		return this.toResult(`Removed story sc-${subTaskPublicId} from its parent story`);
 	}
 	async searchStories(params, nextToken) {
-		const currentUser = await this.client.getCurrentUser();
-		const query = await buildSearchQuery(params, currentUser);
+		const query = await buildSearchQuery(params, await this.client.getCurrentUser());
 		const { stories, total, next_page_token } = await this.client.searchStories(query, nextToken);
 		if (!stories) throw new Error(`Failed to search for stories matching your query: "${query}".`);
 		if (!stories.length) return this.toResult(`Result: No stories found.`);
@@ -1483,15 +1485,13 @@ The story will be added to the default state for the workflow.
 	async createStoryComment({ storyPublicId, text }) {
 		if (!storyPublicId) throw new Error("Story public ID is required");
 		if (!text) throw new Error("Story comment text is required");
-		const story = await this.client.getStory(storyPublicId);
-		if (!story) throw new Error(`Failed to retrieve Shortcut story with public ID: ${storyPublicId}`);
+		if (!await this.client.getStory(storyPublicId)) throw new Error(`Failed to retrieve Shortcut story with public ID: ${storyPublicId}`);
 		const storyComment = await this.client.createStoryComment(storyPublicId, { text });
 		return this.toResult(`Created comment on story sc-${storyPublicId}. Comment URL: ${storyComment.app_url}.`);
 	}
-	async updateStory({ storyPublicId,...updates }) {
+	async updateStory({ storyPublicId, ...updates }) {
 		if (!storyPublicId) throw new Error("Story public ID is required");
-		const story = await this.client.getStory(storyPublicId);
-		if (!story) throw new Error(`Failed to retrieve Shortcut story with public ID: ${storyPublicId}`);
+		if (!await this.client.getStory(storyPublicId)) throw new Error(`Failed to retrieve Shortcut story with public ID: ${storyPublicId}`);
 		const updateParams = {};
 		if (updates.name !== void 0) updateParams.name = updates.name;
 		if (updates.description !== void 0) updateParams.description = updates.description;
@@ -1508,8 +1508,7 @@ The story will be added to the default state for the workflow.
 	async uploadFileToStory(storyPublicId, filePath) {
 		if (!storyPublicId) throw new Error("Story public ID is required");
 		if (!filePath) throw new Error("File path is required");
-		const story = await this.client.getStory(storyPublicId);
-		if (!story) throw new Error(`Failed to retrieve Shortcut story with public ID: ${storyPublicId}`);
+		if (!await this.client.getStory(storyPublicId)) throw new Error(`Failed to retrieve Shortcut story with public ID: ${storyPublicId}`);
 		const uploadedFile = await this.client.uploadFile(storyPublicId, filePath);
 		if (!uploadedFile) throw new Error(`Failed to upload file to story sc-${storyPublicId}`);
 		return this.toResult(`Uploaded file "${uploadedFile.name}" to story sc-${storyPublicId}. File ID is: ${uploadedFile.id}`);
@@ -1517,11 +1516,9 @@ The story will be added to the default state for the workflow.
 	async addTaskToStory({ storyPublicId, taskDescription, taskOwnerIds }) {
 		if (!storyPublicId) throw new Error("Story public ID is required");
 		if (!taskDescription) throw new Error("Task description is required");
-		const story = await this.client.getStory(storyPublicId);
-		if (!story) throw new Error(`Failed to retrieve Shortcut story with public ID: ${storyPublicId}`);
+		if (!await this.client.getStory(storyPublicId)) throw new Error(`Failed to retrieve Shortcut story with public ID: ${storyPublicId}`);
 		if (taskOwnerIds?.length) {
-			const owners = await this.client.getUserMap(taskOwnerIds);
-			if (!owners) throw new Error(`Failed to retrieve users with IDs: ${taskOwnerIds.join(", ")}`);
+			if (!await this.client.getUserMap(taskOwnerIds)) throw new Error(`Failed to retrieve users with IDs: ${taskOwnerIds.join(", ")}`);
 		}
 		const task = await this.client.addTaskToStory(storyPublicId, {
 			description: taskDescription,
@@ -1532,10 +1529,8 @@ The story will be added to the default state for the workflow.
 	async updateTask({ storyPublicId, taskPublicId, taskDescription, taskOwnerIds, isCompleted }) {
 		if (!storyPublicId) throw new Error("Story public ID is required");
 		if (!taskPublicId) throw new Error("Task public ID is required");
-		const story = await this.client.getStory(storyPublicId);
-		if (!story) throw new Error(`Failed to retrieve Shortcut story with public ID: ${storyPublicId}`);
-		const task = await this.client.getTask(storyPublicId, taskPublicId);
-		if (!task) throw new Error(`Failed to retrieve Shortcut task with public ID: ${taskPublicId}`);
+		if (!await this.client.getStory(storyPublicId)) throw new Error(`Failed to retrieve Shortcut story with public ID: ${storyPublicId}`);
+		if (!await this.client.getTask(storyPublicId, taskPublicId)) throw new Error(`Failed to retrieve Shortcut task with public ID: ${taskPublicId}`);
 		const updatedTask = await this.client.updateTask(storyPublicId, taskPublicId, {
 			description: taskDescription,
 			ownerIds: taskOwnerIds,
@@ -1548,10 +1543,8 @@ The story will be added to the default state for the workflow.
 	async addRelationToStory({ storyPublicId, relatedStoryPublicId, relationshipType }) {
 		if (!storyPublicId) throw new Error("Story public ID is required");
 		if (!relatedStoryPublicId) throw new Error("Related story public ID is required");
-		const story = await this.client.getStory(storyPublicId);
-		if (!story) throw new Error(`Failed to retrieve Shortcut story with public ID: ${storyPublicId}`);
-		const relatedStory = await this.client.getStory(relatedStoryPublicId);
-		if (!relatedStory) throw new Error(`Failed to retrieve Shortcut story with public ID: ${relatedStoryPublicId}`);
+		if (!await this.client.getStory(storyPublicId)) throw new Error(`Failed to retrieve Shortcut story with public ID: ${storyPublicId}`);
+		if (!await this.client.getStory(relatedStoryPublicId)) throw new Error(`Failed to retrieve Shortcut story with public ID: ${relatedStoryPublicId}`);
 		let subjectStoryId = storyPublicId;
 		let objectStoryId = relatedStoryPublicId;
 		if (relationshipType === "blocked by" || relationshipType === "duplicated by") {
@@ -1688,4 +1681,4 @@ var WorkflowTools = class WorkflowTools extends BaseTools {
 };
 
 //#endregion
-export { CustomMcpServer, DocumentTools, EpicTools, IterationTools, ObjectiveTools, ShortcutClientWrapper, StoryTools, TeamTools, UserTools, WorkflowTools };
+export { ObjectiveTools as a, DocumentTools as c, ShortcutClientWrapper as d, StoryTools as i, BaseTools as l, UserTools as n, IterationTools as o, TeamTools as r, EpicTools as s, WorkflowTools as t, CustomMcpServer as u };

package/package.json

@@ -12,7 +12,7 @@
 		"modelcontextprotocol"
 	],
 	"license": "MIT",
-	"version": "0.17.0",
+	"version": "0.19.0",
 	"type": "module",
 	"main": "dist/index.js",
 	"bin": {
@@ -35,7 +35,7 @@
 	},
 	"dependencies": {
 		"@modelcontextprotocol/sdk": "^1.25.1",
-		"@shortcut/client": "^3.1.0",
+		"@shortcut/client": "^3.2.0",
 		"express": "^4.18.2",
 		"pino": "^9.5.0",
 		"pino-http": "^10.3.0",