add plotting

Author: Dema-koder

src/main/java/org/example/project/configuration/BotKeyboardConfig.java

@@ -26,6 +26,7 @@ public ReplyKeyboardMarkup mainMenuKeyboard() {
 
         KeyboardRow row3 = new KeyboardRow();
         row3.add(new KeyboardButton("/leetcode"));
+        row3.add(new KeyboardButton("/investplot"));
 
         keyboard.add(row1);
         keyboard.add(row2);

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

@@ -1,23 +1,28 @@
-//package org.example.project.service;
-//
-//import org.springframework.stereotype.Service;
-//
-//import java.net.URI;
-//import java.net.URISyntaxException;
-//import java.net.http.HttpClient;
-//import java.net.http.HttpRequest;
-//
-//@Service
-//public class StoncksPlotService {
-//
-//    public String getPlotImage(String period) throws URISyntaxException {
-//        HttpClient client = HttpClient.newHttpClient();
-//        String pythonServiceUrl = "http://localhost:8000/plot?period=" + period;
-//        HttpRequest request = HttpRequest.newBuilder()
-//                .uri(new URI(pythonServiceUrl))
-//                .GET()
-//                .build();
-//        RestTemplate restTemplate = new RestTemplate();
-//        return restTemplate.getForObject(pythonServiceUrl, String.class);
-//    }
-//}
+package org.example.project.service;
+
+import org.springframework.stereotype.Service;
+
+import java.net.URI;
+import java.net.http.HttpClient;
+import java.net.http.HttpRequest;
+import java.net.http.HttpResponse;
+
+@Service
+public class StoncksPlotService {
+
+    public String getPlotImage(String period) {
+        try {
+            HttpClient client = HttpClient.newHttpClient();
+            String pythonServiceUrl = "http://zealous_grothendieck:8000/plot?period=week";
+            HttpRequest request = HttpRequest.newBuilder()
+                    .uri(new URI(pythonServiceUrl))
+                    .GET()
+                    .build();
+            HttpResponse<String> response = client.send(request, HttpResponse.BodyHandlers.ofString());
+            return response.body();
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+        return "";
+    }
+}

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

@@ -6,9 +6,14 @@
 import org.telegram.telegrambots.meta.api.methods.ParseMode;
 import org.telegram.telegrambots.meta.api.methods.send.SendChatAction;
 import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
+import org.telegram.telegrambots.meta.api.methods.send.SendPhoto;
+import org.telegram.telegrambots.meta.api.objects.InputFile;
 import org.telegram.telegrambots.meta.api.objects.replykeyboard.ReplyKeyboardMarkup;
 import org.telegram.telegrambots.meta.exceptions.TelegramApiException;
 
+import java.io.ByteArrayInputStream;
+import java.util.Base64;
+
 @Slf4j
 @Component
 public class TelegramMessageSender implements MessageSender{
@@ -79,4 +84,27 @@ private String[] splitString(String str, int length) {
 
     }
 
+    public void sendPlotToUser(Long chatId, String htmlResponse) {
+        try {
+            String base64Prefix = "data:image/png;base64,";
+            int startIndex = htmlResponse.indexOf(base64Prefix) + base64Prefix.length();
+            int endIndex = htmlResponse.indexOf("\"", startIndex);
+            String base64Image = htmlResponse.substring(startIndex, endIndex);
+
+            byte[] imageBytes = Base64.getDecoder().decode(base64Image);
+
+            InputFile photo = new InputFile(new ByteArrayInputStream(imageBytes), "plot.png");
+
+            SendPhoto sendPhoto = SendPhoto.builder()
+                    .chatId(chatId.toString())
+                    .photo(photo)
+                    .caption("График изменения портфеля")
+                    .build();
+
+            bot.execute(sendPhoto);
+        } catch (TelegramApiException e) {
+            e.printStackTrace();
+        }
+    }
+
 }

src/main/java/org/example/project/service/command/main/InvestPlotHandler.java

@@ -0,0 +1,32 @@
+package org.example.project.service.command.main;
+
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.example.project.domain.Users;
+import org.example.project.service.StoncksPlotService;
+import org.example.project.service.TelegramMessageSender;
+import org.example.project.service.UserService;
+import org.example.project.service.command.CommandHandler;
+import org.springframework.stereotype.Component;
+import org.telegram.telegrambots.meta.api.objects.Update;
+import org.telegram.telegrambots.meta.api.objects.replykeyboard.ReplyKeyboardMarkup;
+
+@Slf4j
+@Component
+@RequiredArgsConstructor
+public class InvestPlotHandler implements MainCommandHandler, CommandHandler {
+    private final ReplyKeyboardMarkup mainMenuKeyboard;
+    private final TelegramMessageSender messageSender;
+    private final StoncksPlotService stoncksPlotService;
+
+    @Override
+    public boolean canHandle(String command) {
+        return command.equals("/investplot");
+    }
+
+    @Override
+    public void handleCommand(long chatId, Update update) {
+        String plot = stoncksPlotService.getPlotImage("week");
+        messageSender.sendPlotToUser(chatId, plot);
+    }
+}