Update app.py

app.py

@@ -1,19 +1,60 @@
 from flask import Flask, request, jsonify, Response
 import json
-from pywebpush import webpush
+from pywebpush import webpush, WebPushException
+import os
+from cryptography.hazmat.primitives import serialization
+from cryptography.hazmat.primitives.asymmetric import ec
+from base64 import urlsafe_b64encode
 
 app = Flask(__name__)
 PORT = 7860
 
 # Путь к файлу для хранения новостей
 NEWS_FILE = 'news.json'
-# Путь к файлу для хранения подписок на уведомления
+# Путь к файлу для хранения подписок
 SUBSCRIPTIONS_FILE = 'subscriptions.json'
-
-# VAPID keys для push уведомлений. **ЗАМЕНИТЕ ЭТИ PLACEHOLDER КЛЮЧИ НА ВАШИ СОБСТВЕННЫЕ!**
-VAPID_PUBLIC_KEY_BASE64 = "BIdP1w9i3m7C-p9tQ_97z7LgusQ49t8wZ6-L2vcl1kYRt6b-5Zg848p0h7FvL3Qv8K4j_J9z_n-J9y8v6v918" # **ЗАМЕНИТЬ!**
-VAPID_PRIVATE_KEY_BASE64 = "w10j9y0z8x7c6v5b4n3m2l1k0j9h8g7f6e5d4s3a2q1z" # **ЗАМЕНИТЬ!**
-VAPID_CLAIMS = {"subject": "mailto:your-email@example.com"} # Замените на свой email
+# Путь к файлу для хранения VAPID ключей
+VAPID_KEYS_FILE = 'vapid_keys.json'
+
+# --- Генерация и загрузка VAPID ключей ---
+
+def generate_vapid_keys():
+    """Генерирует VAPID ключи и возвращает их в формате base64."""
+    private_key = ec.generate_private_key(ec.SECP256R1())
+    public_key = private_key.public_key()
+
+    private_pem = private_key.private_bytes(
+        encoding=serialization.Encoding.PEM,
+        format=serialization.PrivateFormat.PKCS8,
+        encryption_algorithm=serialization.NoEncryption()
+    )
+    public_pem = public_key.public_bytes(
+        encoding=serialization.Encoding.PEM,
+        format=serialization.PublicFormat.SubjectPublicKeyInfo
+    )
+
+    private_b64 = urlsafe_b64encode(private_pem).decode('utf-8').rstrip('=')
+    public_b64 = urlsafe_b64encode(public_pem).decode('utf-8').rstrip('=')
+
+    return {"private_key": private_b64, "public_key": public_b64}
+
+def load_or_generate_vapid_keys():
+    """Загружает VAPID ключи из файла или генерирует новые, если файла нет."""
+    if os.path.exists(VAPID_KEYS_FILE):
+        with open(VAPID_KEYS_FILE, 'r') as f:
+            return json.load(f)
+    else:
+        keys = generate_vapid_keys()
+        with open(VAPID_KEYS_FILE, 'w') as f:
+            json.dump(keys, f, indent=4)
+        print(f"Сгенерированы новые VAPID ключи и сохранены в {VAPID_KEYS_FILE}")
+        return keys
+
+# Загружаем или генерируем ключи при старте приложения
+vapid_keys = load_or_generate_vapid_keys()
+VAPID_PUBLIC_KEY_BASE64 = vapid_keys["public_key"]
+VAPID_PRIVATE_KEY_BASE64 = vapid_keys["private_key"]
+VAPID_CLAIMS = {"subject": "mailto:your-email@example.com"}  # Замените на свой email
 
 # --- Вспомогательные функции ---
 
@@ -46,7 +87,7 @@ def save_subscriptions(subscriptions):
 def send_push_notification(subscription, message_body):
     """Отправляет push уведомление одному подписчику."""
     try:
-        print(f"Попытка отправить уведомление на endpoint: {subscription['endpoint']}") # Лог перед отправкой
+        print(f"Отправка уведомления на: {subscription['endpoint']}")
         webpush(
             subscription_info=subscription,
             data=message_body,
@@ -54,10 +95,10 @@ def send_push_notification(subscription, message_body):
             vapid_public_key=VAPID_PUBLIC_KEY_BASE64,
             vapid_claims=VAPID_CLAIMS
         )
-        print(f"Уведомление успешно отправлено на endpoint: {subscription['endpoint']}") # Лог после успешной отправки
+        print(f"Уведомление успешно отправлено на: {subscription['endpoint']}")
         return True
-    except Exception as e:
-        print(f"Ошибка отправки уведомления на endpoint: {subscription['endpoint']}. Ошибка: {e}") # Лог ошибки
+    except WebPushException as e:
+        print(f"Ошибка отправки уведомления на: {subscription['endpoint']}. Ошибка: {e}")
         return False
 
 def send_push_to_all(message_body):
@@ -68,9 +109,10 @@ def send_push_to_all(message_body):
         if send_push_notification(sub, message_body):
             valid_subscriptions.append(sub)
         else:
-            print(f"Не удалось отправить уведомление для подписки: {sub['endpoint']}") # Логирование для отладки
-    save_subscriptions(valid_subscriptions) # Сохраняем только валидные подписки
-
+            print(f"Подписка удалена из-за оши��ки: {sub['endpoint']}")
+    if valid_subscriptions != subscriptions:
+        save_subscriptions(valid_subscriptions)
+    return len(valid_subscriptions) > 0
 
 # --- HTML для главной страницы ---
 HTML_INDEX = """
@@ -124,16 +166,14 @@ HTML_INDEX = """
                 Notification.requestPermission().then(permission => {
                     if (permission === "granted") {
                         console.log("Разрешение на уведомления получено.");
-                        subscribeUserToPush(); // Подписываем пользователя после получения разрешения
+                        subscribeUserToPush();
                     } else {
                         console.log("Разрешение на уведомления отклонено.");
                     }
                 });
             } else if (Notification.permission === "granted") {
                 console.log("Разрешение на уведомления уже есть.");
-                subscribeUserToPush(); // Подписываем пользователя, если разрешение уже есть
-            } else {
-                console.log("Уведомления заблокированы.");
+                subscribeUserToPush();
             }
         }
 
@@ -153,10 +193,10 @@ HTML_INDEX = """
             navigator.serviceWorker.ready.then(serviceWorkerRegistration => {
                 serviceWorkerRegistration.pushManager.subscribe({
                     userVisibleOnly: true,
-                    applicationServerKey: urlBase64ToUint8Array(publicKey) // Используем публичный ключ
+                    applicationServerKey: urlBase64ToUint8Array(publicKey)
                 })
                 .then(subscription => {
-                    console.log('Подписка на Push API:', subscription);
+                    console.log('Успешная подписка на Push API:', subscription);
                     sendSubscriptionToServer(subscription);
                 })
                 .catch(error => {
@@ -180,10 +220,10 @@ HTML_INDEX = """
                 return response.json();
             })
             .then(data => {
-                console.log('Подписка успешно сохранена на сервере:', data.message);
+                console.log('Подписка успешно сохранена:', data.message);
             })
             .catch(error => {
-                console.error('Ошибка при сохранении подписки на сервере:', error);
+                console.error('Ошибка при сохранении подписки:', error);
             });
         }
 
@@ -192,13 +232,16 @@ HTML_INDEX = """
                 .then(response => response.json())
                 .then(news => {
                     const newsList = document.getElementById('news-list');
-                    newsList.innerHTML = ''; // Очищаем список перед добавлением
+                    newsList.innerHTML = '';
                     news.forEach(item => {
                         const li = document.createElement('li');
                         li.className = 'list-group-item';
                         li.textContent = item.text;
                         newsList.appendChild(li);
                     });
+                })
+                .catch(error => {
+                    console.error('Ошибка загрузки новостей:', error);
                 });
         }
 
@@ -216,39 +259,34 @@ HTML_INDEX = """
                 .then(response => response.json())
                 .then(data => {
                     console.log(data.message);
-                    document.getElementById('news-text').value = ''; // Очищаем поле ввода
-                    loadNews(); // Обновляем список новостей
+                    document.getElementById('news-text').value = '';
+                    loadNews();
+                })
+                .catch(error => {
+                    console.error('Ошибка добавления новости:', error);
                 });
             }
         }
 
-
         function urlBase64ToUint8Array(base64String) {
             const padding = '='.repeat((4 - base64String.length % 4) % 4);
             const base64 = (base64String + padding)
                 .replace(/-/g, '+')
                 .replace(/_/g, '/');
-
             const rawData = window.atob(base64);
             const outputArray = new Uint8Array(rawData.length);
-
             for (let i = 0; i < rawData.length; ++i) {
                 outputArray[i] = rawData.charCodeAt(i);
             }
             return outputArray;
         }
 
-
-        // Обработка полученных push уведомлений в foreground (если нужно что-то делать на странице)
         navigator.serviceWorker.addEventListener('message', event => {
             if (event.data && event.data.type === 'push-received') {
-                console.log('Получено push уведомление в foreground:', event.data.message);
-                loadNews(); // Обновляем список новостей при получении уведомления
-                // Можно также показать какое-то визуальное уведомление на странице, если нужно
+                console.log('Push уведомление в foreground:', event.data.message);
+                loadNews();
             }
         });
-
-
     </script>
 </body>
 </html>
@@ -261,12 +299,11 @@ self.addEventListener('push', event => {
     const title = 'Новая новость!';
     const options = {
         body: message,
-        icon: '/no-icon.png' // Путь к иконке (теперь в корне, или можно убрать если не нужна иконка)
+        icon: '/no-icon.png' // Убедитесь, что иконка существует, или удалите эту строку
     };
 
     event.waitUntil(self.registration.showNotification(title, options));
 
-    // Отправляем сообщение клиенту (странице) о получении push уведомления
     self.clients.matchAll({ type: 'window' }).then(clients => {
         clients.forEach(client => {
             client.postMessage({ type: 'push-received', message: message });
@@ -276,7 +313,7 @@ self.addEventListener('push', event => {
 
 self.addEventListener('notificationclick', event => {
     event.notification.close();
-    event.waitUntil(clients.openWindow('/')); // Открываем главное окно при клике на уведомление
+    event.waitUntil(clients.openWindow('/'));
 });
 """
 
@@ -292,51 +329,50 @@ MANIFEST_JSON = """
 }
 """
 
-
 # --- Flask маршруты ---
 
 @app.route('/')
 def index():
-    """Главная страница, отображает новости и запрашивает разрешение на уведомления."""
+    """Главная страница."""
     news = load_news()
-    public_key = VAPID_PUBLIC_KEY_BASE64 # Передаем публичный ключ в шаблон
+    public_key = VAPID_PUBLIC_KEY_BASE64
     html = HTML_INDEX.replace('%PUBLIC_KEY%', public_key)
-    # Заполняем список новостей динамически на сервере, чтобы избежать flicker при загрузке
     news_list_html = ''.join([f'<li class="list-group-item">{item["text"]}</li>' for item in news])
     html = html.replace('<ul class="list-group" id="news-list"></ul>', f'<ul class="list-group" id="news-list">{news_list_html}</ul>')
     return html
 
 @app.route('/add_news', methods=['POST'])
 def add_news():
-
-        print("Маршрут /add_news вызван!")  # <--- Отладочный вывод (добавлено)
-        news_text = request.form.get('news_text')
-        if news_text:
-            news_list = load_news()
-            news_list.append({"text": news_text})
-            save_news(news_list)
-
-            # Отправляем push уведомления о новой новости
-            send_push_to_all(news_text)
-
+    """API endpoint для добавления новости."""
+    print("Маршрут /add_news вызван!")
+    news_text = request.form.get('news_text')
+    if news_text:
+        news_list = load_news()
+        news_list.append({"text": news_text})
+        save_news(news_list)
+        success = send_push_to_all(news_text)
+        if success:
             return jsonify({"message": "Новость добавлена и уведомления отправлены"}), 200
-        return jsonify({"error": "Текст новости не предоставлен"}), 400
+        else:
+            return jsonify({"message": "Новость добавлена, но уведомления не отправлены"}), 200
+    return jsonify({"error": "Текст новости не предоставлен"}), 400
 
 @app.route('/get_news')
 def get_news():
-    """API endpoint для получения списка новостей."""
+    """API endpoint для получения новостей."""
     news = load_news()
     return jsonify(news)
 
 @app.route('/save_subscription', methods=['POST'])
 def save_subscription():
-    """API endpoint для сохранения подписки пользователя на уведомления."""
+    """API endpoint для сохранения подписки."""
     subscription_data = request.get_json()
     if subscription_data:
         subscriptions = load_subscriptions()
-        subscriptions.append(subscription_data)
-        save_subscriptions(subscriptions)
-        print(f"Подписка сохранена: {subscription_data['endpoint']}") # Логирование для отладки
+        if not any(sub['endpoint'] == subscription_data['endpoint'] for sub in subscriptions):
+            subscriptions.append(subscription_data)
+            save_subscriptions(subscriptions)
+            print(f"Подписка сохранена: {subscription_data['endpoint']}")
         return jsonify({"message": "Подписка сохранена"}), 201
     return jsonify({"error": "Неверные данные подписки"}), 400
 
@@ -350,6 +386,5 @@ def manifest():
     """Маршрут для manifest.json."""
     return Response(MANIFEST_JSON, mimetype='application/json')
 
-
 if __name__ == '__main__':
-    app.run(debug=True, port=PORT, host='0.0.0.0') # Доступно извне и на порту 7860
+    app.run(debug=True, port=PORT, host='0.0.0.0')