ShareX/ShareX.HelpersLib/CLI/ExternalCLIManager.cs
2018-01-01 17:59:14 +03:00

131 lines
4.2 KiB
C#

#region License Information (GPL v3)
/*
ShareX - A program that allows you to take screenshots and share any file type
Copyright (c) 2007-2018 ShareX Team
This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 2
of the License, or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
Optionally you can also view the license at <http://www.gnu.org/licenses/>.
*/
#endregion License Information (GPL v3)
using System;
using System.Diagnostics;
using System.IO;
using System.Text;
namespace ShareX.HelpersLib
{
public abstract class ExternalCLIManager : IDisposable
{
public event DataReceivedEventHandler OutputDataReceived;
public event DataReceivedEventHandler ErrorDataReceived;
protected Process process;
protected bool processRunning;
public virtual int Open(string path, string args = null)
{
DebugHelper.WriteLine("CLI: \"{0}\" {1}", path, args);
if (File.Exists(path))
{
using (process = new Process())
{
ProcessStartInfo psi = new ProcessStartInfo(path);
psi.UseShellExecute = false;
psi.CreateNoWindow = true;
psi.RedirectStandardInput = true;
psi.RedirectStandardOutput = true;
psi.RedirectStandardError = true;
psi.Arguments = args;
psi.WorkingDirectory = Path.GetDirectoryName(path);
psi.StandardOutputEncoding = Encoding.UTF8;
psi.StandardErrorEncoding = Encoding.UTF8;
process.EnableRaisingEvents = true;
if (psi.RedirectStandardOutput) process.OutputDataReceived += cli_OutputDataReceived;
if (psi.RedirectStandardError) process.ErrorDataReceived += cli_ErrorDataReceived;
process.StartInfo = psi;
process.Start();
if (psi.RedirectStandardOutput) process.BeginOutputReadLine();
if (psi.RedirectStandardError) process.BeginErrorReadLine();
try
{
processRunning = true;
process.WaitForExit();
}
finally
{
processRunning = false;
}
return process.ExitCode;
}
}
return -1;
}
private void cli_OutputDataReceived(object sender, DataReceivedEventArgs e)
{
if (e.Data != null)
{
if (OutputDataReceived != null)
{
OutputDataReceived(sender, e);
}
}
}
private void cli_ErrorDataReceived(object sender, DataReceivedEventArgs e)
{
if (e.Data != null)
{
if (ErrorDataReceived != null)
{
ErrorDataReceived(sender, e);
}
}
}
public void WriteInput(string input)
{
if (processRunning && process != null && process.StartInfo != null && process.StartInfo.RedirectStandardInput)
{
process.StandardInput.WriteLine(input);
}
}
public virtual void Close()
{
if (processRunning && process != null)
{
process.CloseMainWindow();
}
}
public void Dispose()
{
if (process != null)
{
process.Dispose();
}
}
}
}