source: trunk/USvnZero.pas

unit USvnZero;

interface

uses
  Classes, SysUtils, FileUtil, USubversion, UCommon;

type

  { TSvnZero }

  TSvnZero = class(TSubversion)
  private
    Subversion: TSubversion;
    FWorkingCopyRootPath: string;
    FWorkingCopyRootPathCurrentDir: string;
    function GetWorkingCopyRootPath(FileName: string): string;
    function CheckErrorOutput(Text: string): Boolean;
    function GetFileInfoByMd5(Md5Hash: string): TSubversionInfo;
    function GetFileInfoBySha1(Sha1Hash: string): TSubversionInfo;
    procedure ZeroPristine;
    procedure RestorePristine;
    function GetWorkingCopyRootPathFromText(Text: string): string;
    function GetSvnDir(FileName: string): string;
    function GetPristineDir(FileName: string): string;
    function GetCommand(Parameters: array of string): string;
    function IsWorkingCopyCommand(Command: string): Boolean;
  public
    procedure Execute(Parameters: array of string); override;
    constructor Create;
    destructor Destroy; override;
  end;


implementation

uses
  SQLite3, SQLite3Wrap;

const
  SvnBaseExt = '.svn-base';

resourcestring
  SZeroedFilesCount = 'Zeroed files count: %d';
  SPristineRecoveryFail = 'Pristine file recovery not successful';

procedure TSvnZero.ZeroPristine;
var
  Files: TStrings;
  I: Integer;
  Dir: string;
  FileName: string;
  ZeroedFilesCount: Integer;
begin
  ZeroedFilesCount := 0;
  Dir := GetPristineDir('.');
  if DirectoryExists(Dir) then
  try
    Files := FindAllFiles(Dir, '*' + SvnBaseExt);
    for I := 0 to Files.Count - 1 do begin
      FileName := Files[I];
      DeleteFile(FileName);
      MakeFileWriteable(FileName);
      Inc(ZeroedFilesCount);
    end;
  finally
    Files.Free;
  end;
  //StandardOutput := StandardOutput + Format(SZeroedFilesCount, [ZeroedFilesCount]) + LineEnding;
end;

procedure TSvnZero.RestorePristine;
begin
end;

function TSvnZero.GetWorkingCopyRootPathFromText(Text: string): string;
var
  Index: Integer;
const
  StartText = 'Working Copy Root Path: ';
begin
  Result := '';
  Index := Pos(StartText, Text);
  if Index > 0 then begin
    Result := Copy(Text, Index + Length(StartText), MaxInt);
    Index := Pos(LineEnding, Result);
    if Index > 0 then begin
      Delete(Result, Index, MaxInt);
    end;
  end;
end;

function TSvnZero.GetSvnDir(FileName: string): string;
begin
  Result := GetWorkingCopyRootPath(FileName) + DirectorySeparator + '.svn';
end;

function TSvnZero.GetPristineDir(FileName: string): string;
begin
  Result := GetSvnDir(FileName) + DirectorySeparator + 'pristine';
end;

function TSvnZero.GetCommand(Parameters: array of string): string;
var
  I: Integer;
begin
  Result := '';
  I := 0;
  while (I < Length(Parameters)) do begin
    if Copy(Parameters[I], 1, 1) <> '-' then begin
      Result := Parameters[I];
      Break;
    end;
    Inc(I);
  end;
end;

function TSvnZero.IsWorkingCopyCommand(Command: string): Boolean;
begin
  Result := (Command = 'update') or (Command = 'commit');
end;

procedure TSvnZero.Execute(Parameters: array of string);
var
  StandardOutputMerged: string;
  LastErrorOutput: string;
begin
  Subversion.PrintOutput := PrintOutput;
  Subversion.PrintCommand := PrintCommand;
  LastErrorOutput := '';
  StandardOutputMerged := '';
  while True do begin
    Subversion.Execute(Parameters);
    ErrorOutput := Subversion.ErrorOutput;
    StandardOutputMerged := StandardOutputMerged + Subversion.StandardOutput;
    if CheckErrorOutput(ErrorOutput) then begin
      if LastErrorOutput = ErrorOutput then begin
        WriteLn(SPristineRecoveryFail);
        Break;
      end;
    end else Break;
    LastErrorOutput := ErrorOutput;
  end;
  if StandardOutputMerged <> '' then StandardOutput := StandardOutputMerged;
  if IsWorkingCopyCommand(GetCommand(Parameters)) then ZeroPristine;
end;

constructor TSvnZero.Create;
begin
  Subversion := TSubversion.Create;
end;

destructor TSvnZero.Destroy;
begin
  FreeAndNil(Subversion);
  inherited;
end;

function TSvnZero.GetWorkingCopyRootPath(FileName: string): string;
begin
  if (FWorkingCopyRootPathCurrentDir <> GetCurrentDir) or (FWorkingCopyRootPath = '') then begin
    Subversion.Execute(['info', FileName]);
    FWorkingCopyRootPath := GetWorkingCopyRootPathFromText(Subversion.StandardOutput);
    FWorkingCopyRootPathCurrentDir := GetCurrentDir;
  end;
  Result := FWorkingCopyRootPath;
end;

function TSvnZero.CheckErrorOutput(Text: string): Boolean;
var
  FileName: string;
  I: Integer;
  ExportedFileName: string;
  PristineFileName: string;
  Info: TSubversionInfo;
  Md5Hash: string;
  Sha1Hash: string;
const
  StartText = 'expected:';
  StartText2 = 'svn: E000002: Can''t open file ''';
begin
  Result := False;
  I := Pos(StartText2, Text);
  if I > 0 then begin
    Sha1Hash := Copy(Text, I + Length(StartText2), MaxInt);
    I := Pos('.svn-base', Sha1Hash);
    if I > 0 then
      Delete(Sha1Hash, I, MaxInt);
    I := LastPos(DirectorySeparator, Sha1Hash);
    if I > 0 then
      Delete(Sha1Hash, 1, I);

    Info := GetFileInfoBySha1(Sha1Hash);
    ExportedFileName := GetSvnDir('.') + DirectorySeparator + 'tmp' +
      DirectorySeparator + 'ExportedFile.bin';
    PristineFileName := GetPristineDir('.') + DirectorySeparator + Copy(Info.Checksum, 1, 2) +
      DirectorySeparator + Info.Checksum + '.svn-base';
    Subversion.Execute(['export', '-r', Info.Revision, '--force', Info.URL, ExportedFileName]);
    MakeFileWriteable(PristineFileName);
    CopyFile(ExportedFileName, PristineFileName);
    Subversion.Execute(['cleanup']);
    Result := True;
  end;

(*  if Pos(': Checksum mismatch', Text) > 0 then begin
    I := Pos(StartText, Text);
    if I > 0 then begin
      Md5Hash := Copy(Text, I + Length(StartText), MaxInt);
      I := Pos(#10, Md5Hash);
      if I > 0 then
        Delete(Md5Hash, I, MaxInt);
      Md5Hash := Trim(Md5Hash);
      Info := GetFileInfoByMd5(Md5Hash);
      ExportedFileName := GetSvnDir('.') + DirectorySeparator + 'tmp' +
        DirectorySeparator + 'ExportedFile.bin';
      PristineFileName := GetPristineDir('.') + DirectorySeparator + Copy(Info.Checksum, 1, 2) +
        DirectorySeparator + Info.Checksum + '.svn-base';
      Subversion.Execute(['export', '-r', Info.Revision, '--force', Info.URL, ExportedFileName]);
      MakeFileWriteable(PristineFileName);
      CopyFile(ExportedFileName, PristineFileName);
      Subversion.Execute(['cleanup']);
      Result := True;
    end;
 {   I := Pos('''', Text);
    if I > 0 then begin
      FileName := Copy(Text, I + 1, MaxInt);
      I := Pos('''', FileName);
      if I > 0 then
        Delete(FileName, I, MaxInt);

      Subversion.Execute(['info', FileName]);
      Info := GetInfo(Subversion.StandardOutput);
      ExportedFileName := GetSvnDir(FileName) + DirectorySeparator + 'tmp' +
        DirectorySeparator + 'ExportedFile.bin';
      PristineFileName := GetPristineDir(FileName) + DirectorySeparator + Copy(Info.Checksum, 1, 2) +
        DirectorySeparator + Info.Checksum + '.svn-base';
      Subversion.Execute(['export', '-r', Info.Revision, '--force', Info.URL, ExportedFileName]);
      MakeFileWriteable(PristineFileName);
      CopyFile(ExportedFileName, PristineFileName);
      Subversion.Execute(['cleanup']);
      Result := True;
    end;
    }
  end;
*)
end;

function TSvnZero.GetFileInfoByMd5(Md5Hash: string): TSubversionInfo;
var
  Database: TSQLite3Database;
  Statement: TSQLite3Statement;
begin
  Database := TSQLite3Database.Create;
  try
    Database.Open(GetSvnDir('.') + DirectorySeparator + 'wc.db');
    Statement := Database.Prepare('SELECT checksum FROM PRISTINE WHERE md5_checksum=''$md5 $' + Md5Hash + '''');
    try
      while Statement.Step = SQLITE_ROW do begin
        Result.Checksum := Statement.ColumnText(0);
      end;
    finally
      Statement.Free;
    end;

    Statement := Database.Prepare('SELECT repos_path,changed_revision FROM NODES WHERE checksum="' + Result.Checksum + '" LIMIT 1');
    try
      while Statement.Step = SQLITE_ROW do begin
        Result.Path := Statement.ColumnText(0);
        Result.Revision := Statement.ColumnText(1);
      end;
    finally
      Statement.Free;
    end;

    Statement := Database.Prepare('SELECT root FROM REPOSITORY LIMIT 1');
    try
      while Statement.Step = SQLITE_ROW do begin
        Result.URL := Statement.ColumnText(0) + '/' + Result.Path;
      end;
    finally
      Statement.Free;
    end;

    if Result.Checksum.StartsWith('$sha1$') then
      Result.Checksum := Copy(Result.Checksum, 7);
  finally
    Database.Free;
  end;
end;

function TSvnZero.GetFileInfoBySha1(Sha1Hash: string): TSubversionInfo;
var
  Database: TSQLite3Database;
  Statement: TSQLite3Statement;
begin
  Database := TSQLite3Database.Create;
  try
    Database.Open(GetSvnDir('.') + DirectorySeparator + 'wc.db');
    Result.Checksum := Sha1Hash;

    Statement := Database.Prepare('SELECT repos_path,changed_revision FROM NODES WHERE checksum="$sha1$' + Result.Checksum + '" LIMIT 1');
    try
      while Statement.Step = SQLITE_ROW do begin
        Result.Path := Statement.ColumnText(0);
        Result.Revision := Statement.ColumnText(1);
      end;
    finally
      Statement.Free;
    end;

    Statement := Database.Prepare('SELECT root FROM REPOSITORY LIMIT 1');
    try
      while Statement.Step = SQLITE_ROW do begin
        Result.URL := Statement.ColumnText(0) + '/' + Result.Path;
      end;
    finally
      Statement.Free;
    end;

    if Result.Checksum.StartsWith('$sha1$') then
      Result.Checksum := Copy(Result.Checksum, 7);
  finally
    Database.Free;
  end;
end;

end.