Pass workspace

Author: jcs090218

Handlers/CompletionHandler.cs

@@ -0,0 +1,47 @@
+using ShaderlabVS;
+
+namespace ShaderLS
+{
+    public class Buffer
+    {
+        private string _text = "";
+        private HashSet<string> _tokens = new();
+
+        public string GetText() { return _text; }
+        public HashSet<string> Tokens() { return _tokens; }
+
+        public Buffer(string text)
+        {
+            this._text = text;
+            SetWordsInDocuments(text);
+        }
+
+        private void SetWordsInDocuments(string text)
+        {
+            var reader = new StringReader(text);
+
+            string line = reader.ReadLine();
+
+            while (line != null)
+            {
+                if (Utilities.IsCommentLine(line))
+                {
+                    line = reader.ReadLine();
+                    continue;
+                }
+
+                string[] words = line.Split(
+                    new char[] { '{', '}', ' ', '\t', '(', ')', '[', ']', '+', '-', '*', '/', '%', '^', '>', '<', ':',
+                                '.', ';', '\"', '\'', '?', '\\', '&', '|', '`', '$', '#', ','},
+                    StringSplitOptions.RemoveEmptyEntries);
+
+                foreach (var word in words)
+                {
+                    _tokens.Add(word);
+                }
+
+                line = reader.ReadLine();
+            }
+        }
+    }
+}

Management/Workspace.cs

@@ -7,40 +7,45 @@ namespace ShaderLS
 {
     public class BufferService
     {
-        private readonly ConcurrentDictionary<DocumentUri, string> buffers = new();
+        private readonly ConcurrentDictionary<DocumentUri, Buffer> _buffers = new();
 
         public void Add(DocumentUri key, string text)
         {
-            buffers.TryAdd(key, text);
+            _buffers.TryAdd(key, new Buffer(text));
         }
 
         public void Remove(DocumentUri key)
         {
-            buffers.TryRemove(key, out _);
+            _buffers.TryRemove(key, out _);
         }
 
-        public string GetText(DocumentUri key)
+        public void ApplyFullChange(DocumentUri key, string text)
         {
-            return buffers[key];
+            var buffer = _buffers[key];
+            _buffers.TryUpdate(key, new Buffer(text), buffer);
         }
 
-        public void ApplyFullChange(DocumentUri key, string text)
+        public void ApplyIncrementalChange(DocumentUri key, Range range, string text)
         {
-            var buffer = buffers[key];
-            buffers.TryUpdate(key, text, buffer);
+            var buffer = _buffers[key];
+            var newText = Splice(buffer.GetText(), range, text);
+            _buffers.TryUpdate(key, new Buffer(newText), buffer);
         }
 
-        public void ApplyIncrementalChange(DocumentUri key, Range range, string text)
+        public string GetText(DocumentUri key)
+        {
+            return _buffers[key].GetText();
+        }
+
+        public HashSet<string> Tokens(DocumentUri key)
         {
-            var buffer = buffers[key];
-            var newText = Splice(buffer, range, text);
-            buffers.TryUpdate(key, newText, buffer);
+            return _buffers[key].Tokens();
         }
 
         private static int GetIndex(string buffer, Position position)
         {
             var index = 0;
-            for (var i = 0; i < position.Line; i++)
+            for (var i = 0; i < position.Line; ++i)
             {
                 index = buffer.IndexOf('\n', index) + 1;
             }