Add context in endpoint dsl (#2453)

* Add context in endpoint dsl like Grape::Middleware::Helpers

* Add CHANGELOG
Fix rubocop

* Change to self

CHANGELOG.md

@@ -7,6 +7,7 @@
 
 #### Fixes
 
+* [#2453](/~https://github.com/ruby-grape/grape/pull/2453): Fix context in rescue_from - [@ericproulx](/~https://github.com/ericproulx).
 * Your contribution here.
 
 ### 2.1.0 (2024/06/15)

lib/grape/dsl/inside_route.rb

@@ -461,6 +461,10 @@ def entity_representation_for(entity_class, object, options)
       def http_version
         env['HTTP_VERSION'] || env[Rack::SERVER_PROTOCOL]
       end
+
+      def context
+        self
+      end
     end
   end
 end

spec/grape/api_spec.rb

@@ -4404,4 +4404,23 @@ def uniqe_id_route
       expect(last_response.body).to eq('deprecated')
     end
   end
+
+  context 'rescue_from context' do
+    subject { last_response }
+
+    let(:api) do
+      Class.new(described_class) do
+        rescue_from :all do
+          error!(context.env, 400)
+        end
+        get { raise ArgumentError, 'Oops!' }
+      end
+    end
+
+    let(:app) { api }
+
+    before { get '/' }
+
+    it { is_expected.to be_bad_request }
+  end
 end