import os

from config import get_storage_root
from database import get_user_quota


def get_user_files(username):
    user_dir = os.path.join(get_storage_root(), username)
    if not os.path.exists(user_dir):
        return []
    files = []
    for fname in os.listdir(user_dir):
        fpath = os.path.join(user_dir, fname)
        if os.path.isfile(fpath):
            files.append(fname)
    return files


def can_upload(username, file_size_bytes):
    quota_mb, used_mb = get_user_quota(username)
    if quota_mb is None:
        return False
    file_size_mb = file_size_bytes / (1024 * 1024)
    return (used_mb + file_size_mb) <= quota_mb


def get_user_dir(username):
    user_dir = os.path.join(get_storage_root(), username)
    os.makedirs(user_dir, exist_ok=True)
    return user_dir