document change

Author: leigaol

packages/amazonq/src/app/inline/completion.ts

@@ -2,7 +2,7 @@
  * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
  * SPDX-License-Identifier: Apache-2.0
  */
-import * as vscode from 'vscode'
+
 import {
     CancellationToken,
     InlineCompletionContext,
@@ -46,11 +46,7 @@ import { Experiments, getLogger, sleep } from 'aws-core-vscode/shared'
 import { debounce, messageUtils } from 'aws-core-vscode/utils'
 import { showEdits } from './EditRendering/imageRenderer'
 import { ICursorUpdateRecorder } from './cursorUpdateManager'
-
-let lastDocumentDeleteEvent: vscode.TextDocumentChangeEvent | undefined = undefined
-let lastDocumentDeleteTime = 0
-
-let lastDocumentChangeEventMap: Map<string, vscode.TextDocumentChangeEvent> = new Map()
+import { DocumentEventListener } from './documentEventListener'
 
 export class InlineCompletionManager implements Disposable {
     private disposable: Disposable
@@ -62,7 +58,7 @@ export class InlineCompletionManager implements Disposable {
 
     private inlineTutorialAnnotation: InlineTutorialAnnotation
     private readonly logSessionResultMessageName = 'aws/logInlineCompletionSessionResults'
-    private documentChangeListener: Disposable
+    private documentEventListener: DocumentEventListener
 
     constructor(
         languageClient: LanguageClient,
@@ -76,27 +72,19 @@ export class InlineCompletionManager implements Disposable {
         this.lineTracker = lineTracker
         this.recommendationService = new RecommendationService(this.sessionManager, cursorUpdateRecorder)
         this.inlineTutorialAnnotation = inlineTutorialAnnotation
+        this.documentEventListener = new DocumentEventListener()
         this.inlineCompletionProvider = new AmazonQInlineCompletionItemProvider(
             languageClient,
             this.recommendationService,
             this.sessionManager,
-            this.inlineTutorialAnnotation
+            this.inlineTutorialAnnotation,
+            this.documentEventListener
         )
 
-        this.documentChangeListener = vscode.workspace.onDidChangeTextDocument((e) => {
-            if (e.contentChanges.length === 1 && e.contentChanges[0].text === '') {
-                lastDocumentDeleteEvent = e
-                lastDocumentDeleteTime = performance.now()
-            }
-            if (e.contentChanges.length > 0) {
-                lastDocumentChangeEventMap.set(e.document.uri.fsPath, e)
-            }
-        })
         this.disposable = languages.registerInlineCompletionItemProvider(
             CodeWhispererConstants.platformLanguageIds,
             this.inlineCompletionProvider
         )
-
         this.lineTracker.ready()
     }
 
@@ -109,8 +97,8 @@ export class InlineCompletionManager implements Disposable {
             this.disposable.dispose()
             this.lineTracker.dispose()
         }
-        if (this.documentChangeListener) {
-            this.documentChangeListener.dispose()
+        if (this.documentEventListener) {
+            this.documentEventListener.dispose()
         }
     }
 
@@ -216,7 +204,8 @@ export class AmazonQInlineCompletionItemProvider implements InlineCompletionItem
         private readonly languageClient: LanguageClient,
         private readonly recommendationService: RecommendationService,
         private readonly sessionManager: SessionManager,
-        private readonly inlineTutorialAnnotation: InlineTutorialAnnotation
+        private readonly inlineTutorialAnnotation: InlineTutorialAnnotation,
+        private readonly documentEventListener: DocumentEventListener
     ) {}
 
     private readonly logSessionResultMessageName = 'aws/logInlineCompletionSessionResults'
@@ -256,14 +245,11 @@ export class AmazonQInlineCompletionItemProvider implements InlineCompletionItem
         await sleep(1)
         // prevent user deletion invoking auto trigger
         // this is a best effort estimate of deletion
-        const timeDiff = Math.abs(performance.now() - lastDocumentDeleteTime)
-        if (timeDiff < 500 && lastDocumentDeleteEvent && lastDocumentDeleteEvent.document.uri === document.uri) {
+        if (this.documentEventListener.isLastEventDeletion(document.uri.fsPath)) {
             getLogger().debug('Skip auto trigger when deleting code')
             return []
         }
 
-        const event = lastDocumentChangeEventMap.get(document.uri.fsPath) || undefined
-        console.log(event)
         let logstr = `GenerateCompletion metadata:\\n`
         try {
             const t0 = performance.now()

packages/amazonq/src/app/inline/documentEventListener.ts

@@ -0,0 +1,50 @@
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import * as vscode from 'vscode'
+
+export interface DocumentChangeEvent {
+    event: vscode.TextDocumentChangeEvent
+    timestamp: number
+}
+
+export class DocumentEventListener {
+    private lastDocumentChangeEventMap: Map<string, DocumentChangeEvent> = new Map()
+    private documentChangeListener: vscode.Disposable
+    private _maxDocument = 1000
+
+    constructor() {
+        this.documentChangeListener = vscode.workspace.onDidChangeTextDocument((e) => {
+            if (e.contentChanges.length > 0) {
+                if (this.lastDocumentChangeEventMap.size > this._maxDocument) {
+                    this.lastDocumentChangeEventMap.clear()
+                }
+                this.lastDocumentChangeEventMap.set(e.document.uri.fsPath, { event: e, timestamp: performance.now() })
+            }
+        })
+    }
+
+    public isLastEventDeletion(filepath: string): boolean {
+        const result = this.lastDocumentChangeEventMap.get(filepath)
+        if (result) {
+            const event = result.event
+            const eventTime = result.timestamp
+            const isDelete =
+                (event && event.contentChanges.length === 1 && event.contentChanges[0].text === '') || false
+            const timeDiff = Math.abs(performance.now() - eventTime)
+            return timeDiff < 500 && isDelete
+        }
+        return false
+    }
+
+    public getLastDocumentChangeEvent(filepath: string): DocumentChangeEvent | undefined {
+        return this.lastDocumentChangeEventMap.get(filepath)
+    }
+
+    public dispose(): void {
+        if (this.documentChangeListener) {
+            this.documentChangeListener.dispose()
+        }
+    }
+}

packages/amazonq/test/unit/amazonq/apps/inline/completion.test.ts

@@ -28,6 +28,7 @@ import {
 } from 'aws-core-vscode/codewhisperer'
 import { LineTracker } from '../../../../../src/app/inline/stateTracker/lineTracker'
 import { InlineTutorialAnnotation } from '../../../../../src/app/inline/tutorials/inlineTutorialAnnotation'
+import { DocumentEventListener } from '../../../../../src/app/inline/documentEventListener'
 
 describe('InlineCompletionManager', () => {
     let manager: InlineCompletionManager
@@ -243,11 +244,13 @@ describe('InlineCompletionManager', () => {
             let getAllRecommendationsStub: sinon.SinonStub
             let recommendationService: RecommendationService
             let inlineTutorialAnnotation: InlineTutorialAnnotation
+            let documentEventListener: DocumentEventListener
 
             beforeEach(() => {
                 const lineTracker = new LineTracker()
                 inlineTutorialAnnotation = new InlineTutorialAnnotation(lineTracker, mockSessionManager)
                 recommendationService = new RecommendationService(mockSessionManager)
+                documentEventListener = new DocumentEventListener()
                 vsCodeState.isRecommendationsActive = false
                 mockSessionManager = {
                     getActiveSession: getActiveSessionStub,
@@ -271,7 +274,8 @@ describe('InlineCompletionManager', () => {
                         languageClient,
                         recommendationService,
                         mockSessionManager,
-                        inlineTutorialAnnotation
+                        inlineTutorialAnnotation,
+                        documentEventListener
                     )
                     const items = await provider.provideInlineCompletionItems(
                         mockDocument,
@@ -287,7 +291,8 @@ describe('InlineCompletionManager', () => {
                         languageClient,
                         recommendationService,
                         mockSessionManager,
-                        inlineTutorialAnnotation
+                        inlineTutorialAnnotation,
+                        documentEventListener
                     )
                     await provider.provideInlineCompletionItems(mockDocument, mockPosition, mockContext, mockToken)
                 }),
@@ -296,7 +301,8 @@ describe('InlineCompletionManager', () => {
                         languageClient,
                         recommendationService,
                         mockSessionManager,
-                        inlineTutorialAnnotation
+                        inlineTutorialAnnotation,
+                        documentEventListener
                     )
                     getActiveRecommendationStub.returns([
                         {
@@ -326,7 +332,8 @@ describe('InlineCompletionManager', () => {
                         languageClient,
                         recommendationService,
                         mockSessionManager,
-                        inlineTutorialAnnotation
+                        inlineTutorialAnnotation,
+                        documentEventListener
                     )
                     const expectedText = `${mockSuggestions[1].insertText}this is my text`
                     getActiveRecommendationStub.returns([
@@ -352,7 +359,8 @@ describe('InlineCompletionManager', () => {
                         languageClient,
                         recommendationService,
                         mockSessionManager,
-                        inlineTutorialAnnotation
+                        inlineTutorialAnnotation,
+                        documentEventListener
                     )
                     getActiveRecommendationStub.returns([])
                     const messageShown = new Promise((resolve) =>
@@ -385,7 +393,8 @@ describe('InlineCompletionManager', () => {
                         languageClient,
                         recommendationService,
                         mockSessionManager,
-                        inlineTutorialAnnotation
+                        inlineTutorialAnnotation,
+                        documentEventListener
                     )
                     const p1 = provider.provideInlineCompletionItems(mockDocument, mockPosition, mockContext, mockToken)
                     const p2 = provider.provideInlineCompletionItems(mockDocument, mockPosition, mockContext, mockToken)