public override void ElementAdded(ElementAddedEventArgs e) { base.ElementAdded(e); Association element = (Association)e.ModelElement; Store store = element.Store; Transaction current = store.TransactionManager.CurrentTransaction; PluralizationService pluralizationService = ModelRoot.PluralizationService; if (current.IsSerializing) { return; } if (string.IsNullOrEmpty(element.TargetPropertyName)) { string rootName = element.TargetMultiplicity == Multiplicity.ZeroMany && pluralizationService?.IsSingular(element.Target.Name) == true ? pluralizationService.Pluralize(element.Target.Name) : element.Target.Name; string identifierName = rootName; int index = 0; while (element.Source.HasPropertyNamed(identifierName)) { identifierName = $"{rootName}_{++index}"; } element.TargetPropertyName = identifierName; } if (element is BidirectionalAssociation bidirectionalAssociation) { if (string.IsNullOrEmpty(bidirectionalAssociation.SourcePropertyName)) { string rootName = element.SourceMultiplicity == Multiplicity.ZeroMany && pluralizationService?.IsSingular(element.Source.Name) == true ? pluralizationService.Pluralize(element.Source.Name) : element.Source.Name; string identifierName = rootName; int index = 0; while (element.Target.HasPropertyNamed(identifierName)) { identifierName = $"{rootName}_{++index}"; } bidirectionalAssociation.SourcePropertyName = identifierName; } } AssociationChangeRules.SetEndpointRoles(element); }
private void ProcessUnidirectionalAssociations(ParsingModels.ModelClass modelClass) { List<ModelUnidirectionalAssociation> unidirectionalAssociations = modelClass.UnidirectionalAssociations; foreach (ModelUnidirectionalAssociation data in unidirectionalAssociations) { UnidirectionalAssociation existing = Store.Get<UnidirectionalAssociation>() .FirstOrDefault(x => x.Target.Name == data.TargetClassName && x.Source.Name == data.SourceClassName && x.Source.Name == modelClass.Name // just to be sure && x.TargetPropertyName == data.TargetPropertyName); if (existing != null) { if (string.IsNullOrWhiteSpace(existing.FKPropertyName) && !string.IsNullOrWhiteSpace(data.ForeignKey)) { existing.FKPropertyName = data.ForeignKey; existing.Source.ModelRoot.ExposeForeignKeys = true; } continue; } ModelClass source = Store.Get<ModelClass>().FirstOrDefault(c => c.FullName == data.SourceClassFullName); ModelClass target = Store.Get<ModelClass>().FirstOrDefault(c => c.FullName == data.TargetClassFullName); if (source == null || target == null || source.FullName != modelClass.FullName) continue; // ReSharper disable once UnusedVariable UnidirectionalAssociation element = new UnidirectionalAssociation(Store, new[] { new RoleAssignment(UnidirectionalAssociation.UnidirectionalSourceDomainRoleId, source), new RoleAssignment(UnidirectionalAssociation.UnidirectionalTargetDomainRoleId, target) }, new[] { new PropertyAssignment(Association.SourceMultiplicityDomainPropertyId, ConvertMultiplicity(data.SourceMultiplicity)), new PropertyAssignment(Association.TargetMultiplicityDomainPropertyId, ConvertMultiplicity(data.TargetMultiplicity)), new PropertyAssignment(Association.TargetPropertyNameDomainPropertyId, data.TargetPropertyName), new PropertyAssignment(Association.TargetSummaryDomainPropertyId, data.TargetSummary), new PropertyAssignment(Association.TargetDescriptionDomainPropertyId, data.TargetDescription), new PropertyAssignment(Association.FKPropertyNameDomainPropertyId, data.ForeignKey), new PropertyAssignment(Association.SourceRoleDomainPropertyId, ConvertRole(data.SourceRole)), new PropertyAssignment(Association.TargetRoleDomainPropertyId, ConvertRole(data.TargetRole)), }); AssociationChangeRules.SetEndpointRoles(element); element.EnsureForeignKeyAttributes(); } }
public override void ElementPropertyChanged(ElementPropertyChangedEventArgs e) { base.ElementPropertyChanged(e); ModelRoot element = (ModelRoot)e.ModelElement; Store store = element.Store; Transaction current = store.TransactionManager.CurrentTransaction; if (current.IsSerializing) { return; } if (Equals(e.NewValue, e.OldValue)) { return; } List <string> errorMessages = EFCoreValidator.GetErrors(element).ToList(); bool redraw = false; switch (e.DomainProperty.Name) { case "ConnectionString": if (e.NewValue != null) { element.ConnectionStringName = null; } break; case "ConnectionStringName": if (e.NewValue != null) { element.ConnectionString = null; } break; case "DatabaseSchema": if (string.IsNullOrEmpty((string)e.NewValue)) { element.DatabaseSchema = "dbo"; } break; case "EntityFrameworkVersion": element.EntityFrameworkPackageVersion = "Latest"; if (element.EntityFrameworkVersion == EFVersion.EFCore) { element.InheritanceStrategy = CodeStrategy.TablePerHierarchy; } break; case "EnumOutputDirectory": if (string.IsNullOrEmpty((string)e.NewValue) && !string.IsNullOrEmpty(element.EntityOutputDirectory)) { element.EnumOutputDirectory = element.EntityOutputDirectory; } break; case "StructOutputDirectory": if (string.IsNullOrEmpty((string)e.NewValue) && !string.IsNullOrEmpty(element.EntityOutputDirectory)) { element.StructOutputDirectory = element.EntityOutputDirectory; } break; case "EntityOutputDirectory": if (string.IsNullOrEmpty(element.EnumOutputDirectory) || element.EnumOutputDirectory == (string)e.OldValue) { element.EnumOutputDirectory = (string)e.NewValue; } if (string.IsNullOrEmpty(element.StructOutputDirectory) || element.StructOutputDirectory == (string)e.OldValue) { element.StructOutputDirectory = (string)e.NewValue; } break; case "FileNameMarker": string newFileNameMarker = (string)e.NewValue; if (!Regex.Match($"a.{newFileNameMarker}.cs", @"^(?!^(PRN|AUX|CLOCK\$|NUL|CON|COM\d|LPT\d|\..*)(\..+)?$)[^\x00-\x1f\\?*:\"";|/]+$") .Success) { errorMessages.Add("Invalid value to make part of file name"); } break; case "InheritanceStrategy": if ((element.EntityFrameworkVersion == EFVersion.EFCore) && (element.NuGetPackageVersion.MajorMinorVersionNum < 2.1)) { element.InheritanceStrategy = CodeStrategy.TablePerHierarchy; } break; case "LayoutAlgorithm": ModelDisplay.LayoutDiagram(element.Classes.FirstOrDefault()?.GetActiveDiagram() as EFModelDiagram); break; case "Namespace": errorMessages.Add(CommonRules.ValidateNamespace((string)e.NewValue, CodeGenerator.IsValidLanguageIndependentIdentifier)); break; case "ShowCascadeDeletes": // need these change rules to fire even though nothing in Association has changed // so we need to set this early -- requires guarding against recursion. bool newShowCascadeDeletes = (bool)e.NewValue; if (element.ShowCascadeDeletes != newShowCascadeDeletes) { element.ShowCascadeDeletes = newShowCascadeDeletes; foreach (Association association in store.ElementDirectory.FindElements <Association>()) { AssociationChangeRules.UpdateDisplayForCascadeDelete(association); } } redraw = true; break; case "ShowWarningsInDesigner": redraw = true; break; case "WarnOnMissingDocumentation": if (element.ShowWarningsInDesigner) { redraw = true; } ModelRoot.ExecuteValidator?.Invoke(); break; } errorMessages = errorMessages.Where(m => m != null).ToList(); if (errorMessages.Any()) { current.Rollback(); ErrorDisplay.Show(string.Join("\n", errorMessages)); } if (redraw) { element.InvalidateDiagrams(); } }
public override void ElementAdded(ElementAddedEventArgs e) { base.ElementAdded(e); Association element = (Association)e.ModelElement; Store store = element.Store; Transaction current = store.TransactionManager.CurrentTransaction; PluralizationService pluralizationService = ModelRoot.PluralizationService; if (current.IsSerializing || ModelRoot.BatchUpdating) { return; } // add unidirectional // source can't be dependent (connection builder handles this) // if target is dependent, // source cardinality is 0..1 // target cardinality is 0..1 // source is principal if (element is UnidirectionalAssociation && element.Target.IsDependentType) { element.TargetMultiplicity = Multiplicity.ZeroOne; element.SourceRole = EndpointRole.Principal; element.TargetRole = EndpointRole.Dependent; } // add bidirectional // neither can be dependent (connection builder handles this) if (string.IsNullOrEmpty(element.TargetPropertyName)) { string rootName = element.TargetMultiplicity == Multiplicity.ZeroMany && pluralizationService?.IsSingular(element.Target.Name) == true ? pluralizationService.Pluralize(element.Target.Name) : element.Target.Name; string identifierName = rootName; int index = 0; while (element.Source.HasPropertyNamed(identifierName)) { identifierName = $"{rootName}_{++index}"; } element.TargetPropertyName = identifierName; } if (element is BidirectionalAssociation bidirectionalAssociation) { if (string.IsNullOrEmpty(bidirectionalAssociation.SourcePropertyName)) { string rootName = element.SourceMultiplicity == Multiplicity.ZeroMany && pluralizationService?.IsSingular(element.Source.Name) == true ? pluralizationService.Pluralize(element.Source.Name) : element.Source.Name; string identifierName = rootName; int index = 0; while (element.Target.HasPropertyNamed(identifierName)) { identifierName = $"{rootName}_{++index}"; } bidirectionalAssociation.SourcePropertyName = identifierName; } } AssociationChangeRules.SetEndpointRoles(element); }
public override void ElementAdded(ElementAddedEventArgs e) { base.ElementAdded(e); Association element = (Association)e.ModelElement; Store store = element.Store; Transaction current = store.TransactionManager.CurrentTransaction; PluralizationService pluralizationService = ModelRoot.PluralizationService; if (current.IsSerializing) { return; } // TODO: Error if 1..N to an owned type (EFCore) or to a complex type (EF6) if (element.TargetMultiplicity == Multiplicity.ZeroMany && element.Target.IsDependentType) { current.Rollback(); ErrorDisplay.Show($"Can't have a 0..* association from {element.Source.Name} to dependent type {element.Target.Name}"); return; } if (string.IsNullOrEmpty(element.TargetPropertyName)) { string rootName = element.TargetMultiplicity == Multiplicity.ZeroMany && pluralizationService?.IsSingular(element.Target.Name) == true ? pluralizationService.Pluralize(element.Target.Name) : element.Target.Name; string identifierName = rootName; int index = 0; while (element.Source.HasPropertyNamed(identifierName)) { identifierName = $"{rootName}_{++index}"; } element.TargetPropertyName = identifierName; } if (element is BidirectionalAssociation bidirectionalAssociation) { if (string.IsNullOrEmpty(bidirectionalAssociation.SourcePropertyName)) { string rootName = element.SourceMultiplicity == Multiplicity.ZeroMany && pluralizationService?.IsSingular(element.Source.Name) == true ? pluralizationService.Pluralize(element.Source.Name) : element.Source.Name; string identifierName = rootName; int index = 0; while (element.Target.HasPropertyNamed(identifierName)) { identifierName = $"{rootName}_{++index}"; } bidirectionalAssociation.SourcePropertyName = identifierName; } } AssociationChangeRules.SetEndpointRoles(element); }
protected override void OnDocumentLoaded() { base.OnDocumentLoaded(); if (!(RootElement is ModelRoot modelRoot)) { return; } // set to the project's namespace if no namespace set if (string.IsNullOrEmpty(modelRoot.Namespace)) { using (Transaction tx = modelRoot.Store.TransactionManager.BeginTransaction("SetDefaultNamespace")) { DTE dte = Microsoft.VisualStudio.Shell.Package.GetGlobalService(typeof(DTE)) as DTE; Project currentProject = GetActiveProject(dte); modelRoot.Namespace = currentProject.Properties.Item("DefaultNamespace")?.Value as string; tx.Commit(); } } ReadOnlyCollection <Association> associations = modelRoot.Store.ElementDirectory.FindElements <Association>(); if (associations.Any()) { using (Transaction tx = modelRoot.Store.TransactionManager.BeginTransaction("StyleConnectors")) { // style association connectors if needed foreach (Association element in associations) { AssociationChangeRules.UpdateDisplayForPersistence(element); AssociationChangeRules.UpdateDisplayForCascadeDelete(element); // for older diagrams that didn't calculate this initially AssociationChangeRules.SetEndpointRoles(element); } tx.Commit(); } } List <GeneralizationConnector> generalizationConnectors = modelRoot.Store .ElementDirectory .FindElements <GeneralizationConnector>() .Where(x => !x.FromShape.IsVisible || !x.ToShape.IsVisible) .ToList(); List <AssociationConnector> associationConnectors = modelRoot.Store .ElementDirectory .FindElements <AssociationConnector>() .Where(x => !x.FromShape.IsVisible || !x.ToShape.IsVisible) .ToList(); if (generalizationConnectors.Any() || associationConnectors.Any()) { using (Transaction tx = modelRoot.Store.TransactionManager.BeginTransaction("HideConnectors")) { // hide any connectors that may have been hidden due to hidden shapes foreach (GeneralizationConnector connector in generalizationConnectors) { connector.Hide(); } foreach (AssociationConnector connector in associationConnectors) { connector.Hide(); } tx.Commit(); } } }