Add http.route and appsec params to wordpress

DataDog · Dec 28, 2023 · a35d80b · a35d80b
1 parent cd8c66e
commit a35d80b
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/Integrations/Integrations/WordPress/V1/WordPressIntegrationLoader.php b/src/Integrations/Integrations/WordPress/V1/WordPressIntegrationLoader.php
@@ -40,6 +40,23 @@ public function load(WordPressIntegration $integration)
             $span->meta[Tag::COMPONENT] = WordPressIntegration::NAME;
         });
 
+         \DDTrace\hook_method('WP', 'main',  null, function ($This, $scope, $args) {
+            if (\property_exists($This, 'did_permalink') && $This->did_permalink === true) {
+                $rootSpan = \DDTrace\root_span();
+                if (\property_exists($This, 'matched_rule')) {
+                    $rootSpan->meta[Tag::HTTP_ROUTE] = $This->matched_rule;
+                }
+                if (\method_exists($route, 'parameters') &&
+                    function_exists('\datadog\appsec\push_params') &&
+                    \property_exists($This, 'query_vars')) {
+                    $parameters = $This->query_vars;
+                    if (count($parameters) > 0) {
+                        \datadog\appsec\push_params($parameters);
+                    }
+                }
+            }
+         });
+
         \DDTrace\trace_method('WP', 'init', function (SpanData $span) use ($service) {
             $span->name = $span->resource = 'WP.init';
             $span->type = Type::WEB_SERVLET;