260 lines
10 KiB
JavaScript
260 lines
10 KiB
JavaScript
const { Pool } = require('pg');
|
||
const dotenv = require('dotenv');
|
||
const uuid = require('uuid');
|
||
|
||
dotenv.config();
|
||
|
||
const pool = new Pool({
|
||
connectionString: process.env.DATABASE_URL,
|
||
});
|
||
|
||
async function createNotificationTables() {
|
||
const client = await pool.connect();
|
||
|
||
try {
|
||
await client.query('BEGIN');
|
||
|
||
console.log('Creating UUID extension if not exists...');
|
||
await client.query(`
|
||
CREATE EXTENSION IF NOT EXISTS "uuid-ossp";
|
||
`);
|
||
|
||
// Проверяем существование таблицы notifications
|
||
const notificationsExists = await client.query(`
|
||
SELECT EXISTS (
|
||
SELECT FROM information_schema.tables
|
||
WHERE table_name = 'notifications'
|
||
) as exists
|
||
`);
|
||
|
||
if (!notificationsExists.rows[0].exists) {
|
||
console.log('Creating notifications table...');
|
||
await client.query(`
|
||
CREATE TABLE notifications (
|
||
id UUID PRIMARY KEY DEFAULT uuid_generate_v4(),
|
||
user_id UUID NOT NULL REFERENCES users(id) ON DELETE CASCADE,
|
||
type VARCHAR(50) NOT NULL,
|
||
data JSONB,
|
||
is_read BOOLEAN DEFAULT FALSE,
|
||
created_at TIMESTAMP DEFAULT NOW()
|
||
)
|
||
`);
|
||
|
||
console.log('Creating index on notifications...');
|
||
await client.query(`
|
||
CREATE INDEX idx_notifications_user_id ON notifications (user_id);
|
||
CREATE INDEX idx_notifications_type ON notifications (type);
|
||
CREATE INDEX idx_notifications_created_at ON notifications (created_at);
|
||
`);
|
||
} else {
|
||
console.log('Notifications table already exists.');
|
||
}
|
||
|
||
// Проверяем существование таблицы scheduled_notifications
|
||
const scheduledExists = await client.query(`
|
||
SELECT EXISTS (
|
||
SELECT FROM information_schema.tables
|
||
WHERE table_name = 'scheduled_notifications'
|
||
) as exists
|
||
`);
|
||
|
||
if (!scheduledExists.rows[0].exists) {
|
||
console.log('Creating scheduled_notifications table...');
|
||
await client.query(`
|
||
CREATE TABLE scheduled_notifications (
|
||
id UUID PRIMARY KEY DEFAULT uuid_generate_v4(),
|
||
user_id UUID NOT NULL REFERENCES users(id) ON DELETE CASCADE,
|
||
type VARCHAR(50) NOT NULL,
|
||
data JSONB,
|
||
scheduled_at TIMESTAMP NOT NULL,
|
||
processed BOOLEAN DEFAULT FALSE,
|
||
created_at TIMESTAMP DEFAULT NOW()
|
||
)
|
||
`);
|
||
|
||
console.log('Creating index on scheduled_notifications...');
|
||
await client.query(`
|
||
CREATE INDEX idx_scheduled_notifications_user_id ON scheduled_notifications (user_id);
|
||
CREATE INDEX idx_scheduled_notifications_scheduled_at ON scheduled_notifications (scheduled_at);
|
||
CREATE INDEX idx_scheduled_notifications_processed ON scheduled_notifications (processed);
|
||
`);
|
||
} else {
|
||
console.log('Scheduled_notifications table already exists.');
|
||
}
|
||
|
||
// Проверяем существование таблицы notification_templates
|
||
const templatesExists = await client.query(`
|
||
SELECT EXISTS (
|
||
SELECT FROM information_schema.tables
|
||
WHERE table_name = 'notification_templates'
|
||
) as exists
|
||
`);
|
||
|
||
if (!templatesExists.rows[0].exists) {
|
||
console.log('Creating notification_templates table...');
|
||
await client.query(`
|
||
CREATE TABLE notification_templates (
|
||
id UUID PRIMARY KEY DEFAULT uuid_generate_v4(),
|
||
type VARCHAR(50) NOT NULL UNIQUE,
|
||
title TEXT NOT NULL,
|
||
message_template TEXT NOT NULL,
|
||
button_template JSONB,
|
||
created_at TIMESTAMP DEFAULT NOW(),
|
||
updated_at TIMESTAMP DEFAULT NOW()
|
||
)
|
||
`);
|
||
} else {
|
||
console.log('Notification_templates table already exists.');
|
||
}
|
||
|
||
// Проверяем наличие колонки notification_settings в таблице users
|
||
const settingsColumnExists = await client.query(`
|
||
SELECT EXISTS (
|
||
SELECT FROM information_schema.columns
|
||
WHERE table_name = 'users' AND column_name = 'notification_settings'
|
||
) as exists
|
||
`);
|
||
|
||
if (!settingsColumnExists.rows[0].exists) {
|
||
console.log('Adding notification_settings column to users table...');
|
||
await client.query(`
|
||
ALTER TABLE users
|
||
ADD COLUMN notification_settings JSONB DEFAULT '{
|
||
"newMatches": true,
|
||
"newMessages": true,
|
||
"newLikes": true,
|
||
"reminders": true,
|
||
"dailySummary": true,
|
||
"timePreference": "evening",
|
||
"doNotDisturb": false
|
||
}'::jsonb
|
||
`);
|
||
} else {
|
||
console.log('Notification_settings column already exists in users table.');
|
||
}
|
||
|
||
// Заполнение таблицы шаблонов уведомлений базовыми шаблонами
|
||
if (!templatesExists.rows[0].exists) {
|
||
console.log('Populating notification templates...');
|
||
|
||
const templates = [
|
||
{
|
||
type: 'new_like',
|
||
title: 'Новый лайк!',
|
||
message_template: '❤️ *{{name}}* поставил(а) вам лайк!\n\nВозраст: {{age}}\n{{city}}\n\nОтветьте взаимностью или посмотрите профиль.',
|
||
button_template: {
|
||
inline_keyboard: [
|
||
[{ text: '👀 Посмотреть профиль', callback_data: 'view_profile:{{userId}}' }],
|
||
[
|
||
{ text: '❤️ Лайк в ответ', callback_data: 'like_back:{{userId}}' },
|
||
{ text: '⛔️ Пропустить', callback_data: 'dislike_profile:{{userId}}' }
|
||
],
|
||
[{ text: '💕 Открыть все лайки', callback_data: 'view_likes' }]
|
||
]
|
||
}
|
||
},
|
||
{
|
||
type: 'super_like',
|
||
title: 'Супер-лайк!',
|
||
message_template: '⭐️ *{{name}}* отправил(а) вам супер-лайк!\n\nВозраст: {{age}}\n{{city}}\n\nВы произвели особое впечатление! Ответьте взаимностью или посмотрите профиль.',
|
||
button_template: {
|
||
inline_keyboard: [
|
||
[{ text: '👀 Посмотреть профиль', callback_data: 'view_profile:{{userId}}' }],
|
||
[
|
||
{ text: '❤️ Лайк в ответ', callback_data: 'like_back:{{userId}}' },
|
||
{ text: '⛔️ Пропустить', callback_data: 'dislike_profile:{{userId}}' }
|
||
],
|
||
[{ text: '💕 Открыть все лайки', callback_data: 'view_likes' }]
|
||
]
|
||
}
|
||
},
|
||
{
|
||
type: 'new_match',
|
||
title: 'Новый матч!',
|
||
message_template: '🎊 *Ура! Это взаимно!* 🎊\n\nВы и *{{name}}* понравились друг другу!\nВозраст: {{age}}\n{{city}}\n\nСделайте первый шаг - напишите сообщение!',
|
||
button_template: {
|
||
inline_keyboard: [
|
||
[{ text: '💬 Начать общение', callback_data: 'open_chat:{{matchId}}' }],
|
||
[
|
||
{ text: '👀 Посмотреть профиль', callback_data: 'view_profile:{{userId}}' },
|
||
{ text: '📋 Все матчи', callback_data: 'view_matches' }
|
||
]
|
||
]
|
||
}
|
||
},
|
||
{
|
||
type: 'new_message',
|
||
title: 'Новое сообщение!',
|
||
message_template: '💌 *Новое сообщение!*\n\nОт: *{{name}}*\n\n"{{message}}"\n\nОтветьте на сообщение прямо сейчас!',
|
||
button_template: {
|
||
inline_keyboard: [
|
||
[{ text: '📩 Ответить', callback_data: 'open_chat:{{matchId}}' }],
|
||
[
|
||
{ text: '👤 Профиль', callback_data: 'view_profile:{{userId}}' },
|
||
{ text: '📋 Все чаты', callback_data: 'view_matches' }
|
||
]
|
||
]
|
||
}
|
||
},
|
||
{
|
||
type: 'match_reminder',
|
||
title: 'Напоминание о матче',
|
||
message_template: '💕 У вас есть матч с *{{name}}*, но вы еще не начали общение!\n\nНе упустите шанс познакомиться поближе!',
|
||
button_template: {
|
||
inline_keyboard: [
|
||
[{ text: '💬 Начать общение', callback_data: 'open_chat:{{matchId}}' }],
|
||
[{ text: '👀 Посмотреть профиль', callback_data: 'view_profile:{{userId}}' }]
|
||
]
|
||
}
|
||
},
|
||
{
|
||
type: 'inactive_matches',
|
||
title: 'Неактивные матчи',
|
||
message_template: '⏰ У вас {{count}} неактивных матчей!\n\nПродолжите общение, чтобы не упустить интересные знакомства!',
|
||
button_template: {
|
||
inline_keyboard: [
|
||
[{ text: '📋 Открыть матчи', callback_data: 'view_matches' }],
|
||
[{ text: '💕 Смотреть новые анкеты', callback_data: 'start_browsing' }]
|
||
]
|
||
}
|
||
},
|
||
{
|
||
type: 'like_summary',
|
||
title: 'Сводка лайков',
|
||
message_template: '💖 У вас {{count}} новых лайков!\n\nПосмотрите, кто проявил к вам интерес сегодня!',
|
||
button_template: {
|
||
inline_keyboard: [
|
||
[{ text: '👀 Посмотреть лайки', callback_data: 'view_likes' }],
|
||
[{ text: '💕 Начать знакомиться', callback_data: 'start_browsing' }]
|
||
]
|
||
}
|
||
}
|
||
];
|
||
|
||
for (const template of templates) {
|
||
await client.query(`
|
||
INSERT INTO notification_templates (id, type, title, message_template, button_template)
|
||
VALUES ($1, $2, $3, $4, $5)
|
||
`, [
|
||
uuid.v4(),
|
||
template.type,
|
||
template.title,
|
||
template.message_template,
|
||
JSON.stringify(template.button_template)
|
||
]);
|
||
}
|
||
}
|
||
|
||
await client.query('COMMIT');
|
||
console.log('Successfully created notification tables');
|
||
} catch (err) {
|
||
await client.query('ROLLBACK');
|
||
console.error('Error creating notification tables:', err);
|
||
} finally {
|
||
client.release();
|
||
pool.end();
|
||
}
|
||
}
|
||
|
||
createNotificationTables().catch(err => console.error('Failed to create notification tables:', err));
|