From 7d6d8b2494a30121ea21e9064bf0b1f7ea87e1ce Mon Sep 17 00:00:00 2001
From: Heyward Fann <fannheyward@gmail.com>
Date: Mon, 25 Mar 2024 19:24:26 +0800
Subject: [PATCH] feat(list): add --buffer to list buffer diags

---
 .github/ISSUE_TEMPLATE/bug_report.md |  2 +-
 README.md                            |  2 +-
 autoload/coc/list.vim                |  2 +-
 src/__tests__/list/sources.test.ts   | 17 +++++++++++++++++
 src/list/source/diagnostics.ts       | 15 +++++++++++++--
 5 files changed, 33 insertions(+), 5 deletions(-)

diff --git a/.github/ISSUE_TEMPLATE/bug_report.md b/.github/ISSUE_TEMPLATE/bug_report.md
index 3b0eaebbd70..d881da861d1 100644
--- a/.github/ISSUE_TEMPLATE/bug_report.md
+++ b/.github/ISSUE_TEMPLATE/bug_report.md
@@ -6,7 +6,7 @@ about: Create a report to help us improve
 <!--
 **Warning: We will close the bug issue without the issue template and the reproduce ways.**
 
-If you have question, please ask at https://gitter.im/neoclide/coc.nvim
+If you have question, please ask at https://github.com/neoclide/coc.nvim/discussions
 
 If the problem related to specific language server, please checkout: https://git.io/fjCEM
 
diff --git a/README.md b/README.md
index e8dc875e4f5..5d24b414c2e 100644
--- a/README.md
+++ b/README.md
@@ -8,7 +8,7 @@
     <a href="https://github.com/neoclide/coc.nvim/actions"><img alt="Actions" src="https://img.shields.io/github/actions/workflow/status/neoclide/coc.nvim/ci.yml?style=flat-square&branch=master"></a>
       <a href="https://codecov.io/gh/neoclide/coc.nvim"><img alt="Codecov Coverage Status" src="https://img.shields.io/codecov/c/github/neoclide/coc.nvim.svg?style=flat-square"></a>
     <a href="doc/coc.txt"><img alt="Doc" src="https://img.shields.io/badge/doc-%3Ah%20coc.txt-brightgreen.svg?style=flat-square"></a>
-    <a href="https://matrix.to/#/#coc.nvim:matrix.org"><img alt="Gitter" src="https://img.shields.io/gitter/room/neoclide/coc.nvim.svg?style=flat-square"></a>
+    <a href="https://matrix.to/#/#coc.nvim:matrix.org"><img alt="Matrix" src="https://img.shields.io/matrix/coc.nvim%3Amatrix.org?style=flat-square"></a>
   </p>
 </p>
 
diff --git a/autoload/coc/list.vim b/autoload/coc/list.vim
index 9212f223575..dc1af948dad 100644
--- a/autoload/coc/list.vim
+++ b/autoload/coc/list.vim
@@ -40,7 +40,7 @@ function! coc#list#setlines(bufnr, lines, append)
 endfunction
 
 function! coc#list#options(...)
-  let list = ['--top', '--tab', '--normal', '--no-sort', '--input=', '--strict',
+  let list = ['--top', '--tab', '--buffer', '--normal', '--no-sort', '--input=', '--strict',
         \ '--regex', '--interactive', '--number-select', '--auto-preview',
         \ '--ignore-case', '--no-quit', '--first', '--reverse', '--height=']
   if get(g:, 'coc_enabled', 0)
diff --git a/src/__tests__/list/sources.test.ts b/src/__tests__/list/sources.test.ts
index 9d379f93ce5..0a92aecd3ac 100644
--- a/src/__tests__/list/sources.test.ts
+++ b/src/__tests__/list/sources.test.ts
@@ -662,9 +662,26 @@ describe('list sources', () => {
 
     it('should load diagnostics source', async () => {
       await createDocument('a')
+      await createDocument('b')
       await manager.start(['diagnostics'])
       await manager.session?.ui.ready
       expect(manager.isActivated).toBe(true)
+
+      let buf = await nvim.buffer
+      let lines = await buf.lines
+      expect(lines.length).toEqual(10)
+    })
+
+    it('should load diagnostics for buffer only', async () => {
+      await createDocument('a')
+      await createDocument('b')
+      await manager.start(['diagnostics', '--buffer'])
+      await manager.session?.ui.ready
+      expect(manager.isActivated).toBe(true)
+
+      let buf = await nvim.buffer
+      let lines = await buf.lines
+      expect(lines.length).toEqual(5)
     })
 
     it('should refresh on diagnostics refresh', async () => {
diff --git a/src/list/source/diagnostics.ts b/src/list/source/diagnostics.ts
index b3660aa84eb..b37476340b4 100644
--- a/src/list/source/diagnostics.ts
+++ b/src/list/source/diagnostics.ts
@@ -1,11 +1,13 @@
 'use strict'
+import { URI } from 'vscode-uri'
 import diagnosticManager, { DiagnosticItem } from '../../diagnostic/manager'
 import { defaultValue } from '../../util'
 import { isParentFolder } from '../../util/fs'
 import { path } from '../../util/node'
+import workspace from '../../workspace'
 import { formatListItems, formatPath, PathFormatting, UnformattedListItem } from '../formatting'
 import { ListManager } from '../manager'
-import { ListContext, ListItem } from '../types'
+import { ListArgument, ListContext, ListItem } from '../types'
 import LocationList from './location'
 
 export function convertToLabel(item: DiagnosticItem, cwd: string, includeCode: boolean, pathFormat: PathFormatting = 'full'): string[] {
@@ -20,6 +22,11 @@ export default class DiagnosticsList extends LocationList {
   public readonly defaultAction = 'open'
   public readonly description = 'diagnostics of current workspace'
   public name = 'diagnostics'
+  public options: ListArgument[] = [{
+    name: '--buffer',
+    hasValue: true,
+    description: 'list diagnostics of current buffer only',
+  }]
   public constructor(manager: ListManager) {
     super()
     diagnosticManager.onDidRefresh(async () => {
@@ -30,7 +37,11 @@ export default class DiagnosticsList extends LocationList {
 
   public async loadItems(context: ListContext): Promise<ListItem[]> {
     let list = await diagnosticManager.getDiagnosticList()
-    let { cwd } = context
+    let { cwd, args } = context
+    if (args.includes('--buffer')) {
+      const doc = await workspace.document
+      list = list.filter(item => item.file === URI.parse(doc.uri).fsPath)
+    }
     const config = this.getConfig()
     const includeCode = config.get<boolean>('includeCode', true)
     const pathFormat = config.get<PathFormatting>('pathFormat', "full")