插件接口更新，增加ITMPlugin::OnInitialize函数，新增ITrafficMonitor接口

Author: zhongyang219

TrafficMonitor/TrafficMonitor.cpp

@@ -1316,7 +1316,7 @@ bool CTrafficMonitorApp::DPIFromRect(const RECT& rect, UINT* out_dpi_x, UINT* ou
     return hr == S_OK;
 }
 
-COLORREF CTrafficMonitorApp::GetThemeColor() const
+unsigned int CTrafficMonitorApp::GetThemeColor() const
 {
     return m_theme_color;
 }
@@ -1386,3 +1386,64 @@ int CTrafficMonitorApp::ExitInstance()
 
     return CWinApp::ExitInstance();
 }
+
+double CTrafficMonitorApp::GetMonitorData(MonitorItem item)
+{
+    switch (item)
+    {
+    case MI_UP: return m_out_speed;
+    case MI_DOWN: return m_in_speed;
+    case MI_CPU: return m_cpu_usage;
+    case MI_MEMORY: return m_memory_usage;
+    case MI_GPU_USAGE: return m_gpu_usage;
+    case MI_CPU_TEMP: return m_cpu_temperature;
+    case MI_GPU_TEMP: return m_gpu_temperature;
+    case MI_HDD_TEMP: return m_hdd_temperature;
+    case MI_MAIN_BOARD_TEMP: return m_main_board_temperature;
+    case MI_HDD_USAGE: return m_hdd_usage;
+    case MI_CPU_FREQ: return m_cpu_freq;
+    case MI_TODAY_UP_TRAFFIC: return m_today_up_traffic;
+    case MI_TODAY_DOWN_TRAFFIC: return m_today_down_traffic;
+    }
+    return 0.0;
+}
+
+void CTrafficMonitorApp::ShowNotifyMessage(const wchar_t* strMsg)
+{
+    CTrafficMonitorDlg* pMainWnd = dynamic_cast<CTrafficMonitorDlg*>(m_pMainWnd);
+    if (pMainWnd != nullptr)
+    {
+        pMainWnd->ShowNotifyTip(CCommon::LoadText(IDS_TRAFFICMONITOR_PLUGIN_NITIFICATION), strMsg);
+    }
+}
+
+unsigned short CTrafficMonitorApp::GetLanguageId() const
+{
+    return m_general_data.language;
+}
+
+const wchar_t* CTrafficMonitorApp::GetConfigDir() const
+{
+    static std::wstring config_dir;
+    config_dir = m_config_dir;
+    config_dir += L"plugins\\";
+    return config_dir.c_str();
+}
+
+int CTrafficMonitorApp::GetDPI(DPIType type) const
+{
+    CTrafficMonitorDlg* pMainWnd = dynamic_cast<CTrafficMonitorDlg*>(m_pMainWnd);
+    switch (type)
+    {
+    case DPI_MAIN_WND:
+        return m_dpi;
+    case DPI_TASKBAR:
+        if (pMainWnd != nullptr)
+        {
+            if (pMainWnd->IsTaskbarWndValid())
+                return pMainWnd->GetTaskbarWindow()->GetDPI();
+        }
+        return m_dpi;
+    }
+    return 0;
+}

TrafficMonitor/TrafficMonitor.h

@@ -21,13 +21,14 @@
 #include "DllFunctions.h"
 #include "StrTable.h"
 #include "PluginUpdateHelper.h"
+#include "PluginInterface.h"
 
 // CTrafficMonitorApp:
 // 有关此类的实现，请参阅 TrafficMonitor.cpp
 //
 
 
-class CTrafficMonitorApp : public CWinApp
+class CTrafficMonitorApp : public CWinApp, public ITrafficMonitor
 {
 public:
     //各种路径
@@ -168,7 +169,7 @@ class CTrafficMonitorApp : public CWinApp
 
     bool DPIFromRect(const RECT& rect, UINT* out_dpi_x, UINT* out_dpi_y);
 
-    COLORREF GetThemeColor() const;
+    virtual unsigned int GetThemeColor() const override;
     void SetThemeColor(COLORREF color);
 
 private:
@@ -197,6 +198,13 @@ class CTrafficMonitorApp : public CWinApp
     afx_msg void OnFrequentyAskedQuestions();
     afx_msg void OnUpdateLog();
     virtual int ExitInstance();
+
+    // 通过 ITrafficMonitor 继承
+    double GetMonitorData(MonitorItem item) override;
+    void ShowNotifyMessage(const wchar_t* strMsg) override;
+    unsigned short GetLanguageId() const override;
+    const wchar_t* GetConfigDir() const override;
+    int GetDPI(DPIType type) const override;
 };
 
 extern CTrafficMonitorApp theApp;

TrafficMonitor/TrafficMonitorDlg.h

@@ -156,7 +156,9 @@ class CTrafficMonitorDlg : public CDialog
 
     void AddNotifyIcon();       //添加通知区图标
     void DeleteNotifyIcon();
+public:
     void ShowNotifyTip(const wchar_t* title, const wchar_t* message);       //显示通知区提示
+protected:
     void UpdateNotifyIconTip();     //更新通知区图标的鼠标提示
 
     void SaveHistoryTraffic();
@@ -173,7 +175,9 @@ class CTrafficMonitorDlg : public CDialog
     void LoadBackGroundImage();
     void SetTextFont();
 
+public:
     bool IsTaskbarWndValid() const;
+protected:
 
     void TaskbarShowHideItem(DisplayItem type);
 

TrafficMonitor/language.h

@@ -216,6 +216,7 @@
 #define IDS_SECONDARY_DISPLAY L"IDS_SECONDARY_DISPLAY"
 #define IDS_RESTORE_FROM_SLEEP_LOG L"IDS_RESTORE_FROM_SLEEP_LOG"
 #define IDS_PLUGIN_NEW_VERSION_INFO L"IDS_PLUGIN_NEW_VERSION_INFO"
+#define IDS_TRAFFICMONITOR_PLUGIN_NITIFICATION L"IDS_TRAFFICMONITOR_PLUGIN_NITIFICATION"
 
 #define TXT_OK L"TXT_OK"
 #define TXT_CANCEL L"TXT_CANCEL"

TrafficMonitor/language/English.ini

@@ -220,6 +220,7 @@ IDS_PRIMARY_DISPLAY = "Primary display"
 IDS_SECONDARY_DISPLAY = "Secondary display <%1%>"
 IDS_RESTORE_FROM_SLEEP_LOG = "The system has been restored from hibernation, the connection has been reinitialized. (<%1%> times has been reinitialized.)"
 IDS_PLUGIN_NEW_VERSION_INFO = "Update available, latest version: <%1%>"
+IDS_TRAFFICMONITOR_PLUGIN_NITIFICATION = "TrafficMonitor Plugin Notification"
 
 TXT_OK = "OK"
 TXT_CANCEL = "Cancel"

TrafficMonitor/language/Simplified_Chinese.ini

@@ -220,6 +220,7 @@ IDS_PRIMARY_DISPLAY = "主显示器"
 IDS_SECONDARY_DISPLAY = "副显示器 <%1%>"
 IDS_RESTORE_FROM_SLEEP_LOG = "系统已从休眠状态恢复，已重新初始化连接。（已重新初始化<%1%>次）"
 IDS_PLUGIN_NEW_VERSION_INFO = "有更新，最新版本：<%1%>"
+IDS_TRAFFICMONITOR_PLUGIN_NITIFICATION = "TrafficMonitor 插件通知"
 
 TXT_OK = "确定"
 TXT_CANCEL = "取消"

TrafficMonitor/language/Traditional_Chinese.ini

@@ -220,6 +220,7 @@ IDS_PRIMARY_DISPLAY = "主顯示器"
 IDS_SECONDARY_DISPLAY = "副顯示器 <%1%>"
 IDS_RESTORE_FROM_SLEEP_LOG = "系統已從休眠狀態恢復，已重新初始化連線。（已重新初始化<%1%>次）"
 IDS_PLUGIN_NEW_VERSION_INFO = "有更新，最新版本：<%1%>"
+IDS_TRAFFICMONITOR_PLUGIN_NITIFICATION = "TrafficMonitor 外掛通知"
 
 TXT_OK = "確定"
 TXT_CANCEL = "取消"

include/PluginInterface.h

@@ -151,7 +151,9 @@ class IPluginItem
     virtual float GetResourceUsageGraphValue() const { return 0.0; }
 };
 
+class ITrafficMonitor;
 
+///////////////////////////////////////////////////////////////////////////////////////////////////////
 //插件接口
 class ITMPlugin
 {
@@ -161,7 +163,7 @@ class ITMPlugin
      * @attention 插件开发者不应该修改这里的返回值，也不应该重写此虚函数。
      * @return  int
      */
-    virtual int GetAPIVersion() const { return 6; }
+    virtual int GetAPIVersion() const { return 7; }
 
     /**
      * @brief   获取插件显示项目的对象
@@ -313,8 +315,85 @@ class ITMPlugin
     */
     virtual int IsCommandChecked(int command_index) { return false; }
 
+    /**
+     * @brief   插件初始化
+     * @detail  当插件被加载时被调用，传递ITrafficMonitor接口的指针。插件可以保存此指针以调用ITrafficMonitor接口中的函数
+     * @param   pApp
+     */
+    virtual void OnInitialize(ITrafficMonitor* pApp) {}
+};
+
+
+///////////////////////////////////////////////////////////////////////////////////////////////////////
+//主程序接口
+class ITrafficMonitor
+{
+public:
+    /** 主程序的所有监控信息 */
+    enum MonitorItem
+    {
+        MI_UP,                  /**< 上传速度 */
+        MI_DOWN,                /**< 下载速度 */
+        MI_CPU,                 /**< CPU利用率 */
+        MI_MEMORY,              /**< 内存利用率 */
+        MI_GPU_USAGE,           /**< 显卡利用率 */
+        MI_CPU_TEMP,            /**< CPU温度 */
+        MI_GPU_TEMP,            /**< 显卡温度 */
+        MI_HDD_TEMP,            /**< 硬盘温度 */
+        MI_MAIN_BOARD_TEMP,     /**< 主板温度 */
+        MI_HDD_USAGE,           /**< 硬盘利用率 */
+        MI_CPU_FREQ,            /**< CPU频率 */
+        MI_TODAY_UP_TRAFFIC,    /**< 今日上传流量 */
+        MI_TODAY_DOWN_TRAFFIC   /**< 今日下载流量 */
+    };
+
+    /**
+     * @brief   获取一个主程序的监控信息
+     *          （ITMPlugin::OnMonitorInfo将被弃用）
+     * @param   item 要获取监控信息的项目
+     * @return  获取到监控信息
+     */
+    virtual double GetMonitorData(MonitorItem item) = 0;
+
+    /**
+     * @brief   显示一个通知消息
+     * @param   strMsg 要显示的通知消息
+     */
+    virtual void ShowNotifyMessage(const wchar_t* strMsg) = 0;
+
+    /**
+     * @brief   获取当前语言id
+     * @return  当前主程序的语言id
+     */
+    virtual unsigned short GetLanguageId() const = 0;
+
+    /**
+     * @brief   获取配置文件目录
+     * @return  配置文件目录
+     */
+    virtual const wchar_t* GetConfigDir() const = 0;
+
+    /** 主程序DPI类型 */
+    enum DPIType
+    {
+        DPI_MAIN_WND,   /**< 主窗口的DPI */
+        DPI_TASKBAR     /**< 任务栏窗口的DPI */
+    };
+
+    /**
+     * @brief   获取主程序DPI
+     * @return  主程序DPI
+     */
+    virtual int GetDPI(DPIType type) const = 0;
+
+    /**
+     * @brief   获取当前系统主题颜色
+     * @return  COLORREF格式的颜色值
+     */
+    virtual unsigned int GetThemeColor() const = 0;
 };
 
+
 /*
 * 注意：插件dll需导出以下函数
 * ITMPlugin* TMPluginGetInstance();
@@ -340,4 +419,6 @@ class ITMPlugin
 * -------------------------------------------------------------------------
 *     6       | 新增 IPluginItem::GetResourceUsageGraphType IPluginItem::GetResourceUsageGraphValue 函数
 * -------------------------------------------------------------------------
+*     7       | 新增 ITMPlugin::OnInitialize 函数
+* -------------------------------------------------------------------------
 */