"""
File handling utilities for SkillSync
"""

import os
import shutil
from datetime import datetime

class FileHandler:
    """Handles file operations for submissions"""
    
    def __init__(self, upload_dir="data/user_data/submissions", max_size_mb=100):
        self.upload_dir = upload_dir
        self.max_size = max_size_mb * 1024 * 1024
        self.allowed_extensions = [".zip", ".pdf", ".ipynb", ".py", ".txt", ".md"]
    
    def ensure_upload_dir(self):
        """Ensure upload directory exists"""
        os.makedirs(self.upload_dir, exist_ok=True)
    
    def validate_file(self, file):
        """Validate uploaded file"""
        if file is None:
            return False, "No file provided"
        
        file_name = getattr(file, 'name', 'unknown')
        file_ext = os.path.splitext(file_name)[1].lower()
        
        if file_ext not in self.allowed_extensions:
            return False, f"File type {file_ext} not allowed"
        
        return True, "File validated successfully"
    
    def save_file(self, file, user_id, assignment_type):
        """Save uploaded file"""
        try:
            self.ensure_upload_dir()
            original_name = getattr(file, 'name', 'uploaded_file')
            original_name = os.path.basename(original_name)
            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
            new_name = f"{user_id}_{assignment_type}_{timestamp}_{original_name}"
            destination = os.path.join(self.upload_dir, new_name)
            
            if hasattr(file, 'name'):
                shutil.copy2(file.name, destination)
            
            return True, destination
        except Exception as e:
            return False, str(e)
    
    def get_file_info(self, filepath):
        """Get information about a file"""
        if not os.path.exists(filepath):
            return None
        
        stat = os.stat(filepath)
        return {
            "path": filepath,
            "size": stat.st_size,
            "created": datetime.fromtimestamp(stat.st_ctime).strftime("%Y-%m-%d %H:%M:%S"),
            "modified": datetime.fromtimestamp(stat.st_mtime).strftime("%Y-%m-%d %H:%M:%S")
        }