import json

import flask
import flask_login

from . import db
from . import system

blueprint = flask.Blueprint('ipsets', __name__, url_prefix='/ipsets')

@blueprint.route('/', methods=('GET', 'POST'))
@flask_login.login_required
def index():
    if not flask_login.current_user.is_admin:
        return flask.Response('forbidden', status=403, mimetype='text/plain')

    with db.locked():
        ipsets = db.read('ipsets')
        networks = db.read('networks')
        if flask.request.method == 'POST':
            form = flask.request.form
            ipsets = {}
            for name, ip, ip6 in zip(form.getlist('name'), form.getlist('ip'), form.getlist('ip6')):
                if name and name not in networks:
                    ipsets[name] = {
                        'ip': ip.split(),
                        'ip6': ip6.split()
                    }
            db.write('ipsets', ipsets)
            system.run(system.save_config)
            return flask.redirect(flask.url_for('ipsets.index'))
        return flask.render_template('ipsets/index.html', ipsets=ipsets)