export_specific.py

"""
export_specific.py
==================
Exports only the specific channels, DMs, and group DMs you configure below.

Usage:
    python export_specific.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, resolve
)

# ─────────────────────────────────────────────────────────────────────────────
#  CONFIGURE WHAT YOU WANT TO EXPORT
# ─────────────────────────────────────────────────────────────────────────────

# Channels (use exact Slack channel name, no # prefix)
CHANNELS = [
    "android-automation-team",
    "debug-openclaw",
    "general",
]

# DMs — use the person's Slack display name (as shown in Slack sidebar)
DM_USERS = [
    "Fawad Raza",
    "Jop",
    "elicett",
]

# Group DMs — list of member display names (all members must match)
# Leave empty [] to skip group DMs
GROUP_DMS = [
    ["elicett", "Fawad Raza", "Jop", "Ysha Lia"],  # the group from sidebar
]

# ─────────────────────────────────────────────────────────────────────────────
#  OTHER SETTINGS
# ─────────────────────────────────────────────────────────────────────────────
OUTPUT_DIR    = "slack_export/specific"
FETCH_FILES   = True
FETCH_THREADS = True
# ─────────────────────────────────────────────────────────────────────────────


def members_match(conv_members, wanted_names, user_map):
    """Check if a group DM's members match the wanted name list (order-insensitive)."""
    resolved = set(resolve(uid, user_map) for uid in conv_members)
    return resolved == set(wanted_names)


def should_include(conv, user_map):
    """Return True if this conversation matches the user's config."""
    is_dm      = conv.get("is_im", False)
    is_group   = conv.get("is_mpim", False)
    is_channel = not is_dm and not is_group

    if is_channel:
        name = conv.get("name", "")
        return name in CHANNELS

    if is_dm:
        other = resolve(conv.get("user", ""), user_map)
        return other in DM_USERS

    if is_group:
        if not GROUP_DMS:
            return False
        members = conv.get("members", [])
        return any(members_match(members, wanted, user_map) for wanted in GROUP_DMS)

    return False


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 conversation list...")
    all_convs = get_all_pages(
        "conversations.list", "channels",
        {
            "types": "public_channel,private_channel,im,mpim",
            "exclude_archived": False,
            "limit": 200,
        }
    )

    # Filter to only what was configured above
    selected = [c for c in all_convs if should_include(c, user_map)]

    print(f"  Matched {len(selected)} conversation(s) from your config\n")

    if not selected:
        print("  Nothing matched. Check your CHANNELS / DM_USERS / GROUP_DMS names above.")
        return

    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 SPECIFIC EXPORT",
        f"Workspace : {workspace}",
        f"User      : {me}",
        f"Date      : {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}",
        f"Exported  : {len(selected)} conversations",
    ]

    print("Exporting selected conversations...")
    for conv in selected:
        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()