linuxrouter/docker/router-dash/app/action_apply_dhcp_reservations.py

from flask import Blueprint, request, redirect, flash
from auth import require_level
from config_utils import load_core, save_core, verify_core_hash, apply_msg
import sanitize
import validate

bp = Blueprint('action_apply_dhcp_reservations', __name__)

VIEW   = '/view/view_dhcp'


def _row_index():
    try:
        return int(request.form.get('row_index', ''))
    except (ValueError, TypeError):
        return None


def _hash_ok():
    if not verify_core_hash(request.form.get('config_hash', '')):
        flash('Configuration was modified by another session. Please refresh and try again.', 'error')
        return False
    return True


def _flat_index_to_vlan_res(vlans, flat_idx):
    pos = 0
    for vi, vlan in enumerate(vlans):
        for ri in range(len(vlan.get('reservations', []))):
            if pos == flat_idx:
                return vi, ri
            pos += 1
    return None, None


def _parse_ip():
    """Return validated IP string, or None after flashing an error."""
    raw = request.form.get('ip', '').strip()
    if not raw:
        flash('The configuration has not been saved because an IP address is required.', 'error')
        return None
    ip = validate.ip(raw)
    if not ip:
        flash(f'The configuration has not been saved because "{raw}" is not a valid IP address.', 'error')
        return None
    return ip


@bp.route('/action/add_dhcp_reservation', methods=['POST'])
@require_level('administrator')
def add_dhcp_reservation():
    vlan_name     = sanitize.name(request.form.get('vlan_name', ''))
    description   = sanitize.text(request.form.get('description', ''))
    hostname      = sanitize.domainname(request.form.get('hostname', ''))
    mac           = sanitize.mac(request.form.get('mac', ''))
    ip            = _parse_ip()
    radius_client = 'radius_client' in request.form

    if ip is None:
        return redirect(VIEW)

    if not vlan_name:
        flash('The configuration has not been saved because a VLAN is required.', 'error')
        return redirect(VIEW)
    if not mac:
        flash('The configuration has not been saved because a MAC address is required.', 'error')
        return redirect(VIEW)

    if not _hash_ok():
        return redirect(VIEW)

    core  = load_core()
    vlans = core.get('vlans', [])
    vlan  = next((v for v in vlans if v.get('name') == vlan_name), None)
    if vlan is None:
        flash(f'The configuration has not been saved because VLAN "{vlan_name}" was not found.', 'error')
        return redirect(VIEW)

    vlan.setdefault('reservations', []).append({
        'description':   description,
        'hostname':      hostname,
        'mac':           mac,
        'ip':            ip,
        'radius_client': radius_client,
        'enabled':       True,
    })
    save_core(core)

    flash(apply_msg(), 'success')
    return redirect(VIEW)


@bp.route('/action/toggle_dhcp_reservation', methods=['POST'])
@require_level('administrator')
def toggle_dhcp_reservation():
    idx = _row_index()
    if idx is None:
        flash('Invalid request.', 'error')
        return redirect(VIEW)

    if not _hash_ok():
        return redirect(VIEW)

    core  = load_core()
    vlans = core.get('vlans', [])
    vi, ri = _flat_index_to_vlan_res(vlans, idx)
    if vi is None:
        flash('Entry not found.', 'error')
        return redirect(VIEW)

    res = vlans[vi]['reservations'][ri]
    res['enabled'] = not res.get('enabled', True)
    save_core(core)

    flash(apply_msg(), 'success')
    return redirect(VIEW)


@bp.route('/action/edit_dhcp_reservation', methods=['POST'])
@require_level('administrator')
def edit_dhcp_reservation():
    idx = _row_index()
    if idx is None:
        flash('Invalid request.', 'error')
        return redirect(VIEW)

    description   = sanitize.text(request.form.get('description', ''))
    hostname      = sanitize.domainname(request.form.get('hostname', ''))
    mac           = sanitize.mac(request.form.get('mac', ''))
    ip            = _parse_ip()
    radius_client = 'radius_client' in request.form

    if ip is None:
        return redirect(VIEW)
    if not mac:
        flash('The configuration has not been saved because a MAC address is required.', 'error')
        return redirect(VIEW)

    if not _hash_ok():
        return redirect(VIEW)

    core  = load_core()
    vlans = core.get('vlans', [])
    vi, ri = _flat_index_to_vlan_res(vlans, idx)
    if vi is None:
        flash('Entry not found.', 'error')
        return redirect(VIEW)

    res = vlans[vi]['reservations'][ri]
    res.update({
        'description':   description,
        'hostname':      hostname,
        'mac':           mac,
        'ip':            ip,
        'radius_client': radius_client,
        'enabled':       'enabled' in request.form,
    })
    save_core(core)

    flash(apply_msg(), 'success')
    return redirect(VIEW)


@bp.route('/action/delete_dhcp_reservation', methods=['POST'])
@require_level('administrator')
def delete_dhcp_reservation():
    idx = _row_index()
    if idx is None:
        flash('Invalid request.', 'error')
        return redirect(VIEW)

    if not _hash_ok():
        return redirect(VIEW)

    core  = load_core()
    vlans = core.get('vlans', [])
    vi, ri = _flat_index_to_vlan_res(vlans, idx)
    if vi is None:
        flash('Entry not found.', 'error')
        return redirect(VIEW)

    removed = vlans[vi]['reservations'].pop(ri)
    save_core(core)

    flash(apply_msg(), 'success')
    return redirect(VIEW)