fix: minor fixes

Author: psteinroe

.env

@@ -1 +1 @@
-DATABASE_URL=postgresql://postgres:[email protected]:5432/postgres
+DATABASE_URL=postgresql://postgres:[email protected]:54322/postgres

crates/pg_query_ext/src/lib.rs

@@ -26,8 +26,7 @@ pub fn parse(sql: &str) -> Result<NodeEnum> {
             .nodes()
             .iter()
             .find(|n| n.1 == 1)
-            .unwrap()
-            .0
-            .to_enum()
-    })
+            .map(|n| n.0.to_enum())
+            .ok_or_else(|| Error::Parse("Unable to find root node".to_string()))
+    })?
 }

crates/pg_workspace_new/src/workspace/server/pg_query.rs

@@ -41,7 +41,7 @@ impl PgQueryStore {
 
     pub fn modify_statement(&self, change: &ModifiedStatement) {
         self.remove_statement(&change.old_stmt);
-        self.add_statement(&change.new_stmt, &change.change_text);
+        self.add_statement(&change.new_stmt, &change.new_stmt_text);
     }
 
     pub fn get_diagnostics(&self, stmt: &Statement) -> Vec<pg_diagnostics::serde::Diagnostic> {