mirror of
https://github.com/langgenius/dify-plugin-sdks.git
synced 2026-07-24 04:26:00 -04:00
Harry
98 lines
3.6 KiB
Python
98 lines
3.6 KiB
Python
from __future__ import annotations
|
|
|
|
from collections.abc import Mapping
|
|
from typing import Any
|
|
|
|
from werkzeug import Request
|
|
|
|
from dify_plugin.entities.trigger import Variables
|
|
from dify_plugin.errors.trigger import EventIgnoreError
|
|
from dify_plugin.interfaces.trigger import Event
|
|
|
|
|
|
class CheckRunUnifiedEvent(Event):
|
|
"""Unified Check Run event (created/completed)."""
|
|
|
|
def _on_event(self, request: Request, parameters: Mapping[str, Any], payload: Mapping[str, Any]) -> Variables:
|
|
payload = request.get_json()
|
|
if not payload:
|
|
raise ValueError("No payload received")
|
|
|
|
allowed_actions = parameters.get("actions") or []
|
|
action = payload.get("action")
|
|
if allowed_actions and action not in allowed_actions:
|
|
raise EventIgnoreError()
|
|
|
|
run = payload.get("check_run")
|
|
if not isinstance(run, Mapping):
|
|
raise ValueError("No check_run in payload")
|
|
|
|
self._check_name(run, parameters.get("check_name"))
|
|
self._check_branch(run, parameters.get("branch"))
|
|
self._check_app_slug(run, parameters.get("app_slug"))
|
|
self._check_conclusion(run, parameters.get("conclusion"))
|
|
self._check_actor(payload, parameters.get("actor"))
|
|
|
|
return Variables(variables={**payload})
|
|
|
|
def _check_name(self, run: Mapping[str, Any], value: str | None) -> None:
|
|
if not value:
|
|
return
|
|
names = {v.strip() for v in str(value).split(",") if v.strip()}
|
|
if not names:
|
|
return
|
|
if (run.get("name") or "").strip() not in names:
|
|
raise EventIgnoreError()
|
|
|
|
def _check_branch(self, run: Mapping[str, Any], value: str | None) -> None:
|
|
if not value:
|
|
return
|
|
branches = {v.strip() for v in str(value).split(",") if v.strip()}
|
|
if not branches:
|
|
return
|
|
# Prefer PR branch if present
|
|
branch: str | None = None
|
|
for pr in run.get("pull_requests") or []:
|
|
head = pr.get("head") if isinstance(pr, Mapping) else None
|
|
if isinstance(head, Mapping):
|
|
branch = head.get("ref")
|
|
if branch:
|
|
break
|
|
if not branch:
|
|
suite = run.get("check_suite") if isinstance(run, Mapping) else None
|
|
if isinstance(suite, Mapping):
|
|
branch = suite.get("head_branch")
|
|
if branch not in branches:
|
|
raise EventIgnoreError()
|
|
|
|
def _check_app_slug(self, run: Mapping[str, Any], value: str | None) -> None:
|
|
if not value:
|
|
return
|
|
slugs = {v.strip() for v in str(value).split(",") if v.strip()}
|
|
if not slugs:
|
|
return
|
|
app = run.get("app") or {}
|
|
slug = (app.get("slug") if isinstance(app, Mapping) else None) or ""
|
|
if slug not in slugs:
|
|
raise EventIgnoreError()
|
|
|
|
def _check_conclusion(self, run: Mapping[str, Any], value: str | None) -> None:
|
|
if not value:
|
|
return
|
|
allowed = {v.strip().lower() for v in str(value).split(",") if v.strip()}
|
|
if not allowed:
|
|
return
|
|
conclusion = (run.get("conclusion") or "").lower()
|
|
if conclusion and conclusion not in allowed:
|
|
raise EventIgnoreError()
|
|
|
|
def _check_actor(self, payload: Mapping[str, Any], value: str | None) -> None:
|
|
if not value:
|
|
return
|
|
users = {v.strip() for v in str(value).split(",") if v.strip()}
|
|
if not users:
|
|
return
|
|
actor_login = (payload.get("sender") or {}).get("login")
|
|
if actor_login not in users:
|
|
raise EventIgnoreError()
|