fix: High CPU usage in new files check (#446) (#474)

* fix: High CPU usage in new files check (#446)

* Resolve absolute path

Author: reneleonhardt

src/main/java/ee/carlrobert/codegpt/settings/configuration/ConfigurationState.java

@@ -132,13 +132,13 @@ public boolean equals(Object o) {
     if (this == o) {
       return true;
     }
-    if (o == null || getClass() != o.getClass()) {
+    if (!(o instanceof ConfigurationState that)) {
       return false;
     }
-    ConfigurationState that = (ConfigurationState) o;
     return maxTokens == that.maxTokens
-        && Double.compare(that.temperature, temperature) == 0
+        && Double.compare(temperature, that.temperature) == 0
         && checkForPluginUpdates == that.checkForPluginUpdates
+        && checkForNewScreenshots == that.checkForNewScreenshots
         && createNewChatOnEachAction == that.createNewChatOnEachAction
         && ignoreGitCommitTokenLimit == that.ignoreGitCommitTokenLimit
         && methodNameGenerationEnabled == that.methodNameGenerationEnabled
@@ -153,8 +153,9 @@ public boolean equals(Object o) {
   @Override
   public int hashCode() {
     return Objects.hash(systemPrompt, commitMessagePrompt, maxTokens, temperature,
-        checkForPluginUpdates, createNewChatOnEachAction, ignoreGitCommitTokenLimit,
-        methodNameGenerationEnabled, captureCompileErrors, autoFormattingEnabled,
+        checkForPluginUpdates, checkForNewScreenshots, createNewChatOnEachAction,
+        ignoreGitCommitTokenLimit, methodNameGenerationEnabled, captureCompileErrors,
+        autoFormattingEnabled,
         codeCompletionsEnabled, tableData);
   }
 }

src/main/kotlin/ee/carlrobert/codegpt/CodeGPTProjectActivity.kt

@@ -19,10 +19,14 @@ import ee.carlrobert.codegpt.settings.configuration.ConfigurationSettings
 import ee.carlrobert.codegpt.settings.service.you.YouSettings
 import ee.carlrobert.codegpt.toolwindow.chat.ui.textarea.AttachImageNotifier
 import ee.carlrobert.codegpt.ui.OverlayUtil
+import io.ktor.util.*
 import java.nio.file.Paths
+import kotlin.io.path.absolutePathString
 
 class CodeGPTProjectActivity : StartupActivity.Background {
 
+    private val watchExtensions = listOf("jpg", "jpeg", "png")
+
     override fun runActivity(project: Project) {
         EditorActionsUtil.refreshActions()
         CredentialsStore.loadAll()
@@ -34,10 +38,14 @@ class CodeGPTProjectActivity : StartupActivity.Background {
         if (!ApplicationManager.getApplication().isUnitTestMode
             && ConfigurationSettings.getCurrentState().isCheckForNewScreenshots
         ) {
+            val desktopPath = Paths.get(System.getProperty("user.home"), "Desktop")
             project.service<FileWatcher>()
-                .watch(Paths.get(System.getProperty("user.home"), "Desktop").toFile()) {
-                    if (listOf("jpg", "jpeg", "png").contains(it.extension)) {
-                        showImageAttachmentNotification(project, it.absolutePath)
+                .watch(desktopPath) {
+                    if (watchExtensions.contains(it.extension.lowercase())) {
+                        showImageAttachmentNotification(
+                            project,
+                            desktopPath.resolve(it).absolutePathString()
+                        )
                     }
                 }
         }
@@ -95,4 +103,4 @@ class CodeGPTProjectActivity : StartupActivity.Background {
             })
             .notify(project)
     }
-}
+}

src/main/kotlin/ee/carlrobert/codegpt/FileWatcher.kt

@@ -2,28 +2,33 @@ package ee.carlrobert.codegpt
 
 import com.intellij.openapi.Disposable
 import com.intellij.openapi.components.Service
-import org.apache.commons.io.monitor.FileAlterationListenerAdaptor
-import org.apache.commons.io.monitor.FileAlterationMonitor
-import org.apache.commons.io.monitor.FileAlterationObserver
-import java.io.File
+import java.nio.file.FileSystems
+import java.nio.file.Path
+import java.nio.file.StandardWatchEventKinds.ENTRY_CREATE
+import java.nio.file.WatchKey
+import kotlin.concurrent.thread
+
 
 @Service(Service.Level.PROJECT)
 class FileWatcher : Disposable {
 
-    private var fileMonitor: FileAlterationMonitor? = null
+    private var fileMonitor: Thread? = null
 
-    fun watch(pathToWatch: File, onFileCreated: (File) -> Unit) {
-        val observer = FileAlterationObserver(pathToWatch)
-        observer.addListener(object : FileAlterationListenerAdaptor() {
-            override fun onFileCreate(file: File) {
-                onFileCreated(file)
+    fun watch(pathToWatch: Path, onFileCreated: (Path) -> Unit) {
+        val watchService = FileSystems.getDefault().newWatchService()
+        pathToWatch.register(watchService, ENTRY_CREATE) // watch for new files
+        fileMonitor = thread {
+            var key: WatchKey
+            while ((watchService.take().also { key = it }) != null) {
+                for (event in key.pollEvents()) {
+                    onFileCreated(event.context() as Path)
+                }
+                key.reset()
             }
-        })
-        fileMonitor = FileAlterationMonitor(500, observer)
-        fileMonitor?.start()
+        }
     }
 
     override fun dispose() {
-        fileMonitor?.stop()
+        fileMonitor?.interrupt()
     }
-}
+}