"""
「关于本站」后台管理：编辑单条记录 (id=1)。
插件版。
"""
import os

from flask import Blueprint, render_template, request
from flask_wtf.csrf import validate_csrf
from wtforms.validators import ValidationError

from applications.common.utils.http import fail_api, success_api
from applications.common.utils.rights import authorize
from applications.common.utils.validate import str_escape
from applications.extensions import db
from applications.models import About

dir_path = os.path.dirname(os.path.abspath(__file__))
bp = Blueprint(
    'about', __name__,
    url_prefix='/system/about',
    template_folder=os.path.join(dir_path, '..', 'templates'),
)


def _get_about() -> About:
    a = About.query.get(1)
    if a is None:
        a = About(id=1, site_name='旺珂 · 导航')
        db.session.add(a)
        db.session.commit()
    return a


@bp.get('/')
@authorize("site:about:main")
def main():
    about = _get_about()
    return render_template('system/about/main.html', about=about)


@bp.post('/save')
@authorize("site:about:main", log=True)
def save():
    req_json = request.get_json(force=True, silent=True) or {}
    try:
        validate_csrf(req_json.get("csrf_token"))
    except ValidationError:
        return fail_api(msg='非法请求')

    a = _get_about()
    fields = {
        'site_name': (req_json.get('siteName') or '').strip()[:120],
        'content_md': (req_json.get('contentMd') or '').strip(),
        'icp': (req_json.get('icp') or '').strip()[:120],
        'contact_email': (req_json.get('contactEmail') or '').strip()[:120],
        'contact_qq': (req_json.get('contactQq') or '').strip()[:32],
        'contact_wechat': (req_json.get('contactWechat') or '').strip()[:120],
        'contact_telegram': (req_json.get('contactTelegram') or '').strip()[:120],
        'contact_github': (req_json.get('contactGithub') or '').strip()[:255],
        'donate_url': (req_json.get('donateUrl') or '').strip()[:255],
    }
    for k, v in fields.items():
        setattr(a, k, v)
    db.session.commit()
    return success_api(msg='保存成功')