add stat

main.go

@@ -16,7 +16,7 @@ import (
 )
 
 const (
-	version = "v1.1.0"
+	version = "v1.2.0"
 )
 
 type Opts struct {
@@ -88,6 +88,29 @@ func run() {
 			_, _ = fmt.Fprintf(w, `{"result":"ok"}`)
 		})
 
+		http.HandleFunc("/stat", func(w http.ResponseWriter, r *http.Request) {
+			chats, err := db.GetAllChats()
+			if err != nil {
+				fmt.Println(err)
+				w.Header().Add("Content-Type", "application/json")
+				w.WriteHeader(http.StatusInternalServerError)
+				_, _ = fmt.Fprintf(w, `{"result":"error"}`)
+				return
+			}
+			for _, chatID := range chats {
+				if err := sendStatToChat(bot, chatID, "Напоминаю:\n- Cегодня больше не жрем!\n\n"); err != nil {
+					fmt.Println(err)
+					w.Header().Add("Content-Type", "application/json")
+					w.WriteHeader(http.StatusInternalServerError)
+					_, _ = fmt.Fprintf(w, `{"result":"error"}`)
+					return
+				}
+			}
+			w.Header().Add("Content-Type", "application/json")
+			w.WriteHeader(http.StatusOK)
+			_, _ = fmt.Fprintf(w, `{"result":"ok"}`)
+		})
+
 		port := ":10002"
 		log.Println("Server is start up! port", port)
 		log.Fatal(http.ListenAndServe(port, nil))
@@ -147,16 +170,10 @@ func run() {
 		case commands.Go:
 			sendGoToChat(bot, chatID)
 		case commands.Stat:
-			stat, err := db.GetStat(chatID)
-			if err != nil {
-				log.Println(err)
+			if err := sendStatToChat(bot, chatID, ""); err != nil {
+				fmt.Println(err)
 				continue
 			}
-			msgText := "Результаты за сегодня:\n"
-			for k, v := range stat {
-				msgText += fmt.Sprintf("- %s: %d\n", k, v)
-			}
-			_, _ = bot.Send(tgbot.NewMessage(chatID, msgText))
 		}
 	}
 }
@@ -178,3 +195,16 @@ func sendGoToChat(bot *tgbot.BotAPI, chatID int64) {
 		}
 	}
 }
+
+func sendStatToChat(bot *tgbot.BotAPI, chatID int64, prefix string) error {
+	stat, err := db.GetStat(chatID)
+	if err != nil {
+		return err
+	}
+	msgText := prefix + "Результаты за сегодня:\n"
+	for k, v := range stat {
+		msgText += fmt.Sprintf("- %s: %d\n", k, v)
+	}
+	_, _ = bot.Send(tgbot.NewMessage(chatID, msgText))
+	return nil
+}