unit DbRestore;

interface

uses
  Windows, Messages, SysUtils, Classes, Graphics, Controls, Forms, Dialogs,
  StdCtrls,registry, Buttons, ExtCtrls,FileCtrl, Db, DBTables;

type
  TDbRestoreFrm = class(TForm)
    Panel1: TPanel;
    Label1: TLabel;
    Label2: TLabel;
    Label3: TLabel;
    Label7: TLabel;
    BitBtn1: TBitBtn;
    BitBtn2: TBitBtn;
    Label4: TLabel;
    Edit_userid: TEdit;
    Edit_pass: TEdit;
    Edit_string: TEdit;
    SpeedButton1: TSpeedButton;
    Query1: TQuery;
    Query2: TQuery;
    OpenDialog1: TOpenDialog;
    Databasedh: TDatabase;
    procedure FormCreate(Sender: TObject);
    procedure SpeedButton1Click(Sender: TObject);
    procedure BitBtn1Click(Sender: TObject);
  private
    { Private declarations }
    FDir:string;
    //得到ORACLE 的系統路徑
    Function GetOracleDir:string;
    procedure DropAllTable;
    //得到數據庫服務器日期
    Function GetDbServerDate(DBName:string):string;
    Function ExecQuery(PQuery:TQuery;PSql:string):Boolean;
  public
    { Public declarations }
  end;

var
  DbRestoreFrm: TDbRestoreFrm;

implementation

{$R *.DFM}

Function TDbRestoreFrm.GetOracleDir:string;
var
  Reg: TRegistry;
begin
  Reg := TRegistry.Create;
  result:='';
  try
    Reg.RootKey := HKEY_LOCAL_MACHINE;
    if Reg.OpenKey('\SOFTWARE\ORACLE', True) then
      result:=Reg.ReadString('ORACLE_HOME');
  finally
    Reg.CloseKey;
    Reg.Free;
  end;
end;
procedure TDbRestoreFrm.FormCreate(Sender: TObject);
begin
  Edit_userid.text:='';
  Edit_Pass.text  :='';
  Edit_string.text:='';

  FDir:=ExtractFilePath(ParamStr(0))+'data\'+GetDbServerDate('db_vipdl');
  if not DirectoryExists(FDir) then ForceDirectories(FDir);

  label4.Caption:='';
end;

procedure TDbRestoreFrm.SpeedButton1Click(Sender: TObject);
begin
  if OpenDialog1.Execute then
  begin
    label4.Caption:=OpenDialog1.FileName;
  end;
end;

procedure TDbRestoreFrm.BitBtn1Click(Sender: TObject);
var ftem:string;
begin
  //******應該要驗證用戶名和密碼是否正確*********
   Try
     Databasedh.Connected := false;
     Databasedh.Params.Clear;
     Databasedh.Params.Add('SERVER NAME='+Edit_string.Text);
     Databasedh.Params.Add('USER NAME='+Edit_userid.Text);  //用戶名和密碼固定不變
     Databasedh.Params.Add('PASSWORD='+Edit_Pass.Text);
     Databasedh.Connected := True;
   Except
     Application.MessageBox(Pchar('該用戶不能進行恢復!'), '系統提示', MB_OK+MB_ICONERROR);
     Exit;
   End;
   Databasedh.Connected :=false;
  //**********************************************
  if trim(Edit_userid.text) = '' then
  begin
    Application.MessageBox('用戶帳號不能為空!','警告',MB_OK+MB_ICONINFORMATION);
    edit_userid.SetFocus;
    exit;
  end;
  if trim(Edit_Pass.text) = '' then
  begin
    Application.MessageBox('用戶口令不能為空!','警告',MB_OK+MB_ICONINFORMATION);
    edit_pass.SetFocus;
    exit;
  end;
  if trim(Edit_string.text) = '' then
  begin
    Application.MessageBox('連接串不能為空!','警告',MB_OK+MB_ICONINFORMATION);
    edit_string.SetFocus;
    exit;
  end;
  if label4.Caption = '' then
  begin
    Application.MessageBox('請選擇文件!','警告',MB_OK+MB_ICONINFORMATION);

    exit;
  end;
  ftem:= GetOracleDir;
  if ftem = '' then
  begin
    Application.MessageBox('本機沒有ORACLE 數據庫系統,不能進行整個數據備份!','警告',MB_OK+MB_ICONINFORMATION);
    exit;
  end;
  ftem:= GetOracleDir+'\bin\imp.exe';
  if not FileExists(ftem) then
  begin
     ftem:=GetOracleDir+'\bin\imp.exe';
     if not FileExists(ftem+'\bin\exp.exe') then
     begin
        Application.MessageBox('備份工具不存在!','警告',MB_OK+MB_ICONINFORMATION);
        exit;
     end
  end;
  if copy(trim(edit_string.text),1,1)='@' then
    ftem:=ftem + ' '+trim(edit_userid.text)+'/'+trim(edit_pass.text)+trim(edit_string.text)+' file='+label4.Caption+' commit=y full=y ignore=y log='+Fdir+'\restore.log'
  else
    ftem:=ftem + ' '+trim(edit_userid.text)+'/'+trim(edit_pass.text)+'@'+trim(edit_string.text)+' file='+label4.Caption+' commit=y full=y ignore=y log='+Fdir+'\restore.log';

  //先刪除表
  DropAllTable;
  winExec(pchar(ftem),SW_MINIMIZE);
  close;
end;
procedure TDbRestoreFrm.DropAllTable;
var fsql:string;
BEGIN
    query1.open;
    query1.First;
    while not query1.Eof do
    begin
      fsql:='drop table '+Query1.FieldByName('TABLE_NAME').asstring;
      ExecQuery(query2,fsql);
      query1.Next;
    end;
END;
function TDbRestoreFrm.GetDbServerDate(DBName: string): string;
var query:TQuery;
    Present: TDateTime;
    Year, Month, Day: Word;
    str:string;
begin
  try
      query:=TQuery.Create(nil);
  except
      exit;
  end;
  query.DatabaseName:=DBname;
  query.close;
  query.sql.Clear;
  query.sql.Add('select distinct sysdate from dual');
  try
     query.open;
  except
     exit;
  end;
  query.First;
  Present:=Query.Fields[0].AsDateTime;
  DecodeDate(Present, Year, Month, Day);
  str:=inttostr(year);
  if month<10 then str:=str+'0'+inttostr(month)
  else str:=str+inttostr(month);
  if day<10 then str:=str+'0'+inttostr(day)
  else str:=str+inttostr(day);
  result:=str;
  query.free;
end;

function TDbRestoreFrm.ExecQuery(PQuery: TQuery; PSql: string): Boolean;
begin
  Result:=true;
  PQuery.close;
  PQuery.sql.Clear;
  PQuery.Filtered:=false;
  PQuery.sql.add(PSql);
  try
    PQuery.ExecSQL;
  except
    Result:=false;
  end;
end;

end.