Add type definitions for Grape and Sinatra route serializers

Author: y9v

sig/datadog/appsec/api_security/endpoint_collection/grape_routes_serializer.rbs

@@ -0,0 +1,28 @@
+module Datadog
+  module AppSec
+    module APISecurity
+      module EndpointCollection
+        interface _GrapeRoute
+          def request_method: () -> String
+          def pattern: () -> _GrapeRoutePattern
+        end
+
+        interface _GrapeRoutePattern
+          def origin: () -> String
+        end
+
+        class GrapeRoutesSerializer
+          @routes: Array[_GrapeRoute]
+
+          @path_prefix: String
+
+          def initialize: (Array[_GrapeRoute] routes, ?path_prefix: String) -> void
+
+          def to_enum: () -> Enumerator[Core::Telemetry::Event::AppEndpointsLoaded::endpoint]
+
+          def serialize_route: (_GrapeRoute route) -> Core::Telemetry::Event::AppEndpointsLoaded::endpoint
+        end
+      end
+    end
+  end
+end

sig/datadog/appsec/api_security/endpoint_collection/rails_routes_serializer.rbs

@@ -2,30 +2,47 @@ module Datadog
   module AppSec
     module APISecurity
       module EndpointCollection
-        interface _Route
+        interface _RailsRoute
           def dispatcher?: () -> bool
           def verb: () -> String
-          def path: () -> _RoutePath
+          def path: () -> _RailsRoutePath
+          def app: () -> _RailsRouteApp
         end
 
-        interface _RoutePath
-          def spec: () -> _RouteSpec
+        interface _RailsRoutePath
+          def spec: () -> _RailsRouteSpec
         end
 
-        interface _RouteSpec
+        interface _RailsRouteSpec
           def to_s: () -> String
         end
 
+        interface _RailsRouteApp
+          def rack_app: () -> _RackApp
+        end
+
+        interface _RackApp
+          def is_a?: (Class) -> bool
+
+          def routes: () -> untyped
+        end
+
         class RailsRoutesSerializer
           FORMAT_SUFFIX: String
 
-          @routes: Array[_Route]
+          @routes: Array[_RailsRoute]
 
-          def initialize: (Array[_Route] routes) -> void
+          def initialize: (Array[_RailsRoute] routes) -> void
 
           def to_enum: () -> Enumerator[Core::Telemetry::Event::AppEndpointsLoaded::endpoint]
 
-          def serialize_route: (_Route route) -> Core::Telemetry::Event::AppEndpointsLoaded::endpoint
+          def serialize_route: (_RailsRoute route) -> Core::Telemetry::Event::AppEndpointsLoaded::endpoint
+
+          private
+
+          def grape_app?: (_RackApp rack_app) -> bool
+
+          def sinatra_app?: (_RackApp rack_app) -> bool
         end
       end
     end

sig/datadog/appsec/api_security/endpoint_collection/sinatra_routes_serializer.rbs

@@ -0,0 +1,23 @@
+module Datadog
+  module AppSec
+    module APISecurity
+      module EndpointCollection
+        interface _SinatraRoute
+          def safe_string: () -> String
+        end
+
+        class SinatraRoutesSerializer
+          @routes: Hash[String, Array[_SinatraRoute]]
+
+          @path_prefix: String
+
+          def initialize: (Hash[String, Array[_SinatraRoute]] routes, ?path_prefix: String) -> void
+
+          def to_enum: () -> Enumerator[Core::Telemetry::Event::AppEndpointsLoaded::endpoint]
+
+          def serialize_route: (_SinatraRoute route, String method) -> Core::Telemetry::Event::AppEndpointsLoaded::endpoint
+        end
+      end
+    end
+  end
+end

vendor/rbs/grape/0/grape.rbs

@@ -0,0 +1,4 @@
+module Grape
+  class API
+  end
+end

vendor/rbs/sinatra/0/sinatra.rbs

@@ -2,4 +2,7 @@ module Sinatra
   class Response
     def initialize: (::Array[::String] body, ::Integer status, ::Hash[::String, ::String] headers) -> void
   end
+
+  class Base
+  end
 end