mirror of
https://github.com/HeidiSQL/HeidiSQL.git
synced 2025-08-06 18:24:26 +08:00
349 lines
12 KiB
ObjectPascal
349 lines
12 KiB
ObjectPascal
unit copytable;
|
|
|
|
|
|
interface
|
|
|
|
uses
|
|
Windows, SysUtils, Classes, Graphics, Controls, Forms, Dialogs, StdCtrls,
|
|
mysql_connection, VirtualTrees, SynEdit, SynMemo;
|
|
|
|
type
|
|
TCopyTableForm = class(TForm)
|
|
editNewTablename: TEdit;
|
|
lblNewTablename: TLabel;
|
|
btnCancel: TButton;
|
|
comboDatabase: TComboBox;
|
|
btnOK: TButton;
|
|
TreeElements: TVirtualStringTree;
|
|
MemoWhereClause: TSynMemo;
|
|
lblItems: TLabel;
|
|
lblWhere: TLabel;
|
|
procedure editNewTablenameChange(Sender: TObject);
|
|
procedure FormShow(Sender: TObject);
|
|
procedure btnOKClick(Sender: TObject);
|
|
procedure FormCreate(Sender: TObject);
|
|
procedure TreeElementsGetText(Sender: TBaseVirtualTree; Node: PVirtualNode;
|
|
Column: TColumnIndex; TextType: TVSTTextType; var CellText: string);
|
|
procedure TreeElementsInitNode(Sender: TBaseVirtualTree; ParentNode, Node: PVirtualNode;
|
|
var InitialStates: TVirtualNodeInitStates);
|
|
procedure TreeElementsGetImageIndex(Sender: TBaseVirtualTree; Node: PVirtualNode;
|
|
Kind: TVTImageKind; Column: TColumnIndex; var Ghosted: Boolean; var ImageIndex: Integer);
|
|
procedure TreeElementsInitChildren(Sender: TBaseVirtualTree; Node: PVirtualNode;
|
|
var ChildCount: Cardinal);
|
|
procedure FormDestroy(Sender: TObject);
|
|
procedure TreeElementsChecked(Sender: TBaseVirtualTree; Node: PVirtualNode);
|
|
procedure FormClose(Sender: TObject; var Action: TCloseAction);
|
|
private
|
|
{ Private declarations }
|
|
FOrgTableName: String;
|
|
FColumns: TTableColumnList;
|
|
FKeys: TTableKeyList;
|
|
FForeignKeys: TForeignKeyList;
|
|
public
|
|
{ Public declarations }
|
|
end;
|
|
|
|
|
|
implementation
|
|
|
|
uses helpers, main;
|
|
|
|
const
|
|
nColumns = 0;
|
|
nKeys = 1;
|
|
nForeignKeys = 2;
|
|
nData = 3;
|
|
|
|
{$R *.DFM}
|
|
{$I const.inc}
|
|
|
|
|
|
|
|
procedure TCopyTableForm.FormCreate(Sender: TObject);
|
|
begin
|
|
InheritFont(Font);
|
|
Width := GetRegValue(REGNAME_COPYTABLE_WINWIDTH, Width);
|
|
Height := GetRegValue(REGNAME_COPYTABLE_WINHEIGHT, Height);
|
|
SetWindowSizeGrip(Handle, True);
|
|
MainForm.SetupSynEditors;
|
|
FixVT(TreeElements);
|
|
FColumns := TTableColumnList.Create;
|
|
FKeys := TTableKeyList.Create;
|
|
FForeignKeys := TForeignKeyList.Create;
|
|
end;
|
|
|
|
|
|
procedure TCopyTableForm.FormDestroy(Sender: TObject);
|
|
begin
|
|
// Save GUI stuff
|
|
MainReg.WriteInteger(REGNAME_COPYTABLE_WINWIDTH, Width);
|
|
MainReg.WriteInteger(REGNAME_COPYTABLE_WINHEIGHT, Height);
|
|
end;
|
|
|
|
|
|
procedure TCopyTableForm.FormShow(Sender: TObject);
|
|
var
|
|
CreateCode: String;
|
|
begin
|
|
if Mainform.DBtree.Focused then
|
|
FOrgTableName := Mainform.SelectedTable.Name
|
|
else
|
|
FOrgTableName := Mainform.ListTables.Text[Mainform.ListTables.FocusedNode, 0];
|
|
editNewTablename.Text := FOrgTableName + '_copy';
|
|
editNewTablename.SetFocus;
|
|
lblNewTablename.Caption := 'Copy ''' + FOrgTableName + ''' to new db.table:';
|
|
editNewTablename.SetFocus;
|
|
|
|
// Select TargetDatabase
|
|
comboDatabase.Items.Clear;
|
|
comboDatabase.Items.Assign(Mainform.AllDatabases);
|
|
comboDatabase.ItemIndex := comboDatabase.Items.IndexOf(Mainform.ActiveDatabase);
|
|
if comboDatabase.ItemIndex = -1 then
|
|
comboDatabase.ItemIndex := 0;
|
|
|
|
CreateCode := MainForm.Connection.GetVar('SHOW CREATE TABLE '+MainForm.mask(FOrgTableName), 1);
|
|
FColumns.Clear;
|
|
FKeys.Clear;
|
|
FForeignKeys.Clear;
|
|
ParseTableStructure(CreateCode, FColumns, FKeys, FForeignKeys);
|
|
|
|
TreeElements.Clear;
|
|
TreeElements.RootNodeCount := 4;
|
|
end;
|
|
|
|
|
|
|
|
procedure TCopyTableForm.FormClose(Sender: TObject; var Action: TCloseAction);
|
|
var
|
|
Node: PVirtualNode;
|
|
Option: String;
|
|
begin
|
|
// Save first level node check options
|
|
Node := TreeElements.GetFirst;
|
|
while Assigned(Node) do begin
|
|
case Node.Index of
|
|
nColumns: Option := REGNAME_COPYTABLE_COLUMNS;
|
|
nKeys: Option := REGNAME_COPYTABLE_KEYS;
|
|
nForeignKeys: Option := REGNAME_COPYTABLE_FOREIGN;
|
|
nData: Option := REGNAME_COPYTABLE_DATA;
|
|
else raise Exception.Create(SUnhandledNodeIndex);
|
|
end;
|
|
MainReg.WriteBool(Option, Node.CheckState in [csCheckedNormal, csCheckedPressed, csMixedNormal, csMixedPressed]);
|
|
Node := TreeElements.GetNextSibling(Node);
|
|
end;
|
|
end;
|
|
|
|
|
|
procedure TCopyTableForm.TreeElementsChecked(Sender: TBaseVirtualTree; Node: PVirtualNode);
|
|
begin
|
|
// Disable WHERE memo if "Data" was unselected
|
|
if (Sender.GetNodeLevel(Node) = 0) and (Node.Index = nData) then begin
|
|
MemoWhereClause.Enabled := Node.CheckState = csCheckedNormal;
|
|
if MemoWhereClause.Enabled then begin
|
|
MemoWhereClause.Highlighter := MainForm.SynSQLSyn1;
|
|
MemoWhereClause.Color := clWindow;
|
|
end else begin
|
|
MemoWhereClause.Highlighter := nil;
|
|
MemoWhereClause.Color := clBtnFace;
|
|
end;
|
|
end;
|
|
end;
|
|
|
|
|
|
procedure TCopyTableForm.TreeElementsGetImageIndex(Sender: TBaseVirtualTree; Node: PVirtualNode;
|
|
Kind: TVTImageKind; Column: TColumnIndex; var Ghosted: Boolean; var ImageIndex: Integer);
|
|
begin
|
|
// Get node index
|
|
if not (Kind in [ikNormal, ikSelected]) then Exit;
|
|
case Sender.GetNodeLevel(Node) of
|
|
0: case Node.Index of
|
|
nColumns: ImageIndex := ICONINDEX_FIELD;
|
|
nKeys: ImageIndex := 13;
|
|
nForeignKeys: ImageIndex := ICONINDEX_FOREIGNKEY;
|
|
nData: ImageIndex := 41;
|
|
else raise Exception.Create(SUnhandledNodeIndex);
|
|
end;
|
|
1: case Node.Parent.Index of
|
|
nColumns: ImageIndex := ICONINDEX_FIELD;
|
|
nKeys: ImageIndex := GetIndexIcon(FKeys[Node.Index].IndexType);
|
|
nForeignKeys: ImageIndex := ICONINDEX_FOREIGNKEY;
|
|
else raise Exception.Create(SUnhandledNodeIndex);
|
|
end;
|
|
end;
|
|
end;
|
|
|
|
|
|
procedure TCopyTableForm.TreeElementsGetText(Sender: TBaseVirtualTree; Node: PVirtualNode;
|
|
Column: TColumnIndex; TextType: TVSTTextType; var CellText: string);
|
|
begin
|
|
// Get node text
|
|
case Sender.GetNodeLevel(Node) of
|
|
0: case Node.Index of
|
|
nColumns: CellText := 'Columns';
|
|
nKeys: CellText := 'Indexes';
|
|
nForeignKeys: CellText := 'Foreign keys';
|
|
nData: CellText := 'Data';
|
|
else raise Exception.Create(SUnhandledNodeIndex);
|
|
end;
|
|
1: case Node.Parent.Index of
|
|
nColumns: CellText := FColumns[Node.Index].Name;
|
|
nKeys: CellText := FKeys[Node.Index].Name;
|
|
nForeignKeys: CellText := FForeignKeys[Node.Index].KeyName;
|
|
else raise Exception.Create(SUnhandledNodeIndex);
|
|
end;
|
|
end;
|
|
end;
|
|
|
|
|
|
procedure TCopyTableForm.TreeElementsInitChildren(Sender: TBaseVirtualTree; Node: PVirtualNode;
|
|
var ChildCount: Cardinal);
|
|
begin
|
|
// Set child node count
|
|
case Sender.GetNodeLevel(Node) of
|
|
0: case Node.Index of
|
|
nColumns: ChildCount := FColumns.Count;
|
|
nKeys: ChildCount := FKeys.Count;
|
|
nForeignKeys: ChildCount := FForeignKeys.Count;
|
|
nData: ChildCount := 0;
|
|
else raise Exception.Create(SUnhandledNodeIndex);
|
|
end;
|
|
else ChildCount := 0;
|
|
end;
|
|
end;
|
|
|
|
|
|
procedure TCopyTableForm.TreeElementsInitNode(Sender: TBaseVirtualTree; ParentNode,
|
|
Node: PVirtualNode; var InitialStates: TVirtualNodeInitStates);
|
|
var
|
|
Option: String;
|
|
begin
|
|
// First three upper nodes mostly have child nodes
|
|
Node.CheckType := ctTriStateCheckBox;
|
|
case Sender.GetNodeLevel(Node) of
|
|
0: begin
|
|
if Node.Index in [nColumns, nKeys, nForeignKeys] then
|
|
Include(InitialStates, ivsHasChildren);
|
|
case Node.Index of
|
|
nColumns: Option := REGNAME_COPYTABLE_COLUMNS;
|
|
nKeys: Option := REGNAME_COPYTABLE_KEYS;
|
|
nForeignKeys: Option := REGNAME_COPYTABLE_FOREIGN;
|
|
nData: Option := REGNAME_COPYTABLE_DATA;
|
|
else raise Exception.Create(SUnhandledNodeIndex);
|
|
end;
|
|
if GetRegValue(Option, True) then
|
|
Node.CheckState := csCheckedNormal;
|
|
(Sender as TVirtualStringTree).OnChecked(Sender, Node);
|
|
// Disable node to indicate 0 children
|
|
if ((Node.Index = nKeys) and (FKeys.Count = 0))
|
|
or ((Node.Index = nForeignKeys) and (FForeignKeys.Count = 0)) then
|
|
Node.States := Node.States + [vsDisabled];
|
|
end;
|
|
|
|
1: if Node.Parent.CheckState in [csCheckedNormal, csCheckedPressed, csMixedNormal, csMixedPressed] then
|
|
Node.CheckState := csCheckedNormal;
|
|
end;
|
|
end;
|
|
|
|
|
|
procedure TCopyTableForm.editNewTablenameChange(Sender: TObject);
|
|
begin
|
|
// Disable OK button as long as table name is empty
|
|
btnOK.Enabled := editNewTablename.Text <> '';
|
|
end;
|
|
|
|
|
|
procedure TCopyTableForm.btnOKClick(Sender: TObject);
|
|
var
|
|
CreateCode, Clause, DataCols, TableExistance: String;
|
|
ParentNode, Node: PVirtualNode;
|
|
DBObjects: TDBObjectList;
|
|
Obj: TDBObject;
|
|
DoData: Boolean;
|
|
begin
|
|
// Compose and run CREATE query
|
|
TableExistance := MainForm.Connection.GetVar('SHOW TABLES FROM '+MainForm.mask(comboDatabase.Text)+' LIKE '+esc(editNewTablename.Text));
|
|
if TableExistance <> '' then begin
|
|
if MessageDlg('Target table exists. Drop it and overwrite?', mtConfirmation, [mbYes, mbCancel], 0) = mrCancel then begin
|
|
ModalResult := mrNone;
|
|
Exit;
|
|
end;
|
|
MainForm.Connection.Query('DROP TABLE '+MainForm.mask(comboDatabase.Text)+'.'+MainForm.mask(editNewTablename.Text));
|
|
end;
|
|
|
|
Screen.Cursor := crHourglass;
|
|
MainForm.ShowStatusMsg('Generating SQL code ...');
|
|
DataCols := '';
|
|
DoData := False;
|
|
ParentNode := TreeElements.GetFirst;
|
|
while Assigned(ParentNode) do begin
|
|
Node := TreeElements.GetFirstChild(ParentNode);
|
|
while Assigned(Node) do begin
|
|
if Node.CheckState in [csCheckedNormal, csCheckedPressed] then begin
|
|
case ParentNode.Index of
|
|
nColumns: begin
|
|
Clause := FColumns[Node.Index].SQLCode;
|
|
DataCols := DataCols + MainForm.mask(FColumns[Node.Index].Name) + ', ';
|
|
end;
|
|
nKeys: Clause := FKeys[Node.Index].SQLCode;
|
|
nForeignkeys: Clause := FForeignKeys[Node.Index].SQLCode;
|
|
else raise Exception.Create(SUnhandledNodeIndex);
|
|
end;
|
|
CreateCode := CreateCode + #9 + Clause + ',' + CRLF;
|
|
end;
|
|
Node := TreeElements.GetNextSibling(Node);
|
|
end;
|
|
if (ParentNode.Index = nData) then
|
|
DoData := ParentNode.CheckState in [csCheckedNormal, csCheckedPressed];
|
|
ParentNode := TreeElements.GetNextSibling(ParentNode);
|
|
end;
|
|
Delete(CreateCode, Length(CreateCode)-2, 3);
|
|
CreateCode := 'CREATE TABLE '+Mainform.mask(comboDatabase.Text)+'.'+Mainform.mask(editNewTablename.Text)+' ('+CRLF+CreateCode+CRLF+')'+CRLF;
|
|
|
|
// Add collation and engine clauses
|
|
DBObjects := Mainform.Connection.GetDBObjects(Mainform.ActiveDatabase);
|
|
for Obj in DBObjects do begin
|
|
if Obj.Name = FOrgTableName then begin
|
|
if Obj.Collation <> '' then
|
|
CreateCode := CreateCode + ' COLLATE ''' + Obj.Collation + '''';
|
|
if Obj.Engine <> '' then begin
|
|
if Mainform.Connection.ServerVersionInt < 40018 then
|
|
CreateCode := CreateCode + ' TYPE=' + Obj.Engine
|
|
else
|
|
CreateCode := CreateCode + ' ENGINE=' + Obj.Engine;
|
|
end;
|
|
if Obj.RowFormat <> '' then
|
|
CreateCode := CreateCode + ' ROW_FORMAT=' + Obj.RowFormat;
|
|
if Obj.AutoInc > -1 then
|
|
CreateCode := CreateCode + ' AUTO_INCREMENT=' + IntToStr(Obj.AutoInc);
|
|
CreateCode := CreateCode + ' COMMENT=' + esc(Obj.Comment);
|
|
break;
|
|
end;
|
|
end;
|
|
|
|
// Append SELECT .. FROM OrgTable clause
|
|
if DoData and (DataCols <> '') then begin
|
|
DataCols := Trim(DataCols);
|
|
Delete(DataCols, Length(DataCols), 1);
|
|
CreateCode := CreateCode + ' SELECT ' + DataCols + ' FROM ' + MainForm.mask(FOrgTableName);
|
|
if MemoWhereClause.GetTextLen > 0 then
|
|
CreateCode := CreateCode + ' WHERE ' + MemoWhereClause.Text;
|
|
end;
|
|
|
|
// Run query and refresh list
|
|
try
|
|
MainForm.ShowStatusMsg('Creating table ...');
|
|
MainForm.Connection.Query(CreateCode);
|
|
MainForm.actRefresh.Execute;
|
|
except
|
|
on E:EDatabaseError do begin
|
|
Screen.Cursor := crDefault;
|
|
MessageDlg(E.Message, mtError, [mbOk], 0);
|
|
ModalResult := mrNone;
|
|
end;
|
|
end;
|
|
MainForm.ShowStatusMsg;
|
|
Screen.Cursor := crDefault;
|
|
end;
|
|
|
|
end.
|