export_all.py

"""
export_all.py
=============
Exports ALL channels and DMs you are a member of.

Usage:
    python export_all.py
"""

import os
import json
from datetime import datetime
from slack_client import (
    auth_check, fetch_users, get_all_pages,
    export_conversation, write_gpt_file
)

# ─── Config ──────────────────────────────────────────────────────────────────
OUTPUT_DIR    = "slack_export/all"
FETCH_FILES   = True    # download attached files
FETCH_THREADS = True    # fetch thread replies
# ─────────────────────────────────────────────────────────────────────────────


def main():
    os.makedirs(OUTPUT_DIR, exist_ok=True)

    print("Connecting to Slack...")
    me, workspace = auth_check()
    print(f"  Logged in as : {me}")
    print(f"  Workspace    : {workspace}\n")

    user_map = fetch_users()
    print(f"  {len(user_map)} users loaded\n")

    print("Fetching all conversations...")
    conversations = get_all_pages(
        "conversations.list", "channels",
        {
            "types": "public_channel,private_channel,im,mpim",
            "exclude_archived": False,
            "limit": 200,
        }
    )
    # Only keep conversations you're actually a member of
    conversations = [c for c in conversations if c.get("is_member", True)]
    print(f"  Found {len(conversations)} conversations\n")

    # Save metadata
    with open(os.path.join(OUTPUT_DIR, "_conversations.json"), "w") as f:
        json.dump(conversations, f, indent=2, ensure_ascii=False)

    with open(os.path.join(OUTPUT_DIR, "_users.json"), "w") as f:
        json.dump(user_map, f, indent=2, ensure_ascii=False)

    gpt_lines = [
        "SLACK FULL EXPORT",
        f"Workspace : {workspace}",
        f"User      : {me}",
        f"Date      : {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}",
        f"Total     : {len(conversations)} conversations",
    ]

    print("Exporting all conversations...")
    for conv in conversations:
        lines = export_conversation(
            conv, user_map, OUTPUT_DIR,
            fetch_files=FETCH_FILES,
            fetch_threads=FETCH_THREADS,
        )
        gpt_lines.extend(lines)

    write_gpt_file(OUTPUT_DIR, gpt_lines)
    print(f"\nDone! Output saved to: {os.path.abspath(OUTPUT_DIR)}/")


if __name__ == "__main__":
    main()