ssh-agent-workspace 1.0.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +319 -0
- package/dist/__tests__/SSHManager.test.d.ts +2 -0
- package/dist/__tests__/SSHManager.test.d.ts.map +1 -0
- package/dist/__tests__/SSHManager.test.js +134 -0
- package/dist/__tests__/SSHManager.test.js.map +1 -0
- package/dist/__tests__/SessionManager.test.d.ts +2 -0
- package/dist/__tests__/SessionManager.test.d.ts.map +1 -0
- package/dist/__tests__/SessionManager.test.js +141 -0
- package/dist/__tests__/SessionManager.test.js.map +1 -0
- package/dist/__tests__/StorageManager.test.d.ts +2 -0
- package/dist/__tests__/StorageManager.test.d.ts.map +1 -0
- package/dist/__tests__/StorageManager.test.js +171 -0
- package/dist/__tests__/StorageManager.test.js.map +1 -0
- package/dist/__tests__/ansi.test.d.ts +2 -0
- package/dist/__tests__/ansi.test.d.ts.map +1 -0
- package/dist/__tests__/ansi.test.js +41 -0
- package/dist/__tests__/ansi.test.js.map +1 -0
- package/dist/__tests__/security.test.d.ts +2 -0
- package/dist/__tests__/security.test.d.ts.map +1 -0
- package/dist/__tests__/security.test.js +87 -0
- package/dist/__tests__/security.test.js.map +1 -0
- package/dist/__tests__/validation.test.d.ts +2 -0
- package/dist/__tests__/validation.test.d.ts.map +1 -0
- package/dist/__tests__/validation.test.js +23 -0
- package/dist/__tests__/validation.test.js.map +1 -0
- package/dist/core/HostSecurityManager.d.ts +25 -0
- package/dist/core/HostSecurityManager.d.ts.map +1 -0
- package/dist/core/HostSecurityManager.js +76 -0
- package/dist/core/HostSecurityManager.js.map +1 -0
- package/dist/core/SSHManager.d.ts +48 -0
- package/dist/core/SSHManager.d.ts.map +1 -0
- package/dist/core/SSHManager.js +288 -0
- package/dist/core/SSHManager.js.map +1 -0
- package/dist/core/SessionManager.d.ts +15 -0
- package/dist/core/SessionManager.d.ts.map +1 -0
- package/dist/core/SessionManager.js +96 -0
- package/dist/core/SessionManager.js.map +1 -0
- package/dist/core/StorageManager.d.ts +27 -0
- package/dist/core/StorageManager.d.ts.map +1 -0
- package/dist/core/StorageManager.js +87 -0
- package/dist/core/StorageManager.js.map +1 -0
- package/dist/core/TmuxManager.d.ts +21 -0
- package/dist/core/TmuxManager.d.ts.map +1 -0
- package/dist/core/TmuxManager.js +110 -0
- package/dist/core/TmuxManager.js.map +1 -0
- package/dist/core/ToolConfigManager.d.ts +15 -0
- package/dist/core/ToolConfigManager.d.ts.map +1 -0
- package/dist/core/ToolConfigManager.js +57 -0
- package/dist/core/ToolConfigManager.js.map +1 -0
- package/dist/index.d.ts +3 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +169 -0
- package/dist/index.js.map +1 -0
- package/dist/server.d.ts +44 -0
- package/dist/server.d.ts.map +1 -0
- package/dist/server.js +152 -0
- package/dist/server.js.map +1 -0
- package/dist/tools/backup.d.ts +74 -0
- package/dist/tools/backup.d.ts.map +1 -0
- package/dist/tools/backup.js +152 -0
- package/dist/tools/backup.js.map +1 -0
- package/dist/tools/connect.d.ts +46 -0
- package/dist/tools/connect.d.ts.map +1 -0
- package/dist/tools/connect.js +235 -0
- package/dist/tools/connect.js.map +1 -0
- package/dist/tools/connection_status.d.ts +39 -0
- package/dist/tools/connection_status.d.ts.map +1 -0
- package/dist/tools/connection_status.js +67 -0
- package/dist/tools/connection_status.js.map +1 -0
- package/dist/tools/db_query.d.ts +103 -0
- package/dist/tools/db_query.d.ts.map +1 -0
- package/dist/tools/db_query.js +194 -0
- package/dist/tools/db_query.js.map +1 -0
- package/dist/tools/deploy.d.ts +127 -0
- package/dist/tools/deploy.d.ts.map +1 -0
- package/dist/tools/deploy.js +201 -0
- package/dist/tools/deploy.js.map +1 -0
- package/dist/tools/disconnect.d.ts +46 -0
- package/dist/tools/disconnect.d.ts.map +1 -0
- package/dist/tools/disconnect.js +77 -0
- package/dist/tools/disconnect.js.map +1 -0
- package/dist/tools/exec.d.ts +69 -0
- package/dist/tools/exec.d.ts.map +1 -0
- package/dist/tools/exec.js +188 -0
- package/dist/tools/exec.js.map +1 -0
- package/dist/tools/group_exec.d.ts +80 -0
- package/dist/tools/group_exec.d.ts.map +1 -0
- package/dist/tools/group_exec.js +150 -0
- package/dist/tools/group_exec.js.map +1 -0
- package/dist/tools/health_check.d.ts +38 -0
- package/dist/tools/health_check.d.ts.map +1 -0
- package/dist/tools/health_check.js +161 -0
- package/dist/tools/health_check.js.map +1 -0
- package/dist/tools/host_security.d.ts +52 -0
- package/dist/tools/host_security.d.ts.map +1 -0
- package/dist/tools/host_security.js +127 -0
- package/dist/tools/host_security.js.map +1 -0
- package/dist/tools/index.d.ts +24 -0
- package/dist/tools/index.d.ts.map +1 -0
- package/dist/tools/index.js +24 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/tools/interrupt.d.ts +47 -0
- package/dist/tools/interrupt.d.ts.map +1 -0
- package/dist/tools/interrupt.js +77 -0
- package/dist/tools/interrupt.js.map +1 -0
- package/dist/tools/list_hosts.d.ts +15 -0
- package/dist/tools/list_hosts.d.ts.map +1 -0
- package/dist/tools/list_hosts.js +18 -0
- package/dist/tools/list_hosts.js.map +1 -0
- package/dist/tools/list_sessions.d.ts +16 -0
- package/dist/tools/list_sessions.d.ts.map +1 -0
- package/dist/tools/list_sessions.js +20 -0
- package/dist/tools/list_sessions.js.map +1 -0
- package/dist/tools/read_output.d.ts +46 -0
- package/dist/tools/read_output.d.ts.map +1 -0
- package/dist/tools/read_output.js +73 -0
- package/dist/tools/read_output.js.map +1 -0
- package/dist/tools/reconnect_to_tmux.d.ts +53 -0
- package/dist/tools/reconnect_to_tmux.d.ts.map +1 -0
- package/dist/tools/reconnect_to_tmux.js +199 -0
- package/dist/tools/reconnect_to_tmux.js.map +1 -0
- package/dist/tools/send_input.d.ts +45 -0
- package/dist/tools/send_input.d.ts.map +1 -0
- package/dist/tools/send_input.js +83 -0
- package/dist/tools/send_input.js.map +1 -0
- package/dist/tools/sftp_download.d.ts +52 -0
- package/dist/tools/sftp_download.d.ts.map +1 -0
- package/dist/tools/sftp_download.js +90 -0
- package/dist/tools/sftp_download.js.map +1 -0
- package/dist/tools/sftp_list.d.ts +46 -0
- package/dist/tools/sftp_list.d.ts.map +1 -0
- package/dist/tools/sftp_list.js +93 -0
- package/dist/tools/sftp_list.js.map +1 -0
- package/dist/tools/sftp_upload.d.ts +52 -0
- package/dist/tools/sftp_upload.d.ts.map +1 -0
- package/dist/tools/sftp_upload.js +98 -0
- package/dist/tools/sftp_upload.js.map +1 -0
- package/dist/tools/ssh_tunnel.d.ts +116 -0
- package/dist/tools/ssh_tunnel.d.ts.map +1 -0
- package/dist/tools/ssh_tunnel.js +282 -0
- package/dist/tools/ssh_tunnel.js.map +1 -0
- package/dist/tools/sync.d.ts +71 -0
- package/dist/tools/sync.d.ts.map +1 -0
- package/dist/tools/sync.js +310 -0
- package/dist/tools/sync.js.map +1 -0
- package/dist/tools/tail_log.d.ts +61 -0
- package/dist/tools/tail_log.d.ts.map +1 -0
- package/dist/tools/tail_log.js +111 -0
- package/dist/tools/tail_log.js.map +1 -0
- package/dist/tools/tools_config.d.ts +34 -0
- package/dist/tools/tools_config.d.ts.map +1 -0
- package/dist/tools/tools_config.js +98 -0
- package/dist/tools/tools_config.js.map +1 -0
- package/dist/types/index.d.ts +21 -0
- package/dist/types/index.d.ts.map +1 -0
- package/dist/types/index.js +2 -0
- package/dist/types/index.js.map +1 -0
- package/dist/utils/ansi.d.ts +2 -0
- package/dist/utils/ansi.d.ts.map +1 -0
- package/dist/utils/ansi.js +7 -0
- package/dist/utils/ansi.js.map +1 -0
- package/dist/utils/logger.d.ts +3 -0
- package/dist/utils/logger.d.ts.map +1 -0
- package/dist/utils/logger.js +8 -0
- package/dist/utils/logger.js.map +1 -0
- package/dist/utils/security.d.ts +7 -0
- package/dist/utils/security.d.ts.map +1 -0
- package/dist/utils/security.js +58 -0
- package/dist/utils/security.js.map +1 -0
- package/dist/utils/ssh.d.ts +4 -0
- package/dist/utils/ssh.d.ts.map +1 -0
- package/dist/utils/ssh.js +29 -0
- package/dist/utils/ssh.js.map +1 -0
- package/dist/utils/sshConfig.d.ts +4 -0
- package/dist/utils/sshConfig.d.ts.map +1 -0
- package/dist/utils/sshConfig.js +85 -0
- package/dist/utils/sshConfig.js.map +1 -0
- package/dist/utils/validation.d.ts +4 -0
- package/dist/utils/validation.d.ts.map +1 -0
- package/dist/utils/validation.js +12 -0
- package/dist/utils/validation.js.map +1 -0
- package/docs/SECURITY.md +213 -0
- package/docs/TOOLS.md +425 -0
- package/keygen.bat +325 -0
- package/package.json +48 -0
- package/test_check.bat +9 -0
- package/test_delayed.bat +12 -0
- package/vitest.config.ts +14 -0
|
@@ -0,0 +1,310 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
import fs from 'fs';
|
|
3
|
+
import path from 'path';
|
|
4
|
+
import { isReadOnlyMode } from '../utils/security.js';
|
|
5
|
+
import { logger } from '../utils/logger.js';
|
|
6
|
+
export const syncSchema = z.object({
|
|
7
|
+
session_id: z.string().min(1),
|
|
8
|
+
local_path: z.string().min(1).describe('Absolute local directory path'),
|
|
9
|
+
remote_path: z.string().min(1).describe('Absolute remote directory path'),
|
|
10
|
+
direction: z.enum(['upload', 'download', 'bidirectional']).default('upload').describe('Sync direction'),
|
|
11
|
+
dry_run: z.boolean().default(false).describe('Preview changes without actually transferring files'),
|
|
12
|
+
max_depth: z.number().int().min(1).max(20).default(10).describe('Maximum directory recursion depth'),
|
|
13
|
+
});
|
|
14
|
+
export const syncTool = {
|
|
15
|
+
name: 'sync',
|
|
16
|
+
description: 'Bidirectional rsync-like sync between local and remote directories via SFTP. Compares files by mtime and size, transfers only changed/new files. Supports dry-run preview.',
|
|
17
|
+
inputSchema: {
|
|
18
|
+
type: 'object',
|
|
19
|
+
properties: {
|
|
20
|
+
session_id: { type: 'string', description: 'Session ID returned by connect' },
|
|
21
|
+
local_path: { type: 'string', description: 'Absolute local directory path' },
|
|
22
|
+
remote_path: { type: 'string', description: 'Absolute remote directory path' },
|
|
23
|
+
direction: {
|
|
24
|
+
type: 'string',
|
|
25
|
+
description: 'upload (local->remote), download (remote->local), or bidirectional (both ways, newest wins)',
|
|
26
|
+
enum: ['upload', 'download', 'bidirectional'],
|
|
27
|
+
default: 'upload',
|
|
28
|
+
},
|
|
29
|
+
dry_run: { type: 'boolean', description: 'Preview changes without transferring (default: false)', default: false },
|
|
30
|
+
max_depth: { type: 'number', description: 'Max directory depth (default: 10)', default: 10 },
|
|
31
|
+
},
|
|
32
|
+
required: ['session_id', 'local_path', 'remote_path'],
|
|
33
|
+
},
|
|
34
|
+
};
|
|
35
|
+
async function listLocalRecursive(dirPath, maxDepth, currentDepth, prefix) {
|
|
36
|
+
const results = [];
|
|
37
|
+
if (currentDepth > maxDepth)
|
|
38
|
+
return results;
|
|
39
|
+
let entries;
|
|
40
|
+
try {
|
|
41
|
+
entries = fs.readdirSync(dirPath, { withFileTypes: true });
|
|
42
|
+
}
|
|
43
|
+
catch {
|
|
44
|
+
return results;
|
|
45
|
+
}
|
|
46
|
+
for (const entry of entries) {
|
|
47
|
+
if (entry.name.startsWith('.'))
|
|
48
|
+
continue;
|
|
49
|
+
const fullPath = path.join(dirPath, entry.name);
|
|
50
|
+
const relPath = prefix ? `${prefix}/${entry.name}` : entry.name;
|
|
51
|
+
if (entry.isFile()) {
|
|
52
|
+
try {
|
|
53
|
+
const stat = fs.statSync(fullPath);
|
|
54
|
+
results.push({ relPath, size: stat.size, mtime: Math.floor(stat.mtimeMs) });
|
|
55
|
+
}
|
|
56
|
+
catch {
|
|
57
|
+
continue;
|
|
58
|
+
}
|
|
59
|
+
}
|
|
60
|
+
else if (entry.isDirectory() && !entry.isSymbolicLink()) {
|
|
61
|
+
const children = await listLocalRecursive(fullPath, maxDepth, currentDepth + 1, relPath);
|
|
62
|
+
results.push(...children);
|
|
63
|
+
}
|
|
64
|
+
}
|
|
65
|
+
return results;
|
|
66
|
+
}
|
|
67
|
+
async function listRemoteRecursive(sshManager, session, remotePath, maxDepth, currentDepth, prefix) {
|
|
68
|
+
const results = [];
|
|
69
|
+
if (currentDepth > maxDepth)
|
|
70
|
+
return results;
|
|
71
|
+
let entries;
|
|
72
|
+
try {
|
|
73
|
+
entries = await sshManager.sftpList(session.ssh, remotePath);
|
|
74
|
+
}
|
|
75
|
+
catch {
|
|
76
|
+
return results;
|
|
77
|
+
}
|
|
78
|
+
for (const entry of entries) {
|
|
79
|
+
if (entry.filename === '.' || entry.filename === '..')
|
|
80
|
+
continue;
|
|
81
|
+
if (entry.filename.startsWith('.'))
|
|
82
|
+
continue;
|
|
83
|
+
const fullPath = remotePath.endsWith('/') ? `${remotePath}${entry.filename}` : `${remotePath}/${entry.filename}`;
|
|
84
|
+
const relPath = prefix ? `${prefix}/${entry.filename}` : entry.filename;
|
|
85
|
+
if (entry.attrs.isFile) {
|
|
86
|
+
results.push({ relPath, size: entry.attrs.size, mtime: entry.attrs.mtime });
|
|
87
|
+
}
|
|
88
|
+
else if (entry.attrs.isDirectory && !entry.attrs.isSymbolicLink) {
|
|
89
|
+
const children = await listRemoteRecursive(sshManager, session, fullPath, maxDepth, currentDepth + 1, relPath);
|
|
90
|
+
results.push(...children);
|
|
91
|
+
}
|
|
92
|
+
}
|
|
93
|
+
return results;
|
|
94
|
+
}
|
|
95
|
+
export async function handleSync(args, sessionManager, sshManager) {
|
|
96
|
+
const parsed = syncSchema.safeParse(args);
|
|
97
|
+
if (!parsed.success) {
|
|
98
|
+
return { content: [{ type: 'text', text: `Invalid arguments: ${parsed.error.message}` }], isError: true };
|
|
99
|
+
}
|
|
100
|
+
const { session_id, local_path, remote_path, direction, dry_run, max_depth } = parsed.data;
|
|
101
|
+
const session = sessionManager.get(session_id);
|
|
102
|
+
if (!session) {
|
|
103
|
+
return { content: [{ type: 'text', text: `Error: Session '${session_id}' not found` }], isError: true };
|
|
104
|
+
}
|
|
105
|
+
if (!sshManager.isAlive(session.ssh)) {
|
|
106
|
+
return { content: [{ type: 'text', text: `Error: SSH connection dead for session '${session_id}'` }], isError: true };
|
|
107
|
+
}
|
|
108
|
+
if (isReadOnlyMode(session.host)) {
|
|
109
|
+
return { content: [{ type: 'text', text: `Error: Host '${session.host}' is in read-only mode. Sync is disabled.` }], isError: true };
|
|
110
|
+
}
|
|
111
|
+
const resolvedLocal = path.resolve(local_path);
|
|
112
|
+
if (!fs.existsSync(resolvedLocal)) {
|
|
113
|
+
return {
|
|
114
|
+
content: [{ type: 'text', text: `Error: Local directory not found: ${resolvedLocal}` }],
|
|
115
|
+
isError: true,
|
|
116
|
+
};
|
|
117
|
+
}
|
|
118
|
+
if (!fs.statSync(resolvedLocal).isDirectory()) {
|
|
119
|
+
return {
|
|
120
|
+
content: [{ type: 'text', text: `Error: Local path is not a directory: ${resolvedLocal}` }],
|
|
121
|
+
isError: true,
|
|
122
|
+
};
|
|
123
|
+
}
|
|
124
|
+
const remoteExists = await sshManager.sftpExists(session.ssh, remote_path);
|
|
125
|
+
if (!remoteExists) {
|
|
126
|
+
return {
|
|
127
|
+
content: [{ type: 'text', text: `Error: Remote directory not found: ${remote_path}` }],
|
|
128
|
+
isError: true,
|
|
129
|
+
};
|
|
130
|
+
}
|
|
131
|
+
const report = {
|
|
132
|
+
session_id,
|
|
133
|
+
dry_run,
|
|
134
|
+
direction,
|
|
135
|
+
local_path: resolvedLocal,
|
|
136
|
+
remote_path,
|
|
137
|
+
synced: [],
|
|
138
|
+
skipped: 0,
|
|
139
|
+
errors: [],
|
|
140
|
+
};
|
|
141
|
+
let localFiles = [];
|
|
142
|
+
let remoteFiles = [];
|
|
143
|
+
try {
|
|
144
|
+
localFiles = await listLocalRecursive(resolvedLocal, max_depth, 0, '');
|
|
145
|
+
}
|
|
146
|
+
catch (err) {
|
|
147
|
+
report.errors.push(`Local listing failed: ${err.message}`);
|
|
148
|
+
}
|
|
149
|
+
try {
|
|
150
|
+
remoteFiles = await listRemoteRecursive(sshManager, session, remote_path, max_depth, 0, '');
|
|
151
|
+
}
|
|
152
|
+
catch (err) {
|
|
153
|
+
report.errors.push(`Remote listing failed: ${err.message}`);
|
|
154
|
+
}
|
|
155
|
+
const localMap = new Map(localFiles.map((f) => [f.relPath, f]));
|
|
156
|
+
const remoteMap = new Map(remoteFiles.map((f) => [f.relPath, f]));
|
|
157
|
+
const allPaths = new Set([...localMap.keys(), ...remoteMap.keys()]);
|
|
158
|
+
for (const relPath of allPaths) {
|
|
159
|
+
const local = localMap.get(relPath);
|
|
160
|
+
const remote = remoteMap.get(relPath);
|
|
161
|
+
if (!remote && local) {
|
|
162
|
+
if (direction === 'download') {
|
|
163
|
+
report.skipped++;
|
|
164
|
+
continue;
|
|
165
|
+
}
|
|
166
|
+
report.synced.push({ direction: 'upload', rel_path: relPath, size: local.size, reason: 'new' });
|
|
167
|
+
if (!dry_run) {
|
|
168
|
+
try {
|
|
169
|
+
const remoteFile = remote_path.endsWith('/') ? `${remote_path}${relPath}` : `${remote_path}/${relPath}`;
|
|
170
|
+
const remoteDir = path.dirname(remoteFile);
|
|
171
|
+
await ensureRemoteDir(sshManager, session.ssh, remoteDir);
|
|
172
|
+
await sshManager.sftpUpload(session.ssh, path.join(resolvedLocal, relPath), remoteFile);
|
|
173
|
+
}
|
|
174
|
+
catch (err) {
|
|
175
|
+
report.errors.push(`Upload failed ${relPath}: ${err.message}`);
|
|
176
|
+
}
|
|
177
|
+
}
|
|
178
|
+
continue;
|
|
179
|
+
}
|
|
180
|
+
if (!local && remote) {
|
|
181
|
+
if (direction === 'upload') {
|
|
182
|
+
report.skipped++;
|
|
183
|
+
continue;
|
|
184
|
+
}
|
|
185
|
+
report.synced.push({ direction: 'download', rel_path: relPath, size: remote.size, reason: 'new' });
|
|
186
|
+
if (!dry_run) {
|
|
187
|
+
try {
|
|
188
|
+
const localFile = path.join(resolvedLocal, relPath);
|
|
189
|
+
const localDir = path.dirname(localFile);
|
|
190
|
+
if (!fs.existsSync(localDir))
|
|
191
|
+
fs.mkdirSync(localDir, { recursive: true });
|
|
192
|
+
const remoteFile = remote_path.endsWith('/') ? `${remote_path}${relPath}` : `${remote_path}/${relPath}`;
|
|
193
|
+
await sshManager.sftpDownload(session.ssh, remoteFile, localFile);
|
|
194
|
+
}
|
|
195
|
+
catch (err) {
|
|
196
|
+
report.errors.push(`Download failed ${relPath}: ${err.message}`);
|
|
197
|
+
}
|
|
198
|
+
}
|
|
199
|
+
continue;
|
|
200
|
+
}
|
|
201
|
+
if (local && remote) {
|
|
202
|
+
if (local.size === remote.size && local.mtime === remote.mtime) {
|
|
203
|
+
report.skipped++;
|
|
204
|
+
continue;
|
|
205
|
+
}
|
|
206
|
+
if (direction === 'upload') {
|
|
207
|
+
report.synced.push({ direction: 'upload', rel_path: relPath, size: local.size, reason: 'updated' });
|
|
208
|
+
if (!dry_run) {
|
|
209
|
+
try {
|
|
210
|
+
const remoteFile = remote_path.endsWith('/') ? `${remote_path}${relPath}` : `${remote_path}/${relPath}`;
|
|
211
|
+
await sshManager.sftpUpload(session.ssh, path.join(resolvedLocal, relPath), remoteFile);
|
|
212
|
+
}
|
|
213
|
+
catch (err) {
|
|
214
|
+
report.errors.push(`Upload failed ${relPath}: ${err.message}`);
|
|
215
|
+
}
|
|
216
|
+
}
|
|
217
|
+
}
|
|
218
|
+
else if (direction === 'download') {
|
|
219
|
+
report.synced.push({ direction: 'download', rel_path: relPath, size: remote.size, reason: 'updated' });
|
|
220
|
+
if (!dry_run) {
|
|
221
|
+
try {
|
|
222
|
+
const localFile = path.join(resolvedLocal, relPath);
|
|
223
|
+
const remoteFile = remote_path.endsWith('/') ? `${remote_path}${relPath}` : `${remote_path}/${relPath}`;
|
|
224
|
+
await sshManager.sftpDownload(session.ssh, remoteFile, localFile);
|
|
225
|
+
}
|
|
226
|
+
catch (err) {
|
|
227
|
+
report.errors.push(`Download failed ${relPath}: ${err.message}`);
|
|
228
|
+
}
|
|
229
|
+
}
|
|
230
|
+
}
|
|
231
|
+
else {
|
|
232
|
+
if (local.mtime >= remote.mtime) {
|
|
233
|
+
report.synced.push({ direction: 'upload', rel_path: relPath, size: local.size, reason: 'newer' });
|
|
234
|
+
if (!dry_run) {
|
|
235
|
+
try {
|
|
236
|
+
const remoteFile = remote_path.endsWith('/') ? `${remote_path}${relPath}` : `${remote_path}/${relPath}`;
|
|
237
|
+
await sshManager.sftpUpload(session.ssh, path.join(resolvedLocal, relPath), remoteFile);
|
|
238
|
+
}
|
|
239
|
+
catch (err) {
|
|
240
|
+
report.errors.push(`Upload failed ${relPath}: ${err.message}`);
|
|
241
|
+
}
|
|
242
|
+
}
|
|
243
|
+
}
|
|
244
|
+
else {
|
|
245
|
+
report.synced.push({ direction: 'download', rel_path: relPath, size: remote.size, reason: 'newer' });
|
|
246
|
+
if (!dry_run) {
|
|
247
|
+
try {
|
|
248
|
+
const localFile = path.join(resolvedLocal, relPath);
|
|
249
|
+
const remoteFile = remote_path.endsWith('/') ? `${remote_path}${relPath}` : `${remote_path}/${relPath}`;
|
|
250
|
+
await sshManager.sftpDownload(session.ssh, remoteFile, localFile);
|
|
251
|
+
}
|
|
252
|
+
catch (err) {
|
|
253
|
+
report.errors.push(`Download failed ${relPath}: ${err.message}`);
|
|
254
|
+
}
|
|
255
|
+
}
|
|
256
|
+
}
|
|
257
|
+
}
|
|
258
|
+
}
|
|
259
|
+
}
|
|
260
|
+
const uploaded = report.synced.filter((s) => s.direction === 'upload');
|
|
261
|
+
const downloaded = report.synced.filter((s) => s.direction === 'download');
|
|
262
|
+
const totalTransferred = [...uploaded, ...downloaded].reduce((sum, s) => sum + s.size, 0);
|
|
263
|
+
logger.info({
|
|
264
|
+
sessionId: session_id,
|
|
265
|
+
synced: report.synced.length,
|
|
266
|
+
skipped: report.skipped,
|
|
267
|
+
uploaded: uploaded.length,
|
|
268
|
+
downloaded: downloaded.length,
|
|
269
|
+
totalSize: totalTransferred,
|
|
270
|
+
dry_run,
|
|
271
|
+
errors: report.errors.length,
|
|
272
|
+
}, 'Sync complete');
|
|
273
|
+
return {
|
|
274
|
+
content: [{
|
|
275
|
+
type: 'text',
|
|
276
|
+
text: JSON.stringify({
|
|
277
|
+
...report,
|
|
278
|
+
summary: {
|
|
279
|
+
total_synced: report.synced.length,
|
|
280
|
+
uploaded: uploaded.length,
|
|
281
|
+
downloaded: downloaded.length,
|
|
282
|
+
skipped: report.skipped,
|
|
283
|
+
total_bytes: totalTransferred,
|
|
284
|
+
errors: report.errors.length,
|
|
285
|
+
},
|
|
286
|
+
}, null, 2),
|
|
287
|
+
}],
|
|
288
|
+
isError: report.errors.length > 0,
|
|
289
|
+
};
|
|
290
|
+
}
|
|
291
|
+
async function ensureRemoteDir(sshManager, ssh, remotePath) {
|
|
292
|
+
const parts = remotePath.replace(/\\/g, '/').split('/').filter(Boolean);
|
|
293
|
+
let current = '';
|
|
294
|
+
for (const part of parts) {
|
|
295
|
+
current += '/' + part;
|
|
296
|
+
try {
|
|
297
|
+
const exists = await sshManager.sftpExists(ssh, current);
|
|
298
|
+
if (!exists) {
|
|
299
|
+
await sshManager.sftpMkdir(ssh, current);
|
|
300
|
+
}
|
|
301
|
+
}
|
|
302
|
+
catch {
|
|
303
|
+
try {
|
|
304
|
+
await sshManager.sftpMkdir(ssh, current);
|
|
305
|
+
}
|
|
306
|
+
catch { }
|
|
307
|
+
}
|
|
308
|
+
}
|
|
309
|
+
}
|
|
310
|
+
//# sourceMappingURL=sync.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sync.js","sourceRoot":"","sources":["../../src/tools/sync.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AAGxB,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAE5C,MAAM,CAAC,MAAM,UAAU,GAAG,CAAC,CAAC,MAAM,CAAC;IACjC,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;IAC7B,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,+BAA+B,CAAC;IACvE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,gCAAgC,CAAC;IACzE,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,UAAU,EAAE,eAAe,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,QAAQ,CAAC,gBAAgB,CAAC;IACvG,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,qDAAqD,CAAC;IACnG,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,mCAAmC,CAAC;CACrG,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,QAAQ,GAAG;IACtB,IAAI,EAAE,MAAM;IACZ,WAAW,EACT,4KAA4K;IAC9K,WAAW,EAAE;QACX,IAAI,EAAE,QAAiB;QACvB,UAAU,EAAE;YACV,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,gCAAgC,EAAE;YAC7E,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,+BAA+B,EAAE;YAC5E,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,gCAAgC,EAAE;YAC9E,SAAS,EAAE;gBACT,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,6FAA6F;gBAC1G,IAAI,EAAE,CAAC,QAAQ,EAAE,UAAU,EAAE,eAAe,CAAC;gBAC7C,OAAO,EAAE,QAAQ;aAClB;YACD,OAAO,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,uDAAuD,EAAE,OAAO,EAAE,KAAK,EAAE;YAClH,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,mCAAmC,EAAE,OAAO,EAAE,EAAE,EAAE;SAC7F;QACD,QAAQ,EAAE,CAAC,YAAY,EAAE,YAAY,EAAE,aAAa,CAAC;KACtD;CACF,CAAC;AA0BF,KAAK,UAAU,kBAAkB,CAC/B,OAAe,EACf,QAAgB,EAChB,YAAoB,EACpB,MAAc;IAEd,MAAM,OAAO,GAAe,EAAE,CAAC;IAC/B,IAAI,YAAY,GAAG,QAAQ;QAAE,OAAO,OAAO,CAAC;IAE5C,IAAI,OAAoB,CAAC;IACzB,IAAI,CAAC;QACH,OAAO,GAAG,EAAE,CAAC,WAAW,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;IAC7D,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC5B,IAAI,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;YAAE,SAAS;QACzC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QAChD,MAAM,OAAO,GAAG,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC;QAEhE,IAAI,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC;YACnB,IAAI,CAAC;gBACH,MAAM,IAAI,GAAG,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;gBACnC,OAAO,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC9E,CAAC;YAAC,MAAM,CAAC;gBAAC,SAAS;YAAC,CAAC;QACvB,CAAC;aAAM,IAAI,KAAK,CAAC,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,CAAC;YAC1D,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC,QAAQ,EAAE,QAAQ,EAAE,YAAY,GAAG,CAAC,EAAE,OAAO,CAAC,CAAC;YACzF,OAAO,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAC;QAC5B,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,KAAK,UAAU,mBAAmB,CAChC,UAAsB,EACtB,OAAuC,EACvC,UAAkB,EAClB,QAAgB,EAChB,YAAoB,EACpB,MAAc;IAEd,MAAM,OAAO,GAAe,EAAE,CAAC;IAC/B,IAAI,YAAY,GAAG,QAAQ;QAAE,OAAO,OAAO,CAAC;IAE5C,IAAI,OAAoB,CAAC;IACzB,IAAI,CAAC;QACH,OAAO,GAAG,MAAM,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;IAC/D,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC5B,IAAI,KAAK,CAAC,QAAQ,KAAK,GAAG,IAAI,KAAK,CAAC,QAAQ,KAAK,IAAI;YAAE,SAAS;QAChE,IAAI,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC;YAAE,SAAS;QAE7C,MAAM,QAAQ,GAAG,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,UAAU,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,GAAG,UAAU,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;QACjH,MAAM,OAAO,GAAG,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC;QAExE,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;YACvB,OAAO,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;QAC9E,CAAC;aAAM,IAAI,KAAK,CAAC,KAAK,CAAC,WAAW,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,cAAc,EAAE,CAAC;YAClE,MAAM,QAAQ,GAAG,MAAM,mBAAmB,CAAC,UAAU,EAAE,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,YAAY,GAAG,CAAC,EAAE,OAAO,CAAC,CAAC;YAC/G,OAAO,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAC;QAC5B,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,IAAa,EACb,cAA8B,EAC9B,UAAsB;IAEtB,MAAM,MAAM,GAAG,UAAU,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IAC1C,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QACpB,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,sBAAsB,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IACrH,CAAC;IAED,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,SAAS,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC;IAC3F,MAAM,OAAO,GAAG,cAAc,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IAE/C,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,mBAAmB,UAAU,aAAa,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IACnH,CAAC;IACD,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;QACrC,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,2CAA2C,UAAU,GAAG,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IACjI,CAAC;IAED,IAAI,cAAc,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QACjC,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,gBAAgB,OAAO,CAAC,IAAI,2CAA2C,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAChJ,CAAC;IAED,MAAM,aAAa,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAC/C,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;QAClC,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,qCAAqC,aAAa,EAAE,EAAE,CAAC;YAChG,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;IACD,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC;QAC9C,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,yCAAyC,aAAa,EAAE,EAAE,CAAC;YACpG,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;IAED,MAAM,YAAY,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,OAAO,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;IAC3E,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,sCAAsC,WAAW,EAAE,EAAE,CAAC;YAC/F,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;IAED,MAAM,MAAM,GAAe;QACzB,UAAU;QACV,OAAO;QACP,SAAS;QACT,UAAU,EAAE,aAAa;QACzB,WAAW;QACX,MAAM,EAAE,EAAE;QACV,OAAO,EAAE,CAAC;QACV,MAAM,EAAE,EAAE;KACX,CAAC;IAEF,IAAI,UAAU,GAAe,EAAE,CAAC;IAChC,IAAI,WAAW,GAAe,EAAE,CAAC;IAEjC,IAAI,CAAC;QACH,UAAU,GAAG,MAAM,kBAAkB,CAAC,aAAa,EAAE,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;IACzE,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,yBAA0B,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;IACxE,CAAC;IAED,IAAI,CAAC;QACH,WAAW,GAAG,MAAM,mBAAmB,CAAC,UAAU,EAAE,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;IAC9F,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,0BAA2B,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;IACzE,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAChE,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAElE,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,QAAQ,CAAC,IAAI,EAAE,EAAE,GAAG,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IAEpE,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,MAAM,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACpC,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAEtC,IAAI,CAAC,MAAM,IAAI,KAAK,EAAE,CAAC;YACrB,IAAI,SAAS,KAAK,UAAU,EAAE,CAAC;gBAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBAAC,SAAS;YAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;YAChG,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,IAAI,CAAC;oBACH,MAAM,UAAU,GAAG,WAAW,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,WAAW,GAAG,OAAO,EAAE,CAAC,CAAC,CAAC,GAAG,WAAW,IAAI,OAAO,EAAE,CAAC;oBACxG,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;oBAC3C,MAAM,eAAe,CAAC,UAAU,EAAE,OAAO,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;oBAC1D,MAAM,UAAU,CAAC,UAAU,CAAC,OAAO,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,OAAO,CAAC,EAAE,UAAU,CAAC,CAAC;gBAC1F,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,OAAO,KAAM,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;gBAC5E,CAAC;YACH,CAAC;YACD,SAAS;QACX,CAAC;QAED,IAAI,CAAC,KAAK,IAAI,MAAM,EAAE,CAAC;YACrB,IAAI,SAAS,KAAK,QAAQ,EAAE,CAAC;gBAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBAAC,SAAS;YAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;YACnG,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,IAAI,CAAC;oBACH,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;oBACpD,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;oBACzC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC;wBAAE,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;oBAC1E,MAAM,UAAU,GAAG,WAAW,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,WAAW,GAAG,OAAO,EAAE,CAAC,CAAC,CAAC,GAAG,WAAW,IAAI,OAAO,EAAE,CAAC;oBACxG,MAAM,UAAU,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC;gBACpE,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,mBAAmB,OAAO,KAAM,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;gBAC9E,CAAC;YACH,CAAC;YACD,SAAS;QACX,CAAC;QAED,IAAI,KAAK,IAAI,MAAM,EAAE,CAAC;YACpB,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC,IAAI,IAAI,KAAK,CAAC,KAAK,KAAK,MAAM,CAAC,KAAK,EAAE,CAAC;gBAC/D,MAAM,CAAC,OAAO,EAAE,CAAC;gBACjB,SAAS;YACX,CAAC;YAED,IAAI,SAAS,KAAK,QAAQ,EAAE,CAAC;gBAC3B,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;gBACpG,IAAI,CAAC,OAAO,EAAE,CAAC;oBACb,IAAI,CAAC;wBACH,MAAM,UAAU,GAAG,WAAW,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,WAAW,GAAG,OAAO,EAAE,CAAC,CAAC,CAAC,GAAG,WAAW,IAAI,OAAO,EAAE,CAAC;wBACxG,MAAM,UAAU,CAAC,UAAU,CAAC,OAAO,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,OAAO,CAAC,EAAE,UAAU,CAAC,CAAC;oBAC1F,CAAC;oBAAC,OAAO,GAAG,EAAE,CAAC;wBACb,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,OAAO,KAAM,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;oBAC5E,CAAC;gBACH,CAAC;YACH,CAAC;iBAAM,IAAI,SAAS,KAAK,UAAU,EAAE,CAAC;gBACpC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;gBACvG,IAAI,CAAC,OAAO,EAAE,CAAC;oBACb,IAAI,CAAC;wBACH,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;wBACpD,MAAM,UAAU,GAAG,WAAW,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,WAAW,GAAG,OAAO,EAAE,CAAC,CAAC,CAAC,GAAG,WAAW,IAAI,OAAO,EAAE,CAAC;wBACxG,MAAM,UAAU,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC;oBACpE,CAAC;oBAAC,OAAO,GAAG,EAAE,CAAC;wBACb,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,mBAAmB,OAAO,KAAM,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;oBAC9E,CAAC;gBACH,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,IAAI,KAAK,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;oBAChC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;oBAClG,IAAI,CAAC,OAAO,EAAE,CAAC;wBACb,IAAI,CAAC;4BACH,MAAM,UAAU,GAAG,WAAW,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,WAAW,GAAG,OAAO,EAAE,CAAC,CAAC,CAAC,GAAG,WAAW,IAAI,OAAO,EAAE,CAAC;4BACxG,MAAM,UAAU,CAAC,UAAU,CAAC,OAAO,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,OAAO,CAAC,EAAE,UAAU,CAAC,CAAC;wBAC1F,CAAC;wBAAC,OAAO,GAAG,EAAE,CAAC;4BACb,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,OAAO,KAAM,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;wBAC5E,CAAC;oBACH,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;oBACrG,IAAI,CAAC,OAAO,EAAE,CAAC;wBACb,IAAI,CAAC;4BACH,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;4BACpD,MAAM,UAAU,GAAG,WAAW,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,WAAW,GAAG,OAAO,EAAE,CAAC,CAAC,CAAC,GAAG,WAAW,IAAI,OAAO,EAAE,CAAC;4BACxG,MAAM,UAAU,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC;wBACpE,CAAC;wBAAC,OAAO,GAAG,EAAE,CAAC;4BACb,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,mBAAmB,OAAO,KAAM,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;wBAC9E,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC;IACvE,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,UAAU,CAAC,CAAC;IAC3E,MAAM,gBAAgB,GAAG,CAAC,GAAG,QAAQ,EAAE,GAAG,UAAU,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IAE1F,MAAM,CAAC,IAAI,CAAC;QACV,SAAS,EAAE,UAAU;QACrB,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM;QAC5B,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,QAAQ,EAAE,QAAQ,CAAC,MAAM;QACzB,UAAU,EAAE,UAAU,CAAC,MAAM;QAC7B,SAAS,EAAE,gBAAgB;QAC3B,OAAO;QACP,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM;KAC7B,EAAE,eAAe,CAAC,CAAC;IAEpB,OAAO;QACL,OAAO,EAAE,CAAC;gBACR,IAAI,EAAE,MAAe;gBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,GAAG,MAAM;oBACT,OAAO,EAAE;wBACP,YAAY,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM;wBAClC,QAAQ,EAAE,QAAQ,CAAC,MAAM;wBACzB,UAAU,EAAE,UAAU,CAAC,MAAM;wBAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;wBACvB,WAAW,EAAE,gBAAgB;wBAC7B,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM;qBAC7B;iBACF,EAAE,IAAI,EAAE,CAAC,CAAC;aACZ,CAAC;QACF,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC;KAClC,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,eAAe,CAAC,UAAsB,EAAE,GAA0B,EAAE,UAAkB;IACnG,MAAM,KAAK,GAAG,UAAU,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACxE,IAAI,OAAO,GAAG,EAAE,CAAC;IACjB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,OAAO,IAAI,GAAG,GAAG,IAAI,CAAC;QACtB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;YACzD,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,UAAU,CAAC,SAAS,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;YAC3C,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,IAAI,CAAC;gBAAC,MAAM,UAAU,CAAC,SAAS,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC,CAAA,CAAC;QAC5D,CAAC;IACH,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,61 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
import { SessionManager } from '../core/SessionManager.js';
|
|
3
|
+
import { SSHManager } from '../core/SSHManager.js';
|
|
4
|
+
export declare const tailLogSchema: z.ZodObject<{
|
|
5
|
+
session_id: z.ZodString;
|
|
6
|
+
file_path: z.ZodString;
|
|
7
|
+
lines: z.ZodDefault<z.ZodNumber>;
|
|
8
|
+
follow_ms: z.ZodDefault<z.ZodNumber>;
|
|
9
|
+
}, "strip", z.ZodTypeAny, {
|
|
10
|
+
session_id: string;
|
|
11
|
+
lines: number;
|
|
12
|
+
file_path: string;
|
|
13
|
+
follow_ms: number;
|
|
14
|
+
}, {
|
|
15
|
+
session_id: string;
|
|
16
|
+
file_path: string;
|
|
17
|
+
lines?: number | undefined;
|
|
18
|
+
follow_ms?: number | undefined;
|
|
19
|
+
}>;
|
|
20
|
+
export declare const tailLogTool: {
|
|
21
|
+
name: string;
|
|
22
|
+
description: string;
|
|
23
|
+
inputSchema: {
|
|
24
|
+
type: "object";
|
|
25
|
+
properties: {
|
|
26
|
+
session_id: {
|
|
27
|
+
type: string;
|
|
28
|
+
description: string;
|
|
29
|
+
};
|
|
30
|
+
file_path: {
|
|
31
|
+
type: string;
|
|
32
|
+
description: string;
|
|
33
|
+
};
|
|
34
|
+
lines: {
|
|
35
|
+
type: string;
|
|
36
|
+
description: string;
|
|
37
|
+
default: number;
|
|
38
|
+
};
|
|
39
|
+
follow_ms: {
|
|
40
|
+
type: string;
|
|
41
|
+
description: string;
|
|
42
|
+
default: number;
|
|
43
|
+
};
|
|
44
|
+
};
|
|
45
|
+
required: string[];
|
|
46
|
+
};
|
|
47
|
+
};
|
|
48
|
+
export declare function handleTailLog(args: unknown, sessionManager: SessionManager, sshManager: SSHManager): Promise<{
|
|
49
|
+
content: {
|
|
50
|
+
type: "text";
|
|
51
|
+
text: string;
|
|
52
|
+
}[];
|
|
53
|
+
isError: boolean;
|
|
54
|
+
} | {
|
|
55
|
+
content: {
|
|
56
|
+
type: "text";
|
|
57
|
+
text: string;
|
|
58
|
+
}[];
|
|
59
|
+
isError?: undefined;
|
|
60
|
+
}>;
|
|
61
|
+
//# sourceMappingURL=tail_log.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tail_log.d.ts","sourceRoot":"","sources":["../../src/tools/tail_log.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAC3D,OAAO,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAGnD,eAAO,MAAM,aAAa;;;;;;;;;;;;;;;EAKxB,CAAC;AAEH,eAAO,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;CA4BvB,CAAC;AAEF,wBAAsB,aAAa,CACjC,IAAI,EAAE,OAAO,EACb,cAAc,EAAE,cAAc,EAC9B,UAAU,EAAE,UAAU;;;;;;;;;;;;GAgFvB"}
|
|
@@ -0,0 +1,111 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
import { logger } from '../utils/logger.js';
|
|
3
|
+
export const tailLogSchema = z.object({
|
|
4
|
+
session_id: z.string().min(1),
|
|
5
|
+
file_path: z.string().min(1).describe('Absolute path to the log file on the remote host'),
|
|
6
|
+
lines: z.number().min(1).max(5000).default(100).describe('Number of lines to fetch (default: 100)'),
|
|
7
|
+
follow_ms: z.number().min(0).max(30000).default(0).describe('If >0, poll for new lines this many milliseconds and return appended output'),
|
|
8
|
+
});
|
|
9
|
+
export const tailLogTool = {
|
|
10
|
+
name: 'tail_log',
|
|
11
|
+
description: 'Read the last N lines of a remote log file via SSH exec channel. Optionally follow (poll) for new output.',
|
|
12
|
+
inputSchema: {
|
|
13
|
+
type: 'object',
|
|
14
|
+
properties: {
|
|
15
|
+
session_id: {
|
|
16
|
+
type: 'string',
|
|
17
|
+
description: 'Session ID returned by connect',
|
|
18
|
+
},
|
|
19
|
+
file_path: {
|
|
20
|
+
type: 'string',
|
|
21
|
+
description: 'Absolute path to the log file on the remote host',
|
|
22
|
+
},
|
|
23
|
+
lines: {
|
|
24
|
+
type: 'number',
|
|
25
|
+
description: 'Number of lines to fetch from the end (default: 100, max: 5000)',
|
|
26
|
+
default: 100,
|
|
27
|
+
},
|
|
28
|
+
follow_ms: {
|
|
29
|
+
type: 'number',
|
|
30
|
+
description: 'Milliseconds to poll for new output after initial fetch (0 = no follow, max: 30000)',
|
|
31
|
+
default: 0,
|
|
32
|
+
},
|
|
33
|
+
},
|
|
34
|
+
required: ['session_id', 'file_path'],
|
|
35
|
+
},
|
|
36
|
+
};
|
|
37
|
+
export async function handleTailLog(args, sessionManager, sshManager) {
|
|
38
|
+
const parsed = tailLogSchema.safeParse(args);
|
|
39
|
+
if (!parsed.success) {
|
|
40
|
+
return {
|
|
41
|
+
content: [{ type: 'text', text: `Invalid arguments: ${parsed.error.message}` }],
|
|
42
|
+
isError: true,
|
|
43
|
+
};
|
|
44
|
+
}
|
|
45
|
+
const { session_id, file_path, lines, follow_ms } = parsed.data;
|
|
46
|
+
const session = sessionManager.get(session_id);
|
|
47
|
+
if (!session) {
|
|
48
|
+
return {
|
|
49
|
+
content: [{ type: 'text', text: `Error: Session '${session_id}' not found or has been disconnected` }],
|
|
50
|
+
isError: true,
|
|
51
|
+
};
|
|
52
|
+
}
|
|
53
|
+
if (!sshManager.isAlive(session.ssh)) {
|
|
54
|
+
return {
|
|
55
|
+
content: [{ type: 'text', text: `Error: SSH connection for session '${session_id}' is dead` }],
|
|
56
|
+
isError: true,
|
|
57
|
+
};
|
|
58
|
+
}
|
|
59
|
+
try {
|
|
60
|
+
const tailCmd = `tail -n ${lines} '${file_path.replace(/'/g, "'\\''")}' 2>&1`;
|
|
61
|
+
const initial = await sshManager.exec(session.ssh, tailCmd, 10000);
|
|
62
|
+
if (initial.code !== 0 && !initial.stdout) {
|
|
63
|
+
return {
|
|
64
|
+
content: [{ type: 'text', text: `Error: Failed to read '${file_path}': ${initial.stderr || initial.stdout || 'unknown error'}` }],
|
|
65
|
+
isError: true,
|
|
66
|
+
};
|
|
67
|
+
}
|
|
68
|
+
let output = initial.stdout;
|
|
69
|
+
if (follow_ms > 0) {
|
|
70
|
+
const pollInterval = 500;
|
|
71
|
+
let elapsed = 0;
|
|
72
|
+
while (elapsed < follow_ms) {
|
|
73
|
+
await new Promise((r) => setTimeout(r, pollInterval));
|
|
74
|
+
elapsed += pollInterval;
|
|
75
|
+
try {
|
|
76
|
+
const next = await sshManager.exec(session.ssh, tailCmd, 5000);
|
|
77
|
+
if (next.stdout && next.stdout !== output) {
|
|
78
|
+
const prevLen = output.length;
|
|
79
|
+
if (next.stdout.length > prevLen && next.stdout.endsWith(output.slice(-200))) {
|
|
80
|
+
output = next.stdout;
|
|
81
|
+
}
|
|
82
|
+
else if (next.stdout !== output) {
|
|
83
|
+
const diff = next.stdout.slice(Math.max(0, next.stdout.length - (next.stdout.length - prevLen + 2000)));
|
|
84
|
+
if (diff.length > output.length) {
|
|
85
|
+
output = diff;
|
|
86
|
+
}
|
|
87
|
+
else if (next.stdout.length > output.length) {
|
|
88
|
+
output = next.stdout.slice(-Math.max(lines * 2, output.length + 2000));
|
|
89
|
+
}
|
|
90
|
+
}
|
|
91
|
+
}
|
|
92
|
+
}
|
|
93
|
+
catch {
|
|
94
|
+
logger.warn({ sessionId: session_id, file_path }, 'Follow poll failed, continuing');
|
|
95
|
+
}
|
|
96
|
+
}
|
|
97
|
+
}
|
|
98
|
+
logger.info({ sessionId: session_id, file_path, lines, outputLength: output.length }, 'Log tailed');
|
|
99
|
+
return {
|
|
100
|
+
content: [{ type: 'text', text: output }],
|
|
101
|
+
};
|
|
102
|
+
}
|
|
103
|
+
catch (err) {
|
|
104
|
+
logger.error({ sessionId: session_id, file_path, error: err.message }, 'Tail log failed');
|
|
105
|
+
return {
|
|
106
|
+
content: [{ type: 'text', text: `Error: Tail log failed: ${err.message}` }],
|
|
107
|
+
isError: true,
|
|
108
|
+
};
|
|
109
|
+
}
|
|
110
|
+
}
|
|
111
|
+
//# sourceMappingURL=tail_log.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tail_log.js","sourceRoot":"","sources":["../../src/tools/tail_log.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAE5C,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,CAAC,MAAM,CAAC;IACpC,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;IAC7B,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,kDAAkD,CAAC;IACzF,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC,yCAAyC,CAAC;IACnG,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,6EAA6E,CAAC;CAC3I,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,WAAW,GAAG;IACzB,IAAI,EAAE,UAAU;IAChB,WAAW,EACT,2GAA2G;IAC7G,WAAW,EAAE;QACX,IAAI,EAAE,QAAiB;QACvB,UAAU,EAAE;YACV,UAAU,EAAE;gBACV,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,gCAAgC;aAC9C;YACD,SAAS,EAAE;gBACT,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,kDAAkD;aAChE;YACD,KAAK,EAAE;gBACL,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,iEAAiE;gBAC9E,OAAO,EAAE,GAAG;aACb;YACD,SAAS,EAAE;gBACT,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,qFAAqF;gBAClG,OAAO,EAAE,CAAC;aACX;SACF;QACD,QAAQ,EAAE,CAAC,YAAY,EAAE,WAAW,CAAC;KACtC;CACF,CAAC;AAEF,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,IAAa,EACb,cAA8B,EAC9B,UAAsB;IAEtB,MAAM,MAAM,GAAG,aAAa,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IAC7C,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QACpB,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,sBAAsB,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC;YACxF,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;IAED,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC;IAChE,MAAM,OAAO,GAAG,cAAc,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IAE/C,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,mBAAmB,UAAU,sCAAsC,EAAE,CAAC;YAC/G,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;QACrC,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,sCAAsC,UAAU,WAAW,EAAE,CAAC;YACvG,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,WAAW,KAAK,KAAK,SAAS,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,QAAQ,CAAC;QAC9E,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;QAEnE,IAAI,OAAO,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YAC1C,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,0BAA0B,SAAS,MAAM,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,IAAI,eAAe,EAAE,EAAE,CAAC;gBAC1I,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;QAED,IAAI,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAE5B,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;YAClB,MAAM,YAAY,GAAG,GAAG,CAAC;YACzB,IAAI,OAAO,GAAG,CAAC,CAAC;YAEhB,OAAO,OAAO,GAAG,SAAS,EAAE,CAAC;gBAC3B,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC,CAAC;gBACtD,OAAO,IAAI,YAAY,CAAC;gBACxB,IAAI,CAAC;oBACH,MAAM,IAAI,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;oBAC/D,IAAI,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;wBAC1C,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC;wBAC9B,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,OAAO,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;4BAC7E,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;wBACvB,CAAC;6BAAM,IAAI,IAAI,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;4BAClC,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;4BACxG,IAAI,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;gCAChC,MAAM,GAAG,IAAI,CAAC;4BAChB,CAAC;iCAAM,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;gCAC9C,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,GAAG,CAAC,EAAE,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC;4BACzE,CAAC;wBACH,CAAC;oBACH,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,UAAU,EAAE,SAAS,EAAE,EAAE,gCAAgC,CAAC,CAAC;gBACtF,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,UAAU,EAAE,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,MAAM,CAAC,MAAM,EAAE,EAAE,YAAY,CAAC,CAAC;QAEpG,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;SACnD,CAAC;IACJ,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,CAAC,KAAK,CAAC,EAAE,SAAS,EAAE,UAAU,EAAE,SAAS,EAAE,KAAK,EAAG,GAAa,CAAC,OAAO,EAAE,EAAE,iBAAiB,CAAC,CAAC;QACrG,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,2BAA4B,GAAa,CAAC,OAAO,EAAE,EAAE,CAAC;YAC/F,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,34 @@
|
|
|
1
|
+
import { ToolConfigManager } from '../core/ToolConfigManager.js';
|
|
2
|
+
export declare const toolsConfigTool: {
|
|
3
|
+
name: string;
|
|
4
|
+
description: string;
|
|
5
|
+
inputSchema: {
|
|
6
|
+
type: "object";
|
|
7
|
+
properties: {
|
|
8
|
+
action: {
|
|
9
|
+
type: string;
|
|
10
|
+
enum: string[];
|
|
11
|
+
description: string;
|
|
12
|
+
};
|
|
13
|
+
tool: {
|
|
14
|
+
type: string;
|
|
15
|
+
description: string;
|
|
16
|
+
};
|
|
17
|
+
};
|
|
18
|
+
required: string[];
|
|
19
|
+
};
|
|
20
|
+
};
|
|
21
|
+
export declare function handleToolsConfig(args: unknown, toolConfigManager: ToolConfigManager): Promise<{
|
|
22
|
+
content: {
|
|
23
|
+
type: "text";
|
|
24
|
+
text: string;
|
|
25
|
+
}[];
|
|
26
|
+
isError: boolean;
|
|
27
|
+
} | {
|
|
28
|
+
content: {
|
|
29
|
+
type: "text";
|
|
30
|
+
text: string;
|
|
31
|
+
}[];
|
|
32
|
+
isError?: undefined;
|
|
33
|
+
}>;
|
|
34
|
+
//# sourceMappingURL=tools_config.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tools_config.d.ts","sourceRoot":"","sources":["../../src/tools/tools_config.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AAEjE,eAAO,MAAM,eAAe;;;;;;;;;;;;;;;;;;CAmB3B,CAAC;AAOF,wBAAsB,iBAAiB,CACrC,IAAI,EAAE,OAAO,EACb,iBAAiB,EAAE,iBAAiB;;;;;;;;;;;;GA2ErC"}
|
|
@@ -0,0 +1,98 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
export const toolsConfigTool = {
|
|
3
|
+
name: 'tools_config',
|
|
4
|
+
description: 'Manage tool enable/disable state. Use to disable tools you do not need, reducing token overhead in the MCP tool list. tools_config itself cannot be disabled.',
|
|
5
|
+
inputSchema: {
|
|
6
|
+
type: 'object',
|
|
7
|
+
properties: {
|
|
8
|
+
action: {
|
|
9
|
+
type: 'string',
|
|
10
|
+
enum: ['list', 'enable', 'disable', 'reset'],
|
|
11
|
+
description: 'list=show all, enable/disable=toggle one, reset=re-enable all',
|
|
12
|
+
},
|
|
13
|
+
tool: {
|
|
14
|
+
type: 'string',
|
|
15
|
+
description: 'Tool name (required for enable/disable)',
|
|
16
|
+
},
|
|
17
|
+
},
|
|
18
|
+
required: ['action'],
|
|
19
|
+
},
|
|
20
|
+
};
|
|
21
|
+
const schema = z.object({
|
|
22
|
+
action: z.enum(['list', 'enable', 'disable', 'reset']),
|
|
23
|
+
tool: z.string().optional(),
|
|
24
|
+
});
|
|
25
|
+
export async function handleToolsConfig(args, toolConfigManager) {
|
|
26
|
+
const parsed = schema.safeParse(args);
|
|
27
|
+
if (!parsed.success) {
|
|
28
|
+
return {
|
|
29
|
+
content: [
|
|
30
|
+
{ type: 'text', text: `Invalid arguments: ${parsed.error.message}` },
|
|
31
|
+
],
|
|
32
|
+
isError: true,
|
|
33
|
+
};
|
|
34
|
+
}
|
|
35
|
+
const { action, tool } = parsed.data;
|
|
36
|
+
switch (action) {
|
|
37
|
+
case 'list': {
|
|
38
|
+
const entries = toolConfigManager.getAll();
|
|
39
|
+
const list = Object.entries(entries).map(([name, enabled]) => ({
|
|
40
|
+
tool: name,
|
|
41
|
+
enabled,
|
|
42
|
+
}));
|
|
43
|
+
return {
|
|
44
|
+
content: [
|
|
45
|
+
{
|
|
46
|
+
type: 'text',
|
|
47
|
+
text: list.length > 0 ? JSON.stringify(list, null, 2) : 'All tools enabled (default)',
|
|
48
|
+
},
|
|
49
|
+
],
|
|
50
|
+
};
|
|
51
|
+
}
|
|
52
|
+
case 'enable': {
|
|
53
|
+
if (!tool) {
|
|
54
|
+
return {
|
|
55
|
+
content: [
|
|
56
|
+
{ type: 'text', text: 'Error: "tool" parameter required for enable' },
|
|
57
|
+
],
|
|
58
|
+
isError: true,
|
|
59
|
+
};
|
|
60
|
+
}
|
|
61
|
+
toolConfigManager.setEnabled(tool, true);
|
|
62
|
+
return {
|
|
63
|
+
content: [{ type: 'text', text: `Tool '${tool}' enabled` }],
|
|
64
|
+
};
|
|
65
|
+
}
|
|
66
|
+
case 'disable': {
|
|
67
|
+
if (!tool) {
|
|
68
|
+
return {
|
|
69
|
+
content: [
|
|
70
|
+
{ type: 'text', text: 'Error: "tool" parameter required for disable' },
|
|
71
|
+
],
|
|
72
|
+
isError: true,
|
|
73
|
+
};
|
|
74
|
+
}
|
|
75
|
+
if (tool === 'tools_config') {
|
|
76
|
+
return {
|
|
77
|
+
content: [
|
|
78
|
+
{ type: 'text', text: 'Error: Cannot disable tools_config itself' },
|
|
79
|
+
],
|
|
80
|
+
isError: true,
|
|
81
|
+
};
|
|
82
|
+
}
|
|
83
|
+
toolConfigManager.setEnabled(tool, false);
|
|
84
|
+
return {
|
|
85
|
+
content: [{ type: 'text', text: `Tool '${tool}' disabled` }],
|
|
86
|
+
};
|
|
87
|
+
}
|
|
88
|
+
case 'reset': {
|
|
89
|
+
toolConfigManager.reset();
|
|
90
|
+
return {
|
|
91
|
+
content: [
|
|
92
|
+
{ type: 'text', text: 'All tools reset to enabled (default)' },
|
|
93
|
+
],
|
|
94
|
+
};
|
|
95
|
+
}
|
|
96
|
+
}
|
|
97
|
+
}
|
|
98
|
+
//# sourceMappingURL=tools_config.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tools_config.js","sourceRoot":"","sources":["../../src/tools/tools_config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,MAAM,CAAC,MAAM,eAAe,GAAG;IAC7B,IAAI,EAAE,cAAc;IACpB,WAAW,EACT,+JAA+J;IACjK,WAAW,EAAE;QACX,IAAI,EAAE,QAAiB;QACvB,UAAU,EAAE;YACV,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,CAAC;gBAC5C,WAAW,EAAE,+DAA+D;aAC7E;YACD,IAAI,EAAE;gBACJ,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,yCAAyC;aACvD;SACF;QACD,QAAQ,EAAE,CAAC,QAAQ,CAAC;KACrB;CACF,CAAC;AAEF,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC;IACtB,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;IACtD,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CAC5B,CAAC,CAAC;AAEH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,IAAa,EACb,iBAAoC;IAEpC,MAAM,MAAM,GAAG,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IACtC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QACpB,OAAO;YACL,OAAO,EAAE;gBACP,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,sBAAsB,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE;aAC9E;YACD,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;IAED,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC;IAErC,QAAQ,MAAM,EAAE,CAAC;QACf,KAAK,MAAM,CAAC,CAAC,CAAC;YACZ,MAAM,OAAO,GAAG,iBAAiB,CAAC,MAAM,EAAE,CAAC;YAC3C,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC7D,IAAI,EAAE,IAAI;gBACV,OAAO;aACR,CAAC,CAAC,CAAC;YACJ,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,6BAA6B;qBACtF;iBACF;aACF,CAAC;QACJ,CAAC;QACD,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,OAAO;oBACL,OAAO,EAAE;wBACP,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,6CAA6C,EAAE;qBAC/E;oBACD,OAAO,EAAE,IAAI;iBACd,CAAC;YACJ,CAAC;YACD,iBAAiB,CAAC,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YACzC,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,SAAS,IAAI,WAAW,EAAE,CAAC;aACrE,CAAC;QACJ,CAAC;QACD,KAAK,SAAS,CAAC,CAAC,CAAC;YACf,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,OAAO;oBACL,OAAO,EAAE;wBACP,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,8CAA8C,EAAE;qBAChF;oBACD,OAAO,EAAE,IAAI;iBACd,CAAC;YACJ,CAAC;YACD,IAAI,IAAI,KAAK,cAAc,EAAE,CAAC;gBAC5B,OAAO;oBACL,OAAO,EAAE;wBACP,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,2CAA2C,EAAE;qBAC7E;oBACD,OAAO,EAAE,IAAI;iBACd,CAAC;YACJ,CAAC;YACD,iBAAiB,CAAC,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAC1C,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,SAAS,IAAI,YAAY,EAAE,CAAC;aACtE,CAAC;QACJ,CAAC;QACD,KAAK,OAAO,CAAC,CAAC,CAAC;YACb,iBAAiB,CAAC,KAAK,EAAE,CAAC;YAC1B,OAAO;gBACL,OAAO,EAAE;oBACP,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,sCAAsC,EAAE;iBACxE;aACF,CAAC;QACJ,CAAC;IACH,CAAC;AACH,CAAC"}
|