221 lines
7.9 KiB
JavaScript
221 lines
7.9 KiB
JavaScript
import 'dotenv/config';
|
||
import { Telegraf, Markup } from 'telegraf';
|
||
import { generatePassword, strength } from './lib/generator.js';
|
||
|
||
const TOKEN = process.env.BOT_TOKEN;
|
||
const OWNER_ID = Number(process.env.OWNER_ID);
|
||
|
||
if (!TOKEN) {
|
||
console.error('BOT_TOKEN not set in .env');
|
||
process.exit(1);
|
||
}
|
||
|
||
const bot = new Telegraf(TOKEN, { handlerTimeout: 5_000 });
|
||
|
||
/* ---------- User settings state (in-memory) ---------- */
|
||
const defaults = () => ({
|
||
length: 16,
|
||
upper: true,
|
||
lower: true,
|
||
digits: true,
|
||
symbols: true,
|
||
excludeAmbiguous: false,
|
||
pronounceable: false,
|
||
});
|
||
const users = new Map();
|
||
const history = new Map(); // userId -> [passwords]
|
||
|
||
const state = (id) => {
|
||
if (!users.has(id)) users.set(id, defaults());
|
||
return users.get(id);
|
||
};
|
||
|
||
const addHistory = (id, pwd) => {
|
||
if (!history.has(id)) history.set(id, []);
|
||
const h = history.get(id);
|
||
h.unshift(pwd);
|
||
if (h.length > 5) h.pop();
|
||
};
|
||
|
||
/* ---------- Keyboard builders ---------- */
|
||
const mainMenu = () =>
|
||
Markup.inlineKeyboard([
|
||
[Markup.button.callback('🔑 Сгенерировать пароль', 'gen')],
|
||
[Markup.button.callback('⚙️ Настройки', 'settings')],
|
||
[Markup.button.callback('📋 История', 'history')],
|
||
[Markup.button.callback('ℹ️ О боте', 'about')],
|
||
]);
|
||
|
||
const settingsKb = (s) => {
|
||
const on = (v) => (v ? '✅' : '❎');
|
||
return Markup.inlineKeyboard([
|
||
[
|
||
Markup.button.callback('➖ 1', 'len:-1'),
|
||
Markup.button.callback(`📏 Длина: ${s.length}`, 'noop'),
|
||
Markup.button.callback('➕ 1', 'len:+1'),
|
||
],
|
||
[
|
||
Markup.button.callback('➖ 5', 'len:-5'),
|
||
Markup.button.callback('➕ 5', 'len:+5'),
|
||
],
|
||
[Markup.button.callback(`${on(s.upper)} Заглавные (A-Z)`, 'set:upper')],
|
||
[Markup.button.callback(`${on(s.lower)} Строчные (a-z)`, 'set:lower')],
|
||
[Markup.button.callback(`${on(s.digits)} Цифры (0-9)`, 'set:digits')],
|
||
[Markup.button.callback(`${on(s.symbols)} Символы (!@#$)`, 'set:symbols')],
|
||
[Markup.button.callback(`${on(s.excludeAmbiguous)} Без похожих (il1Lo0O)`, 'set:excludeAmbiguous')],
|
||
[Markup.button.callback(`${on(s.pronounceable)} Произносимый`, 'set:pronounceable')],
|
||
[Markup.button.callback('🔑 Сгенерировать', 'gen'), Markup.button.callback('⬅️ Назад', 'main')],
|
||
]);
|
||
};
|
||
|
||
const resultKb = () =>
|
||
Markup.inlineKeyboard([
|
||
[Markup.button.callback('🔁 Ещё раз', 'gen')],
|
||
[Markup.button.callback('⚙️ Настройки', 'settings'), Markup.button.callback('⬅️ Меню', 'main')],
|
||
]);
|
||
|
||
const backKb = (cb = 'main') =>
|
||
Markup.inlineKeyboard([[Markup.button.callback('⬅️ Назад', cb)]]);
|
||
|
||
const passwordText = (pwd, s) => {
|
||
const st = strength(pwd);
|
||
return [
|
||
'🔑 Ваш пароль:',
|
||
'',
|
||
'`' + pwd + '`',
|
||
'',
|
||
`${st.bar} ${st.level} · ~${st.bits} бит энтропии`,
|
||
`⚙️ Длина ${s.length} · ${modeLabel(s)}`,
|
||
'',
|
||
'🛡 crypto.randomInt, без Math.random.',
|
||
].join('\n');
|
||
};
|
||
|
||
const modeLabel = (s) => {
|
||
if (s.pronounceable) return 'произносимый';
|
||
const parts = [];
|
||
if (s.upper) parts.push('A-Z');
|
||
if (s.lower) parts.push('a-z');
|
||
if (s.digits) parts.push('0-9');
|
||
if (s.symbols) parts.push('!@#');
|
||
return parts.join(' + ') || 'a-z';
|
||
};
|
||
|
||
/* ---------- Access restriction: OWNER_ID only ---------- */
|
||
const ownerOnly = async (ctx, next) => {
|
||
if (!Number.isFinite(OWNER_ID) || ctx.from?.id === OWNER_ID) return next();
|
||
if (ctx.callbackQuery) await ctx.answerCbQuery('Доступ запрещен').catch(() => {});
|
||
return ctx.reply('⛔ Доступ разрешен только владельцу бота.').catch(() => {});
|
||
};
|
||
bot.use(ownerOnly);
|
||
|
||
/* ---------- Handlers ---------- */
|
||
bot.start((ctx) => {
|
||
state(ctx.from.id);
|
||
return ctx.reply(
|
||
'👋 Привет! Я генератор паролей.\n\n' +
|
||
'🔑 Жми «Сгенерировать» — пароль создаётся криптографически стойким генератором Node.js (crypto.randomInt).\n' +
|
||
'⚙️ В настройках — длина, наборы символов, режимы.',
|
||
mainMenu()
|
||
);
|
||
});
|
||
|
||
bot.action('noop', (ctx) => ctx.answerCbQuery());
|
||
|
||
bot.action('main', (ctx) =>
|
||
ctx.editMessageText('Выбери действие:', mainMenu()).catch(() => {})
|
||
);
|
||
|
||
bot.action('settings', (ctx) => {
|
||
const s = state(ctx.from.id);
|
||
return ctx.editMessageText('⚙️ Настройки генерации:', settingsKb(s)).catch(() => {});
|
||
});
|
||
|
||
bot.action(/^len:(-?\d+)$/, (ctx) => {
|
||
const delta = Number(ctx.match[1]);
|
||
const s = state(ctx.from.id);
|
||
s.length = Math.min(128, Math.max(4, s.length + delta));
|
||
return ctx
|
||
.editMessageText(`⚙️ Настройки генерации:\nДлина: ${s.length} (4–128)`, settingsKb(s))
|
||
.then(() => ctx.answerCbQuery(`Длина: ${s.length}`))
|
||
.catch(() => {});
|
||
});
|
||
|
||
bot.action(/^set:(\w+)$/, (ctx) => {
|
||
const key = ctx.match[1];
|
||
if (!(key in defaults())) return ctx.answerCbQuery('Неизвестная настройка');
|
||
const s = state(ctx.from.id);
|
||
if (key === 'pronounceable' || key === 'excludeAmbiguous') {
|
||
s[key] = !s[key];
|
||
} else {
|
||
const others = ['upper', 'lower', 'digits', 'symbols'].filter((k) => k !== key);
|
||
if (s[key] && !others.some((k) => s[k])) {
|
||
return ctx.answerCbQuery('Нужен хотя бы один набор символов');
|
||
}
|
||
s[key] = !s[key];
|
||
}
|
||
return ctx.editMessageText('⚙️ Настройки генерации:', settingsKb(s)).catch(() => {});
|
||
});
|
||
|
||
bot.action('gen', async (ctx) => {
|
||
const s = state(ctx.from.id);
|
||
const pwd = generatePassword(s);
|
||
addHistory(ctx.from.id, pwd);
|
||
await ctx.answerCbQuery('Готово').catch(() => {});
|
||
return ctx.replyWithMarkdown(passwordText(pwd, s), resultKb()).catch(() => {});
|
||
});
|
||
|
||
bot.action('history', (ctx) => {
|
||
const h = history.get(ctx.from.id) || [];
|
||
const text = h.length
|
||
? '📋 Последние 5 паролей:\n\n' + h.map((p, i) => `${i + 1}. \`${p}\``).join('\n')
|
||
: '📋 История пуста — сначала сгенерируй пароль.';
|
||
const kb = Markup.inlineKeyboard([
|
||
[Markup.button.callback('🧹 Очистить', 'clearhist')],
|
||
[Markup.button.callback('⬅️ Назад', 'main')],
|
||
]);
|
||
return ctx.replyWithMarkdown(text, kb).catch(() => {});
|
||
});
|
||
|
||
bot.action('clearhist', (ctx) => {
|
||
history.delete(ctx.from.id);
|
||
return ctx.editMessageText('🧹 История очищена.', backKb()).catch(() => {});
|
||
});
|
||
|
||
bot.action('about', (ctx) =>
|
||
ctx
|
||
.editMessageText(
|
||
'ℹ️ tg_bot_gen — генератор паролей в Telegram\n\n' +
|
||
'🛡 Node.js crypto (crypto.randomInt), Fisher–Yates перемешивание, гарантия минимум одного символа из каждого набора.\n' +
|
||
'📏 Длина 4–128 символов.\n' +
|
||
'🔤 Наборы: A-Z, a-z, 0-9, символы; опции «без похожих» и «произносимый».\n' +
|
||
'🔒 Пароли не отправляются на серверы и не сохраняются на диске.',
|
||
backKb()
|
||
)
|
||
.catch(() => {})
|
||
);
|
||
|
||
bot.on(['text', 'message'], (ctx) => {
|
||
const t = ctx.message?.text;
|
||
if (t === '/start' || t === '/help' || t === '/menu') return bot.start(ctx);
|
||
return ctx.reply('Используй /start 🙂', mainMenu());
|
||
});
|
||
|
||
bot.catch((err) => console.error('Bot error:', err?.response?.description || err));
|
||
|
||
/* ---------- Start (long polling) ---------- */
|
||
bot
|
||
.launch({ dropPendingUpdates: true })
|
||
.then(() => console.log(`Bot @${bot.botInfo?.username ?? '?'} started (polling)`))
|
||
.catch((e) => {
|
||
console.error('Failed to launch:', e?.response?.description || e.message);
|
||
process.exit(1);
|
||
});
|
||
|
||
const shutdown = () => {
|
||
bot.stop('shutting down');
|
||
process.exit(0);
|
||
};
|
||
process.once('SIGINT', shutdown);
|
||
process.once('SIGTERM', shutdown);
|