import json
import os
import uuid
from datetime import datetime
import logging
from hf_storage import HFStorageManager

logger = logging.getLogger(__name__)

class ReportManager:
    def __init__(self, history_file='saved_reports.json'):
        self.history_file = history_file
        self.storage = HFStorageManager()

    def _load_history(self):
        data = self.storage.load_file(self.history_file)
        if data is None:
            return {}
        if isinstance(data, dict):
            return data
        return {}

    def _save_history(self, history):
        self.storage.save_file(self.history_file, history)

    def get_student_reports(self, enrollment_no):
        history = self._load_history()
        # Return list of reports, sorted by timestamp (newest first)
        reports = history.get(str(enrollment_no), [])
        reports.sort(key=lambda x: x.get('timestamp', ''), reverse=True)
        return reports

    def save_report(self, enrollment_no, report_data):
        history = self._load_history()
        enrollment_no = str(enrollment_no)
        
        if enrollment_no not in history:
            history[enrollment_no] = []

        report_id = str(uuid.uuid4())
        timestamp = datetime.now().isoformat()

        # Create a summary title (e.g., "Report - Dec 03, 2025")
        date_str = datetime.now().strftime("%b %d, %Y")
        title = f"Report - {date_str}"

        new_report = {
            'id': report_id,
            'title': title,
            'timestamp': timestamp,
            'data': report_data
        }

        history[enrollment_no].append(new_report)
        self._save_history(history)
        return new_report

    def get_report(self, enrollment_no, report_id):
        history = self._load_history()
        enrollment_no = str(enrollment_no)
        
        if enrollment_no in history:
            for report in history[enrollment_no]:
                if report['id'] == report_id:
                    return report
        return None