dnmgmt/whoisdb/models.py

from django.db import models
from django.db.models import Q
from django.urls import reverse

from dncore.models import User
from .validators import HandleValidator, HandleValidatorWithSuffix

import ipaddress


class WhoisObject(models.Model):
	class Meta:
		abstract = True
	handleSuffix = ""

	handle = models.SlugField(max_length=32, unique=True, verbose_name='handle', validators=[HandleValidator()])
	created = models.DateTimeField(auto_now_add=True)
	last_modified = models.DateTimeField(auto_now_add=True)

	#def __init__(self, *args, **kwargs):
	#	super(WhoisObject, self).__init__(*args, **kwargs)

	#	if getattr(self, "handle"):
	#		field = self._meta.get_field("handle")
	#		if HandleValidatorWithSuffix not in map(type, field.validators):
	#			print(self.handle, "NOOOOT")
	#			field.validators.append(HandleValidatorWithSuffix(self.handleSuffix))
	#		else:
	#			print(self.handle, list(map(type, field.validators)))

	def getPK(self):
		return self.handle

	def __str__(self):
		return self.handle

	def getClassName(self):
		return self._meta.object_name

	def genHandle(self, main=None):
		if not main:
			main = self.name
		return self.genGenericHandle(main)

	@classmethod
	def genGenericHandle(clazz, main):
		prefix = ""
		if " " in main:
			parts = main.split(" ")
			prefix = "".join(map(lambda _x: _x[0], parts))
			if len(prefix) < 3 and len(parts[-1]) > 1:
				prefix += parts[-1][1:4 - len(prefix)]
		else:
			prefix = main[0:3]
		prefix = prefix.upper()

		i = 1
		handle = "%s%%d-%s" % (prefix, clazz.handleSuffix)
		while True:
			try:
				prefix
				clazz.objects.get(handle=handle % i)
				i += 1
			except clazz.DoesNotExist:
				break

		return handle % i

	def getNoDeleteReasons(self):
		raise NotImplementedError()

	def canBeDeleted(self):
		return not bool(self.getNoDeleteReasons())

	def handleAuto(self, name=None):
		if self.handle == "AUTO":
			self.handle = self.genHandle(name)


class Maintainer(WhoisObject):
	handleSuffix = "MNT"

	auth = models.ManyToManyField(User)
	handle = models.SlugField(max_length=32, unique=True, verbose_name='handle', validators=[HandleValidatorWithSuffix('MNT')], help_text="Must end with -MNT, eg FOO3-MNT")
	description = models.CharField(max_length=64, blank=True, help_text="Short description what this maintainer is for")

	admin_c = models.ManyToManyField("Contact")

	rir = models.BooleanField(default=False)
	lir = models.BooleanField(default=False)

	# autoInclude = models.BooleanField(default=True)

	def get_absolute_url(self):
		return reverse("whoisdb:handle-detail", kwargs={"handle": self.handle})

	def getNoDeleteReasons(self):
		reasons = []

		mntables = [Contact, ASBlock, ASNumber, InetNum]
		for mntable in mntables:
			candidates = mntable.objects.filter(mnt_by=self).annotate(mntCount=models.Count('mnt_by')).filter(mntCount__lte=1)
			for candidate in candidates:
				reasons.append("Object %s would have no maintainers left." % candidate.handle)

		return reasons

	def canEdit(self, user):
		return user in self.auth.all()


class MntdObject(WhoisObject):
	class Meta:
		abstract = True

	mnt_by = models.ManyToManyField(Maintainer)

	def canEdit(self, user):
		mnts = user.maintainer_set.all()
		objmnts = self.mnt_by.all()
		if hasattr(self, "mnt_lower"):
			objmnts |= self.mnt_lower.all()

		for objmnt in objmnts:
			if objmnt in mnts:
				return True
		return False

	@classmethod
	def getMntQueryset(clazz, mnts, instance, attr=None):
		mntQ = Q(mnt_by__in=mnts)
		if hasattr(clazz, "mnt_lower"):
			mntQ |= Q(mnt_lower__in=mnts)

		qs = clazz.objects.filter(mntQ)
		if attr and instance and instance.pk:
			if type(instance._meta.get_field(attr)) == models.ManyToManyField:
				qs |= getattr(instance, attr).all()
			elif getattr(instance, attr) is not None:
				qs |= clazz.objects.filter(pk=getattr(instance, attr).pk)

		return qs.distinct()


class Contact(MntdObject):
	handleSuffix = "DN"
	TYPE_PERSON = 'PERSON'
	TYPE_ROLE = 'ROLE'
	TYPE = (('person', TYPE_PERSON), ('role', TYPE_ROLE))
	TYPE = (('person', TYPE_PERSON),)

	name = models.CharField(max_length=128)
	type = models.CharField(max_length=10, choices=TYPE, default=TYPE_PERSON)

	def get_absolute_url(self):
		return reverse("whoisdb:handle-detail", kwargs={"handle": self.handle})

	def getNoDeleteReasons(self):
		reasons = []

		contactables = [Maintainer]
		for contactable in contactables:
			candidates = contactable.objects.filter(admin_c=self).annotate(contactCount=models.Count('admin_c')).filter(contactCount__lte=1)
			for candidate in candidates:
				reasons.append("Object %s would have no contact left." % candidate.handle)

		return reasons


class ASBlock(MntdObject):
	handleSuffix = "ASB"

	parent_block = models.ForeignKey("ASBlock", models.CASCADE, null=True, blank=True, default=None)
	name = models.CharField(max_length=32)
	asBegin = models.PositiveIntegerField()
	asEnd = models.PositiveIntegerField()
	description = models.CharField(max_length=64, blank=True)
	admin_c = models.ManyToManyField("Contact")

	mnt_lower = models.ManyToManyField(Maintainer, related_name='lower_asblock_set', blank=True)

	def getResource(self):
		return "%s - %s" % (self.asBegin, self.asEnd)

	def get_absolute_url(self):
		return reverse("whoisdb:handle-detail", kwargs={"handle": self.handle})

	def getNoDeleteReasons(self):
		reasons = []

		if self.asblock_set.count() > 0:
			reasons.append("The AS block is referenced by the following other blocks: %s" % (", ".join(map(lambda _x: _x.handle, self.asblock_set.all()))))

		if self.asnumber_set.count() > 0:
			reasons.append("The AS block is referenced by the following as numbers: %s" % (", ".join(map(lambda _x: _x.handle, self.asnumber_set.all()))))

		return reasons


class ASNumber(MntdObject):
	handleSuffix = "AS"

	number = models.PositiveIntegerField(unique=True, db_index=True)
	volatile = models.BooleanField(default=False, help_text="Check if this AS is not going to be online 24/7 (for example on a laptop)")
	asblock = models.ForeignKey(ASBlock, models.CASCADE)
	name = models.CharField(max_length=32)
	description = models.CharField(max_length=64, blank=True)
	admin_c = models.ManyToManyField("Contact")

	mnt_lower = models.ManyToManyField(Maintainer, related_name='lower_asnumber_set', blank=True)

	def get_absolute_url(self):
		return reverse("whoisdb:handle-detail", kwargs={"handle": self.handle})

	def getNoDeleteReasons(self):
		reasons = []

		return reasons

	def getResource(self):
		return str(self.number)


class InetNum(MntdObject):
	handleSuffix = "NET"

	IPv4 = "ipv4"
	IPv6 = "ipv6"

	PROTO = ((IPv4, 'IPv4'), (IPv6, 'IPv6'))
	protocol = models.CharField(max_length=4, choices=PROTO)
	address = models.GenericIPAddressField(db_index=True)
	netmask = models.PositiveIntegerField()
	parent_range = models.ForeignKey("InetNum", models.CASCADE, null=True, blank=True, default=None)
	name = models.CharField(max_length=64)
	description = models.CharField(max_length=64, blank=True)
	admin_c = models.ManyToManyField("Contact")

	mnt_lower = models.ManyToManyField(Maintainer, related_name='lower_inetnum_set', blank=True)

	def getResource(self):
		return self.prefix()

	def prefix(self):
		""" Helper function, mainly used in templates """
		return "%s/%s" % (self.address, self.netmask)

	def getNetwork(self):
		return ipaddress.ip_network(self.prefix())

	def get_absolute_url(self):
		return reverse("whoisdb:handle-detail", kwargs={"handle": self.handle})

	def getNoDeleteReasons(self):
		reasons = []
		if self.inetnum_set.all().count() > 0:
			reasons.append("The following networks depend on this network: %s" % ", ".join(map(lambda _x: _x.handle, self.inetnum_set.all())))

		return reasons