butter-portal/app/routes.py

from email.mime import image

from alembic.util import obfuscate_url_pw

from app import app
from flask import render_template, flash, redirect, url_for, request, session
from app.forms import LoginForm, SettingsForm
from flask_login import login_user, current_user, logout_user, login_required
import sqlalchemy as sa
from app import db
from app.models import User, Setting
from werkzeug.datastructures import FileStorage
import json
from flask_babel import _
import base64
import random
from wordfreq import top_n_list
import secrets
import string

def gen_username():
    words = top_n_list("en", 5000)
    prefix = random.randint(1000, 9999)
    return f"{random.choice(words)}{random.choice(words)}{prefix}"

def gen_password():
    characters = string.ascii_letters + string.digits
    password = ''.join(secrets.choice(characters) for i in range(20))
    return password



def get_setting(name) -> str:
    setting = db.session.scalar(sa.select(Setting).where(Setting.key == name))
    return str(setting.value)

def set_setting(name: str, value: str):
    setting = db.session.scalar(sa.select(Setting).where(Setting.key == name))
    print(f"I have changed {setting.key}")
    setting.value = value
    db.session.add(setting)

def dump_settings(filename: str) -> None:
    settings = db.session.execute(sa.select(Setting)).scalars().all()
    settings_dict = {s.key: s.value for s in settings}
    print(settings_dict)
    with open(filename, "w") as f:
        json.dump(settings_dict, f, indent=4)
@app.route('/')
@app.route('/index')
def index():
    enable_chat = get_setting("enable_chat")
    enable_app_store = get_setting("enable_app_store")
    enable_map_viewer = get_setting("enable_map_viewer")
    enable_file_viewer = get_setting("enable_file_viewer")
    enable_deltachat = get_setting("enable_deltachat")
    service_array = []
    usb_inserted = False # actual test of whether USB is inserted
    usb_has_maps = False # actual test of whether USB has maps folder
    usb_has_appstore = False # actual test of whether USB has an appstore
    if enable_deltachat == 'true':
        service_array.append({"name": "Secure Messaging", "image": url_for("static", filename="images/deltachat-icon.png"), "url": url_for("messaging") })
    if enable_chat == 'true':
        service_array.append({"name": "Message Board", "image": url_for("static", filename="images/chat-icon.png"), "url": app.config["CONVENE_INSTALL_PATH"] })
    if enable_app_store == 'true' and usb_has_appstore:
        service_array.append({"name": "Apps", "image": url_for("static", filename="images/appstore-icon.svg")})
    if enable_map_viewer == 'true' and usb_has_maps:
        service_array.append({"name": "Offline Maps", "image": url_for("static", filename="images/maps-icon.png")})
    if enable_file_viewer == 'true':
        name = "Files"
        if not usb_inserted:
            name = "Insert USB to browse files"
        service_array.append({
                             "name": name,
                             "image": url_for("static", filename="images/explore-icon.svg"),
                             "url": url_for("usb")})
    return render_template('index.html', title='Home', get_setting=get_setting, services=service_array)

@app.route('/usb')
def usb():
    return render_template('usb-file-viewer.html', title='File Viewer')

@app.route('/login', methods=['GET', 'POST'])
def login():
    if current_user.is_authenticated:
        return redirect(url_for('admin'))
    form = LoginForm()
    form.remember_me.data = False
    if form.validate_on_submit():
        user = db.session.scalar(sa.select(User).where(User.username == form.username.data))
        if user is None or not user.check_password(form.password.data):
            flash(_('Invalid username or password'))
            return redirect(url_for('login'))
        login_user(user)
        return redirect(url_for('admin'))
    return render_template('login.html', title='Sign in', form=form, get_setting=get_setting)


@app.route('/logout')
def logout():
    logout_user()
    return redirect(url_for('index'))
@app.route('/admin', methods=['GET', 'POST'])
@login_required
def admin():
    form = SettingsForm()
    populate_settings = ['butterbox_name', 'wifi_password', 'enable_access_point', 'ssid', 'enable_file_viewer', 'enable_map_viewer', 'enable_app_store', 'enable_chat', 'enable_deltachat' ]

    bool_settings = ['enable_access_point','enable_file_viewer', 'enable_map_viewer', 'enable_app_store', 'enable_chat', 'enable_deltachat']

    if not form.is_submitted():
        for s in populate_settings:
            if s in bool_settings:
                getattr(form, s).data = (get_setting(s) == "true")
            else:
                getattr(form, s).data = get_setting(s)
    non_admin_settings_changed = False
    if form.validate_on_submit():
        if form.submit.data:
            for s in populate_settings:
                new_value = getattr(form, s).data
                if s in bool_settings:
                    new_value = str(new_value).lower() # all settings are str fow now
                existing_value = get_setting(s)
                if new_value != existing_value:
                    print(f"New value was changed for {s}. Existing value was {existing_value}, new value was {new_value}")
                    set_setting(s, new_value)
                    non_admin_settings_changed = True
                    if s in ['butterbox_name', 'wifi_password', 'ssid', 'enable_access_point', 'enable_chat', 'enable_delta_chat']:
                        app.config['SETTINGS_CHANGED'] = True

            new_logo = form.butterbox_logo.data
            if new_logo.filename:
                logo_stream = form.butterbox_logo.data.stream
                b64_logo = base64.b64encode(logo_stream.read()).decode('utf-8')
                file_mimetype = form.butterbox_logo.data.mimetype
                new_value = f"data:{file_mimetype};base64,{b64_logo}"
                existing_value = get_setting('butterbox_logo')
                if new_value != existing_value:
                    set_setting('butterbox_logo', new_value)
                    non_admin_settings_changed = True
            new_admin_password = form.admin_password.data
            if new_admin_password:
                existing_admin_password = get_setting('admin_password')
                if new_admin_password != existing_admin_password:
                    set_setting('admin_password', new_admin_password)
                    non_admin_settings_changed = True

            if app.config['SETTINGS_CHANGED']:
                flash(_("⚠️ Some settings won't fully take effect until the Butter Box restarts. Click 'Apply Changes' to restart."))
            else:
                if non_admin_settings_changed:
                    flash(
                        _("Settings successfully changed."))
            db.session.commit()

        if form.apply_changes.data:
            set_setting('apply_changes', "true")
            dump_settings("settings.txt")
            flash(_("⚠️ Changes applied! Please wait for the box to restart."))


    return render_template('admin.html', get_setting=get_setting, form=form)


@app.route('/messaging', methods=['GET', 'POST'])
def messaging():
    return render_template('messaging.html', get_setting=get_setting)

@app.route("/deltachat_credentials", methods=["POST"])
def generate_random_deltachat_credentials():
    ip = app.config['BUTTERBOX_DEFAULT_IP']
    username = gen_username()
    password = gen_password()

    flash(f"Username:    {username}")
    flash(f"Password:    {password}")
    flash(f"IP:  {ip}")
    return redirect(url_for("messaging"))