yt-local/youtube/__init__.py

from youtube import util
from .get_app_version import app_version
import flask
from flask import request
import jinja2
import settings
import traceback
import logging
import re
from sys import exc_info
from flask_babel import Babel

yt_app = flask.Flask(__name__)
yt_app.config['TEMPLATES_AUTO_RELOAD'] = True
yt_app.url_map.strict_slashes = False

# Don't log full tracebacks for handled FetchErrors
class FetchErrorFilter(logging.Filter):
    def filter(self, record):
        if record.exc_info and record.exc_info[0] == util.FetchError:
            return False
        return True

yt_app.logger.addFilter(FetchErrorFilter())
# yt_app.jinja_env.trim_blocks = True
# yt_app.jinja_env.lstrip_blocks = True

# Configure Babel for i18n
import os
yt_app.config['BABEL_DEFAULT_LOCALE'] = 'en'
# Use absolute path for translations directory to avoid issues with package structure changes
_app_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
yt_app.config['BABEL_TRANSLATION_DIRECTORIES'] = os.path.join(_app_root, 'translations')

def get_locale():
    """Determine the best locale based on user preference or browser settings"""
    # Check if user has a language preference in settings
    if hasattr(settings, 'language') and settings.language:
        locale = settings.language
        print(f'[i18n] Using user preference: {locale}')
        return locale
    # Otherwise, use browser's Accept-Language header
    # Only match languages with available translations
    locale = request.accept_languages.best_match(['en', 'es'])
    print(f'[i18n] Using browser language: {locale}')
    return locale or 'en'

babel = Babel(yt_app, locale_selector=get_locale)


yt_app.add_url_rule('/settings', 'settings_page', settings.settings_page, methods=['POST', 'GET'])


@yt_app.route('/')
def homepage():
    return flask.render_template('home.html', title="YT Local")


@yt_app.route('/licenses')
def licensepage():
    return flask.render_template(
        'licenses.html',
        title="Licenses - YT Local"
    )


theme_names = {
    0: 'light_theme',
    1: 'gray_theme',
    2: 'dark_theme',
}


@yt_app.context_processor
def inject_theme_preference():
    return {
        'theme_path': '/youtube.com/static/' + theme_names[settings.theme] + '.css',
        'settings': settings,
        # Detect version
        'current_version': app_version()['version'],
        'current_branch': app_version()['branch'],
        'current_commit': app_version()['commit'],
    }


@yt_app.template_filter('commatize')
def commatize(num):
    if num is None:
        return ''
    if isinstance(num, str):
        try:
            num = int(num)
        except ValueError:
            return num
    return '{:,}'.format(num)


def timestamp_replacement(match):
    time_seconds = 0
    for part in match.group(0).split(':'):
        time_seconds = 60*time_seconds + int(part)
    return (
        """
          <a href="#" id="timestamp%s">%s</a>
          <script>
           // @license magnet:?xt=urn:btih:0b31508aeb0634b347b8270c7bee4d411b5d4109&dn=agpl-3.0.txt AGPL-v3-or-Later
           (function main() {
             'use strict';
             const player = document.getElementById('js-video-player');
             const a = document.getElementById('timestamp%s');
             a.addEventListener('click', function(event) {
               player.currentTime = %s
             });
           }());
           // @license-end
          </script>
        """ % (
            str(time_seconds),
            match.group(0),
            str(time_seconds),
            str(time_seconds)
        )
    )


TIMESTAMP_RE = re.compile(r'\b(\d?\d:)?\d?\d:\d\d\b')


@yt_app.template_filter('timestamps')
def timestamps(text):
    return TIMESTAMP_RE.sub(timestamp_replacement, text)


@yt_app.errorhandler(500)
def error_page(e):
    slim = request.args.get('slim', False) # whether it was an ajax request
    if exc_info()[0] == util.FetchError:
        fetch_err = exc_info()[1]
        error_code = fetch_err.code

        if error_code == '429' and settings.route_tor:
            error_message = ('Error: YouTube blocked the request because the Tor'
                ' exit node is overutilized. Try getting a new exit node by'
                ' using the New Identity button in the Tor Browser.')
            if fetch_err.error_message:
                error_message += '\n\n' + fetch_err.error_message
            if fetch_err.ip:
                error_message += '\n\nExit node IP address: ' + fetch_err.ip
            return flask.render_template('error.html', error_message=error_message, slim=slim), 502

        elif error_code == '429':
            error_message = ('YouTube is temporarily blocking requests from your IP address (429 Too Many Requests).\n\n'
                'Try:\n'
                '• Wait a few minutes and refresh\n'
                '• Enable Tor routing in Settings for automatic IP rotation\n'
                '• Use a VPN to change your IP address')
            if fetch_err.ip:
                error_message += '\n\nYour IP: ' + fetch_err.ip
            return flask.render_template('error.html', error_message=error_message, slim=slim), 429

        elif error_code == '502' and ('Failed to resolve' in str(fetch_err) or 'Failed to establish' in str(fetch_err)):
            error_message = ('Could not connect to YouTube.\n\n'
                'Check your internet connection and try again.')
            return flask.render_template('error.html', error_message=error_message, slim=slim), 502

        elif error_code == '403':
            error_message = ('YouTube blocked this request (403 Forbidden).\n\n'
                'Try enabling Tor routing in Settings.')
            return flask.render_template('error.html', error_message=error_message, slim=slim), 403

        elif error_code == '404':
            error_message = 'Error: The page you are looking for isn\'t here.'
            return flask.render_template('error.html', error_code=error_code,
                                         error_message=error_message, slim=slim), 404

        else:
            # Catch-all for any other FetchError (400, etc.)
            error_message = f'Error communicating with YouTube ({error_code}).'
            if fetch_err.error_message:
                error_message += '\n\n' + fetch_err.error_message
            return flask.render_template('error.html', error_message=error_message, slim=slim), 502

    return flask.render_template('error.html', traceback=traceback.format_exc(),
                                 slim=slim), 500


font_choices = {
    0: 'initial',
    1: '"liberation serif", "times new roman", calibri, carlito, serif',
    2: 'arial, "liberation sans", sans-serif',
    3: 'verdana, sans-serif',
    4: 'tahoma, sans-serif',
}


@yt_app.route('/shared.css')
def get_css():
    return flask.Response(
        flask.render_template(
            'shared.css',
            font_family=font_choices[settings.font]
        ),
        mimetype='text/css',
    )


# This is okay because the flask urlize function puts the href as the first
# property
YOUTUBE_LINK_RE = re.compile(r'<a href="(' + util.YOUTUBE_URL_RE_STR + ')"')
old_urlize = jinja2.filters.urlize


def prefix_urlize(*args, **kwargs):
    result = old_urlize(*args, **kwargs)
    return YOUTUBE_LINK_RE.sub(r'<a href="/\1"', result)


jinja2.filters.urlize = prefix_urlize