preselect last inserted snippet in CocList

Author: chemzqm

Readme.md

@@ -175,7 +175,8 @@ some regex patterns can't be supported by javascript, including
 
 ## Commands
 
-- Use `:CocList snippets` to open snippets list used by current buffer.
+- Use `:CocList snippets` to open snippets list used by current buffer, the last
+  used snippet item would be preselected.
 - Use `:CocCommand snippets.openSnippetFiles` to choose and open a snippet file
   that used by current document.
 - Use `:CocCommand snippets.editSnippets` to edit user's ultisnips snippets of

src/index.ts

@@ -9,7 +9,7 @@ import { SnipmateProvider } from './snipmateProvider'
 import { TextmateProvider } from './textmateProvider'
 import { UltiSnipsConfig } from './types'
 import { getSnippetsDirectory, UltiSnippetsProvider } from './ultisnipsProvider'
-import { addFiletypes, getAdditionalFiletype, getSnippetFiletype, insertSnippetEdit, sameFile, waitDocument } from './util'
+import { addFiletypes, getAdditionalFiletype, getSnippetFiletype, insertSnippetEdit, sameFile, setLastSnippet, waitDocument } from './util'
 
 interface API {
   expandable: () => Promise<boolean>
@@ -114,6 +114,13 @@ export async function activate(context: ExtensionContext): Promise<API> {
     manager.loadSnippetsByFiletype(filetypes.join('.'))
   }))
 
+  events.on('CompleteDone', item => {
+    if (item && item['data'] && item['data']['snip']) {
+      let snip = item['data']['snip']
+      setLastSnippet(snip.filepath, snip.lnum)
+    }
+  }, null, context.subscriptions)
+
   let excludes = configuration.get<string[]>('excludePatterns', [])
   if (!Array.isArray(excludes)) excludes = []
   excludes = excludes.map(p => workspace.expand(p))

src/list/snippet.ts

@@ -5,7 +5,7 @@ Author Qiming Zhao <chemzqm@gmail> (https://github.com/chemzqm)
 import { BasicList, ListContext, ListItem, Location, Position, Range, Uri, workspace } from 'coc.nvim'
 import os from 'os'
 import { ProviderManager } from '../provider'
-import { getSnippetFiletype } from '../util'
+import { getLastSnippet, getSnippetFiletype } from '../util'
 
 function formatPrefix(prefix: string): string {
   if (prefix.length >= 20) return prefix.slice(0, 17) + '...'
@@ -30,13 +30,16 @@ export default class SnippetsList extends BasicList {
     let filetype = getSnippetFiletype(doc)
     let snippets = this.manager.getSnippets(filetype)
     let res: ListItem[] = []
+    let last = getLastSnippet()
     for (let snip of snippets) {
       let pos: Position = Position.create(snip.lnum, 0)
       let location: Location = Location.create(Uri.file(snip.filepath).toString(), Range.create(pos, Position.create(snip.lnum, 1)))
       let prefix = snip.prefix.length ? snip.prefix : snip.originRegex ?? ''
+      let preselect = last && last.filepath == snip.filepath && last.lnum == snip.lnum
       res.push({
-        label: `${formatPrefix(prefix)}\t${snip.description}\t${snip.filepath.replace(os.homedir(), '~')}`,
+        label: `${formatPrefix(prefix)}\t${snip.description}\t${snip.filepath.replace(os.homedir(), '~')}:${snip.lnum}`,
         filterText: `${snip.prefix} ${snip.description}`,
+        preselect,
         location,
         data: { prefix }
       })

src/massCodeProvider.ts

@@ -1,7 +1,7 @@
 import { Document, OutputChannel, Position, Range, window, workspace } from 'coc.nvim'
 import http from 'http'
 
-import type { Snippet, SnippetEdit, MassCodeConfig } from './types'
+import type { MassCodeConfig, Snippet, SnippetEdit } from './types'
 import { TriggerKind } from './types'
 
 import BaseProvider from './baseProvider'
@@ -122,6 +122,7 @@ export class MassCodeProvider extends BaseProvider {
         prefix: s.prefix,
         description: s.description,
         location: s.filepath,
+        lnum: s.lnum,
         priority: s.priority,
         regex: s.originRegex,
         context: s.context,

src/snipmateProvider.ts

@@ -200,6 +200,7 @@ export class SnipmateProvider extends BaseProvider {
         prefix: s.prefix,
         description: s.description,
         location: s.filepath,
+        lnum: s.lnum,
         range,
         newText: s.body,
         priority: -1

src/textmateProvider.ts

@@ -165,6 +165,7 @@ export class TextmateProvider extends BaseProvider {
         range: Range.create(position.line, position.character - prefix.length, position.line, position.character),
         newText: snip.body,
         location: snip.filepath,
+        lnum: snip.lnum,
         description: snip.description,
         priority: snip.priority ?? -1
       })

src/types.ts

@@ -126,6 +126,7 @@ export interface SnippetEdit {
   prefix: string
   newText: string
   location: string
+  lnum: number
   description: string
   priority: number
   regex?: string

src/ultisnipsProvider.ts

@@ -228,6 +228,7 @@ export class UltiSnippetsProvider extends BaseProvider {
         prefix: s.prefix,
         description: s.description,
         location: s.filepath,
+        lnum: s.lnum,
         priority: s.priority,
         regex: s.originRegex,
         context: s.context,

src/util.ts

@@ -2,24 +2,38 @@
 MIT License http://www.opensource.org/licenses/mit-license.php
 Author Qiming Zhao <chemzqm@gmail> (https://github.com/chemzqm)
 *******************************************************************/
-import { commands, workspace, Document, TextEdit, Uri } from 'coc.nvim'
-import { promisify } from 'util'
+import { commands, Document, TextEdit, Uri, workspace } from 'coc.nvim'
 import crypto from 'crypto'
 import fs from 'fs'
 import os from 'os'
 import path from 'path'
+import { promisify } from 'util'
 import { ReplaceItem, SnippetEditWithSource, UltiSnippetOption } from './types'
 
 export interface CodeInfo {
   readonly hash: string
   readonly code: string
 }
 
+export interface LastSnippet {
+  filepath: string
+  lnum: number
+}
+
 export const pythonCodes: Map<string, CodeInfo> = new Map()
 
 const caseInsensitive = os.platform() == 'win32' || os.platform() == 'darwin'
 const BASE64 = '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz-_'
 const additionalFiletypes: Map<number, string[]> = new Map()
+var lastSnippet: LastSnippet = undefined
+
+export function setLastSnippet(filepath: string, lnum: number): void {
+  lastSnippet = { filepath, lnum }
+}
+
+export function getLastSnippet(): LastSnippet | undefined {
+  return lastSnippet
+}
 
 export async function insertSnippetEdit(edit: SnippetEditWithSource) {
   let ultisnips = edit.source == 'ultisnips' || edit.source == 'snipmate'
@@ -35,6 +49,7 @@ export async function insertSnippetEdit(edit: SnippetEditWithSource) {
       removeWhiteSpace: formatOptions.removeWhiteSpace
     }
   }
+  setLastSnippet(edit.location, edit.lnum)
   await commands.executeCommand('editor.action.insertSnippet', TextEdit.replace(edit.range, edit.newText), option)
 }