zontreck/ForgeCore
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Restructure

Browse source
This commit is contained in:
Zontreck 2019-12-20 17:38:55 -07:00
parent 437cc0e0e1
commit da91200c74
16 changed files with 43 additions and 110 deletions
Download patch file Download diff file Expand all files Collapse all files

253
CommandSystem/CommandRegistry.cs Normal file
View file

@ -0,0 +1,253 @@
using System;
using System.Collections.Generic;
using System.Linq;
using OpenMetaverse;
using Newtonsoft.Json;
using System.Reflection;
using System.Threading;
namespace Bot.CommandSystem
{
public sealed class CommandRegistry
{
/*
* ===============================
* START SINGLETON PATTERN
* ===============================
*/
private static CommandRegistry _instance = null;
private static readonly object lockhandle = new object();
static CommandRegistry()
{
}
public static CommandRegistry Instance
{
get
{
lock (lockhandle)
{
if (_instance == null)
{
BotSession bs = BotSession.Instance;
_instance = new CommandRegistry();
_instance.client = bs.grid;
_instance.config = bs.ConfigurationHandle;
_instance.Log = bs.Logger;
_instance.MHEx = bs.MHE;
_instance.LocateCommands();
}
return _instance;
}
}
}
/*
* ==============================
* SINGLETON PATTERN END
* Usage: CommandRegistry reg = CommandRegistry.Instance
* ==============================
*/
// Define the registry
public Dictionary<string, CommandGroup> Cmds = new Dictionary<string, CommandGroup>();
public GridClient client;
public SysOut Log;
public IConfig config;
public MessageHandler.MessageHandleEvent MHEx;
public void LocateCommands()
{
try
{
int i = 0;
// Locate all commands--
for (i = 0; i < AppDomain.CurrentDomain.GetAssemblies().Length; i++)
{
Assembly A = null;
try
{
A = AppDomain.CurrentDomain.GetAssemblies()[i];
}
catch (Exception e)
{
// MHEx(MessageHandler.Destinations.DEST_LOCAL, UUID.Zero, "")
}
if (A != null)
{
int ii = 0;
for (ii = 0; ii < A.GetTypes().Length; ii++)
{
Type T = null;
try
{
T = A.GetTypes()[ii];
}
catch (Exception e)
{
}
if (T != null)
{
if (T.IsClass)
{
foreach (MethodInfo MI in T.GetMethods())
{
CommandGroup[] Command = (CommandGroup[])MI.GetCustomAttributes(typeof(CommandGroup), false);
//var CommandO = MI.GetCustomAttributes(typeof(CommandGroup), false);
if (Command.Length > 0)
{
for (int ix = 0; ix < Command.Length; ix++)
{
CommandGroup CG = Command[ix];
CG.AssignedMethod = MI;
if (Cmds.ContainsKey(CG.Command) == false)
{
Console.WriteLine("DISCOVER: " + CG.Command);
Cmds.Add(CG.Command, CG);
}
}
}
}
}
}
}
}
}
Console.WriteLine("Discovered " + Cmds.Count.ToString() + " total commands");
}
catch (ReflectionTypeLoadException e)
{
MHEx(MessageHandler.Destinations.DEST_LOCAL, UUID.Zero, "FAILURE!!!\n \n[Assembly load failure]");
foreach (Exception X in e.LoaderExceptions)
{
MHEx(MessageHandler.Destinations.DEST_LOCAL, UUID.Zero, X.Message + "\n \nSTACK: " + X.StackTrace);
}
}
}
public void PrintHelpAll(MessageHandler.Destinations dest, UUID uid)
{
for (int i = 0; i < Cmds.Count; i++)
{
KeyValuePair<string, CommandGroup> kvp = Cmds.ElementAt(i);
CommandHelp HE = kvp.Value.cmdUsage;
if (dest == MessageHandler.Destinations.DEST_GROUP)
{
if (!HE.hasGroupFlag())
{
//return;
}
else
{
MHEx(dest, uid, HE.GetUsage());
}
}
else
{
MHEx(dest, uid, HE.GetUsage());
}
// MHEx(dest, uid, kvp.Value.cmdUsage.GetUsage());
}
}
public void PrintHelp(MessageHandler.Destinations dest, string cmd, UUID uid)
{
try
{
CommandHelp HE = Cmds[cmd].cmdUsage;
if (dest == MessageHandler.Destinations.DEST_GROUP)
{
if (!HE.hasGroupFlag())
{
//return; // DO NOT SCHEDULE THIS HELP INFO FOR GROUP!!!
}
}
MHEx(dest, uid, Cmds[cmd].cmdUsage.GetUsage());
}
catch (Exception e)
{
MHEx(dest, uid, "Error: Unrecognized command");
}
}
public void RunCommand(string cmdString, UUID user, int level, MessageHandler.MessageHandleEvent MHE, MessageHandler.Destinations source, UUID agentKey, string agentName)
{
MHEx = MHE;
int pos = 0;
string[] cmdStruct = cmdString.Split(' ');
int IgnoreCount = 0;
foreach (string S in cmdStruct)
{
if (IgnoreCount > 0) { IgnoreCount--; }
else
{
// Search for Command
if (Cmds.ContainsKey(S))
{
// this must be a command
// argument types will ALWAYS be structured like this:
// UUID, int, GridClient, [args up to argCount], optional:{SysOut}
CommandGroup cgX = Cmds[S];
if (level >= cgX.minLevel)
{
// Check that the destination is allowed.
// If not then skip this command entirely
MessageHandler.Destinations dests = cgX.CommandSource;
bool Allowed = false;
if ((dests & MessageHandler.Destinations.DEST_AGENT) == source) Allowed = true;
if ((dests & MessageHandler.Destinations.DEST_GROUP) == source) Allowed = true;
if ((dests & MessageHandler.Destinations.DEST_LOCAL) == source) Allowed = true;
if (!Allowed)
{
IgnoreCount = cgX.arguments;
}
else
{
var ovj = Activator.CreateInstance(cgX.AssignedMethod.DeclaringType);
string[] additionalArgs = new string[cgX.arguments];
IgnoreCount = cgX.arguments;
for (int i = 1; i <= cgX.arguments; i++)
{
additionalArgs[i - 1] = cmdStruct[pos + i];
}
pos++;
//(UUID client, int level, GridClient grid, string[] additionalArgs,
//SysOut log, MessageHandler.MessageHandleEvent MHE, MessageHandler.Destinations source,
//CommandRegistry registry, UUID agentKey, string agentName)
Thread CommandThread = new Thread(() => cgX.AssignedMethod.Invoke(ovj, new object[] { user, level, client, additionalArgs, Log, MHE, source, this, agentKey, agentName }));
CommandThread.Start();
}
}
}
}
}
}
}
}