commit
52da4e4423
@ -0,0 +1,43 @@
|
|||||||
|
using System.Collections.Generic;
|
||||||
|
using System.Linq;
|
||||||
|
using NzbDrone.Common.Crypto;
|
||||||
|
using NzbDrone.Common.Extensions;
|
||||||
|
|
||||||
|
namespace NzbDrone.Api.Frontend.Mappers
|
||||||
|
{
|
||||||
|
public interface ICacheBreakerProvider
|
||||||
|
{
|
||||||
|
string AddCacheBreakerToPath(string resourceUrl);
|
||||||
|
}
|
||||||
|
|
||||||
|
public class CacheBreakerProvider : ICacheBreakerProvider
|
||||||
|
{
|
||||||
|
private readonly IEnumerable<IMapHttpRequestsToDisk> _diskMappers;
|
||||||
|
private readonly IHashProvider _hashProvider;
|
||||||
|
|
||||||
|
public CacheBreakerProvider(IEnumerable<IMapHttpRequestsToDisk> diskMappers, IHashProvider hashProvider)
|
||||||
|
{
|
||||||
|
_diskMappers = diskMappers;
|
||||||
|
_hashProvider = hashProvider;
|
||||||
|
}
|
||||||
|
|
||||||
|
public string AddCacheBreakerToPath(string resourceUrl)
|
||||||
|
{
|
||||||
|
if (!ShouldBreakCache(resourceUrl))
|
||||||
|
{
|
||||||
|
return resourceUrl;
|
||||||
|
}
|
||||||
|
|
||||||
|
var mapper = _diskMappers.Single(m => m.CanHandle(resourceUrl));
|
||||||
|
var pathToFile = mapper.Map(resourceUrl);
|
||||||
|
var hash = _hashProvider.ComputeMd5(pathToFile).ToBase64();
|
||||||
|
|
||||||
|
return resourceUrl + "?h=" + hash;
|
||||||
|
}
|
||||||
|
|
||||||
|
private static bool ShouldBreakCache(string path)
|
||||||
|
{
|
||||||
|
return !path.EndsWith(".ics") && !path.EndsWith("main");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,31 @@
|
|||||||
|
using System.Security.Cryptography;
|
||||||
|
using NzbDrone.Common.Disk;
|
||||||
|
|
||||||
|
namespace NzbDrone.Common.Crypto
|
||||||
|
{
|
||||||
|
public interface IHashProvider
|
||||||
|
{
|
||||||
|
byte[] ComputeMd5(string path);
|
||||||
|
}
|
||||||
|
|
||||||
|
public class HashProvider : IHashProvider
|
||||||
|
{
|
||||||
|
private readonly IDiskProvider _diskProvider;
|
||||||
|
|
||||||
|
public HashProvider(IDiskProvider diskProvider)
|
||||||
|
{
|
||||||
|
_diskProvider = diskProvider;
|
||||||
|
}
|
||||||
|
|
||||||
|
public byte[] ComputeMd5(string path)
|
||||||
|
{
|
||||||
|
using (var md5 = MD5.Create())
|
||||||
|
{
|
||||||
|
using (var stream = _diskProvider.StreamFile(path))
|
||||||
|
{
|
||||||
|
return md5.ComputeHash(stream);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,17 @@
|
|||||||
|
using System;
|
||||||
|
|
||||||
|
namespace NzbDrone.Common.Extensions
|
||||||
|
{
|
||||||
|
public static class Base64Extentions
|
||||||
|
{
|
||||||
|
public static string ToBase64(this byte[] bytes)
|
||||||
|
{
|
||||||
|
return Convert.ToBase64String(bytes);
|
||||||
|
}
|
||||||
|
|
||||||
|
public static string ToBase64(this long input)
|
||||||
|
{
|
||||||
|
return BitConverter.GetBytes(input).ToBase64();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in new issue