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.
116 lines
4.1 KiB
116 lines
4.1 KiB
using System;
|
|
using System.IO;
|
|
using System.Text.RegularExpressions;
|
|
using Nancy;
|
|
using NLog;
|
|
using NzbDrone.Common.Disk;
|
|
using NzbDrone.Common.EnvironmentInfo;
|
|
using NzbDrone.Core.Analytics;
|
|
using NzbDrone.Core.Configuration;
|
|
|
|
namespace NzbDrone.Api.Frontend.Mappers
|
|
{
|
|
public class IndexHtmlMapper : StaticResourceMapperBase
|
|
{
|
|
private readonly IDiskProvider _diskProvider;
|
|
private readonly IConfigFileProvider _configFileProvider;
|
|
private readonly IAnalyticsService _analyticsService;
|
|
private readonly Func<ICacheBreakerProvider> _cacheBreakProviderFactory;
|
|
private readonly string _indexPath;
|
|
private static readonly Regex ReplaceRegex = new Regex(@"(?:(?<attribute>href|src)=\"")(?<path>.*?(?<extension>css|js|png|ico|ics))(?:\"")(?:\s(?<nohash>data-no-hash))?", 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,
|
|
IAnalyticsService analyticsService,
|
|
Func<ICacheBreakerProvider> cacheBreakProviderFactory,
|
|
Logger logger)
|
|
: base(diskProvider, logger)
|
|
{
|
|
_diskProvider = diskProvider;
|
|
_configFileProvider = configFileProvider;
|
|
_analyticsService = analyticsService;
|
|
_cacheBreakProviderFactory = cacheBreakProviderFactory;
|
|
_indexPath = Path.Combine(appFolderInfo.StartUpFolder, _configFileProvider.UiFolder, "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(".") && !resourceUrl.StartsWith("/login");
|
|
}
|
|
|
|
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 (RuntimeInfo.IsProduction && _generatedContent != null)
|
|
{
|
|
return _generatedContent;
|
|
}
|
|
|
|
var text = _diskProvider.ReadAllText(_indexPath);
|
|
|
|
var cacheBreakProvider = _cacheBreakProviderFactory();
|
|
|
|
text = ReplaceRegex.Replace(text, match =>
|
|
{
|
|
string url;
|
|
|
|
if (match.Groups["nohash"].Success)
|
|
{
|
|
url = match.Groups["path"].Value;
|
|
}
|
|
|
|
else
|
|
{
|
|
url = cacheBreakProvider.AddCacheBreakerToPath(match.Groups["path"].Value);
|
|
}
|
|
|
|
return string.Format("{0}=\"{1}{2}\"", match.Groups["attribute"].Value, URL_BASE, url);
|
|
});
|
|
|
|
text = text.Replace("API_ROOT", URL_BASE + "/api");
|
|
text = text.Replace("API_KEY", API_KEY);
|
|
text = text.Replace("APP_VERSION", BuildInfo.Version.ToString());
|
|
text = text.Replace("APP_BRANCH", _configFileProvider.Branch.ToLower());
|
|
text = text.Replace("APP_ANALYTICS", _analyticsService.IsEnabled.ToString().ToLowerInvariant());
|
|
text = text.Replace("URL_BASE", URL_BASE);
|
|
text = text.Replace("PRODUCTION", RuntimeInfo.IsProduction.ToString().ToLowerInvariant());
|
|
|
|
_generatedContent = text;
|
|
|
|
return _generatedContent;
|
|
}
|
|
}
|
|
} |