Update huggingface_storage.py

huggingface_storage.py

@@ -132,4 +132,91 @@ def delete_project_folder_from_hf(
         # Si le dossier est déjà supprimé, HfHubHTTPError 404 est levée par delete_folder
         if "404" in str(e):
              return True, "Dossier non trouvé sur Hugging Face (déjà supprimé ou ID invalide)."
-        return False, f"Échec de la suppression Hugging Face: {str(e)}"
+        return False, f"Échec de la suppression Hugging Face: {str(e)}"
+
+def create_empty_repo_folder_on_hf(
+    repo_slug: str,
+    user_id: str,
+    repo_id: Optional[str] = None
+) -> tuple[bool, str]:
+    """
+    Crée une structure de dossier vide (représentant le dépôt) sur le Dataset Hugging Face.
+    Ceci est souvent implémenté en créant un fichier .gitkeep ou README.md initial.
+    Nous allons créer un fichier README.md minimal.
+    """
+    repo_id = repo_id if repo_id else HF_DATASET_REPO_ID
+    if not repo_id:
+        return False, "Erreur de configuration: Repository Hugging Face non défini."
+
+    readme_content = f"# Dépôt {repo_slug}\n\nBienvenue sur votre dépôt GitForge, hébergé sur Mailix.\n\nCommencez par ajouter ou créer des fichiers."
+    path_in_repo = f"{user_id}/{repo_slug}/README.md"
+    
+    try:
+        # Utilisation de HfApi.upload_file pour créer un fichier unique, ce qui crée le dossier
+        commit_url = api.upload_file(
+            path_or_fileobj=readme_content.encode('utf-8'),
+            path_in_repo=path_in_repo,
+            repo_id=repo_id,
+            repo_type="dataset",
+            commit_message=f"Initialisation du dépôt {repo_slug}"
+        )
+        
+        logger.info(f"Dépôt vide créé sur HF pour {repo_slug}. Commit: {commit_url}")
+        return True, "Dépôt initialisé avec succès sur Hugging Face."
+        
+    except Exception as e:
+        logger.error(f"Échec de l'initialisation du dépôt HF pour {repo_slug}: {e}")
+        return False, f"Échec de l'initialisation Hugging Face: {str(e)}"
+
+
+def list_repo_files_from_hf(
+    repo_slug: str,
+    user_id: str,
+    repo_id: Optional[str] = None
+) -> tuple[list[Dict], str]:
+    """
+    Liste tous les fichiers et dossiers dans le chemin spécifique (user_id/repo_slug) 
+    du Dataset Hugging Face.
+    """
+    repo_id = repo_id if repo_id else HF_DATASET_REPO_ID
+    if not repo_id:
+        return [], "Erreur de configuration: Repository Hugging Face non défini."
+
+    # Le chemin dans le repo est 'user_id/repo_slug'
+    path_to_repo = f"{user_id}/{repo_slug}"
+
+    try:
+        # Utiliser HfFileSystem pour lister le contenu du dossier
+        items = fs.ls(f"datasets/{repo_id}/{path_to_repo}", detail=True)
+        
+        file_list = []
+        for item in items:
+            # item est un dictionnaire avec 'name', 'type', 'size', 'last_modified', etc.
+            # On ignore le nom du dossier racine pour l'affichage utilisateur
+            file_name = item['name'].split('/')[-1]
+
+            # Ignorer les dossiers et les fichiers système cachés (.DS_Store, .gitattributes, etc.)
+            if item['type'] == 'directory' or file_name.startswith('.'):
+                 # Pour une future étape, on pourrait inclure les dossiers
+                 continue
+
+            file_list.append({
+                "name": file_name,
+                "path": item['name'], # Chemin complet pour référence interne
+                "size": item.get('size', 0),
+                "type": "file", # Dans cette version, nous listons seulement les fichiers
+                "last_modified": item.get('last_modified', 'N/A')
+            })
+            
+        logger.info(f"Liste des fichiers récupérée pour {path_to_repo} ({len(file_list)} fichiers).")
+        return file_list, "Liste des fichiers récupérée avec succès."
+
+    except HfHubHTTPError as e:
+        if "404" in str(e):
+            # Le dossier n'existe pas encore (dépôt vide sans initialisation)
+            return [], f"Le dépôt '{repo_slug}' n'a pas encore de fichiers sur Hugging Face (404)."
+        logger.error(f"Erreur HTTP HF lors de la liste des fichiers pour {path_to_repo}: {e}")
+        return [], f"Erreur Hugging Face: {str(e)}"
+    except Exception as e:
+        logger.error(f"Erreur inconnue lors de la liste des fichiers HF pour {path_to_repo}: {e}")
+        return [], f"Erreur inconnue lors de l'accès à Hugging Face: {str(e)}"