Source code for src.commands.social_connector

# src/commands/social_connector.py
import logging
import random
from typing import Optional

logger = logging.getLogger(__name__)




def register_social_connector_command(bolt_app):
    @bolt_app.command("/connect")
    def handle_social_connector(ack, respond, body, client):
        ack()

        channel = body.get("channel_id")
        team_id = body.get("team_id") or (body.get("authorizations") or [{}])[0].get("team_id") or ""
        if not channel or not team_id:
            respond("Could not determine the current channel or workspace for the social connector test.")
            return

        posted = send_social_connector_message(client, channel, team_id)
        if not posted:
            respond("No matching pair found yet. Have a few people set their tags with `/picktags` and try again.")





def find_matching_pair(team_id: str) -> tuple[Optional[str], Optional[str], list[str]]:
    """
    Find two users from the workspace with at least one shared interest tag.
    Returns (user1_id, user2_id, shared_tags) or (None, None, []) if no pair is found.
    """
    from services.mongo_service import get_all_user_interests

    all_users = get_all_user_interests(team_id)
    users_with_tags = [user for user in all_users if user.get("tags")]

    if len(users_with_tags) < 2:
        return None, None, []

    matching_pairs = []
    for first_index in range(len(users_with_tags)):
        for second_index in range(first_index + 1, len(users_with_tags)):
            first_user = users_with_tags[first_index]
            second_user = users_with_tags[second_index]
            shared_tags = sorted(set(first_user["tags"]) & set(second_user["tags"]))
            if shared_tags:
                matching_pairs.append((first_user["user_id"], second_user["user_id"], shared_tags))

    if not matching_pairs:
        return None, None, []

    return random.choice(matching_pairs)





def send_social_connector_message(client, channel: str, team_id: str) -> bool:
    """
    Find two users with shared interest tags and post a soft introduction in the channel.
    """
    from services.llm_service import get_social_connector_icebreaker, get_social_connector_message

    user1_id, user2_id, shared_tags = find_matching_pair(team_id)
    if not user1_id or not user2_id:
        logger.info("[SOCIAL] No matching pair found for team %s", team_id)
        return False

    message = get_social_connector_message(user1_id, user2_id, shared_tags)
    icebreaker = get_social_connector_icebreaker(shared_tags)
    try:
        client.chat_postMessage(channel=channel, text=message)
        client.chat_postMessage(channel=channel, text=icebreaker)
        logger.info("[SOCIAL] Posted connector message for %s and %s", user1_id, user2_id)
        return True
    except Exception as error:
        logger.error("[SOCIAL] Failed to post connector message: %s", error)
        return False