Ejemplo n.º 1
0
        private void GetCompleteStatement(
            ITextSnapshot snapshot, SnapshotPoint snapPos, out StringBuilder sbErrors, out ITree treeStmt)
        {
            string sql = snapshot.GetText();

            treeStmt  = null;
            sbErrors  = new StringBuilder();
            _position = snapPos.Position;
            _tokens   = RemoveToken(sql, snapPos);
            if (_tokens.Count == 1 && _tokens.Get(0).Type == MySQL51Lexer.EOF)
            {
                return;
            }
            MySQL51Parser.program_return r =
                LanguageServiceUtil.ParseSql(sql, false, out sbErrors, _tokens);
            if (r == null)
            {
                return;
            }
            ITree t = r.Tree as ITree;

            treeStmt = t;
            // locate current statement's AST
            if (t.IsNil)
            {
                ITree tmp = FindStmt(t);
                if (tmp != null)
                {
                    treeStmt = tmp;
                }
            }
        }
Ejemplo n.º 2
0
        /// <summary>
        /// Removes a token using the enhanced token stream class.
        /// </summary>
        /// <param name="sql"></param>
        /// <param name="snapPos"></param>
        /// <returns></returns>
        private CommonTokenStream RemoveToken(string sql, SnapshotPoint snapPos)
        {
            Version ver = LanguageServiceUtil.GetVersion(LanguageServiceUtil.GetConnection().ServerVersion);
            TokenStreamRemovable tokens = LanguageServiceUtil.GetTokenStream(sql, ver);
            IToken tr       = null;
            int    position = snapPos.Position;

            tokens.Fill();
            if (!char.IsWhiteSpace(snapPos.GetChar()))
            {
                foreach (IToken t in tokens.GetTokens())
                {
                    if ((t.StartIndex <= position) && (t.StopIndex >= position))
                    {
                        tr = t;
                        break;
                    }
                }
                tokens.Remove(tr);
            }
            return(tokens);
        }
Ejemplo n.º 3
0
        void ReParse()
        {
            ITextSnapshot newSnapshot = buffer.CurrentSnapshot;
            List <Region> newRegions  = new List <Region>();

            string        sql = newSnapshot.GetText();
            StringBuilder sb  = new StringBuilder();

            MySQL51Parser.statement_list_return r = LanguageServiceUtil.ParseSql(sql, false, out sb);

            if (sb.Length != 0)
            {
                return;
            }

            CommonTree ct = (CommonTree)r.Tree;

            foreach (ITree c in ct.Children)
            {
                newRegions.Add(new Region()
                {
                    Level         = 1,
                    StartLine     = c.Line,
                    StartOffset   = c.TokenStartIndex,
                    EndLine       = -1,
                    EndOffset     = c.TokenStopIndex,
                    PartialParent = null
                });
            }

            ITextSnapshot snap = this.snapshot;

            if (this.TagsChanged != null)
            {
                this.TagsChanged(this, new SnapshotSpanEventArgs(
                                     new SnapshotSpan(this.snapshot, Span.FromBounds(0, snap.Length))));
            }
        }
Ejemplo n.º 4
0
        void ICompletionSource.AugmentCompletionSession(ICompletionSession session, IList <CompletionSet> completionSets)
        {
            DbConnection connection = LanguageServiceUtil.GetConnection();

            if (connection != null)
            {
                string database = LanguageServiceUtil.GetCurrentDatabase();
                if (string.IsNullOrEmpty(database))
                {
                    database = connection.Database;
                }
                if (string.IsNullOrEmpty(database))
                {
                    database = "mysql";
                }
                if (session.TextView.Caret.Position.BufferPosition.Position == 0)
                {
                    return;
                }
                SnapshotPoint           currentPoint = (session.TextView.Caret.Position.BufferPosition) - 1;
                ITextStructureNavigator navigator    = _mSourceProvider.NavigatorService.GetTextStructureNavigator(_mTextBuffer);
                TextExtent    extent = navigator.GetExtentOfWord(currentPoint);
                ITrackingSpan span   = currentPoint.Snapshot.CreateTrackingSpan(extent.Span, SpanTrackingMode.EdgeInclusive);

                StringBuilder     sbErrors;
                ITextSnapshot     snapshot = currentPoint.Snapshot;
                ITextSnapshotLine line     = currentPoint.GetContainingLine();
                int position = currentPoint.Position;
                // Get starting token
                ITree t;
                GetCompleteStatement(snapshot, currentPoint, out sbErrors, out t);
                if ((snapshot.Length == 0) || (t == null))
                {
                    return;
                }

                string s             = sbErrors.ToString();
                Match  m             = new Regex(@"Expected (?<item>.*)\.").Match(s);
                string expectedToken = "";
                if (m.Success)
                {
                    expectedToken = m.Groups["item"].Value;
                }
                if (expectedToken == "table_factor" ||
                    expectedToken == "simple_table_ref_no_alias_existing")
                {
                    _mCompList = new List <Completion>();
                    DataTable schema = connection.GetSchema("Tables", new string[] { null, database });
                    schema.Merge(connection.GetSchema("Views", new string[] { null, database }));
                    string completionItem = null, completionItemUnq = null;

                    foreach (DataRow row in schema.Rows)
                    {
                        completionItemUnq = row["TABLE_NAME"].ToString();
                        completionItem    = string.Format("`{0}`", row["TABLE_NAME"].ToString());
                        _mCompList.Add(new Completion(completionItemUnq, completionItem, completionItem, null, null));
                    }

                    completionSets.Add(new CompletionSet(
                                           "MySqlTokens",  //the non-localized title of the tab
                                           "MySQL Tokens", //the display title of the tab
                                           FindTokenSpanAtPosition(session.GetTriggerPoint(_mTextBuffer), session),
                                           _mCompList,
                                           null));
                }
                if (expectedToken == "proc_name")
                {
                    _mCompList = new List <Completion>();
                    DataTable schema = connection.GetSchema("PROCEDURES WITH PARAMETERS", new string[] { null, database });
                    DataView  vi     = schema.DefaultView;
                    vi.Sort = "specific_name asc";
                    string completionItem = null;
                    string description    = null;
                    foreach (DataRowView row in vi)
                    {
                        if ("procedure".CompareTo(row["routine_type"].ToString().ToLower()) == 0)
                        {
                            completionItem = row["specific_name"].ToString();
                            description    = string.Format("procedure {0}.{1}({2})",
                                                           row["routine_schema"], row["specific_name"], row["ParameterList"]);
                            _mCompList.Add(new Completion(completionItem, completionItem,
                                                          description, null, null));
                        }
                    }

                    completionSets.Add(new CompletionSet(
                                           "MySqlTokens",  //the non-localized title of the tab
                                           "MySQL Tokens", //the display title of the tab
                                           FindTokenSpanAtPosition(session.GetTriggerPoint(_mTextBuffer), session),
                                           _mCompList,
                                           null));
                }
                else if (expectedToken == "column_name")
                {
                    if (t != null)
                    {
                        if ((t.ChildCount != 0) ||
                            ((t is CommonErrorNode) &&
                             ((t as CommonErrorNode).Text.Equals("SELECT",
                                                                 StringComparison.CurrentCultureIgnoreCase))))
                        {
                            List <TableWithAlias> tables = new List <TableWithAlias>();
                            ParserUtils.GetTables(t, tables);
                            List <string> cols = GetColumns(connection, tables, database);
                            CreateCompletionList(cols, session, completionSets);
                        }
                    }
                }
            }
        }
Ejemplo n.º 5
0
        public void AugmentQuickInfoSession(IQuickInfoSession session, IList <object> qiContent, out ITrackingSpan applicableToSpan)
        {
            applicableToSpan = null;
            if (m_dictionary == null)
            {
                DbConnection connection = LanguageServiceUtil.GetConnection();
                if (connection == null || string.IsNullOrEmpty(connection.Database))
                {
                    return;
                }
                LoadDictionary(connection);
            }
            // Map the trigger point down to our buffer.
            SnapshotPoint?subjectTriggerPoint = session.GetTriggerPoint(m_subjectBuffer.CurrentSnapshot);

            if (!subjectTriggerPoint.HasValue)
            {
                applicableToSpan = null;
                return;
            }

            ITextSnapshot currentSnapshot = subjectTriggerPoint.Value.Snapshot;
            SnapshotSpan  querySpan       = new SnapshotSpan(subjectTriggerPoint.Value, 0);

            //look for occurrences of our QuickInfo words in the span
            ITextStructureNavigator navigator =
                m_provider.NavigatorService.GetTextStructureNavigator(m_subjectBuffer);
            TextExtent extent     = navigator.GetExtentOfWord(subjectTriggerPoint.Value);
            string     searchText = extent.Span.GetText();

            foreach (string key in m_dictionary.Keys)
            {
                if (key == searchText)
                {
                    int foundIndex = 0;
                    int span       = querySpan.Start.Add(foundIndex).Position;
                    applicableToSpan = currentSnapshot.CreateTrackingSpan
                                       (
                        span,
                        Math.Min(
                            span + currentSnapshot.Length - subjectTriggerPoint.Value.Position,
                            currentSnapshot.Length - span),
                        SpanTrackingMode.EdgeInclusive
                                       );

                    string value;
                    m_dictionary.TryGetValue(key, out value);
                    if (value != null)
                    {
                        qiContent.Add(value);
                    }
                    else
                    {
                        qiContent.Add("");
                    }

                    return;
                }
            }

            applicableToSpan = null;
        }