You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
90 lines
3.1 KiB
90 lines
3.1 KiB
using System;
|
|
using System.IO;
|
|
using System.Runtime.InteropServices;
|
|
using System.Text.RegularExpressions;
|
|
using Nancy;
|
|
using NLog;
|
|
using NzbDrone.Common.Disk;
|
|
using NzbDrone.Common.EnvironmentInfo;
|
|
using NzbDrone.Core.Configuration;
|
|
|
|
namespace NzbDrone.Api.Frontend.Mappers
|
|
{
|
|
public class IndexHtmlMapper : StaticResourceMapperBase
|
|
{
|
|
private readonly IDiskProvider _diskProvider;
|
|
private readonly Func<ICacheBreakerProvider> _cacheBreakProviderFactory;
|
|
private readonly string _indexPath;
|
|
private static readonly Regex ReplaceRegex = new Regex("(?<=(?:href|src|data-main)=\").*?(?=\")", RegexOptions.Compiled | RegexOptions.IgnoreCase);
|
|
|
|
private static String API_KEY;
|
|
private static String URL_BASE;
|
|
private string _generatedContent
|
|
;
|
|
|
|
public IndexHtmlMapper(IAppFolderInfo appFolderInfo,
|
|
IDiskProvider diskProvider,
|
|
IConfigFileProvider configFileProvider,
|
|
Func<ICacheBreakerProvider> cacheBreakProviderFactory,
|
|
Logger logger)
|
|
: base(diskProvider, logger)
|
|
{
|
|
_diskProvider = diskProvider;
|
|
_cacheBreakProviderFactory = cacheBreakProviderFactory;
|
|
_indexPath = Path.Combine(appFolderInfo.StartUpFolder, "UI", "index.html");
|
|
|
|
API_KEY = configFileProvider.ApiKey;
|
|
URL_BASE = configFileProvider.UrlBase;
|
|
}
|
|
|
|
public override string Map(string resourceUrl)
|
|
{
|
|
return _indexPath;
|
|
}
|
|
|
|
public override bool CanHandle(string resourceUrl)
|
|
{
|
|
return !resourceUrl.Contains(".");
|
|
}
|
|
|
|
public override Response GetResponse(string resourceUrl)
|
|
{
|
|
var response = base.GetResponse(resourceUrl);
|
|
response.Headers["X-UA-Compatible"] = "IE=edge";
|
|
|
|
return response;
|
|
}
|
|
|
|
protected override Stream GetContentStream(string filePath)
|
|
{
|
|
var text = GetIndexText();
|
|
|
|
var stream = new MemoryStream();
|
|
var writer = new StreamWriter(stream);
|
|
writer.Write(text);
|
|
writer.Flush();
|
|
stream.Position = 0;
|
|
return stream;
|
|
}
|
|
|
|
private string GetIndexText()
|
|
{
|
|
if (RuntimeInfoBase.IsProduction && _generatedContent != null)
|
|
{
|
|
return _generatedContent;
|
|
}
|
|
|
|
_generatedContent = _diskProvider.ReadAllText(_indexPath);
|
|
|
|
var cacheBreakProvider = _cacheBreakProviderFactory();
|
|
|
|
_generatedContent = ReplaceRegex.Replace(_generatedContent, match => cacheBreakProvider.AddCacheBreakerToPath(URL_BASE + match.Value));
|
|
|
|
_generatedContent = _generatedContent.Replace("API_ROOT", URL_BASE + "/api");
|
|
_generatedContent = _generatedContent.Replace("API_KEY", API_KEY);
|
|
_generatedContent = _generatedContent.Replace("APP_VERSION", BuildInfo.Version.ToString());
|
|
|
|
return _generatedContent;
|
|
}
|
|
}
|
|
} |