frog/src/extension.ts

'use strict';
// The module 'vscode' contains the VS Code extensibility API
// Import the module and reference it with the alias vscode in your code below
import * as vscode from 'vscode';
import { ConnectionProvider, ConnectionNode } from './connections'
import { TextDocumentContentProvider } from './resultsetview'

const { spawn } = require('child_process');
const extensionName = "frog";

var dbSession = null;


// this method is called when your extension is activated
// your extension is activated the very first time the command is executed
export function activate(context: vscode.ExtensionContext) {
    let previewUri = vscode.Uri.parse('frog-view://authority/resultset');

    const connectionProvider = new ConnectionProvider();
    let myOutputChannel = vscode.window.createOutputChannel('SQL Query Results');
    context.subscriptions.push(myOutputChannel);

    let diagnosticCollection = vscode.languages.createDiagnosticCollection(extensionName);
    context.subscriptions.push(diagnosticCollection);

    vscode.window.registerTreeDataProvider('connections', connectionProvider);

    let provider = new TextDocumentContentProvider();
	let registration = vscode.workspace.registerTextDocumentContentProvider('frog-view', provider);

        // Use the console to output diagnostic information (console.log) and errors (console.error)
    // This line of code will only be executed once when your extension is activated
    console.log('Congratulations, your extension "frog" is now active!');

    let disposable = vscode.commands.registerCommand('extension.executeSnippet', () => {
        // The code you place here will be executed every time your command is executed

        
        let editor = vscode.window.activeTextEditor;

        if (!editor) return;

        let selection = editor.selection;

        if (selection.isEmpty) return;

        let text = editor.document.getText(selection);        

        if (!dbSession) {
            let extPath = vscode.extensions.getExtension("todie.frog").extensionPath;
            dbSession = spawn("java", ['-jar', extPath + '/frog-runner.jar', '172.19.23.18', '1521', 'aax2sm', 'aax2sm', 'aax2sm']);

            let outputBuffer = "";

            dbSession.stdout.on('data', (data) => {
                outputBuffer += data.toString();
                
                try {
                    let json = JSON.parse(outputBuffer);

                    console.log(json);

                    const diagnostics = [];
                    let cd = json['columnDefinitions']

                    if (!json['rows']) {
                        if (json['userErrors']) {
                            myOutputChannel.append("Errors found:\n");

                            const document = vscode.window.activeTextEditor.document;
                            
                            for (let error of json['userErrors'][0]['lines']) {
                                let range = document.lineAt(selection.start.line + error['line'] - 1).range;
                                let diagnostic = new vscode.Diagnostic(range, error['text'], vscode.DiagnosticSeverity.Error);
                                diagnostic.source = extensionName;
                                diagnostic.code = "code";
                                diagnostics.push(diagnostic);
                                myOutputChannel.append(error['line'] + ": " + error['text'] + "\n");
                            }

                            diagnosticCollection.set(document.uri, diagnostics);
                        } else if (json['exception']) {
                            myOutputChannel.append("Exception caught: " + json['exceptionMessage'] + "\n");
                        } else {
                            myOutputChannel.append("SQL executed successfully.\n");
                        }
                    } else {
                        provider.setData(json);
                        provider.update(previewUri);

                        vscode.commands.executeCommand('vscode.previewHtml', previewUri, vscode.ViewColumn.Two, 'Result Set View').then((success) => {
                        }, (reason) => {
                            vscode.window.showErrorMessage(reason);
                        });
                
                        myOutputChannel.append("Query successful.\n");
                    }

                    myOutputChannel.append("-- " + new Date() + " -------------------------------\n");
                    outputBuffer = "";
                } catch (e) {
                    console.log(e);
                }
            });
              
            dbSession.stderr.on('data', (data) => {
                myOutputChannel.show();
                myOutputChannel.append(data.toString());
            });

            dbSession.on('close', (code) => {
                console.log(`child process exited with code ${code}`);
                dbSession = null;
            });
        }

        if (!text) {
            vscode.window.showInformationMessage('No text selected');
            return;
        }

        
        myOutputChannel.append("-- " + new Date() + " -------------------------------\n"); //Executing: '" + text + "'\n");
        dbSession.stdin.write(text + "\n--- END ---\n");
    });

    context.subscriptions.push(disposable, registration);
}

// this method is called when your extension is deactivated
export function deactivate() {
    if (dbSession) dbSession.stdin.end();
}