mirror of
https://github.com/PostHog/posthog.git
synced 2024-12-01 04:12:23 +01:00
99 lines
3.5 KiB
Python
99 lines
3.5 KiB
Python
import re
|
|
from typing import Any, Dict
|
|
from urllib.parse import urlparse
|
|
|
|
import structlog
|
|
from django.conf import settings
|
|
|
|
from ee.tasks.subscriptions.subscription_utils import generate_assets
|
|
from posthog.models.exported_asset import ExportedAsset
|
|
from posthog.models.integration import Integration, SlackIntegration
|
|
from posthog.models.sharing_configuration import SharingConfiguration
|
|
|
|
logger = structlog.get_logger(__name__)
|
|
|
|
from posthog.celery import app
|
|
|
|
SHARED_LINK_REGEX = r"\/(?:shared_dashboard|shared|embedded)\/(.+)"
|
|
|
|
|
|
def _block_for_asset(asset: ExportedAsset) -> Dict:
|
|
image_url = asset.get_public_content_url()
|
|
alt_text = None
|
|
if asset.insight:
|
|
alt_text = asset.insight.name or asset.insight.derived_name
|
|
|
|
if settings.DEBUG:
|
|
image_url = "https://source.unsplash.com/random"
|
|
|
|
return {"type": "image", "image_url": image_url, "alt_text": alt_text}
|
|
|
|
|
|
def _handle_slack_event(event_payload: Any) -> None:
|
|
slack_team_id = event_payload.get("team_id")
|
|
channel = event_payload.get("event").get("channel")
|
|
message_ts = event_payload.get("event").get("message_ts")
|
|
unfurl_id = event_payload.get("event").get("unfurl_id")
|
|
source = event_payload.get("event").get("source")
|
|
links_to_unfurl = event_payload.get("event").get("links")
|
|
|
|
unfurls = {}
|
|
|
|
for link_obj in links_to_unfurl:
|
|
link = link_obj.get("url")
|
|
parsed = urlparse(link)
|
|
matches = re.search(SHARED_LINK_REGEX, parsed.path)
|
|
|
|
if matches:
|
|
share_token = matches[1]
|
|
|
|
# First we try and get the sharingconfig for the given link
|
|
try:
|
|
sharing_config: SharingConfiguration = SharingConfiguration.objects.get(
|
|
access_token=share_token, enabled=True
|
|
)
|
|
except SharingConfiguration.DoesNotExist:
|
|
logger.info("No SharingConfiguration found")
|
|
continue
|
|
|
|
team_id = sharing_config.team_id
|
|
|
|
# Now we try and get the SlackIntegration for the specificed Posthog team and Slack Team
|
|
try:
|
|
integration = Integration.objects.get(kind="slack", team=team_id, config__team__id=slack_team_id)
|
|
slack_integration = SlackIntegration(integration)
|
|
|
|
except Integration.DoesNotExist:
|
|
logger.info("No SlackIntegration found for this team")
|
|
continue
|
|
|
|
# With both the integration and the resource we are good to go!!
|
|
|
|
insights, assets = generate_assets(sharing_config, 1)
|
|
|
|
if assets:
|
|
unfurls[link] = {
|
|
"blocks": [
|
|
{
|
|
"type": "section",
|
|
"text": {"type": "mrkdwn", "text": insights[0].name or insights[0].derived_name},
|
|
"accessory": _block_for_asset(assets[0]),
|
|
}
|
|
]
|
|
}
|
|
|
|
if unfurls:
|
|
try:
|
|
slack_integration.client.chat_unfurl(unfurls=unfurls, unfurl_id=unfurl_id, source=source, channel="", ts="")
|
|
except Exception as e:
|
|
# NOTE: This is temporary as a test to understand if the channel and ts are actually required as the docs are not clear
|
|
slack_integration.client.chat_unfurl(
|
|
unfurls=unfurls, unfurl_id=unfurl_id, source=source, channel=channel, ts=message_ts
|
|
)
|
|
raise e
|
|
|
|
|
|
@app.task()
|
|
def handle_slack_event(payload: Any) -> None:
|
|
return _handle_slack_event(payload)
|