ваше сообщение коммита
This commit is contained in:
@@ -145,6 +145,61 @@ class AuthService {
|
||||
return false;
|
||||
}
|
||||
|
||||
/**
|
||||
* Получение балансов токенов для адреса
|
||||
* @param {string} address - Адрес кошелька
|
||||
* @returns {Promise<Object>} - Объект с балансами токенов
|
||||
*/
|
||||
async getTokenBalances(address) {
|
||||
if (!address) {
|
||||
logger.error('No address provided for getTokenBalances');
|
||||
return {
|
||||
eth: '0',
|
||||
bsc: '0',
|
||||
arbitrum: '0',
|
||||
polygon: '0'
|
||||
};
|
||||
}
|
||||
|
||||
const balances = {};
|
||||
|
||||
for (const contract of ADMIN_CONTRACTS) {
|
||||
try {
|
||||
const provider = this.providers[contract.network];
|
||||
if (!provider) {
|
||||
logger.error(`No provider for network ${contract.network}`);
|
||||
balances[contract.network] = '0';
|
||||
continue;
|
||||
}
|
||||
|
||||
const tokenContract = new ethers.Contract(
|
||||
contract.address,
|
||||
ERC20_ABI,
|
||||
provider
|
||||
);
|
||||
|
||||
const balance = await tokenContract.balanceOf(address);
|
||||
const formattedBalance = ethers.formatUnits(balance, 18);
|
||||
|
||||
logger.info(`Token balance for ${address} on ${contract.network}:`, {
|
||||
contract: contract.address,
|
||||
balance: formattedBalance
|
||||
});
|
||||
|
||||
balances[contract.network] = formattedBalance;
|
||||
} catch (error) {
|
||||
logger.error(`Error getting balance for ${contract.network}:`, {
|
||||
address,
|
||||
contract: contract.address,
|
||||
error: error.message
|
||||
});
|
||||
balances[contract.network] = '0';
|
||||
}
|
||||
}
|
||||
|
||||
return balances;
|
||||
}
|
||||
|
||||
// Создание сессии с проверкой роли
|
||||
async createSession(session, { userId, authenticated, authType, guestId, address }) {
|
||||
try {
|
||||
@@ -209,12 +264,18 @@ class AuthService {
|
||||
try {
|
||||
// Проверяем наличие связанного кошелька
|
||||
const wallet = await this.getLinkedWallet(userId);
|
||||
if (wallet) {
|
||||
// Если есть кошелек, проверяем админские токены
|
||||
const isAdmin = await this.checkAdminRole(wallet);
|
||||
return isAdmin ? 'admin' : 'user';
|
||||
|
||||
// Если кошелек не привязан, пользователь получает роль user
|
||||
// с базовым доступом к чату и истории сообщений
|
||||
if (!wallet) {
|
||||
logger.info(`No wallet linked for user ${userId}, assigning basic user role`);
|
||||
return 'user';
|
||||
}
|
||||
return 'user';
|
||||
|
||||
// Если есть кошелек, проверяем админские токены
|
||||
const isAdmin = await this.checkAdminRole(wallet);
|
||||
logger.info(`Role check for user ${userId} with wallet ${wallet}: ${isAdmin ? 'admin' : 'user'}`);
|
||||
return isAdmin ? 'admin' : 'user';
|
||||
} catch (error) {
|
||||
logger.error('Error checking user role:', error);
|
||||
return 'user';
|
||||
@@ -243,17 +304,93 @@ class AuthService {
|
||||
if (userResult.rows.length === 0) {
|
||||
return { verified: false };
|
||||
}
|
||||
|
||||
// Проверяем наличие кошелька и определяем роль
|
||||
const wallet = await this.getLinkedWallet(userId);
|
||||
let role = 'user'; // Базовая роль для доступа к чату
|
||||
|
||||
if (wallet) {
|
||||
// Если есть кошелек, проверяем баланс токенов
|
||||
const isAdmin = await this.checkAdminRole(wallet);
|
||||
role = isAdmin ? 'admin' : 'user';
|
||||
logger.info(`User ${userId} has wallet ${wallet}, role set to ${role}`);
|
||||
} else {
|
||||
logger.info(`User ${userId} has no wallet, using basic user role`);
|
||||
}
|
||||
|
||||
return {
|
||||
verified: true,
|
||||
userId,
|
||||
email
|
||||
email,
|
||||
role,
|
||||
wallet: wallet || null
|
||||
};
|
||||
} catch (error) {
|
||||
logger.error('Error checking email verification:', error);
|
||||
return { verified: false };
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Проверка Telegram аутентификации
|
||||
*/
|
||||
async verifyTelegramAuth(telegramId, verificationCode) {
|
||||
try {
|
||||
logger.info(`Verifying Telegram auth for ID: ${telegramId} with code: ${verificationCode}`);
|
||||
|
||||
// Находим или создаем пользователя
|
||||
const userResult = await db.query(
|
||||
`SELECT u.* FROM users u
|
||||
JOIN user_identities ui ON u.id = ui.user_id
|
||||
WHERE ui.provider = 'telegram' AND ui.provider_id = $1`,
|
||||
[telegramId]
|
||||
);
|
||||
|
||||
let userId;
|
||||
if (userResult.rows.length > 0) {
|
||||
userId = userResult.rows[0].id;
|
||||
logger.info(`Found existing user ${userId} for Telegram ID ${telegramId}`);
|
||||
} else {
|
||||
// Создаем нового пользователя с ролью user
|
||||
const newUserResult = await db.query(
|
||||
'INSERT INTO users (role) VALUES ($1) RETURNING id',
|
||||
['user']
|
||||
);
|
||||
userId = newUserResult.rows[0].id;
|
||||
|
||||
// Добавляем Telegram идентификатор
|
||||
await db.query(
|
||||
'INSERT INTO user_identities (user_id, provider, provider_id) VALUES ($1, $2, $3)',
|
||||
[userId, 'telegram', telegramId]
|
||||
);
|
||||
logger.info(`Created new user ${userId} for Telegram ID ${telegramId}`);
|
||||
}
|
||||
|
||||
// Проверяем наличие кошелька и определяем роль
|
||||
const wallet = await this.getLinkedWallet(userId);
|
||||
let role = 'user'; // Базовая роль для доступа к чату
|
||||
|
||||
if (wallet) {
|
||||
// Если есть кошелек, проверяем баланс токенов
|
||||
const isAdmin = await this.checkAdminRole(wallet);
|
||||
role = isAdmin ? 'admin' : 'user';
|
||||
logger.info(`User ${userId} has wallet ${wallet}, role set to ${role}`);
|
||||
} else {
|
||||
logger.info(`User ${userId} has no wallet, using basic user role`);
|
||||
}
|
||||
|
||||
return {
|
||||
success: true,
|
||||
userId,
|
||||
role,
|
||||
telegramId,
|
||||
wallet: wallet || null
|
||||
};
|
||||
} catch (error) {
|
||||
logger.error('Error in Telegram auth verification:', error);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Создаем и экспортируем единственный экземпляр
|
||||
|
||||
@@ -1,209 +1,106 @@
|
||||
const { pool } = require('../db');
|
||||
const verificationService = require('./verification-service');
|
||||
const logger = require('../utils/logger');
|
||||
const emailBot = require('./emailBot');
|
||||
const db = require('../db');
|
||||
const authService = require('./auth-service');
|
||||
const verificationService = require('./verification-service');
|
||||
const { EmailBotService } = require('./emailBot');
|
||||
|
||||
// Инициализация процесса аутентификации по email
|
||||
async function initEmailAuth(session, email) {
|
||||
try {
|
||||
if (!email || !email.match(/^[^\s@]+@[^\s@]+\.[^\s@]+$/)) {
|
||||
throw new Error('Некорректный формат email');
|
||||
}
|
||||
|
||||
// Сохраняем email в сессии для последующей верификации
|
||||
session.pendingEmail = email.toLowerCase();
|
||||
|
||||
// Создаем или получаем ID пользователя
|
||||
let userId;
|
||||
|
||||
// Проверяем, существует ли пользователь с этим email
|
||||
const existingEmailUser = await db.query(
|
||||
`SELECT ui.user_id
|
||||
FROM user_identities ui
|
||||
WHERE ui.provider = 'email' AND ui.provider_id = $1`,
|
||||
[email.toLowerCase()]
|
||||
);
|
||||
|
||||
if (existingEmailUser.rows.length > 0) {
|
||||
// Используем существующего пользователя
|
||||
userId = existingEmailUser.rows[0].user_id;
|
||||
logger.info(`Using existing user ${userId} for email ${email}`);
|
||||
|
||||
// Связываем гостевой ID с существующим пользователем, если еще нет
|
||||
if (session.guestId) {
|
||||
const guestIdentity = await db.query(
|
||||
`SELECT * FROM user_identities
|
||||
WHERE user_id = $1 AND provider = 'guest' AND provider_id = $2`,
|
||||
[userId, session.guestId]
|
||||
);
|
||||
|
||||
if (guestIdentity.rows.length === 0) {
|
||||
await db.query(
|
||||
`INSERT INTO user_identities
|
||||
(user_id, provider, provider_id, created_at)
|
||||
VALUES ($1, $2, $3, NOW())
|
||||
ON CONFLICT (provider, provider_id) DO UPDATE SET user_id = $1`,
|
||||
[userId, 'guest', session.guestId]
|
||||
);
|
||||
}
|
||||
class EmailAuth {
|
||||
constructor() {
|
||||
this.emailBot = emailBot;
|
||||
}
|
||||
|
||||
async initEmailAuth(session, email) {
|
||||
try {
|
||||
if (!email || !email.match(/^[^\s@]+@[^\s@]+\.[^\s@]+$/)) {
|
||||
throw new Error('Некорректный формат email');
|
||||
}
|
||||
} else if (session.authenticated && session.userId) {
|
||||
// Если пользователь уже аутентифицирован, используем его ID
|
||||
userId = session.userId;
|
||||
} else if (session.guestId) {
|
||||
// Проверяем, есть ли пользователь с текущим guestId
|
||||
const guestUserResult = await db.query(
|
||||
`SELECT u.id
|
||||
FROM users u
|
||||
JOIN user_identities ui ON u.id = ui.user_id
|
||||
WHERE ui.provider = 'guest' AND ui.provider_id = $1`,
|
||||
[session.guestId]
|
||||
);
|
||||
|
||||
if (guestUserResult.rows.length > 0) {
|
||||
// Используем существующего пользователя с guestId
|
||||
userId = guestUserResult.rows[0].id;
|
||||
// Создаем или получаем ID пользователя
|
||||
let userId;
|
||||
|
||||
if (session.authenticated && session.userId) {
|
||||
userId = session.userId;
|
||||
} else {
|
||||
// Создаем нового пользователя
|
||||
const userResult = await db.query(
|
||||
'INSERT INTO users (created_at) VALUES (NOW()) RETURNING id'
|
||||
'INSERT INTO users (role) VALUES ($1) RETURNING id',
|
||||
['user']
|
||||
);
|
||||
userId = userResult.rows[0].id;
|
||||
|
||||
// Связываем гостевой ID с новым пользователем
|
||||
if (session.guestId) {
|
||||
await db.query(
|
||||
`INSERT INTO user_identities
|
||||
(user_id, provider, provider_id, created_at)
|
||||
VALUES ($1, $2, $3, NOW())`,
|
||||
[userId, 'guest', session.guestId]
|
||||
);
|
||||
}
|
||||
session.tempUserId = userId;
|
||||
}
|
||||
} else {
|
||||
// Создаем нового пользователя без гостевого ID
|
||||
const userResult = await db.query(
|
||||
'INSERT INTO users (created_at) VALUES (NOW()) RETURNING id'
|
||||
|
||||
// Сохраняем email в сессии
|
||||
session.pendingEmail = email.toLowerCase();
|
||||
|
||||
// Создаем код через сервис верификации
|
||||
const verificationCode = await verificationService.createVerificationCode(
|
||||
'email',
|
||||
email.toLowerCase(),
|
||||
userId
|
||||
);
|
||||
userId = userResult.rows[0].id;
|
||||
|
||||
// Отправляем код на email
|
||||
await this.emailBot.sendVerificationCode(email, verificationCode);
|
||||
|
||||
logger.info(`Generated verification code for Email auth for ${email} and sent to user's email`);
|
||||
|
||||
return { success: true, verificationCode };
|
||||
} catch (error) {
|
||||
logger.error('Error in email auth initialization:', error);
|
||||
throw error;
|
||||
}
|
||||
|
||||
session.tempUserId = userId;
|
||||
|
||||
// Создаем код через сервис верификации
|
||||
const code = await verificationService.createVerificationCode(
|
||||
'email',
|
||||
email.toLowerCase(),
|
||||
userId
|
||||
);
|
||||
|
||||
// Создаем экземпляр EmailBotService для отправки кода
|
||||
const emailService = new EmailBotService(
|
||||
process.env.EMAIL_USER,
|
||||
process.env.EMAIL_PASSWORD
|
||||
);
|
||||
|
||||
// Отправляем код на email пользователя
|
||||
await emailService.sendVerificationCode(email.toLowerCase(), userId);
|
||||
|
||||
logger.info(`Generated verification code for Email auth for ${email} and sent to user's email`);
|
||||
return { success: true };
|
||||
} catch (error) {
|
||||
logger.error('Error initializing email auth:', error);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
// Проверка кода верификации
|
||||
async function checkEmailVerification(code, session) {
|
||||
try {
|
||||
if (!session?.pendingEmail) {
|
||||
return { verified: false, message: "Email не найден в сессии" };
|
||||
}
|
||||
async checkEmailVerification(code, session) {
|
||||
try {
|
||||
if (!code) {
|
||||
return { verified: false, message: 'Код верификации не предоставлен' };
|
||||
}
|
||||
|
||||
// Проверяем код через сервис верификации
|
||||
const result = await verificationService.verifyCode(code, 'email', session.pendingEmail);
|
||||
|
||||
if (!result.success) {
|
||||
return { verified: false, message: result.error || "Неверный код верификации" };
|
||||
}
|
||||
|
||||
const userId = result.userId;
|
||||
|
||||
// Проверяем, существует ли пользователь
|
||||
const userResult = await db.query(
|
||||
'SELECT * FROM users WHERE id = $1',
|
||||
[userId]
|
||||
);
|
||||
|
||||
if (userResult.rows.length === 0) {
|
||||
return { verified: false, message: "Пользователь не найден" };
|
||||
}
|
||||
|
||||
// Проверяем, есть ли у пользователя связанный email
|
||||
const emailIdentity = await db.query(
|
||||
`SELECT * FROM user_identities
|
||||
WHERE user_id = $1 AND provider = 'email' AND provider_id = $2`,
|
||||
[userId, session.pendingEmail]
|
||||
);
|
||||
|
||||
if (emailIdentity.rows.length === 0) {
|
||||
// Связываем Email с пользователем
|
||||
if (!session.pendingEmail) {
|
||||
return { verified: false, message: 'Email не найден в сессии' };
|
||||
}
|
||||
|
||||
// Проверяем код через сервис верификации
|
||||
const result = await verificationService.verifyCode(code, 'email', session.pendingEmail);
|
||||
|
||||
if (!result.success) {
|
||||
// Используем сообщение об ошибке из сервиса верификации
|
||||
return { verified: false, message: result.error || 'Неверный код верификации' };
|
||||
}
|
||||
|
||||
const userId = result.userId || session.tempUserId;
|
||||
const email = session.pendingEmail;
|
||||
|
||||
// Добавляем email в базу данных
|
||||
await db.query(
|
||||
`INSERT INTO user_identities
|
||||
(user_id, provider, provider_id, created_at)
|
||||
VALUES ($1, $2, $3, NOW())
|
||||
ON CONFLICT (provider, provider_id) DO UPDATE SET user_id = $1`,
|
||||
[userId, 'email', session.pendingEmail]
|
||||
(user_id, provider, provider_id)
|
||||
VALUES ($1, $2, $3)
|
||||
ON CONFLICT (provider, provider_id)
|
||||
DO UPDATE SET user_id = $1`,
|
||||
[userId, 'email', email.toLowerCase()]
|
||||
);
|
||||
}
|
||||
|
||||
// Связываем гостевой ID с пользователем, если его еще нет
|
||||
if (session.guestId) {
|
||||
const guestIdentity = await db.query(
|
||||
`SELECT * FROM user_identities
|
||||
WHERE user_id = $1 AND provider = 'guest' AND provider_id = $2`,
|
||||
[userId, session.guestId]
|
||||
);
|
||||
|
||||
if (guestIdentity.rows.length === 0) {
|
||||
await db.query(
|
||||
`INSERT INTO user_identities
|
||||
(user_id, provider, provider_id, created_at)
|
||||
VALUES ($1, $2, $3, NOW())
|
||||
ON CONFLICT (provider, provider_id) DO UPDATE SET user_id = $1`,
|
||||
[userId, 'guest', session.guestId]
|
||||
);
|
||||
}
|
||||
|
||||
// Связываем гостевые сообщения с пользователем
|
||||
try {
|
||||
const messagesExist = await db.query(
|
||||
'SELECT EXISTS(SELECT 1 FROM guest_messages WHERE guest_id = $1) as exists',
|
||||
[session.guestId]
|
||||
);
|
||||
|
||||
if (messagesExist.rows[0].exists) {
|
||||
await db.query('SELECT link_guest_messages($1, $2)', [userId, session.guestId]);
|
||||
}
|
||||
} catch (linkError) {
|
||||
logger.error(`Error linking messages: ${linkError}`);
|
||||
// Очищаем временные данные
|
||||
delete session.pendingEmail;
|
||||
if (session.tempUserId) {
|
||||
delete session.tempUserId;
|
||||
}
|
||||
|
||||
return {
|
||||
verified: true,
|
||||
userId,
|
||||
email: email.toLowerCase()
|
||||
};
|
||||
} catch (error) {
|
||||
logger.error('Error checking email verification:', error);
|
||||
return { verified: false, message: 'Ошибка при проверке кода верификации' };
|
||||
}
|
||||
|
||||
return {
|
||||
verified: true,
|
||||
userId,
|
||||
email: session.pendingEmail
|
||||
};
|
||||
} catch (error) {
|
||||
logger.error('Error in Email verification:', error);
|
||||
return { verified: false, message: "Ошибка при проверке кода" };
|
||||
}
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
initEmailAuth,
|
||||
checkEmailVerification
|
||||
};
|
||||
// Создаем и экспортируем единственный экземпляр
|
||||
const emailAuth = new EmailAuth();
|
||||
module.exports = emailAuth;
|
||||
@@ -5,17 +5,19 @@ const simpleParser = require('mailparser').simpleParser;
|
||||
const { processMessage } = require('./ai-assistant');
|
||||
const { inspect } = require('util');
|
||||
const logger = require('../utils/logger');
|
||||
const verificationService = require('./verification-service');
|
||||
|
||||
// Конфигурация для отправки писем
|
||||
const transporter = nodemailer.createTransport({
|
||||
host: process.env.EMAIL_SMTP_HOST,
|
||||
port: process.env.EMAIL_SMTP_PORT,
|
||||
secure: process.env.EMAIL_SMTP_PORT === '465',
|
||||
host: process.env.EMAIL_SMTP_HOST || 'smtp.hostland.ru',
|
||||
port: process.env.EMAIL_SMTP_PORT || 465,
|
||||
secure: true,
|
||||
auth: {
|
||||
user: process.env.EMAIL_USER,
|
||||
pass: process.env.EMAIL_PASSWORD,
|
||||
},
|
||||
pool: true,
|
||||
maxConnections: 3,
|
||||
maxMessages: 5,
|
||||
tls: {
|
||||
rejectUnauthorized: false
|
||||
}
|
||||
@@ -30,26 +32,22 @@ const imapConfig = {
|
||||
tls: true,
|
||||
tlsOptions: { rejectUnauthorized: false },
|
||||
keepalive: {
|
||||
interval: 10000, // Проверка соединения каждые 10 секунд
|
||||
idleInterval: 300000, // Сброс соединения через 5 минут простоя
|
||||
forceNoop: true // Принудительная отправка NOOP для поддержания соединения
|
||||
interval: 10000,
|
||||
idleInterval: 300000,
|
||||
forceNoop: true
|
||||
}
|
||||
};
|
||||
|
||||
class EmailBotService {
|
||||
constructor(user, password) {
|
||||
this.user = user;
|
||||
this.password = password;
|
||||
constructor() {
|
||||
this.transporter = transporter;
|
||||
this.imap = new Imap(imapConfig);
|
||||
this.initialize();
|
||||
this.listenForReplies();
|
||||
}
|
||||
|
||||
initialize() {
|
||||
this.imap.once('error', (err) => {
|
||||
logger.error(`IMAP connection error: ${err.message}`);
|
||||
// Пробуем переподключиться через 1 минуту при ошибке
|
||||
setTimeout(() => {
|
||||
try {
|
||||
if (this.imap.state !== 'connected') {
|
||||
@@ -63,59 +61,52 @@ class EmailBotService {
|
||||
});
|
||||
}
|
||||
|
||||
async sendVerificationCode(toEmail, userId) {
|
||||
// Метод для инициализации email верификации
|
||||
async initEmailVerification(email, userId, code) {
|
||||
try {
|
||||
// Создаем код через сервис верификации
|
||||
const code = await verificationService.createVerificationCode(
|
||||
'email',
|
||||
toEmail.toLowerCase(),
|
||||
userId
|
||||
);
|
||||
// Отправляем код на email
|
||||
await this.sendVerificationCode(email, code);
|
||||
|
||||
// Отправляем письмо с кодом
|
||||
return { success: true };
|
||||
} catch (error) {
|
||||
logger.error('Error initializing email verification:', error);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
// Отправка кода верификации
|
||||
async sendVerificationCode(email, code) {
|
||||
try {
|
||||
const mailOptions = {
|
||||
from: this.user,
|
||||
to: toEmail,
|
||||
subject: 'Код подтверждения для DApp for Business',
|
||||
text: `Ваш код подтверждения: ${code}\n\nДля завершения аутентификации, пожалуйста, введите этот код на сайте.\n\nКод действителен в течение 15 минут.`,
|
||||
from: process.env.EMAIL_USER,
|
||||
to: email,
|
||||
subject: 'Код подтверждения',
|
||||
text: `Ваш код подтверждения: ${code}\n\nКод действителен в течение 15 минут.`,
|
||||
html: `
|
||||
<div style="font-family: Arial, sans-serif; max-width: 600px; margin: 0 auto; padding: 20px; border: 1px solid #ddd; border-radius: 5px;">
|
||||
<h2 style="color: #333;">Код подтверждения для DApp for Business</h2>
|
||||
<p>Ваш код подтверждения:</p>
|
||||
<div style="font-size: 24px; font-weight: bold; padding: 15px; background-color: #f5f5f5; border-radius: 5px; text-align: center; margin: 20px 0;">
|
||||
${code}
|
||||
<div style="font-family: Arial, sans-serif; max-width: 600px; margin: 0 auto;">
|
||||
<h2 style="color: #333;">Код подтверждения</h2>
|
||||
<p style="font-size: 16px; color: #666;">Ваш код подтверждения:</p>
|
||||
<div style="background-color: #f5f5f5; padding: 15px; border-radius: 5px; text-align: center; margin: 20px 0;">
|
||||
<span style="font-size: 24px; font-weight: bold; color: #333;">${code}</span>
|
||||
</div>
|
||||
<p>Для завершения аутентификации, пожалуйста, введите этот код в форме на сайте.</p>
|
||||
<p>Код действителен в течение 15 минут.</p>
|
||||
<hr style="margin: 20px 0; border: none; border-top: 1px solid #ddd;">
|
||||
<p style="font-size: 12px; color: #777;">Это автоматическое сообщение, пожалуйста, не отвечайте на него.</p>
|
||||
<p style="font-size: 14px; color: #999;">Код действителен в течение 15 минут.</p>
|
||||
</div>
|
||||
`
|
||||
};
|
||||
|
||||
const info = await this.transporter.sendMail(mailOptions);
|
||||
logger.info(`Email sent: ${info.messageId}`);
|
||||
|
||||
return { success: true, code };
|
||||
await this.transporter.sendMail(mailOptions);
|
||||
logger.info(`Verification code sent to ${email}`);
|
||||
} catch (error) {
|
||||
logger.error(`Error sending email: ${error}`);
|
||||
return { success: false, error: error.message };
|
||||
logger.error('Error sending verification code:', error);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
listenForReplies() {
|
||||
// Запускаем проверку почты каждые 60 секунд
|
||||
setInterval(() => {
|
||||
this.checkEmails();
|
||||
}, 60000);
|
||||
}
|
||||
|
||||
checkEmails() {
|
||||
try {
|
||||
// Добавляем обработчики ошибок
|
||||
this.imap.once('error', (err) => {
|
||||
logger.error(`IMAP connection error during check: ${err.message}`);
|
||||
// Пытаемся закрыть соединение при ошибке
|
||||
try {
|
||||
this.imap.end();
|
||||
} catch (e) {
|
||||
@@ -145,7 +136,6 @@ class EmailBotService {
|
||||
return;
|
||||
}
|
||||
|
||||
// Защищаемся от пустых результатов
|
||||
try {
|
||||
const f = this.imap.fetch(results, { bodies: '' });
|
||||
|
||||
@@ -156,23 +146,6 @@ class EmailBotService {
|
||||
logger.error(`Error parsing message: ${err}`);
|
||||
return;
|
||||
}
|
||||
|
||||
// Обработка входящего письма для Ollama
|
||||
try {
|
||||
// Проверяем, что это действительно письмо (защита от ошибок)
|
||||
if (parsed && parsed.text && parsed.from && parsed.from.value &&
|
||||
parsed.from.value.length > 0 && parsed.from.value[0].address) {
|
||||
|
||||
const fromEmail = parsed.from.value[0].address.toLowerCase();
|
||||
const subject = parsed.subject || '';
|
||||
const text = parsed.text || '';
|
||||
|
||||
// Передаем письмо в Ollama для обработки
|
||||
await this.processOllamaEmail(fromEmail, subject, text);
|
||||
}
|
||||
} catch (e) {
|
||||
logger.error(`Error processing email for Ollama: ${e.message}`);
|
||||
}
|
||||
});
|
||||
});
|
||||
});
|
||||
@@ -203,7 +176,6 @@ class EmailBotService {
|
||||
this.imap.connect();
|
||||
} catch (error) {
|
||||
logger.error(`Global error checking emails: ${error.message}`);
|
||||
// Обеспечиваем корректное завершение IMAP сессии
|
||||
try {
|
||||
this.imap.end();
|
||||
} catch (e) {
|
||||
@@ -212,74 +184,25 @@ class EmailBotService {
|
||||
}
|
||||
}
|
||||
|
||||
// Метод для обработки письма с помощью Ollama
|
||||
async processOllamaEmail(fromEmail, subject, text) {
|
||||
try {
|
||||
// Проверяем, есть ли текст для обработки
|
||||
if (!text || text.trim() === '') {
|
||||
logger.info(`Empty message from ${fromEmail}, skipping Ollama processing`);
|
||||
return;
|
||||
}
|
||||
|
||||
logger.info(`Processing message from ${fromEmail} for Ollama`);
|
||||
|
||||
// Получаем ответ от Ollama
|
||||
const response = await processMessage(text);
|
||||
|
||||
if (response) {
|
||||
// Отправляем ответ обратно пользователю
|
||||
await this.transporter.sendMail({
|
||||
from: this.user,
|
||||
to: fromEmail,
|
||||
subject: `Re: ${subject}`,
|
||||
text: response
|
||||
});
|
||||
|
||||
logger.info(`Ollama response sent to ${fromEmail}`);
|
||||
}
|
||||
} catch (error) {
|
||||
logger.error(`Error in Ollama email processing: ${error}`);
|
||||
|
||||
// Отправляем сообщение об ошибке пользователю
|
||||
try {
|
||||
await this.transporter.sendMail({
|
||||
from: this.user,
|
||||
to: fromEmail,
|
||||
subject: 'Error processing your request',
|
||||
text: 'Sorry, we encountered an error processing your message. Please try again later.'
|
||||
});
|
||||
} catch (e) {
|
||||
logger.error(`Error sending error notification: ${e}`);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Метод для проверки кода без IMAP
|
||||
async verifyCode(email, code) {
|
||||
return await verificationService.verifyCode(code, 'email', email.toLowerCase());
|
||||
}
|
||||
|
||||
// Оставляем существующий метод для отправки электронных писем
|
||||
// Метод для отправки email
|
||||
async sendEmail(to, subject, text) {
|
||||
try {
|
||||
const mailOptions = {
|
||||
from: this.user,
|
||||
from: process.env.EMAIL_USER,
|
||||
to,
|
||||
subject,
|
||||
text
|
||||
};
|
||||
|
||||
const info = await this.transporter.sendMail(mailOptions);
|
||||
logger.info(`Email sent: ${info.messageId}`);
|
||||
await this.transporter.sendMail(mailOptions);
|
||||
logger.info(`Email sent to ${to}`);
|
||||
return true;
|
||||
} catch (error) {
|
||||
logger.error(`Error sending email: ${error}`);
|
||||
return false;
|
||||
logger.error('Error sending email:', error);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
EmailBotService,
|
||||
transporter
|
||||
};
|
||||
// Экспортируем singleton instance
|
||||
module.exports = new EmailBotService();
|
||||
|
||||
@@ -1,5 +1,7 @@
|
||||
const { initTelegramBot } = require('./telegram-service');
|
||||
const { initEmailBot, sendEmail, checkEmails } = require('./emailBot');
|
||||
const emailBot = require('./emailBot');
|
||||
const telegramBot = require('./telegramBot');
|
||||
const aiAssistant = require('./ai-assistant');
|
||||
const {
|
||||
initializeVectorStore,
|
||||
getVectorStore,
|
||||
@@ -14,9 +16,9 @@ module.exports = {
|
||||
initTelegramBot,
|
||||
|
||||
// Email
|
||||
initEmailBot,
|
||||
sendEmail,
|
||||
checkEmails,
|
||||
emailBot,
|
||||
sendEmail: emailBot.sendEmail,
|
||||
checkEmails: emailBot.checkEmails,
|
||||
|
||||
// Vector Store
|
||||
initializeVectorStore,
|
||||
@@ -28,5 +30,7 @@ module.exports = {
|
||||
processMessage,
|
||||
getUserInfo,
|
||||
getConversationHistory,
|
||||
// ... другие экспорты
|
||||
|
||||
telegramBot,
|
||||
aiAssistant
|
||||
};
|
||||
|
||||
@@ -9,7 +9,9 @@ class VerificationService {
|
||||
|
||||
// Генерация кода
|
||||
generateCode() {
|
||||
return Math.random().toString(36).substring(2, 2 + this.codeLength).toUpperCase();
|
||||
const code = Math.random().toString(36).substring(2, 2 + this.codeLength).toUpperCase();
|
||||
logger.info(`Generated verification code: ${code}`);
|
||||
return code;
|
||||
}
|
||||
|
||||
// Создание кода верификации
|
||||
@@ -18,6 +20,8 @@ class VerificationService {
|
||||
const expiresAt = new Date(Date.now() + this.expirationMinutes * 60 * 1000);
|
||||
|
||||
try {
|
||||
logger.info(`Creating verification code for ${provider}:${providerId}, userId: ${userId}`);
|
||||
|
||||
await db.query(
|
||||
`INSERT INTO verification_codes
|
||||
(code, provider, provider_id, user_id, expires_at)
|
||||
@@ -25,9 +29,15 @@ class VerificationService {
|
||||
[code, provider, providerId, userId, expiresAt]
|
||||
);
|
||||
|
||||
logger.info(`Verification code created successfully for ${provider}:${providerId}`);
|
||||
return code;
|
||||
} catch (error) {
|
||||
logger.error('Error creating verification code:', error);
|
||||
logger.error('Error creating verification code:', {
|
||||
error: error.message,
|
||||
provider,
|
||||
providerId,
|
||||
userId
|
||||
});
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
@@ -35,6 +45,28 @@ class VerificationService {
|
||||
// Проверка кода
|
||||
async verifyCode(code, provider, providerId) {
|
||||
try {
|
||||
logger.info(`Verifying code for ${provider}:${providerId}`);
|
||||
|
||||
// Преобразуем код в верхний регистр для сравнения
|
||||
const normalizedCode = code.toUpperCase();
|
||||
logger.info(`Normalized code: ${normalizedCode}`);
|
||||
|
||||
// Проверим, есть ли такой код в базе (для отладки)
|
||||
const checkResult = await db.query(
|
||||
`SELECT code FROM verification_codes
|
||||
WHERE provider = $1
|
||||
AND provider_id = $2
|
||||
AND used = false
|
||||
AND expires_at > NOW()`,
|
||||
[provider, providerId]
|
||||
);
|
||||
|
||||
if (checkResult.rows.length > 0) {
|
||||
logger.info(`Found codes for ${provider}:${providerId}: ${JSON.stringify(checkResult.rows.map(r => r.code))}`);
|
||||
} else {
|
||||
logger.warn(`No active codes found for ${provider}:${providerId}`);
|
||||
}
|
||||
|
||||
const result = await db.query(
|
||||
`SELECT * FROM verification_codes
|
||||
WHERE code = $1
|
||||
@@ -42,10 +74,11 @@ class VerificationService {
|
||||
AND provider_id = $3
|
||||
AND used = false
|
||||
AND expires_at > NOW()`,
|
||||
[code, provider, providerId]
|
||||
[normalizedCode, provider, providerId]
|
||||
);
|
||||
|
||||
if (result.rows.length === 0) {
|
||||
logger.warn(`Invalid or expired code for ${provider}:${providerId}. Input: ${normalizedCode}`);
|
||||
return { success: false, error: 'Неверный или истекший код' };
|
||||
}
|
||||
|
||||
@@ -57,13 +90,19 @@ class VerificationService {
|
||||
[verification.id]
|
||||
);
|
||||
|
||||
logger.info(`Code verified successfully for ${provider}:${providerId}`);
|
||||
return {
|
||||
success: true,
|
||||
userId: verification.user_id,
|
||||
providerId: verification.provider_id
|
||||
};
|
||||
} catch (error) {
|
||||
logger.error('Error verifying code:', error);
|
||||
logger.error('Error verifying code:', {
|
||||
error: error.message,
|
||||
code,
|
||||
provider,
|
||||
providerId
|
||||
});
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
@@ -71,13 +110,15 @@ class VerificationService {
|
||||
// Очистка истекших кодов
|
||||
async cleanupExpiredCodes() {
|
||||
try {
|
||||
await db.query(
|
||||
'DELETE FROM verification_codes WHERE expires_at <= NOW()'
|
||||
const result = await db.query(
|
||||
'DELETE FROM verification_codes WHERE expires_at <= NOW() RETURNING id'
|
||||
);
|
||||
logger.info(`Cleaned up ${result.rowCount} expired verification codes`);
|
||||
} catch (error) {
|
||||
logger.error('Error cleaning up expired codes:', error);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
module.exports = new VerificationService();
|
||||
const verificationService = new VerificationService();
|
||||
module.exports = verificationService;
|
||||
Reference in New Issue
Block a user