feat(selectRange): add select range support

Author: iamcco

.vim/coc-settings.json

@@ -3,6 +3,7 @@
     "Runtimepath",
     "Vimruntime",
     "Workdir",
+    "endfunction",
     "fitem",
     "fitems",
     "flist",
@@ -12,6 +13,7 @@
     "lnum",
     "msglog",
     "neco",
+    "rlist",
     "runtimepath's",
     "runtimepaths",
     "shvl",

README.md

@@ -16,6 +16,7 @@
 - go to references
 - document highlight
 - folding range
+- select range
 - rename
 - snippets
 - diagnostic

src/handles/selectionRange.ts

@@ -1,27 +1,72 @@
-import {FoldingRange, FoldingRangeParams} from "vscode-languageserver";
+import {SelectionRangeParams, SelectionRange, Range, Position} from "vscode-languageserver";
 
 import {workspace} from "../server/workspaces";
+import {documents} from "../server/documents";
 
-export const selectionRangeProvider = (params: FoldingRangeParams) => {
-  const res: FoldingRange[] = [];
-  const { textDocument } = params;
+export const selectionRangeProvider = (params: SelectionRangeParams): SelectionRange[] => {
+  const selectRanges: SelectionRange[] = [];
+  const { textDocument, positions } = params;
+  if (!positions || positions.length === 0) {
+    return selectRanges
+  }
   const buffer = workspace.getBufferByUri(textDocument.uri);
-  if (!buffer) {
-    return res;
+  const document = documents.get(textDocument.uri)
+  if (!buffer || !document) {
+    return selectRanges;
   }
   const globalFunctions = buffer.getGlobalFunctions();
   const scriptFunctions = buffer.getScriptFunctions();
-  return Object.values(globalFunctions).concat(Object.values(scriptFunctions))
+
+  const funcs = Object.values(globalFunctions).concat(Object.values(scriptFunctions))
     .reduce((pre, cur) => {
       return pre.concat(cur);
     }, [])
-    .map<FoldingRange>((func) => {
-      return {
-        startLine: func.startLine - 1,
-        startCharacter: func.startCol - 1,
-        endLine: func.endLine - 1,
-        endCharacter: func.endCol - 1,
-        kind: "region",
-      };
-    });
+  let range = Range.create(positions[0], positions[0])
+  if (positions.length > 1) {
+    range = Range.create(positions[0], positions[positions.length - 1])
+  }
+  const ranges: Range[] = []
+  funcs.forEach(item => {
+    const p = item.range
+    const line = document.getText(Range.create(
+      Position.create(p.endLine - 1, 0),
+      Position.create(p.endLine, 0)
+    ))
+    const newRange = Range.create(
+      Position.create(p.startLine - 1, p.startCol - 1),
+      Position.create(p.endLine - 1, p.endCol - 1 + line.slice(p.endCol - 1).split(' ')[0].length)
+    )
+    if (range.start.line >= newRange.start.line
+      && range.end.line <= newRange.end.line
+      && !(range.start.line === newRange.start.line && range.end.line === newRange.end.line)) {
+      if (ranges.length === 0) {
+        ranges.push(newRange)
+      } else {
+        let i = 0;
+        for(const len = ranges.length; i < len; i++) {
+          if (ranges[i].start.line <= newRange.start.line && ranges[i].end.line >= newRange.end.line) {
+            ranges.splice(i, 0, newRange)
+            break
+          }
+        }
+        if (i === ranges.length) {
+          ranges.push(newRange)
+        }
+      }
+    }
+  })
+  if (ranges.length) {
+    selectRanges.push(
+      ranges.reverse().reduce((pre, cur, idx) => {
+        if (idx === 0) {
+          return pre
+        }
+        return {
+          range: cur,
+          parent: pre
+        }
+      }, {range: ranges[0]} as SelectionRange)
+    )
+  }
+  return selectRanges
 };

src/index.ts

@@ -17,6 +17,7 @@ import config from "./server/config";
 import { connection } from "./server/connection";
 import { documents } from "./server/documents";
 import { next, unsubscribe } from "./server/parser";
+import {selectionRangeProvider} from "./handles/selectionRange";
 
 // lsp initialize
 connection.onInitialize((param: InitializeParams) => {
@@ -74,6 +75,7 @@ connection.onInitialize((param: InitializeParams) => {
       textDocumentSync: TextDocumentSyncKind.Incremental,
       documentHighlightProvider: true,
       foldingRangeProvider: true,
+      selectionRangeProvider: true,
       hoverProvider: true,
       completionProvider: {
         triggerCharacters: [".", ":", "#", "[", "&", "$", "<", '"', "'"],
@@ -131,5 +133,8 @@ connection.onDocumentHighlight(documentHighlightProvider);
 // folding range
 connection.onFoldingRanges(foldingRangeProvider);
 
+// select range
+connection.onSelectionRanges(selectionRangeProvider)
+
 // lsp start
 connection.listen();

src/lib/vimparser.d.ts

@@ -33,7 +33,7 @@ export declare interface INode {
 
 export declare class StringReader {
   public buf: string[];
-  public pos: Array<[number, number, number]>;
+  public pos: [number, number, number][];
   constructor(lines: string[])
 }
 

src/server/buffer.ts

@@ -111,6 +111,12 @@ export interface IFunction {
   startCol: number;
   endLine: number;
   endCol: number;
+  range: {
+    startLine: number;
+    startCol: number;
+    endLine: number;
+    endCol: number;
+  }
 }
 
 export interface IFunRef {
@@ -570,6 +576,12 @@ export class Buffer {
       startCol: pos.col,
       endLine: endfunction!.pos.lnum,
       endCol: endfunction!.pos.col,
+      range: {
+        startLine: node.pos.lnum,
+        startCol: node.pos.col,
+        endLine: endfunction!.pos.lnum,
+        endCol: endfunction!.pos.col,
+      }
     };
     if (globalFuncPattern.test(name)) {
       if (!this.globalFunctions[name] || !Array.isArray(this.globalFunctions[name])) {
@@ -618,6 +630,12 @@ export class Buffer {
       startCol: pos.col,
       endLine: pos.lnum,
       endCol: pos.col,
+      range: {
+        startLine: pos.lnum,
+        startCol: pos.col,
+        endLine: pos.lnum,
+        endCol: pos.col,
+      }
     };
     if (globalFuncPattern.test(name)) {
       if (!this.globalFunctions[name] || !Array.isArray(this.globalFunctions[name])) {