public TreeViewNode(string name,TreeViewNode parent)
{
this.Name = name;
this.Children = new List<TreeViewNode>();
this._parent =parent;
this.IsNodeEnabled = true;
}
private void PopulateTree(List<DBTableInfo> sqlTableList, SPGenerator.UserControls.TreeViewWithCheckBox treeView1)
{
TreeViewNode root = new TreeViewNode(Constants.rootTreeNodeText, null);
foreach (var tbl in sqlTableList)
{
root.Children.Add(CreateTableNode(tbl, root));
}
root.IsNodeExpanded = true;
treeView1.DataContext = new List<TreeViewNode> { root };
rootNode = root;
}
private List<DBTableColumnInfo> GetWhereClauseFields(TreeViewNode tableChildNode)
{
var selectedFields = new List<DBTableColumnInfo>();
var whereClauseNode = GetChildNodeByText(tableChildNode, Constants.whereConditionTreeNodeText);
if (whereClauseNode == null)
return selectedFields;
foreach (TreeViewNode fieldNode in whereClauseNode.Children)
{
if (fieldNode.IsChecked ?? true)
{
selectedFields.Add((DBTableColumnInfo)fieldNode.Tag);
}
}
return selectedFields;
}
private List<DBTableColumnInfo> GetSelectedFields(TreeViewNode tableChildNode)
{
var selectedFields = new List<DBTableColumnInfo>();
foreach (TreeViewNode fieldNode in tableChildNode.Children)
{
if ((fieldNode.IsChecked ?? true) && fieldNode.Name != Constants.whereConditionTreeNodeText)
{
selectedFields.Add((DBTableColumnInfo)fieldNode.Tag);
}
}
return selectedFields;
}
private TreeViewNode GetChildNodeByText(TreeViewNode parentNode, string childNodeText)
{
var childNode = parentNode.Children.OfType<TreeViewNode>()
.FirstOrDefault(node => node.Name.Equals(childNodeText));
return childNode;
}
private void GenerateSPForSingleTable(TreeViewNode tblNode, StringBuilder sb)
{
foreach (TreeViewNode childNode in tblNode.Children)
{
var selectedFields = GetSelectedFields(childNode);
var whereClauseSelectedFields = GetWhereClauseFields(childNode);
if (childNode.IsChecked ?? true)
{
model.GenerateSp(tblNode.Name, childNode.Name, ref sb, selectedFields, whereClauseSelectedFields);
}
}
}
private TreeViewNode CreateTableNode(DBTableInfo sqlTableInfo, TreeViewNode parent)
{
TreeViewNode tblNode = new TreeViewNode(sqlTableInfo.TableName, parent);
tblNode.Tag = sqlTableInfo;
TreeViewNode insertSp = new TreeViewNode(Constants.insertTreeNodeText, tblNode);
TreeViewNode updateSp = new TreeViewNode(Constants.updateTreeNodeText, tblNode);
TreeViewNode whereCondition = new TreeViewNode(Constants.whereConditionTreeNodeText, updateSp);
AddColumnNodes(insertSp, sqlTableInfo.Columns, true);
AddColumnNodes(updateSp, sqlTableInfo.Columns, true);
AddColumnNodes(whereCondition, sqlTableInfo.Columns, false);
updateSp.Children.Add(whereCondition);
tblNode.Children.Add(insertSp);
tblNode.Children.Add(updateSp);
return tblNode;
}
private TreeViewNode AddColumnNodes(TreeViewNode parentNode, List<DBTableColumnInfo> columns, bool disableExludeColumn)
{
foreach (var colInfo in columns)
{
TreeViewNode colNode = new TreeViewNode(GetColumnDispalyName(colInfo), parentNode);
colNode.Tag = colInfo;
if (disableExludeColumn)
{
colNode.IsNodeEnabled = !colInfo.Exclude;
}
parentNode.Children.Add(colNode);
}
return parentNode;
}
