fix impl unwrap block #2207

pyrefly/lib/lsp/non_wasm/server.rs

@@ -5451,6 +5451,10 @@ impl Server {
                 "convert_dict",
                 transaction.convert_dict_code_actions(&handle, range)
             );
+            timed_refactor_action!(
+                "unwrap_block",
+                transaction.unwrap_block_code_actions(&handle, range)
+            );
             timed_refactor_action!(
                 "pytest_fixture_type_annotation",
                 transaction.pytest_fixture_type_annotation_code_actions(

pyrefly/lib/state/lsp.rs

@@ -4190,6 +4190,14 @@ impl<'a> Transaction<'a> {
         quick_fixes::convert_dict::convert_dict_code_actions(self, handle, selection)
     }
 
+    pub fn unwrap_block_code_actions(
+        &self,
+        handle: &Handle,
+        selection: TextRange,
+    ) -> Option<Vec<LocalRefactorCodeAction>> {
+        quick_fixes::unwrap_block::unwrap_block_code_actions(self, handle, selection)
+    }
+
     pub fn change_signature_code_actions(
         &self,
         handle: &Handle,

pyrefly/lib/state/lsp/quick_fixes.rs

@@ -32,3 +32,4 @@ pub(crate) mod types;
 pub(crate) mod unnecessary_type_conversion;
 pub(crate) mod unused_ignore;
 pub(crate) mod unused_import;
+pub(crate) mod unwrap_block;