public bool Clone(bool full)
{
if (Workspace != null)
return false;
try
{
if (!full)
{
ProtoBuf.Serializer.SerializeWithLengthPrefix<NetCommand>(Connection.GetStream(), new NetCommand() { Type = NetCommandType.Clone }, ProtoBuf.PrefixStyle.Fixed32);
var clonePack = Utilities.ReceiveEncrypted<ClonePayload>(SharedInfo);
Workspace = Area.InitRemote(BaseDirectory, clonePack, SkipContainmentCheck);
}
else
{
ProtoBuf.Serializer.SerializeWithLengthPrefix<NetCommand>(Connection.GetStream(), new NetCommand() { Type = NetCommandType.FullClone }, ProtoBuf.PrefixStyle.Fixed32);
var response = ProtoBuf.Serializer.DeserializeWithLengthPrefix<NetCommand>(Connection.GetStream(), ProtoBuf.PrefixStyle.Fixed32);
if (response.Type == NetCommandType.Acknowledge)
{
int dbVersion = (int)response.Identifier;
if (!WorkspaceDB.AcceptRemoteDBVersion(dbVersion))
{
Printer.PrintError("Server database version is incompatible (v{0}). Use non-full clone to perform the operation.", dbVersion);
ProtoBuf.Serializer.SerializeWithLengthPrefix<NetCommand>(Connection.GetStream(), new NetCommand() { Type = NetCommandType.Error }, ProtoBuf.PrefixStyle.Fixed32);
return false;
}
else
ProtoBuf.Serializer.SerializeWithLengthPrefix<NetCommand>(Connection.GetStream(), new NetCommand() { Type = NetCommandType.Acknowledge }, ProtoBuf.PrefixStyle.Fixed32);
}
System.IO.FileInfo fsInfo = new System.IO.FileInfo(System.IO.Path.GetRandomFileName());
Printer.PrintMessage("Attempting to import metadata file to temp path {0}", fsInfo.FullName);
var printer = Printer.CreateSimplePrinter("Progress", (obj) =>
{
return string.Format("#b#{0}## received.", Versionr.Utilities.Misc.FormatSizeFriendly((long)obj));
});
try
{
long total = 0;
using (var stream = fsInfo.OpenWrite())
{
while (true)
{
var data = Utilities.ReceiveEncrypted<DataPayload>(SharedInfo);
stream.Write(data.Data, 0, data.Data.Length);
total += data.Data.Length;
printer.Update(total);
if (data.EndOfStream)
break;
}
printer.End(total);
response = ProtoBuf.Serializer.DeserializeWithLengthPrefix<NetCommand>(Connection.GetStream(), ProtoBuf.PrefixStyle.Fixed32);
if (response.Type == NetCommandType.Error)
{
Printer.PrintError("Server failed to clone the database.");
return false;
}
}
Printer.PrintMessage("Metadata written, importing DB.");
Area area = new Area(Area.GetAdminFolderForDirectory(BaseDirectory));
try
{
fsInfo.MoveTo(area.MetadataFile.FullName);
if (!area.ImportDB())
throw new Exception("Couldn't import data.");
Workspace = Area.Load(BaseDirectory);
SharedInfo.Workspace = Workspace;
return true;
}
catch
{
if (area.MetadataFile.Exists)
area.MetadataFile.Delete();
area.AdministrationFolder.Delete();
throw;
}
}
catch
{
if (fsInfo.Exists)
fsInfo.Delete();
return false;
}
}
SharedInfo.Workspace = Workspace;
return true;
}
catch (Exception e)
{
Printer.PrintError(e.ToString());
return false;
}
}
public AreaVM(string path, string name, AreaInitMode areaInitMode, string host = null, int port = 0)
{
RefreshCommand = new DelegateCommand(Refresh);
PullCommand = new DelegateCommand(Pull);
PushCommand = new DelegateCommand(Push);
_name = name;
DirectoryInfo dir = new DirectoryInfo(path);
switch (areaInitMode)
{
case AreaInitMode.Clone:
// Spawn another dialog for the source (or put it in the Clone New button)
Client client = new Client(dir);
if (client.Connect(host, port, null, true))
{
bool result = client.Clone(true);
if (!result)
result = client.Clone(false);
if (result)
{
string remoteName = "default";
client.Workspace.SetRemote(client.Host, client.Port, client.Module, remoteName);
client.Pull(false, client.Workspace.CurrentBranch.ID.ToString());
_area = Area.Load(client.Workspace.Root);
_area.Checkout(null, false, false);
client.SyncRecords();
}
}
else
{
MessageBox.Show(string.Format("Couldn't connect to {0}:{1}", host, port), "Clone Failed", MessageBoxButton.OK, MessageBoxImage.Error);
}
client.Close();
break;
case AreaInitMode.InitNew:
// Tell versionr to initialize at path
try
{
dir.Create();
}
catch
{
MessageBox.Show("Error - couldn't create subdirectory \"{0}\"", dir.FullName);
break;
}
_area = Area.Init(dir, name);
break;
case AreaInitMode.UseExisting:
// Add it to settings and refresh UI, get status etc.
_area = Area.Load(dir);
break;
}
}
