Support use of named host variables

Signed-off-by: worksofliam <mrliamallan@live.co.uk>

Author: worksofliam

src/language/providers/actionProvider.ts

@@ -0,0 +1,60 @@
+import { CodeAction, CodeActionKind, languages, TextDocument, Uri, WorkspaceEdit } from "vscode";
+import { remoteAssistIsEnabled } from "./logic/available";
+import { getSqlDocument } from "./logic/parse";
+
+class SqlCodeAction extends CodeAction {
+  constructor(title: string, kind: CodeActionKind, public file: {document: TextDocument, statementOffset: number, bindCount: number}) {
+    super(title, kind);
+  }
+}
+
+const invalidBindingLabels = [`bind`, `cl`];
+
+export const actionProvider = languages.registerCodeActionsProvider({ language: `sql` }, {
+  provideCodeActions(document, range, context, token) {
+    if (range.isEmpty) {
+      const offset = document.offsetAt(range.start);
+
+      const enabled = remoteAssistIsEnabled();
+      if (!enabled) return;
+
+      const sqlDoc = getSqlDocument(document);
+      if (!sqlDoc) return;
+
+      const currentStatement = sqlDoc.getStatementByOffset(offset);
+      const label = currentStatement.getLabel()?.toLowerCase() || ``;
+
+      if (currentStatement && !invalidBindingLabels.includes(label)) {
+        const markers = currentStatement.getEmbeddedStatementAreas().filter(a => a.type === `marker`);
+        const codeActions: SqlCodeAction[] = [];
+
+        if (markers.length > 0) {
+          const action = new SqlCodeAction(`Generate bind statement`, CodeActionKind.QuickFix, {
+            document,
+            statementOffset: currentStatement.range.end,
+            bindCount: markers.length
+          });
+
+          codeActions.push(action);
+
+          return codeActions;
+        }
+      }
+
+      return [];
+    }
+  },
+  resolveCodeAction(codeAction: SqlCodeAction, token) {
+    if (!(codeAction instanceof SqlCodeAction)) return codeAction;
+    codeAction.edit = new WorkspaceEdit();
+    const document = codeAction.file.document;
+
+    const endOfStatementPos = document.positionAt(codeAction.file.statementOffset);
+    const lineOfStatement = document.lineAt(endOfStatementPos.line);
+
+    let statement = `bind: ${new Array(codeAction.file.bindCount).fill(`v`).join(`, `)}`;
+
+    codeAction.edit.insert(document.uri, lineOfStatement.range.end, `\n${statement};`);
+    return codeAction;
+  }
+});

src/language/providers/index.ts

@@ -1,3 +1,4 @@
+import { actionProvider } from "./actionProvider";
 import { completionProvider } from "./completionProvider";
 import { formatProvider } from "./formatProvider";
 import { hoverProvider, openProvider } from "./hoverProvider";
@@ -20,7 +21,8 @@ export function languageInit() {
     // peekProvider,
     ...problemProvider,
     checkDocumentDefintion,
-    sqlLanguageStatus
+    sqlLanguageStatus,
+    actionProvider
   );
 
   return functionality;

src/language/sql/document.ts

@@ -223,7 +223,7 @@ export default class Document {
     })
   }
 
-  removeEmbeddedAreas(statement: Statement, snippetString?: boolean): ParsedEmbeddedStatement {
+  removeEmbeddedAreas(statement: Statement, replacement: `snippet`|`?`): ParsedEmbeddedStatement {
     const areas = statement.getEmbeddedStatementAreas();
 
     const totalParameters = areas.filter(a => a.type === `marker`).length;
@@ -242,7 +242,14 @@ export default class Document {
         case `marker`:
           const markerContent = newContent.substring(start, end);
 
-          newContent = newContent.substring(0, start) + (snippetString ? `\${${totalParameters-parameterCount}:${markerContent}}` : `?`) + newContent.substring(end) + (snippetString ? `$0` : ``);
+          switch (replacement) {
+            case `snippet`:
+              newContent = newContent.substring(0, start) + `\${${totalParameters-parameterCount}:${markerContent}}` + newContent.substring(end) + `$0`;
+              break;
+            case `?`:
+              newContent = newContent.substring(0, start) + `?` + newContent.substring(end);
+              break;
+          }
 
           parameterCount++;
           break;

src/views/results/binding.ts

@@ -18,10 +18,10 @@ export function getPriorBindableStatement(editor: TextEditor, offset: number): {
     if (group.statements.length === 1) {
       const statement = group.statements[0];
       if (!statement.getLabel()) {
-        const markers = statement.getEmbeddedStatementAreas().filter(a => a.type === `marker`);
+        const newStatement = sqlDocument.removeEmbeddedAreas(statement, `?`);
         return {
-          statement: sqlDocument.content.substring(group.range.start, group.range.end),
-          parameters: markers.length
+          statement: newStatement.content,
+          parameters: newStatement.parameterCount
         };
       }
     }
@@ -40,7 +40,7 @@ export function getLiteralsFromStatement(group: StatementGroup): SqlParameter[]
         literals.push(token.value.substring(1, token.value.length - 1)); // Remove quotes
       } else if (token.type === `number`) {
         // Handle decimal numbers
-        if (tokenIs(tokens[i+1], `number`) && tokenIs(tokens[i+2], `number`)) {
+        if (tokenIs(tokens[i+1], `dot`) && tokenIs(tokens[i+2], `number`)) {
           literals.push(Number(`${token.value}.${tokens[i+2].value}`));
           i += 2; // Skip the next two tokens as they are part of the decimal number
         } else {

src/views/results/index.ts

@@ -600,7 +600,7 @@ export function parseStatement(editor?: vscode.TextEditor, existingInfo?: Statem
 
   if (sqlDocument) {
     if (statementInfo.qualifier !== `cl`) {
-      statementInfo.embeddedInfo = sqlDocument.removeEmbeddedAreas(statementInfo.statement, true);
+      statementInfo.embeddedInfo = sqlDocument.removeEmbeddedAreas(statementInfo.statement, `snippet`);
     }
   }