cpptripleslash/TripleSlashCompletionSource.cs at master · tcbhat/cpptripleslash · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
namespace CppTripleSlash
{
    using Microsoft.VisualStudio.Language.Intellisense;
    using Microsoft.VisualStudio.Text;
    using Microsoft.VisualStudio.Text.Operations;
    using System;
    using System.Collections.Generic;
    using System.Linq;
    using System.Windows.Media;

    public class TripleSlashCompletionSource : ICompletionSource
    {
        private TripleSlashCompletionSourceProvider m_sourceProvider;
        private ITextBuffer m_textBuffer;
        private List<Completion> m_compList = new List<Completion>();
        private bool m_isDisposed;

        public TripleSlashCompletionSource(TripleSlashCompletionSourceProvider sourceProvider, ITextBuffer textBuffer)
        {
            m_sourceProvider = sourceProvider;
            m_textBuffer = textBuffer;
            ImageSource image = null;

            try
            {
                image = this.m_sourceProvider.GlyphService.GetGlyph(StandardGlyphGroup.GlyphKeyword, StandardGlyphItem.GlyphItemPublic);
            }
            catch
            {
            }


            m_compList.Add(new Completion("<!-->", "<!---->", string.Empty, image, string.Empty));
            m_compList.Add(new Completion("<![CDATA[>", "<![CDATA[]]>", string.Empty, image, string.Empty));
            m_compList.Add(new Completion("<c>", "<c></c>", string.Empty, image, string.Empty));
            m_compList.Add(new Completion("<code>", "<code></code>", string.Empty, image, string.Empty));
            m_compList.Add(new Completion("<example>", "<example></example>", string.Empty, image, string.Empty));
            m_compList.Add(new Completion("<exception>", "<exception cref=\"\"></exception>", string.Empty, image, string.Empty));
            m_compList.Add(new Completion("<include>", "<include file='' path='[@name=\"\"]'/>", string.Empty, image, string.Empty));
            m_compList.Add(new Completion("<list>", "<list></list>", string.Empty, image, string.Empty));
            m_compList.Add(new Completion("<para>", "<para></para>", string.Empty, image, string.Empty));
            m_compList.Add(new Completion("<param>", "<param name=\"\"></param>", string.Empty, image, string.Empty));
            m_compList.Add(new Completion("<paramref>", "<paramref name=\"\"/>", string.Empty, image, string.Empty));
            m_compList.Add(new Completion("<permission>", "<permission cref=\"\"></permission>", string.Empty, image, string.Empty));
            m_compList.Add(new Completion("<remarks>", "<remarks></remarks>", string.Empty, image, string.Empty));
            m_compList.Add(new Completion("<returns>", "<returns></returns>", string.Empty, image, string.Empty));
            m_compList.Add(new Completion("<see>", "<see cref=\"\"/>", string.Empty, image, string.Empty));
            m_compList.Add(new Completion("<seealso>", "<seealso cref=\"\"/>", string.Empty, image, string.Empty));
            m_compList.Add(new Completion("<typeparam>", "<typeparam name=\"\"></typeparam>", string.Empty, image, string.Empty));
            m_compList.Add(new Completion("<typeparamref>", "<typeparamref name=\"\"/>", string.Empty, image, string.Empty));
            m_compList.Add(new Completion("<value>", "<value></value>", string.Empty, image, string.Empty));

        }

        void ICompletionSource.AugmentCompletionSession(ICompletionSession session, IList<CompletionSet> completionSets)
        {
            try
            {
                if (m_isDisposed)
                {
                    return;
                }

                SnapshotPoint? snapshotPoint = session.GetTriggerPoint(m_textBuffer.CurrentSnapshot);
                if (!snapshotPoint.HasValue)
                {
                    return;
                }

                string text = snapshotPoint.Value.GetContainingLine().GetText();
                if (m_textBuffer.ContentType.TypeName != TripleSlashCompletionCommandHandler.CppTypeName)
                {
                    return;
                }

                if (!text.TrimStart().StartsWith("///"))
                {
                    return;
                }

                ITrackingSpan trackingSpan = FindTokenSpanAtPosition(session.GetTriggerPoint(m_textBuffer), session);
                var newCompletionSet = new CompletionSet(
                    "TripleSlashCompletionSet",
                    "TripleSlashCompletionSet",
                    trackingSpan,
                    m_compList,
                    Enumerable.Empty<Completion>());
                completionSets.Add(newCompletionSet);
            }
            catch
            {
            }
        }

        public void Dispose()
        {
            if (!m_isDisposed)
            {
                GC.SuppressFinalize(this);
                m_isDisposed = true;
            }
        }

        private ITrackingSpan FindTokenSpanAtPosition(ITrackingPoint point, ICompletionSession session)
        {
            try
            {
                SnapshotPoint currentPoint = session.TextView.Caret.Position.BufferPosition - 1;
                return currentPoint.Snapshot.CreateTrackingSpan(currentPoint, 1, SpanTrackingMode.EdgeInclusive);
            }
            catch
            {
            }

            return null;
        }
    }
}