Exemplo n.º 1
0
        protected override void CommandCallback(object sender, EventArgs e)
        {
            IWpfTextView textView = ServiceProvider.GetWpfTextView();

            if (textView == null)
            {
                return;
            }

            BqlFormatter formatter = CreateFormatter(textView);

            SnapshotPoint caretPosition = textView.Caret.Position.BufferPosition;
            Document      document      = caretPosition.Snapshot.GetOpenDocumentInCurrentContextWithChanges();

            if (document == null)
            {
                return;
            }

            (SyntaxNode syntaxRoot, SemanticModel semanticModel) = ThreadHelper.JoinableTaskFactory.Run(
                async() => (await document.GetSyntaxRootAsync(), await document.GetSemanticModelAsync()));

            if (syntaxRoot == null || semanticModel == null)
            {
                return;
            }

            SyntaxNode formattedRoot;

            if (textView.Selection.IsActive && !textView.Selection.IsEmpty)             // if has selection
            {
                // Find all nodes within the span and format them
                var        selectionSpan = TextSpan.FromBounds(textView.Selection.Start.Position, textView.Selection.End.Position);
                SyntaxNode topNode       = syntaxRoot.FindNode(selectionSpan);           // can, return top node that intersects with selectionSpan, so we need SpanWalker here

                if (topNode == null)
                {
                    return;                     // nothing to format (e.g. selection contains only trivia)
                }
                var spanWalker = new SpanWalker(selectionSpan);
                spanWalker.Visit(topNode);

                if (spanWalker.NodesWithinSpan.Count == 0)
                {
                    return;
                }

                formattedRoot = syntaxRoot.ReplaceNodes(spanWalker.NodesWithinSpan, (o, r) => formatter.Format(o, semanticModel));
            }
            else
            {
                formattedRoot = formatter.Format(syntaxRoot, semanticModel);
            }

            if (!textView.TextBuffer.EditInProgress)
            {
                var formattedDocument = document.WithSyntaxRoot(formattedRoot);
                ApplyChanges(document, formattedDocument);
            }
        }
Exemplo n.º 2
0
        private async System.Threading.Tasks.Task CommandCallbackAsync()
        {
            await ThreadHelper.JoinableTaskFactory.SwitchToMainThreadAsync();

            IWpfTextView textView = await ServiceProvider.GetWpfTextViewAsync();

            if (textView == null || Package.DisposalToken.IsCancellationRequested)
            {
                return;
            }

            SnapshotPoint caretPosition = textView.Caret.Position.BufferPosition;
            Document      document      = caretPosition.Snapshot.GetOpenDocumentInCurrentContextWithChanges();

            if (document == null)
            {
                return;
            }

            await TaskScheduler.Default;                //Go to background thread

            BqlFormatter formatter  = BqlFormatter.FromTextView(textView);
            SyntaxNode   syntaxRoot = await document.GetSyntaxRootAsync();

            SemanticModel semanticModel = await document.GetSemanticModelAsync();

            if (syntaxRoot == null || semanticModel == null)
            {
                return;
            }

            bool isPlatformReferenced = semanticModel.Compilation.GetTypeByMetadataName(TypeFullNames.PXGraph) != null;

            if (!isPlatformReferenced)
            {
                return;
            }

            SyntaxNode formattedRoot;

            if (textView.Selection.IsActive && !textView.Selection.IsEmpty)             // if has selection
            {
                // Find all nodes within the span and format them
                var        selectionSpan = TextSpan.FromBounds(textView.Selection.Start.Position, textView.Selection.End.Position);
                SyntaxNode topNode       = syntaxRoot.FindNode(selectionSpan);           // can, return top node that intersects with selectionSpan, so we need SpanWalker here

                if (topNode == null)
                {
                    return;                     // nothing to format (e.g. selection contains only trivia)
                }
                var spanWalker = new SpanWalker(selectionSpan);
                spanWalker.Visit(topNode);

                if (spanWalker.NodesWithinSpan.Count == 0)
                {
                    return;
                }

                formattedRoot = syntaxRoot.ReplaceNodes(spanWalker.NodesWithinSpan,
                                                        (oldNode, replacementNode) => formatter.Format(oldNode, semanticModel) ?? replacementNode);
            }
            else
            {
                formattedRoot = formatter.Format(syntaxRoot, semanticModel) ?? syntaxRoot;
            }

            await ThreadHelper.JoinableTaskFactory.SwitchToMainThreadAsync();             // Return to UI thread

            if (!textView.TextBuffer.EditInProgress && formattedRoot != syntaxRoot)
            {
                var formattedDocument = document.WithSyntaxRoot(formattedRoot);
                ApplyChanges(document, formattedDocument);
            }
        }