import subprocess
import json

gapi_bin = "/root/.hermes/profiles/khoi/skills/productivity/google-workspace/scripts/google_api.py"

# Folders inside Micro
# 1ai6-k8LMopsVeMEsYQXcRH3ZzU4vWmHL is '02. Bài đọc'
r = subprocess.run(['python3', gapi_bin, 'drive', 'search', '--raw-query', "'1ai6-k8LMopsVeMEsYQXcRH3ZzU4vWmHL' in parents and trashed = false"], capture_output=True, text=True)
folders = json.loads(r.stdout)

all_drive_files = []

for folder in folders:
    folder_name = folder['name'] # e.g. 26.11.03
    folder_id = folder['id']
    q = f"'{folder_id}' in parents and trashed = false"
    sub_r = subprocess.run(['python3', gapi_bin, 'drive', 'search', '--raw-query', q], capture_output=True, text=True)
    try:
        files = json.loads(sub_r.stdout)
        for f in files:
            all_drive_files.append({
                'folder_date': folder_name,
                'id': f['id'],
                'name': f['name'],
                'mimeType': f['mimeType']
            })
    except Exception as e:
        print(f"Error {folder_name}: {e}")

# Also check Micro PA folder (1Q_QOGHW6dIqYbKgC775QH-G4VSKZf4Sm)
r_pa = subprocess.run(['python3', gapi_bin, 'drive', 'search', '--raw-query', "'1Q_QOGHW6dIqYbKgC775QH-G4VSKZf4Sm' in parents and trashed = false"], capture_output=True, text=True)
files_pa = json.loads(r_pa.stdout)
for f in files_pa:
    all_drive_files.append({
        'folder_date': 'Micro PA',
        'id': f['id'],
        'name': f['name'],
        'mimeType': f['mimeType']
    })

print(f"Total files on Drive right now: {len(all_drive_files)}")
for item in all_drive_files:
    print(f"[{item['folder_date']}] {item['name']} ({item['id']})")