unit connections;

// -------------------------------------

// Connections (start-window)

// -------------------------------------

interface

uses

Windows, SysUtils, Classes, Controls, Forms, Dialogs, StdCtrls, ExtCtrls, ComCtrls,

VirtualTrees, Menus, Graphics, Generics.Collections, ActiveX, extra_controls, Messages,

dbconnection, gnugettext, SynRegExpr, System.Types, Vcl.GraphUtil, ADODB, StrUtils,

System.Math, System.Actions, Vcl.ActnList, Vcl.StdActns;

type

Tconnform = class(TExtForm)

btnCancel: TButton;

btnOpen: TButton;

btnSave: TButton;

btnNew: TButton;

btnDelete: TButton;

popupSessions: TPopupMenu;

menuSave: TMenuItem;

menuDelete: TMenuItem;

menuSaveAs: TMenuItem;

TimerStatistics: TTimer;

PageControlDetails: TPageControl;

tabSettings: TTabSheet;

lblPort: TLabel;

lblPassword: TLabel;

lblHost: TLabel;

lblUsername: TLabel;

lblNetworkType: TLabel;

chkCompressed: TCheckBox;

editPort: TEdit;

updownPort: TUpDown;

editPassword: TEdit;

editUsername: TEdit;

editHost: TButtonedEdit;

tabAdvanced: TTabSheet;

tabStatistics: TTabSheet;

lblLastConnectLeft: TLabel;

lblCounterLeft: TLabel;

lblCreatedLeft: TLabel;

lblCreatedRight: TLabel;

lblCounterRight1: TLabel;

lblLastConnectRight: TLabel;

tabSSHtunnel: TTabSheet;

editSSHlocalport: TEdit;

editSSHUser: TEdit;

editSSHPassword: TEdit;

lblSSHLocalPort: TLabel;

lblSSHUser: TLabel;

lblSSHPassword: TLabel;

editSSHPlinkExe: TButtonedEdit;

lblSSHPlinkExe: TLabel;

comboNetType: TComboBoxEx;

lblSSHhost: TLabel;

editSSHhost: TEdit;

editSSHport: TEdit;

editSSHPrivateKey: TButtonedEdit;

lblSSHkeyfile: TLabel;

editDatabases: TButtonedEdit;

lblDatabase: TLabel;

chkLoginPrompt: TCheckBox;

lblPlinkTimeout: TLabel;

editSSHTimeout: TEdit;

updownSSHTimeout: TUpDown;

chkWindowsAuth: TCheckBox;

chkCleartextPluginEnabled: TCheckBox;

splitterMain: TSplitter;

tabStart: TTabSheet;

lblHelp: TLabel;

btnImportSettings: TButton;

timerSettingsImport: TTimer;

popupNew: TPopupMenu;

menuNewSessionInRoot: TMenuItem;

menuNewFolderInRoot: TMenuItem;

menuContextNewFolderInFolder: TMenuItem;

menuContextNewSessionInFolder: TMenuItem;

menuNewSessionInFolder: TMenuItem;

menuNewFolderInFolder: TMenuItem;

chkLocalTimeZone: TCheckBox;

editStartupScript: TButtonedEdit;

lblStartupScript: TLabel;

chkFullTableStatus: TCheckBox;

btnMore: TButton;

popupMore: TPopupMenu;

Checkforupdates1: TMenuItem;

About1: TMenuItem;

Preferences1: TMenuItem;

Exportsettingsfile1: TMenuItem;

Importsettingsfile1: TMenuItem;

lblComment: TLabel;

memoComment: TMemo;

lblQueryTimeout: TLabel;

editQueryTimeout: TEdit;

updownQueryTimeout: TUpDown;

menuMoreGeneralHelp: TMenuItem;

menuRename: TMenuItem;

lblKeepAlive: TLabel;

editKeepAlive: TEdit;

updownKeepAlive: TUpDown;

lblCounterRight2: TLabel;

lblCounterLeft2: TLabel;

TimerButtonAnimation: TTimer;

lblBackgroundColor: TLabel;

ColorBoxBackgroundColor: TColorBox;

comboLibrary: TComboBox;

lblLibrary: TLabel;

pnlLeft: TPanel;

ListSessions: TVirtualStringTree;

editSearch: TButtonedEdit;

popupHost: TPopupMenu;

menuFindDatabaseFiles: TMenuItem;

menuAddDatabaseFiles: TMenuItem;

lblIgnoreDatabasePattern: TLabel;

editIgnoreDatabasePattern: TEdit;

ActionListConnections: TActionList;

actFilter: TAction;

Filter1: TMenuItem;

chkLogFileDdl: TCheckBox;

editLogFilePath: TButtonedEdit;

tabSSL: TTabSheet;

chkWantSSL: TCheckBox;

lblSSLPrivateKey: TLabel;

lblSSLCACertificate: TLabel;

lblSSLCertificate: TLabel;

lblSSLcipher: TLabel;

editSSLcipher: TEdit;

editSSLCertificate: TButtonedEdit;

editSSLCACertificate: TButtonedEdit;

editSSLPrivateKey: TButtonedEdit;

lblLogFile: TLabel;

chkLogFileDml: TCheckBox;

procedure FormCreate(Sender: TObject);

procedure btnOpenClick(Sender: TObject);

procedure FormShow(Sender: TObject);

procedure btnSaveClick(Sender: TObject);

procedure btnSaveAsClick(Sender: TObject);

procedure btnNewClick(Sender: TObject);

procedure btnDeleteClick(Sender: TObject);

procedure Modification(Sender: TObject);

procedure ListSessionsGetText(Sender: TBaseVirtualTree; Node: PVirtualNode;

Column: TColumnIndex; TextType: TVSTTextType; var CellText: String);

procedure ListSessionsFocusChanged(Sender: TBaseVirtualTree;

Node: PVirtualNode; Column: TColumnIndex);

procedure ListSessionsGetImageIndex(Sender: TBaseVirtualTree;

Node: PVirtualNode; Kind: TVTImageKind; Column: TColumnIndex;

var Ghosted: Boolean; var ImageIndex: TImageIndex);

procedure ListSessionsNewText(Sender: TBaseVirtualTree; Node: PVirtualNode;

Column: TColumnIndex; NewText: String);

procedure ListSessionsFocusChanging(Sender: TBaseVirtualTree; OldNode,

NewNode: PVirtualNode; OldColumn, NewColumn: TColumnIndex;

var Allowed: Boolean);

procedure FormCloseQuery(Sender: TObject; var CanClose: Boolean);

procedure TimerStatisticsTimer(Sender: TObject);

procedure FormClose(Sender: TObject; var Action: TCloseAction);

procedure ListSessionsCreateEditor(Sender: TBaseVirtualTree; Node: PVirtualNode; Column: TColumnIndex;

out EditLink: IVTEditLink);

procedure PickFile(Sender: TObject);

procedure editSSHPlinkExeChange(Sender: TObject);

procedure editHostChange(Sender: TObject);

procedure editDatabasesRightButtonClick(Sender: TObject);

procedure chkLoginPromptClick(Sender: TObject);

procedure ListSessionsGetNodeDataSize(Sender: TBaseVirtualTree;

var NodeDataSize: Integer);

procedure comboNetTypeChange(Sender: TObject);

procedure splitterMainMoved(Sender: TObject);

procedure btnImportSettingsClick(Sender: TObject);

procedure timerSettingsImportTimer(Sender: TObject);

procedure ListSessionsStructureChange(Sender: TBaseVirtualTree;

Node: PVirtualNode; Reason: TChangeReason);

procedure ListSessionsDragOver(Sender: TBaseVirtualTree; Source: TObject;

Shift: TShiftState; State: TDragState; Pt: TPoint; Mode: TDropMode;

var Effect: Integer; var Accept: Boolean);

procedure ListSessionsDragDrop(Sender: TBaseVirtualTree; Source: TObject;

DataObject: IDataObject; Formats: TFormatArray; Shift: TShiftState;

Pt: TPoint; var Effect: Integer; Mode: TDropMode);

procedure btnMoreClick(Sender: TObject);

procedure menuRenameClick(Sender: TObject);

procedure TimerButtonAnimationTimer(Sender: TObject);

procedure ColorBoxBackgroundColorGetColors(Sender: TCustomColorBox;

Items: TStrings);

procedure editTrim(Sender: TObject);

procedure editSearchChange(Sender: TObject);

procedure editSearchRightButtonClick(Sender: TObject);

procedure editHostDblClick(Sender: TObject);

procedure ListSessionsNodeDblClick(Sender: TBaseVirtualTree;

const HitInfo: THitInfo);

procedure FindAddDatabaseFilesClick(Sender: TObject);

procedure FormResize(Sender: TObject);

procedure ListSessionsBeforeCellPaint(Sender: TBaseVirtualTree;

TargetCanvas: TCanvas; Node: PVirtualNode; Column: TColumnIndex;

CellPaintMode: TVTCellPaintMode; CellRect: TRect; var ContentRect: TRect);

procedure actFilterExecute(Sender: TObject);

private

{ Private declarations }

FLoaded: Boolean;

FSessionModified, FOnlyPasswordModified: Boolean;

FServerVersion: String;

FSettingsImportWaitTime: Cardinal;

FPopupDatabases: TPopupMenu;

FButtonAnimationStep: Integer;

FLastSelectedNetTypeGroup: TNetTypeGroup;

function GetSelectedNetType: TNetType;

procedure SetSelectedNetType(Value: TNetType);

procedure RefreshSessions(ParentNode: PVirtualNode);

function SelectedSessionPath: String;

function CurrentParams: TConnectionParameters;

procedure FinalizeModifications(var CanProceed: Boolean);

procedure ValidateControls;

function NodeSessionNames(Node: PVirtualNode; var RegKey: String): TStringList;

procedure MenuDatabasesClick(Sender: TObject);

procedure WMNCLBUTTONDOWN(var Msg: TWMNCLButtonDown) ; message WM_NCLBUTTONDOWN;

procedure WMNCLBUTTONUP(var Msg: TWMNCLButtonUp) ; message WM_NCLBUTTONUP;

procedure RefreshBackgroundColors;

property SelectedNetType: TNetType read GetSelectedNetType write SetSelectedNetType;

public

{ Public declarations }

end;

implementation

uses Main, apphelpers, grideditlinks;

{$I const.inc}

{$R *.DFM}

procedure Tconnform.WMNCLBUTTONDOWN(var Msg: TWMNCLButtonDown) ;

begin

if Msg.HitTest = HTHELP then

Msg.Result := 0 // "eat" the message

else

inherited;

end;

procedure Tconnform.WMNCLBUTTONUP(var Msg: TWMNCLButtonUp) ;

begin

if Msg.HitTest = HTHELP then begin

Msg.Result := 0;

Help(Self, 'connecting');

end else

inherited;

end;

procedure Tconnform.FormCreate(Sender: TObject);

var

NetTypeStr, FilenameHint: String;

nt: TNetType;

ntg: TNetTypeGroup;

Params: TConnectionParameters;

ComboItem: TComboExItem;

Placeholders: TStringList;

i: Integer;

begin

// Fix GUI stuff

HasSizeGrip := True;

Width := AppSettings.ReadInt(asSessionManagerWindowWidth);

Height := AppSettings.ReadInt(asSessionManagerWindowHeight);

Left := AppSettings.ReadInt(asSessionManagerWindowLeft, '', Left);

Top := AppSettings.ReadInt(asSessionManagerWindowTop, '', Top);

// Move to visible area if window was on a now plugged off monitor previously

MakeFullyVisible;

pnlLeft.Width := AppSettings.ReadInt(asSessionManagerListWidth);

splitterMain.OnMoved(Sender);

FixVT(ListSessions);

MainForm.RestoreListSetup(ListSessions);

ListSessions.OnCompareNodes := MainForm.AnyGridCompareNodes;

ListSessions.OnHeaderClick := MainForm.AnyGridHeaderClick;

ListSessions.OnHeaderDraggedOut := MainForm.AnyGridHeaderDraggedOut;

btnImportSettings.Caption := MainForm.actImportSettings.Caption;

FLoaded := False;

comboNetType.Clear;

Params := TConnectionParameters.Create;

for ntg := Low(ntg) to High(ntg) do begin

for nt:=Low(nt) to High(nt) do begin

Params.NetType := nt;

if Params.GetNetTypeGroup <> ntg then

Continue;

NetTypeStr := Params.NetTypeName(True);

ComboItem := TComboExItem.Create(comboNetType.ItemsEx);

ComboItem.Caption := NetTypeStr;

ComboItem.ImageIndex := Params.ImageIndex;

ComboItem.Data := Pointer(nt);

end;

end;

Params.Free;

// Create filename placeholders hint

Placeholders := GetOutputFilenamePlaceholders;

FilenameHint := _('Allows the following replacement patterns:');

for i:=0 to Placeholders.Count-1 do begin

FilenameHint := FilenameHint + CRLF + '%' + Placeholders.Names[i] + ': ' + Placeholders.ValueFromIndex[i];

end;

Placeholders.Free;

editLogFilePath.Hint := FilenameHint;

end;

procedure Tconnform.RefreshSessions(ParentNode: PVirtualNode);

var

SessionNames: TStringList;

RegKey: String;

i: Integer;

Params: TConnectionParameters;

SessNode: PVirtualNode;

begin

// Initialize session tree

// And while we're at it, collect custom colors for background color selector

if ParentNode=nil then begin

ListSessions.Clear;

end else begin

ListSessions.DeleteChildren(ParentNode, True);

end;

SessionNames := NodeSessionNames(ParentNode, RegKey);

for i:=0 to SessionNames.Count-1 do begin

Params := TConnectionParameters.Create(RegKey+SessionNames[i]);

SessNode := ListSessions.AddChild(ParentNode, PConnectionParameters(Params));

if Params.IsFolder then begin

RefreshSessions(SessNode);

end;

end;

if not Assigned(ParentNode) then

RefreshBackgroundColors;

end;

procedure Tconnform.FormResize(Sender: TObject);

begin

splitterMainMoved(splitterMain);

end;

procedure Tconnform.FormCloseQuery(Sender: TObject; var CanClose: Boolean);

begin

// Modifications? Ask if they should be saved.

FinalizeModifications(CanClose);

end;

procedure Tconnform.FormClose(Sender: TObject; var Action: TCloseAction);

begin

// Suspend calculating statistics as long as they're not visible

TimerStatistics.Enabled := False;

// Save GUI stuff

AppSettings.WriteInt(asSessionManagerListWidth, pnlLeft.Width);

AppSettings.WriteInt(asSessionManagerWindowWidth, Width);

AppSettings.WriteInt(asSessionManagerWindowHeight, Height);

AppSettings.WriteInt(asSessionManagerWindowLeft, Left);

AppSettings.WriteInt(asSessionManagerWindowTop, Top);

MainForm.SaveListSetup(ListSessions);

end;

procedure Tconnform.FormShow(Sender: TObject);

var

LastActiveSession: String;

LastSessions: TStringList;

PSess: PConnectionParameters;

Node: PVirtualNode;

begin

// Init sessions tree

RefreshSessions(nil);

// Focus last session

SelectNode(ListSessions, nil);

LastSessions := Explode(DELIM, AppSettings.ReadString(asLastSessions));

LastActiveSession := AppSettings.ReadString(asLastActiveSession);

if (LastActiveSession = '') and (LastSessions.Count > 0) then

LastActiveSession := LastSessions[0];

Node := ListSessions.GetFirst;

while Assigned(Node) do begin

PSess := ListSessions.GetNodeData(Node);

if PSess.SessionPath = LastActiveSession then

SelectNode(ListSessions, Node);

Node := ListSessions.GetNext(Node);

end;

ListSessions.SetFocus;

// Reactivate statistics

TimerStatistics.Enabled := True;

TimerStatistics.OnTimer(Sender);

FLoaded := True;

end;

function Tconnform.GetSelectedNetType: TNetType;

begin

Result := TNetType(comboNetType.ItemsEx[comboNetType.ItemIndex].Data);

end;

procedure Tconnform.SetSelectedNetType(Value: TNetType);

var

i: Integer;

begin

for i:=0 to comboNetType.ItemsEx.Count-1 do begin

if TNetType(comboNetType.ItemsEx[i].Data) = Value then begin

comboNetType.ItemIndex := i;

Break;

end;

end;

end;

procedure Tconnform.btnOpenClick(Sender: TObject);

var

Connection: TDBConnection;

Params: TConnectionParameters;

begin

// Connect to selected session

Params := CurrentParams;

if not btnOpen.Enabled then

Exit;

btnOpen.Enabled := False;

FButtonAnimationStep := 0;

TimerButtonAnimation.Enabled := True;

Screen.Cursor := crHourglass;

if Mainform.InitConnection(Params, True, Connection) then

ModalResult := mrOK

else begin

TimerStatistics.OnTimer(Sender);

ModalResult := mrNone;

end;

TimerButtonAnimation.Enabled := False;

btnOpen.Enabled := True;

btnOpen.Caption := _('Open');

Screen.Cursor := crDefault;

end;

procedure Tconnform.btnSaveClick(Sender: TObject);

var

Sess: PConnectionParameters;

Conn: TDBConnection;

begin

// Overtake edited values for current parameter object and save to registry

Sess := ListSessions.GetNodeData(ListSessions.FocusedNode);

Sess.Hostname := editHost.Text;

Sess.Username := editUsername.Text;

Sess.Password := editPassword.Text;

Sess.LoginPrompt := chkLoginPrompt.Checked;

Sess.WindowsAuth := chkWindowsAuth.Checked;

Sess.CleartextPluginEnabled := chkCleartextPluginEnabled.Checked;

Sess.Port := updownPort.Position;

Sess.NetType := SelectedNetType;

Sess.Compressed := chkCompressed.Checked;

Sess.QueryTimeout := updownQueryTimeout.Position;

Sess.KeepAlive := updownKeepAlive.Position;

Sess.LocalTimeZone := chkLocalTimeZone.Checked;

Sess.FullTableStatus := chkFullTableStatus.Checked;

Sess.SessionColor := ColorBoxBackgroundColor.Selected;

Sess.LibraryOrProvider := comboLibrary.Text;

Sess.AllDatabasesStr := editDatabases.Text;

Sess.Comment := memoComment.Text;

Sess.StartupScriptFilename := editStartupScript.Text;

Sess.SSHPlinkExe := editSSHPlinkExe.Text;

Sess.SSHHost := editSSHhost.Text;

Sess.SSHPort := MakeInt(editSSHport.Text);

Sess.SSHUser := editSSHUser.Text;

Sess.SSHPassword := editSSHPassword.Text;

Sess.SSHTimeout := updownSSHTimeout.Position;

Sess.SSHPrivateKey := editSSHPrivateKey.Text;

Sess.SSHLocalPort := MakeInt(editSSHlocalport.Text);

Sess.WantSSL := chkWantSSL.Checked;

Sess.SSLPrivateKey := editSSLPrivateKey.Text;

Sess.SSLCertificate := editSSLCertificate.Text;

Sess.SSLCACertificate := editSSLCACertificate.Text;

Sess.SSLCipher := editSSLCipher.Text;

Sess.IgnoreDatabasePattern := editIgnoreDatabasePattern.Text;

Sess.LogFileDdl := chkLogFileDdl.Checked;

Sess.LogFileDml := chkLogFileDml.Checked;

Sess.LogFilePath := editLogFilePath.Text;

Sess.SaveToRegistry;

FSessionModified := False;

ListSessions.Invalidate;

RefreshBackgroundColors;

ValidateControls;

// Apply session color (and othher settings) to opened connection(s)

for Conn in MainForm.Connections do begin

if Conn.Parameters.SessionPath = Sess.SessionPath then begin

Conn.Parameters.SessionColor := Sess.SessionColor;

MainForm.DBtree.Repaint;

end;

end;

end;

procedure Tconnform.btnMoreClick(Sender: TObject);

var

btn: TButton;

begin

btn := Sender as TButton;

btn.DropDownMenu.Popup(btn.ClientOrigin.X, btn.ClientOrigin.Y+btn.Height);

end;

procedure Tconnform.btnSaveAsClick(Sender: TObject);

var

newName, ParentKey: String;

NameOK: Boolean;

NewSess: TConnectionParameters;

Node: PVirtualNode;

SessionNames: TStringList;

begin

// Save session as ...

newName := _('Enter new session name ...');

NameOK := False;

SessionNames := NodeSessionNames(ListSessions.FocusedNode.Parent, ParentKey);

while not NameOK do begin

if not InputQuery(_('Clone session ...'), _('New session name:'), newName) then

Exit; // Cancelled

NameOK := SessionNames.IndexOf(newName) = -1;

if not NameOK then

ErrorDialog(f_('Session "%s" already exists!', [ParentKey+newName]))

else begin

// Create the key and save its values

NewSess := CurrentParams;

NewSess.SessionPath := ParentKey+newName;

NewSess.SaveToRegistry;

Node := ListSessions.InsertNode(ListSessions.FocusedNode, amInsertAfter, PConnectionParameters(NewSess));

FSessionModified := False;

SelectNode(ListSessions, Node);

end;

end;

SessionNames.Free;

end;

procedure Tconnform.btnImportSettingsClick(Sender: TObject);

begin

MainForm.actImportSettings.Execute;

FSettingsImportWaitTime := 0;

timerSettingsImport.Enabled := MainForm.ImportSettingsDone;

end;

procedure Tconnform.timerSettingsImportTimer(Sender: TObject);

begin

Inc(FSettingsImportWaitTime, timerSettingsImport.Interval);

RefreshSessions(nil);

if ListSessions.RootNodeCount > 0 then

timerSettingsImport.Enabled := False;

if FSettingsImportWaitTime >= 10000 then begin

timerSettingsImport.Enabled := False;

MessageDialog(f_('Imported sessions could not be detected. Restarting %s may solve that.', [APPNAME]), mtWarning, [mbOK]);

end;

end;

procedure Tconnform.btnNewClick(Sender: TObject);

var

i: Integer;

CanProceed, CreateInRoot: Boolean;

NewSess: TConnectionParameters;

ParentSess: PConnectionParameters;

ParentNode, NewNode: PVirtualNode;

ParentPath: String;

SiblingSessionNames: TStringList;

begin

// Create new session or folder

FinalizeModifications(CanProceed);

if not CanProceed then

Exit;

CreateInRoot := (Sender = menuNewSessionInRoot) or (Sender = menuNewFolderInRoot);

if Assigned(ListSessions.FocusedNode) then

ParentSess := ListSessions.GetNodeData(ListSessions.FocusedNode)

else

ParentSess := nil;

if CreateInRoot then

ParentNode := nil

else begin

if ParentSess = nil then

ParentNode := nil

else if ParentSess.IsFolder then

ParentNode := ListSessions.FocusedNode

else

ParentNode := ListSessions.FocusedNode.Parent;

end;

SiblingSessionNames := NodeSessionNames(ParentNode, ParentPath);

NewSess := TConnectionParameters.Create;

NewSess.IsFolder := (Sender = menuNewFolderInRoot) or (Sender = menuContextNewFolderInFolder) or (Sender = menuNewFolderInFolder);

NewSess.SessionPath := ParentPath + 'Unnamed';

i := 0;

while SiblingSessionNames.IndexOf(NewSess.SessionName) > -1 do begin

inc(i);

NewSess.SessionPath := ParentPath + 'Unnamed-' + IntToStr(i);

end;

NewSess.SaveToRegistry;

SiblingSessionNames.Free;

NewNode := ListSessions.AddChild(ParentNode, PConnectionParameters(NewSess));

// Select it

SelectNode(ListSessions, NewNode);

ValidateControls;

ListSessions.EditNode(NewNode, 0);

end;

procedure Tconnform.actFilterExecute(Sender: TObject);

begin

editSearch.SetFocus;

end;

procedure Tconnform.btnDeleteClick(Sender: TObject);

var

Sess: PConnectionParameters;

Node, FocusNode: PVirtualNode;

begin

Node := ListSessions.FocusedNode;

Sess := ListSessions.GetNodeData(Node);

if MessageDialog(f_('Delete session "%s"?', [Sess.SessionName]), mtConfirmation, [mbYes, mbCancel]) = mrYes then

begin

AppSettings.SessionPath := Sess.SessionPath;

AppSettings.DeleteCurrentKey;

if Assigned(Node.NextSibling) then

FocusNode := Node.NextSibling

else if Assigned(Node.PrevSibling) then

FocusNode := Node.PrevSibling

else

FocusNode := Node.Parent;

ListSessions.DeleteNode(Node);

SelectNode(ListSessions, FocusNode);

ListSessions.SetFocus;

end;

end;

function Tconnform.SelectedSessionPath: String;

var

Sess: PConnectionParameters;

begin

if not Assigned(ListSessions.FocusedNode) then

Result := ''

else begin

Sess := ListSessions.GetNodeData(ListSessions.FocusedNode);

Result := Sess.SessionPath;

end;

end;

function Tconnform.CurrentParams: TConnectionParameters;

var

FromReg: PConnectionParameters;

begin

// Return non-stored parameters

FromReg := ListSessions.GetNodeData(ListSessions.FocusedNode);

if FromReg.IsFolder then begin

Result := FromReg^;

end else begin

Result := TConnectionParameters.Create;

Result.SessionPath := SelectedSessionPath;

Result.Counter := FromReg.Counter;

Result.SessionColor := ColorBoxBackgroundColor.Selected;

Result.NetType := SelectedNetType;

Result.ServerVersion := FServerVersion;

Result.Hostname := editHost.Text;

Result.Username := editUsername.Text;

Result.Password := editPassword.Text;

Result.LoginPrompt := chkLoginPrompt.Checked;

Result.WindowsAuth := chkWindowsAuth.Checked;

Result.CleartextPluginEnabled := chkCleartextPluginEnabled.Checked;

if updownPort.Enabled then

Result.Port := updownPort.Position

else

Result.Port := 0;

Result.AllDatabasesStr := editDatabases.Text;

Result.LibraryOrProvider := comboLibrary.Text;

Result.Comment := memoComment.Text;

Result.SSHHost := editSSHHost.Text;

Result.SSHPort := MakeInt(editSSHPort.Text);

Result.SSHUser := editSSHuser.Text;

Result.SSHPassword := editSSHpassword.Text;

Result.SSHTimeout := updownSSHTimeout.Position;

Result.SSHPrivateKey := editSSHPrivateKey.Text;

Result.SSHLocalPort := MakeInt(editSSHlocalport.Text);

Result.SSHPlinkExe := editSSHplinkexe.Text;

Result.WantSSL := chkWantSSL.Checked;

Result.SSLPrivateKey := editSSLPrivateKey.Text;

Result.SSLCertificate := editSSLCertificate.Text;

Result.SSLCACertificate := editSSLCACertificate.Text;

Result.SSLCipher := editSSLCipher.Text;

Result.StartupScriptFilename := editStartupScript.Text;

Result.Compressed := chkCompressed.Checked;

Result.QueryTimeout := updownQueryTimeout.Position;

Result.KeepAlive := updownKeepAlive.Position;

Result.LocalTimeZone := chkLocalTimeZone.Checked;

Result.FullTableStatus := chkFullTableStatus.Checked;

Result.SessionColor := ColorBoxBackgroundColor.Selected;

Result.IgnoreDatabasePattern := editIgnoreDatabasePattern.Text;

Result.LogFileDdl := chkLogFileDdl.Checked;

Result.LogFileDml := chkLogFileDml.Checked;

Result.LogFilePath := editLogFilePath.Text;

end;

end;

procedure Tconnform.ListSessionsGetImageIndex(Sender: TBaseVirtualTree;

Node: PVirtualNode; Kind: TVTImageKind; Column: TColumnIndex;

var Ghosted: Boolean; var ImageIndex: TImageIndex);

var

Sess: PConnectionParameters;

begin

// An edited session gets an additional pencil symbol

if Column > 0 then

ImageIndex := -1

else case Kind of

ikNormal, ikSelected: begin

Sess := Sender.GetNodeData(Node);

ImageIndex := Sess.ImageIndex;

end;

ikOverlay:

if (Node = Sender.FocusedNode) and FSessionModified then

ImageIndex := 162;

end;

end;

procedure Tconnform.ListSessionsGetNodeDataSize(Sender: TBaseVirtualTree;

var NodeDataSize: Integer);

begin

NodeDataSize := SizeOf(TConnectionParameters);

end;

procedure Tconnform.ListSessionsGetText(Sender: TBaseVirtualTree;

Node: PVirtualNode; Column: TColumnIndex; TextType: TVSTTextType;

var CellText: String);

var

Sess: PConnectionParameters;

begin

// Display session name cell

Sess := Sender.GetNodeData(Node);

if Sess.IsFolder then begin

case Column of

0: CellText := Sess.SessionName;

else CellText := '';

end;

end else begin

case Column of

0: begin

CellText := Sess.SessionName;

if FSessionModified and (Node = Sender.FocusedNode) and (not Sender.IsEditing) then

CellText := CellText + ' *';

end;

1: CellText := Sess.Hostname;

2: CellText := Sess.Username;

3: CellText := Sess.ServerVersion;

4: if Sess.LastConnect>0 then

CellText := DateTimeToStr(Sess.LastConnect)

else

CellText := '';

5: CellText := FormatNumber(Sess.Counter);

6: CellText := Sess.Comment;

end;

end;

end;

function Tconnform.NodeSessionNames(Node: PVirtualNode; var RegKey: String): TStringList;

var

Sess: PConnectionParameters;

Folders: TStringList;

begin

// Find sibling session names in a folder node

if Node = nil then

Node := ListSessions.RootNode;

// Find registry sub path for given node

RegKey := '';

if Node <> ListSessions.RootNode then begin

Sess := ListSessions.GetNodeData(Node);

RegKey := Sess.SessionPath + '\';

end;

// Fetch from registry

Folders := TStringList.Create;

Result := AppSettings.GetSessionNames(RegKey, Folders);

Result.AddStrings(Folders);

Folders.Free;

end;

procedure Tconnform.ListSessionsBeforeCellPaint(Sender: TBaseVirtualTree;

TargetCanvas: TCanvas; Node: PVirtualNode; Column: TColumnIndex;

CellPaintMode: TVTCellPaintMode; CellRect: TRect; var ContentRect: TRect);

var

Session: PConnectionParameters;

begin

// Paint custom background color

if CellPaintMode=cpmPaint then begin

Session := Sender.GetNodeData(Node);

if Session.SessionColor <> AppSettings.GetDefaultInt(asTreeBackground) then begin

TargetCanvas.Brush.Color := Session.SessionColor;

TargetCanvas.FillRect(CellRect);

end;

end;

end;

procedure Tconnform.ListSessionsCreateEditor(Sender: TBaseVirtualTree; Node: PVirtualNode;

Column: TColumnIndex; out EditLink: IVTEditLink);

begin

// Use our own text editor to rename a session

EditLink := TInplaceEditorLink.Create(Sender as TVirtualStringTree, True, nil);

end;

procedure Tconnform.ListSessionsDragDrop(Sender: TBaseVirtualTree;

Source: TObject; DataObject: IDataObject; Formats: TFormatArray;

Shift: TShiftState; Pt: TPoint; var Effect: Integer; Mode: TDropMode);

var

TargetNode, ParentNode: PVirtualNode;

AttachMode: TVTNodeAttachMode;

TargetSess, FocusedSess: PConnectionParameters;

ParentKey: String;

SiblingSessions: TStringList;

begin

TargetNode := Sender.GetNodeAt(Pt.X, Pt.Y);

if not Assigned(TargetNode) then begin

MessageBeep(MB_ICONEXCLAMATION);

Exit;

end;

TargetSess := Sender.GetNodeData(TargetNode);

FocusedSess := Sender.GetNodeData(ListSessions.FocusedNode);

case Mode of

dmAbove:

AttachMode := amInsertBefore;

dmOnNode:

if TargetSess.IsFolder then

AttachMode := amAddChildFirst

else

AttachMode := amInsertBefore;

dmBelow:

AttachMode := amInsertAfter;

else

AttachMode := amInsertAfter;

end;

if AttachMode in [amInsertBefore, amInsertAfter] then

ParentNode := TargetNode.Parent

else

ParentNode := TargetNode;

SiblingSessions := NodeSessionNames(ParentNode, ParentKey);

// Test if target folder has an equal named node

if SiblingSessions.IndexOf(FocusedSess.SessionName) > -1 then

ErrorDialog(f_('Session "%s" already exists!', [ParentKey+FocusedSess.SessionName]))

else begin

try

AppSettings.SessionPath := FocusedSess.SessionPath;

AppSettings.MoveCurrentKey(REGKEY_SESSIONS+'\'+ParentKey+FocusedSess.SessionName);

ListSessions.MoveTo(ListSessions.FocusedNode, TargetNode, AttachMode, False);

FocusedSess.SessionPath := ParentKey+FocusedSess.SessionName;

except

on E:Exception do

ErrorDialog(f_('Error while moving registry key: %s', [E.Message]));

end;

end;

SiblingSessions.Free;

end;

procedure Tconnform.ListSessionsDragOver(Sender: TBaseVirtualTree;

Source: TObject; Shift: TShiftState; State: TDragState; Pt: TPoint;

Mode: TDropMode; var Effect: Integer; var Accept: Boolean);

var

TargetNode, ParentNode: PVirtualNode;

TargetSess: PConnectionParameters;

begin

// Allow node dragging everywhere except within the current folder

TargetNode := Sender.GetNodeAt(Pt.X, Pt.Y);

TargetSess := Sender.GetNodeData(TargetNode);

Accept := (Source = Sender)

and Assigned(TargetSess)

and (Mode <> dmNowhere)

and (TargetNode <> ListSessions.FocusedNode.Parent);

// Moving a folder into itself would create an infinite folder structure

if Accept and TargetSess.IsFolder then

Accept := Accept and (TargetNode <> ListSessions.FocusedNode);

if Accept and (not TargetSess.IsFolder) then

Accept := Accept and (TargetNode.Parent <> ListSessions.FocusedNode.Parent);

if Accept then begin

// Do not allow focused node to be moved somewhere below itself

ParentNode := TargetNode.Parent;

while Assigned(ParentNode) do begin

Accept := Accept and (ParentNode <> ListSessions.FocusedNode);

if not Accept then

Break;

ParentNode := ParentNode.Parent;

end;

// Shows the right tooltip on Aero GUI

Effect := DROPEFFECT_MOVE;

end;

end;

procedure Tconnform.ListSessionsFocusChanged(Sender: TBaseVirtualTree;

Node: PVirtualNode; Column: TColumnIndex);

var

SessionFocused, InFolder: Boolean;

Sess: PConnectionParameters;

begin

// select one connection!

Screen.Cursor := crHourglass;

TimerStatistics.Enabled := False;

SessionFocused := False;

InFolder := False;

Sess := nil;

if Assigned(Node) then begin

Sess := Sender.GetNodeData(Node);

SessionFocused := not Sess.IsFolder;

InFolder := (ListSessions.GetNodeLevel(Node) > 0) or Sess.IsFolder;

end;

FLoaded := False;

tabStart.TabVisible := not SessionFocused;

tabSettings.TabVisible := SessionFocused;

tabSSHtunnel.TabVisible := SessionFocused;

tabAdvanced.TabVisible := SessionFocused;

tabStatistics.TabVisible := SessionFocused;

menuRename.Enabled := Assigned(Node);

menuNewSessionInFolder.Enabled := InFolder;

menuNewFolderInFolder.Enabled := InFolder;

FreeAndNil(FPopupDatabases);

if not SessionFocused then begin

PageControlDetails.ActivePage := tabStart;

if ListSessions.RootNodeCount = 0 then

lblHelp.Caption := f_('New here? In order to connect to a server, you have to create a so called '+

'"session" at first. Just click the "New" button on the bottom left to create your first session.'+

'Give it a friendly name (e.g. "Local DB server") so you''ll recall it the next time you start %s.', [APPNAME])

else

lblHelp.Caption := _('Please click a session on the left list to edit parameters, doubleclick to open it.');

end else begin

PageControlDetails.ActivePage := tabSettings;

SelectedNetType := Sess.NetType;

FLastSelectedNetTypeGroup := Sess.NetTypeGroup;

editHost.Text := Sess.Hostname;

editUsername.Text := Sess.Username;

editPassword.Text := Sess.Password;

chkLoginPrompt.Checked := Sess.LoginPrompt;

chkWindowsAuth.Checked := Sess.WindowsAuth;

chkCleartextPluginEnabled.Checked := Sess.CleartextPluginEnabled;

updownPort.Position := Sess.Port;

chkCompressed.Checked := Sess.Compressed;

updownQueryTimeout.Position := Sess.QueryTimeout;

updownKeepAlive.Position := Sess.KeepAlive;

chkLocalTimeZone.Checked := Sess.LocalTimeZone;

chkFullTableStatus.Checked := Sess.FullTableStatus;

ColorBoxBackgroundColor.Selected := Sess.SessionColor;

editDatabases.Text := Sess.AllDatabasesStr;

comboLibrary.Items := Sess.GetLibraries;

comboLibrary.ItemIndex := comboLibrary.Items.IndexOf(Sess.LibraryOrProvider);

if (comboLibrary.ItemIndex = -1) and (comboLibrary.Items.Count > 0) then begin

comboLibrary.ItemIndex := 0;

end;

memoComment.Text := Sess.Comment;

editStartupScript.Text := Sess.StartupScriptFilename;

editSSHPlinkExe.Text := Sess.SSHPlinkExe;

editSSHHost.Text := Sess.SSHHost;

editSSHport.Text := IntToStr(Sess.SSHPort);

editSSHUser.Text := Sess.SSHUser;

editSSHPassword.Text := Sess.SSHPassword;

updownSSHTimeout.Position := Sess.SSHTimeout;

editSSHPrivateKey.Text := Sess.SSHPrivateKey;

editSSHlocalport.Text := IntToStr(Sess.SSHLocalPort);

chkWantSSL.Checked := Sess.WantSSL;

editSSLPrivateKey.Text := Sess.SSLPrivateKey;

editSSLCertificate.Text := Sess.SSLCertificate;

editSSLCACertificate.Text := Sess.SSLCACertificate;

editSSLCipher.Text := Sess.SSLCipher;

editIgnoreDatabasePattern.Text := Sess.IgnoreDatabasePattern;

chkLogFileDdl.Checked := Sess.LogFileDdl;

chkLogFileDml.Checked := Sess.LogFileDml;

editLogFilePath.Text := Sess.LogFilePath;

FServerVersion := Sess.ServerVersion;

end;

FLoaded := True;

FSessionModified := False;

ListSessions.Repaint;

ValidateControls;