Add Uri to the extension call

Author: MikhailArkhipov

src/Analysis/Engine/Test/LanguageServerTests.cs

@@ -644,7 +644,7 @@ public Task<IReadOnlyDictionary<string, object>> ExecuteCommand(string command,
             #endregion
 
             #region ICompletionExtension
-            public Task HandleCompletionAsync(IModuleAnalysis analysis, PythonAst tree, SourceLocation location, CompletionList completions, CancellationToken token) {
+            public Task HandleCompletionAsync(Uri documentUri, IModuleAnalysis analysis, PythonAst tree, SourceLocation location, CompletionList completions, CancellationToken token) {
                 Assert.IsNotNull(tree);
                 Assert.IsNotNull(analysis);
                 for (int i = 0; i < completions.items.Length; ++i) {

src/LanguageServer/Impl/Extensions/ICompletionExtension.cs

@@ -14,6 +14,7 @@
 // See the Apache Version 2.0 License for specific language governing
 // permissions and limitations under the License.
 
+using System;
 using System.Threading;
 using System.Threading.Tasks;
 using Microsoft.PythonTools;
@@ -22,6 +23,6 @@
 
 namespace Microsoft.Python.LanguageServer.Extensions {
     public interface ICompletionExtension {
-        Task HandleCompletionAsync(IModuleAnalysis analysis, PythonAst tree, SourceLocation location, CompletionList completions, CancellationToken token);
+        Task HandleCompletionAsync(Uri documentUri, IModuleAnalysis analysis, PythonAst tree, SourceLocation location, CompletionList completions, CancellationToken token);
     }
 }

src/LanguageServer/Impl/Implementation/Server.Completion.cs

@@ -98,7 +98,7 @@ public override async Task<CompletionList> Completion(CompletionParams @params,
             }
 
             await InvokeExtensionsAsync((ext, token)
-                => (ext as ICompletionExtension)?.HandleCompletionAsync(analysis, tree, @params.position, res, cancellationToken), cancellationToken);
+                => (ext as ICompletionExtension)?.HandleCompletionAsync(uri, analysis, tree, @params.position, res, cancellationToken), cancellationToken);
 
             return res;
         }