一、引言
上篇文章已經在Notepad++的插件開發中集成了選中即問AI
的功能,這一篇文章將在此基礎上進一步集成,支持AI對話窗口
以及常見的代碼功能菜單:
- 顯示AI的Dock窗口,可以用自然語言向 AI 提問或要求執行任務
- 選中代碼后使用,AI 會詳細解釋代碼功能
- 需要 AI 幫助改進或修復代碼時使用
- 自動生成代碼注釋
- 選中即問,直接把選中內容丟給AI
- 參數設置,基于插件配置切換AI平臺等參數,提供對話框設置及調整平臺參數
本篇的主要難點在于如何集成Dock窗口,是自己裸寫Window窗口,還是找第三方庫,或者從Notepad++的源碼中剝離相關代碼?
最終選擇的是從Notepad++的源碼中剝離相關代碼的方案,但是直接剝離很難,發現依賴越來越多,因此需要在剝離的基礎上做一些改造,刪減非必須的功能,比如NppDarkMode
裁剪。
注
:項目已開源、鏡像,歡迎使用及指正
二、Notepad++的源碼中剝離Dock窗口
1. 剝離后的Dock文件列表
Common.h
Docking.h
DockingDlgInterface.h
dockingResource.h
dpiManagerV2.cpp
dpiManagerV2.h
NppDarkMode.cpp
NppDarkMode.h
StaticDialog.cpp
StaticDialog.h
Window.h
2. 要點說明
直接從Notepad++中拷貝上述文件代碼到項目中引用的話,會發現還要包含其他文件,然后試著把別的文件引入的時候,后面發現引入的文件越來越多,所以需要對部分文件進行裁剪,主要裁剪的代碼是NppDarkMode.cpp
。 我把NppDarkMode.cpp
的代碼全刪了,然后自己實現了項目中調用了的函數,采集后的NppDarkMode.cpp
文件內容如下:
#include "NppDarkMode.h"enum class SystemVersion
{Unknown,Windows10,Windows11
};SystemVersion GetWindowsVersion()
{// 使用RtlGetVersion替代已廢棄的GetVersionExtypedef NTSTATUS(WINAPI* RtlGetVersionPtr)(PRTL_OSVERSIONINFOW);OSVERSIONINFOW osInfo = { 0 };HMODULE hMod = GetModuleHandleW(L"ntdll.dll");if (hMod){auto RtlGetVersion = reinterpret_cast<RtlGetVersionPtr>(GetProcAddress(hMod, "RtlGetVersion"));if (RtlGetVersion) {osInfo.dwOSVersionInfoSize = sizeof(osInfo);if (RtlGetVersion(&osInfo) == 0){ // STATUS_SUCCESS// Windows 11的版本號為10.0.22000+if (osInfo.dwMajorVersion == 10 &&osInfo.dwMinorVersion == 0){if (osInfo.dwBuildNumber >= 22000){return SystemVersion::Windows11;}else if (osInfo.dwBuildNumber >= 10240){return SystemVersion::Windows10;}}}}}return SystemVersion::Unknown;
}bool NppDarkMode::isWindows10() { return GetWindowsVersion() == SystemVersion::Windows10; }
bool NppDarkMode::isWindows11() { return GetWindowsVersion() == SystemVersion::Windows11; }
void NppDarkMode::setDarkTitleBar(HWND hwnd) {}
3. AI窗口實現
現在只需要引入DockingDlgInterface.h
文件,實現一個基于DockingDlgInterface
的窗口即可,其中AiAssistWnd.h
:
// AiAssistWnd.h
#pragma once
#include "DockingDlgInterface.h"
#include "PluginInterface.h"class AiAssistWnd : public DockingDlgInterface {
public:AiAssistWnd(HINSTANCE hInst, const NppData& nppData);~AiAssistWnd();// 必需實現的虛函數virtual void init();virtual INT_PTR run_dlgProc(UINT message, WPARAM wParam, LPARAM lParam);// 功能接口void updateModelList(const std::vector<std::wstring>& models);void appendAnswer(const std::wstring& answer);void clearConversation();private:void initControls();void layoutControls(int width, int height);void handleUserInput();// NppHINSTANCE _hInst;NppData _nppData;// 控件句柄HWND _hModelCombo = nullptr;HWND _hInputEdit = nullptr;HWND _hAnswerView = nullptr;// 配置參數const int CONTROL_MARGIN = 5;const int COMBO_HEIGHT = 25;const int INPUT_HEIGHT = 80;// 字體資源HFONT _hFont = nullptr;HFONT _hBoldFont = nullptr;// 控件ID定義enum ControlID {IDC_MODEL_COMBO = 2000,IDC_INPUT_EDIT,IDC_ANSWER_VIEW};
};
代碼實現:
// AiAssistWnd.cpp
#include "AiAssistWnd.h"
#include "resource.h"
#include <richedit.h>
#include <commctrl.h>
#include <format>AiAssistWnd::AiAssistWnd(HINSTANCE hInst, const NppData& nppData): DockingDlgInterface(IDD_DIALOG_AI_ASSIST), _hInst(hInst), _nppData(nppData)
{this->_hParent = _nppData._nppHandle;
}AiAssistWnd::~AiAssistWnd()
{if (_hFont) DeleteObject(_hFont);if (_hBoldFont) DeleteObject(_hBoldFont);
}void AiAssistWnd::init()
{DockingDlgInterface::init(_hInst, _hParent);// 注冊Dock窗口tTbData tbData = { 0 };DockingDlgInterface::create(&tbData);tbData.uMask = DWS_DF_CONT_RIGHT | DWS_ICONTAB;;tbData.pszModuleName = L"AI Assistant";;tbData.dlgID = _dlgID;::SendMessage(_nppData._nppHandle, NPPM_DMMREGASDCKDLG, 0, (LPARAM)&tbData);// 初始化UI控件initControls();// 設置初始大小RECT rc;GetClientRect(_hSelf, &rc);::SetWindowPos(_hSelf, nullptr, rc.left, rc.top, 300, rc.bottom, SWP_NOZORDER | SWP_NOMOVE);layoutControls(rc.right, rc.bottom);
}void AiAssistWnd::initControls()
{// 創建控件_hModelCombo = ::CreateWindowExW(0, WC_COMBOBOX, L"",CBS_DROPDOWNLIST | WS_CHILD | WS_VISIBLE | WS_TABSTOP,0, 0, 0, 0, _hSelf, (HMENU)IDC_MODEL_COMBO, _hInst, nullptr);_hInputEdit = ::CreateWindowExW(WS_EX_CLIENTEDGE, L"EDIT", L"",WS_CHILD | WS_VISIBLE | ES_MULTILINE | ES_AUTOVSCROLL | WS_VSCROLL,0, 0, 0, 0, _hSelf, (HMENU)IDC_INPUT_EDIT, _hInst, nullptr);// 使用RichEdit 4.1LoadLibraryW(L"Msftedit.dll");_hAnswerView = ::CreateWindowExW(WS_EX_CLIENTEDGE, MSFTEDIT_CLASS, L"",WS_CHILD | WS_VISIBLE | ES_MULTILINE | ES_READONLY | WS_VSCROLL | WS_HSCROLL,0, 0, 0, 0, _hSelf, (HMENU)IDC_ANSWER_VIEW, _hInst, nullptr);// 初始化字體_hFont = CreateFontW(14, 0, 0, 0, FW_NORMAL, FALSE, FALSE, FALSE,DEFAULT_CHARSET, OUT_DEFAULT_PRECIS, CLIP_DEFAULT_PRECIS,DEFAULT_QUALITY, DEFAULT_PITCH, L"Segoe UI");_hBoldFont = CreateFontW(14, 0, 0, 0, FW_SEMIBOLD, FALSE, FALSE, FALSE,DEFAULT_CHARSET, OUT_DEFAULT_PRECIS, CLIP_DEFAULT_PRECIS,DEFAULT_QUALITY, DEFAULT_PITCH, L"Segoe UI");// 應用字體SendMessageW(_hModelCombo, WM_SETFONT, (WPARAM)_hFont, TRUE);SendMessageW(_hInputEdit, WM_SETFONT, (WPARAM)_hFont, TRUE);SendMessageW(_hAnswerView, WM_SETFONT, (WPARAM)_hFont, TRUE);
}INT_PTR AiAssistWnd::run_dlgProc(UINT message, WPARAM wParam, LPARAM lParam)
{switch (message) {case WM_SIZE:if (wParam != SIZE_MINIMIZED) {RECT rc;GetClientRect(_hSelf, &rc);layoutControls(rc.right, rc.bottom);}return TRUE;case WM_COMMAND:if (HIWORD(wParam) == EN_MAXTEXT && LOWORD(wParam) == IDC_INPUT_EDIT) {handleUserInput();return TRUE;}break;case WM_NOTIFY:// 處理其他通知消息break;case WM_CTLCOLOREDIT:// 設置輸入框背景色if ((HWND)lParam == _hInputEdit) {SetBkColor((HDC)wParam, RGB(255, 255, 255));return (INT_PTR)GetStockObject(WHITE_BRUSH);}break;}return DockingDlgInterface::run_dlgProc(message, wParam, lParam);
}void AiAssistWnd::layoutControls(int width, int height)
{int yPos = CONTROL_MARGIN;int nHeight = CONTROL_MARGIN;// 回答區域int answerHeight = height - 3*CONTROL_MARGIN - COMBO_HEIGHT - INPUT_HEIGHT;::MoveWindow(_hAnswerView,CONTROL_MARGIN, yPos,width - 2 * CONTROL_MARGIN, answerHeight, TRUE);yPos += CONTROL_MARGIN + answerHeight;// 模型選擇框::MoveWindow(_hModelCombo,CONTROL_MARGIN, yPos,width - 2 * CONTROL_MARGIN, COMBO_HEIGHT, TRUE);yPos += COMBO_HEIGHT + CONTROL_MARGIN;// 輸入框::MoveWindow(_hInputEdit,CONTROL_MARGIN, yPos,width - 2 * CONTROL_MARGIN, INPUT_HEIGHT, TRUE);yPos += INPUT_HEIGHT + CONTROL_MARGIN;
}void AiAssistWnd::updateModelList(const std::vector<std::wstring>& models)
{SendMessageW(_hModelCombo, CB_RESETCONTENT, 0, 0);for (const auto& model : models) {SendMessageW(_hModelCombo, CB_ADDSTRING, 0, (LPARAM)model.c_str());}if (!models.empty()) {SendMessageW(_hModelCombo, CB_SETCURSEL, 0, 0);}
}void AiAssistWnd::appendAnswer(const std::wstring& answer)
{// 添加時間戳SYSTEMTIME st;GetLocalTime(&st);std::wstring timestamp = std::format(L"[{:02}:{:02}:{:02}] ",st.wHour, st.wMinute, st.wSecond);// 設置富文本格式CHARFORMAT2W cf = { sizeof(CHARFORMAT2W) };cf.dwMask = CFM_COLOR | CFM_BOLD;cf.crTextColor = RGB(0, 128, 0);cf.dwEffects = CFE_BOLD;SendMessageW(_hAnswerView, EM_SETCHARFORMAT, SCF_SELECTION, (LPARAM)&cf);SendMessageW(_hAnswerView, EM_REPLACESEL, FALSE, (LPARAM)timestamp.c_str());// 恢復默認格式cf.dwMask = CFM_COLOR;cf.crTextColor = RGB(0, 0, 0);cf.dwEffects = 0;SendMessageW(_hAnswerView, EM_SETCHARFORMAT, SCF_SELECTION, (LPARAM)&cf);SendMessageW(_hAnswerView, EM_REPLACESEL, FALSE, (LPARAM)answer.c_str());// 自動滾動到底部SendMessageW(_hAnswerView, WM_VSCROLL, SB_BOTTOM, 0);
}void AiAssistWnd::clearConversation()
{if (!IsWindow(_hAnswerView))return;// 使用SETTEXTEX結構清空內容SETTEXTEX st = {ST_DEFAULT, // 標志位1200 // 使用UTF-16編碼};// 方法1:直接設置空文本(保留格式)SendMessageW(_hAnswerView, EM_SETTEXTEX, (WPARAM)&st, (LPARAM)L"");// 方法2:通過選擇全部刪除(更徹底)// SendMessageW(_hAnswerView, EM_SETSEL, 0, -1); // 全選// SendMessageW(_hAnswerView, EM_REPLACESEL, 0, (LPARAM)L""); // 替換為空// 可選:重置滾動條位置SendMessageW(_hAnswerView, WM_VSCROLL, SB_TOP, 0);// 可選:清除Undo緩沖區SendMessageW(_hAnswerView, EM_EMPTYUNDOBUFFER, 0, 0);
}
void AiAssistWnd::handleUserInput()
{// 獲取輸入文本int len = GetWindowTextLengthW(_hInputEdit) + 1;std::wstring input(len, L'\0');GetWindowTextW(_hInputEdit, &input[0], len);input.resize(len - 1); // 移除結尾的null字符if (!input.empty()) {// TODO: 觸發AI處理邏輯appendAnswer(L"Received: " + input);// 清空輸入框SetWindowTextW(_hInputEdit, L"");}
}
三、主要菜單功能實現
1.定義菜單
// PluginDefinition.h
//-----------------------------------------------//
//-- STEP 2. DEFINE YOUR PLUGIN COMMAND NUMBER --//
//-----------------------------------------------//
//
// Here define the number of your plugin commands
//
const int nbFunc = 6;//
// Your plugin command functions
//
//
// 參數設置
void PluginConfig();
// 打開Ai助手窗口
void OpenAiAssistWnd();
// 解讀代碼
void ReadCode();
// 代碼優化
void OptimizeCode();
// 添加代碼注釋
void AddCodeComment();
// 選中即問
void AskBySelectedText();
2.初始化菜單
// PluginDefinition.cpp
//
// Initialization of your plugin commands
// You should fill your plugins commands here
void commandMenuInit()
{//--------------------------------------------////-- STEP 3. CUSTOMIZE YOUR PLUGIN COMMANDS --////--------------------------------------------//// with function :// setCommand(int index, // zero based number to indicate the order of command// TCHAR *commandName, // the command name that you want to see in plugin menu// PFUNCPLUGINCMD functionPointer, // the symbol of function (function pointer) associated with this command. The body should be defined below. See Step 4.// ShortcutKey *shortcut, // optional. Define a shortcut to trigger this command// bool check0nInit // optional. Make this menu item be checked visually// );// 初始化數據g_pNppImp = new NppImp(g_nppData);// 初始化菜單ShortcutKey* pSck = new ShortcutKey[nbFunc];g_pShortcutKeys = pSck;size_t nCid = 0;setCommand(nCid, L"參數配置", PluginConfig, NULL, false); ++nCid;pSck[nCid] = { false, true, false, 'K' };setCommand(nCid, L"顯示窗口", OpenAiAssistWnd, pSck + nCid, false); ++nCid;pSck[nCid] = { false, true, false, 'J' };setCommand(nCid, L"解讀代碼", ReadCode, pSck + nCid, false); ++nCid;pSck[nCid] = { false, true, false, 'Y' };setCommand(nCid, L"優化代碼", OptimizeCode, pSck + nCid, false); ++nCid;pSck[nCid] = { false, true, false, 'Z' };setCommand(nCid, L"代碼注釋", AddCodeComment, pSck + nCid, false); ++nCid;pSck[nCid] = { false, true, false, 'A' };setCommand(nCid, L"選中即問", AskBySelectedText, pSck + nCid, false); ++nCid;
}
3.打開AI窗口
// PluginDefinition.cpp
// 打開Ai助手窗口
void OpenAiAssistWnd()
{// 初始化Dock窗口if (g_pAiWnd == nullptr && g_hModule != nullptr && g_nppData._nppHandle != nullptr){g_pAiWnd = new AiAssistWnd((HINSTANCE)g_hModule, g_nppData);g_pAiWnd->init();}if (g_pAiWnd){g_pAiWnd->display(true);}
}
四、效果展示
1.插件菜單
2.AI窗口
五、結語
到這里,已經在Notepad++中支持AI對話窗口了,已經有一點Cursor
的AI編輯器的意思了,下一步將進一步完善功能實現,包括配置加載、配置窗口以及界面功能實現。