AnalyzerForm.cs

Go to the documentation of this file.

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Drawing;
using System.Linq;
using System.Text;
using System.Windows.Forms;
using Microsoft.Win32;
using MySql.Data.MySqlClient;

namespace RoutineAnalyzer
{
    public partial class AnalyzerForm : Form
    {
        public AnalyzerForm()
        {
            //required call
            InitializeComponent();

            //performed read from persistence
            try
            {
                //load the form values from the registry (using last params used)
                RegistryKey key = GetRegistryKey("Software\\MySQL Routine Analyzer");
                _login.Text = key.GetValue("Login").ToString();
                _password.Text = key.GetValue("Password").ToString();
                _schema.Text = key.GetValue("Schema").ToString();
            }
            catch
            {
                //if it fails, we don't really care
            }
        }

        private RegistryKey GetRegistryKey(string path)
        {
            //otherwise create the key and return it
            return Registry.CurrentUser.CreateSubKey(path);
        }

        private RoutineBase CreateRoutineDesc(string decl, bool isStreamed)
        {
            if (_cppRadioBtn.Checked)
            {
                if (isStreamed)
                    return new CppGetRoutine(decl);
                else
                    return new CppSetRoutine(decl);
            }
            else if (_uscriptRadioBtn.Checked)
            {
                if (isStreamed)
                    return new UScriptGetRoutine(decl);
                else
                    return new UScriptSetRoutine(decl);
            }
            else
            {
                if (isStreamed)
                    return new CppWrapperGetRoutine(decl);
                else
                    return new CppWrapperSetRoutine(decl);
            }
        }

        void FormatCode()
        {
            string[] lines = _code.Text.Split('\n');
            string tabs = "";
            if (_cppRadioBtn.Checked)
                tabs = "\t\t";
            int increaseForNextIter = 0;
            StringBuilder builder = new StringBuilder();
            foreach (string line in lines)
            {
                //when an open brace is found, we actually wait until
                //the next line to increase the tabs
                //when it's a closed brace, we do it immediately
                if (increaseForNextIter == 1)
                    tabs += "\t";

                increaseForNextIter = 0;
                if (line == "{")
                    increaseForNextIter = 1;
                else if (line == "}")
                    tabs = tabs.Remove(0, 1);

                builder.AppendLine(line.Insert(0, tabs));
            }

            _code.Text = builder.ToString();
        }

        private MySqlConnectionStringBuilder CreateBuilder()
        {
            //build the connection string to be returned
            MySqlConnectionStringBuilder builder = new MySqlConnectionStringBuilder();
            builder.Server = "localhost";
            builder.UserID = _login.Text;
            builder.Password = _password.Text;
            builder.Database = _schema.Text;

            ParseHelper.SqlBuilder = builder;

            try
            {
                //save the current values to the registry, so next time we start the app we can use them
                using (RegistryKey key = GetRegistryKey("Software\\MySQL Routine Analyzer"))
                {
                    key.SetValue("Login", _login.Text);
                    key.SetValue("Password", _password.Text);
                    key.SetValue("Schema", _schema.Text);
                    key.Flush();
                }
            }
            catch
            {
                //if this fails, we don't really care (it's not a critical feature)
            }

            return builder;
        }

        private void GenerateCode()
        {
            try
            {
                //create the connection criteria
                MySqlConnectionStringBuilder builder = CreateBuilder();
                //pass it off to the routine reader to retrieve the procedure code
                RoutineReader reader = new RoutineReader(builder);
                //get a list of the definitions for further parsing
                List<string> definitions = reader.GetProcedureDefintions();

                //each definition must be parsed and added to the textbox
                StringBuilder b = new StringBuilder();

                //build the code
                DataTable table = _sprocGrid.DataSource as DataTable;
                for (int i = 0; i < definitions.Count; ++i)
                {
                    bool isStreamed = false;
                    if (table != null)
                    {
                        string tableRowsToString = table.Rows[i][1].ToString();
                        isStreamed = tableRowsToString == "True";
                    }

                    //create a routine descriptor to pull apart the definition
                    RoutineBase desc = CreateRoutineDesc(definitions[i], isStreamed);
                    //forward engineer the C++ and add it
                    b.Append(desc.GetDeclaration());
                    b.AppendLine(";");
                }

                b.AppendLine();
                b.AppendLine();

                for (int i = 0; i < definitions.Count; ++i)
                {
                    bool isStreamed = false;
                    if (table != null)
                    {
                        string tableRowsToString = table.Rows[i][1].ToString();
                        isStreamed = tableRowsToString == "True";
                    }

                    //create a routine descriptor to pull apart the definition
                    RoutineBase desc = CreateRoutineDesc(definitions[i], isStreamed);
                    //forward engineer the C++ and add it
                    b.AppendLine(desc.GetCode());
                    b.AppendLine();
                }

                _code.Text = b.ToString();

                //format the code for tab spacing
                FormatCode();
            }
            catch (Exception ex)
            {
                MessageBox.Show("Error during analysis: " + ex.Message);
            }
        }

        #region Event Handlers

        private void OnAnalyzeBtnClick(object sender, EventArgs e)
        {
            GenerateCode();
        }

        private void OnScanDatabaseClick(object sender, EventArgs e)
        {
            MySqlConnectionStringBuilder builder = CreateBuilder();
            //pass it off to the routine reader to retrieve the procedure code
            RoutineReader reader = new RoutineReader(builder);
            DataTable table = reader.GetProcedureDefinitionAsDataTable();
            table.Columns.Add("isStream");
            foreach (DataRow row in table.Rows)
            {
                string sprocName = row[0].ToString();
                row[1] = sprocName.Contains("get");
            }
            _sprocGrid.DataSource = table;
        }

        #endregion
    }
}