14 Commits

Author SHA1 Message Date
4974796f0a Добавил сущности role и user 2023-08-10 22:21:30 +03:00
047502916a Добавил spring jpa и liquibase 2023-08-09 22:18:02 +03:00
c174f36848 Merge pull request 'Фикс багов' (#5) from bugfix/fix-issues into master
Reviewed-on: L_DelOff/ServerMonitorBot#5
2023-08-08 22:36:44 +03:00
619996b913 Фикс багов 2023-08-08 22:36:06 +03:00
0cfdd37702 Merge pull request 'feature/add-uname-command' (#4) from feature/add-uname-command into master
Reviewed-on: L_DelOff/ServerMonitorBot#4
2023-08-08 21:02:32 +03:00
e6fac12e15 Если сервер один, то не показываю кнопку "все сервера" 2023-08-08 21:01:10 +03:00
b6147e9e91 Реализована команда uname 2023-08-07 23:49:02 +03:00
fcd4929e0c Рефактор 2023-08-06 23:01:58 +03:00
7b18162d30 Добавлена кнопка uname, пока не полный функционал 2023-08-06 00:55:44 +03:00
26d9f83917 Merge pull request 'Добавлена кнопка статус соединения' (#3) from feature/add-status-button into master
Reviewed-on: L_DelOff/ServerMonitorBot#3
2023-08-05 23:01:51 +03:00
6d56771999 Добавлена кнопка статус соединения 2023-08-05 22:58:19 +03:00
91efa22e12 Merge pull request 'При запуске идёт подключение ко всем серверам' (#2) from feature/add_ssh_connect into master
Reviewed-on: L_DelOff/ServerMonitorBot#2
2023-08-05 19:35:01 +03:00
61aaba99a6 При запуске идёт подключение ко всем серверам 2023-08-05 19:34:06 +03:00
687c99b80d Merge pull request 'Настойка телеграм бота' (#1) from feature/TelegramBotExample into master
Reviewed-on: L_DelOff/ServerMonitorBot#1
2023-08-05 17:25:37 +03:00
31 changed files with 819 additions and 44 deletions

19
pom.xml
View File

@@ -42,8 +42,23 @@
<artifactId>telegrambots</artifactId> <artifactId>telegrambots</artifactId>
<version>6.7.0</version> <version>6.7.0</version>
</dependency> </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> </dependencies>
<build> <build>

View File

@@ -2,8 +2,11 @@ package ru.ldeloff.servermonitorbot;
import org.springframework.boot.SpringApplication; import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication; import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import ru.ldeloff.servermonitorbot.config.SshConfig;
@SpringBootApplication @SpringBootApplication
@EnableConfigurationProperties(SshConfig.class)
public class ServerMonitorBotApplication { public class ServerMonitorBotApplication {
public static void main(String[] args) { public static void main(String[] args) {

View File

@@ -0,0 +1,21 @@
package ru.ldeloff.servermonitorbot.config;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Configuration;
import ru.ldeloff.servermonitorbot.model.SshServer;
import java.util.List;
@Configuration
@ConfigurationProperties(prefix = "ssh")
public class SshConfig {
private List<SshServer> servers;
public List<SshServer> getServers() {
return servers;
}
public void setServers(List<SshServer> servers) {
this.servers = servers;
}
}

View File

@@ -0,0 +1,92 @@
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.methods.send.SendMessage;
import org.telegram.telegrambots.meta.api.objects.Update;
import org.telegram.telegrambots.meta.exceptions.TelegramApiException;
import ru.ldeloff.servermonitorbot.model.TelegramBot;
import ru.ldeloff.servermonitorbot.repository.SshRepository;
import ru.ldeloff.servermonitorbot.service.TelegramBotService;
import ru.ldeloff.servermonitorbot.utils.ui.TelegramBotKeyboard;
import ru.ldeloff.servermonitorbot.utils.ui.uname.UnameChatButtonAggregate;
@Service
@Slf4j
@RequiredArgsConstructor
public class TelegramBotController extends TelegramLongPollingBot {
final TelegramBot telegramBot;
final SshRepository sshRepository;
final TelegramBotService telegramBotService;
@Override
public void onUpdateReceived(Update update) {
if (update.hasMessage()) {
if (update.getMessage().hasText()) {
String messageText = update.getMessage().getText();
switch (messageText) {
case "/start" -> {
log.info("Получена команда '" + messageText
+ "' от " + update.getMessage().getChat().getUserName());
telegramBotService.firstUse(update, this);
}
case "Статус" -> {
log.info("Получена команда '" + messageText
+ "' от " + update.getMessage().getChat().getUserName());
telegramBotService.getStatusSessions(update, this);
}
case "uname" -> {
log.info("Получена команда '" + messageText
+ "' от " + update.getMessage().getChat().getUserName());
telegramBotService.sendUnameAggregate(update, this);
}
default -> {
log.warn("Неизвестная команда '" + messageText
+ "' от " + update.getMessage().getChat().getUserName());
telegramBotService.switchToMainMenu(update, this);
}
}
}
} else if (update.hasCallbackQuery()) {
String messageText = update.getCallbackQuery().getData();
String [] tags = messageText.split(":");
if (tags.length > 1) {
switch (tags[0]) {
case "uname":
log.info("Получена команда '" + messageText
+ "' от " + update.getCallbackQuery().getMessage().getChat().getUserName());
telegramBotService.uname(update, this);
break;
default:
log.warn("Неизвестная команда '" + messageText
+ "' от " + update.getCallbackQuery().getMessage().getChat().getUserName());
telegramBotService.switchToMainMenu(update, this);
break;
}
} else {
log.warn("Неверный формат команды '" + messageText
+ "' от " + update.getCallbackQuery().getMessage().getChat().getUserName());
telegramBotService.switchToMainMenu(update, this);
}
}
}
@Override
public String getBotUsername() {
return telegramBot.getBotUsername();
}
@Override
public String getBotToken() {
return telegramBot.getBotToken();
}
@Override
public void onClosing() {
super.onClosing();
sshRepository.disconnectSessions();
}
}

View File

@@ -0,0 +1,20 @@
package ru.ldeloff.servermonitorbot.init;
import org.springframework.boot.ApplicationArguments;
import org.springframework.boot.ApplicationRunner;
import org.springframework.stereotype.Component;
import ru.ldeloff.servermonitorbot.repository.SshRepository;
@Component
public class SshConnect implements ApplicationRunner {
final SshRepository sshRepository;
public SshConnect(SshRepository sshRepository) {
this.sshRepository = sshRepository;
}
@Override
public void run(ApplicationArguments args) throws Exception {
sshRepository.connectToAllServer();
}
}

View File

@@ -1,4 +1,4 @@
package ru.ldeloff.servermonitorbot; package ru.ldeloff.servermonitorbot.init;
import lombok.NoArgsConstructor; import lombok.NoArgsConstructor;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
@@ -9,16 +9,16 @@ import org.springframework.stereotype.Component;
import org.telegram.telegrambots.meta.TelegramBotsApi; import org.telegram.telegrambots.meta.TelegramBotsApi;
import org.telegram.telegrambots.meta.exceptions.TelegramApiException; import org.telegram.telegrambots.meta.exceptions.TelegramApiException;
import org.telegram.telegrambots.updatesreceivers.DefaultBotSession; import org.telegram.telegrambots.updatesreceivers.DefaultBotSession;
import ru.ldeloff.servermonitorbot.model.TelegramBot; import ru.ldeloff.servermonitorbot.controller.TelegramBotController;
@Slf4j @Slf4j
@Component @Component
@NoArgsConstructor @NoArgsConstructor
public class StartBot implements ApplicationRunner { public class StartBot implements ApplicationRunner {
private TelegramBot telegramBot; private TelegramBotController telegramBot;
@Autowired @Autowired
public StartBot(TelegramBot telegramBot) { public StartBot(TelegramBotController telegramBot) {
this.telegramBot = telegramBot; this.telegramBot = telegramBot;
} }

View File

@@ -0,0 +1,25 @@
package ru.ldeloff.servermonitorbot.model;
import jakarta.validation.constraints.Min;
import jakarta.validation.constraints.NotNull;
import lombok.Getter;
import lombok.Setter;
@Getter
@Setter
public class Command {
@NotNull
private String command;
@NotNull
@Min(100)
private int timeout;
@NotNull
private SshServer sshServer;
private String response;
public Command(String command, int timeout, SshServer sshServer) {
this.command = command;
this.timeout = timeout;
this.sshServer = sshServer;
}
}

View File

@@ -0,0 +1,17 @@
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 = "role")
private String role;
}

View File

@@ -0,0 +1,24 @@
package ru.ldeloff.servermonitorbot.model;
import com.jcraft.jsch.Session;
import jakarta.validation.constraints.NotNull;
import lombok.Getter;
import lombok.Setter;
import org.springframework.stereotype.Component;
@Setter
@Getter
@Component
public class SshServer {
@NotNull
private String name;
@NotNull
private String host;
@NotNull
private int port;
@NotNull
private String user;
@NotNull
private String password;
private Session session;
}

View File

@@ -1,47 +1,17 @@
package ru.ldeloff.servermonitorbot.model; package ru.ldeloff.servermonitorbot.model;
import lombok.AllArgsConstructor; import lombok.Getter;
import lombok.NoArgsConstructor;
import org.springframework.beans.factory.annotation.Value; import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
import org.telegram.telegrambots.bots.TelegramLongPollingBot;
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
import org.telegram.telegrambots.meta.api.objects.Update;
import org.telegram.telegrambots.meta.exceptions.TelegramApiException;
@Getter
@Component @Component
@NoArgsConstructor public class TelegramBot{
@AllArgsConstructor private final String botUsername;
public class TelegramBot extends TelegramLongPollingBot { private final String botToken;
@Value("${bot.name}")
private String botUsername;
@Value("${bot.token}") public TelegramBot(@Value("${bot.name}") String botUsername, @Value("${bot.token}") String botToken) {
private String botToken; this.botUsername = botUsername;
this.botToken = botToken;
@Override
public void onUpdateReceived(Update update) {
// We check if the update has a message and the message has text
if (update.hasMessage() && update.getMessage().hasText()) {
SendMessage message = new SendMessage(); // Create a SendMessage object with mandatory fields
message.setChatId(update.getMessage().getChatId().toString());
message.setText(update.getMessage().getText());
try {
execute(message); // Call method to send the message
} catch (TelegramApiException e) {
e.printStackTrace();
}
}
}
@Override
public String getBotUsername() {
return botUsername;
}
@Override
public String getBotToken() {
return botToken;
} }
} }

View File

@@ -0,0 +1,21 @@
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 = "login")
private Long login;
@ManyToOne
@JoinColumn(name = "id_role")
private Role role;
}

View File

@@ -0,0 +1,16 @@
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;
public interface SshRepository {
void connectToAllServer();
Session connectToServer(SshServer sshServer) throws JSchException;
void disconnectSessions();
String getStatusSessions();
List<SshServer> getSshServers();
}

View File

@@ -0,0 +1,88 @@
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 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 java.util.ArrayList;
import java.util.List;
import java.util.Objects;
@Slf4j
@Repository
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<>();
@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 {
Session session = new JSch().getSession(sshServer.getUser(),
sshServer.getHost(),
sshServer.getPort());
session.setPassword(sshServer.getPassword());
session.setConfig("StrictHostKeyChecking", "no");
session.connect();
return 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" : "отключён";
}
}
}

View File

@@ -0,0 +1,13 @@
package ru.ldeloff.servermonitorbot.service;
import org.telegram.telegrambots.meta.api.objects.Update;
import ru.ldeloff.servermonitorbot.controller.TelegramBotController;
public interface TelegramBotService {
void switchToMainMenu(Update update, TelegramBotController bot);
void getStatusSessions(Update update, TelegramBotController bot);
void sendUnameAggregate(Update update, TelegramBotController bot);
void uname(Update update, TelegramBotController bot);
void firstUse(Update update, TelegramBotController telegramBotController);
}

View File

@@ -0,0 +1,69 @@
package ru.ldeloff.servermonitorbot.service;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
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.repository.SshRepository;
import ru.ldeloff.servermonitorbot.service.ssh.SshService;
import ru.ldeloff.servermonitorbot.service.uname.UnameService;
import ru.ldeloff.servermonitorbot.utils.ui.TelegramBotKeyboard;
import ru.ldeloff.servermonitorbot.utils.ui.uname.UnameChatButtonAggregate;
@Slf4j
@Service
@RequiredArgsConstructor
public class TelegramBotServiceImpl implements TelegramBotService {
final TelegramBotKeyboard telegramBotKeyboard;
final UnameChatButtonAggregate unameChatButtonAggregate;
final UnameService unameService;
final SshService sshService;
@Override
public void firstUse(Update update, TelegramBotController bot) {
SendMessage message = new SendMessage();
long chatId = 0L;
chatId = update.getMessage().getChatId();
message.setText("Добро пожаловать " + update.getMessage().getChat().getUserName() + "!");
message.setChatId(chatId);
sendMessage(telegramBotKeyboard.uiForm(message), bot);
}
@Override
public void switchToMainMenu(Update update, TelegramBotController bot) {
SendMessage message = new SendMessage();
long chatId = 0L;
try {
chatId = update.getMessage().getChatId();
} catch (Exception e) {
chatId = update.getCallbackQuery().getMessage().getChatId();
}
message.setText("Неверная команда и или формат. Попробуем сначала.");
message.setChatId(chatId);
sendMessage(telegramBotKeyboard.uiForm(message), bot);
}
@Override
public void getStatusSessions(Update update, TelegramBotController bot) {
sendMessage(sshService.getStatusSessions(update), bot);
}
@Override
public void sendUnameAggregate(Update update, TelegramBotController bot) {
sendMessage(unameService.sendUnameAggregate(update), bot);
}
@Override
public void uname(Update update, TelegramBotController bot) {
sendMessage(unameService.uname(update), bot);
}
private void sendMessage(SendMessage message, TelegramBotController bot) {
try {
bot.execute(message);
} catch (TelegramApiException e) {
e.printStackTrace();
log.warn(e.getMessage());
}
}
}

View File

@@ -0,0 +1,16 @@
package ru.ldeloff.servermonitorbot.service.ssh;
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
import org.telegram.telegrambots.meta.api.objects.Update;
import ru.ldeloff.servermonitorbot.model.Command;
import ru.ldeloff.servermonitorbot.model.SshServer;
import java.util.List;
public interface SshService {
SendMessage getStatusSessions(Update update);
List<SshServer> getSshServers();
Command execute(Command command);
List<Command> execute(List<Command> commands);
}

View File

@@ -0,0 +1,78 @@
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 org.telegram.telegrambots.meta.api.objects.Update;
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(Update update) {
SendMessage message = new SendMessage();
long chatId = update.getMessage().getChatId();
message.setText(sshRepository.getStatusSessions());
message.setChatId(chatId);
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;
}
}

View File

@@ -0,0 +1,10 @@
package ru.ldeloff.servermonitorbot.service.uname;
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
import org.telegram.telegrambots.meta.api.objects.Update;
public interface UnameService {
SendMessage sendUnameAggregate(Update update);
SendMessage uname(Update update);
}

View File

@@ -0,0 +1,76 @@
package ru.ldeloff.servermonitorbot.service.uname;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
import org.telegram.telegrambots.meta.api.objects.Update;
import ru.ldeloff.servermonitorbot.model.Command;
import ru.ldeloff.servermonitorbot.model.SshServer;
import ru.ldeloff.servermonitorbot.service.ssh.SshService;
import ru.ldeloff.servermonitorbot.utils.SshServerUtils;
import ru.ldeloff.servermonitorbot.utils.ui.uname.UnameChatButtonAggregate;
import java.util.Arrays;
import java.util.List;
import java.util.Objects;
import java.util.Optional;
@Service
@RequiredArgsConstructor
@Slf4j
public class UnameServiceImpl implements UnameService {
final SshService sshService;
final UnameChatButtonAggregate unameChatButtonAggregate;
@Override
public SendMessage sendUnameAggregate(Update update) {
SendMessage message = new SendMessage();
long chatId = update.getMessage().getChatId();
message.setChatId(chatId);
return unameChatButtonAggregate.uiForm(message);
}
@Override
public SendMessage uname(Update update) {
String messageText = update.getCallbackQuery().getData();
String [] tags = Arrays.stream(messageText.split(":"))
.map(String::trim)
.toArray(String[]::new);
String result;
if (tags[1].equals("all")) {
result = unameAllHost();
} else {
result = unameSpecificHost(tags[1]);
}
SendMessage message = new SendMessage();
long chatId = update.getCallbackQuery().getMessage().getChatId();
message.setChatId(chatId);
message.setText(result);
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;
}
}
}

View File

@@ -0,0 +1,17 @@
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;
}
}

View File

@@ -0,0 +1,35 @@
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";
@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));
replyKeyboardMarkup.setKeyboard(keyboardRows);
message.setReplyMarkup(replyKeyboardMarkup);
return message;
}
}

View File

@@ -0,0 +1,7 @@
package ru.ldeloff.servermonitorbot.utils.ui;
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
public interface UiFormer {
SendMessage uiForm(SendMessage message);
}

View File

@@ -0,0 +1,57 @@
package ru.ldeloff.servermonitorbot.utils.ui.uname;
import lombok.RequiredArgsConstructor;
import org.springframework.beans.factory.annotation.Autowired;
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.repository.SshRepository;
import ru.ldeloff.servermonitorbot.service.ssh.SshService;
import ru.ldeloff.servermonitorbot.utils.ui.UiFormer;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
@Component
@RequiredArgsConstructor
public class UnameChatButtonAggregate implements UiFormer {
private final SshService sshService;
@Override
public SendMessage uiForm(SendMessage message) {
List<List<InlineKeyboardButton>> keyboard = new ArrayList<>();
InlineKeyboardMarkup inlineKeyboardMarkup = new InlineKeyboardMarkup();
sshService.getSshServers().forEach(sshServer -> {
if (Objects.nonNull(sshServer.getSession())) {
if (sshServer.getSession().isConnected()) {
List<InlineKeyboardButton> keyboardRow = new ArrayList<>();
InlineKeyboardButton inlineKeyboardButton = new InlineKeyboardButton();
inlineKeyboardButton.setText(sshServer.getName());
inlineKeyboardButton.setCallbackData("uname: " + sshServer.getName());
keyboardRow.add(inlineKeyboardButton);
keyboard.add(keyboardRow);
}
}
});
if (keyboard.size() > 1) {
List<InlineKeyboardButton> keyboardRow = new ArrayList<>();
InlineKeyboardButton inlineKeyboardButton = new InlineKeyboardButton();
inlineKeyboardButton.setText("Все сервера");
inlineKeyboardButton.setCallbackData("uname: all");
keyboardRow.add(inlineKeyboardButton);
keyboard.add(keyboardRow);
}
inlineKeyboardMarkup.setKeyboard(keyboard);
message.setReplyMarkup(inlineKeyboardMarkup);
message.setText("Выполнить uname для");
return message;
}
}

View File

@@ -1,5 +1,32 @@
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: bot:
name: "ServerMonitorBot" name: "ServerMonitorBot"
token: "token" token: "token"
ssh:
servers:
-
name: "Server Name"
host: "url"
port: 22
user: "user"
password: "pass"

View File

@@ -0,0 +1,10 @@
<?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>

View File

@@ -0,0 +1,12 @@
<?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"/>
</databaseChangeLog>

View File

@@ -0,0 +1,17 @@
--liquibase formatted sql
--changeset L_DelOff:create_table_users rollbackSplitStatements:true
--comment: Создание таблицы пользователей
CREATE TABLE users
(
id integer 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;

View File

@@ -0,0 +1,15 @@
--liquibase formatted sql
--changeset L_DelOff:create_table_roles rollbackSplitStatements:true
--comment: Создание таблицы ролей
CREATE TABLE roles
(
id integer 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;

View File

@@ -0,0 +1,2 @@
ALTER TABLE users
ADD COLUMN role_id INT NOT NULL REFERENCES roles(id);

View File

@@ -0,0 +1,2 @@
INSERT INTO roles (id, role) VALUES (1, 'ADMIN');
INSERT INTO roles (id, role) VALUES (2, 'USER');