798 lines
28 KiB
TypeScript
798 lines
28 KiB
TypeScript
import * as plugins from './plugins.js';
|
|
import * as paths from './paths.js';
|
|
import { tspmIpcClient } from './classes.ipcclient.js';
|
|
import { TspmServiceManager } from './classes.servicemanager.js';
|
|
import { Logger, LogLevel } from './utils.errorhandler.js';
|
|
import type { IProcessConfig } from './classes.tspm.js';
|
|
|
|
export interface CliArguments {
|
|
verbose?: boolean;
|
|
watch?: boolean;
|
|
memory?: string;
|
|
cwd?: string;
|
|
daemon?: boolean;
|
|
test?: boolean;
|
|
name?: string;
|
|
autorestart?: boolean;
|
|
watchPaths?: string[];
|
|
[key: string]: any;
|
|
}
|
|
|
|
// Helper function to parse memory strings (e.g., "512MB", "2GB")
|
|
function parseMemoryString(memStr: string): number {
|
|
const units = {
|
|
KB: 1024,
|
|
MB: 1024 * 1024,
|
|
GB: 1024 * 1024 * 1024,
|
|
};
|
|
|
|
const match = memStr.toUpperCase().match(/^(\d+(?:\.\d+)?)\s*(KB|MB|GB)?$/);
|
|
if (!match) {
|
|
throw new Error(
|
|
`Invalid memory format: ${memStr}. Use format like "512MB" or "2GB"`,
|
|
);
|
|
}
|
|
|
|
const value = parseFloat(match[1]);
|
|
const unit = (match[2] || 'MB') as keyof typeof units;
|
|
|
|
return Math.floor(value * units[unit]);
|
|
}
|
|
|
|
// Helper function to format memory for display
|
|
function formatMemory(bytes: number): string {
|
|
if (bytes >= 1024 * 1024 * 1024) {
|
|
return `${(bytes / (1024 * 1024 * 1024)).toFixed(1)} GB`;
|
|
} else if (bytes >= 1024 * 1024) {
|
|
return `${(bytes / (1024 * 1024)).toFixed(1)} MB`;
|
|
} else if (bytes >= 1024) {
|
|
return `${(bytes / 1024).toFixed(1)} KB`;
|
|
} else {
|
|
return `${bytes} B`;
|
|
}
|
|
}
|
|
|
|
// Helper function to handle daemon connection errors
|
|
function handleDaemonError(error: any, action: string): void {
|
|
if (error.message?.includes('daemon is not running') ||
|
|
error.message?.includes('Not connected') ||
|
|
error.message?.includes('ECONNREFUSED')) {
|
|
console.error(`Error: Cannot ${action} - TSPM daemon is not running.`);
|
|
console.log('\nTo start the daemon, run one of:');
|
|
console.log(' tspm daemon start - Start for this session only');
|
|
console.log(' tspm enable - Enable as system service (recommended)');
|
|
} else {
|
|
console.error(`Error ${action}:`, error.message);
|
|
}
|
|
process.exit(1);
|
|
}
|
|
|
|
// Helper function for padding strings
|
|
function pad(str: string, length: number): string {
|
|
return str.length > length
|
|
? str.substring(0, length - 3) + '...'
|
|
: str.padEnd(length);
|
|
}
|
|
|
|
export const run = async (): Promise<void> => {
|
|
const cliLogger = new Logger('CLI');
|
|
const tspmProjectinfo = new plugins.projectinfo.ProjectInfo(paths.packageDir);
|
|
|
|
// Check if debug mode is enabled
|
|
const debugMode = process.env.TSPM_DEBUG === 'true';
|
|
if (debugMode) {
|
|
cliLogger.setLevel(LogLevel.DEBUG);
|
|
cliLogger.debug('Debug mode enabled');
|
|
}
|
|
|
|
const smartcliInstance = new plugins.smartcli.Smartcli();
|
|
smartcliInstance.addVersion(tspmProjectinfo.npm.version);
|
|
|
|
// Default command - show help and list processes
|
|
smartcliInstance.standardCommand().subscribe({
|
|
next: async (argvArg: CliArguments) => {
|
|
console.log(
|
|
`TSPM - TypeScript Process Manager v${tspmProjectinfo.npm.version}`,
|
|
);
|
|
console.log('Usage: tspm [command] [options]');
|
|
console.log('\nService Management:');
|
|
console.log(' enable Enable TSPM as system service (systemd)');
|
|
console.log(' disable Disable TSPM system service');
|
|
console.log('\nProcess Commands:');
|
|
console.log(' start <script> Start a process');
|
|
console.log(' list List all processes');
|
|
console.log(' stop <id> Stop a process');
|
|
console.log(' restart <id> Restart a process');
|
|
console.log(' delete <id> Delete a process');
|
|
console.log(' describe <id> Show details for a process');
|
|
console.log(' logs <id> Show logs for a process');
|
|
console.log(' start-all Start all saved processes');
|
|
console.log(' stop-all Stop all processes');
|
|
console.log(' restart-all Restart all processes');
|
|
console.log('\nDaemon Commands:');
|
|
console.log(' daemon start Start daemon manually (current session)');
|
|
console.log(' daemon stop Stop the daemon');
|
|
console.log(' daemon status Show daemon status');
|
|
console.log(
|
|
'\nUse tspm [command] --help for more information about a command.',
|
|
);
|
|
|
|
// Show current process list
|
|
console.log('\nProcess List:');
|
|
|
|
try {
|
|
const response = await tspmIpcClient.request('list', {});
|
|
const processes = response.processes;
|
|
|
|
if (processes.length === 0) {
|
|
console.log(
|
|
' No processes running. Use "tspm start" to start a process.',
|
|
);
|
|
} else {
|
|
console.log(
|
|
'┌─────────┬─────────────┬───────────┬───────────┬──────────┐',
|
|
);
|
|
console.log(
|
|
'│ ID │ Name │ Status │ Memory │ Restarts │',
|
|
);
|
|
console.log(
|
|
'├─────────┼─────────────┼───────────┼───────────┼──────────┤',
|
|
);
|
|
|
|
for (const proc of processes) {
|
|
const statusColor =
|
|
proc.status === 'online'
|
|
? '\x1b[32m'
|
|
: proc.status === 'errored'
|
|
? '\x1b[31m'
|
|
: '\x1b[33m';
|
|
const resetColor = '\x1b[0m';
|
|
|
|
console.log(
|
|
`│ ${pad(proc.id, 7)} │ ${pad(proc.id, 11)} │ ${statusColor}${pad(proc.status, 9)}${resetColor} │ ${pad(formatMemory(proc.memory), 9)} │ ${pad(proc.restarts.toString(), 8)} │`,
|
|
);
|
|
}
|
|
|
|
console.log(
|
|
'└─────────┴─────────────┴───────────┴───────────┴──────────┘',
|
|
);
|
|
}
|
|
} catch (error) {
|
|
console.error('Error: TSPM daemon is not running.');
|
|
console.log('\nTo start the daemon, run one of:');
|
|
console.log(' tspm daemon start - Start for this session only');
|
|
console.log(' tspm enable - Enable as system service (recommended)');
|
|
}
|
|
},
|
|
error: (err) => {
|
|
cliLogger.error(err);
|
|
},
|
|
complete: () => {},
|
|
});
|
|
|
|
// Start command
|
|
smartcliInstance.addCommand('start').subscribe({
|
|
next: async (argvArg: CliArguments) => {
|
|
try {
|
|
const script = argvArg._[1];
|
|
if (!script) {
|
|
console.error('Error: Please provide a script to run');
|
|
console.log('Usage: tspm start <script> [options]');
|
|
console.log('\nOptions:');
|
|
console.log(' --name <name> Name for the process');
|
|
console.log(
|
|
' --memory <size> Memory limit (e.g., "512MB", "2GB")',
|
|
);
|
|
console.log(' --cwd <path> Working directory');
|
|
console.log(
|
|
' --watch Watch for file changes and restart',
|
|
);
|
|
console.log(' --watch-paths <paths> Comma-separated paths to watch');
|
|
console.log(' --autorestart Auto-restart on crash');
|
|
return;
|
|
}
|
|
|
|
const memoryLimit = argvArg.memory
|
|
? parseMemoryString(argvArg.memory)
|
|
: 512 * 1024 * 1024; // Default 512MB
|
|
const projectDir = argvArg.cwd || process.cwd();
|
|
const name = argvArg.name || script;
|
|
const watch = argvArg.watch || false;
|
|
const autorestart = argvArg.autorestart !== false; // Default true
|
|
const watchPaths = argvArg.watchPaths
|
|
? typeof argvArg.watchPaths === 'string'
|
|
? (argvArg.watchPaths as string).split(',')
|
|
: argvArg.watchPaths
|
|
: undefined;
|
|
|
|
const processConfig: IProcessConfig = {
|
|
id: name.replace(/[^a-zA-Z0-9-_]/g, '_'),
|
|
name,
|
|
command: script,
|
|
projectDir,
|
|
memoryLimitBytes: memoryLimit,
|
|
autorestart,
|
|
watch,
|
|
watchPaths,
|
|
};
|
|
|
|
console.log(`Starting process: ${name}`);
|
|
console.log(` Command: ${script}`);
|
|
console.log(` Directory: ${projectDir}`);
|
|
console.log(` Memory limit: ${formatMemory(memoryLimit)}`);
|
|
console.log(` Auto-restart: ${autorestart}`);
|
|
if (watch) {
|
|
console.log(` Watch mode: enabled`);
|
|
if (watchPaths) {
|
|
console.log(` Watch paths: ${watchPaths.join(', ')}`);
|
|
}
|
|
}
|
|
|
|
const response = await tspmIpcClient.request('start', {
|
|
config: processConfig,
|
|
});
|
|
|
|
console.log(`✓ Process started successfully`);
|
|
console.log(` ID: ${response.processId}`);
|
|
console.log(` PID: ${response.pid || 'N/A'}`);
|
|
console.log(` Status: ${response.status}`);
|
|
} catch (error) {
|
|
handleDaemonError(error, 'start process');
|
|
}
|
|
},
|
|
error: (err) => {
|
|
cliLogger.error(err);
|
|
},
|
|
complete: () => {},
|
|
});
|
|
|
|
// Stop command
|
|
smartcliInstance.addCommand('stop').subscribe({
|
|
next: async (argvArg: CliArguments) => {
|
|
try {
|
|
const id = argvArg._[1];
|
|
if (!id) {
|
|
console.error('Error: Please provide a process ID');
|
|
console.log('Usage: tspm stop <id>');
|
|
return;
|
|
}
|
|
|
|
console.log(`Stopping process: ${id}`);
|
|
const response = await tspmIpcClient.request('stop', { id });
|
|
|
|
if (response.success) {
|
|
console.log(`✓ ${response.message}`);
|
|
} else {
|
|
console.error(`✗ Failed to stop process: ${response.message}`);
|
|
}
|
|
} catch (error) {
|
|
handleDaemonError(error, 'stop process');
|
|
}
|
|
},
|
|
error: (err) => {
|
|
cliLogger.error(err);
|
|
},
|
|
complete: () => {},
|
|
});
|
|
|
|
// Restart command
|
|
smartcliInstance.addCommand('restart').subscribe({
|
|
next: async (argvArg: CliArguments) => {
|
|
try {
|
|
const id = argvArg._[1];
|
|
if (!id) {
|
|
console.error('Error: Please provide a process ID');
|
|
console.log('Usage: tspm restart <id>');
|
|
return;
|
|
}
|
|
|
|
console.log(`Restarting process: ${id}`);
|
|
const response = await tspmIpcClient.request('restart', { id });
|
|
|
|
console.log(`✓ Process restarted successfully`);
|
|
console.log(` ID: ${response.processId}`);
|
|
console.log(` PID: ${response.pid || 'N/A'}`);
|
|
console.log(` Status: ${response.status}`);
|
|
} catch (error) {
|
|
handleDaemonError(error, 'restart process');
|
|
}
|
|
},
|
|
error: (err) => {
|
|
cliLogger.error(err);
|
|
},
|
|
complete: () => {},
|
|
});
|
|
|
|
// Delete command
|
|
smartcliInstance.addCommand('delete').subscribe({
|
|
next: async (argvArg: CliArguments) => {
|
|
try {
|
|
const id = argvArg._[1];
|
|
if (!id) {
|
|
console.error('Error: Please provide a process ID');
|
|
console.log('Usage: tspm delete <id>');
|
|
return;
|
|
}
|
|
|
|
console.log(`Deleting process: ${id}`);
|
|
const response = await tspmIpcClient.request('delete', { id });
|
|
|
|
if (response.success) {
|
|
console.log(`✓ ${response.message}`);
|
|
} else {
|
|
console.error(`✗ Failed to delete process: ${response.message}`);
|
|
}
|
|
} catch (error) {
|
|
handleDaemonError(error, 'delete process');
|
|
}
|
|
},
|
|
error: (err) => {
|
|
cliLogger.error(err);
|
|
},
|
|
complete: () => {},
|
|
});
|
|
|
|
// List command
|
|
smartcliInstance.addCommand('list').subscribe({
|
|
next: async (argvArg: CliArguments) => {
|
|
try {
|
|
const response = await tspmIpcClient.request('list', {});
|
|
const processes = response.processes;
|
|
|
|
if (processes.length === 0) {
|
|
console.log('No processes running.');
|
|
} else {
|
|
console.log('Process List:');
|
|
console.log(
|
|
'┌─────────┬─────────────┬───────────┬───────────┬──────────┬──────────┐',
|
|
);
|
|
console.log(
|
|
'│ ID │ Name │ Status │ PID │ Memory │ Restarts │',
|
|
);
|
|
console.log(
|
|
'├─────────┼─────────────┼───────────┼───────────┼──────────┼──────────┤',
|
|
);
|
|
|
|
for (const proc of processes) {
|
|
const statusColor =
|
|
proc.status === 'online'
|
|
? '\x1b[32m'
|
|
: proc.status === 'errored'
|
|
? '\x1b[31m'
|
|
: '\x1b[33m';
|
|
const resetColor = '\x1b[0m';
|
|
|
|
console.log(
|
|
`│ ${pad(proc.id, 7)} │ ${pad(proc.id, 11)} │ ${statusColor}${pad(proc.status, 9)}${resetColor} │ ${pad((proc.pid || '-').toString(), 9)} │ ${pad(formatMemory(proc.memory), 8)} │ ${pad(proc.restarts.toString(), 8)} │`,
|
|
);
|
|
}
|
|
|
|
console.log(
|
|
'└─────────┴─────────────┴───────────┴───────────┴──────────┴──────────┘',
|
|
);
|
|
}
|
|
} catch (error) {
|
|
handleDaemonError(error, 'list processes');
|
|
}
|
|
},
|
|
error: (err) => {
|
|
cliLogger.error(err);
|
|
},
|
|
complete: () => {},
|
|
});
|
|
|
|
// Describe command
|
|
smartcliInstance.addCommand('describe').subscribe({
|
|
next: async (argvArg: CliArguments) => {
|
|
try {
|
|
const id = argvArg._[1];
|
|
if (!id) {
|
|
console.error('Error: Please provide a process ID');
|
|
console.log('Usage: tspm describe <id>');
|
|
return;
|
|
}
|
|
|
|
const response = await tspmIpcClient.request('describe', { id });
|
|
|
|
console.log(`Process Details: ${id}`);
|
|
console.log('─'.repeat(40));
|
|
console.log(`Status: ${response.processInfo.status}`);
|
|
console.log(`PID: ${response.processInfo.pid || 'N/A'}`);
|
|
console.log(
|
|
`Memory: ${formatMemory(response.processInfo.memory)}`,
|
|
);
|
|
console.log(
|
|
`CPU: ${response.processInfo.cpu ? response.processInfo.cpu.toFixed(1) + '%' : 'N/A'}`,
|
|
);
|
|
console.log(
|
|
`Uptime: ${response.processInfo.uptime ? Math.floor(response.processInfo.uptime / 1000) + 's' : 'N/A'}`,
|
|
);
|
|
console.log(`Restarts: ${response.processInfo.restarts}`);
|
|
console.log('\nConfiguration:');
|
|
console.log(`Command: ${response.config.command}`);
|
|
console.log(`Directory: ${response.config.projectDir}`);
|
|
console.log(
|
|
`Memory Limit: ${formatMemory(response.config.memoryLimitBytes)}`,
|
|
);
|
|
console.log(`Auto-restart: ${response.config.autorestart}`);
|
|
if (response.config.watch) {
|
|
console.log(`Watch: enabled`);
|
|
if (response.config.watchPaths) {
|
|
console.log(
|
|
`Watch Paths: ${response.config.watchPaths.join(', ')}`,
|
|
);
|
|
}
|
|
}
|
|
} catch (error) {
|
|
handleDaemonError(error, 'describe process');
|
|
}
|
|
},
|
|
error: (err) => {
|
|
cliLogger.error(err);
|
|
},
|
|
complete: () => {},
|
|
});
|
|
|
|
// Logs command
|
|
smartcliInstance.addCommand('logs').subscribe({
|
|
next: async (argvArg: CliArguments) => {
|
|
try {
|
|
const id = argvArg._[1];
|
|
if (!id) {
|
|
console.error('Error: Please provide a process ID');
|
|
console.log('Usage: tspm logs <id> [options]');
|
|
console.log('\nOptions:');
|
|
console.log(' --lines <n> Number of lines to show (default: 50)');
|
|
console.log(' --follow Stream logs in real-time (like tail -f)');
|
|
return;
|
|
}
|
|
|
|
const lines = argvArg.lines || 50;
|
|
const follow = argvArg.follow || argvArg.f || false;
|
|
|
|
// Get initial logs
|
|
const response = await tspmIpcClient.request('getLogs', { id, lines });
|
|
|
|
if (!follow) {
|
|
// Static log output
|
|
console.log(`Logs for process: ${id} (last ${lines} lines)`);
|
|
console.log('─'.repeat(60));
|
|
|
|
for (const log of response.logs) {
|
|
const timestamp = new Date(log.timestamp).toLocaleTimeString();
|
|
const prefix = log.type === 'stdout' ? '[OUT]' : log.type === 'stderr' ? '[ERR]' : '[SYS]';
|
|
console.log(`${timestamp} ${prefix} ${log.message}`);
|
|
}
|
|
} else {
|
|
// Streaming log output
|
|
console.log(`Logs for process: ${id} (streaming...)`);
|
|
console.log('─'.repeat(60));
|
|
|
|
// Display initial logs
|
|
let lastSeq = 0;
|
|
for (const log of response.logs) {
|
|
const timestamp = new Date(log.timestamp).toLocaleTimeString();
|
|
const prefix = log.type === 'stdout' ? '[OUT]' : log.type === 'stderr' ? '[ERR]' : '[SYS]';
|
|
console.log(`${timestamp} ${prefix} ${log.message}`);
|
|
if (log.seq !== undefined) {
|
|
lastSeq = Math.max(lastSeq, log.seq);
|
|
}
|
|
}
|
|
|
|
// Subscribe to real-time updates
|
|
await tspmIpcClient.subscribe(id, (log: any) => {
|
|
// Check for sequence gap or duplicate
|
|
if (log.seq !== undefined && log.seq <= lastSeq) {
|
|
return; // Skip duplicate
|
|
}
|
|
if (log.seq !== undefined && log.seq > lastSeq + 1) {
|
|
console.log(`[WARNING] Log gap detected: expected seq ${lastSeq + 1}, got ${log.seq}`);
|
|
}
|
|
|
|
const timestamp = new Date(log.timestamp).toLocaleTimeString();
|
|
const prefix = log.type === 'stdout' ? '[OUT]' : log.type === 'stderr' ? '[ERR]' : '[SYS]';
|
|
console.log(`${timestamp} ${prefix} ${log.message}`);
|
|
|
|
if (log.seq !== undefined) {
|
|
lastSeq = log.seq;
|
|
}
|
|
});
|
|
|
|
// Handle Ctrl+C gracefully
|
|
let isCleaningUp = false;
|
|
const cleanup = async () => {
|
|
if (isCleaningUp) return;
|
|
isCleaningUp = true;
|
|
console.log('\n\nStopping log stream...');
|
|
try {
|
|
await tspmIpcClient.unsubscribe(id);
|
|
await tspmIpcClient.disconnect();
|
|
} catch (err) {
|
|
// Ignore cleanup errors
|
|
}
|
|
process.exit(0);
|
|
};
|
|
|
|
process.on('SIGINT', cleanup);
|
|
process.on('SIGTERM', cleanup);
|
|
|
|
// Keep the process alive
|
|
await new Promise(() => {}); // Block forever until interrupted
|
|
}
|
|
} catch (error) {
|
|
handleDaemonError(error, 'get logs');
|
|
}
|
|
},
|
|
error: (err) => {
|
|
cliLogger.error(err);
|
|
},
|
|
complete: () => {},
|
|
});
|
|
|
|
// Start-all command
|
|
smartcliInstance.addCommand('start-all').subscribe({
|
|
next: async (argvArg: CliArguments) => {
|
|
try {
|
|
console.log('Starting all processes...');
|
|
const response = await tspmIpcClient.request('startAll', {});
|
|
|
|
if (response.started.length > 0) {
|
|
console.log(`✓ Started ${response.started.length} processes:`);
|
|
for (const id of response.started) {
|
|
console.log(` - ${id}`);
|
|
}
|
|
}
|
|
|
|
if (response.failed.length > 0) {
|
|
console.log(`✗ Failed to start ${response.failed.length} processes:`);
|
|
for (const failure of response.failed) {
|
|
console.log(` - ${failure.id}: ${failure.error}`);
|
|
}
|
|
}
|
|
} catch (error) {
|
|
handleDaemonError(error, 'start all processes');
|
|
}
|
|
},
|
|
error: (err) => {
|
|
cliLogger.error(err);
|
|
},
|
|
complete: () => {},
|
|
});
|
|
|
|
// Stop-all command
|
|
smartcliInstance.addCommand('stop-all').subscribe({
|
|
next: async (argvArg: CliArguments) => {
|
|
try {
|
|
console.log('Stopping all processes...');
|
|
const response = await tspmIpcClient.request('stopAll', {});
|
|
|
|
if (response.stopped.length > 0) {
|
|
console.log(`✓ Stopped ${response.stopped.length} processes:`);
|
|
for (const id of response.stopped) {
|
|
console.log(` - ${id}`);
|
|
}
|
|
}
|
|
|
|
if (response.failed.length > 0) {
|
|
console.log(`✗ Failed to stop ${response.failed.length} processes:`);
|
|
for (const failure of response.failed) {
|
|
console.log(` - ${failure.id}: ${failure.error}`);
|
|
}
|
|
}
|
|
} catch (error) {
|
|
handleDaemonError(error, 'stop all processes');
|
|
}
|
|
},
|
|
error: (err) => {
|
|
cliLogger.error(err);
|
|
},
|
|
complete: () => {},
|
|
});
|
|
|
|
// Restart-all command
|
|
smartcliInstance.addCommand('restart-all').subscribe({
|
|
next: async (argvArg: CliArguments) => {
|
|
try {
|
|
console.log('Restarting all processes...');
|
|
const response = await tspmIpcClient.request('restartAll', {});
|
|
|
|
if (response.restarted.length > 0) {
|
|
console.log(`✓ Restarted ${response.restarted.length} processes:`);
|
|
for (const id of response.restarted) {
|
|
console.log(` - ${id}`);
|
|
}
|
|
}
|
|
|
|
if (response.failed.length > 0) {
|
|
console.log(
|
|
`✗ Failed to restart ${response.failed.length} processes:`,
|
|
);
|
|
for (const failure of response.failed) {
|
|
console.log(` - ${failure.id}: ${failure.error}`);
|
|
}
|
|
}
|
|
} catch (error) {
|
|
handleDaemonError(error, 'restart all processes');
|
|
}
|
|
},
|
|
error: (err) => {
|
|
cliLogger.error(err);
|
|
},
|
|
complete: () => {},
|
|
});
|
|
|
|
// Daemon commands
|
|
smartcliInstance.addCommand('daemon').subscribe({
|
|
next: async (argvArg: CliArguments) => {
|
|
const subCommand = argvArg._[1];
|
|
|
|
switch (subCommand) {
|
|
case 'start':
|
|
try {
|
|
const status = await tspmIpcClient.getDaemonStatus();
|
|
if (status) {
|
|
console.log('TSPM daemon is already running');
|
|
console.log(` PID: ${status.pid}`);
|
|
console.log(
|
|
` Uptime: ${Math.floor((status.uptime || 0) / 1000)}s`,
|
|
);
|
|
console.log(` Processes: ${status.processCount}`);
|
|
return;
|
|
}
|
|
|
|
console.log('Starting TSPM daemon manually...');
|
|
|
|
// Import spawn to start daemon process
|
|
const { spawn } = await import('child_process');
|
|
const daemonScript = plugins.path.join(
|
|
paths.packageDir,
|
|
'dist_ts',
|
|
'daemon.js',
|
|
);
|
|
|
|
// Start daemon as a regular background process (not detached)
|
|
const daemonProcess = spawn(process.execPath, [daemonScript], {
|
|
stdio: 'ignore',
|
|
env: {
|
|
...process.env,
|
|
TSPM_DAEMON_MODE: 'true',
|
|
},
|
|
});
|
|
|
|
console.log(`Started daemon with PID: ${daemonProcess.pid}`);
|
|
|
|
// Wait for daemon to be ready
|
|
await new Promise(resolve => setTimeout(resolve, 2000));
|
|
|
|
const newStatus = await tspmIpcClient.getDaemonStatus();
|
|
if (newStatus) {
|
|
console.log('✓ TSPM daemon started successfully');
|
|
console.log(` PID: ${newStatus.pid}`);
|
|
console.log('\nNote: This daemon will run until you stop it or logout.');
|
|
console.log('For automatic startup, use "tspm enable" instead.');
|
|
}
|
|
} catch (error) {
|
|
console.error('Error starting daemon:', error.message);
|
|
process.exit(1);
|
|
}
|
|
break;
|
|
|
|
case 'start-service':
|
|
// This is called by systemd - start the daemon directly
|
|
console.log('Starting TSPM daemon for systemd service...');
|
|
const { startDaemon } = await import('./classes.daemon.js');
|
|
await startDaemon();
|
|
break;
|
|
|
|
case 'stop':
|
|
try {
|
|
console.log('Stopping TSPM daemon...');
|
|
await tspmIpcClient.stopDaemon(true);
|
|
console.log('✓ TSPM daemon stopped successfully');
|
|
} catch (error) {
|
|
console.error('Error stopping daemon:', error.message);
|
|
process.exit(1);
|
|
}
|
|
break;
|
|
|
|
case 'status':
|
|
try {
|
|
const status = await tspmIpcClient.getDaemonStatus();
|
|
if (!status) {
|
|
console.log('TSPM daemon is not running');
|
|
console.log('Use "tspm daemon start" to start it');
|
|
return;
|
|
}
|
|
|
|
console.log('TSPM Daemon Status:');
|
|
console.log('─'.repeat(40));
|
|
console.log(`Status: ${status.status}`);
|
|
console.log(`PID: ${status.pid}`);
|
|
console.log(
|
|
`Uptime: ${Math.floor((status.uptime || 0) / 1000)}s`,
|
|
);
|
|
console.log(`Processes: ${status.processCount}`);
|
|
console.log(
|
|
`Memory: ${formatMemory(status.memoryUsage || 0)}`,
|
|
);
|
|
console.log(`CPU: ${status.cpuUsage?.toFixed(1) || 0}s`);
|
|
|
|
// Disconnect from daemon after getting status
|
|
await tspmIpcClient.disconnect();
|
|
} catch (error) {
|
|
console.error('Error getting daemon status:', error.message);
|
|
process.exit(1);
|
|
}
|
|
break;
|
|
|
|
default:
|
|
console.log('Usage: tspm daemon <command>');
|
|
console.log('\nCommands:');
|
|
console.log(' start Start the TSPM daemon');
|
|
console.log(' stop Stop the TSPM daemon');
|
|
console.log(' status Show daemon status');
|
|
break;
|
|
}
|
|
},
|
|
error: (err) => {
|
|
cliLogger.error(err);
|
|
},
|
|
complete: () => {},
|
|
});
|
|
|
|
// Enable command - Enable TSPM daemon as systemd service
|
|
smartcliInstance.addCommand('enable').subscribe({
|
|
next: async (argvArg: CliArguments) => {
|
|
try {
|
|
const serviceManager = new TspmServiceManager();
|
|
console.log('Enabling TSPM daemon as system service...');
|
|
|
|
await serviceManager.enableService();
|
|
|
|
console.log('✓ TSPM daemon enabled and started as system service');
|
|
console.log(' The daemon will now start automatically on system boot');
|
|
console.log(' Use "tspm disable" to remove the service');
|
|
} catch (error) {
|
|
console.error('Error enabling service:', error.message);
|
|
if (error.message.includes('permission') || error.message.includes('denied')) {
|
|
console.log('\nNote: You may need to run this command with sudo');
|
|
}
|
|
process.exit(1);
|
|
}
|
|
},
|
|
error: (err) => {
|
|
cliLogger.error(err);
|
|
},
|
|
complete: () => {},
|
|
});
|
|
|
|
// Disable command - Disable TSPM daemon systemd service
|
|
smartcliInstance.addCommand('disable').subscribe({
|
|
next: async (argvArg: CliArguments) => {
|
|
try {
|
|
const serviceManager = new TspmServiceManager();
|
|
console.log('Disabling TSPM daemon service...');
|
|
|
|
await serviceManager.disableService();
|
|
|
|
console.log('✓ TSPM daemon service disabled');
|
|
console.log(' The daemon will no longer start on system boot');
|
|
console.log(' Use "tspm enable" to re-enable the service');
|
|
} catch (error) {
|
|
console.error('Error disabling service:', error.message);
|
|
if (error.message.includes('permission') || error.message.includes('denied')) {
|
|
console.log('\nNote: You may need to run this command with sudo');
|
|
}
|
|
process.exit(1);
|
|
}
|
|
},
|
|
error: (err) => {
|
|
cliLogger.error(err);
|
|
},
|
|
complete: () => {},
|
|
});
|
|
|
|
// Start parsing commands
|
|
smartcliInstance.startParse();
|
|
};
|