DNS settings first commit and static host management

dns/admin.py

@@ -0,0 +1,3 @@
+from django.contrib import admin
+
+# Register your models here.

dns/apps.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class DnsConfig(AppConfig):
+    default_auto_field = 'django.db.models.BigAutoField'
+    name = 'dns'

dns/forms.py

@@ -0,0 +1,50 @@
+import requests
+from .models import DNSSettings, StaticHost
+
+from django import forms
+from crispy_forms.helper import FormHelper
+from crispy_forms.layout import Layout, Fieldset, ButtonHolder, Submit, Div, Field, HTML
+from crispy_forms.bootstrap import FormActions, StrictButton
+from django.core.exceptions import ValidationError
+from datetime import datetime
+from django.core.exceptions import ValidationError
+import re
+
+
+
+class StaticHostForm(forms.ModelForm):
+    class Meta:
+        model = StaticHost
+        fields = ['hostname', 'ip_address']
+
+    def __init__(self, *args, **kwargs):
+        super(StaticHostForm, self).__init__(*args, **kwargs)
+        self.helper = FormHelper()
+        self.helper.form_method = 'post'
+        if self.instance.pk:
+            delete_html = "<a href='javascript:void(0)' class='btn btn-outline-danger' data-command='delete' onclick='openCommandDialog(this)'>Delete</a>"
+        else:
+            delete_html = ''
+        self.helper.layout = Layout(
+            Fieldset(
+                'Static DNS',
+                Div(
+                    Field('hostname', css_class='form-control'),
+                    Field('ip_address', css_class='form-control'),
+                    css_class='col-md-6'
+                ),
+            ),
+            FormActions(
+                Submit('save', 'Save', css_class='btn btn-primary'),
+                HTML('<a class="btn btn-outline-primary" href="/dns/">Back</a> '),
+                HTML(delete_html),
+            )
+        )
+
+    def clean(self):
+        hostname = self.cleaned_data.get('hostname')
+        if hostname:
+            regex = r'^[a-zA-Z][a-zA-Z0-9-\.]*[a-zA-Z0-9]$'
+            if not re.match(regex, hostname):
+                raise ValidationError('Invalid hostname')
+        return

dns/migrations/0001_initial.py

@@ -0,0 +1,39 @@
+# Generated by Django 5.0.4 on 2024-04-26 17:42
+
+import uuid
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='DNSSettings',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('name', models.CharField(default='dns_settings', max_length=100)),
+                ('dns_primary', models.GenericIPAddressField(blank=True, default='1.1.1.1', null=True)),
+                ('dns_secondary', models.GenericIPAddressField(blank=True, default='1.0.0.1', null=True)),
+                ('pending_changes', models.BooleanField(default=True)),
+                ('created', models.DateTimeField(auto_now_add=True)),
+                ('updated', models.DateTimeField(auto_now=True)),
+                ('uuid', models.UUIDField(default=uuid.uuid4, editable=False, unique=True)),
+            ],
+        ),
+        migrations.CreateModel(
+            name='StaticHost',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('hostname', models.CharField(max_length=100, unique=True)),
+                ('ip_address', models.GenericIPAddressField()),
+                ('created', models.DateTimeField(auto_now_add=True)),
+                ('updated', models.DateTimeField(auto_now=True)),
+                ('uuid', models.UUIDField(default=uuid.uuid4, editable=False, unique=True)),
+            ],
+        ),
+    ]

dns/models.py

@@ -0,0 +1,25 @@
+from django.db import models
+import uuid
+
+
+class DNSSettings(models.Model):
+    name = models.CharField(default='dns_settings', max_length=100)
+    dns_primary = models.GenericIPAddressField(blank=True, null=True, default='1.1.1.1')
+    dns_secondary = models.GenericIPAddressField(blank=True, null=True, default='1.0.0.1')
+    pending_changes = models.BooleanField(default=True)
+
+    created = models.DateTimeField(auto_now_add=True)
+    updated = models.DateTimeField(auto_now=True)
+    uuid = models.UUIDField(unique=True, default=uuid.uuid4, editable=False)
+
+
+class StaticHost(models.Model):
+    hostname = models.CharField(max_length=100, unique=True)
+    ip_address = models.GenericIPAddressField()
+
+    created = models.DateTimeField(auto_now_add=True)
+    updated = models.DateTimeField(auto_now=True)
+    uuid = models.UUIDField(unique=True, default=uuid.uuid4, editable=False)
+
+    def __str__(self):
+        return self.hostname

dns/tests.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

dns/views.py

@@ -0,0 +1,53 @@
+from django.contrib.auth.decorators import login_required
+from django.shortcuts import render, get_object_or_404, redirect
+from django.contrib import messages
+from user_manager.models import UserAcl
+from .models import DNSSettings, StaticHost
+from .forms import StaticHostForm
+
+
+@login_required
+def view_static_host_list(request):
+    dns_settings, _ = DNSSettings.objects.get_or_create(name='dns_settings')
+    static_host_list = StaticHost.objects.all().order_by('hostname')
+    context = {
+        'dns_settings': dns_settings,
+        'static_host_list': static_host_list,
+    }
+    return render(request, 'dns/static_host_list.html', context=context)
+
+
+@login_required
+def view_manage_static_host(request):
+    if not UserAcl.objects.filter(user=request.user).filter(user_level__gte=40).exists():
+        return render(request, 'access_denied.html', {'page_title': 'Access Denied'})
+    dns_settings, _ = DNSSettings.objects.get_or_create(name='dns_settings')
+    if request.GET.get('uuid'):
+        static_dns = get_object_or_404(StaticHost, uuid=request.GET.get('uuid'))
+        if request.GET.get('action') == 'delete':
+            if request.GET.get('confirmation') == 'delete':
+                static_dns.delete()
+                dns_settings.pending_changes = True
+                dns_settings.save()
+                messages.success(request, 'Static DNS deleted successfully')
+                return redirect('/dns/')
+            else:
+                messages.warning(request, 'Static DNS not deleted|Invalid confirmation')
+                return redirect('/dns/')
+    else:
+        static_dns = None
+
+    form = StaticHostForm(request.POST or None, instance=static_dns)
+    if form.is_valid():
+        form.save()
+        dns_settings.pending_changes = True
+        dns_settings.save()
+        messages.success(request, 'Static DNS saved successfully')
+        return redirect('/dns/')
+
+    context = {
+        'dns_settings': dns_settings,
+        'form': form,
+        'instance': static_dns,
+    }
+    return render(request, 'generic_form.html', context=context)

templates/base.html

@@ -113,6 +113,15 @@
                       </p>
                     </a>
                 </li>
+        
+                <li class="nav-item">
+                    <a href="/dns/" class="nav-link {% if '/dns/' in request.path %}active{% endif %}">
+                      <i class="fas fa-address-book nav-icon"></i>
+                      <p>
+                        DNS
+                      </p>
+                    </a>
+                </li>
 
               <li class="nav-item">
                   <a href="/firewall/rule_list/" class="nav-link {% if '/firewall/' in request.path %}active{% endif %}">

templates/dns/static_host_list.html

@@ -0,0 +1,55 @@
+{% extends 'base.html' %}
+
+{% block content %}
+    <div class='row'>
+        <div class='col-lg-6'>
+            <div class="card card-primary card-outline">
+                <div class="card-header">
+                    <h3 class="card-title">Static Host List</h3>
+                </div>
+
+                <div class="card-body">
+                <div class="row">
+                    <div class="col-lg-12">
+                        <table class="table table-hover">
+                            <thead>
+                            <tr>
+                                <th>Hostname</th>
+                                <th>IP Address</th>
+                                <th></th>
+                            </tr>
+                            </thead>
+                            <tbody>
+                            {% if static_host_list %}
+                            {% for static_host in static_host_list %}
+                                <tr>
+                                    <td>{{ static_host.hostname }}</td>
+                                    <td>{{ static_host.ip_address }}</td>
+                                    <td class="min-width">
+                                        <a href="/dns/manage_static_host/?uuid={{ static_host.uuid }}"><i class="fas fa-edit"></i></a>
+                                    </td>
+                                </tr>
+                            {% endfor %}
+                            {% else %}
+                                <tr>
+                                    <td colspan="3"></td>
+                                </tr>
+                            {% endif %}
+                            </tbody>
+                        </table>
+                    </div>
+                </div>
+                
+                <div class="row">
+                    <div class="col-lg-12">
+                        <a href="/dns/manage_static_host/" class="btn btn-primary">Add Static Host</a>
+                        <a href="/dns/manage_settings/" class="btn btn-primary">DNS Settings</a>
+                        <a href="/dns/apply_config/" class="btn btn-primary">Apply Config</a>
+                    </div>
+                </div>
+                </div>
+            </div>
+        </div>
+    </div>
+{% endblock %}
+

wireguard_webadmin/settings.py

@@ -36,10 +36,13 @@ INSTALLED_APPS = [
     'django.contrib.sessions',
     'django.contrib.messages',
     'django.contrib.staticfiles',
+    'crispy_forms',
+    'crispy_bootstrap4',
     'wireguard',
     'user_manager',
     'wireguard_tools',
     'firewall',
+    'dns'
 ]
 
 MIDDLEWARE = [
@@ -51,6 +54,8 @@ MIDDLEWARE = [
     'django.contrib.messages.middleware.MessageMiddleware',
     'django.middleware.clickjacking.XFrameOptionsMiddleware',
 ]
+CRISPY_ALLOWED_TEMPLATE_PACKS = "bootstrap4"
+CRISPY_TEMPLATE_PACK = "bootstrap4"
 
 ROOT_URLCONF = 'wireguard_webadmin.urls'
 

wireguard_webadmin/urls.py

@@ -24,12 +24,15 @@ from accounts.views import view_create_first_user, view_login, view_logout
 from wireguard_tools.views import export_wireguard_configs, download_config_or_qrcode, restart_wireguard_interfaces
 from api.views import wireguard_status, cron_check_updates, cron_update_peer_latest_handshake, routerfleet_get_user_token, routerfleet_authenticate_session
 from firewall.views import view_redirect_rule_list, manage_redirect_rule, view_firewall_rule_list, manage_firewall_rule, view_manage_firewall_settings, view_generate_iptables_script, view_reset_firewall, view_firewall_migration_required
+from dns.views import view_static_host_list, view_manage_static_host
 
 
 urlpatterns = [
     path('admin/', admin.site.urls),
     path('', view_welcome, name='welcome'),
     path('status/', view_wireguard_status, name='wireguard_status'),
+    path('dns/', view_static_host_list, name='static_host_list'),
+    path('dns/manage_static_host/', view_manage_static_host, name='manage_static_host'),
     path('peer/list/', view_wireguard_peer_list, name='wireguard_peer_list'),
     path('peer/manage/', view_wireguard_peer_manage, name='wireguard_peer_manage'),
     path('peer/manage_ip_address/', view_manage_ip_address, name='manage_ip_address'),