Fix LSP initialisation bug. (#5771)

## Description

I noticed that LSP wasn't loading the correct syntax highlighting and
that other LSP features like go_to, hover etc.. were not working when a
new project was loaded. It would start working after the first key press
though.

We now wait for parsing on the code_action LSP feature. We also enable
the wait_for_parsing function to wait until the compilation thread is
started so it doesn't return early in the case where the LSP requests
were received by the server before the did_open notification.

before
![Screenshot 2024-03-26 at 11 36 54
am](https://github.com/FuelLabs/sway/assets/1289413/807055fd-12a5-4d14-b165-10a41bd17187)

after
![Screenshot 2024-03-26 at 11 35 39
am](https://github.com/FuelLabs/sway/assets/1289413/f7a440a2-eb0b-48e1-ba82-c78f5e5fa252)

Author: JoshuaBatty

sway-lsp/src/handlers/request.rs

@@ -240,6 +240,7 @@ pub async fn handle_code_action(
     state: &ServerState,
     params: lsp_types::CodeActionParams,
 ) -> Result<Option<lsp_types::CodeActionResponse>> {
+    let _ = state.wait_for_parsing().await;
     match state
         .sessions
         .uri_and_session_from_workspace(&params.text_document.uri)

sway-lsp/src/server_state.rs

@@ -205,7 +205,11 @@ impl ServerState {
     /// this process until `is_compiling` becomes false.
     pub async fn wait_for_parsing(&self) {
         loop {
-            if !self.is_compiling.load(Ordering::SeqCst) {
+            // Check both the is_compiling flag and the last_compilation_state.
+            // Wait if is_compiling is true or if the last_compilation_state is Uninitialized.
+            if !self.is_compiling.load(Ordering::SeqCst)
+                && *self.last_compilation_state.read() != LastCompilationState::Uninitialized
+            {
                 // compilation is finished, lets check if there are pending compilation requests.
                 if self.cb_rx.is_empty() {
                     // no pending compilation work, safe to break.
@@ -225,7 +229,6 @@ impl ServerState {
 
         // Set the retrigger_compilation flag to true so that the compilation exits early
         self.retrigger_compilation.store(true, Ordering::SeqCst);
-        self.wait_for_parsing().await;
 
         // Send a terminate message to the compilation thread
         self.cb_tx

sway-lsp/tests/lib.rs

@@ -209,7 +209,7 @@ fn did_change_stress_test() {
 #[test]
 fn did_change_stress_test_random_wait() {
     run_async!({
-        let test_duration = tokio::time::Duration::from_secs(12 * 60); // 5 minutes timeout
+        let test_duration = tokio::time::Duration::from_secs(5 * 60); // 5 minutes timeout
         let test_future = async {
             std::env::set_var("RUST_BACKTRACE", "1");
             let default_panic = std::panic::take_hook();