from django.db import models


class EngineType(models.Model):
    engine_name = models.CharField(max_length=200)
    subdomain_discovery = models.BooleanField()
    dir_file_search = models.BooleanField()
    subdomain_takeover = models.BooleanField()
    port_scan = models.BooleanField()
    fetch_url = models.BooleanField()
    yaml_configuration = models.TextField()
    default_engine = models.BooleanField(null=True, default=False)

    def __str__(self):
        return self.engine_name


class Wordlist(models.Model):
    name = models.CharField(max_length=200)
    short_name = models.CharField(max_length=50)
    count = models.IntegerField(default=0)

    def __str__(self):
        return self.name


class Configuration(models.Model):
    name = models.CharField(max_length=200)
    short_name = models.CharField(max_length=50)
    content = models.TextField()

    def __str__(self):
        return self.name