private void SerializeAsYaml(DiagramElements diagramElementsObj)
{
var serializer = new YamlDotNet.Serialization.SerializerBuilder()
.WithEmissionPhaseObjectGraphVisitor(args => new YamlIEnumerableSkipEmptyObjectGraphVisitor(args.InnerVisitor))
.Build();
using (var writer = new StringWriter())
{
serializer.Serialize(writer, diagramElementsObj);
yamlData = writer.ToString();
outputYamlString = yamlData.Replace("'", string.Empty);
}
}
private void SaveAsYaml(DiagramElements diagramElementsObj)
{
SaveFileDialog savefile = new SaveFileDialog();
savefile.FileName = diagramElementsObj.refDiagramName;
savefile.Filter = "YAML files (*.yaml)|*.yaml|All files (*.*)|*.*";
if (savefile.ShowDialog() == DialogResult.OK)
{
using (StreamWriter sw = new StreamWriter(savefile.FileName))
sw.WriteLine(outputYamlString);
}
}
private void GetTransitionsByState(EA.Repository Rep, DiagramElements diagramElementsObj, EA.Element element)
{
foreach (EA.Connector item in element.Connectors)
{
bool isOld = false;
int clientId = item.ClientID;
int supplierId = item.SupplierID;
EA.Element clientElement = Rep.GetElementByID(clientId);
EA.Element supplierElement = Rep.GetElementByID(supplierId);
Transition transitionObj = new Transition();
transitionObj.from = Utilities.FormatElementName(clientElement.FQName);
transitionObj.to = Utilities.FormatElementName(supplierElement.FQName);
transitionObj.trigger = item.TransitionEvent;
effectsList = item.TransitionAction;
effectsList = effectsList.ReplaceAll(charsToReplaceFromEffects, ',');
effectsList = Utilities.TruncateCommas(effectsList);
if (string.IsNullOrEmpty(effectsList))
{
transitionObj.effects = "";
}
else
{
transitionObj.effects = "[" + effectsList + "]";
}
foreach (var transItem in diagramElementsObj.transitions)
{
if (transItem.from.Equals(transitionObj.from) && transItem.to.Equals(transitionObj.to))
{
isOld = true;
break;
}
}
if (!isOld)
{
diagramElementsObj.transitions.Add(transitionObj);
}
}
}
public void EA_MenuClick(EA.Repository rep, string location, string menuName, string itemName)
{
if (itemName == "Save diagram as YAML")
{
EA.Diagram diag;
DiagramElements diagramElementsObj = new DiagramElements();
switch (rep.GetContextItemType())
{
case EA.ObjectType.otPackage:
{
MessageBox.Show("Please select a diagram");
break;
}
case EA.ObjectType.otDiagram:
{
diag = rep.GetContextObject();
diagramElementsObj.refDiagramName = diag.Name;
diagramElementsObj.refDiagramId = diag.DiagramGUID.Trim(charsToReplaceFromDiagramId);
diagramElementsObj.states = new List <State>();
diagramElementsObj.transitions = new HashSet <Transition>();
foreach (EA.DiagramObject diagramObj in diag.DiagramObjects)
{
int diagramId = diagramObj.DiagramID;
EA.Diagram diagram = rep.GetDiagramByID(diagramId);
int elementId = diagramObj.ElementID;
EA.Element element = rep.GetElementByID(elementId);
State stateObj = new State();
if (element.MetaType == "Pseudostate")
{
diagramElementsObj.initialState = Utilities.FormatElementName(element.Name);
}
else if (element.MetaType == "FinalState")
{
diagramElementsObj.finalState = Utilities.FormatElementName(element.Name);
}
else
{
if (element.MetaType == "State")
{
stateObj.name = Utilities.FormatElementName(element.FQName);
diagramElementsObj.states.Add(stateObj);
}
}
if (element.Methods.Count > 0)
{
GetActionsByState(element, stateObj);
}
GetTransitionsByState(rep, diagramElementsObj, element);
}
SerializeAsYaml(diagramElementsObj);
SaveAsYaml(diagramElementsObj);
break;
}
case EA.ObjectType.otElement:
{
MessageBox.Show("Please select a diagram");
break;
}
}
}
else if (itemName == "About")
{
MessageBox.Show("Yaml Generator - Version 1.0");
}
}