using System;

using RGiesecke.DllExport;

using System.Runtime.InteropServices;

using System.Windows.Forms;

using System.Drawing;

using System.Reflection;

using System.IO;

using System.Collections.Generic;

namespace utPLSQL

{

//*FUNC: 11*/ BOOL (*IDE_Connected)();

internal delegate bool IdeConnected();

//*FUNC: 12*/ void (*IDE_GetConnectionInfo)(char **Username, char **Password, char **Database);

internal delegate void IdeGetConnectionInfo(out IntPtr username, out IntPtr password, out IntPtr database);

//*FUNC: 20*/ void (*IDE_CreateWindow)(int WindowType, char *Text, BOOL Execute);

internal delegate void IdeCreateWindow(int windowType, string text, bool execute);

//*FUNC: 69*/ void *(*IDE_CreatePopupItem)(int ID, int Index, char *Name, char *ObjectType);

internal delegate void IdeCreatePopupItem(int id, int index, string name, string objectType);

//*FUNC: 74*/ int (*IDE_GetPopupObject)(char **ObjectType, char **ObjectOwner, char **ObjectName, char **SubObject);

internal delegate int IdeGetPopupObject(out IntPtr objectType, out IntPtr objectOwner, out IntPtr objectName, out IntPtr subObject);

//*FUNC: 79*/ char *(*IDE_GetObjectSource)(char *ObjectType, char *ObjectOwner, char *ObjectName);

internal delegate IntPtr IdeGetObjectSource(string objectType, string objectOwner, string objectName);

//*FUNC: 150*/ void (*IDE_CreateToolButton)(int ID, int Index, char *Name, char *BitmapFile, int BitmapHandle);

internal delegate void IdeCreateToolButton(int id, int index, string name, string bitmapFile, long bitmapHandle);

public class PlsqlDeveloperUtPlsqlPlugin

{

private const string PLUGIN_NAME = "utPLSQL Plugin";

private const int PLUGIN_MENU_INDEX_ALLTESTS = 3;

private const int PLUGIN_MENU_INDEX_ALLTESTS_WITH_COVERAGE = 4;

private const int PLUGIN_POPUP_INDEX = 1;

private const int PLUGIN_POPUP_INDEX_WITH_COVERAGE = 2;

private static PlsqlDeveloperUtPlsqlPlugin plugin;

internal static IdeConnected connected;

internal static IdeGetConnectionInfo getConnectionInfo;

internal static IdeCreateWindow createWindow;

internal static IdeCreatePopupItem createPopupItem;

internal static IdeGetPopupObject getPopupObject;

internal static IdeGetObjectSource getObjectSource;

internal static IdeCreateToolButton createToolButton;

internal static int pluginId;

internal static string username;

internal static string password;

internal static string database;

private static RealTimeTestRunner testRunner;

private static readonly List<TestRunnerWindow> windows = new List<TestRunnerWindow>();

private PlsqlDeveloperUtPlsqlPlugin()

{

testRunner = new RealTimeTestRunner();

}

#region DLL exported API

[DllExport("IdentifyPlugIn", CallingConvention = CallingConvention.Cdecl)]

public static string IdentifyPlugIn(int id)

{

if (plugin == null)

{

plugin = new PlsqlDeveloperUtPlsqlPlugin();

pluginId = id;

}

return PLUGIN_NAME;

}

[DllExport("OnActivate", CallingConvention = CallingConvention.Cdecl)]

public static void OnActivate()

{

try

{

PlsqlDeveloperUtPlsqlPlugin.ConnectToDatabase();

// Seperate streams are needed!

var assembly = Assembly.GetExecutingAssembly();

using (Stream stream = assembly.GetManifestResourceStream("utPLSQL.utPLSQL.bmp"))

{

PlsqlDeveloperUtPlsqlPlugin.createToolButton(pluginId, PLUGIN_MENU_INDEX_ALLTESTS, "utPLSQL", "utPLSQL.bmp", new Bitmap(stream).GetHbitmap().ToInt64());

}

using (Stream stream = assembly.GetManifestResourceStream("utPLSQL.utPLSQL_coverage.bmp"))

{

PlsqlDeveloperUtPlsqlPlugin.createToolButton(pluginId, PLUGIN_MENU_INDEX_ALLTESTS_WITH_COVERAGE, "utPLSQL", "utPLSQL_coverage.bmp", new Bitmap(stream).GetHbitmap().ToInt64());

}

using (Stream stream = assembly.GetManifestResourceStream("utPLSQL.utPLSQL.bmp"))

{

PlsqlDeveloperUtPlsqlPlugin.createToolButton(pluginId, PLUGIN_POPUP_INDEX, "utPLSQL", "utPLSQL.bmp", new Bitmap(stream).GetHbitmap().ToInt64());

}

using (Stream stream = assembly.GetManifestResourceStream("utPLSQL.utPLSQL_coverage.bmp"))

{

PlsqlDeveloperUtPlsqlPlugin.createToolButton(pluginId, PLUGIN_POPUP_INDEX_WITH_COVERAGE, "utPLSQL", "utPLSQL_coverage.bmp", new Bitmap(stream).GetHbitmap().ToInt64());

}

}

catch (Exception e)

{

MessageBox.Show(e.Message);

}

PlsqlDeveloperUtPlsqlPlugin.createPopupItem(pluginId, PLUGIN_POPUP_INDEX, "Run utPLSQL Test", "USER");

PlsqlDeveloperUtPlsqlPlugin.createPopupItem(pluginId, PLUGIN_POPUP_INDEX_WITH_COVERAGE, "Run Code Coverage", "USER");

PlsqlDeveloperUtPlsqlPlugin.createPopupItem(pluginId, PLUGIN_POPUP_INDEX, "Run utPLSQL Test", "PACKAGE");

PlsqlDeveloperUtPlsqlPlugin.createPopupItem(pluginId, PLUGIN_POPUP_INDEX_WITH_COVERAGE, "Run Code Coverage", "PACKAGE");

}

[DllExport("CanClose", CallingConvention = CallingConvention.Cdecl)]

public static bool CanClose()

{

foreach (TestRunnerWindow window in windows)

{

if (window.Running)

{

var confirmResult = MessageBox.Show("utPLSQL Tests are still running.\r\n\r\nDo you really want to close PL/SQL Developer?", "Running utPLSQL Tests", MessageBoxButtons.YesNo, MessageBoxIcon.Warning);

if (confirmResult == DialogResult.Yes)

{

return true;

}

else

{

return false;

}

}

}

return true;

}

[DllExport("RegisterCallback", CallingConvention = CallingConvention.Cdecl)]

public static void RegisterCallback(int index, IntPtr function)

{

switch (index)

{

case 11:

PlsqlDeveloperUtPlsqlPlugin.connected = (IdeConnected)Marshal.GetDelegateForFunctionPointer(function, typeof(IdeConnected));

break;

case 12:

PlsqlDeveloperUtPlsqlPlugin.getConnectionInfo = (IdeGetConnectionInfo)Marshal.GetDelegateForFunctionPointer(function, typeof(IdeGetConnectionInfo));

break;

case 20:

PlsqlDeveloperUtPlsqlPlugin.createWindow = (IdeCreateWindow)Marshal.GetDelegateForFunctionPointer(function, typeof(IdeCreateWindow));

break;

case 69:

PlsqlDeveloperUtPlsqlPlugin.createPopupItem = (IdeCreatePopupItem)Marshal.GetDelegateForFunctionPointer(function, typeof(IdeCreatePopupItem));

break;

case 74:

PlsqlDeveloperUtPlsqlPlugin.getPopupObject = (IdeGetPopupObject)Marshal.GetDelegateForFunctionPointer(function, typeof(IdeGetPopupObject));

break;

case 79:

PlsqlDeveloperUtPlsqlPlugin.getObjectSource = (IdeGetObjectSource)Marshal.GetDelegateForFunctionPointer(function, typeof(IdeGetObjectSource));

break;

case 150:

PlsqlDeveloperUtPlsqlPlugin.createToolButton = (IdeCreateToolButton)Marshal.GetDelegateForFunctionPointer(function, typeof(IdeCreateToolButton));

break;

}

}

[DllExport("OnConnectionChange", CallingConvention = CallingConvention.Cdecl)]

public static void OnConnectionChange()

{

PlsqlDeveloperUtPlsqlPlugin.ConnectToDatabase();

}

[DllExport("CreateMenuItem", CallingConvention = CallingConvention.Cdecl)]

public static string CreateMenuItem(int index)

{

switch (index)

{

case 1:

return "TAB=Tools";

case 2:

return "GROUP=utPLSQL";

case PLUGIN_MENU_INDEX_ALLTESTS:

return "LARGEITEM=Run all tests of current user";

case PLUGIN_MENU_INDEX_ALLTESTS_WITH_COVERAGE:

return "LARGEITEM=Run code coverage for current user";

default:

return "";

}

}

[DllExport("OnMenuClick", CallingConvention = CallingConvention.Cdecl)]

public static void OnMenuClick(int index)

{

if (index == PLUGIN_MENU_INDEX_ALLTESTS)

{

if (PlsqlDeveloperUtPlsqlPlugin.connected())

{

var testResultWindow = new TestRunnerWindow(testRunner);

windows.Add(testResultWindow);

testResultWindow.RunTestsAsync("_ALL", username, null, null, false);

}

}

else if (index == PLUGIN_MENU_INDEX_ALLTESTS_WITH_COVERAGE)

{

if (PlsqlDeveloperUtPlsqlPlugin.connected())

{

var testResultWindow = new TestRunnerWindow(testRunner);

windows.Add(testResultWindow);

testResultWindow.RunTestsAsync("_ALL", username, null, null, true);

}

}

else if (index == PLUGIN_POPUP_INDEX)

{

if (PlsqlDeveloperUtPlsqlPlugin.connected())

{

IntPtr type;

IntPtr owner;

IntPtr name;

IntPtr subType;

PlsqlDeveloperUtPlsqlPlugin.getPopupObject(out type, out owner, out name, out subType);

var testResultWindow = new TestRunnerWindow(testRunner);

windows.Add(testResultWindow);

testResultWindow.RunTestsAsync(Marshal.PtrToStringAnsi(type), Marshal.PtrToStringAnsi(owner), Marshal.PtrToStringAnsi(name), Marshal.PtrToStringAnsi(subType), false);

}

}

else if (index == PLUGIN_POPUP_INDEX_WITH_COVERAGE)

{

if (PlsqlDeveloperUtPlsqlPlugin.connected())

{

IntPtr type;

IntPtr owner;

IntPtr name;

IntPtr subType;

PlsqlDeveloperUtPlsqlPlugin.getPopupObject(out type, out owner, out name, out subType);

var testResultWindow = new TestRunnerWindow(testRunner);

windows.Add(testResultWindow);

testResultWindow.RunTestsAsync(Marshal.PtrToStringAnsi(type), Marshal.PtrToStringAnsi(owner), Marshal.PtrToStringAnsi(name), Marshal.PtrToStringAnsi(subType), true);

}

}

}

[DllExport("About", CallingConvention = CallingConvention.Cdecl)]

public static string About()

{

new AboutDialog().Show();

return "";

}

#endregion

internal static void OpenPackageBody(string owner, string name)

{

IntPtr source = getObjectSource("PACKAGE BODY", owner, name);

createWindow(3, Marshal.PtrToStringAnsi(source), false);

}

private static void ConnectToDatabase()

{

try

{

testRunner.Close();

if (PlsqlDeveloperUtPlsqlPlugin.connected())

{

IntPtr ptrUsername;

IntPtr ptrPassword;

IntPtr ptrDatabase;

PlsqlDeveloperUtPlsqlPlugin.getConnectionInfo(out ptrUsername, out ptrPassword, out ptrDatabase);

username = Marshal.PtrToStringAnsi(ptrUsername);

password = Marshal.PtrToStringAnsi(ptrPassword);

database = Marshal.PtrToStringAnsi(ptrDatabase);

testRunner.Connect(username, password, database);

}

}

catch (Exception e)

{

MessageBox.Show(e.Message);

}

}

}

}