from django.db import models

from django.conf import settings
import magic

mime = magic.Magic(mime=True)


class Post(models.Model):
    content = models.TextField(blank=True)
    created_at = models.DateTimeField(auto_now_add=True)
    updated_at = models.DateTimeField(auto_now=True)

    author = models.ForeignKey(
        settings.AUTH_USER_MODEL,
        on_delete=models.CASCADE,
        related_name='posts'
    )
    
    hub = models.ForeignKey(
        'pages.Hub',
        on_delete=models.CASCADE,
        null=True,
        blank=True
    )
    
    def __str__(self):
        return f"Post {self.id}"
    
class PostContent(models.Model):
    post = models.ForeignKey(Post, related_name='contents', on_delete=models.CASCADE)
    mime_type = models.CharField(max_length=100)
    file = models.FileField(upload_to='post_contents/', null=True, blank=True)  # For images/videos
    alt_text = models.TextField(blank=True, null=True)  # For text content

    def __str__(self):
        return f"Content for Post {self.post.id} - Type: {self.mime_type}"
    
    def save(self, *args, **kwargs):
        if self.file and not self.file._committed:
            self.mime_type = mime.from_buffer(self.file.read(1024))
        return super().save(*args, **kwargs)