init

Author: juleswritescode

.cargo/config.toml

@@ -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.lock

@@ -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_configuration/src/database.rs

@@ -28,6 +28,7 @@ pgt_schema_cache          = { workspace = true }
 pgt_statement_splitter    = { workspace = true }
 pgt_text_size.workspace   = true
 pgt_typecheck             = { workspace = true }
+pgt_commands              = { workspace = true }
 rustc-hash                = { workspace = true }
 schemars                  = { workspace = true, optional = true }
 serde                     = { workspace = true, features = ["derive"] }

crates/pgt_configuration/src/lib.rs

@@ -0,0 +1,42 @@
+use pgt_configuration::RuleSelector;
+use pgt_fs::PgTPath;
+use pgt_text_size::TextRange;
+
+use crate::Workspace;
+
+#[derive(Debug, serde::Serialize, serde::Deserialize)]
+#[cfg_attr(feature = "schema", derive(schemars::JsonSchema))]
+pub struct CodeActionsParams {
+    pub path: PgTPath,
+    pub range: Option<TextRange>,
+    pub only: Vec<RuleSelector>,
+    pub skip: Vec<RuleSelector>,
+}
+
+#[derive(Debug, serde::Serialize, serde::Deserialize)]
+#[cfg_attr(feature = "schema", derive(schemars::JsonSchema))]
+pub struct CodeActionsResult {
+    pub actions: Vec<CodeAction>,
+}
+
+#[derive(Debug, serde::Serialize, serde::Deserialize)]
+#[cfg_attr(feature = "schema", derive(schemars::JsonSchema))]
+#[serde(rename_all = "camelCase")]
+pub struct CodeAction {
+    pub category: CodeActionCategory,
+}
+
+#[derive(Debug, serde::Serialize, serde::Deserialize)]
+#[cfg_attr(feature = "schema", derive(schemars::JsonSchema))]
+pub enum CodeActionCategory {
+    ExecuteCommand(String),
+}
+
+impl CodeActionCategory {
+    fn perform(self, workspace: &dyn Workspace) -> Result<(), String> {
+        match self {
+            Self::ExecuteCommand(stmt) => {}
+        }
+        Ok(())
+    }
+}

crates/pgt_workspace/Cargo.toml

@@ -3,6 +3,7 @@ use std::ops::{Deref, DerefMut};
 use pgt_console::Console;
 use pgt_fs::{FileSystem, OsFileSystem};
 
+mod code_actions;
 pub mod configuration;
 pub mod diagnostics;
 pub mod dome;

crates/pgt_workspace/src/code_actions.rs

@@ -1,6 +1,7 @@
 use std::{panic::RefUnwindSafe, path::PathBuf, sync::Arc};
 
 pub use self::client::{TransportRequest, WorkspaceClient, WorkspaceTransport};
+use crate::code_actions::*;
 use pgt_analyse::RuleCategories;
 use pgt_configuration::{PartialConfiguration, RuleSelector};
 use pgt_fs::PgTPath;
@@ -115,6 +116,12 @@ pub trait Workspace: Send + Sync + RefUnwindSafe {
         params: PullDiagnosticsParams,
     ) -> Result<PullDiagnosticsResult, WorkspaceError>;
 
+    /// Retrieves a list of available code_actions for a file/cursor_position
+    fn pull_code_actions(
+        &self,
+        params: CodeActionsParams,
+    ) -> Result<CodeActionsResult, WorkspaceError>;
+
     fn get_completions(
         &self,
         params: GetCompletionsParams,

crates/pgt_workspace/src/lib.rs

@@ -89,6 +89,13 @@ impl<T> Workspace for WorkspaceClient<T>
 where
     T: WorkspaceTransport + RefUnwindSafe + Send + Sync,
 {
+    fn pull_code_actions(
+        &self,
+        params: crate::code_actions::CodeActionsParams,
+    ) -> Result<crate::code_actions::CodeActionsResult, WorkspaceError> {
+        self.request("pgt/code_actions", params)
+    }
+
     fn open_file(&self, params: OpenFileParams) -> Result<(), WorkspaceError> {
         self.request("pgt/open_file", params)
     }

crates/pgt_workspace/src/workspace.rs

@@ -19,6 +19,7 @@ use tree_sitter::TreeSitterStore;
 
 use crate::{
     WorkspaceError,
+    code_actions::{CodeAction, CodeActionCategory, CodeActionsResult},
     configuration::to_analyser_rules,
     settings::{Settings, SettingsHandle, SettingsHandleMut},
     workspace::PullDiagnosticsResult,
@@ -253,6 +254,35 @@ impl Workspace for WorkspaceServer {
         Ok(self.is_ignored(params.pgt_path.as_path()))
     }
 
+    fn pull_code_actions(
+        &self,
+        params: crate::code_actions::CodeActionsParams,
+    ) -> Result<crate::code_actions::CodeActionsResult, WorkspaceError> {
+        let cursor_position = params.range;
+
+        let doc = self
+            .documents
+            .get(&params.path)
+            .ok_or(WorkspaceError::not_found())?;
+
+        let eligible_statements = doc.iter_statements_with_text_and_range().filter(
+            |(_, range, _)| match cursor_position {
+                None => true,
+                Some(r) => range.contains(r),
+            },
+        );
+
+        let mut actions: Vec<CodeAction> = vec![];
+
+        for (stmt, range, txt) in eligible_statements {
+            actions.push(CodeAction {
+                category: CodeActionCategory::ExecuteCommand(txt.into()),
+            });
+        }
+
+        Ok(CodeActionsResult { actions })
+    }
+
     fn pull_diagnostics(
         &self,
         params: super::PullDiagnosticsParams,

crates/pgt_workspace/src/workspace/client.rs

@@ -38,7 +38,7 @@ postgrestools init
 
 After running the `init` command, you’ll have a `postgrestools.jsonc` file in your directory:
 
-[//]: # (BEGIN DEFAULT_CONFIGURATION)
+[//]: # "BEGIN DEFAULT_CONFIGURATION"
 
 ```json
 {
@@ -63,12 +63,13 @@ After running the `init` command, you’ll have a `postgrestools.jsonc` file in
     "username": "postgres",
     "password": "postgres",
     "database": "postgres",
-    "connTimeoutSecs": 10
+    "connTimeoutSecs": 10,
+    "allowStatementExecutionsAgainst": ["127.0.0.1/*", "localhost/*"]
   }
 }
 ```
 
-[//]: # (END DEFAULT_CONFIGURATION)
+[//]: # "END DEFAULT_CONFIGURATION"
 
 Make sure to point the database connection settings at your local development database. To see what else can be configured, run `--help`.
 
@@ -91,9 +92,7 @@ We recommend installing an editor plugin to get the most out of Postgres Languag
 > [!NOTE]
 > We will update this section once we have published the binaries.
 
-
 ## CI Setup
 
 > [!NOTE]
 > We will update this section once we have published the binaries.
-