from django.db import models
from django.conf import settings
from django_tidb.fields import BigAutoRandomField


class LabSSHAuthorization(models.Model):
    class StatusChoices(models.TextChoices):
        PENDING = "pending"
        APPROVED = "approved"
        DENIED = "denied"

    id = BigAutoRandomField(primary_key=True)
    user = models.ForeignKey(
        settings.AUTH_USER_MODEL, on_delete=models.CASCADE, blank=True, null=True
    )
    lab_instance = models.ForeignKey(
        "labs.LabInstance", on_delete=models.CASCADE, blank=True, null=True
    )
    status = models.CharField(
        max_length=20,
        choices=StatusChoices.choices,
        default=StatusChoices.PENDING,
    )
    authorized_at = models.DateTimeField(blank=True, null=True)
    created_at = models.DateTimeField(auto_now_add=True)

    class Meta:
        db_table = "lab_ssh_authorizations"