Put EE routes in EE files under EE directories

1 files changed, 21 insertions, 0 deletions

diff --git a/lib/gitlab/patch/draw_route.rb b/lib/gitlab/patch/draw_route.rb
new file mode 100644
index 00000000000..4396e811a8c
--- /dev/null
+++ b/lib/gitlab/patch/draw_route.rb
@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+# We're patching `ActionDispatch::Routing::Mapper` in
+# config/initializers/routing_draw.rb
+module Gitlab
+  module Patch
+    module DrawRoute
+      def draw(routes_name)
+        instance_eval(File.read(Rails.root.join("config/routes/#{routes_name}.rb")))
+
+        draw_ee(routes_name)
+      end
+
+      def draw_ee(routes_name)
+        path = Rails.root.join("ee/config/routes/#{routes_name}.rb")
+
+        instance_eval(File.read(path)) if File.exist?(path)
+      end
+    end
+  end
+end