Sonarr/NzbDrone.Common/HttpProvider.cs

142 lines
4.5 KiB
C#
Raw Normal View History

2013-04-10 23:41:45 +00:00
using System;
using System.Collections.Generic;
using System.Diagnostics;
2011-04-25 18:16:38 +00:00
using System.IO;
using System.Net;
using System.Text;
using NLog;
using NzbDrone.Common.EnvironmentInfo;
2010-09-28 03:40:01 +00:00
2012-02-11 00:48:20 +00:00
namespace NzbDrone.Common
2010-09-28 03:40:01 +00:00
{
2013-04-10 23:41:45 +00:00
public interface IHttpProvider
{
string DownloadString(string url);
string DownloadString(string url, string username, string password);
2013-06-03 06:12:31 +00:00
Dictionary<string, string> GetHeader(string url);
2013-04-10 23:41:45 +00:00
Stream DownloadStream(string url, NetworkCredential credential = null);
void DownloadFile(string url, string fileName);
string PostCommand(string address, string username, string password, string command);
}
public class HttpProvider : IHttpProvider
2010-09-28 03:40:01 +00:00
{
private readonly Logger _logger;
public const string CONTENT_LENGTH_HEADER = "Content-Length";
private readonly string _userAgent;
public HttpProvider(Logger logger)
{
_logger = logger;
_userAgent = String.Format("NzbDrone {0}", BuildInfo.Version);
}
public string DownloadString(string url)
{
return DownloadString(url, null);
}
public string DownloadString(string url, string username, string password)
{
return DownloadString(url, new NetworkCredential(username, password));
}
2013-08-20 05:53:18 +00:00
private string DownloadString(string url, ICredentials identity)
{
try
{
var client = new WebClient { Credentials = identity };
client.Headers.Add(HttpRequestHeader.UserAgent, _userAgent);
return client.DownloadString(url);
}
catch (WebException e)
{
_logger.Warn("Failed to get response from: {0} {1}", url, e.Message);
throw;
}
catch (Exception e)
{
_logger.WarnException("Failed to get response from: " + url, e);
2011-04-05 05:30:13 +00:00
throw;
}
2011-04-05 05:30:13 +00:00
}
2013-06-03 06:12:31 +00:00
public Dictionary<string, string> GetHeader(string url)
{
var headers = new Dictionary<string, string>();
var request = WebRequest.Create(url);
request.Method = "HEAD";
var response = request.GetResponse();
2013-06-03 06:12:31 +00:00
foreach (var key in response.Headers.AllKeys)
{
headers.Add(key, response.Headers[key]);
}
return headers;
}
2013-04-10 23:41:45 +00:00
public Stream DownloadStream(string url, NetworkCredential credential = null)
2011-04-05 05:30:13 +00:00
{
var request = (HttpWebRequest)WebRequest.Create(url);
request.UserAgent = _userAgent;
request.Timeout = 20 * 1000;
2011-04-25 18:16:38 +00:00
request.Credentials = credential;
2011-04-25 18:16:38 +00:00
var response = request.GetResponse();
return response.GetResponseStream();
}
2011-04-25 18:16:38 +00:00
2013-04-10 23:41:45 +00:00
public void DownloadFile(string url, string fileName)
{
try
{
var fileInfo = new FileInfo(fileName);
if (fileInfo.Directory != null && !fileInfo.Directory.Exists)
{
fileInfo.Directory.Create();
}
_logger.Trace("Downloading [{0}] to [{1}]", url, fileName);
var stopWatch = Stopwatch.StartNew();
var webClient = new WebClient();
webClient.Headers.Add(HttpRequestHeader.UserAgent, _userAgent);
webClient.DownloadFile(url, fileName);
stopWatch.Stop();
_logger.Trace("Downloading Completed. took {0:0}s", stopWatch.Elapsed.Seconds);
}
catch (WebException e)
{
_logger.Warn("Failed to get response from: {0} {1}", url, e.Message);
throw;
}
catch (Exception e)
{
_logger.WarnException("Failed to get response from: " + url, e);
throw;
}
}
2013-04-10 23:41:45 +00:00
public 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("&nbsp;", " ");
}
2010-09-28 03:40:01 +00:00
}
2010-09-28 05:58:49 +00:00
}