protected override async Task ExtractDatabaseEntityDeclarationsAsync(Solution solution, IProgress <ExtractionProgress> progress)
{
Dictionary <ClassDeclarationSyntax, SemanticModel> classes = await solution.GetClassesSignedWithAttributeType <TableAttribute>();
foreach (var item in classes.Keys)
{
var dbEntityDeclaration = new DatabaseEntityDeclaration <LINQToSQL>(item.Identifier.ToString(), item.GetCompilationInfo(classes[item]));
DatabaseEntityDeclarations.Add(dbEntityDeclaration);
}
}
protected override async Task ExtractDatabaseEntityDeclarationsAsync(Solution solution, IProgress <ExtractionProgress> progress)
{
string extractionNote = "Extracting Database Entity Declarations by finding all IQueryable<T> properties in Data Context Declarations";
progress.Report(new ExtractionProgress(extractionNote));
int totalAmountOfDataContextClasses = GetTotalAmountOfDataContextClasses();
int counter = 0;
foreach (var dataContextClassDeclarationSyntax in Context.DataContextDeclarations)
{
counter++;
progress.Report(GetExtractionProgress(totalAmountOfDataContextClasses, counter));
CompilationInfo compInfo = dataContextClassDeclarationSyntax.CompilationInfo;
foreach (var propertyDeclarationSyntax in compInfo.SyntaxNode.DescendantNodes().OfType <PropertyDeclarationSyntax>())
{
SemanticModel model = compInfo.SemanticModel;
if (model.IsOfType <IQueryable>(propertyDeclarationSyntax))
{
TypeSyntax propertyType = propertyDeclarationSyntax.Type;
//Get T from DbSet<T> or IQueryable<T>
string entityClassName = (propertyType as GenericNameSyntax).TypeArgumentList.Arguments[0].ToFullString();
Dictionary <ClassDeclarationSyntax, SemanticModel> entityClass = await solution.GetClassesOfType(entityClassName);
if (entityClass.Keys.Count > 0)
{
ClassDeclarationSyntax entityClassDeclarationSyntax = entityClass.Keys.First();
SemanticModel modelForEntityClass = entityClass[entityClassDeclarationSyntax];
DatabaseEntityDeclarations.Add(new DatabaseEntityDeclaration <EntityFramework>(entityClassName,
entityClassDeclarationSyntax.GetCompilationInfo(modelForEntityClass)));
}
}
}
}
}