github/Discord.Net
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Implemented new CommandMap

Browse Source
This commit is contained in:
RogueException
2016-06-29 07:03:44 -03:00
parent e4762b1522
commit 7723130713
3 changed files with 178 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

76
src/Discord.Net.Commands/Map/CommandMap.cs Normal file
View File

@@ -0,0 +1,76 @@
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Collections.Immutable;
using System.Linq;
namespace Discord.Commands
{
internal class CommandMap
{
private readonly ConcurrentDictionary<string, CommandMapNode> _nodes;
public CommandMap()
{
_nodes = new ConcurrentDictionary<string, CommandMapNode>();
}
public void AddCommand(Command command)
{
string text = command.Text;
int nextSpace = text.IndexOf(' ');
string name;
lock (this)
{
if (nextSpace == -1)
name = command.Text;
else
name = command.Text.Substring(0, nextSpace);
var nextNode = _nodes.GetOrAdd(name, x => new CommandMapNode(x));
nextNode.AddCommand(nextSpace == -1 ? "" : text, nextSpace + 1, command);
}
}
public void RemoveCommand(Command command)
{
string text = command.Text;
int nextSpace = text.IndexOf(' ');
string name;
lock (this)
{
if (nextSpace == -1)
name = command.Text;
else
name = command.Text.Substring(0, nextSpace);
CommandMapNode nextNode;
if (_nodes.TryGetValue(name, out nextNode))
{
nextNode.AddCommand(nextSpace == -1 ? "" : text, nextSpace + 1, command);
if (nextNode.IsEmpty)
_nodes.TryRemove(name, out nextNode);
}
}
}
public IEnumerable<Command> GetCommands(string text)
{
int nextSpace = text.IndexOf(' ');
string name;
lock (this)
{
if (nextSpace == -1)
name = text;
else
name = text.Substring(0, nextSpace);
CommandMapNode nextNode;
if (_nodes.TryGetValue(name, out nextNode))
return nextNode.GetCommands(text, nextSpace + 1);
else
return Enumerable.Empty<Command>();
}
}
}
}

95
src/Discord.Net.Commands/Map/CommandMapNode.cs Normal file
View File

@@ -0,0 +1,95 @@
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Collections.Immutable;
namespace Discord.Commands
{
internal class CommandMapNode
{
private readonly ConcurrentDictionary<string, CommandMapNode> _nodes;
private readonly string _name;
private ImmutableArray<Command> _commands;
public bool IsEmpty => _commands.Length == 0 && _nodes.Count == 0;
public CommandMapNode(string name)
{
_name = name;
_nodes = new ConcurrentDictionary<string, CommandMapNode>();
_commands = ImmutableArray.Create<Command>();
}
public void AddCommand(string text, int index, Command command)
{
int nextSpace = text.IndexOf(' ', index);
string name;
lock (this)
{
if (text == "")
_commands = _commands.Add(command);
else
{
if (nextSpace == -1)
name = text.Substring(index);
else
name = text.Substring(index, nextSpace - index);
var nextNode = _nodes.GetOrAdd(name, x => new CommandMapNode(x));
nextNode.AddCommand(nextSpace == -1 ? "" : text, nextSpace + 1, command);
}
}
}
public void RemoveCommand(string text, int index, Command command)
{
int nextSpace = text.IndexOf(' ', index);
string name;
lock (this)
{
if (text == "")
_commands = _commands.Remove(command);
else
{
if (nextSpace == -1)
name = text.Substring(index);
else
name = text.Substring(index, nextSpace - index);
CommandMapNode nextNode;
if (_nodes.TryGetValue(name, out nextNode))
{
nextNode.RemoveCommand(nextSpace == -1 ? "" : text, nextSpace + 1, command);
if (nextNode.IsEmpty)
_nodes.TryRemove(name, out nextNode);
}
}
}
}
public IEnumerable<Command> GetCommands(string text, int index)
{
int nextSpace = text.IndexOf(' ', index);
string name;
var commands = _commands;
for (int i = 0; i < commands.Length; i++)
yield return _commands[i];
if (text != "")
{
if (nextSpace == -1)
name = text.Substring(index);
else
name = text.Substring(index, nextSpace - index);
CommandMapNode nextNode;
if (_nodes.TryGetValue(name, out nextNode))
{
foreach (var cmd in nextNode.GetCommands(nextSpace == -1 ? "" : text, nextSpace + 1))
yield return cmd;
}
}
}
}
}