1
0
mirror of https://github.com/Sonarr/Sonarr.git synced 2024-10-30 15:32:31 +01:00

added Expansive

This commit is contained in:
Keivan Beigi 2013-04-15 17:07:58 -07:00
parent 652a9d4a8a
commit 6a5c10a456
8 changed files with 435 additions and 0 deletions

View File

@ -0,0 +1,12 @@
using System;
namespace NzbDrone.Common.Expansive
{
public class CircularReferenceException : Exception
{
public CircularReferenceException(string message)
: base(message)
{
}
}
}

View File

@ -0,0 +1,272 @@
using System;
using System.Collections.Generic;
using System.Collections.Specialized;
using System.Dynamic;
using System.Linq;
using System.Text.RegularExpressions;
namespace NzbDrone.Common.Expansive
{
public static class Expansive
{
private static Dictionary<TokenStyle, PatternStyle> _patternStyles;
public static bool RequireAllExpansions { get; set; }
public static Func<string, string> DefaultExpansionFactory { get; set; }
public static TokenStyle DefaultTokenStyle { get; set; }
static Expansive()
{
Initialize();
}
public static string Expand(this string source)
{
return source.Expand(DefaultExpansionFactory);
}
public static string Expand(this string source, TokenStyle tokenStyle)
{
return source.ExpandInternal(DefaultExpansionFactory, tokenStyle);
}
public static string Expand(this string source, params string[] args)
{
var output = source;
var tokens = new List<string>();
var patternStyle = _patternStyles[DefaultTokenStyle];
var pattern = new Regex(patternStyle.TokenMatchPattern, RegexOptions.IgnoreCase);
var calls = new Stack<string>();
string callingToken = null;
while (pattern.IsMatch(output))
{
foreach (Match match in pattern.Matches(output))
{
var token = patternStyle.TokenReplaceFilter(match.Value);
var tokenIndex = 0;
if (!tokens.Contains(token))
{
tokens.Add(token);
tokenIndex = tokens.Count - 1;
}
else
{
tokenIndex = tokens.IndexOf(token);
}
output = Regex.Replace(output, patternStyle.OutputFilter(match.Value), "{" + tokenIndex + "}");
}
}
var newArgs = new List<string>();
foreach (var arg in args)
{
var newArg = arg;
var tokenPattern = new Regex(patternStyle.TokenFilter(String.Join("|", tokens)));
while (tokenPattern.IsMatch(newArg))
{
foreach (Match match in tokenPattern.Matches(newArg))
{
var token = patternStyle.TokenReplaceFilter(match.Value);
if (calls.Contains(string.Format("{0}:{1}", callingToken, token))) throw new CircularReferenceException(string.Format("Circular Reference Detected for token '{0}'.", callingToken));
calls.Push(string.Format("{0}:{1}", callingToken, token));
callingToken = token;
newArg = Regex.Replace(newArg, patternStyle.OutputFilter(match.Value), args[tokens.IndexOf(token)]);
}
}
newArgs.Add(newArg);
}
return string.Format(output, newArgs.ToArray());
}
public static string Expand(this string source, Func<string, string> expansionFactory)
{
return source.ExpandInternal(expansionFactory, DefaultTokenStyle);
}
public static string Expand(this string source, Func<string, string> expansionFactory, TokenStyle tokenStyle)
{
return source.ExpandInternal(expansionFactory, tokenStyle);
}
public static string Expand(this string source, object model)
{
return source.Expand(model, DefaultTokenStyle);
}
public static string Expand(this string source, params object[] models)
{
var mergedModel = new ExpandoObject().ToDictionary();
models.ToList().ForEach(m =>
{
var md = m.ToDictionary();
var keys = md.Keys;
keys.ToList().ForEach(k =>
{
if (!mergedModel.ContainsKey(k))
{
mergedModel.Add(k, md[k]);
}
});
});
return source.Expand(mergedModel as ExpandoObject);
}
public static string Expand(this string source, object model, TokenStyle tokenStyle)
{
return source.ExpandInternal(
name =>
{
IDictionary<string, object> modelDict = model.ToDictionary();
if (RequireAllExpansions && !modelDict.ContainsKey(name))
{
return "";
}
if (modelDict[name] == null)
{
return "";
}
return modelDict[name].ToString();
}
, tokenStyle);
}
#region : Private Helper Methods :
private static void Initialize()
{
DefaultTokenStyle = TokenStyle.MvcRoute;
_patternStyles = new Dictionary<TokenStyle, PatternStyle>
{
{
TokenStyle.MvcRoute, new PatternStyle
{
TokenMatchPattern = @"\{[a-zA-Z]\w*\}",
TokenReplaceFilter = token => token.Replace("{", "").Replace("}", ""),
OutputFilter = output => (output.StartsWith("{") && output.EndsWith("}") ? output : @"\{" + output + @"\}"),
TokenFilter = tokens => "{(" + tokens + ")}"
}
}
,
{
TokenStyle.Razor, new PatternStyle
{
TokenMatchPattern = @"@([a-zA-Z]\w*|\([a-zA-Z]\w*\))",
TokenReplaceFilter = token => token.Replace("@", "").Replace("(", "").Replace(")", ""),
OutputFilter = output => (output.StartsWith("@") ? output.Replace("(", @"\(").Replace(")",@"\)") : "@" + output.Replace("(", @"\(").Replace(")",@"\)")),
TokenFilter = tokens => @"@(" + tokens + @"|\(" + tokens + @"\))"
}
}
,
{
TokenStyle.NAnt, new PatternStyle
{
TokenMatchPattern = @"\$\{[a-zA-Z]\w*\}",
TokenReplaceFilter = token => token.Replace("${", "").Replace("}", ""),
OutputFilter = output => (output.StartsWith("${") && output.EndsWith("}") ? output.Replace("$",@"\$").Replace("{",@"\{").Replace("}",@"\}") : @"\$\{" + output + @"\}"),
TokenFilter = tokens => @"\$\{(" + tokens + @")\}"
}
}
,
{
TokenStyle.MSBuild, new PatternStyle
{
TokenMatchPattern = @"\$\([a-zA-Z]\w*\)",
TokenReplaceFilter = token => token.Replace("$(", "").Replace(")", ""),
OutputFilter = output => (output.StartsWith("$(") && output.EndsWith(")") ? output.Replace("$",@"\$").Replace("(",@"\(").Replace(")",@"\)") : @"\$\(" + output + @"\)"),
TokenFilter = tokens => @"\$\((" + tokens + @")\)"
}
}
};
}
private static string ExpandInternal(this string source, Func<string, string> expansionFactory, TokenStyle tokenStyle)
{
if (expansionFactory == null) throw new ApplicationException("ExpansionFactory not defined.\nDefine a DefaultExpansionFactory or call Expand(source, Func<string, string> expansionFactory))");
var patternStyle = _patternStyles[tokenStyle];
var pattern = new Regex(patternStyle.TokenMatchPattern, RegexOptions.IgnoreCase);
var callTreeParent = new Tree<string>("root").Root;
return source.Explode(pattern, patternStyle, expansionFactory, callTreeParent);
}
private static string Explode(this string source, Regex pattern, PatternStyle patternStyle, Func<string, string> expansionFactory, TreeNode<string> parent)
{
var output = source;
while (output.HasChildren(pattern))
{
foreach (Match match in pattern.Matches(source))
{
var child = match.Value;
var token = patternStyle.TokenReplaceFilter(match.Value);
var thisNode = parent.Children.Add(token);
// if we have already encountered this token in this call tree, we have a circular reference
if (thisNode.CallTree.Contains(token))
throw new CircularReferenceException(string.Format("Circular Reference Detected for token '{0}'. Call Tree: {1}->{2}",
token,
String.Join("->", thisNode.CallTree.ToArray().Reverse()), token));
// expand this match
var expandedValue = expansionFactory(token);
// Replace the match with the expanded value
child = Regex.Replace(child, patternStyle.OutputFilter(match.Value), expandedValue);
// Recursively expand the child until we no longer encounter nested tokens (or hit a circular reference)
child = child.Explode(pattern, patternStyle, expansionFactory, thisNode);
// finally, replace the match in the output with the fully-expanded value
output = Regex.Replace(output, patternStyle.OutputFilter(match.Value), child);
}
}
return output;
}
private static bool HasChildren(this string token, Regex pattern)
{
return pattern.IsMatch(token);
}
/// <summary>
/// Turns the object into an ExpandoObject
/// </summary>
private static dynamic ToExpando(this object o)
{
var result = new ExpandoObject();
var d = result as IDictionary<string, object>; //work with the Expando as a Dictionary
if (o is ExpandoObject) return o; //shouldn't have to... but just in case
if (o is NameValueCollection || o.GetType().IsSubclassOf(typeof(NameValueCollection)))
{
var nv = (NameValueCollection)o;
nv.Cast<string>().Select(key => new KeyValuePair<string, object>(key, nv[key])).ToList().ForEach(i => d.Add(i));
}
else
{
var props = o.GetType().GetProperties();
foreach (var item in props)
{
d.Add(item.Name, item.GetValue(o, null));
}
}
return result;
}
/// <summary>
/// Turns the object into a Dictionary
/// </summary>
private static IDictionary<string, object> ToDictionary(this object thingy)
{
return (IDictionary<string, object>)thingy.ToExpando();
}
#endregion
}
}

View File

@ -0,0 +1,12 @@
using System;
namespace NzbDrone.Common.Expansive
{
internal class PatternStyle
{
public string TokenMatchPattern { get; set; }
public Func<string, string> TokenFilter { get; set; }
public Func<string, string> TokenReplaceFilter { get; set; }
public Func<string, string> OutputFilter { get; set; }
}
}

View File

@ -0,0 +1,11 @@
namespace NzbDrone.Common.Expansive
{
internal class Tree<T> : TreeNode<T>
{
public Tree(T rootValue)
: base(rootValue)
{
Value = rootValue;
}
}
}

View File

@ -0,0 +1,85 @@
using System.Collections.Generic;
namespace NzbDrone.Common.Expansive
{
internal class TreeNode<T>
{
private List<T> _CallTree;
private TreeNode<T> _Parent;
public TreeNode(T Value)
{
this.Value = Value;
Parent = null;
Children = new TreeNodeList<T>(this);
_CallTree = new List<T>();
}
public TreeNode(T Value, TreeNode<T> Parent)
{
this.Value = Value;
this.Parent = Parent;
Children = new TreeNodeList<T>(this);
_CallTree = new List<T>();
}
public TreeNode<T> Parent
{
get { return _Parent; }
set
{
if (value == _Parent)
{
return;
}
if (_Parent != null)
{
_Parent.Children.Remove(this);
}
if (value != null && !value.Children.Contains(this))
{
value.Children.Add(this);
}
_Parent = value;
}
}
public TreeNode<T> Root
{
get
{
//return (Parent == null) ? this : Parent.Root;
TreeNode<T> node = this;
while (node.Parent != null)
{
node = node.Parent;
}
return node;
}
}
public TreeNodeList<T> Children { get; private set; }
public List<T> CallTree
{
get
{
_CallTree = new List<T>();
TreeNode<T> node = this;
while (node.Parent != null)
{
node = node.Parent;
_CallTree.Add(node.Value);
}
return _CallTree;
}
private set { _CallTree = value; }
}
public T Value { get; set; }
}
}

View File

@ -0,0 +1,32 @@
using System.Collections.Generic;
namespace NzbDrone.Common.Expansive
{
internal class TreeNodeList<T> : List<TreeNode<T>>
{
public TreeNode<T> Parent;
public TreeNodeList(TreeNode<T> Parent)
{
this.Parent = Parent;
}
public new TreeNode<T> Add(TreeNode<T> Node)
{
base.Add(Node);
Node.Parent = Parent;
return Node;
}
public TreeNode<T> Add(T Value)
{
return Add(new TreeNode<T>(Value));
}
public override string ToString()
{
return "Count=" + Count.ToString();
}
}
}

View File

@ -0,0 +1,2 @@
Source: https://github.com/anderly/Expansive
Microsoft Public License (MS-PL): http://opensource.org/licenses/MS-PL

View File

@ -107,6 +107,12 @@
<Compile Include="Eventing\IEvent.cs" />
<Compile Include="Eventing\IEventAggregator.cs" />
<Compile Include="Eventing\IHandle.cs" />
<Compile Include="Expansive\CircularReferenceException.cs" />
<Compile Include="Expansive\Expansive.cs" />
<Compile Include="Expansive\PatternStyle.cs" />
<Compile Include="Expansive\Tree.cs" />
<Compile Include="Expansive\TreeNode.cs" />
<Compile Include="Expansive\TreeNodeList.cs" />
<Compile Include="HostController.cs" />
<Compile Include="Instrumentation\VersionLayoutRenderer.cs" />
<Compile Include="StringExtention.cs" />
@ -152,6 +158,9 @@
<ItemGroup>
<EmbeddedResource Include="EnsureThat\Resources\ExceptionMessages.resx" />
</ItemGroup>
<ItemGroup>
<Content Include="Expansive\license.txt" />
</ItemGroup>
<Import Project="$(MSBuildToolsPath)\Microsoft.CSharp.targets" />
<Import Project="$(SolutionDir)\.nuget\nuget.targets" />
<!-- To modify your build process, add your task inside one of the targets below and uncomment it.