added morning wishes for users

Author: Dema-koder

src/main/java/org/example/project/BotApplication.java

@@ -2,7 +2,9 @@
 
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.scheduling.annotation.EnableScheduling;
 
+@EnableScheduling
 @SpringBootApplication
 public class BotApplication {
     public static void main(String[] args) {

src/main/java/org/example/project/repository/UserRepository.java

@@ -2,8 +2,13 @@
 
 import org.example.project.domain.Users;
 import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.data.jpa.repository.Query;
+
+import java.util.List;
 
 public interface UserRepository extends JpaRepository<Users, Long> {
     Users findByName(String name);
     Users findByChatId(Long chatId);
+    @Query(value = "select * from users", nativeQuery = true)
+    List<Users> getAllUsers();
 }

src/main/java/org/example/project/service/TelegramBot.java

@@ -31,6 +31,8 @@
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
 
+
+// TODO: разбить класс, тут есть функциональность которая нудна для бота, а также функциональность которая отвечает за работу с пользователями
 @Slf4j
 @Component
 public class TelegramBot extends TelegramLongPollingBot {
@@ -171,7 +173,6 @@ private void executeUpdate(Update update) {
         }
     }
 
-    // TODO: Изменить взаимодействие с ChatGPT в боте
     private void questionToGptReceived(long chatId, String question, String model, List<Message> messageHistory) {
         log.info("Sending message to ChatGPT for chatId: {}", chatId);
 
@@ -375,7 +376,8 @@ private void startCommandReceived(long chatId, String name) {
         }
     }
 
-    private void sendMessage(long chatId, String textToSend) {
+    // TODO: убрать из этого класса, так как эта функция нужна и в других классах
+    public void sendMessage(long chatId, String textToSend) {
         var messages = splitString(textToSend, 4096);
         SendMessage sendMessage = new SendMessage();
         for (var message: messages) {
@@ -390,6 +392,7 @@ private void sendMessage(long chatId, String textToSend) {
         }
     }
 
+    // TODO: убрать из этого класса, так как эта функция нужна и в других классах
     private void sendMessageWithKeyboard(long chatId, String textToSend, ReplyKeyboardMarkup keyboard) {
         var messages = splitString(textToSend, 4096);
         SendMessage sendMessage = new SendMessage();

src/main/java/org/example/project/service/job/MorningWishesJob.java

@@ -0,0 +1,37 @@
+package org.example.project.service.job;
+
+import org.example.project.repository.UserRepository;
+import org.example.project.service.ChatGPTService;
+import org.example.project.service.TelegramBot;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.scheduling.annotation.Scheduled;
+import org.springframework.stereotype.Service;
+
+import java.util.ArrayList;
+
+@Service
+public class MorningWishesJob {
+
+    @Autowired
+    private UserRepository userRepository;
+
+    @Autowired
+    private TelegramBot telegramBot;
+
+    @Autowired
+    private ChatGPTService chatGPTService;
+
+    private final String PROMPT = "Напиши пожелание с добрым утром. Это пожелание должно поднимать настроение и " +
+            "настраивать на продуктивный день, оно должно быть не сильно большим, но и не сильно маленьким. " +
+            "В ответ пришли только пожелание, без лишнего текста";
+
+    @Scheduled(cron = "* * 9 * * *")
+    public void scheduleMorningWishes() {
+        var users = userRepository.getAllUsers();
+
+        var message = chatGPTService.sendMessage("", PROMPT, "gpt-3.5-turbo", new ArrayList<>());
+        for (var user: users) {
+            telegramBot.sendMessage(user.getChatId(), message);
+        }
+    }
+}