Object 146

lua/fittencode/chat/controller.lua

@@ -17,13 +17,16 @@ function Controller:new(opts)
         basic_chat_template_id = opts.basic_chat_template_id,
         conversation_types_provider = opts.conversation_types_provider,
         observers = {},
+        augroups = {},
     }, Controller)
     return obj
 end
 
-function Controller:init_singleton()
-    self.selection_changed_autocmd = vim.api.nvim_create_autocmd('User', {
-        pattern = 'fittencode.SelectionChanged',
+function Controller:init_integration()
+    self.augroups.event = vim.api.nvim_create_augroup('Fittencode.Chat.Controller.Event', { clear = true })
+    vim.api.nvim_create_autocmd('User', {
+        pattern = 'Fittencode.SelectionChanged',
+        group = self.augroups.event,
         once = false,
         callback = function(args)
             self:update_view()
@@ -33,6 +36,12 @@ function Controller:init_singleton()
     self:register_observer(self.status)
 end
 
+function Controller:destory()
+    for _, id in pairs(self.augroups) do
+        vim.api.nvim_del_augroup(id)
+    end
+end
+
 function Controller:register_observer(observer)
     table.insert(self.observers, observer)
 end

lua/fittencode/chat/init.lua

@@ -32,7 +32,7 @@ local function setup()
         conversation_types_provider = conversation_types_provider,
         basic_chat_template_id = basic_chat_template_id
     })
-    controller:init_singleton()
+    controller:init_integration()
     view:register_message_receiver(function(message)
         controller:receive_view_message(message)
     end)

lua/fittencode/inline/controller.lua

@@ -34,7 +34,7 @@ function Controller:new(opts)
     return obj
 end
 
-function Controller:init_singleton()
+function Controller:init_integration()
     self.status = Status:new()
     self:register_observer(self.status)
     self.generate_one_stage = Fn.debounce(Client.generate_one_stage, Config.delay_completion.delaytime)
@@ -45,6 +45,14 @@ function Controller:init_singleton()
     self:enable(Config.inline_completion.enable)
 end
 
+function Controller:destory()
+    self:enable(false)
+    for _, id in pairs(self.augroups) do
+        vim.api.nvim_del_augroup(id)
+    end
+    self.augroups = {}
+end
+
 function Controller:register_observer(observer)
     table.insert(self.observers, observer)
 end

lua/fittencode/inline/init.lua

@@ -9,7 +9,7 @@ local controller = nil
 
 local function setup()
     controller = Controller:new()
-    controller:init_singleton()
+    controller:init_integration()
 end
 
 local function get_status()

lua/fittencode/types.lua

@@ -197,6 +197,7 @@
 ---@class FittenCode.Chat.Controller
 ---@field view FittenCode.Chat.View
 ---@field model FittenCode.Chat.Model
+---@field augroups table
 ---@field basic_chat_template_id string
 ---@field conversation_types_provider FittenCode.Chat.ConversationTypeProvider
 ---@field status fittencode.Chat.Status