public async Task <SignalrTypeDocument> GenerateForHubsAsync(ISignalRTypesBuilder signalRTypesBuilder)
{
Dictionary <string, Type> hubs = new Dictionary <string, Type>();
var type = typeof(Hub);
var types = AppDomain.CurrentDomain.GetAssemblies()
.SelectMany(s => s.GetTypes())
.Where(p => type.IsAssignableFrom(p) && !p.IsAbstract).ToList();
foreach (var item in types)
{
hubs.Add(item.Name, item);
}
var document = new SignalrTypeDocument();
return(await GenerateForHubsAsync(hubs, document, signalRTypesBuilder));
}
public async Task <SignalrTypeDocument> GenerateForHubsAsync(IReadOnlyDictionary <string, Type> hubs, SignalrTypeDocument template, ISignalRTypesBuilder signalRTypesBuilder)
{
var document = template;
var resolver = new SignalrTypeSchemaResolver(document, _settings);
var generator = new JsonSchemaGenerator(_settings);
foreach (var h in hubs)
{
var type = h.Value;
var hub = new SignalrTypeHub();
hub.Name = type.Name.EndsWith("Hub") ? type.Name.Substring(0, type.Name.Length - 3) : type.Name;
hub.Description = type.GetXmlDocsSummary();
foreach (var method in GetOperationMethods(type))
{
var operation = await GenerateOperationAsync(type, method, generator, resolver, SignalrTypeOperationType.Sync);
hub.Operations[method.Name] = operation;
}
foreach (var method in GetChannelMethods(type))
{
hub.Operations[method.Name] = await GenerateOperationAsync(type, method, generator, resolver, SignalrTypeOperationType.Observable);
}
// handle with service this comment
//var ctors = type.GetConstructors();
//if (ctors != null && ctors.Length > 0)
//{
// var allTypes = ctors.Select(
// x => x.GetParameters()
// ?.Where(
// z => z.ParameterType.IsGenericType && z.ParameterType.GetGenericTypeDefinition() == typeof(IHubContext<,>))
// ).Select(e => e.Select(s => s.ParameterType.GetGenericArguments()[1]))
// ?.ToList();
// foreach (var itemTypes in allTypes)
// {
// foreach (var itemType in itemTypes)
// {
// foreach (var callbackMethod in GetOperationMethods(itemType))
// {
// var callback = await GenerateOperationAsync(type, callbackMethod, generator, resolver, SignalrTypeOperationType.Sync);
// var methodName = callbackMethod.Name;
// var hubMethodName = callbackMethod.CustomAttributes.FirstOrDefault(x => x.AttributeType == typeof(HubMethodNameAttribute));
// if (hubMethodName != null && hubMethodName.ConstructorArguments.Count > 0)
// {
// var argName = hubMethodName.ConstructorArguments[0].Value.ToString();
// if (!string.IsNullOrEmpty(argName))
// {
// methodName = argName;
// }
// }
// hub.Callbacks[methodName] = callback;
// }
// }
// }
//}
//var baseTypeGenericArguments = type.BaseType.GetGenericArguments();
//if (baseTypeGenericArguments.Length == 1)
//{
// var callbackType = baseTypeGenericArguments[0];
// foreach (var callbackMethod in GetOperationMethods(callbackType))
// {
// var callback = await GenerateOperationAsync(type, callbackMethod, generator, resolver, SignalrTypeOperationType.Sync);
// var methodName = callbackMethod.Name;
// var hubMethodName = callbackMethod.CustomAttributes.FirstOrDefault(x => x.AttributeType == typeof(HubMethodNameAttribute));
// if (hubMethodName != null && hubMethodName.ConstructorArguments.Count > 0)
// {
// var argName = hubMethodName.ConstructorArguments[0].Value.ToString();
// if (!string.IsNullOrEmpty(argName))
// {
// methodName = argName;
// }
// }
// hub.Callbacks[methodName] = callback;
// }
//}
var callbacks = signalRTypesBuilder.GetCallBack();
if (callbacks.Count > 0)
{
foreach (var callbackItem in callbacks)
{
var callback = await GenerateOperationAsync(callbackItem, generator, resolver, SignalrTypeOperationType.Sync);
hub.Callbacks[callbackItem.MethodName] = callback;
}
}
document.Hubs[h.Key] = hub;
}
return(document);
}
public SignalrTypeSchemaResolver(SignalrTypeDocument document, SignalrTypeGeneratorSettings settings)
: base(document, settings)
{
_document = document;
_settings = settings;
}
