web_server.py

from flask import Flask, render_template, request, jsonify, send_from_directory, redirect, url_for, session
from flask_login import LoginManager, UserMixin, login_user, logout_user, login_required, current_user
from datetime import datetime
import config
from database import Database
from translations import get_translation

app = Flask(__name__)
app.config['SECRET_KEY'] = config.WEB_SECRET_KEY
db = Database()

login_manager = LoginManager()
login_manager.init_app(app)
login_manager.login_view = 'login'

class User(UserMixin):
    def __init__(self, id):
        self.id = id

@login_manager.user_loader
def load_user(user_id):
    return User(user_id)

@app.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        username = request.form.get('username')
        password = request.form.get('password')
        
        if username == config.WEB_USERNAME and password == config.WEB_PASSWORD:
            user = User(username)
            login_user(user)
            return redirect(url_for('index'))
        
        return render_template('login.html', error=True, lang=session.get('language', 'en'))
    
    return render_template('login.html', error=False, lang=session.get('language', 'en'))

@app.route('/logout')
@login_required
def logout():
    logout_user()
    return redirect(url_for('login'))

@app.route('/set_language/<lang>')
def set_language(lang):
    if lang in ['en', 'uk']:
        session['language'] = lang
    return redirect(request.referrer or url_for('login'))

@app.route('/')
@login_required
def index():
    lang = session.get('language', config.DEFAULT_LANGUAGE)
    return render_template('index.html', lang=lang)


@app.route('/api/stats')
@login_required
def get_stats():
    stats = db.get_stats()
    return jsonify(stats)


@app.route('/api/chats')
@login_required
def get_chats():
    limit = request.args.get('limit', 100, type=int)
    chats = db.get_chats(limit=limit)
    return jsonify(chats)


@app.route('/api/chats/<chat_id>/messages')
@login_required
def get_messages(chat_id):
    try:
        chat_id = int(chat_id)
    except ValueError:
        return jsonify({'error': 'Invalid chat_id'}), 400
    
    limit = request.args.get('limit', 100, type=int)
    offset = request.args.get('offset', 0, type=int)
    include_deleted = request.args.get('include_deleted', 'true').lower() == 'true'
    
    messages = db.get_messages(chat_id, limit=limit, offset=offset, 
                              include_deleted=include_deleted)
    return jsonify(messages)


@app.route('/api/messages/<int:message_id>/edits')
@login_required
def get_message_edits(message_id):
    chat_id = request.args.get('chat_id', type=int)
    if not chat_id:
        return jsonify({'error': 'chat_id is required'}), 400
    
    edits = db.get_message_edits(message_id, chat_id)
    return jsonify(edits)


@app.route('/api/search')
@login_required
def search_messages():
    query = request.args.get('q', '')
    chat_id = request.args.get('chat_id', type=int)
    limit = request.args.get('limit', 100, type=int)
    
    if not query:
        return jsonify({'error': 'Query parameter "q" is required'}), 400
    
    results = db.search_messages(query, chat_id=chat_id, limit=limit)
    return jsonify(results)


@app.route('/media/<path:filename>')
@login_required
def serve_media(filename):
    return send_from_directory(config.MEDIA_FOLDER, filename)


@app.route('/favicon.ico')
def favicon():
    """Favicon placeholder"""
    return '', 204


@app.template_filter('datetime')
def format_datetime(value):
    if isinstance(value, str):
        try:
            value = datetime.fromisoformat(value)
        except:
            return value
    return value.strftime('%Y-%m-%d %H:%M:%S') if value else ''


def run_server():
    """Run web server"""
    print("=" * 50)
    print("  Telegram Logger - Web Interface")
    print("=" * 50)
    print(f"✓ Server running: http://{config.WEB_HOST}:{config.WEB_PORT}")
    print("✓ Press Ctrl+C to stop")
    print("=" * 50)
    
    app.run(
        host=config.WEB_HOST,
        port=config.WEB_PORT,
        debug=True,
        use_reloader=False
    )


if __name__ == '__main__':
    run_server()