from django.db import models

from labs.utils.database_fields import EncryptedTextField


class LabManagementSSHKey(models.Model):
    class KeyType(models.TextChoices):
        RSA = "rsa"
        ED25519 = "ed25519"

    name = models.CharField(max_length=40)
    cloud_provider = models.CharField(max_length=20)
    region_name = models.CharField(max_length=40)
    private_key = EncryptedTextField()
    public_key = models.TextField()
    key_fingerprint = models.CharField(max_length=128)
    key_type = models.CharField(
        max_length=20, choices=KeyType.choices, default=KeyType.RSA
    )
    is_active = models.BooleanField(default=True)
    created_at = models.DateTimeField(auto_now_add=True)
    updated_at = models.DateTimeField(auto_now=True)

    class Meta:
        db_table = "lab_management_ssh_keys"