Compare commits
106 Commits
feature/ta
...
master
Author | SHA1 | Date |
---|---|---|
|
406f3ba6b3 | |
|
0cd6dd0808 | |
|
7bdcf10cd2 | |
|
a648865083 | |
|
68d0fb2ced | |
|
d23910af9e | |
|
93e04a7e56 | |
|
23b5f02438 | |
|
a03a6abdfc | |
|
0fe1d7e670 | |
|
c3711bb571 | |
|
39c2d6e6fc | |
|
21f4a71c63 | |
|
25b73feb48 | |
|
3a967e399b | |
|
bebf1b2916 | |
|
62a473000a | |
|
b842114bb3 | |
|
187e20ad9d | |
|
6545e01f9b | |
|
f2a74b9f84 | |
|
15d6894013 | |
|
5bd78701e9 | |
|
74e5e2b017 | |
|
30772ec123 | |
|
c5d7689b97 | |
|
5b0decff19 | |
|
2bca814b3f | |
|
24839dc3fe | |
|
2b8a5f326c | |
|
180e1e6be4 | |
|
1a1fca9325 | |
|
e416a84bf7 | |
|
dcd357c6e0 | |
|
43958b0d7c | |
|
c8ff9725e5 | |
|
bac9f84092 | |
|
38a989e223 | |
|
2b782867c9 | |
|
3a0866dfa4 | |
|
386dceed36 | |
|
5d4370d932 | |
|
23effc5fe6 | |
|
e259bb0262 | |
|
9849a46a40 | |
|
757850aae6 | |
|
431e777988 | |
|
8b8833cd2b | |
|
754ae67714 | |
|
7d929254dd | |
|
d822059e4e | |
|
0fecaa501b | |
|
034387af16 | |
|
07b27cb6de | |
|
28ec8dcd64 | |
|
de303fcd8e | |
|
35a8d6a8c7 | |
|
bf070c29a1 | |
|
395da1739b | |
|
34744f93ec | |
|
a20b0c4963 | |
|
94485ab9d2 | |
|
204dbc5f73 | |
|
98bc9633be | |
|
8bdc56bd8a | |
|
051e201525 | |
|
b68520ffa4 | |
|
74078b120c | |
|
94285d88b7 | |
|
008b173b70 | |
|
0ba76e16b2 | |
|
bdb19c81ca | |
|
d56b5a5617 | |
|
182f850a3b | |
|
11c0b4f369 | |
|
a6151fb01e | |
|
3c53ea17a5 | |
|
3b505398e4 | |
|
ead94bea2f | |
|
ecfd0a9ae5 | |
|
7a0c2fde66 | |
|
56a74a8b49 | |
|
0863650ddc | |
|
02a4b5081f | |
|
9ef9ad18cf | |
|
fd39b88296 | |
|
74f02e7827 | |
|
dc07591612 | |
|
0f9721f562 | |
|
742d246cdb | |
|
0ac12deed3 | |
|
9a4652b8f2 | |
|
edd29ea029 | |
|
7838e60a82 | |
|
02a56ee434 | |
|
d60187cc07 | |
|
6359c59af9 | |
|
c84e935f05 | |
|
f9ad7fddbd | |
|
f011205a22 | |
|
5920067b02 | |
|
9e6cba52f6 | |
|
638dc42a46 | |
|
fd4248d722 | |
|
b71b4db66e | |
|
9ee1a4de3c |
|
@ -0,0 +1,11 @@
|
|||
# Используем официальный образ OpenJDK 17 как базовый образ
|
||||
FROM openjdk:17
|
||||
|
||||
# Установка рабочей директории внутри контейнера
|
||||
WORKDIR /app
|
||||
|
||||
# Копируем JAR-файл приложения из локальной директории внутрь контейнера
|
||||
COPY ServerMonitorBot-0.0.1-SNAPSHOT.jar app.jar
|
||||
|
||||
# Команда для запуска Spring Boot приложения при старте контейнера
|
||||
CMD ["java", "-jar", "app.jar"]
|
|
@ -0,0 +1,71 @@
|
|||
pipeline {
|
||||
agent {
|
||||
docker {
|
||||
image 'maven:3.9.5-eclipse-temurin-17'
|
||||
args '-v /root/.m2:/root/.m2'
|
||||
}
|
||||
}
|
||||
|
||||
environment {
|
||||
SSH_HOST = credentials('ROCK_PI_5_SSH_HOST')
|
||||
SSH_PORT = credentials('ROCK_PI_5_SSH_PORT')
|
||||
SSH_USER = credentials('ROCK_PI_5_SSH_USER')
|
||||
SSH_PASS = credentials('ROCK_PI_5_SSH_PASS')
|
||||
PATH_TO_DEV_FOLDER = credentials('SERVERMONITORBOT_PATH_TO_DEV_FOLDER')
|
||||
IMAGE_NAME = 'servermonitorbot'
|
||||
IMAGE_VERSION = 'latest'
|
||||
}
|
||||
|
||||
stages {
|
||||
stage('Init container') {
|
||||
steps {
|
||||
sh 'apt-get update && apt-get install -y sshpass openssh-client'
|
||||
}
|
||||
}
|
||||
stage('Build') {
|
||||
steps {
|
||||
withCredentials([file(credentialsId: 'SERVERMONITORBOT_APPLICATION_YAML', variable: 'application_yaml')]) {
|
||||
sh "cp -f \$application_yaml src/main/resources/application.yml"
|
||||
}
|
||||
sh 'mvn -B -DskipTests -X clean package spring-boot:repackage'
|
||||
}
|
||||
}
|
||||
stage('Stop container') {
|
||||
steps {
|
||||
sh """
|
||||
sshpass -p ${SSH_PASS} ssh -o StrictHostKeyChecking=no ${SSH_USER}@${SSH_HOST} -p ${SSH_PORT} 'cd ${PATH_TO_DEV_FOLDER} && docker compose down'
|
||||
"""
|
||||
}
|
||||
}
|
||||
stage('Remove old image') {
|
||||
steps {
|
||||
sh """
|
||||
sshpass -p ${SSH_PASS} ssh -o StrictHostKeyChecking=no ${SSH_USER}@${SSH_HOST} -p ${SSH_PORT} 'docker image rm -f ${IMAGE_NAME}-app:${IMAGE_VERSION} || true'
|
||||
"""
|
||||
}
|
||||
}
|
||||
stage('Deploy') {
|
||||
steps {
|
||||
script {
|
||||
sh """
|
||||
sshpass -p ${SSH_PASS} ssh -o StrictHostKeyChecking=no ${SSH_USER}@${SSH_HOST} -p ${SSH_PORT} 'mkdir -p ${PATH_TO_DEV_FOLDER}'
|
||||
sshpass -p ${SSH_PASS} ssh -o StrictHostKeyChecking=no ${SSH_USER}@${SSH_HOST} -p ${SSH_PORT} 'rm -f ${PATH_TO_DEV_FOLDER}/{*.*}'
|
||||
sshpass -p ${SSH_PASS} scp -o StrictHostKeyChecking=no -P ${SSH_PORT} target/*.jar ${SSH_USER}@${SSH_HOST}:${PATH_TO_DEV_FOLDER}/
|
||||
sshpass -p ${SSH_PASS} scp -o StrictHostKeyChecking=no -P ${SSH_PORT} Dockerfile ${SSH_USER}@${SSH_HOST}:${PATH_TO_DEV_FOLDER}/
|
||||
sshpass -p ${SSH_PASS} scp -o StrictHostKeyChecking=no -P ${SSH_PORT} docker-compose.yml ${SSH_USER}@${SSH_HOST}:${PATH_TO_DEV_FOLDER}/
|
||||
sshpass -p ${SSH_PASS} ssh -o StrictHostKeyChecking=no ${SSH_USER}@${SSH_HOST} -p ${SSH_PORT} 'cd ${PATH_TO_DEV_FOLDER} && docker build --no-cache -t ${IMAGE_NAME}:${IMAGE_VERSION} .'
|
||||
sshpass -p ${SSH_PASS} ssh -o StrictHostKeyChecking=no ${SSH_USER}@${SSH_HOST} -p ${SSH_PORT} 'cd ${PATH_TO_DEV_FOLDER} && docker compose up -d'
|
||||
"""
|
||||
}
|
||||
}
|
||||
}
|
||||
stage('Remove useless images') {
|
||||
steps {
|
||||
sh """
|
||||
sshpass -p ${SSH_PASS} ssh -o StrictHostKeyChecking=no ${SSH_USER}@${SSH_HOST} -p ${SSH_PORT} 'docker image prune --all --force'
|
||||
"""
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,15 @@
|
|||
pipeline {
|
||||
agent {
|
||||
docker {
|
||||
image 'maven:3.9.5-eclipse-temurin-17'
|
||||
args '-v /root/.m2:/root/.m2'
|
||||
}
|
||||
}
|
||||
stages {
|
||||
stage('Build') {
|
||||
steps {
|
||||
sh 'mvn -B -DskipTests -X clean package spring-boot:repackage'
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
|
@ -0,0 +1,6 @@
|
|||
version: '3'
|
||||
services:
|
||||
app:
|
||||
build:
|
||||
context: .
|
||||
restart: always
|
25
pom.xml
25
pom.xml
|
@ -5,7 +5,7 @@
|
|||
<parent>
|
||||
<groupId>org.springframework.boot</groupId>
|
||||
<artifactId>spring-boot-starter-parent</artifactId>
|
||||
<version>3.1.2</version>
|
||||
<version>3.4.0</version>
|
||||
<relativePath/> <!-- lookup parent from repository -->
|
||||
</parent>
|
||||
<groupId>ru.ldeloff</groupId>
|
||||
|
@ -15,13 +15,13 @@
|
|||
<description>ServerMonitorBot</description>
|
||||
<properties>
|
||||
<java.version>17</java.version>
|
||||
<start-class>ru.ldeloff.servermonitorbot.ServerMonitorBotApplication</start-class>
|
||||
</properties>
|
||||
<dependencies>
|
||||
<dependency>
|
||||
<groupId>org.springframework.boot</groupId>
|
||||
<artifactId>spring-boot-starter</artifactId>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>org.projectlombok</groupId>
|
||||
<artifactId>lombok</artifactId>
|
||||
|
@ -40,25 +40,13 @@
|
|||
<dependency>
|
||||
<groupId>org.telegram</groupId>
|
||||
<artifactId>telegrambots</artifactId>
|
||||
<version>6.7.0</version>
|
||||
<version>6.9.7.1</version>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>com.jcraft</groupId>
|
||||
<artifactId>jsch</artifactId>
|
||||
<version>0.1.55</version>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.springframework.boot</groupId>
|
||||
<artifactId>spring-boot-starter-data-jpa</artifactId>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.liquibase</groupId>
|
||||
<artifactId>liquibase-core</artifactId>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.postgresql</groupId>
|
||||
<artifactId>postgresql</artifactId>
|
||||
</dependency>
|
||||
</dependencies>
|
||||
|
||||
<build>
|
||||
|
@ -74,6 +62,13 @@
|
|||
</exclude>
|
||||
</excludes>
|
||||
</configuration>
|
||||
<executions>
|
||||
<execution>
|
||||
<goals>
|
||||
<goal>repackage</goal>
|
||||
</goals>
|
||||
</execution>
|
||||
</executions>
|
||||
</plugin>
|
||||
</plugins>
|
||||
</build>
|
||||
|
|
|
@ -3,10 +3,12 @@ package ru.ldeloff.servermonitorbot;
|
|||
import org.springframework.boot.SpringApplication;
|
||||
import org.springframework.boot.autoconfigure.SpringBootApplication;
|
||||
import org.springframework.boot.context.properties.EnableConfigurationProperties;
|
||||
import ru.ldeloff.servermonitorbot.config.SshConfig;
|
||||
import org.springframework.scheduling.annotation.EnableScheduling;
|
||||
import ru.ldeloff.servermonitorbot.model.server.SshServers;
|
||||
|
||||
@SpringBootApplication
|
||||
@EnableConfigurationProperties(SshConfig.class)
|
||||
@EnableScheduling
|
||||
@EnableConfigurationProperties(SshServers.class)
|
||||
public class ServerMonitorBotApplication {
|
||||
|
||||
public static void main(String[] args) {
|
||||
|
|
|
@ -0,0 +1,149 @@
|
|||
package ru.ldeloff.servermonitorbot.commands;
|
||||
|
||||
import lombok.Getter;
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import lombok.Setter;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import org.telegram.telegrambots.meta.api.objects.Update;
|
||||
import ru.ldeloff.servermonitorbot.model.user.Role;
|
||||
import ru.ldeloff.servermonitorbot.model.server.SshServer;
|
||||
import ru.ldeloff.servermonitorbot.service.MessagingService;
|
||||
import ru.ldeloff.servermonitorbot.util.UpdateUtil;
|
||||
import ru.ldeloff.servermonitorbot.model.user.User;
|
||||
import ru.ldeloff.servermonitorbot.service.SshService;
|
||||
import ru.ldeloff.servermonitorbot.view.ServerListButtons;
|
||||
|
||||
import java.util.Arrays;
|
||||
import java.util.List;
|
||||
import java.util.Objects;
|
||||
|
||||
@Slf4j
|
||||
@RequiredArgsConstructor
|
||||
public abstract class CommandTemplate {
|
||||
|
||||
private final UpdateUtil updateUtil;
|
||||
private final ServerListButtons serverListButtons;
|
||||
public final SshService sshService;
|
||||
private final MessagingService messagingService;
|
||||
@Getter
|
||||
@Setter
|
||||
private List<Role> expectedRole = List.of(Role.ADMIN);
|
||||
@Getter
|
||||
@Setter
|
||||
private String name;
|
||||
|
||||
public boolean execute(Update update) {
|
||||
if (!isExecute(update)) {
|
||||
return false;
|
||||
}
|
||||
|
||||
User user = updateUtil.getUser(update);
|
||||
String message = updateUtil.getMessage(update);
|
||||
SendMessage answer = new SendMessage();
|
||||
answer.setChatId(user.getTelegramId());
|
||||
answer.setText(message);
|
||||
|
||||
if (!updateUtil.roleChecker(update, expectedRole)) {
|
||||
logNotAuth(user, message);
|
||||
messagingService.send(actionForNotAuth(user, answer));
|
||||
return true;
|
||||
}
|
||||
|
||||
String type = checkTypeCommand(update);
|
||||
switch (type) {
|
||||
case "aggregate":
|
||||
logSuccess(user, message);
|
||||
messagingService.send(executeAggregate(user, answer));
|
||||
return true;
|
||||
case "command":
|
||||
logSuccess(user, message);
|
||||
messagingService.send(executeCommand(user, answer));
|
||||
return true;
|
||||
default:
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
// Методы для проверки команды
|
||||
private boolean isExecute(Update update) {
|
||||
if (update.hasMessage()) {
|
||||
if (update.getMessage().hasText()) {
|
||||
String messageText = update.getMessage().getText();
|
||||
if (Objects.equals(messageText, name)) {
|
||||
log.debug("Команда {} будет выполнена", messageText);
|
||||
return true;
|
||||
}
|
||||
}
|
||||
} else if (update.hasCallbackQuery()) {
|
||||
String [] tags = update.getCallbackQuery().getData().split(":");
|
||||
if (tags.length > 1) {
|
||||
if (Objects.equals(tags[0], name)) {
|
||||
log.debug("Команда {} будет выполнена", tags[0]);
|
||||
return true;
|
||||
}
|
||||
} else {
|
||||
log.debug("Команда не будет выполнена");
|
||||
return false;
|
||||
}
|
||||
}
|
||||
log.debug("Команда не будет выполнена");
|
||||
return false;
|
||||
}
|
||||
private String checkTypeCommand(Update update) {
|
||||
if (update.hasMessage()) {
|
||||
if (update.getMessage().hasText()) {
|
||||
return "aggregate";
|
||||
}
|
||||
} else if (update.hasCallbackQuery()) {
|
||||
String [] tags = update.getCallbackQuery().getData().split(":");
|
||||
if (tags.length > 1) {
|
||||
return "command";
|
||||
} else {
|
||||
return "error";
|
||||
}
|
||||
}
|
||||
return "error";
|
||||
}
|
||||
|
||||
|
||||
// Методы при верных условиях
|
||||
void logSuccess(User user, String message) {
|
||||
log.info("Получена команда '{}' от {} ({}). OK", message, user.getLogin(), user.getTelegramId());
|
||||
}
|
||||
|
||||
public SendMessage executeAggregate(User user, SendMessage message) {
|
||||
return serverListButtons.uiForm(message);
|
||||
}
|
||||
|
||||
SendMessage executeCommand(User user, SendMessage message) {
|
||||
String [] tags = Arrays.stream(message.getText().split(":"))
|
||||
.map(String::trim)
|
||||
.toArray(String[]::new);
|
||||
|
||||
if (tags[1].equals("all")) {
|
||||
message.setText(executeCommandAllHost());
|
||||
} else {
|
||||
message.setText(Objects.requireNonNull(executeCommandSpecificHost(tags[1])));
|
||||
}
|
||||
return message;
|
||||
}
|
||||
private String executeCommandAllHost() {
|
||||
List<SshServer> servers = sshService.getHealthSshServers();
|
||||
StringBuilder response = new StringBuilder();
|
||||
servers.forEach(server -> response.append(executeCommandSpecificHost(server.getName())).append("\n"));
|
||||
return response.toString();
|
||||
}
|
||||
public abstract String executeCommandSpecificHost(String serverName);
|
||||
|
||||
// Методы при неверных условиях
|
||||
void logNotAuth(User user, String message) {
|
||||
log.warn("Получена команда '" + message + "' от " + user.getLogin()
|
||||
+ " (" + user.getTelegramId() + "). Нет прав");
|
||||
}
|
||||
private SendMessage actionForNotAuth(User user, SendMessage message) {
|
||||
message.setText("У пользователя " + user.getLogin() + " (" + user.getTelegramId()
|
||||
+ ") недостаточно прав для выполнения этой команды");
|
||||
return message;
|
||||
}
|
||||
}
|
|
@ -0,0 +1,49 @@
|
|||
package ru.ldeloff.servermonitorbot.commands.cpu.temp.service;
|
||||
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.stereotype.Service;
|
||||
import ru.ldeloff.servermonitorbot.commands.CommandTemplate;
|
||||
import ru.ldeloff.servermonitorbot.service.MessagingService;
|
||||
import ru.ldeloff.servermonitorbot.util.UpdateUtil;
|
||||
import ru.ldeloff.servermonitorbot.model.SshCommand;
|
||||
import ru.ldeloff.servermonitorbot.model.server.SshServer;
|
||||
import ru.ldeloff.servermonitorbot.service.SshService;
|
||||
import ru.ldeloff.servermonitorbot.view.ServerListButtons;
|
||||
|
||||
import java.util.Objects;
|
||||
import java.util.Optional;
|
||||
|
||||
@Service
|
||||
@Slf4j
|
||||
public class CpuTemp extends CommandTemplate {
|
||||
|
||||
public CpuTemp(UpdateUtil updateUtil, ServerListButtons serverListButtons, SshService sshService, MessagingService messagingService) {
|
||||
super(updateUtil, serverListButtons, sshService, messagingService);
|
||||
setName("CPU.temp");
|
||||
}
|
||||
|
||||
@Override
|
||||
public String executeCommandSpecificHost(String serverName) {
|
||||
try {
|
||||
Optional<SshServer> server = sshService.getAllSshServers()
|
||||
.stream()
|
||||
.filter(x -> x.getName().equals(serverName))
|
||||
.findFirst();
|
||||
if (server.isPresent()) {
|
||||
SshCommand result = sshService.execute(new SshCommand("cat /sys/class/thermal/thermal_zone0/temp", 100, server.get()));
|
||||
return server.get().getName() + ": \n" +
|
||||
(Objects.isNull(result.getResponse()) ? "ошибка при выполнении команды" :
|
||||
String.format("%.1f",
|
||||
Double.parseDouble(result.getResponse().replaceAll("\n", ""))/1000))
|
||||
+ "°C";
|
||||
} else {
|
||||
log.error("Ошибка при выполнении команды 'CPUtemp'. Искомый сервер ({}) не найден", serverName);
|
||||
return null;
|
||||
}
|
||||
} catch (NumberFormatException e) {
|
||||
log.error("Ошибка при выполнении команды 'CPUtemp'" ,e.getMessage());
|
||||
return "Сервер " + serverName + " не поддерживает команду 'CPUtemp'";
|
||||
}
|
||||
|
||||
}
|
||||
}
|
|
@ -0,0 +1,87 @@
|
|||
package ru.ldeloff.servermonitorbot.commands.hdd.mdadm.service;
|
||||
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.stereotype.Service;
|
||||
import ru.ldeloff.servermonitorbot.commands.CommandTemplate;
|
||||
import ru.ldeloff.servermonitorbot.service.MessagingService;
|
||||
import ru.ldeloff.servermonitorbot.util.UpdateUtil;
|
||||
import ru.ldeloff.servermonitorbot.model.SshCommand;
|
||||
import ru.ldeloff.servermonitorbot.model.server.SshServer;
|
||||
import ru.ldeloff.servermonitorbot.service.SshService;
|
||||
import ru.ldeloff.servermonitorbot.view.ServerListButtons;
|
||||
|
||||
import java.util.*;
|
||||
|
||||
@Service
|
||||
@Slf4j
|
||||
public class MdadmStatus extends CommandTemplate {
|
||||
|
||||
public MdadmStatus(UpdateUtil updateUtil, ServerListButtons serverListButtons, SshService sshService, MessagingService messagingService) {
|
||||
super(updateUtil, serverListButtons, sshService, messagingService);
|
||||
setName("MDADM");
|
||||
}
|
||||
|
||||
@Override
|
||||
public String executeCommandSpecificHost(String serverName) {
|
||||
Optional<SshServer> server = sshService.getAllSshServers()
|
||||
.stream()
|
||||
.filter(x -> x.getName().equals(serverName))
|
||||
.findFirst();
|
||||
if (server.isPresent()) {
|
||||
// узнаю скок рейдов
|
||||
SshCommand sshCommandRaid = sshService.execute(new SshCommand("mdadm --detail --scan", 500, server.get()));
|
||||
List<String> raidNames = parseRaidNames(sshCommandRaid);
|
||||
// Для каждого рейда надо выяснить статус
|
||||
StringBuilder result = new StringBuilder();
|
||||
raidNames.forEach(raid -> {
|
||||
result.append(raid).append(":\n");
|
||||
SshCommand sshCommandHdd = sshService.execute(new SshCommand("mdadm --detail " + raid, 500, server.get()));
|
||||
List<String> hdds = parseHddState(sshCommandHdd);
|
||||
hdds.forEach(hdd -> result.append(hdd).append("\n"));
|
||||
});
|
||||
return server.get().getName() + ": \n" + result;
|
||||
} else {
|
||||
log.error("Ошибка при выполнении команды 'mdadm'. Искомый сервер ({}) не найден", serverName);
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
private List<String> parseRaidNames(SshCommand result) {
|
||||
List<String> names = new ArrayList<>();
|
||||
Arrays.stream(result.getResponse().split("\n"))
|
||||
.map(line -> line.split(" "))
|
||||
.filter(lineArray -> lineArray.length >= 2)
|
||||
.map(lineArray -> lineArray[1])
|
||||
.forEach(names::add);
|
||||
return names;
|
||||
}
|
||||
|
||||
private List<String> parseHddState(SshCommand result) {
|
||||
List<String> names = new ArrayList<>();
|
||||
names.add("Number Name State");
|
||||
|
||||
String[] lines = result.getResponse().split("\n");
|
||||
|
||||
boolean foundHeader = false;
|
||||
|
||||
for (String line : lines) {
|
||||
if (!foundHeader) {
|
||||
if ( line.contains("Number") &&
|
||||
line.contains("Major") &&
|
||||
line.contains("Minor") &&
|
||||
line.contains("RaidDevice") ) {
|
||||
foundHeader = true;
|
||||
}
|
||||
} else {
|
||||
List<String> elements = Arrays.stream(line.split(" "))
|
||||
.filter(x -> !Objects.equals(x, "")).toList();
|
||||
|
||||
names.add(elements.get(0) + ", "
|
||||
+ elements.get(6) + ", "
|
||||
+ elements.get(4) + " "
|
||||
+ elements.get(5));
|
||||
}
|
||||
}
|
||||
return names;
|
||||
}
|
||||
}
|
|
@ -0,0 +1,39 @@
|
|||
package ru.ldeloff.servermonitorbot.commands.hdd.temp.service;
|
||||
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.stereotype.Service;
|
||||
import ru.ldeloff.servermonitorbot.commands.CommandTemplate;
|
||||
import ru.ldeloff.servermonitorbot.service.MessagingService;
|
||||
import ru.ldeloff.servermonitorbot.util.UpdateUtil;
|
||||
import ru.ldeloff.servermonitorbot.model.SshCommand;
|
||||
import ru.ldeloff.servermonitorbot.model.server.SshServer;
|
||||
import ru.ldeloff.servermonitorbot.service.SshService;
|
||||
import ru.ldeloff.servermonitorbot.view.ServerListButtons;
|
||||
|
||||
import java.util.Objects;
|
||||
import java.util.Optional;
|
||||
|
||||
@Service
|
||||
@Slf4j
|
||||
public class HddTemp extends CommandTemplate {
|
||||
public HddTemp(UpdateUtil updateUtil, ServerListButtons serverListButtons, SshService sshService, MessagingService messagingService) {
|
||||
super(updateUtil, serverListButtons, sshService, messagingService);
|
||||
setName("HDD.temp");
|
||||
}
|
||||
|
||||
@Override
|
||||
public String executeCommandSpecificHost(String serverName) {
|
||||
Optional<SshServer> server = sshService.getAllSshServers()
|
||||
.stream()
|
||||
.filter(x -> x.getName().equals(serverName))
|
||||
.findFirst();
|
||||
if (server.isPresent()) {
|
||||
SshCommand result = sshService.execute(new SshCommand("hddtemp /dev/sd*", 500, server.get()));
|
||||
return server.get().getName() + ": \n" +
|
||||
(Objects.isNull(result.getResponse()) ? "ошибка при выполнении команды" : result.getResponse());
|
||||
} else {
|
||||
log.error("Ошибка при выполнении команды 'hddtemp'. Искомый сервер (" + serverName + ") не найден");
|
||||
return null;
|
||||
}
|
||||
}
|
||||
}
|
|
@ -0,0 +1,18 @@
|
|||
package ru.ldeloff.servermonitorbot.commands.statuschecker.scheduler;
|
||||
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import org.springframework.scheduling.annotation.Scheduled;
|
||||
import org.springframework.stereotype.Service;
|
||||
import ru.ldeloff.servermonitorbot.service.SshService;
|
||||
|
||||
@Service
|
||||
@RequiredArgsConstructor
|
||||
public class CheckSessionStatusScheduler {
|
||||
|
||||
private final SshService sshService;
|
||||
|
||||
@Scheduled(fixedRateString = "${ssh.checkConnectionDelay}")
|
||||
public void updateSessionStatusCheck() {
|
||||
sshService.updateConnection();
|
||||
}
|
||||
}
|
|
@ -0,0 +1,28 @@
|
|||
package ru.ldeloff.servermonitorbot.commands.statuschecker.service;
|
||||
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import ru.ldeloff.servermonitorbot.commands.CommandTemplate;
|
||||
import ru.ldeloff.servermonitorbot.service.MessagingService;
|
||||
import ru.ldeloff.servermonitorbot.util.UpdateUtil;
|
||||
import ru.ldeloff.servermonitorbot.model.user.User;
|
||||
import ru.ldeloff.servermonitorbot.service.SshService;
|
||||
import ru.ldeloff.servermonitorbot.view.ServerListButtons;
|
||||
|
||||
@Service
|
||||
public class CheckSessionStatus extends CommandTemplate {
|
||||
|
||||
public CheckSessionStatus(UpdateUtil updateUtil, ServerListButtons serverListButtons, SshService sshService, MessagingService messagingService) {
|
||||
super(updateUtil, serverListButtons, sshService, messagingService);
|
||||
setName("Статус");
|
||||
}
|
||||
|
||||
public SendMessage executeAggregate(User user, SendMessage message) {
|
||||
return super.sshService.getStatusSessions(message);
|
||||
}
|
||||
|
||||
@Override
|
||||
public String executeCommandSpecificHost(String serverName) {
|
||||
return null;
|
||||
}
|
||||
}
|
|
@ -0,0 +1,40 @@
|
|||
package ru.ldeloff.servermonitorbot.commands.uname.service;
|
||||
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.stereotype.Service;
|
||||
import ru.ldeloff.servermonitorbot.commands.CommandTemplate;
|
||||
import ru.ldeloff.servermonitorbot.service.MessagingService;
|
||||
import ru.ldeloff.servermonitorbot.util.UpdateUtil;
|
||||
import ru.ldeloff.servermonitorbot.model.SshCommand;
|
||||
import ru.ldeloff.servermonitorbot.model.server.SshServer;
|
||||
import ru.ldeloff.servermonitorbot.service.SshService;
|
||||
import ru.ldeloff.servermonitorbot.view.ServerListButtons;
|
||||
|
||||
import java.util.Objects;
|
||||
import java.util.Optional;
|
||||
|
||||
@Service
|
||||
@Slf4j
|
||||
public class Uname extends CommandTemplate {
|
||||
|
||||
public Uname(UpdateUtil updateUtil, ServerListButtons serverListButtons, SshService sshService, MessagingService messagingService) {
|
||||
super(updateUtil, serverListButtons, sshService, messagingService);
|
||||
setName("uname");
|
||||
}
|
||||
|
||||
@Override
|
||||
public String executeCommandSpecificHost(String serverName) {
|
||||
Optional<SshServer> server = sshService.getAllSshServers()
|
||||
.stream()
|
||||
.filter(x -> x.getName().equals(serverName))
|
||||
.findFirst();
|
||||
if (server.isPresent()) {
|
||||
SshCommand result = sshService.execute(new SshCommand("uname -a", 100, server.get()));
|
||||
return server.get().getName() + ": " +
|
||||
(Objects.isNull(result.getResponse()) ? "ошибка при выполнении команды" : result.getResponse());
|
||||
} else {
|
||||
log.error("Ошибка при выполнении команды 'uname'. Искомый сервер ({}) не найден", serverName);
|
||||
return null;
|
||||
}
|
||||
}
|
||||
}
|
|
@ -2,16 +2,37 @@ package ru.ldeloff.servermonitorbot.config;
|
|||
|
||||
import lombok.Getter;
|
||||
import lombok.Setter;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.boot.context.properties.ConfigurationProperties;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
import ru.ldeloff.servermonitorbot.model.dto.InitUserDto;
|
||||
import org.springframework.stereotype.Component;
|
||||
import ru.ldeloff.servermonitorbot.model.user.User;
|
||||
import ru.ldeloff.servermonitorbot.service.RoleService;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
@Configuration
|
||||
@Component
|
||||
@ConfigurationProperties(prefix = "bot")
|
||||
@Getter
|
||||
@Setter
|
||||
public class UserConfig {
|
||||
private List<InitUserDto> users;
|
||||
|
||||
private List<UserProperty> users;
|
||||
|
||||
@Autowired
|
||||
private RoleService roleService;
|
||||
|
||||
@Getter
|
||||
@Setter
|
||||
public static class UserProperty {
|
||||
private Long telegramId;
|
||||
private String role;
|
||||
}
|
||||
|
||||
public List<User> getUsers() {
|
||||
return users.stream().map(user -> new User()
|
||||
.setTelegramId(user.getTelegramId())
|
||||
.setRole(roleService.findRoleByName(user.getRole()))
|
||||
).toList();
|
||||
}
|
||||
|
||||
}
|
||||
|
|
|
@ -1,98 +1,60 @@
|
|||
package ru.ldeloff.servermonitorbot.controller;
|
||||
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.telegram.telegrambots.bots.TelegramLongPollingBot;
|
||||
import org.telegram.telegrambots.meta.api.objects.Message;
|
||||
import org.telegram.telegrambots.meta.api.objects.Update;
|
||||
import ru.ldeloff.servermonitorbot.model.TelegramBot;
|
||||
import ru.ldeloff.servermonitorbot.repository.SshRepository;
|
||||
import ru.ldeloff.servermonitorbot.service.command.FirstUseCommand;
|
||||
import ru.ldeloff.servermonitorbot.service.command.GetStatusSessions;
|
||||
import ru.ldeloff.servermonitorbot.service.command.SwitchToMainMenu;
|
||||
import ru.ldeloff.servermonitorbot.service.command.cputemp.CpuTempAggregateCommand;
|
||||
import ru.ldeloff.servermonitorbot.service.command.cputemp.CpuTempCommand;
|
||||
import ru.ldeloff.servermonitorbot.service.command.hddtemp.HddTempAggregateCommand;
|
||||
import ru.ldeloff.servermonitorbot.service.command.hddtemp.HddTempCommand;
|
||||
import ru.ldeloff.servermonitorbot.service.command.uname.UnameAggregateCommand;
|
||||
import ru.ldeloff.servermonitorbot.service.command.uname.UnameCommand;
|
||||
import ru.ldeloff.servermonitorbot.service.SshService;
|
||||
import ru.ldeloff.servermonitorbot.commands.CommandTemplate;
|
||||
import ru.ldeloff.servermonitorbot.view.SwitchToMainMenu;
|
||||
|
||||
import java.util.Map;
|
||||
|
||||
@Service
|
||||
@Slf4j
|
||||
@RequiredArgsConstructor
|
||||
public class TelegramBotController extends TelegramLongPollingBot {
|
||||
final TelegramBot telegramBot;
|
||||
final SshRepository sshRepository;
|
||||
|
||||
final FirstUseCommand firstUseCommand;
|
||||
final GetStatusSessions getStatusSessions;
|
||||
final UnameAggregateCommand unameAggregateCommand;
|
||||
final SwitchToMainMenu switchToMainMenu;
|
||||
final UnameCommand unameCommand;
|
||||
final CpuTempAggregateCommand cpuTempAggregateCommand;
|
||||
final CpuTempCommand cpuTempCommand;
|
||||
final HddTempAggregateCommand hddTempAggregateCommand;
|
||||
final HddTempCommand hddTempCommand;
|
||||
private final Map<String, CommandTemplate> commands;
|
||||
|
||||
private final TelegramBot telegramBot;
|
||||
private final SwitchToMainMenu switchToMainMenu;
|
||||
private final SshService sshService;
|
||||
|
||||
|
||||
@Override
|
||||
public void onUpdateReceived(Update update) {
|
||||
if (update.hasMessage()) {
|
||||
if (update.getMessage().hasText()) {
|
||||
String messageText = update.getMessage().getText();
|
||||
switch (messageText) {
|
||||
case "/start" -> {
|
||||
firstUseCommand.execute(update, this);
|
||||
}
|
||||
case "Статус" -> {
|
||||
getStatusSessions.execute(update, this);
|
||||
}
|
||||
case "uname" -> {
|
||||
unameAggregateCommand.execute(update, this);
|
||||
}
|
||||
case "CPU.temp" -> {
|
||||
cpuTempAggregateCommand.execute(update, this);
|
||||
}
|
||||
case "HDD.temp" -> {
|
||||
hddTempAggregateCommand.execute(update, this);
|
||||
}
|
||||
default -> {
|
||||
switchToMainMenu.execute(update, this);
|
||||
boolean result = false;
|
||||
for (Map.Entry<String, CommandTemplate> entry : commands.entrySet()) {
|
||||
result = entry.getValue().execute(update);
|
||||
if (result) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (!result) {
|
||||
Message message = update.getMessage();
|
||||
if (message!=null) {
|
||||
message.setText("Not found");
|
||||
}
|
||||
} else if (update.hasCallbackQuery()) {
|
||||
String [] tags = update.getCallbackQuery().getData().split(":");
|
||||
if (tags.length > 1) {
|
||||
switch (tags[0]) {
|
||||
case "uname" -> {
|
||||
unameCommand.execute(update, this);
|
||||
}
|
||||
case "CPU.temp" -> {
|
||||
cpuTempCommand.execute(update, this);
|
||||
}
|
||||
case "HDD.temp" -> {
|
||||
hddTempCommand.execute(update, this);
|
||||
}
|
||||
default -> {
|
||||
switchToMainMenu.execute(update, this);
|
||||
}
|
||||
}
|
||||
} else {
|
||||
switchToMainMenu.execute(update, this);
|
||||
}
|
||||
update.setMessage(message);
|
||||
switchToMainMenu.execute(update);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getBotUsername() {
|
||||
return telegramBot.getBotUsername();
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getBotToken() {
|
||||
return telegramBot.getBotToken();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void onClosing() {
|
||||
super.onClosing();
|
||||
sshRepository.disconnectSessions();
|
||||
sshService.disconnectAllConnections();
|
||||
}
|
||||
}
|
||||
|
|
|
@ -1,34 +1,26 @@
|
|||
package ru.ldeloff.servermonitorbot.init;
|
||||
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.boot.ApplicationArguments;
|
||||
import org.springframework.boot.ApplicationRunner;
|
||||
import org.springframework.core.annotation.Order;
|
||||
import org.springframework.stereotype.Component;
|
||||
import ru.ldeloff.servermonitorbot.config.UserConfig;
|
||||
import ru.ldeloff.servermonitorbot.mapper.UserMapper;
|
||||
import ru.ldeloff.servermonitorbot.model.User;
|
||||
import ru.ldeloff.servermonitorbot.model.dto.InitUserDto;
|
||||
import ru.ldeloff.servermonitorbot.service.user.UserService;
|
||||
|
||||
import java.util.List;
|
||||
import ru.ldeloff.servermonitorbot.service.UserService;
|
||||
|
||||
@Slf4j
|
||||
@Component
|
||||
@RequiredArgsConstructor
|
||||
@Order(2)
|
||||
public class AddUsers implements ApplicationRunner {
|
||||
final List<InitUserDto> initUsers;
|
||||
final UserService userService;
|
||||
final UserMapper userMapper;
|
||||
|
||||
@Autowired
|
||||
public AddUsers(UserConfig userConfig, UserService userService, UserMapper userMapper) {
|
||||
this.initUsers = userConfig.getUsers();
|
||||
this.userService = userService;
|
||||
this.userMapper = userMapper;
|
||||
}
|
||||
final UserConfig userConfig;
|
||||
final UserService userService;
|
||||
|
||||
@Override
|
||||
public void run(ApplicationArguments args) throws Exception {
|
||||
initUsers.forEach(initUserDto -> {
|
||||
User user = userService.saveOrUpdateUser(userMapper.dtoToUser(initUserDto));
|
||||
});
|
||||
public void run(ApplicationArguments args) {
|
||||
userConfig.getUsers().forEach(userService::saveOrUpdateUser);
|
||||
log.info("Загружены и обновлены пользователи");
|
||||
}
|
||||
}
|
||||
|
|
|
@ -0,0 +1,33 @@
|
|||
package ru.ldeloff.servermonitorbot.init;
|
||||
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.boot.ApplicationArguments;
|
||||
import org.springframework.boot.ApplicationRunner;
|
||||
import org.springframework.core.annotation.Order;
|
||||
import org.springframework.stereotype.Component;
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import ru.ldeloff.servermonitorbot.service.MessagingService;
|
||||
import ru.ldeloff.servermonitorbot.service.UserService;
|
||||
import ru.ldeloff.servermonitorbot.view.TelegramBotKeyboard;
|
||||
|
||||
@Slf4j
|
||||
@Component
|
||||
@RequiredArgsConstructor
|
||||
@Order(3)
|
||||
public class ReloadNotification implements ApplicationRunner {
|
||||
|
||||
private final TelegramBotKeyboard telegramBotKeyboard;
|
||||
private final MessagingService messagingService;
|
||||
private final UserService userService;
|
||||
|
||||
@Override
|
||||
public void run(ApplicationArguments args) {
|
||||
userService.getAllUsers().forEach(user -> {
|
||||
SendMessage sendMessage = new SendMessage();
|
||||
sendMessage.setChatId(user.getTelegramId());
|
||||
sendMessage.setText("Бот перезапущен");
|
||||
messagingService.send(telegramBotKeyboard.uiForm(sendMessage));
|
||||
});
|
||||
}
|
||||
}
|
|
@ -1,20 +1,21 @@
|
|||
package ru.ldeloff.servermonitorbot.init;
|
||||
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import org.springframework.boot.ApplicationArguments;
|
||||
import org.springframework.boot.ApplicationRunner;
|
||||
import org.springframework.core.annotation.Order;
|
||||
import org.springframework.stereotype.Component;
|
||||
import ru.ldeloff.servermonitorbot.repository.SshRepository;
|
||||
import ru.ldeloff.servermonitorbot.service.SshService;
|
||||
|
||||
@Component
|
||||
@RequiredArgsConstructor
|
||||
@Order(4)
|
||||
public class SshConnect implements ApplicationRunner {
|
||||
final SshRepository sshRepository;
|
||||
|
||||
public SshConnect(SshRepository sshRepository) {
|
||||
this.sshRepository = sshRepository;
|
||||
}
|
||||
final SshService sshService;
|
||||
|
||||
@Override
|
||||
public void run(ApplicationArguments args) throws Exception {
|
||||
sshRepository.connectToAllServer();
|
||||
public void run(ApplicationArguments args) {
|
||||
sshService.updateConnection();
|
||||
}
|
||||
}
|
||||
|
|
|
@ -1,26 +1,24 @@
|
|||
package ru.ldeloff.servermonitorbot.init;
|
||||
|
||||
import lombok.NoArgsConstructor;
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.boot.ApplicationArguments;
|
||||
import org.springframework.boot.ApplicationRunner;
|
||||
import org.springframework.core.annotation.Order;
|
||||
import org.springframework.stereotype.Component;
|
||||
import org.telegram.telegrambots.meta.TelegramBotsApi;
|
||||
import org.telegram.telegrambots.meta.exceptions.TelegramApiException;
|
||||
import org.telegram.telegrambots.updatesreceivers.DefaultBotSession;
|
||||
import ru.ldeloff.servermonitorbot.controller.TelegramBotController;
|
||||
|
||||
|
||||
@Slf4j
|
||||
@Component
|
||||
@NoArgsConstructor
|
||||
@RequiredArgsConstructor
|
||||
@Order(1)
|
||||
public class StartBot implements ApplicationRunner {
|
||||
private TelegramBotController telegramBot;
|
||||
|
||||
@Autowired
|
||||
public StartBot(TelegramBotController telegramBot) {
|
||||
this.telegramBot = telegramBot;
|
||||
}
|
||||
private final TelegramBotController telegramBot;
|
||||
|
||||
@Override
|
||||
public void run(ApplicationArguments args) {
|
||||
|
@ -29,7 +27,6 @@ public class StartBot implements ApplicationRunner {
|
|||
botsApi.registerBot(telegramBot);
|
||||
log.info("Бот запущен");
|
||||
} catch (TelegramApiException e) {
|
||||
e.printStackTrace();
|
||||
log.error(e.getMessage());
|
||||
}
|
||||
}
|
||||
|
|
|
@ -1,22 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.mapper;
|
||||
|
||||
import org.springframework.stereotype.Component;
|
||||
import ru.ldeloff.servermonitorbot.model.User;
|
||||
import ru.ldeloff.servermonitorbot.model.dto.InitUserDto;
|
||||
import ru.ldeloff.servermonitorbot.service.role.RoleService;
|
||||
|
||||
@Component
|
||||
public class UserMapper {
|
||||
final RoleService roleService;
|
||||
|
||||
public UserMapper(RoleService roleService) {
|
||||
this.roleService = roleService;
|
||||
}
|
||||
|
||||
public User dtoToUser(InitUserDto initUserDto) {
|
||||
User user = new User();
|
||||
user.setTelegramId(initUserDto.getTelegramId());
|
||||
user.setRole(roleService.findRoleByName(initUserDto.getRole()));
|
||||
return user;
|
||||
}
|
||||
}
|
|
@ -1,17 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.model;
|
||||
|
||||
import jakarta.persistence.*;
|
||||
import lombok.Data;
|
||||
|
||||
@Entity
|
||||
@Data
|
||||
@Table(name = "roles")
|
||||
public class Role {
|
||||
@Id
|
||||
@Column(name = "id")
|
||||
@GeneratedValue(strategy = GenerationType.IDENTITY)
|
||||
private long id;
|
||||
|
||||
@Column(name = "name")
|
||||
private String name;
|
||||
}
|
|
@ -4,10 +4,11 @@ import jakarta.validation.constraints.Min;
|
|||
import jakarta.validation.constraints.NotNull;
|
||||
import lombok.Getter;
|
||||
import lombok.Setter;
|
||||
import ru.ldeloff.servermonitorbot.model.server.SshServer;
|
||||
|
||||
@Getter
|
||||
@Setter
|
||||
public class Command {
|
||||
public class SshCommand {
|
||||
@NotNull
|
||||
private String command;
|
||||
@NotNull
|
||||
|
@ -17,7 +18,7 @@ public class Command {
|
|||
private SshServer sshServer;
|
||||
private String response;
|
||||
|
||||
public Command(String command, int timeout, SshServer sshServer) {
|
||||
public SshCommand(String command, int timeout, SshServer sshServer) {
|
||||
this.command = command;
|
||||
this.timeout = timeout;
|
||||
this.sshServer = sshServer;
|
|
@ -1,24 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.model;
|
||||
|
||||
import jakarta.persistence.*;
|
||||
import lombok.Data;
|
||||
|
||||
@Entity
|
||||
@Data
|
||||
@Table(name = "users")
|
||||
public class User {
|
||||
@Id
|
||||
@Column(name = "id")
|
||||
@GeneratedValue(strategy = GenerationType.IDENTITY)
|
||||
private Long id;
|
||||
|
||||
@Column(name = "telegram_id")
|
||||
private Long telegramId;
|
||||
|
||||
@ManyToOne
|
||||
@JoinColumn(name = "role_id")
|
||||
private Role role;
|
||||
|
||||
@Transient
|
||||
private String login;
|
||||
}
|
|
@ -1,14 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.model.dto;
|
||||
|
||||
import jakarta.validation.constraints.NotNull;
|
||||
import lombok.Data;
|
||||
import org.springframework.stereotype.Component;
|
||||
|
||||
@Data
|
||||
@Component
|
||||
public class InitUserDto {
|
||||
@NotNull
|
||||
private long telegramId;
|
||||
@NotNull
|
||||
private String role;
|
||||
}
|
|
@ -0,0 +1,7 @@
|
|||
package ru.ldeloff.servermonitorbot.model.server;
|
||||
|
||||
public enum ServerStatus {
|
||||
ONLINE,
|
||||
OFFLINE,
|
||||
CONNECTION_LOST
|
||||
}
|
|
@ -1,4 +1,4 @@
|
|||
package ru.ldeloff.servermonitorbot.model;
|
||||
package ru.ldeloff.servermonitorbot.model.server;
|
||||
|
||||
import com.jcraft.jsch.Session;
|
||||
import jakarta.validation.constraints.NotNull;
|
||||
|
@ -21,4 +21,5 @@ public class SshServer {
|
|||
@NotNull
|
||||
private String password;
|
||||
private Session session;
|
||||
private ServerStatus status;
|
||||
}
|
|
@ -1,11 +1,9 @@
|
|||
package ru.ldeloff.servermonitorbot.config;
|
||||
package ru.ldeloff.servermonitorbot.model.server;
|
||||
|
||||
import lombok.Getter;
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import lombok.Setter;
|
||||
import org.springframework.boot.context.properties.ConfigurationProperties;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
import ru.ldeloff.servermonitorbot.model.SshServer;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
|
@ -13,6 +11,6 @@ import java.util.List;
|
|||
@ConfigurationProperties(prefix = "ssh")
|
||||
@Getter
|
||||
@Setter
|
||||
public class SshConfig {
|
||||
public class SshServers {
|
||||
private List<SshServer> servers;
|
||||
}
|
|
@ -0,0 +1,7 @@
|
|||
package ru.ldeloff.servermonitorbot.model.user;
|
||||
|
||||
public enum Role {
|
||||
ADMIN,
|
||||
USER,
|
||||
ANONYMOUS
|
||||
}
|
|
@ -0,0 +1,12 @@
|
|||
package ru.ldeloff.servermonitorbot.model.user;
|
||||
|
||||
import lombok.Data;
|
||||
import lombok.experimental.Accessors;
|
||||
|
||||
@Data
|
||||
@Accessors(chain = true)
|
||||
public class User {
|
||||
private Long telegramId;
|
||||
private Role role;
|
||||
private String login;
|
||||
}
|
|
@ -0,0 +1,13 @@
|
|||
package ru.ldeloff.servermonitorbot.model.user;
|
||||
|
||||
import lombok.Getter;
|
||||
import org.springframework.stereotype.Component;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
|
||||
@Getter
|
||||
@Component
|
||||
public class Users {
|
||||
private final List<User> users = new ArrayList<>();
|
||||
}
|
|
@ -1,13 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.repository;
|
||||
|
||||
import org.springframework.data.jpa.repository.JpaRepository;
|
||||
import org.springframework.stereotype.Repository;
|
||||
import ru.ldeloff.servermonitorbot.model.Role;
|
||||
|
||||
import java.util.Optional;
|
||||
|
||||
@Repository
|
||||
public interface RoleRepository extends JpaRepository<Role, Long> {
|
||||
Optional<Role> findRoleByName(String name);
|
||||
Optional<Role> findRoleById(Long id);
|
||||
}
|
|
@ -1,16 +1,12 @@
|
|||
package ru.ldeloff.servermonitorbot.repository;
|
||||
|
||||
import com.jcraft.jsch.JSchException;
|
||||
import com.jcraft.jsch.Session;
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import ru.ldeloff.servermonitorbot.model.SshServer;
|
||||
|
||||
import java.util.List;
|
||||
import ru.ldeloff.servermonitorbot.model.server.SshServer;
|
||||
|
||||
public interface SshRepository {
|
||||
void connectToAllServer();
|
||||
Session connectToServer(SshServer sshServer) throws JSchException;
|
||||
void disconnectSessions();
|
||||
String getStatusSessions();
|
||||
List<SshServer> getSshServers();
|
||||
|
||||
void connect(SshServer sshServer) throws JSchException;
|
||||
|
||||
void disconnect(SshServer sshServer);
|
||||
|
||||
}
|
||||
|
|
|
@ -3,86 +3,35 @@ package ru.ldeloff.servermonitorbot.repository;
|
|||
import com.jcraft.jsch.JSch;
|
||||
import com.jcraft.jsch.JSchException;
|
||||
import com.jcraft.jsch.Session;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import org.springframework.beans.factory.annotation.Value;
|
||||
import org.springframework.stereotype.Repository;
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import ru.ldeloff.servermonitorbot.config.SshConfig;
|
||||
import ru.ldeloff.servermonitorbot.model.SshServer;
|
||||
import ru.ldeloff.servermonitorbot.model.server.SshServer;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.Objects;
|
||||
|
||||
@Slf4j
|
||||
@Repository
|
||||
@RequiredArgsConstructor
|
||||
public class SshRepositoryImpl implements SshRepository {
|
||||
|
||||
private final List<SshServer> sshServers;
|
||||
|
||||
@Autowired
|
||||
public SshRepositoryImpl(SshConfig sshConfig) {
|
||||
this.sshServers = sshConfig.getServers();
|
||||
}
|
||||
|
||||
private List<Session> sessions = new ArrayList<>();
|
||||
@Value("${ssh.timeout:5000}")
|
||||
private int TIMEOUT;
|
||||
|
||||
@Override
|
||||
public void connectToAllServer() {
|
||||
sshServers.forEach(sshServer -> {
|
||||
try {
|
||||
Session session = connectToServer(sshServer);
|
||||
sshServer.setSession(session);
|
||||
log.info("Успешно подключён к " + sshServer.getHost());
|
||||
sessions.add(session);
|
||||
} catch (JSchException e) {
|
||||
log.warn("Не удалось соединиться с " + sshServer.getHost() + ": " + e.getMessage());
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
@Override
|
||||
public Session connectToServer(SshServer sshServer) throws JSchException {
|
||||
public void connect(SshServer sshServer) throws JSchException {
|
||||
Session session = new JSch().getSession(sshServer.getUser(),
|
||||
sshServer.getHost(),
|
||||
sshServer.getPort());
|
||||
session.setPassword(sshServer.getPassword());
|
||||
session.setConfig("StrictHostKeyChecking", "no");
|
||||
session.setTimeout(TIMEOUT);
|
||||
|
||||
session.connect();
|
||||
return session;
|
||||
|
||||
sshServer.setSession(session);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void disconnectSessions() {
|
||||
if (!sessions.isEmpty()) {
|
||||
sessions.forEach(Session::disconnect);
|
||||
sessions.forEach(sessions::remove);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getStatusSessions() {
|
||||
StringBuilder text = new StringBuilder("Статус соединения: \n");
|
||||
sshServers.forEach(server -> {
|
||||
text.append(server.getName())
|
||||
.append(": ")
|
||||
.append(checkStatusServer(server))
|
||||
.append("\n");
|
||||
});
|
||||
return String.valueOf(text);
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<SshServer> getSshServers() {
|
||||
return sshServers;
|
||||
}
|
||||
|
||||
private String checkStatusServer(SshServer server) {
|
||||
if (Objects.isNull(server.getSession())) {
|
||||
return "нет соединения";
|
||||
} else {
|
||||
return server.getSession().isConnected() ? "OK" : "отключён";
|
||||
}
|
||||
public void disconnect(SshServer sshServer) {
|
||||
sshServer.getSession().disconnect();
|
||||
}
|
||||
|
||||
}
|
||||
|
|
|
@ -1,10 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.repository;
|
||||
|
||||
import org.springframework.data.jpa.repository.JpaRepository;
|
||||
import org.springframework.stereotype.Repository;
|
||||
import ru.ldeloff.servermonitorbot.model.User;
|
||||
|
||||
@Repository
|
||||
public interface UserRepository extends JpaRepository<User, Long> {
|
||||
User getByTelegramId(long id);
|
||||
}
|
|
@ -0,0 +1,11 @@
|
|||
package ru.ldeloff.servermonitorbot.service;
|
||||
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
|
||||
public interface MessagingService {
|
||||
void sendMessageToAll(String message);
|
||||
|
||||
void sendMessageToAdmins(String message);
|
||||
|
||||
void send(SendMessage message);
|
||||
}
|
|
@ -0,0 +1,48 @@
|
|||
package ru.ldeloff.servermonitorbot.service;
|
||||
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.context.ApplicationContext;
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import org.telegram.telegrambots.meta.exceptions.TelegramApiException;
|
||||
import ru.ldeloff.servermonitorbot.controller.TelegramBotController;
|
||||
|
||||
|
||||
@Slf4j
|
||||
@Service
|
||||
@RequiredArgsConstructor
|
||||
public class MessagingServiceImpl implements MessagingService {
|
||||
|
||||
private final UserService userService;
|
||||
private final ApplicationContext applicationContext;
|
||||
|
||||
@Override
|
||||
public void sendMessageToAll(String message) {
|
||||
userService.getAllUsers().forEach(user -> {
|
||||
SendMessage sendMessage = new SendMessage();
|
||||
sendMessage.setChatId(user.getTelegramId());
|
||||
sendMessage.setText(message);
|
||||
send(sendMessage);
|
||||
});
|
||||
}
|
||||
|
||||
@Override
|
||||
public void sendMessageToAdmins(String message) {
|
||||
userService.getAllAdmins().forEach(user -> {
|
||||
SendMessage sendMessage = new SendMessage();
|
||||
sendMessage.setChatId(user.getTelegramId());
|
||||
sendMessage.setText(message);
|
||||
send(sendMessage);
|
||||
});
|
||||
}
|
||||
|
||||
@Override
|
||||
public void send(SendMessage message) {
|
||||
try {
|
||||
applicationContext.getBean(TelegramBotController.class).execute(message);
|
||||
} catch (TelegramApiException e) {
|
||||
log.warn("Ошибка при отправке сообщения", e.getMessage());
|
||||
}
|
||||
}
|
||||
}
|
|
@ -0,0 +1,7 @@
|
|||
package ru.ldeloff.servermonitorbot.service;
|
||||
|
||||
import ru.ldeloff.servermonitorbot.model.user.Role;
|
||||
|
||||
public interface RoleService {
|
||||
Role findRoleByName(String name);
|
||||
}
|
|
@ -0,0 +1,14 @@
|
|||
package ru.ldeloff.servermonitorbot.service;
|
||||
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import org.springframework.stereotype.Service;
|
||||
import ru.ldeloff.servermonitorbot.model.user.Role;
|
||||
|
||||
@Service
|
||||
@RequiredArgsConstructor
|
||||
public class RoleServiceImpl implements RoleService {
|
||||
@Override
|
||||
public Role findRoleByName(String name) {
|
||||
return Role.valueOf(name);
|
||||
}
|
||||
}
|
|
@ -0,0 +1,25 @@
|
|||
package ru.ldeloff.servermonitorbot.service;
|
||||
|
||||
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import ru.ldeloff.servermonitorbot.model.SshCommand;
|
||||
import ru.ldeloff.servermonitorbot.model.server.SshServer;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
public interface SshService {
|
||||
|
||||
void updateConnection();
|
||||
|
||||
void disconnectAllConnections();
|
||||
|
||||
SendMessage getStatusSessions(SendMessage update);
|
||||
|
||||
List<SshServer> getAllSshServers();
|
||||
|
||||
List<SshServer> getHealthSshServers();
|
||||
|
||||
SshCommand execute(SshCommand sshCommand);
|
||||
|
||||
List<SshCommand> execute(List<SshCommand> sshCommands);
|
||||
}
|
|
@ -0,0 +1,146 @@
|
|||
package ru.ldeloff.servermonitorbot.service;
|
||||
|
||||
import com.jcraft.jsch.ChannelExec;
|
||||
import com.jcraft.jsch.JSchException;
|
||||
import com.jcraft.jsch.Session;
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import ru.ldeloff.servermonitorbot.model.server.ServerStatus;
|
||||
import ru.ldeloff.servermonitorbot.model.server.SshServers;
|
||||
import ru.ldeloff.servermonitorbot.model.SshCommand;
|
||||
import ru.ldeloff.servermonitorbot.model.server.SshServer;
|
||||
import ru.ldeloff.servermonitorbot.repository.SshRepository;
|
||||
|
||||
import java.io.ByteArrayOutputStream;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.Objects;
|
||||
|
||||
@Service
|
||||
@Slf4j
|
||||
@RequiredArgsConstructor
|
||||
public class SshServiceImpl implements SshService {
|
||||
|
||||
private final SshServers sshServers;
|
||||
private final SshRepository sshRepository;
|
||||
private final MessagingService messagingService;
|
||||
|
||||
@Override
|
||||
public void updateConnection() {
|
||||
sshServers.getServers().forEach(sshServer -> {
|
||||
try {
|
||||
sshRepository.connect(sshServer);
|
||||
updateServerStatus(sshServer, ServerStatus.ONLINE);
|
||||
log.info("Успешно подключён к {}", sshServer.getHost());
|
||||
} catch (JSchException e) {
|
||||
if (sshServer.getStatus() == null) {
|
||||
updateServerStatus(sshServer, ServerStatus.OFFLINE);
|
||||
}
|
||||
if (sshServer.getStatus() == ServerStatus.ONLINE) {
|
||||
updateServerStatus(sshServer, ServerStatus.CONNECTION_LOST);
|
||||
}
|
||||
log.warn("Не удалось соединиться с {}: {}", sshServer.getHost(), e.getMessage());
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
@Override
|
||||
public void disconnectAllConnections() {
|
||||
sshServers.getServers().forEach(sshRepository::disconnect);
|
||||
}
|
||||
|
||||
@Override
|
||||
public SendMessage getStatusSessions(SendMessage message) {
|
||||
StringBuilder text = new StringBuilder("Статус соединения (может выполняться долго): \n");
|
||||
sshServers.getServers().forEach(server -> text.append(server.getName())
|
||||
.append(": ")
|
||||
.append(checkStatusServer(server))
|
||||
.append("\n"));
|
||||
message.setText(text.toString());
|
||||
return message;
|
||||
}
|
||||
|
||||
private String checkStatusServer(SshServer server) {
|
||||
if (Objects.isNull(server.getSession())) {
|
||||
return "нет соединения";
|
||||
}
|
||||
|
||||
if (server.getSession().isConnected()) {
|
||||
return "OK";
|
||||
} else {
|
||||
return "потеряно соединение";
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
private boolean checkHealth(SshServer server) {
|
||||
if (Objects.isNull(server.getSession())) {
|
||||
return false;
|
||||
}
|
||||
|
||||
return server.getSession().isConnected();
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<SshServer> getAllSshServers() {
|
||||
return sshServers.getServers();
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<SshServer> getHealthSshServers() {
|
||||
return sshServers.getServers().stream().filter(this::checkHealth).toList();
|
||||
}
|
||||
|
||||
@Override
|
||||
public SshCommand execute(SshCommand sshCommand) {
|
||||
List<SshCommand> sshCommands = new ArrayList<>();
|
||||
sshCommands.add(sshCommand);
|
||||
return execute(sshCommands).get(0);
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<SshCommand> execute(List<SshCommand> sshCommands) {
|
||||
Session session = sshCommands.get(0).getSshServer().getSession(); // подразумевается, что в листе с командами сервер один и тот же
|
||||
ChannelExec channel = null;
|
||||
try {
|
||||
channel = (ChannelExec) session.openChannel("exec");
|
||||
ChannelExec finalChannel = channel;
|
||||
|
||||
sshCommands.forEach(command -> {
|
||||
try {
|
||||
finalChannel.setCommand(command.getCommand());
|
||||
ByteArrayOutputStream responseStream = new ByteArrayOutputStream();
|
||||
finalChannel.setOutputStream(responseStream);
|
||||
finalChannel.connect();
|
||||
while (finalChannel.isConnected()) {
|
||||
Thread.sleep(command.getTimeout());
|
||||
}
|
||||
String responseString = new String(responseStream.toByteArray());
|
||||
command.setResponse(responseString);
|
||||
} catch (JSchException | InterruptedException e) {
|
||||
log.warn(e.getMessage());
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
});
|
||||
} catch (JSchException e) {
|
||||
log.warn(e.getMessage());
|
||||
throw new RuntimeException(e);
|
||||
} finally {
|
||||
if (channel != null) {
|
||||
channel.disconnect();
|
||||
}
|
||||
}
|
||||
return sshCommands;
|
||||
}
|
||||
|
||||
public void updateServerStatus(SshServer server, ServerStatus newStatus) {
|
||||
ServerStatus oldStatus = server.getStatus();
|
||||
if (oldStatus != newStatus) {
|
||||
server.setStatus(newStatus);
|
||||
messagingService.sendMessageToAdmins("Server status changed for " + server.getName() +
|
||||
": " + oldStatus + " -> " + newStatus);
|
||||
}
|
||||
}
|
||||
}
|
|
@ -0,0 +1,16 @@
|
|||
package ru.ldeloff.servermonitorbot.service;
|
||||
|
||||
import ru.ldeloff.servermonitorbot.model.user.User;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
public interface UserService {
|
||||
|
||||
void saveOrUpdateUser(User user);
|
||||
|
||||
User getByTelegramId(Long id);
|
||||
|
||||
List<User> getAllUsers();
|
||||
|
||||
List<User> getAllAdmins();
|
||||
}
|
|
@ -0,0 +1,42 @@
|
|||
package ru.ldeloff.servermonitorbot.service;
|
||||
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.stereotype.Service;
|
||||
import ru.ldeloff.servermonitorbot.model.user.Role;
|
||||
import ru.ldeloff.servermonitorbot.model.user.User;
|
||||
import ru.ldeloff.servermonitorbot.model.user.Users;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
@Slf4j
|
||||
@Service
|
||||
@RequiredArgsConstructor
|
||||
public class UserServiceImpl implements UserService {
|
||||
|
||||
private final Users users;
|
||||
|
||||
@Override
|
||||
public void saveOrUpdateUser(User user) {
|
||||
if (getByTelegramId(user.getTelegramId()) == null) {
|
||||
users.getUsers().add(user);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public User getByTelegramId(Long telegramId) {
|
||||
return users.getUsers().stream()
|
||||
.filter(user -> user.getTelegramId().equals(telegramId))
|
||||
.findFirst()
|
||||
.orElse(null);
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<User> getAllUsers() {return users.getUsers(); }
|
||||
|
||||
@Override
|
||||
public List<User> getAllAdmins() {
|
||||
return users.getUsers().stream().filter(user -> user.getRole().equals(Role.ADMIN)).toList();
|
||||
}
|
||||
|
||||
}
|
|
@ -1,87 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.service.command;
|
||||
|
||||
import lombok.Data;
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import org.telegram.telegrambots.meta.api.objects.Update;
|
||||
import org.telegram.telegrambots.meta.exceptions.TelegramApiException;
|
||||
import ru.ldeloff.servermonitorbot.controller.TelegramBotController;
|
||||
import ru.ldeloff.servermonitorbot.model.User;
|
||||
import ru.ldeloff.servermonitorbot.service.role.RoleService;
|
||||
import ru.ldeloff.servermonitorbot.service.user.UserService;
|
||||
|
||||
@Data
|
||||
@Slf4j
|
||||
@RequiredArgsConstructor
|
||||
public abstract class CommandTemplate {
|
||||
final UserService userService;
|
||||
final RoleService roleService;
|
||||
long expectedRole = 2L;
|
||||
public abstract SendMessage actionForAuth(User user, SendMessage message);
|
||||
|
||||
public SendMessage actionForNotAuth(User user, SendMessage message) {
|
||||
message.setText("У пользователя " + user.getLogin() + " (" + user.getTelegramId()
|
||||
+ ") недостаточно прав для выполнения этой команды");
|
||||
return message;
|
||||
}
|
||||
public void execute(Update update, TelegramBotController bot) {
|
||||
User user = getUser(update);
|
||||
String message = getMessage(update);
|
||||
SendMessage answer = new SendMessage();
|
||||
answer.setChatId(user.getTelegramId());
|
||||
answer.setText(message);
|
||||
if (user.getRole().getId() <= expectedRole) {
|
||||
logSuccess(user, message);
|
||||
sendMessage(actionForAuth(user, answer), bot);
|
||||
} else {
|
||||
logNotAuth(user, message);
|
||||
sendMessage(actionForNotAuth(user, answer), bot);
|
||||
}
|
||||
}
|
||||
protected void logSuccess(User user, String message) {
|
||||
log.info("Получена команда '" + message + "' от " + user.getLogin()
|
||||
+ " (" + user.getTelegramId() + "). OK");
|
||||
}
|
||||
protected void logNotAuth(User user, String message) {
|
||||
log.warn("Получена команда '" + message + "' от " + user.getLogin()
|
||||
+ " (" + user.getTelegramId() + "). Нет прав");
|
||||
}
|
||||
private User getUser(Update update) {
|
||||
long id = -1L;
|
||||
String login = null;
|
||||
if (update.hasMessage()) {
|
||||
id = update.getMessage().getChat().getId();
|
||||
login = update.getMessage().getChat().getUserName();
|
||||
} else if (update.hasCallbackQuery()) {
|
||||
id = update.getCallbackQuery().getMessage().getChat().getId();
|
||||
login = update.getCallbackQuery().getMessage().getChat().getUserName();
|
||||
}
|
||||
User user = userService.getByTelegramId(id);
|
||||
if (user == null) {
|
||||
user = new User();
|
||||
user.setTelegramId(id);
|
||||
user.setRole(roleService.findRoleById(3L));
|
||||
}
|
||||
user.setLogin(login);
|
||||
return user;
|
||||
}
|
||||
|
||||
private String getMessage(Update update) {
|
||||
if (update.hasMessage()) {
|
||||
return update.getMessage().getText();
|
||||
} else if (update.hasCallbackQuery()) {
|
||||
return update.getCallbackQuery().getData();
|
||||
}
|
||||
return "";
|
||||
}
|
||||
|
||||
private void sendMessage(SendMessage message, TelegramBotController bot) {
|
||||
try {
|
||||
bot.execute(message);
|
||||
} catch (TelegramApiException e) {
|
||||
e.printStackTrace();
|
||||
log.warn(e.getMessage());
|
||||
}
|
||||
}
|
||||
}
|
|
@ -1,23 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.service.command;
|
||||
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import ru.ldeloff.servermonitorbot.model.User;
|
||||
import ru.ldeloff.servermonitorbot.service.role.RoleService;
|
||||
import ru.ldeloff.servermonitorbot.service.user.UserService;
|
||||
import ru.ldeloff.servermonitorbot.utils.ui.TelegramBotKeyboard;
|
||||
|
||||
@Service
|
||||
public class FirstUseCommand extends CommandTemplate {
|
||||
final TelegramBotKeyboard telegramBotKeyboard;
|
||||
|
||||
public FirstUseCommand(UserService userService, RoleService roleService, TelegramBotKeyboard telegramBotKeyboard) {
|
||||
super(userService, roleService);
|
||||
this.telegramBotKeyboard = telegramBotKeyboard;
|
||||
}
|
||||
@Override
|
||||
public SendMessage actionForAuth(User user, SendMessage message) {
|
||||
message.setText("Добро пожаловать " + user.getLogin() + "!");
|
||||
return telegramBotKeyboard.uiForm(message);
|
||||
}
|
||||
}
|
|
@ -1,23 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.service.command;
|
||||
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import ru.ldeloff.servermonitorbot.model.User;
|
||||
import ru.ldeloff.servermonitorbot.service.role.RoleService;
|
||||
import ru.ldeloff.servermonitorbot.service.ssh.SshService;
|
||||
import ru.ldeloff.servermonitorbot.service.user.UserService;
|
||||
|
||||
@Service
|
||||
public class GetStatusSessions extends CommandTemplate {
|
||||
final SshService sshService;
|
||||
|
||||
public GetStatusSessions(UserService userService, RoleService roleService, SshService sshService) {
|
||||
super(userService, roleService);
|
||||
this.sshService = sshService;
|
||||
}
|
||||
|
||||
@Override
|
||||
public SendMessage actionForAuth(User user, SendMessage message) {
|
||||
return sshService.getStatusSessions(message);
|
||||
}
|
||||
}
|
|
@ -1,35 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.service.command;
|
||||
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import ru.ldeloff.servermonitorbot.model.User;
|
||||
import ru.ldeloff.servermonitorbot.service.role.RoleService;
|
||||
import ru.ldeloff.servermonitorbot.service.user.UserService;
|
||||
import ru.ldeloff.servermonitorbot.utils.ui.TelegramBotKeyboard;
|
||||
|
||||
@Service
|
||||
@Slf4j
|
||||
public class SwitchToMainMenu extends CommandTemplate {
|
||||
final TelegramBotKeyboard telegramBotKeyboard;
|
||||
|
||||
public SwitchToMainMenu(UserService userService, RoleService roleService, TelegramBotKeyboard telegramBotKeyboard) {
|
||||
super(userService, roleService);
|
||||
this.telegramBotKeyboard = telegramBotKeyboard;
|
||||
}
|
||||
@Override
|
||||
public SendMessage actionForAuth(User user, SendMessage message) {
|
||||
message.setText("Неверная команда. Попробуем сначала.");
|
||||
return telegramBotKeyboard.uiForm(message);
|
||||
}
|
||||
@Override
|
||||
public void logSuccess(User user, String message) {
|
||||
log.warn("Получена несуществующая команда '" + message + "' от " + user.getLogin()
|
||||
+ " (" + user.getTelegramId() + "). OK");
|
||||
}
|
||||
@Override
|
||||
public void logNotAuth(User user, String message) {
|
||||
log.warn("Получена несуществующая команда '" + message + "' от " + user.getLogin()
|
||||
+ " (" + user.getTelegramId() + "). Нет прав");
|
||||
}
|
||||
}
|
|
@ -1,24 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.service.command.cputemp;
|
||||
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import ru.ldeloff.servermonitorbot.model.User;
|
||||
import ru.ldeloff.servermonitorbot.service.command.CommandTemplate;
|
||||
import ru.ldeloff.servermonitorbot.service.role.RoleService;
|
||||
import ru.ldeloff.servermonitorbot.service.user.UserService;
|
||||
import ru.ldeloff.servermonitorbot.utils.ui.uname.ServerListButtons;
|
||||
|
||||
@Service
|
||||
public class CpuTempAggregateCommand extends CommandTemplate {
|
||||
final ServerListButtons serverListButtons;
|
||||
|
||||
public CpuTempAggregateCommand(UserService userService, RoleService roleService, ServerListButtons serverListButtons) {
|
||||
super(userService, roleService);
|
||||
this.serverListButtons = serverListButtons;
|
||||
}
|
||||
|
||||
@Override
|
||||
public SendMessage actionForAuth(User user, SendMessage message) {
|
||||
return serverListButtons.uiForm(message);
|
||||
}
|
||||
}
|
|
@ -1,67 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.service.command.cputemp;
|
||||
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import ru.ldeloff.servermonitorbot.model.Command;
|
||||
import ru.ldeloff.servermonitorbot.model.SshServer;
|
||||
import ru.ldeloff.servermonitorbot.model.User;
|
||||
import ru.ldeloff.servermonitorbot.service.command.CommandTemplate;
|
||||
import ru.ldeloff.servermonitorbot.service.role.RoleService;
|
||||
import ru.ldeloff.servermonitorbot.service.ssh.SshService;
|
||||
import ru.ldeloff.servermonitorbot.service.user.UserService;
|
||||
import ru.ldeloff.servermonitorbot.utils.SshServerUtils;
|
||||
|
||||
import java.util.Arrays;
|
||||
import java.util.List;
|
||||
import java.util.Objects;
|
||||
import java.util.Optional;
|
||||
|
||||
@Service
|
||||
@Slf4j
|
||||
public class CpuTempCommand extends CommandTemplate {
|
||||
final SshService sshService;
|
||||
public CpuTempCommand(UserService userService, RoleService roleService, SshService sshService) {
|
||||
super(userService, roleService);
|
||||
this.sshService = sshService;
|
||||
}
|
||||
|
||||
@Override
|
||||
public SendMessage actionForAuth(User user, SendMessage message) {
|
||||
String [] tags = Arrays.stream(message.getText().split(":"))
|
||||
.map(String::trim)
|
||||
.toArray(String[]::new);
|
||||
|
||||
if (tags[1].equals("all")) {
|
||||
message.setText(hddTempAllHost());
|
||||
} else {
|
||||
message.setText(Objects.requireNonNull(hddTempSpecificHost(tags[1])));
|
||||
}
|
||||
return message;
|
||||
}
|
||||
|
||||
private String hddTempAllHost() {
|
||||
List<SshServer> servers = SshServerUtils.filterGoodServers(sshService.getSshServers());
|
||||
StringBuilder response = new StringBuilder();
|
||||
servers.forEach(server -> response.append(hddTempSpecificHost(server.getName())).append("\n"));
|
||||
return response.toString();
|
||||
}
|
||||
|
||||
private String hddTempSpecificHost(String serverName) {
|
||||
Optional<SshServer> server = sshService.getSshServers()
|
||||
.stream()
|
||||
.filter(x -> x.getName().equals(serverName))
|
||||
.findFirst();
|
||||
if (server.isPresent()) {
|
||||
Command result = sshService.execute(new Command("cat /sys/class/thermal/thermal_zone0/temp", 100, server.get()));
|
||||
return server.get().getName() + ": \n" +
|
||||
(Objects.isNull(result.getResponse()) ? "ошибка при выполнении команды" :
|
||||
String.format("%.1f",
|
||||
Double.parseDouble(result.getResponse().replaceAll("\n", ""))/1000))
|
||||
+ "°C";
|
||||
} else {
|
||||
log.error("Ошибка при выполнении команды 'CPUtemp'. Искомый сервер (" + serverName + ") не найден");
|
||||
return null;
|
||||
}
|
||||
}
|
||||
}
|
|
@ -1,24 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.service.command.hddtemp;
|
||||
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import ru.ldeloff.servermonitorbot.model.User;
|
||||
import ru.ldeloff.servermonitorbot.service.command.CommandTemplate;
|
||||
import ru.ldeloff.servermonitorbot.service.role.RoleService;
|
||||
import ru.ldeloff.servermonitorbot.service.user.UserService;
|
||||
import ru.ldeloff.servermonitorbot.utils.ui.uname.ServerListButtons;
|
||||
|
||||
@Service
|
||||
public class HddTempAggregateCommand extends CommandTemplate {
|
||||
final ServerListButtons serverListButtons;
|
||||
|
||||
public HddTempAggregateCommand(UserService userService, RoleService roleService, ServerListButtons serverListButtons) {
|
||||
super(userService, roleService);
|
||||
this.serverListButtons = serverListButtons;
|
||||
}
|
||||
|
||||
@Override
|
||||
public SendMessage actionForAuth(User user, SendMessage message) {
|
||||
return serverListButtons.uiForm(message);
|
||||
}
|
||||
}
|
|
@ -1,64 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.service.command.hddtemp;
|
||||
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import ru.ldeloff.servermonitorbot.model.Command;
|
||||
import ru.ldeloff.servermonitorbot.model.SshServer;
|
||||
import ru.ldeloff.servermonitorbot.model.User;
|
||||
import ru.ldeloff.servermonitorbot.service.command.CommandTemplate;
|
||||
import ru.ldeloff.servermonitorbot.service.role.RoleService;
|
||||
import ru.ldeloff.servermonitorbot.service.ssh.SshService;
|
||||
import ru.ldeloff.servermonitorbot.service.user.UserService;
|
||||
import ru.ldeloff.servermonitorbot.utils.SshServerUtils;
|
||||
|
||||
import java.util.Arrays;
|
||||
import java.util.List;
|
||||
import java.util.Objects;
|
||||
import java.util.Optional;
|
||||
|
||||
@Service
|
||||
@Slf4j
|
||||
public class HddTempCommand extends CommandTemplate {
|
||||
final SshService sshService;
|
||||
public HddTempCommand(UserService userService, RoleService roleService, SshService sshService) {
|
||||
super(userService, roleService);
|
||||
this.sshService = sshService;
|
||||
}
|
||||
|
||||
@Override
|
||||
public SendMessage actionForAuth(User user, SendMessage message) {
|
||||
String [] tags = Arrays.stream(message.getText().split(":"))
|
||||
.map(String::trim)
|
||||
.toArray(String[]::new);
|
||||
|
||||
if (tags[1].equals("all")) {
|
||||
message.setText(hddTempAllHost());
|
||||
} else {
|
||||
message.setText(Objects.requireNonNull(hddTempSpecificHost(tags[1])));
|
||||
}
|
||||
return message;
|
||||
}
|
||||
|
||||
private String hddTempAllHost() {
|
||||
List<SshServer> servers = SshServerUtils.filterGoodServers(sshService.getSshServers());
|
||||
StringBuilder response = new StringBuilder();
|
||||
servers.forEach(server -> response.append(hddTempSpecificHost(server.getName())).append("\n"));
|
||||
return response.toString();
|
||||
}
|
||||
|
||||
private String hddTempSpecificHost(String serverName) {
|
||||
Optional<SshServer> server = sshService.getSshServers()
|
||||
.stream()
|
||||
.filter(x -> x.getName().equals(serverName))
|
||||
.findFirst();
|
||||
if (server.isPresent()) {
|
||||
Command result = sshService.execute(new Command("hddtemp /dev/sd*", 500, server.get()));
|
||||
return server.get().getName() + ": \n" +
|
||||
(Objects.isNull(result.getResponse()) ? "ошибка при выполнении команды" : result.getResponse());
|
||||
} else {
|
||||
log.error("Ошибка при выполнении команды 'hddtemp'. Искомый сервер (" + serverName + ") не найден");
|
||||
return null;
|
||||
}
|
||||
}
|
||||
}
|
|
@ -1,24 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.service.command.uname;
|
||||
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import ru.ldeloff.servermonitorbot.model.User;
|
||||
import ru.ldeloff.servermonitorbot.service.command.CommandTemplate;
|
||||
import ru.ldeloff.servermonitorbot.service.role.RoleService;
|
||||
import ru.ldeloff.servermonitorbot.service.user.UserService;
|
||||
import ru.ldeloff.servermonitorbot.utils.ui.uname.ServerListButtons;
|
||||
|
||||
@Service
|
||||
public class UnameAggregateCommand extends CommandTemplate {
|
||||
final ServerListButtons serverListButtons;
|
||||
|
||||
public UnameAggregateCommand(UserService userService, RoleService roleService, ServerListButtons serverListButtons) {
|
||||
super(userService, roleService);
|
||||
this.serverListButtons = serverListButtons;
|
||||
}
|
||||
|
||||
@Override
|
||||
public SendMessage actionForAuth(User user, SendMessage message) {
|
||||
return serverListButtons.uiForm(message);
|
||||
}
|
||||
}
|
|
@ -1,63 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.service.command.uname;
|
||||
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import ru.ldeloff.servermonitorbot.model.Command;
|
||||
import ru.ldeloff.servermonitorbot.model.SshServer;
|
||||
import ru.ldeloff.servermonitorbot.model.User;
|
||||
import ru.ldeloff.servermonitorbot.service.command.CommandTemplate;
|
||||
import ru.ldeloff.servermonitorbot.service.role.RoleService;
|
||||
import ru.ldeloff.servermonitorbot.service.ssh.SshService;
|
||||
import ru.ldeloff.servermonitorbot.service.user.UserService;
|
||||
import ru.ldeloff.servermonitorbot.utils.SshServerUtils;
|
||||
|
||||
import java.util.Arrays;
|
||||
import java.util.List;
|
||||
import java.util.Objects;
|
||||
import java.util.Optional;
|
||||
|
||||
@Service
|
||||
@Slf4j
|
||||
public class UnameCommand extends CommandTemplate {
|
||||
final SshService sshService;
|
||||
public UnameCommand(UserService userService, RoleService roleService, SshService sshService) {
|
||||
super(userService, roleService);
|
||||
this.sshService = sshService;
|
||||
}
|
||||
@Override
|
||||
public SendMessage actionForAuth(User user, SendMessage message) {
|
||||
String [] tags = Arrays.stream(message.getText().split(":"))
|
||||
.map(String::trim)
|
||||
.toArray(String[]::new);
|
||||
|
||||
if (tags[1].equals("all")) {
|
||||
message.setText(unameAllHost());
|
||||
} else {
|
||||
message.setText(Objects.requireNonNull(unameSpecificHost(tags[1])));
|
||||
}
|
||||
return message;
|
||||
}
|
||||
|
||||
private String unameAllHost() {
|
||||
List<SshServer> servers = SshServerUtils.filterGoodServers(sshService.getSshServers());
|
||||
StringBuilder response = new StringBuilder();
|
||||
servers.forEach(server -> response.append(unameSpecificHost(server.getName())).append("\n"));
|
||||
return response.toString();
|
||||
}
|
||||
|
||||
private String unameSpecificHost(String serverName) {
|
||||
Optional<SshServer> server = sshService.getSshServers()
|
||||
.stream()
|
||||
.filter(x -> x.getName().equals(serverName))
|
||||
.findFirst();
|
||||
if (server.isPresent()) {
|
||||
Command result = sshService.execute(new Command("uname -a", 100, server.get()));
|
||||
return server.get().getName() + ": " +
|
||||
(Objects.isNull(result.getResponse()) ? "ошибка при выполнении команды" : result.getResponse());
|
||||
} else {
|
||||
log.error("Ошибка при выполнении команды 'uname'. Искомый сервер (" + serverName + ") не найден");
|
||||
return null;
|
||||
}
|
||||
}
|
||||
}
|
|
@ -1,9 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.service.role;
|
||||
|
||||
import ru.ldeloff.servermonitorbot.model.Role;
|
||||
|
||||
public interface RoleService {
|
||||
Role findRoleByName(String name);
|
||||
|
||||
Role findRoleById(Long id);
|
||||
}
|
|
@ -1,28 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.service.role;
|
||||
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import org.springframework.stereotype.Service;
|
||||
import ru.ldeloff.servermonitorbot.model.Role;
|
||||
import ru.ldeloff.servermonitorbot.repository.RoleRepository;
|
||||
|
||||
import java.util.Locale;
|
||||
import java.util.Optional;
|
||||
|
||||
@Service
|
||||
@RequiredArgsConstructor
|
||||
public class RoleServiceImpl implements RoleService {
|
||||
private final RoleRepository roleRepository;
|
||||
|
||||
@Override
|
||||
public Role findRoleByName(String name) {
|
||||
Optional<Role> role = roleRepository.findRoleByName(name.toUpperCase(Locale.ROOT));
|
||||
//noinspection OptionalGetWithoutIsPresent
|
||||
return role.orElseGet(() -> roleRepository.findRoleByName("ANONYMOUS").get());
|
||||
}
|
||||
|
||||
@Override
|
||||
public Role findRoleById(Long id) {
|
||||
Optional<Role> role = roleRepository.findRoleById(id);
|
||||
return role.orElseGet(() -> roleRepository.findRoleByName("ANONYMOUS").get());
|
||||
}
|
||||
}
|
|
@ -1,16 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.service.ssh;
|
||||
|
||||
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import ru.ldeloff.servermonitorbot.model.Command;
|
||||
import ru.ldeloff.servermonitorbot.model.SshServer;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
public interface SshService {
|
||||
SendMessage getStatusSessions(SendMessage update);
|
||||
|
||||
List<SshServer> getSshServers();
|
||||
Command execute(Command command);
|
||||
List<Command> execute(List<Command> commands);
|
||||
}
|
|
@ -1,74 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.service.ssh;
|
||||
|
||||
import com.jcraft.jsch.ChannelExec;
|
||||
import com.jcraft.jsch.JSchException;
|
||||
import com.jcraft.jsch.Session;
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import ru.ldeloff.servermonitorbot.model.Command;
|
||||
import ru.ldeloff.servermonitorbot.model.SshServer;
|
||||
import ru.ldeloff.servermonitorbot.repository.SshRepository;
|
||||
|
||||
import java.io.ByteArrayOutputStream;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
|
||||
@Service
|
||||
@Slf4j
|
||||
@RequiredArgsConstructor
|
||||
public class SshServiceImpl implements SshService {
|
||||
final SshRepository sshRepository;
|
||||
@Override
|
||||
public SendMessage getStatusSessions(SendMessage message) {
|
||||
message.setText(sshRepository.getStatusSessions());
|
||||
return message;
|
||||
}
|
||||
@Override
|
||||
public List<SshServer> getSshServers() {
|
||||
return sshRepository.getSshServers();
|
||||
}
|
||||
|
||||
@Override
|
||||
public Command execute(Command command) {
|
||||
List<Command> commands = new ArrayList<>();
|
||||
commands.add(command);
|
||||
return execute(commands).get(0);
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<Command> execute(List<Command> commands) {
|
||||
Session session = commands.get(0).getSshServer().getSession(); // подразумевается, что в листе с командами сервер один и тот же
|
||||
ChannelExec channel = null;
|
||||
try {
|
||||
channel = (ChannelExec) session.openChannel("exec");
|
||||
ChannelExec finalChannel = channel;
|
||||
|
||||
commands.forEach(command -> {
|
||||
try {
|
||||
finalChannel.setCommand(command.getCommand());
|
||||
ByteArrayOutputStream responseStream = new ByteArrayOutputStream();
|
||||
finalChannel.setOutputStream(responseStream);
|
||||
finalChannel.connect();
|
||||
while (finalChannel.isConnected()) {
|
||||
Thread.sleep(command.getTimeout());
|
||||
}
|
||||
String responseString = new String(responseStream.toByteArray());
|
||||
command.setResponse(responseString);
|
||||
} catch (JSchException | InterruptedException e) {
|
||||
log.warn(e.getMessage());
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
});
|
||||
} catch (JSchException e) {
|
||||
log.warn(e.getMessage());
|
||||
throw new RuntimeException(e);
|
||||
} finally {
|
||||
if (channel != null) {
|
||||
channel.disconnect();
|
||||
}
|
||||
}
|
||||
return commands;
|
||||
}
|
||||
}
|
|
@ -1,9 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.service.user;
|
||||
|
||||
import ru.ldeloff.servermonitorbot.model.User;
|
||||
|
||||
public interface UserService {
|
||||
User saveOrUpdateUser(User user);
|
||||
|
||||
User getByTelegramId(Long id);
|
||||
}
|
|
@ -1,29 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.service.user;
|
||||
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.stereotype.Service;
|
||||
import ru.ldeloff.servermonitorbot.model.User;
|
||||
import ru.ldeloff.servermonitorbot.repository.UserRepository;
|
||||
|
||||
@Service
|
||||
@RequiredArgsConstructor
|
||||
@Slf4j
|
||||
public class UserServiceImpl implements UserService {
|
||||
|
||||
final UserRepository userRepository;
|
||||
|
||||
@Override
|
||||
public User saveOrUpdateUser(User user) {
|
||||
User existUser = userRepository.getByTelegramId(user.getTelegramId());
|
||||
if (existUser != null) {
|
||||
user.setId(existUser.getId());
|
||||
}
|
||||
return userRepository.save(user);
|
||||
}
|
||||
|
||||
@Override
|
||||
public User getByTelegramId(Long id) {
|
||||
return userRepository.getByTelegramId(id);
|
||||
}
|
||||
}
|
|
@ -0,0 +1,51 @@
|
|||
package ru.ldeloff.servermonitorbot.util;
|
||||
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.telegram.telegrambots.meta.api.objects.Update;
|
||||
import ru.ldeloff.servermonitorbot.model.user.Role;
|
||||
import ru.ldeloff.servermonitorbot.model.user.User;
|
||||
import ru.ldeloff.servermonitorbot.service.UserService;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
@Service
|
||||
@RequiredArgsConstructor
|
||||
public class UpdateUtil {
|
||||
|
||||
private final UserService userService;
|
||||
|
||||
public User getUser(Update update) {
|
||||
long id = -1L;
|
||||
String login = null;
|
||||
if (update.hasMessage()) {
|
||||
id = update.getMessage().getChat().getId();
|
||||
login = update.getMessage().getChat().getUserName();
|
||||
} else if (update.hasCallbackQuery()) {
|
||||
id = update.getCallbackQuery().getMessage().getChatId();
|
||||
login = update.getCallbackQuery().getFrom().getUserName();
|
||||
}
|
||||
User user = userService.getByTelegramId(id);
|
||||
if (user == null) {
|
||||
user = new User();
|
||||
user.setTelegramId(id);
|
||||
user.setRole(Role.ANONYMOUS);
|
||||
}
|
||||
user.setLogin(login);
|
||||
return user;
|
||||
}
|
||||
|
||||
public String getMessage(Update update) {
|
||||
if (update.hasMessage()) {
|
||||
return update.getMessage().getText();
|
||||
} else if (update.hasCallbackQuery()) {
|
||||
return update.getCallbackQuery().getData();
|
||||
}
|
||||
return "";
|
||||
}
|
||||
|
||||
public boolean roleChecker(Update update, List<Role> expectedRoles) {
|
||||
User user = getUser(update);
|
||||
return expectedRoles.contains(user.getRole());
|
||||
}
|
||||
}
|
|
@ -1,17 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.utils;
|
||||
|
||||
import ru.ldeloff.servermonitorbot.model.SshServer;
|
||||
|
||||
import java.util.List;
|
||||
import java.util.Objects;
|
||||
|
||||
public class SshServerUtils {
|
||||
public static List<SshServer> filterGoodServers(List<SshServer> sshServers) {
|
||||
List<SshServer> goodSshServers = sshServers.stream()
|
||||
.filter(Objects::nonNull)
|
||||
.filter(server -> Objects.nonNull(server.getSession()))
|
||||
.filter(server -> server.getSession().isConnected())
|
||||
.toList();
|
||||
return goodSshServers;
|
||||
}
|
||||
}
|
|
@ -1,40 +0,0 @@
|
|||
package ru.ldeloff.servermonitorbot.utils.ui;
|
||||
|
||||
import org.springframework.stereotype.Component;
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import org.telegram.telegrambots.meta.api.objects.replykeyboard.ReplyKeyboardMarkup;
|
||||
import org.telegram.telegrambots.meta.api.objects.replykeyboard.buttons.KeyboardButton;
|
||||
import org.telegram.telegrambots.meta.api.objects.replykeyboard.buttons.KeyboardRow;
|
||||
|
||||
|
||||
import java.util.ArrayList;
|
||||
|
||||
@Component
|
||||
public class TelegramBotKeyboard implements UiFormer {
|
||||
private final String STATUS = "Статус";
|
||||
private final String UNAME = "uname";
|
||||
private final String CPU_TEMP = "CPU.temp";
|
||||
private final String HDD_TEMP = "HDD.temp";
|
||||
|
||||
|
||||
@Override
|
||||
public SendMessage uiForm(SendMessage message) {
|
||||
ReplyKeyboardMarkup replyKeyboardMarkup = new ReplyKeyboardMarkup();
|
||||
replyKeyboardMarkup.setResizeKeyboard(true);
|
||||
replyKeyboardMarkup.setOneTimeKeyboard(false);
|
||||
|
||||
ArrayList<KeyboardRow> keyboardRows = new ArrayList<>();
|
||||
|
||||
KeyboardRow keyboardRow = new KeyboardRow();
|
||||
keyboardRows.add(keyboardRow);
|
||||
|
||||
keyboardRow.add(new KeyboardButton(STATUS));
|
||||
keyboardRow.add(new KeyboardButton(UNAME));
|
||||
keyboardRow.add(new KeyboardButton(CPU_TEMP));
|
||||
keyboardRow.add(new KeyboardButton(HDD_TEMP));
|
||||
|
||||
replyKeyboardMarkup.setKeyboard(keyboardRows);
|
||||
message.setReplyMarkup(replyKeyboardMarkup);
|
||||
return message;
|
||||
}
|
||||
}
|
|
@ -0,0 +1,33 @@
|
|||
package ru.ldeloff.servermonitorbot.view;
|
||||
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import ru.ldeloff.servermonitorbot.commands.CommandTemplate;
|
||||
import ru.ldeloff.servermonitorbot.service.MessagingService;
|
||||
import ru.ldeloff.servermonitorbot.util.UpdateUtil;
|
||||
import ru.ldeloff.servermonitorbot.model.user.User;
|
||||
import ru.ldeloff.servermonitorbot.service.SshService;
|
||||
|
||||
@Service
|
||||
public class FirstUseCommand extends CommandTemplate {
|
||||
|
||||
private final TelegramBotKeyboard telegramBotKeyboard;
|
||||
|
||||
public FirstUseCommand(UpdateUtil updateUtil, ServerListButtons serverListButtons, SshService sshService, TelegramBotKeyboard telegramBotKeyboard, MessagingService messagingService) {
|
||||
super(updateUtil, serverListButtons, sshService, messagingService);
|
||||
this.telegramBotKeyboard = telegramBotKeyboard;
|
||||
setName("/start");
|
||||
}
|
||||
|
||||
|
||||
@Override
|
||||
public SendMessage executeAggregate(User user, SendMessage message) {
|
||||
message.setText("Добро пожаловать " + user.getLogin() + "!");
|
||||
return telegramBotKeyboard.uiForm(message);
|
||||
}
|
||||
|
||||
@Override
|
||||
public String executeCommandSpecificHost(String serverName) {
|
||||
return null;
|
||||
}
|
||||
}
|
|
@ -1,12 +1,11 @@
|
|||
package ru.ldeloff.servermonitorbot.utils.ui.uname;
|
||||
package ru.ldeloff.servermonitorbot.view;
|
||||
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import org.springframework.stereotype.Component;
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import org.telegram.telegrambots.meta.api.objects.replykeyboard.InlineKeyboardMarkup;
|
||||
import org.telegram.telegrambots.meta.api.objects.replykeyboard.buttons.InlineKeyboardButton;
|
||||
import ru.ldeloff.servermonitorbot.service.ssh.SshService;
|
||||
import ru.ldeloff.servermonitorbot.utils.ui.UiFormer;
|
||||
import ru.ldeloff.servermonitorbot.service.SshService;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
|
@ -21,10 +20,9 @@ public class ServerListButtons implements UiFormer {
|
|||
@Override
|
||||
public SendMessage uiForm(SendMessage message) {
|
||||
List<List<InlineKeyboardButton>> keyboard = new ArrayList<>();
|
||||
|
||||
InlineKeyboardMarkup inlineKeyboardMarkup = new InlineKeyboardMarkup();
|
||||
|
||||
sshService.getSshServers().forEach(sshServer -> {
|
||||
sshService.getAllSshServers().forEach(sshServer -> {
|
||||
if (Objects.nonNull(sshServer.getSession())) {
|
||||
if (sshServer.getSession().isConnected()) {
|
||||
List<InlineKeyboardButton> keyboardRow = new ArrayList<>();
|
|
@ -0,0 +1,37 @@
|
|||
package ru.ldeloff.servermonitorbot.view;
|
||||
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import ru.ldeloff.servermonitorbot.commands.CommandTemplate;
|
||||
import ru.ldeloff.servermonitorbot.service.MessagingService;
|
||||
import ru.ldeloff.servermonitorbot.util.UpdateUtil;
|
||||
import ru.ldeloff.servermonitorbot.model.user.User;
|
||||
import ru.ldeloff.servermonitorbot.service.SshService;
|
||||
|
||||
@Service
|
||||
@Slf4j
|
||||
public class SwitchToMainMenu extends CommandTemplate {
|
||||
|
||||
private final TelegramBotKeyboard telegramBotKeyboard;
|
||||
|
||||
public SwitchToMainMenu(UpdateUtil updateUtil, ServerListButtons serverListButtons, SshService sshService, TelegramBotKeyboard telegramBotKeyboard, MessagingService messagingService) {
|
||||
super(updateUtil, serverListButtons, sshService, messagingService);
|
||||
this.telegramBotKeyboard = telegramBotKeyboard;
|
||||
setName("Not found");
|
||||
}
|
||||
|
||||
@Override
|
||||
public String executeCommandSpecificHost(String serverName) {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public SendMessage executeAggregate(User user, SendMessage message) {
|
||||
SendMessage answer = new SendMessage();
|
||||
answer.setChatId(user.getTelegramId());
|
||||
answer.setText("Неверная команда. Попробуем сначала.");
|
||||
return telegramBotKeyboard.uiForm(answer);
|
||||
}
|
||||
|
||||
}
|
|
@ -0,0 +1,70 @@
|
|||
package ru.ldeloff.servermonitorbot.view;
|
||||
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import org.springframework.context.ApplicationContext;
|
||||
import org.springframework.stereotype.Component;
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
import org.telegram.telegrambots.meta.api.objects.replykeyboard.ReplyKeyboardMarkup;
|
||||
import org.telegram.telegrambots.meta.api.objects.replykeyboard.buttons.KeyboardButton;
|
||||
import org.telegram.telegrambots.meta.api.objects.replykeyboard.buttons.KeyboardRow;
|
||||
import ru.ldeloff.servermonitorbot.model.user.Role;
|
||||
import ru.ldeloff.servermonitorbot.model.user.User;
|
||||
import ru.ldeloff.servermonitorbot.service.UserService;
|
||||
import ru.ldeloff.servermonitorbot.commands.CommandTemplate;
|
||||
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
import static ru.ldeloff.servermonitorbot.model.user.Role.ADMIN;
|
||||
import static ru.ldeloff.servermonitorbot.model.user.Role.USER;
|
||||
|
||||
@Component
|
||||
@RequiredArgsConstructor
|
||||
public class TelegramBotKeyboard implements UiFormer {
|
||||
|
||||
private final UserService userService;
|
||||
private final ApplicationContext applicationContext;
|
||||
|
||||
@Override
|
||||
public SendMessage uiForm(SendMessage message) {
|
||||
ReplyKeyboardMarkup replyKeyboardMarkup = new ReplyKeyboardMarkup();
|
||||
replyKeyboardMarkup.setResizeKeyboard(true);
|
||||
replyKeyboardMarkup.setOneTimeKeyboard(false);
|
||||
|
||||
String chatId = message.getChatId();
|
||||
User user = userService.getByTelegramId(Long.parseLong(chatId));
|
||||
|
||||
ArrayList<KeyboardRow> keyboardRows = formKeyboard(user);
|
||||
replyKeyboardMarkup.setKeyboard(keyboardRows);
|
||||
message.setReplyMarkup(replyKeyboardMarkup);
|
||||
return message;
|
||||
}
|
||||
|
||||
private ArrayList<KeyboardRow> formKeyboard(User user) {
|
||||
Role role = user.getRole();
|
||||
Map<String, CommandTemplate> commands = applicationContext.getBeansOfType(CommandTemplate.class);
|
||||
commands.remove("firstUseCommand");
|
||||
commands.remove("switchToMainMenu");
|
||||
|
||||
int buttonOnLine = 3;
|
||||
|
||||
ArrayList<KeyboardRow> keyboardRows = new ArrayList<>();
|
||||
KeyboardRow keyboardRow = new KeyboardRow();
|
||||
|
||||
for (Map.Entry<String, CommandTemplate> entry : commands.entrySet()) {
|
||||
if (entry.getValue().getExpectedRole().contains(role)) {
|
||||
keyboardRow.add(new KeyboardButton(entry.getValue().getName()));
|
||||
}
|
||||
if (keyboardRow.size()>=buttonOnLine) {
|
||||
keyboardRows.add(keyboardRow);
|
||||
keyboardRow = new KeyboardRow();
|
||||
}
|
||||
}
|
||||
if (!keyboardRow.isEmpty()) {
|
||||
keyboardRows.add(keyboardRow);
|
||||
}
|
||||
return keyboardRows;
|
||||
}
|
||||
}
|
|
@ -1,4 +1,4 @@
|
|||
package ru.ldeloff.servermonitorbot.utils.ui;
|
||||
package ru.ldeloff.servermonitorbot.view;
|
||||
|
||||
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
|
||||
|
|
@ -1,21 +1,6 @@
|
|||
spring:
|
||||
application:
|
||||
name: ServerMonitorBot
|
||||
datasource:
|
||||
url: jdbc:postgresql://localhost:5432/servermonitorbot
|
||||
username: servermonitorbot
|
||||
password: servermonitorbot
|
||||
jpa:
|
||||
hibernate:
|
||||
ddl-auto: none
|
||||
liquibase:
|
||||
enabled: true
|
||||
change-log: classpath:db/scripts/changelog-master.xml
|
||||
url: jdbc:postgresql://localhost:5432/servermonitorbot
|
||||
user: servermonitorbot
|
||||
password: servermonitorbot
|
||||
liquibase-schema: "liquibase"
|
||||
default-schema: "servermonitorbot"
|
||||
bot:
|
||||
name: "ServerMonitorBot"
|
||||
token: "token"
|
||||
|
@ -23,6 +8,8 @@ bot:
|
|||
- telegramId: 123456789
|
||||
role: admin
|
||||
ssh:
|
||||
checkConnectionDelay: 300000
|
||||
timeout: 5000
|
||||
servers:
|
||||
-
|
||||
name: "Server Name"
|
||||
|
@ -30,6 +17,12 @@ ssh:
|
|||
port: 22
|
||||
user: "user"
|
||||
password: "pass"
|
||||
-
|
||||
name: "Server Name 2"
|
||||
host: "url 2"
|
||||
port: 22
|
||||
user: "user 2"
|
||||
password: "pass 2"
|
||||
|
||||
|
||||
|
||||
|
|
|
@ -1,10 +0,0 @@
|
|||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<databaseChangeLog
|
||||
xmlns="http://www.liquibase.org/xml/ns/dbchangelog"
|
||||
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
|
||||
xsi:schemaLocation="http://www.liquibase.org/xml/ns/dbchangelog http://www.liquibase.org/xml/ns/dbchangelog/dbchangelog-3.1.xsd">
|
||||
|
||||
<include file="release_0_0_1/changelog.xml" relativeToChangelogFile="true"/>
|
||||
|
||||
|
||||
</databaseChangeLog>
|
|
@ -1,16 +0,0 @@
|
|||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<databaseChangeLog
|
||||
xmlns="http://www.liquibase.org/xml/ns/dbchangelog"
|
||||
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
|
||||
xsi:schemaLocation="http://www.liquibase.org/xml/ns/dbchangelog http://www.liquibase.org/xml/ns/dbchangelog/dbchangelog-3.1.xsd">
|
||||
|
||||
<include file="scripts/01_create_table_users.sql" relativeToChangelogFile="true"/>
|
||||
<include file="scripts/02_create_table_roles.sql" relativeToChangelogFile="true"/>
|
||||
<include file="scripts/03_add_FK_users_to_roles.sql" relativeToChangelogFile="true"/>
|
||||
<include file="scripts/04_add_roles.sql" relativeToChangelogFile="true"/>
|
||||
<include file="scripts/05_add_anonymous_role.sql" relativeToChangelogFile="true"/>
|
||||
<include file="scripts/06_refactor_users_table.sql" relativeToChangelogFile="true"/>
|
||||
<include file="scripts/07_refactor_roles_table.sql" relativeToChangelogFile="true"/>
|
||||
<include file="scripts/08_add_roles.sql" relativeToChangelogFile="true"/>
|
||||
|
||||
</databaseChangeLog>
|
|
@ -1,17 +0,0 @@
|
|||
--liquibase formatted sql
|
||||
|
||||
--changeset L_DelOff:create_table_users rollbackSplitStatements:true
|
||||
--comment: Создание таблицы пользователей
|
||||
CREATE TABLE users
|
||||
(
|
||||
id SERIAL PRIMARY KEY,
|
||||
login VARCHAR(128),
|
||||
id_role integer
|
||||
);
|
||||
|
||||
COMMENT ON TABLE users IS 'Пользователи';
|
||||
COMMENT ON COLUMN users.id IS 'Идентификатор пользователя';
|
||||
COMMENT ON COLUMN users.login IS 'Логин пользователя';
|
||||
COMMENT ON COLUMN users.id_role IS 'Ссылка на роль пользователя';
|
||||
|
||||
--rollback DROP TABLE users;
|
|
@ -1,15 +0,0 @@
|
|||
--liquibase formatted sql
|
||||
|
||||
--changeset L_DelOff:create_table_roles rollbackSplitStatements:true
|
||||
--comment: Создание таблицы ролей
|
||||
CREATE TABLE roles
|
||||
(
|
||||
id SERIAL PRIMARY KEY,
|
||||
role VARCHAR(128)
|
||||
);
|
||||
|
||||
COMMENT ON TABLE roles IS 'Пользователи';
|
||||
COMMENT ON COLUMN roles.id IS 'Идентификатор пользователя';
|
||||
COMMENT ON COLUMN roles.role IS 'Логин пользователя';
|
||||
|
||||
--rollback DROP TABLE roles;
|
|
@ -1,2 +0,0 @@
|
|||
ALTER TABLE users
|
||||
ADD COLUMN role_id INT NOT NULL REFERENCES roles(id);
|
|
@ -1,2 +0,0 @@
|
|||
INSERT INTO roles (id, role) VALUES (1, 'ADMIN');
|
||||
INSERT INTO roles (id, role) VALUES (2, 'USER');
|
|
@ -1 +0,0 @@
|
|||
INSERT INTO roles (id, role) VALUES (3, 'ANONYMOUS');
|
|
@ -1,9 +0,0 @@
|
|||
ALTER TABLE users
|
||||
DROP COLUMN login;
|
||||
|
||||
ALTER TABLE users
|
||||
ADD COLUMN telegram_id INTEGER UNIQUE;
|
||||
COMMENT ON COLUMN users.telegram_id IS 'Telegram ID пользователя';
|
||||
|
||||
ALTER TABLE users
|
||||
DROP COLUMN id_role;
|
|
@ -1,6 +0,0 @@
|
|||
ALTER TABLE roles
|
||||
DROP COLUMN role;
|
||||
|
||||
ALTER TABLE roles
|
||||
ADD COLUMN name VARCHAR(128);
|
||||
COMMENT ON COLUMN roles.name IS 'Имя роли';
|
|
@ -1,4 +0,0 @@
|
|||
TRUNCATE TABLE roles CASCADE;
|
||||
INSERT INTO roles (id, name) VALUES (1, 'ADMIN');
|
||||
INSERT INTO roles (id, name) VALUES (2, 'USER');
|
||||
INSERT INTO roles (id, name) VALUES (3, 'ANONYMOUS');
|
Loading…
Reference in New Issue