refactor: drop skill file, rework client
This commit was merged in pull request #3.
This commit is contained in:
+106
-53
@@ -2,7 +2,8 @@ import { type McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
|
||||
import fs from "node:fs";
|
||||
import path from "node:path";
|
||||
import { z } from "zod/v4";
|
||||
import { EwsClient, loadConfig } from "../ews_client.ts";
|
||||
import { loadConfig } from "../utils/login.ts";
|
||||
import { EwsClient } from "../client/ews_client.ts";
|
||||
|
||||
function getClient(): EwsClient {
|
||||
return new EwsClient(loadConfig());
|
||||
@@ -12,31 +13,41 @@ export function registerEmailTools(server: McpServer): void {
|
||||
server.registerTool(
|
||||
"get_emails",
|
||||
{
|
||||
description: "Get emails from a mailbox folder",
|
||||
description: "Get emails from a mailbox folder."
|
||||
+ " Supports optional text search via query/queryScope.",
|
||||
inputSchema: z.object({
|
||||
folder: z.string().default("Inbox").describe(
|
||||
"Folder name (Inbox, Sent, Drafts, Deleted, Junk, or custom)",
|
||||
"Folder name (Inbox, Sent, Drafts, Deleted, Junk, or custom)."
|
||||
+ " Supports Russian folder names",
|
||||
),
|
||||
limit: z.number().default(10).describe(
|
||||
"Maximum number of emails to return (default 10, max 50)",
|
||||
"Maximum number of emails to return",
|
||||
),
|
||||
offset: z.number().default(0).describe(
|
||||
"Number of emails to skip for pagination",
|
||||
),
|
||||
includeBody: z.boolean().default(false).describe(
|
||||
"If True, fetch full body for each email (slower)",
|
||||
"If true, fetches full email body for each email (slower)",
|
||||
),
|
||||
unreadOnly: z.boolean().default(false).describe(
|
||||
"If True, only return unread emails",
|
||||
),
|
||||
idsOnly: z.boolean().default(false).describe(
|
||||
"If True, return only item IDs and dates (max limit 500)",
|
||||
"If true, only return unread emails",
|
||||
),
|
||||
query: z.string().optional().describe(
|
||||
"Optional text to search for within the folder",
|
||||
),
|
||||
queryScope: z.enum(["all", "subject", "body", "from"]).default("all")
|
||||
.describe("Where to search (only used when query is set)"),
|
||||
.describe("Search scope: all, subject, body, or from"),
|
||||
}),
|
||||
outputSchema: z.object({
|
||||
emails: z.array(z.object({
|
||||
itemId: z.string(),
|
||||
subject: z.string(),
|
||||
from: z.string(),
|
||||
date: z.string(),
|
||||
isRead: z.boolean(),
|
||||
hasAttachments: z.boolean(),
|
||||
})),
|
||||
count: z.number(),
|
||||
}),
|
||||
},
|
||||
async (
|
||||
@@ -46,19 +57,18 @@ export function registerEmailTools(server: McpServer): void {
|
||||
offset,
|
||||
includeBody,
|
||||
unreadOnly,
|
||||
idsOnly,
|
||||
query,
|
||||
queryScope,
|
||||
},
|
||||
) => {
|
||||
try {
|
||||
const client = getClient();
|
||||
const maxLimit = idsOnly ? 500 : 50;
|
||||
const maxLimit = 50;
|
||||
if (limit > maxLimit) limit = maxLimit;
|
||||
offset = Math.max(0, offset);
|
||||
|
||||
const folderId = await client.getFolderId(folder);
|
||||
const baseShape = idsOnly ? "IdOnly" : "AllProperties";
|
||||
const baseShape = "AllProperties";
|
||||
|
||||
if (query && !query.trim()) query = undefined;
|
||||
|
||||
@@ -109,16 +119,16 @@ export function registerEmailTools(server: McpServer): void {
|
||||
let restriction = "";
|
||||
if (restrictions.length === 1) {
|
||||
restriction = `\
|
||||
<m:Restriction>
|
||||
${restrictions[0]}
|
||||
</m:Restriction>`;
|
||||
<m:Restriction>
|
||||
${restrictions[0]}
|
||||
</m:Restriction>`;
|
||||
} else if (restrictions.length > 1) {
|
||||
restriction = `\
|
||||
<m:Restriction>
|
||||
<t:And>
|
||||
${restrictions.join("\n")}
|
||||
</t:And>
|
||||
</m:Restriction>`;
|
||||
<m:Restriction>
|
||||
<t:And>
|
||||
${restrictions.join("\n")}
|
||||
</t:And>
|
||||
</m:Restriction>`;
|
||||
}
|
||||
|
||||
const items = await client.findItems(folderId, {
|
||||
@@ -128,33 +138,9 @@ export function registerEmailTools(server: McpServer): void {
|
||||
restriction,
|
||||
});
|
||||
|
||||
if (idsOnly) {
|
||||
const result = items.map((item: any) => ({
|
||||
itemId: item.ItemId?.["@_Id"] ?? "",
|
||||
date: item.DateTimeReceived ?? "",
|
||||
subject: item.Subject ?? "",
|
||||
}));
|
||||
return {
|
||||
content: [{
|
||||
type: "text" as const,
|
||||
text: JSON.stringify({ itemIds: result, count: result.length }),
|
||||
}],
|
||||
};
|
||||
}
|
||||
|
||||
const emails = [];
|
||||
for (const item of items) {
|
||||
const email = client.extractEmailSummary(item);
|
||||
|
||||
if (includeBody && email.itemId) {
|
||||
const details = client.extractEmailDetails(item);
|
||||
email.to = details.to;
|
||||
email.cc = details.cc;
|
||||
email.body = details.body;
|
||||
email.hasLinks = details.hasLinks;
|
||||
}
|
||||
|
||||
emails.push(email);
|
||||
emails.push(client.extractEmail(item));
|
||||
}
|
||||
|
||||
return {
|
||||
@@ -162,6 +148,7 @@ export function registerEmailTools(server: McpServer): void {
|
||||
type: "text" as const,
|
||||
text: JSON.stringify({ emails, count: emails.length }),
|
||||
}],
|
||||
structuredContent: { emails, count: emails.length },
|
||||
};
|
||||
} catch (error: any) {
|
||||
return {
|
||||
@@ -177,21 +164,43 @@ export function registerEmailTools(server: McpServer): void {
|
||||
server.registerTool(
|
||||
"get_email",
|
||||
{
|
||||
description: "Get a single email with full body and details",
|
||||
description:
|
||||
"Get a single email with full body and details by Exchange ItemId.",
|
||||
inputSchema: z.object({
|
||||
itemId: z.string().describe(
|
||||
"The Exchange ItemId of the email to retrieve",
|
||||
),
|
||||
}),
|
||||
outputSchema: z.object({
|
||||
itemId: z.string(),
|
||||
subject: z.string(),
|
||||
from: z.string(),
|
||||
fromName: z.string().optional(),
|
||||
to: z.array(z.string()).optional(),
|
||||
cc: z.array(z.string()).optional(),
|
||||
date: z.string(),
|
||||
body: z.string().optional(),
|
||||
isRead: z.boolean().optional(),
|
||||
hasAttachments: z.boolean().optional(),
|
||||
hasLinks: z.boolean().optional(),
|
||||
attachments: z.array(z.object({
|
||||
name: z.string(),
|
||||
size: z.number(),
|
||||
contentType: z.string(),
|
||||
attachmentId: z.string().optional(),
|
||||
isInline: z.boolean().optional(),
|
||||
})).optional(),
|
||||
}),
|
||||
},
|
||||
async ({ itemId }) => {
|
||||
try {
|
||||
const client = getClient();
|
||||
const item = await client.getItem(itemId);
|
||||
const email = client.extractEmailDetails(item);
|
||||
const email = client.extractEmail(item);
|
||||
|
||||
return {
|
||||
content: [{ type: "text" as const, text: JSON.stringify(email) }],
|
||||
structuredContent: { emails: [email] },
|
||||
};
|
||||
} catch (error: any) {
|
||||
return {
|
||||
@@ -207,15 +216,21 @@ export function registerEmailTools(server: McpServer): void {
|
||||
server.registerTool(
|
||||
"mark_email_read",
|
||||
{
|
||||
description: "Mark one or more emails as read or unread",
|
||||
description:
|
||||
"Mark one or more emails as read or unread by their Exchange ItemIds.",
|
||||
inputSchema: z.object({
|
||||
itemIds: z.array(z.string()).describe(
|
||||
"List of Exchange ItemIds to update",
|
||||
),
|
||||
isRead: z.boolean().default(true).describe(
|
||||
"True to mark as read, False to mark as unread",
|
||||
"True to mark as read, false to mark as unread",
|
||||
),
|
||||
}),
|
||||
outputSchema: z.object({
|
||||
success: z.boolean(),
|
||||
message: z.string().optional(),
|
||||
error: z.string().optional(),
|
||||
}),
|
||||
},
|
||||
async ({ itemIds, isRead }) => {
|
||||
try {
|
||||
@@ -234,6 +249,10 @@ export function registerEmailTools(server: McpServer): void {
|
||||
type: "text" as const,
|
||||
text: JSON.stringify({ error: errors.join("; ") }),
|
||||
}],
|
||||
structuredContent: {
|
||||
success: false,
|
||||
error: errors.join("; "),
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
@@ -246,6 +265,10 @@ export function registerEmailTools(server: McpServer): void {
|
||||
message: `Marked ${itemIds.length} email(s) as ${status}.`,
|
||||
}),
|
||||
}],
|
||||
structuredContent: {
|
||||
success: true,
|
||||
message: `Marked ${itemIds.length} email(s) as ${status}.`,
|
||||
},
|
||||
};
|
||||
} catch (error: any) {
|
||||
return {
|
||||
@@ -253,6 +276,10 @@ export function registerEmailTools(server: McpServer): void {
|
||||
type: "text" as const,
|
||||
text: JSON.stringify({ error: error.message ?? String(error) }),
|
||||
}],
|
||||
structuredContent: {
|
||||
success: false,
|
||||
error: error.message ?? String(error),
|
||||
},
|
||||
};
|
||||
}
|
||||
},
|
||||
@@ -261,19 +288,36 @@ export function registerEmailTools(server: McpServer): void {
|
||||
server.registerTool(
|
||||
"download_attachments",
|
||||
{
|
||||
description: "Download all file attachments from an email to disk",
|
||||
description: "Download all file attachments from an email to disk."
|
||||
+ " Inline (embedded) attachments are skipped —"
|
||||
+ " only standalone file attachments are downloaded.",
|
||||
inputSchema: z.object({
|
||||
itemId: z.string().describe("The Exchange ItemId of the email"),
|
||||
targetFolder: z.string().default("/tmp/attachments").describe(
|
||||
"Local directory to save files (default /tmp/attachments)",
|
||||
"Local directory to save files",
|
||||
),
|
||||
}),
|
||||
outputSchema: z.object({
|
||||
success: z.boolean(),
|
||||
downloaded: z.array(z.object({
|
||||
name: z.string(),
|
||||
path: z.string(),
|
||||
size: z.number(),
|
||||
contentType: z.string(),
|
||||
})),
|
||||
count: z.number(),
|
||||
errors: z.array(z.object({
|
||||
name: z.string(),
|
||||
error: z.string(),
|
||||
})).optional(),
|
||||
message: z.string().optional(),
|
||||
}),
|
||||
},
|
||||
async ({ itemId, targetFolder }) => {
|
||||
try {
|
||||
const client = new EwsClient(loadConfig());
|
||||
const item = await client.getItem(itemId);
|
||||
const email = client.extractEmailDetails(item);
|
||||
const email = client.extractEmail(item);
|
||||
const attachments = email.attachments ?? [];
|
||||
|
||||
const fileAttachments = attachments.filter(
|
||||
@@ -291,6 +335,12 @@ export function registerEmailTools(server: McpServer): void {
|
||||
message: "No downloadable file attachments.",
|
||||
}),
|
||||
}],
|
||||
structuredContent: {
|
||||
success: true,
|
||||
downloaded: [],
|
||||
count: 0,
|
||||
message: "No downloadable file attachments.",
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
@@ -348,6 +398,9 @@ export function registerEmailTools(server: McpServer): void {
|
||||
...(errors.length ? { errors } : {}),
|
||||
}),
|
||||
}],
|
||||
structuredContent: errors.length === 0
|
||||
? { success: true, downloaded, count: downloaded.length }
|
||||
: { success: false, downloaded, count: downloaded.length, errors },
|
||||
};
|
||||
} catch (error: any) {
|
||||
return {
|
||||
|
||||
Reference in New Issue
Block a user