import time
import base64
from flask import Flask, render_template, request

app = Flask(__name__)

VERSION = '1.0.9'


@app.route('/')
def index():
    return render_template('index.html', version=VERSION)


@app.route('/health')
def health():
    return {'ok': True, 'version': VERSION}


@app.route('/upload', methods=['POST'])
def upload():
    t0 = time.time()
    total = 0
    # base64-режим: клиент отправляет поле "data"
    data = request.form.get('data', '')
    if data:
        # убрать префикс data:...;base64,
        if ',' in data:
            data = data.split(',', 1)[1]
        # form-encode превращает + в пробелы — восстановить
        data = data.replace(' ', '+')
        # добавить padding если потерялся
        missing = len(data) % 4
        if missing:
            data += '=' * (4 - missing)
        raw = base64.b64decode(data)
        total = len(raw)
    # файловый режим
    elif request.files:
        for f in request.files.values():
            while True:
                chunk = f.read(8192)
                if not chunk:
                    break
                total += len(chunk)
    # raw-режим: тело как есть
    else:
        total = len(request.get_data())
    elapsed = round((time.time() - t0) * 1000)
    return {'ok': True, 'bytes': total, 'elapsed_ms': elapsed}


if __name__ == '__main__':
    import gunicorn.app.base

    class StandaloneApp(gunicorn.app.base.BaseApplication):
        def __init__(self, app, options=None):
            self.application = app
            self.options = options or {}
            super().__init__()

        def load_config(self):
            for k, v in self.options.items():
                self.cfg.set(k, v)

        def load(self):
            return self.application

    StandaloneApp(app, {
        'bind': '0.0.0.0:5000',
        'workers': 2,
        'worker_class': 'gevent',
        'timeout': 120,
        'keepalive': 75,
    }).run()