Skip to content

Add ability to upload from stream, and name remote file differently #237

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
pnorms wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Add ability to upload from stream, and name remote file differently #237

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
276 changes: 202 additions & 74 deletions PoshSSH/PoshSSH/SetSftpFile.cs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -56,11 +56,25 @@ public string RemotePath
set { _remotepath = value; }
}

/// <summary>
/// File name on remote target, required when file stream is used.
/// </summary>
private String _remotefilename;
[ValidateNotNullOrEmpty]
[Parameter(Mandatory = false,
ValueFromPipelineByPropertyName = true,
Position = 4)]
public string RemoteFileName
{
get { return _remotefilename; }
set { _remotefilename = value; }
}

/// <summary>
/// The local file to be uploaded.
/// </summary>
private String[] _localfile;
[Parameter(Mandatory = true,
[Parameter(Mandatory = false,
ValueFromPipelineByPropertyName = true,
Position = 1)]
[Alias("PSPath")]
Expand All @@ -70,6 +84,20 @@ public String[] LocalFile
set { _localfile = value; }
}

/// <summary>
/// The local file to be uploaded.
/// </summary>
private MemoryStream _localfilestream;
[Parameter(Mandatory = false,
ValueFromPipelineByPropertyName = true,
Position = 1)]
[Alias("PSStream")]
public MemoryStream LocalFileStream
{
get { return _localfilestream; }
set { _localfilestream = value; }
}

/// <summary>
/// If a file on the target should be overwritten or not.
/// </summary>
Expand Down Expand Up @@ -119,112 +147,212 @@ protected override void ProcessRecord()
{
foreach (var sftpSession in ToProcess)
{
// check if the file specified actually exists.
// Resolve the path even if a relative one is given.
foreach (var localfile in _localfile)
// Check for local file or local file stream
if ((_localfilestream != null) && (_remotefilename != null))
{
ProviderInfo provider;
var pathinfo = GetResolvedProviderPathFromPSPath(localfile, out provider);
var localfullPath = pathinfo[0];
var remoteFullpath = RemotePath.TrimEnd(new[] { '/' }) + "/" + _remotefilename;
WriteVerbose("Uploading to " + remoteFullpath + " on " + sftpSession.Host);

// Setup Action object for showing download progress.

if (File.Exists(@localfullPath))
var res = new Action<ulong>(rs =>
{
WriteVerbose("Uploading " + localfullPath);
var fil = new FileInfo(@localfullPath);
var remoteFullpath = RemotePath.TrimEnd(new[] { '/' }) + "/" + fil.Name;
WriteVerbose("Uploading to " + remoteFullpath + " on " + sftpSession.Host);
//if (!MyInvocation.BoundParameters.ContainsKey("Verbose")) return;
if (_localfilestream.Length > 1240000)
{
var percent = (int)((((double)rs) / _localfilestream.Length) * 100.0);
if (percent % 10 == 0)
{
// This will prevent the progress message from being stuck on the screen.
if (percent == 90 || percent > 90)
{
return;
}

// Setup Action object for showing download progress.
var progressRecord = new ProgressRecord(1,
"Uploading ",
String.Format("{0} Bytes Uploaded of {1}", rs, _localfilestream.Length))
{ PercentComplete = percent };

var res = new Action<ulong>(rs =>
{
//if (!MyInvocation.BoundParameters.ContainsKey("Verbose")) return;
if (fil.Length > 1240000)
Host.UI.WriteProgress(1, progressRecord);
//Host.UI.WriteVerboseLine(percent.ToString(CultureInfo.InvariantCulture) + "% Completed.");
}
}
});

// Check that the path we are uploading to actually exists on the target.
if (sftpSession.Session.Exists(RemotePath))
{
var percent = (int)((((double)rs) / fil.Length) * 100.0);
if (percent % 10 == 0)
// Ensure the remote path is a directory.
var attribs = sftpSession.Session.GetAttributes(RemotePath);
if (!attribs.IsDirectory)
{
throw new SftpPathNotFoundException("Specified path is not a directory");
}
// Check if the file already exists on the target system.
var present = sftpSession.Session.Exists(remoteFullpath);
if ((present & _overwrite) || (!present))
{
// This will prevent the progress message from being stuck on the screen.
if (percent == 90 || percent > 90)
try
{
return;
sftpSession.Session.UploadFile(_localfilestream, remoteFullpath, res);
_localfilestream.Close();
}
catch (Exception ex)
{
_localfilestream.Close();
WriteError(new ErrorRecord(
ex,
"Error while Uploading",
ErrorCategory.InvalidOperation,
sftpSession));

var progressRecord = new ProgressRecord(1,
"Uploading " + fil.Name,
String.Format("{0} Bytes Uploaded of {1}", rs, fil.Length))
{ PercentComplete = percent };

Host.UI.WriteProgress(1, progressRecord);
//Host.UI.WriteVerboseLine(percent.ToString(CultureInfo.InvariantCulture) + "% Completed.");
}
}
else
{
var ex = new SftpPermissionDeniedException("File already exists on remote host.");
WriteError(new ErrorRecord(
ex,
"File already exists on remote host",
ErrorCategory.InvalidOperation,
sftpSession));
}

}
});
else
{
var ex = new SftpPathNotFoundException(RemotePath + " does not exist.");
WriteError(new ErrorRecord(
ex,
RemotePath + " does not exist.",
ErrorCategory.InvalidOperation,
sftpSession));
}
}
else if (_localfile != null)
{
// check if the file specified actually exists.
// Resolve the path even if a relative one is given.
foreach (var localfile in _localfile)
{
ProviderInfo provider;
var pathinfo = GetResolvedProviderPathFromPSPath(localfile, out provider);
var localfullPath = pathinfo[0];

// Check that the path we are uploading to actually exists on the target.
if (sftpSession.Session.Exists(RemotePath))

if (File.Exists(@localfullPath))
{
// Ensure the remote path is a directory.
var attribs = sftpSession.Session.GetAttributes(RemotePath);
if (!attribs.IsDirectory)
WriteVerbose("Uploading " + localfullPath);
var fil = new FileInfo(@localfullPath);

// Check for remote name overwrite
if (_remotefilename == null)
_remotefilename = fil.Name;

var remoteFullpath = RemotePath.TrimEnd(new[] { '/' }) + "/" + _remotefilename;
WriteVerbose("Uploading to " + remoteFullpath + " on " + sftpSession.Host);

// Setup Action object for showing download progress.

var res = new Action<ulong>(rs =>
{
throw new SftpPathNotFoundException("Specified path is not a directory");
}
// Check if the file already exists on the target system.
var present = sftpSession.Session.Exists(remoteFullpath);
if ((present & _overwrite) || (!present))
//if (!MyInvocation.BoundParameters.ContainsKey("Verbose")) return;
if (fil.Length > 1240000)
{
var percent = (int)((((double)rs) / fil.Length) * 100.0);
if (percent % 10 == 0)
{
// This will prevent the progress message from being stuck on the screen.
if (percent == 90 || percent > 90)
{
return;
}

var progressRecord = new ProgressRecord(1,
"Uploading " + fil.Name,
String.Format("{0} Bytes Uploaded of {1}", rs, fil.Length))
{ PercentComplete = percent };

Host.UI.WriteProgress(1, progressRecord);
//Host.UI.WriteVerboseLine(percent.ToString(CultureInfo.InvariantCulture) + "% Completed.");
}
}
});

// Check that the path we are uploading to actually exists on the target.
if (sftpSession.Session.Exists(RemotePath))
{
var localstream = File.OpenRead(localfullPath);
try
// Ensure the remote path is a directory.
var attribs = sftpSession.Session.GetAttributes(RemotePath);
if (!attribs.IsDirectory)
{
sftpSession.Session.UploadFile(localstream, remoteFullpath, res);
localstream.Close();
throw new SftpPathNotFoundException("Specified path is not a directory");
}
catch (Exception ex)
// Check if the file already exists on the target system.
var present = sftpSession.Session.Exists(remoteFullpath);
if ((present & _overwrite) || (!present))
{
localstream.Close();
WriteError(new ErrorRecord(
ex,
"Error while Uploading",
ErrorCategory.InvalidOperation,
sftpSession));
var localstream = File.OpenRead(localfullPath);
try
{
sftpSession.Session.UploadFile(localstream, remoteFullpath, res);
localstream.Close();
}
catch (Exception ex)
{
localstream.Close();
WriteError(new ErrorRecord(
ex,
"Error while Uploading",
ErrorCategory.InvalidOperation,
sftpSession));

}
}
else
{
var ex = new SftpPermissionDeniedException("File already exists on remote host.");
WriteError(new ErrorRecord(
ex,
"File already exists on remote host",
ErrorCategory.InvalidOperation,
sftpSession));
}

}
else
{
var ex = new SftpPermissionDeniedException("File already exists on remote host.");
var ex = new SftpPathNotFoundException(RemotePath + " does not exist.");
WriteError(new ErrorRecord(
ex,
"File already exists on remote host",
ErrorCategory.InvalidOperation,
sftpSession));
ex,
RemotePath + " does not exist.",
ErrorCategory.InvalidOperation,
sftpSession));
}

}
else
{
var ex = new SftpPathNotFoundException(RemotePath + " does not exist.");
var ex = new FileNotFoundException("File to upload " + localfullPath + " was not found.");

WriteError(new ErrorRecord(
ex,
RemotePath + " does not exist.",
ErrorCategory.InvalidOperation,
sftpSession));
}
ex,
"File to upload " + localfullPath + " was not found.",
ErrorCategory.InvalidOperation,
localfullPath));
} // check if file exists.
} // foreach local file
}
else
{
var ex = new FileNotFoundException("File path or File Stream AND Remote File Name was not specified.");

}
else
{
var ex = new FileNotFoundException("File to upload " + localfullPath + " was not found.");

WriteError(new ErrorRecord(
ex,
"File to upload " + localfullPath + " was not found.",
ErrorCategory.InvalidOperation,
localfullPath));
} // check if file exists.
} // foreach local file
WriteError(new ErrorRecord(
ex,
"File path or File Stream AND Remote File Name was not specified.",
ErrorCategory.InvalidOperation, sftpSession));
}
} // sftp session.
} // Process Record.
}
Expand Down