Sonarr/NzbDrone.Common/HttpProvider.cs

102 lines
3.2 KiB
C#

using System.Linq;
using System;
using System.Diagnostics;
using System.IO;
using System.Net;
using System.Text;
using NLog;
namespace NzbDrone.Common
{
public class HttpProvider
{
private readonly EnvironmentProvider _environmentProvider;
private static readonly Logger logger = LogManager.GetCurrentClassLogger();
public HttpProvider(EnvironmentProvider environmentProvider)
{
_environmentProvider = environmentProvider;
}
public HttpProvider()
{
}
public virtual string DownloadString(string address)
{
return DownloadString(address, null);
}
public virtual string DownloadString(string address, string username, string password)
{
return DownloadString(address, new NetworkCredential(username, password));
}
public virtual string DownloadString(string address, ICredentials identity)
{
try
{
var client = new WebClient { Credentials = identity };
return client.DownloadString(address);
}
catch (Exception ex)
{
logger.Trace(ex.Message, ex.ToString());
throw;
}
}
public virtual Stream DownloadStream(string url, NetworkCredential credential)
{
var request = (HttpWebRequest)WebRequest.Create(url);
request.UserAgent = String.Format("NzbDrone {0}", _environmentProvider.Version);
request.Credentials = credential;
var response = request.GetResponse();
return response.GetResponseStream();
}
public virtual void DownloadFile(string url, string fileName)
{
try
{
var fileInfo = new FileInfo(fileName);
if (!fileInfo.Directory.Exists)
{
fileInfo.Directory.Create();
}
logger.Trace("Downloading [{0}] to [{1}]", url, fileName);
var stopWatch = Stopwatch.StartNew();
var webClient = new WebClient();
webClient.DownloadFile(url, fileName);
stopWatch.Stop();
logger.Trace("Downloading Completed. took {0:0}s", stopWatch.Elapsed.Seconds);
}
catch (Exception ex)
{
logger.Warn("Failed to get response from: {0}", url);
logger.TraceException(ex.Message, ex);
throw;
}
}
public virtual string PostCommand(string address, string username, string password, string command)
{
address = String.Format("http://{0}/jsonrpc", address);
logger.Trace("Posting command: {0}, to {1}", command, address);
byte[] byteArray = Encoding.ASCII.GetBytes(command);
var wc = new WebClient();
wc.Credentials = new NetworkCredential(username, password);
var response = wc.UploadData(address, "POST", byteArray);
var text = Encoding.ASCII.GetString(response);
return text.Replace(" ", " ");
}
}
}