telegram-ssh-bot 2.2.0 → 2.4.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "telegram-ssh-bot",
-  "version": "2.2.0",
+  "version": "2.4.0",
   "description": "A Telegram bot for secure SSH server management and remote command execution",
   "type": "module",
   "main": "dist/index.js",
@@ -64,11 +64,13 @@
     "dotenv": "^16.3.1",
     "joi": "^17.11.0",
     "node-telegram-bot-api": "^0.61.0",
+    "prompts": "^2.4.2",
     "ssh2": "^1.14.0"
   },
   "devDependencies": {
     "@types/node": "^20.10.0",
     "@types/node-telegram-bot-api": "^0.61.0",
+    "@types/prompts": "^2.4.9",
     "@types/ssh2": "^1.15.0",
     "@yao-pkg/pkg": "^5.11.0",
     "nodemon": "^3.0.1",

package/scripts/setup-env.d.ts

@@ -0,0 +1,28 @@
+/**
+ * Type declarations for setup-env.js
+ */
+
+export function getConfigDir(): string;
+export function getEnvFilePath(): string;
+export function getEnvExamplePath(): string | null;
+export function ensureConfigDir(): string;
+export function generateEncryptionKey(): string;
+export function envFileExists(): boolean;
+export function createEnvFile(options?: { generateKey?: boolean }): boolean;
+export function writeEnvFile(values: {
+  botToken: string;
+  botChatId: string;
+  botOwnerIds: string[];
+  encryptionKey: string;
+}): boolean;
+export function main(options?: {
+  force?: boolean;
+  generateKey?: boolean;
+  silent?: boolean;
+}): {
+  success: boolean;
+  created: boolean;
+  envPath: string;
+  configDir: string;
+  message: string;
+};

package/scripts/setup-env.js

@@ -6,11 +6,11 @@
 
 import { randomBytes } from "crypto";
 import {
-    copyFileSync,
-    existsSync,
-    mkdirSync,
-    readFileSync,
-    writeFileSync,
+  copyFileSync,
+  existsSync,
+  mkdirSync,
+  readFileSync,
+  writeFileSync,
 } from "fs";
 import { homedir } from "os";
 import { join } from "path";
@@ -164,6 +164,62 @@ MONITORING_INTERVAL_MS=300000
   return true;
 }
 
+/**
+ * Write a .env file with specific values
+ * Used by the installation wizard
+ * @param {Object} values - Configuration values
+ * @param {string} values.botToken - Telegram bot token
+ * @param {string} values.botChatId - Telegram chat ID
+ * @param {string[]} values.botOwnerIds - Array of owner IDs
+ * @param {string} values.encryptionKey - Encryption key (64 hex chars)
+ * @returns {boolean} True if the file was written successfully
+ */
+export function writeEnvFile(values) {
+  const { botToken, botChatId, botOwnerIds, encryptionKey } = values;
+
+  // Ensure config directory exists
+  ensureConfigDir();
+
+  const envPath = getEnvFilePath();
+
+  // Build .env content
+  const envContent = `# Telegram Bot Configuration
+BOT_TOKEN=${botToken}
+BOT_CHAT_ID=${botChatId}
+BOT_OWNER_IDS=${botOwnerIds.join(",")}
+
+# Security
+ENCRYPTION_KEY=${encryptionKey}
+
+# SSH Configuration
+SSH_DEFAULT_PORT=22
+SSH_CONNECTION_TIMEOUT=30000
+SSH_COMMAND_TIMEOUT=30000
+SSH_DEFAULT_PRIVATE_KEY_PATH=
+
+# Rate Limiting
+RATE_LIMIT_ENABLED=true
+RATE_LIMIT_WINDOW_MS=60000
+RATE_LIMIT_MAX_REQUESTS=100
+
+# Backup
+BACKUP_ENABLED=true
+BACKUP_INTERVAL_MS=3600000
+BACKUP_MAX_COUNT=10
+
+# Monitoring
+MONITORING_ENABLED=true
+MONITORING_INTERVAL_MS=300000
+
+# Logging
+LOG_LEVEL=info
+LOG_FORMAT=json
+`;
+
+  writeFileSync(envPath, envContent);
+  return true;
+}
+
 /**
  * Main function to set up the environment
  * @param {Object} options - Setup options