244 lines
7.3 KiB
TypeScript
244 lines
7.3 KiB
TypeScript
/**
|
|
* X Integration - MCP Tool Definitions (Agent/Container Side)
|
|
*
|
|
* These tools run inside the container and communicate with the host via IPC.
|
|
* The host-side implementation is in host.ts.
|
|
*
|
|
* Note: This file is compiled in the container, not on the host.
|
|
* The @ts-ignore is needed because the SDK is only available in the container.
|
|
*/
|
|
|
|
// @ts-ignore - SDK available in container environment only
|
|
import { tool } from '@anthropic-ai/claude-agent-sdk';
|
|
import { z } from 'zod';
|
|
import fs from 'fs';
|
|
import path from 'path';
|
|
|
|
// IPC directories (inside container)
|
|
const IPC_DIR = '/workspace/ipc';
|
|
const TASKS_DIR = path.join(IPC_DIR, 'tasks');
|
|
const RESULTS_DIR = path.join(IPC_DIR, 'x_results');
|
|
|
|
function writeIpcFile(dir: string, data: object): string {
|
|
fs.mkdirSync(dir, { recursive: true });
|
|
const filename = `${Date.now()}-${Math.random().toString(36).slice(2, 8)}.json`;
|
|
const filepath = path.join(dir, filename);
|
|
const tempPath = `${filepath}.tmp`;
|
|
fs.writeFileSync(tempPath, JSON.stringify(data, null, 2));
|
|
fs.renameSync(tempPath, filepath);
|
|
return filename;
|
|
}
|
|
|
|
async function waitForResult(requestId: string, maxWait = 60000): Promise<{ success: boolean; message: string }> {
|
|
const resultFile = path.join(RESULTS_DIR, `${requestId}.json`);
|
|
const pollInterval = 1000;
|
|
let elapsed = 0;
|
|
|
|
while (elapsed < maxWait) {
|
|
if (fs.existsSync(resultFile)) {
|
|
try {
|
|
const result = JSON.parse(fs.readFileSync(resultFile, 'utf-8'));
|
|
fs.unlinkSync(resultFile);
|
|
return result;
|
|
} catch (err) {
|
|
return { success: false, message: `Failed to read result: ${err}` };
|
|
}
|
|
}
|
|
await new Promise(resolve => setTimeout(resolve, pollInterval));
|
|
elapsed += pollInterval;
|
|
}
|
|
|
|
return { success: false, message: 'Request timed out' };
|
|
}
|
|
|
|
export interface SkillToolsContext {
|
|
groupFolder: string;
|
|
isMain: boolean;
|
|
}
|
|
|
|
/**
|
|
* Create X integration MCP tools
|
|
*/
|
|
export function createXTools(ctx: SkillToolsContext) {
|
|
const { groupFolder, isMain } = ctx;
|
|
|
|
return [
|
|
tool(
|
|
'x_post',
|
|
`Post a tweet to X (Twitter). Main group only.
|
|
|
|
The host machine will execute the browser automation to post the tweet.
|
|
Make sure the content is appropriate and within X's character limit (280 chars for text).`,
|
|
{
|
|
content: z.string().max(280).describe('The tweet content to post (max 280 characters)')
|
|
},
|
|
async (args: { content: string }) => {
|
|
if (!isMain) {
|
|
return {
|
|
content: [{ type: 'text', text: 'Only the main group can post tweets.' }],
|
|
isError: true
|
|
};
|
|
}
|
|
|
|
if (args.content.length > 280) {
|
|
return {
|
|
content: [{ type: 'text', text: `Tweet exceeds 280 character limit (current: ${args.content.length})` }],
|
|
isError: true
|
|
};
|
|
}
|
|
|
|
const requestId = `xpost-${Date.now()}-${Math.random().toString(36).slice(2, 8)}`;
|
|
writeIpcFile(TASKS_DIR, {
|
|
type: 'x_post',
|
|
requestId,
|
|
content: args.content,
|
|
groupFolder,
|
|
timestamp: new Date().toISOString()
|
|
});
|
|
|
|
const result = await waitForResult(requestId);
|
|
return {
|
|
content: [{ type: 'text', text: result.message }],
|
|
isError: !result.success
|
|
};
|
|
}
|
|
),
|
|
|
|
tool(
|
|
'x_like',
|
|
`Like a tweet on X (Twitter). Main group only.
|
|
|
|
Provide the tweet URL or tweet ID to like.`,
|
|
{
|
|
tweet_url: z.string().describe('The tweet URL (e.g., https://x.com/user/status/123) or tweet ID')
|
|
},
|
|
async (args: { tweet_url: string }) => {
|
|
if (!isMain) {
|
|
return {
|
|
content: [{ type: 'text', text: 'Only the main group can interact with X.' }],
|
|
isError: true
|
|
};
|
|
}
|
|
|
|
const requestId = `xlike-${Date.now()}-${Math.random().toString(36).slice(2, 8)}`;
|
|
writeIpcFile(TASKS_DIR, {
|
|
type: 'x_like',
|
|
requestId,
|
|
tweetUrl: args.tweet_url,
|
|
groupFolder,
|
|
timestamp: new Date().toISOString()
|
|
});
|
|
|
|
const result = await waitForResult(requestId);
|
|
return {
|
|
content: [{ type: 'text', text: result.message }],
|
|
isError: !result.success
|
|
};
|
|
}
|
|
),
|
|
|
|
tool(
|
|
'x_reply',
|
|
`Reply to a tweet on X (Twitter). Main group only.
|
|
|
|
Provide the tweet URL and your reply content.`,
|
|
{
|
|
tweet_url: z.string().describe('The tweet URL (e.g., https://x.com/user/status/123) or tweet ID'),
|
|
content: z.string().max(280).describe('The reply content (max 280 characters)')
|
|
},
|
|
async (args: { tweet_url: string; content: string }) => {
|
|
if (!isMain) {
|
|
return {
|
|
content: [{ type: 'text', text: 'Only the main group can interact with X.' }],
|
|
isError: true
|
|
};
|
|
}
|
|
|
|
const requestId = `xreply-${Date.now()}-${Math.random().toString(36).slice(2, 8)}`;
|
|
writeIpcFile(TASKS_DIR, {
|
|
type: 'x_reply',
|
|
requestId,
|
|
tweetUrl: args.tweet_url,
|
|
content: args.content,
|
|
groupFolder,
|
|
timestamp: new Date().toISOString()
|
|
});
|
|
|
|
const result = await waitForResult(requestId);
|
|
return {
|
|
content: [{ type: 'text', text: result.message }],
|
|
isError: !result.success
|
|
};
|
|
}
|
|
),
|
|
|
|
tool(
|
|
'x_retweet',
|
|
`Retweet a tweet on X (Twitter). Main group only.
|
|
|
|
Provide the tweet URL to retweet.`,
|
|
{
|
|
tweet_url: z.string().describe('The tweet URL (e.g., https://x.com/user/status/123) or tweet ID')
|
|
},
|
|
async (args: { tweet_url: string }) => {
|
|
if (!isMain) {
|
|
return {
|
|
content: [{ type: 'text', text: 'Only the main group can interact with X.' }],
|
|
isError: true
|
|
};
|
|
}
|
|
|
|
const requestId = `xretweet-${Date.now()}-${Math.random().toString(36).slice(2, 8)}`;
|
|
writeIpcFile(TASKS_DIR, {
|
|
type: 'x_retweet',
|
|
requestId,
|
|
tweetUrl: args.tweet_url,
|
|
groupFolder,
|
|
timestamp: new Date().toISOString()
|
|
});
|
|
|
|
const result = await waitForResult(requestId);
|
|
return {
|
|
content: [{ type: 'text', text: result.message }],
|
|
isError: !result.success
|
|
};
|
|
}
|
|
),
|
|
|
|
tool(
|
|
'x_quote',
|
|
`Quote tweet on X (Twitter). Main group only.
|
|
|
|
Retweet with your own comment added.`,
|
|
{
|
|
tweet_url: z.string().describe('The tweet URL (e.g., https://x.com/user/status/123) or tweet ID'),
|
|
comment: z.string().max(280).describe('Your comment for the quote tweet (max 280 characters)')
|
|
},
|
|
async (args: { tweet_url: string; comment: string }) => {
|
|
if (!isMain) {
|
|
return {
|
|
content: [{ type: 'text', text: 'Only the main group can interact with X.' }],
|
|
isError: true
|
|
};
|
|
}
|
|
|
|
const requestId = `xquote-${Date.now()}-${Math.random().toString(36).slice(2, 8)}`;
|
|
writeIpcFile(TASKS_DIR, {
|
|
type: 'x_quote',
|
|
requestId,
|
|
tweetUrl: args.tweet_url,
|
|
comment: args.comment,
|
|
groupFolder,
|
|
timestamp: new Date().toISOString()
|
|
});
|
|
|
|
const result = await waitForResult(requestId);
|
|
return {
|
|
content: [{ type: 'text', text: result.message }],
|
|
isError: !result.success
|
|
};
|
|
}
|
|
)
|
|
];
|
|
}
|