feat(notify):新增只提醒红字或关注个股新闻的设置

- 在 App 中实现只推送红字或关注个股新闻的逻辑
- 在前端添加 enableOnlyPushRedNews 配置项
- 更新后端设置 API 以支持新功能

app.go

@@ -256,7 +256,7 @@ func (a *App) CheckUpdate(flag int) {
 		}
 		go runtime.EventsEmit(a.ctx, "newsPush", map[string]any{
 			"time":    "发现新版本：" + releaseVersion.TagName,
-			"isRed":   false,
+			"isRed":   true,
 			"source":  "go-stock",
 			"content": fmt.Sprintf("%s", commit.Message),
 		})
@@ -299,7 +299,7 @@ func (a *App) CheckUpdate(flag int) {
 		if flag == 1 {
 			go runtime.EventsEmit(a.ctx, "newsPush", map[string]any{
 				"time":    "当前版本：" + Version,
-				"isRed":   false,
+				"isRed":   true,
 				"source":  "go-stock",
 				"content": "当前版本无更新",
 			})
@@ -545,10 +545,21 @@ func (a *App) CheckStockBaseInfo(ctx context.Context) {
 
 }
 func (a *App) NewsPush(news *[]models.Telegraph) {
+
+	follows := data.NewStockDataApi().GetFollowList(0)
+	stockNames := slice.Map(*follows, func(index int, item data.FollowedStock) string {
+		return item.Name
+	})
+
 	for _, telegraph := range *news {
-		//if telegraph.IsRed {
-		go runtime.EventsEmit(a.ctx, "newsPush", telegraph)
-		go data.NewAlertWindowsApi("go-stock", telegraph.Source+" "+telegraph.Time, telegraph.Content, string(icon)).SendNotification()
+		if a.GetConfig().EnableOnlyPushRedNews {
+			if telegraph.IsRed || strutil.ContainsAny(telegraph.Content, stockNames) {
+				go runtime.EventsEmit(a.ctx, "newsPush", telegraph)
+			}
+		} else {
+			go runtime.EventsEmit(a.ctx, "newsPush", telegraph)
+		}
+		//go data.NewAlertWindowsApi("go-stock", telegraph.Source+" "+telegraph.Time, telegraph.Content, string(icon)).SendNotification()
 		//}
 	}
 }

backend/data/openai_api.go

@@ -219,17 +219,24 @@ func (o *OpenAi) NewSummaryStockNewsStreamWithTools(userQuestion string, sysProm
 		go func() {
 			defer wg.Done()
 			var market strings.Builder
+			market.WriteString(getZSInfo("上证指数", "sh000001", 30) + "\n")
+			market.WriteString(getZSInfo("深证成指", "sz399001", 30) + "\n")
 			market.WriteString(getZSInfo("创业板指数", "sz399006", 30) + "\n")
-			market.WriteString(getZSInfo("上证综合指数", "sh000001", 30) + "\n")
+			market.WriteString(getZSInfo("科创50", "sh000688", 30) + "\n")
 			market.WriteString(getZSInfo("沪深300指数", "sh000300", 30) + "\n")
+			market.WriteString(getZSInfo("中证银行", "sz399986", 30) + "\n")
+			market.WriteString(getZSInfo("科创芯片", "sh000685", 30) + "\n")
+			market.WriteString(getZSInfo("上证医药", "sh000037", 30) + "\n")
+			market.WriteString(getZSInfo("证券龙头", "sz399437", 30) + "\n")
+			market.WriteString(getZSInfo("中证白酒", "sz399997", 30) + "\n")
 			//logger.SugaredLogger.Infof("NewChatStream getZSInfo=\n%s", market.String())
 			msg = append(msg, map[string]interface{}{
 				"role":    "user",
-				"content": "当前市场指数行情",
+				"content": "当前市场/大盘/行业/指数行情",
 			})
 			msg = append(msg, map[string]interface{}{
 				"role":    "assistant",
-				"content": "当前市场指数行情情况如下：\n" + market.String(),
+				"content": "当前市场/大盘/行业/指数行情如下：\n" + market.String(),
 			})
 		}()
 
@@ -302,7 +309,7 @@ func (o *OpenAi) NewSummaryStockNewsStreamWithTools(userQuestion string, sysProm
 		}()
 		wg.Wait()
 
-		news := NewMarketNewsApi().GetNewsList("财联社电报", random.RandInt(50, 150))
+		news := NewMarketNewsApi().GetNewsList("财联社电报", random.RandInt(100, 500))
 		messageText := strings.Builder{}
 		for _, telegraph := range *news {
 			messageText.WriteString("## " + telegraph.Time + ":" + "\n")
@@ -378,9 +385,16 @@ func (o *OpenAi) NewSummaryStockNewsStream(userQuestion string, sysPromptId *int
 		go func() {
 			defer wg.Done()
 			var market strings.Builder
+			market.WriteString(getZSInfo("上证指数", "sh000001", 30) + "\n")
+			market.WriteString(getZSInfo("深证成指", "sz399001", 30) + "\n")
 			market.WriteString(getZSInfo("创业板指数", "sz399006", 30) + "\n")
-			market.WriteString(getZSInfo("上证综合指数", "sh000001", 30) + "\n")
+			market.WriteString(getZSInfo("科创50", "sh000688", 30) + "\n")
 			market.WriteString(getZSInfo("沪深300指数", "sh000300", 30) + "\n")
+			market.WriteString(getZSInfo("中证银行", "sz399986", 30) + "\n")
+			market.WriteString(getZSInfo("科创芯片", "sh000685", 30) + "\n")
+			market.WriteString(getZSInfo("上证医药", "sh000037", 30) + "\n")
+			market.WriteString(getZSInfo("证券龙头", "sz399437", 30) + "\n")
+			market.WriteString(getZSInfo("中证白酒", "sz399997", 30) + "\n")
 			//logger.SugaredLogger.Infof("NewChatStream getZSInfo=\n%s", market.String())
 			msg = append(msg, map[string]interface{}{
 				"role":    "user",

backend/data/openai_api_test.go

@@ -57,3 +57,9 @@ func TestSearchGuShiTongStockInfo(t *testing.T) {
 	SearchGuShiTongStockInfo("gb_goog", 60)
 
 }
+
+func TestGetZSInfo(t *testing.T) {
+	db.Init("../../data/stock.db")
+	getZSInfo("中证银行", "sz399986", 30)
+	getZSInfo("科创50", "sh000688", 30)
+}

backend/data/settings_api.go

@@ -31,6 +31,7 @@ type Settings struct {
 	BrowserPoolSize        int    `json:"browserPoolSize"`
 	EnableFund             bool   `json:"enableFund"`
 	EnablePushNews         bool   `json:"enablePushNews"`
+	EnableOnlyPushRedNews  bool   `json:"enableOnlyPushRedNews"`
 	SponsorCode            string `json:"sponsorCode"`
 	HttpProxy              string `json:"httpProxy"`
 	HttpProxyEnabled       bool   `json:"httpProxyEnabled"`
@@ -100,6 +101,7 @@ func UpdateConfig(s *SettingConfig) string {
 			"dark_theme":                 s.DarkTheme,
 			"enable_fund":                s.EnableFund,
 			"enable_push_news":           s.EnablePushNews,
+			"enable_only_push_red_news":  s.EnableOnlyPushRedNews,
 			"sponsor_code":               s.SponsorCode,
 			"http_proxy":                 s.HttpProxy,
 			"http_proxy_enabled":         s.HttpProxyEnabled,

frontend/src/components/settings.vue

@@ -41,6 +41,7 @@ const formValue = ref({
   darkTheme: true,
   enableFund: false,
   enablePushNews: false,
+  enableOnlyPushRedNews: false,
   sponsorCode: "",
   httpProxy:"",
   httpProxyEnabled:false,
@@ -98,9 +99,11 @@ onMounted(() => {
     formValue.value.darkTheme = res.darkTheme
     formValue.value.enableFund = res.enableFund
     formValue.value.enablePushNews = res.enablePushNews
+    formValue.value.enableOnlyPushRedNews = res.enableOnlyPushRedNews
     formValue.value.sponsorCode = res.sponsorCode
     formValue.value.httpProxy=res.httpProxy;
     formValue.value.httpProxyEnabled=res.httpProxyEnabled;
+
   })
 
   GetPromptTemplates("", "").then(res => {
@@ -135,6 +138,7 @@ function saveConfig() {
     darkTheme: formValue.value.darkTheme,
     enableFund: formValue.value.enableFund,
     enablePushNews: formValue.value.enablePushNews,
+    enableOnlyPushRedNews: formValue.value.enableOnlyPushRedNews,
     sponsorCode: formValue.value.sponsorCode,
     httpProxy:formValue.value.httpProxy,
     httpProxyEnabled:formValue.value.httpProxyEnabled,
@@ -223,6 +227,7 @@ function importConfig() {
       formValue.value.darkTheme = config.darkTheme
       formValue.value.enableFund = config.enableFund
       formValue.value.enablePushNews = config.enablePushNews
+      formValue.value.enableOnlyPushRedNews = config.enableOnlyPushRedNews
       formValue.value.sponsorCode = config.sponsorCode
       formValue.value.httpProxy=config.httpProxy
       formValue.value.httpProxyEnabled=config.httpProxyEnabled
@@ -329,21 +334,25 @@ function deletePrompt(ID) {
 
         <n-card :title="() => h(NTag, { type: 'primary', bordered: false }, () => '通知设置')" size="small">
           <n-grid :cols="24" :x-gap="24" style="text-align: left">
-            <n-form-item-gi :span="4" label="钉钉推送：" path="dingPush.enable">
+            <n-form-item-gi :span="3" label="钉钉推送：" path="dingPush.enable">
               <n-switch v-model:value="formValue.dingPush.enable"/>
             </n-form-item-gi>
-            <n-form-item-gi :span="4" label="本地推送：" path="localPush.enable">
+            <n-form-item-gi :span="3" label="本地推送：" path="localPush.enable">
               <n-switch v-model:value="formValue.localPush.enable"/>
             </n-form-item-gi>
-            <n-form-item-gi :span="4" label="弹幕功能：" path="enableDanmu">
+            <n-form-item-gi :span="3" label="弹幕功能：" path="enableDanmu">
               <n-switch v-model:value="formValue.enableDanmu"/>
             </n-form-item-gi>
-            <n-form-item-gi :span="4" label="显示滚动快讯：" path="enableNews">
+            <n-form-item-gi :span="3" label="显示滚动快讯：" path="enableNews">
               <n-switch v-model:value="formValue.enableNews"/>
             </n-form-item-gi>
-            <n-form-item-gi :span="4" label="市场资讯提醒：" path="enablePushNews">
+            <n-form-item-gi :span="3" label="市场资讯提醒：" path="enablePushNews">
               <n-switch v-model:value="formValue.enablePushNews"/>
             </n-form-item-gi>
+            <n-form-item-gi v-if="formValue.enablePushNews" :span="4" label="只提醒红字或关注个股的新闻：" path="enableOnlyPushRedNews">
+              <n-switch v-model:value="formValue.enableOnlyPushRedNews"/>
+            </n-form-item-gi>
+
             <n-form-item-gi :span="22" v-if="formValue.dingPush.enable" label="钉钉机器人接口地址："
                             path="dingPush.dingRobot">
               <n-input placeholder="请输入钉钉机器人接口地址" v-model:value="formValue.dingPush.dingRobot"/>
@@ -461,7 +470,6 @@ function deletePrompt(ID) {
 
           </n-grid>
         </n-card>
-
       </n-space>
     </n-form>
   </n-flex>

frontend/wailsjs/go/models.ts

@@ -389,6 +389,7 @@ export namespace data {
 	    browserPoolSize: number;
 	    enableFund: boolean;
 	    enablePushNews: boolean;
+	    enableOnlyPushRedNews: boolean;
 	    sponsorCode: string;
 	    httpProxy: string;
 	    httpProxyEnabled: boolean;
@@ -423,6 +424,7 @@ export namespace data {
 	        this.browserPoolSize = source["browserPoolSize"];
 	        this.enableFund = source["enableFund"];
 	        this.enablePushNews = source["enablePushNews"];
+	        this.enableOnlyPushRedNews = source["enableOnlyPushRedNews"];
 	        this.sponsorCode = source["sponsorCode"];
 	        this.httpProxy = source["httpProxy"];
 	        this.httpProxyEnabled = source["httpProxyEnabled"];