private void _SetRelation(IRelationSetter relationSetter)
{
if (relationSetter is EmptyRelationSetter)
{
return;
}
RelationDialog relationDialog = new RelationDialog();
relationDialog.ShowDialog();
if (relationSetter.SetRelation(relationDialog.Relation))
{
this.UpdateCanvas();
}
relationDialog.Dispose();
}
protected override async Task ExecuteAsync(OleMenuCmdEventArgs e)
{
await ThreadHelper.JoinableTaskFactory.SwitchToMainThreadAsync();
Document document;
document = ApplicationObject.VisualStudioApplication.ActiveDocument;
if (document == null)
{
return;
}
TextDocument textDoc = (TextDocument)document.Object("TextDocument");
if (textDoc == null)
{
return;
}
StringBuilder sbResult = new System.Text.StringBuilder(500);
RelationDialog dlg = new RelationDialog();
DialogResult result = dlg.ShowDialog();
if (result == DialogResult.Cancel)
{
return;
}
string openBracket;
string closeBracket;
string typeOf;
textDoc.Selection.StartOfLine(vsStartOfLineOptions.vsStartOfLineOptionsFirstColumn, false);
string listLeft = null;
string listRight = null;
string startGeneric = null;
string endGeneric = null;
string spc = null;
bool isVC;
if (isVC = (Path.GetExtension(document.FullName).ToLower() == ".cs"))
{
openBracket = "[";
closeBracket = "]";
typeOf = "typeof";
spc = "\t\t";
startGeneric = "<";
endGeneric = ">";
}
else
{
openBracket = "<";
closeBracket = "> _";
typeOf = "GetType";
spc = "\t";
startGeneric = "(Of ";
endGeneric = ")";
}
if (dlg.List)
{
if (!dlg.UseGenerics)
{
switch (dlg.ListType)
{
case ListType.IList:
listLeft = "IList";
listRight = "ArrayList";
break;
default:
listLeft = listRight = "ArrayList";
break;
}
}
else
{
switch (dlg.ListType)
{
case ListType.IList:
listLeft = "IList" + startGeneric + dlg.Type + endGeneric;
listRight = "List" + startGeneric + dlg.Type + endGeneric;
break;
default:
listLeft = listRight = "List" + startGeneric + dlg.Type + endGeneric;
break;
}
}
}
try
{
sbResult.Append(spc + openBracket + "NDORelation");
if ((dlg.List && !dlg.UseGenerics) || dlg.Composite || dlg.RelationName.Trim() != string.Empty)
{
bool needsComma = false;
sbResult.Append("(");
if (dlg.List && !dlg.UseGenerics)
{
sbResult.Append(typeOf + '(' + dlg.Type + ')');
needsComma = true;
}
if (dlg.Composite)
{
if (needsComma)
{
sbResult.Append(", ");
}
sbResult.Append("RelationInfo.Composite");
needsComma = true;
}
if (dlg.RelationName.Trim() != string.Empty)
{
if (needsComma)
{
sbResult.Append(", ");
}
sbResult.Append("\"" + dlg.RelationName + "\"");
}
sbResult.Append(")");
}
sbResult.Append(closeBracket + '\n');
if (isVC)
{
if (dlg.List)
{
sbResult.Append(spc + listLeft + " ");
}
else
{
sbResult.Append(spc + dlg.Type + " ");
}
sbResult.Append(dlg.FieldName);
if (dlg.List)
{
sbResult.Append(" = new " + listRight);
sbResult.Append("()");
}
sbResult.Append(";\n");
}
else
{
sbResult.Append(spc + "Dim ");
sbResult.Append(dlg.FieldName);
sbResult.Append(" As ");
if (dlg.List)
{
sbResult.Append(listLeft);
}
else
{
sbResult.Append(dlg.Type);
}
if (dlg.List)
{
sbResult.Append(" = new " + listRight);
sbResult.Append("()");
}
sbResult.Append("\n");
}
textDoc.Selection.Insert(sbResult.ToString(), (int)vsInsertFlags.vsInsertFlagsInsertAtStart);
}
catch (Exception ex)
{
MessageBox.Show(ex.Message, "Add Relation Add-in");
}
}