"""
Data storage utilities for QR codes
"""

import uuid
from datetime import datetime

# In-memory storage for QR codes (in production, use a database)
qr_codes_db = {}

class QRDataManager:
    def __init__(self):
        pass
    
    def save_qr_record(self, qr_type, content, settings, image_data, page_id=None):
        """Save QR code record to database"""
        qr_id = str(uuid.uuid4())
        qr_record = {
            'id': qr_id,
            'type': qr_type,
            'content': content,
            'settings': settings,
            'created_at': datetime.now().isoformat(),
            'image_data': image_data
        }
        
        if page_id:
            qr_record['page_id'] = page_id
            
        qr_codes_db[qr_id] = qr_record
        return qr_id
    
    def get_qr_record(self, qr_id):
        """Get QR code record"""
        return qr_codes_db.get(qr_id)
    
    def delete_qr_record(self, qr_id):
        """Delete QR code record"""
        if qr_id in qr_codes_db:
            del qr_codes_db[qr_id]
            return True
        return False
    
    def list_qr_codes(self):
        """List all QR codes"""
        qr_list = []
        for qr_id, qr_data in qr_codes_db.items():
            qr_list.append({
                'id': qr_id,
                'type': qr_data['type'],
                'created_at': qr_data['created_at'],
                'preview': f'data:image/png;base64,{qr_data["image_data"]}'
            })
        return qr_list
    
    def qr_exists(self, qr_id):
        """Check if QR code exists"""
        return qr_id in qr_codes_db