from django.db import models
from targetApp.models import Domain
from scanEngine.models import EngineType


class ScanHistory(models.Model):
    last_scan_date = models.DateTimeField()
    scan_status = models.IntegerField()
    domain_name = models.ForeignKey(Domain, on_delete=models.CASCADE)
    scan_type = models.ForeignKey(EngineType, on_delete=models.CASCADE)

    def __str__(self):
        # debug purpose remove scan type and id in prod
        return self.domain_name.domain_name + self.scan_type.scan_type_name + str(self.id)

class ScannedHost(models.Model):
    subdomain = models.CharField(max_length=1000)
    scan_history = models.ForeignKey(ScanHistory, on_delete=models.CASCADE)
    open_ports = models.CharField(max_length=1000)
    http_status = models.IntegerField(default=0)
    content_length = models.IntegerField(default=0)
    page_title = models.CharField(max_length=1000)
    http_url = models.CharField(max_length=1000)
    ip_address = models.CharField(max_length=1000)
    screenshot_path = models.CharField(max_length=1000, null=True)
    http_header_path = models.CharField(max_length=1000, null=True)
    technology_stack = models.CharField(max_length=1500, null=True)

    def __str__(self):
        return str(self.scan_history.id)


class WayBackEndPoint(models.Model):
    url_of = models.ForeignKey(ScanHistory, on_delete=models.CASCADE)
    http_url = models.CharField(max_length=1000)
    content_length = models.IntegerField(default=0)
    page_title = models.CharField(max_length=1000)
    http_status = models.IntegerField(default=0)

    def __str__(self):
        return self.page_title

class ScanActivity(models.Model):
    scan_of = models.ForeignKey(ScanHistory, on_delete=models.CASCADE)
    title = models.CharField(max_length=1000)
    time = models.CharField(max_length=100)
    status = models.IntegerField()

    def __str__(self):
        return str(self.title)