private static async Task <Document> FixMigrationHash(Document document, TypeDeclarationSyntax typeDecl, CancellationToken ct)
{
var semanticModel = await document.GetSemanticModelAsync(ct);
var newTypeDecl = MigrationHashHelper.UpdateMigrationHash(typeDecl, ct, semanticModel);
var root = await document.GetSyntaxRootAsync(ct);
var newRoot = root.ReplaceNode(typeDecl, newTypeDecl);
return(document.WithSyntaxRoot(newRoot));
}
private static async Task <Document> AddMigrationMethod(Document document, TypeDeclarationSyntax typeDecl, CancellationToken ct)
{
var semanticModel = await document.GetSemanticModelAsync(ct);
var typeSymbol = semanticModel.GetDeclaredSymbol(typeDecl, ct);
var latestMigrationMethod = MigrationHashHelper.GetMigrationMethods(typeSymbol).LastOrDefault();
var latestVersion = latestMigrationMethod?.ToVersion ?? 0;
var dataArgumentTypeName = latestMigrationMethod?.ReturnType.Name ?? "JToken";
var method = GetMigrationMethod(latestVersion + 1, dataArgumentTypeName, ct);
var typeDeclWithUpdatedMigrationHash = MigrationHashHelper.UpdateMigrationHash(typeDecl, ct, semanticModel);
var typeDeclWithAddedMigrationMethod = AddMember(typeDeclWithUpdatedMigrationHash, method);
var root = (CompilationUnitSyntax)await document.GetSyntaxRootAsync(ct);
var rootWithNewTypeDecl = root.ReplaceNode(typeDecl, typeDeclWithAddedMigrationMethod);
return(document.WithSyntaxRoot(rootWithNewTypeDecl));
}
