Beispiel #1
0
        /// <summary>
        /// Creates a DBMLTable for a document, with the details held in xml element
        /// </summary>
        /// <param name="doc">The DBMLDocument this table belongs to</param>
        /// <param name="e">The XmlElement containing the details of the table</param>
        public DBMLTable(DBMLDoc doc, XmlElement e)
        {
            _elem = e;
            _tableName = _elem.GetAttribute("Name");
            if(string.IsNullOrEmpty(_tableName))
                throw new ArgumentException("Cannot get table name");

            XmlNodeList nl = _elem.GetElementsByTagName("Type");
            if(nl.Count != 1 || nl[0].NodeType != XmlNodeType.Element)
                throw new ArgumentException("Cannot Identity type for table " + _tableName);
            _typeElem = nl[0] as XmlElement;

            _cols = new DBMLColumnList();
            try
            {
                foreach(XmlNode n in _typeElem.ChildNodes)
                {
                    XmlElement ce = n as XmlElement;
                    if(ce != null && ce.Name == "Column")
                        _cols.Add(new DBMLColumn(ce));
                }
            }
            catch(Exception ex)
            {
                throw new ArgumentException(string.Format("Problem parsing table {0} {1}", _tableName, ex.Message));
            }
        }
Beispiel #2
0
        /// <summary>
        /// Creates an Xml element with the details of the table
        /// </summary>
        /// <param name="doc">The parent doc</param>
        /// <returns>XmlElement for the new document</returns>
        public XmlElement CreateDBML(DBMLDoc doc)
        {
            XmlElement table = doc.AddElement("Table");

            doc.AddAttributes(table, _elem.Attributes, string.Empty);

            XmlElement type = doc.AddElement("Type");

            doc.AddAttributes(type, _typeElem.Attributes, string.Empty);

            _cols.Sort();
            foreach (DBMLColumn c in _cols)
            {
                if (c.Found)
                {
                    type.AppendChild(c.CreateDBML(doc));
                }
            }

            doc.AddNodes(type, _typeElem.ChildNodes, "Column");
            table.AppendChild(type);

            doc.AddNodes(table, _elem.ChildNodes, "Type");
            return(table);
        }
Beispiel #3
0
        /// <summary>
        /// Creates a DBMLTable for a document, with the details held in xml element
        /// </summary>
        /// <param name="doc">The DBMLDocument this table belongs to</param>
        /// <param name="e">The XmlElement containing the details of the table</param>
        public DBMLTable(DBMLDoc doc, XmlElement e)
        {
            _elem      = e;
            _tableName = _elem.GetAttribute("Name");
            if (string.IsNullOrEmpty(_tableName))
            {
                throw new ArgumentException("Cannot get table name");
            }

            XmlNodeList nl = _elem.GetElementsByTagName("Type");

            if (nl.Count != 1 || nl[0].NodeType != XmlNodeType.Element)
            {
                throw new ArgumentException("Cannot Identity type for table " + _tableName);
            }
            _typeElem = nl[0] as XmlElement;

            _cols = new DBMLColumnList();
            try
            {
                foreach (XmlNode n in _typeElem.ChildNodes)
                {
                    XmlElement ce = n as XmlElement;
                    if (ce != null && ce.Name == "Column")
                    {
                        _cols.Add(new DBMLColumn(ce));
                    }
                }
            }
            catch (Exception ex)
            {
                throw new ArgumentException(string.Format("Problem parsing table {0} {1}", _tableName, ex.Message));
            }
        }
Beispiel #4
0
        /// <summary>
        /// Creates an Xml Element with the details of the column for a new DBML file
        /// </summary>
        /// <param name="doc">The parent document</param>
        /// <returns>An Xml Element cotaining the details of the column</returns>
        public XmlElement CreateDBML(DBMLDoc doc)
        {
            if (!_found)
            {
                return(null);
            }

            XmlElement col = doc.AddElement("Column");

            doc.AddAttribute(col, "Name", _name);
            doc.AddAttribute(col, "Type", clrType);
            doc.AddAttribute(col, "DbType", dbType);
            if (_isPrimaryKey)
            {
                doc.AddAttribute(col, "IsPrimaryKey", "true");
            }
            if (_isIdentity || _dbGenerated)
            {
                doc.AddAttribute(col, "IsDbGenerated", "true");
            }
            doc.AddAttribute(col, "CanBeNull", _canBeNull ? "true" : "false");

            if (_elem != null)
            {
                doc.AddAttributes(col, _elem.Attributes, COL_KNOWNATTR);
            }

            return(col);
        }
Beispiel #5
0
        /// <summary>Implements the Exec method of the IDTCommandTarget interface. This is called when the command is invoked.</summary>
        /// <param term='commandName'>The name of the command to execute.</param>
        /// <param term='executeOption'>Describes how the command should be run.</param>
        /// <param term='varIn'>Parameters passed from the caller to the command handler.</param>
        /// <param term='varOut'>Parameters passed from the command handler to the caller.</param>
        /// <param term='handled'>Informs the caller if the command was handled or not.</param>
        /// <seealso class='Exec' />
        public void Exec(string commandName, vsCommandExecOption executeOption, ref object varIn, ref object varOut, ref bool handled)
        {
            handled = false;
            if(executeOption == vsCommandExecOption.vsCommandExecOptionDoDefault)
            {
                if(commandName == "DBMLUpdate.Connect.DBMLUpdate")
                {
                    // Some initial setup create an output pain and grab status bar
                    if(_output == null)
                        _output = _applicationObject.ToolWindows.OutputWindow.OutputWindowPanes.Add("DBML Update");
                    _output.Clear();
                    EnvDTE.StatusBar status = _applicationObject.StatusBar;
                    _output.Activate();

                    try
                    {
                        //Check where dealing with a dbml file
                        if(_applicationObject.ActiveDocument != null && _applicationObject.ActiveDocument.FullName.EndsWith(".dbml", StringComparison.InvariantCultureIgnoreCase))
                        {
                            _output.OutputString("Beginning Processing...\r\n");

                            // Grab dbml and save it if its unsaved
                            Document actDoc = _applicationObject.ActiveDocument;
                            ProjectItem pi = actDoc.ProjectItem;
                            string projPath = pi.ContainingProject.FullName;
                            string docFileName = actDoc.FullName;
                            if(!actDoc.Saved)
                            {
                                status.Text = "Saving Linq-to-SQL desinger diagram...";
                                actDoc.Save(docFileName);
                            }

                            UpdateStatus(_output, status, "Parsing DBML file...");
                            // Real work begins in DBMLDoc
                            DBMLDoc doc = new DBMLDoc(docFileName, pi);
                            _output.OutputString(string.Format("Parsed {0} Tables\r\n", doc.TableCount));

                            UpdateStatus(_output, status, "Updating from Database...");
                            doc.DBUpdate(_output);

                            status.Text = "Closing the DBML designer...";
                            actDoc.Close(actDoc.Saved ? vsSaveChanges.vsSaveChangesNo : vsSaveChanges.vsSaveChangesYes);

                            UpdateStatus(_output, status, "Creating new DBML file...");
                            doc.CreateDBML(docFileName, _output);

                            status.Text = "Opening designer...";
                            Window win = OpenDesigner(pi, docFileName);

                            UpdateStatus(_output, status, "Re-generating classes...");
                            try
                            {
                                ((VSProjectItem)win.ProjectItem.Object).RunCustomTool();
                            }
                            catch(Exception ex)
                            {
                                try
                                {
                                    win.Document.Save(docFileName);
                                }
                                catch
                                {
                                    UpdateStatus(_output, status, "Code re-generation failed: " + ex.Message);
                                }
                            }

                            _output.OutputString("Finished Processing\r\n");
                        }
                        else
                            _output.OutputString("The active document is not a Linq-to-SQL file.\r\n");
                    }
                    catch(Exception ex)
                    {
                        _output.OutputString("Refreshing entities failed due to exception:" + ex.ToString() + " Message:" + ex.Message + "\r\n");
                    }
                    finally
                    {
                        status.Text = string.Empty;
                        status.Progress(false, "", 0, 0);
                        if(_output != null)
                            _output.Activate();
                    }

                    handled = true;
                }
            }
        }
Beispiel #6
0
        /// <summary>Implements the Exec method of the IDTCommandTarget interface. This is called when the command is invoked.</summary>
        /// <param term='commandName'>The name of the command to execute.</param>
        /// <param term='executeOption'>Describes how the command should be run.</param>
        /// <param term='varIn'>Parameters passed from the caller to the command handler.</param>
        /// <param term='varOut'>Parameters passed from the command handler to the caller.</param>
        /// <param term='handled'>Informs the caller if the command was handled or not.</param>
        /// <seealso class='Exec' />
        public void Exec(string commandName, vsCommandExecOption executeOption, ref object varIn, ref object varOut, ref bool handled)
        {
            handled = false;
            if (executeOption == vsCommandExecOption.vsCommandExecOptionDoDefault)
            {
                if (commandName == "DBMLUpdate.Connect.DBMLUpdate")
                {
                    // Some initial setup create an output pain and grab status bar
                    if (_output == null)
                    {
                        _output = _applicationObject.ToolWindows.OutputWindow.OutputWindowPanes.Add("DBML Update");
                    }
                    _output.Clear();
                    EnvDTE.StatusBar status = _applicationObject.StatusBar;
                    _output.Activate();

                    try
                    {
                        //Check where dealing with a dbml file
                        if (_applicationObject.ActiveDocument != null && _applicationObject.ActiveDocument.FullName.EndsWith(".dbml", StringComparison.InvariantCultureIgnoreCase))
                        {
                            _output.OutputString("Beginning Processing...\r\n");

                            // Grab dbml and save it if its unsaved
                            Document    actDoc      = _applicationObject.ActiveDocument;
                            ProjectItem pi          = actDoc.ProjectItem;
                            string      projPath    = pi.ContainingProject.FullName;
                            string      docFileName = actDoc.FullName;
                            if (!actDoc.Saved)
                            {
                                status.Text = "Saving Linq-to-SQL desinger diagram...";
                                actDoc.Save(docFileName);
                            }

                            UpdateStatus(_output, status, "Parsing DBML file...");
                            // Real work begins in DBMLDoc
                            DBMLDoc doc = new DBMLDoc(docFileName, pi);
                            _output.OutputString(string.Format("Parsed {0} Tables\r\n", doc.TableCount));

                            UpdateStatus(_output, status, "Updating from Database...");
                            doc.DBUpdate(_output);

                            status.Text = "Closing the DBML designer...";
                            actDoc.Close(actDoc.Saved ? vsSaveChanges.vsSaveChangesNo : vsSaveChanges.vsSaveChangesYes);

                            UpdateStatus(_output, status, "Creating new DBML file...");
                            doc.CreateDBML(docFileName, _output);

                            status.Text = "Opening designer...";
                            Window win = OpenDesigner(pi, docFileName);

                            UpdateStatus(_output, status, "Re-generating classes...");
                            try
                            {
                                ((VSProjectItem)win.ProjectItem.Object).RunCustomTool();
                            }
                            catch (Exception ex)
                            {
                                try
                                {
                                    win.Document.Save(docFileName);
                                }
                                catch
                                {
                                    UpdateStatus(_output, status, "Code re-generation failed: " + ex.Message);
                                }
                            }

                            _output.OutputString("Finished Processing\r\n");
                        }
                        else
                        {
                            _output.OutputString("The active document is not a Linq-to-SQL file.\r\n");
                        }
                    }
                    catch (Exception ex)
                    {
                        _output.OutputString("Refreshing entities failed due to exception:" + ex.ToString() + " Message:" + ex.Message + "\r\n");
                    }
                    finally
                    {
                        status.Text = string.Empty;
                        status.Progress(false, "", 0, 0);
                        if (_output != null)
                        {
                            _output.Activate();
                        }
                    }

                    handled = true;
                }
            }
        }
Beispiel #7
0
        /// <summary>
        /// Creates an Xml element with the details of the table
        /// </summary>
        /// <param name="doc">The parent doc</param>
        /// <returns>XmlElement for the new document</returns>
        public XmlElement CreateDBML(DBMLDoc doc)
        {
            XmlElement table = doc.AddElement("Table");
            doc.AddAttributes(table, _elem.Attributes, string.Empty);

            XmlElement type = doc.AddElement("Type");
            doc.AddAttributes(type, _typeElem.Attributes, string.Empty);

            _cols.Sort();
            foreach(DBMLColumn c in _cols)
            {
                if(c.Found)
                    type.AppendChild(c.CreateDBML(doc));
            }

            doc.AddNodes(type, _typeElem.ChildNodes, "Column");
            table.AppendChild(type);

            doc.AddNodes(table, _elem.ChildNodes, "Type");
            return table;
        }