feat: execute stmt under cursor (supabase-community#257)

Author: juleswritescode

.cargo/config.toml

@@ -0,0 +1,3 @@
+{
+  "postgrestools.bin": "./target/debug/postgrestools"
+}

.vscode/settings.json

@@ -39,6 +39,7 @@ serde                    = "1.0.195"
 serde_json               = "1.0.114"
 similar                  = "2.6.0"
 smallvec                 = { version = "1.13.2", features = ["union", "const_new", "serde"] }
+strum                    = { version = "0.27.1", features = ["derive"] }
 # this will use tokio if available, otherwise async-std
 sqlx               = { version = "0.8.2", features = ["runtime-tokio", "runtime-async-std", "postgres", "json"] }
 syn                = "1.0.109"
@@ -56,7 +57,6 @@ pgt_analyse                = { path = "./crates/pgt_analyse", version = "0.0.0"
 pgt_analyser               = { path = "./crates/pgt_analyser", version = "0.0.0" }
 pgt_base_db                = { path = "./crates/pgt_base_db", version = "0.0.0" }
 pgt_cli                    = { path = "./crates/pgt_cli", version = "0.0.0" }
-pgt_commands               = { path = "./crates/pgt_commands", version = "0.0.0" }
 pgt_completions            = { path = "./crates/pgt_completions", version = "0.0.0" }
 pgt_configuration          = { path = "./crates/pgt_configuration", version = "0.0.0" }
 pgt_console                = { path = "./crates/pgt_console", version = "0.0.0" }

Cargo.lock

@@ -1,3 +1,4 @@
+use biome_deserialize::StringSet;
 use biome_deserialize_macros::{Merge, Partial};
 use bpaf::Bpaf;
 use serde::{Deserialize, Serialize};
@@ -28,6 +29,9 @@ pub struct DatabaseConfiguration {
     #[partial(bpaf(long("database")))]
     pub database: String,
 
+    #[partial(bpaf(long("allow_statement_executions_against")))]
+    pub allow_statement_executions_against: StringSet,
+
     /// The connection timeout in seconds.
     #[partial(bpaf(long("conn_timeout_secs"), fallback(Some(10)), debug_fallback))]
     pub conn_timeout_secs: u16,
@@ -41,6 +45,7 @@ impl Default for DatabaseConfiguration {
             username: "postgres".to_string(),
             password: "postgres".to_string(),
             database: "postgres".to_string(),
+            allow_statement_executions_against: Default::default(),
             conn_timeout_secs: 10,
         }
     }

Cargo.toml

@@ -111,6 +111,7 @@ impl PartialConfiguration {
                 password: Some("postgres".to_string()),
                 database: Some("postgres".to_string()),
                 conn_timeout_secs: Some(10),
+                allow_statement_executions_against: Default::default(),
             }),
         }
     }

crates/pgt_commands/Cargo.toml

@@ -28,6 +28,7 @@ pgt_workspace           = { workspace = true }
 rustc-hash              = { workspace = true }
 serde                   = { workspace = true, features = ["derive"] }
 serde_json              = { workspace = true }
+strum                   = { workspace = true }
 tokio                   = { workspace = true, features = ["rt", "io-std"] }
 tower-lsp               = { version = "0.20.0" }
 tracing                 = { workspace = true, features = ["attributes"] }

crates/pgt_commands/src/command.rs

@@ -1,10 +1,15 @@
 use pgt_lsp_converters::{PositionEncoding, WideEncoding, negotiated_encoding};
+use pgt_workspace::code_actions::{CommandActionCategory, CommandActionCategoryIter};
+use strum::{EnumIter, IntoEnumIterator};
 use tower_lsp::lsp_types::{
-    ClientCapabilities, CompletionOptions, PositionEncodingKind, SaveOptions, ServerCapabilities,
-    TextDocumentSyncCapability, TextDocumentSyncKind, TextDocumentSyncOptions,
-    TextDocumentSyncSaveOptions, WorkDoneProgressOptions,
+    ClientCapabilities, CodeActionOptions, CompletionOptions, ExecuteCommandOptions,
+    PositionEncodingKind, SaveOptions, ServerCapabilities, TextDocumentSyncCapability,
+    TextDocumentSyncKind, TextDocumentSyncOptions, TextDocumentSyncSaveOptions,
+    WorkDoneProgressOptions,
 };
 
+use crate::handlers::code_actions::command_id;
+
 /// The capabilities to send from server as part of [`InitializeResult`]
 ///
 /// [`InitializeResult`]: lspower::lsp::InitializeResult
@@ -46,10 +51,19 @@ pub(crate) fn server_capabilities(capabilities: &ClientCapabilities) -> ServerCa
                 work_done_progress: None,
             },
         }),
+        execute_command_provider: Some(ExecuteCommandOptions {
+            commands: CommandActionCategory::iter()
+                .map(|c| command_id(&c))
+                .collect::<Vec<String>>(),
+
+            ..Default::default()
+        }),
         document_formatting_provider: None,
         document_range_formatting_provider: None,
         document_on_type_formatting_provider: None,
-        code_action_provider: None,
+        code_action_provider: Some(tower_lsp::lsp_types::CodeActionProviderCapability::Simple(
+            true,
+        )),
         rename_provider: None,
         ..Default::default()
     }

crates/pgt_commands/src/execute_statement.rs

@@ -1,2 +1,4 @@
+pub(crate) mod code_actions;
 pub(crate) mod completions;
+mod helper;
 pub(crate) mod text_document;