Added new Select option

Author: pacoweb

extension.js

@@ -1,7 +1,8 @@
 const vscode = require('vscode');
-const azdata = require('azdata');
+
 const {getSqlScriptAsInsertAsync} = require('./scriptInsertAs.js');
 const {getSqlScriptAsUpdateAsync} = require('./scriptUpdateAs.js');
+const {getSqlScriptAsSelectAsync} = require('./scriptSelectAs.js');
 const sqlUtils = require('./scriptSqlUtils.js');
 
 function activate(context) 
@@ -137,6 +138,56 @@ function activate(context)
         }
     );
 
+    let selectTableCommandToClipBoard = vscode.commands.registerCommand("extraSqlScriptAs.selectTableToClipboard"
+    , function(context) 
+        {
+            let databaseName = context.connectionProfile.databaseName;
+            let schemaName = context.nodeInfo.metadata.schema;
+            let tableName = context.nodeInfo.metadata.name;
+
+            getSqlScriptAsSelectAsync(context.connectionProfile, databaseName, schemaName, tableName)
+                .then(scriptText => 
+                {
+                    vscode.env.clipboard.writeText(scriptText).then((text)=>{
+                        vscode.window.showInformationMessage('Script copied to clipboard.');
+                    });
+                })
+                .catch(reason => 
+                        {
+                            vscode.window.showErrorMessage(reason);
+                        }
+                );      
+        }
+    );
+   
+    let selectTableCommand = vscode.commands.registerCommand("extraSqlScriptAs.selectTable"
+        , function(context) 
+        {
+            let databaseName = context.connectionProfile.databaseName;
+            let schemaName = context.nodeInfo.metadata.schema;
+            let tableName = context.nodeInfo.metadata.name;
+            
+            //Test
+            getSqlScriptAsSelectAsync(context.connectionProfile, databaseName, schemaName, tableName)
+                .then(scriptText => 
+                {
+                    vscode.commands.executeCommand('newQuery').then(s => {
+                        
+                        let editor = vscode.window.activeTextEditor;
+
+                        editor.edit(edit => {
+                            edit.insert(new vscode.Position(0, 0), scriptText);
+                        });
+                    });
+                })
+                .catch(reason => 
+                    {
+                        vscode.window.showErrorMessage(reason);
+                    }
+            );      
+        }
+    );
+
     context.subscriptions.push(insertTableCommand);
     context.subscriptions.push(insertTableCommandToClipBoard);
 
@@ -145,6 +196,9 @@ function activate(context)
 
     context.subscriptions.push(deleteTableCommand);
     context.subscriptions.push(deleteTableCommandToClipBoard);
+
+    context.subscriptions.push(selectTableCommand);
+    context.subscriptions.push(selectTableCommandToClipBoard);
 };
 
 function deactivate() {

package.json

@@ -3,7 +3,7 @@
 	"displayName": "Extra Sql Script As",
 	"description": "This extension adds several missing options to the context menu of the object tree: Script Table as INSERT, Script Table as UPDATE...",
 	"publisher": "pacoweb",
-	"version": "0.4.1",
+	"version": "0.5.0",
 	"license": "https://raw.githubusercontent.com/pacoweb/extraSqlScriptAs/master/LICENSE",
 	"icon": "images/default_icon.png",
 	"repository": {
@@ -21,9 +21,11 @@
 		"onCommand:extraSqlScriptAs.insertTable",
 		"onCommand:extraSqlScriptAs.updateTable",
 		"onCommand:extraSqlScriptAs.deleteTable",
+		"onCommand:extraSqlScriptAs.selectTable",
 		"onCommand:extraSqlScriptAs.insertTableToClipboard",
 		"onCommand:extraSqlScriptAs.updateTableToClipboard",
-		"onCommand:extraSqlScriptAs.deleteTableToClipboard"
+		"onCommand:extraSqlScriptAs.deleteTableToClipboard",
+		"onCommand:extraSqlScriptAs.selectTableToClipboard"
 	],
 	"main": "./extension.js",
 	"contributes": {
@@ -40,6 +42,10 @@
 				"command": "extraSqlScriptAs.deleteTable",
 				"title": "Script Table as DELETE"
 			},
+			{
+				"command": "extraSqlScriptAs.selectTable",
+				"title": "Script Table as SELECT"
+			},
 			{
 				"command": "extraSqlScriptAs.insertTableToClipboard",
 				"title": "Script Table as INSERT to clipboard"
@@ -51,6 +57,10 @@
 			{
 				"command": "extraSqlScriptAs.deleteTableToClipboard",
 				"title": "Script Table as DELETE to clipboard"
+			},
+			{
+				"command": "extraSqlScriptAs.selectTableToClipboard",
+				"title": "Script Table as SELECT to clipboard"
 			}
 		],
 		"menus": {
@@ -70,6 +80,11 @@
 					"when": "connectionProvider == MSSQL && nodeType && nodeType == Table",
 					"group": "1data1"
 				},
+				{
+					"command": "extraSqlScriptAs.selectTable",
+					"when": "connectionProvider == MSSQL && nodeType && nodeType == Table",
+					"group": "1data1"
+				},
 				{
 					"command": "extraSqlScriptAs.insertTableToClipboard",
 					"when": "connectionProvider == MSSQL && nodeType && nodeType == Table",
@@ -84,6 +99,11 @@
 					"command": "extraSqlScriptAs.deleteTableToClipboard",
 					"when": "connectionProvider == MSSQL && nodeType && nodeType == Table",
 					"group": "1data1"
+				},
+				{
+					"command": "extraSqlScriptAs.selectTableToClipboard",
+					"when": "connectionProvider == MSSQL && nodeType && nodeType == Table",
+					"group": "1data1"
 				}
 			]
 		}

scriptSelectAs.js

@@ -0,0 +1,45 @@
+'use strict';
+
+const sqlUtils = require('./scriptSqlUtils.js');
+
+const colNameOrdinal = 0;
+
+async function getSqlScriptAsSelectAsync(connectionProfile, tableCatalog, tableSchema, tableName) 
+{
+    let queryText = sqlUtils.getColumnInfoQuerySql(tableCatalog, tableSchema, tableName);
+
+    let results = await sqlUtils.getResultsFromQuerySql(connectionProfile, "MSSQL", queryText);
+
+    if (!results || results.rowCount === 0) {
+        throw "No se han obtenido resultados de la consulta";
+    }
+
+    let updateSqlScript = buildFinalScript(results, tableCatalog, tableSchema, tableName);
+
+    return updateSqlScript;
+}
+
+function buildFinalScript(results, tableCatalog, tableSchema, tableName)
+{
+    let fullScript = [];
+    let columsScriptPart = [];
+
+    fullScript.push("SELECT ");
+
+    let columnIndex = 0;
+
+    for (let i= 0; i !== results.rowCount; i++) 
+    {
+        let rowData = results.rows[i];
+
+        const separator = (columnIndex === 0) ? " " : ",";
+        
+        columsScriptPart.push("\t\t" + separator + "[" + rowData[colNameOrdinal].displayValue + "]");
+
+        columnIndex += 1;
+    }
+
+    return fullScript.concat(columsScriptPart).concat([`FROM [${tableCatalog}].[${tableSchema}].[${tableName}] `]).join('\n');
+}
+
+module.exports.getSqlScriptAsSelectAsync = getSqlScriptAsSelectAsync;

scriptUpdateAs.js

@@ -1,6 +1,5 @@
 'use strict';
 
-const vscode = require('vscode');
 const sqlUtils = require('./scriptSqlUtils.js');
 
 const colNameOrdinal = 0;