Compare commits
9 Commits
feature/ad
...
feature/ta
| Author | SHA1 | Date | |
|---|---|---|---|
| ad7ee9a96c | |||
| e1ea86b36b | |||
| 148c511b9e | |||
| f08d88355a | |||
| 4974796f0a | |||
| 047502916a | |||
| c174f36848 | |||
| 619996b913 | |||
| 0cfdd37702 |
14
pom.xml
14
pom.xml
@@ -47,8 +47,18 @@
|
||||
<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>
|
||||
|
||||
@@ -1,5 +1,8 @@
|
||||
package ru.ldeloff.servermonitorbot.config;
|
||||
|
||||
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;
|
||||
@@ -8,14 +11,8 @@ import java.util.List;
|
||||
|
||||
@Configuration
|
||||
@ConfigurationProperties(prefix = "ssh")
|
||||
@Getter
|
||||
@Setter
|
||||
public class SshConfig {
|
||||
private List<SshServer> servers;
|
||||
|
||||
public List<SshServer> getServers() {
|
||||
return servers;
|
||||
}
|
||||
|
||||
public void setServers(List<SshServer> servers) {
|
||||
this.servers = servers;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,17 @@
|
||||
package ru.ldeloff.servermonitorbot.config;
|
||||
|
||||
import lombok.Getter;
|
||||
import lombok.Setter;
|
||||
import org.springframework.boot.context.properties.ConfigurationProperties;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
import ru.ldeloff.servermonitorbot.model.dto.InitUserDto;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
@Configuration
|
||||
@ConfigurationProperties(prefix = "bot")
|
||||
@Getter
|
||||
@Setter
|
||||
public class UserConfig {
|
||||
private List<InitUserDto> users;
|
||||
}
|
||||
@@ -7,9 +7,13 @@ 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.Role;
|
||||
import ru.ldeloff.servermonitorbot.model.TelegramBot;
|
||||
import ru.ldeloff.servermonitorbot.model.User;
|
||||
import ru.ldeloff.servermonitorbot.repository.SshRepository;
|
||||
import ru.ldeloff.servermonitorbot.service.TelegramBotService;
|
||||
import ru.ldeloff.servermonitorbot.service.role.RoleService;
|
||||
import ru.ldeloff.servermonitorbot.service.user.UserService;
|
||||
import ru.ldeloff.servermonitorbot.utils.ui.TelegramBotKeyboard;
|
||||
import ru.ldeloff.servermonitorbot.utils.ui.uname.UnameChatButtonAggregate;
|
||||
|
||||
@@ -20,31 +24,68 @@ public class TelegramBotController extends TelegramLongPollingBot {
|
||||
|
||||
final TelegramBot telegramBot;
|
||||
final SshRepository sshRepository;
|
||||
final TelegramBotKeyboard telegramBotKeyboard;
|
||||
final UnameChatButtonAggregate unameChatButtonAggregate;
|
||||
final TelegramBotService telegramBotService;
|
||||
final UserService userService;
|
||||
final RoleService roleService;
|
||||
|
||||
@Override
|
||||
public void onUpdateReceived(Update update) {
|
||||
Role role = getRole(update);
|
||||
if (update.hasMessage()) {
|
||||
if (update.getMessage().hasText()) {
|
||||
String messageText = update.getMessage().getText();
|
||||
switch (messageText) {
|
||||
case "/start" -> {
|
||||
log.info("Получена команда /start");
|
||||
telegramBotService.firstUse(update, this);
|
||||
if (role.getId() < 3) {
|
||||
log.info("Получена команда '" + messageText
|
||||
+ "' от " + update.getMessage().getChat().getUserName()
|
||||
+ " (" + update.getMessage().getChat().getId() + "). OK");
|
||||
telegramBotService.firstUse(update, this);
|
||||
} else {
|
||||
log.warn("Получена команда '" + messageText
|
||||
+ "' от " + update.getMessage().getChat().getUserName()
|
||||
+ " (" + update.getMessage().getChat().getId() + "). Нет прав");
|
||||
telegramBotService.notAuthorization(update, this);
|
||||
}
|
||||
}
|
||||
case "Статус" -> {
|
||||
log.info("Получена команда 'Статус'");
|
||||
telegramBotService.getStatusSessions(update, this);
|
||||
if (role.getId() < 3) {
|
||||
log.info("Получена команда '" + messageText
|
||||
+ "' от " + update.getMessage().getChat().getUserName()
|
||||
+ " (" + update.getMessage().getChat().getId() + "). OK");
|
||||
telegramBotService.getStatusSessions(update, this);
|
||||
} else {
|
||||
log.warn("Получена команда '" + messageText
|
||||
+ "' от " + update.getMessage().getChat().getUserName()
|
||||
+ " (" + update.getMessage().getChat().getId() + "). Нет прав");
|
||||
telegramBotService.notAuthorization(update, this);
|
||||
}
|
||||
}
|
||||
case "uname" -> {
|
||||
log.info("Получена команда 'uname'");
|
||||
telegramBotService.sendUnameAggregate(update, this);
|
||||
if (role.getId() < 3) {
|
||||
log.info("Получена команда '" + messageText
|
||||
+ "' от " + update.getMessage().getChat().getUserName()
|
||||
+ " (" + update.getMessage().getChat().getId() + "). OK");
|
||||
telegramBotService.sendUnameAggregate(update, this);
|
||||
} else {
|
||||
log.warn("Получена команда '" + messageText
|
||||
+ "' от " + update.getMessage().getChat().getUserName()
|
||||
+ " (" + update.getMessage().getChat().getId() + "). Нет прав");
|
||||
telegramBotService.notAuthorization(update, this);
|
||||
}
|
||||
}
|
||||
default -> {
|
||||
log.warn("Неизвестная команда:" + messageText);
|
||||
telegramBotService.switchToMainMenu(update, this);
|
||||
if (role.getId() < 3) {
|
||||
log.warn("Неизвестная команда '" + messageText
|
||||
+ "' от " + update.getMessage().getChat().getUserName()
|
||||
+ " (" + update.getMessage().getChat().getId() + "). ОК");
|
||||
telegramBotService.switchToMainMenu(update, this);
|
||||
} else {
|
||||
log.warn("Неизвестная команда '" + messageText
|
||||
+ "' от " + update.getMessage().getChat().getUserName()
|
||||
+ " (" + update.getMessage().getChat().getId() + "). Нет прав");
|
||||
telegramBotService.notAuthorization(update, this);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -54,17 +95,44 @@ public class TelegramBotController extends TelegramLongPollingBot {
|
||||
if (tags.length > 1) {
|
||||
switch (tags[0]) {
|
||||
case "uname":
|
||||
log.info("Получена команда '" + messageText + "'");
|
||||
telegramBotService.uname(update, this);
|
||||
if (role.getId() < 3) {
|
||||
log.info("Получена команда '" + messageText
|
||||
+ "' от " + update.getCallbackQuery().getMessage().getChat().getUserName()
|
||||
+ " (" + update.getCallbackQuery().getMessage().getChat().getId() + "). ОК");
|
||||
telegramBotService.uname(update, this);
|
||||
} else {
|
||||
log.info("Получена команда '" + messageText
|
||||
+ "' от " + update.getCallbackQuery().getMessage().getChat().getUserName()
|
||||
+ " (" + update.getCallbackQuery().getMessage().getChat().getId() + "). Нет прав");
|
||||
telegramBotService.notAuthorization(update, this);
|
||||
}
|
||||
break;
|
||||
default:
|
||||
log.warn("Неизвестная команда:" + messageText);
|
||||
telegramBotService.switchToMainMenu(update, this);
|
||||
if (role.getId() < 3) {
|
||||
log.warn("Неизвестная команда '" + messageText
|
||||
+ "' от " + update.getCallbackQuery().getMessage().getChat().getUserName()
|
||||
+ " (" + update.getCallbackQuery().getMessage().getChat().getId() + "). ОК");
|
||||
telegramBotService.switchToMainMenu(update, this);
|
||||
} else {
|
||||
log.warn("Неизвестная команда '" + messageText
|
||||
+ "' от " + update.getCallbackQuery().getMessage().getChat().getUserName()
|
||||
+ " (" + update.getCallbackQuery().getMessage().getChat().getId() + "). Нет прав");
|
||||
telegramBotService.notAuthorization(update, this);
|
||||
}
|
||||
break;
|
||||
}
|
||||
} else {
|
||||
log.warn("Неверный формат команды:" + messageText);
|
||||
if (role.getId() < 3) {
|
||||
log.warn("Неверный формат команды '" + messageText
|
||||
+ "' от " + update.getCallbackQuery().getMessage().getChat().getUserName()
|
||||
+ " (" + update.getCallbackQuery().getMessage().getChat().getId() + "). ОК");
|
||||
telegramBotService.switchToMainMenu(update, this);
|
||||
} else {
|
||||
log.warn("Неверный формат команды '" + messageText
|
||||
+ "' от " + update.getCallbackQuery().getMessage().getChat().getUserName()
|
||||
+ " (" + update.getCallbackQuery().getMessage().getChat().getId() + "). Нет прав");
|
||||
telegramBotService.notAuthorization(update, this);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -84,4 +152,19 @@ public class TelegramBotController extends TelegramLongPollingBot {
|
||||
super.onClosing();
|
||||
sshRepository.disconnectSessions();
|
||||
}
|
||||
|
||||
private Role getRole(Update update) {
|
||||
Long id = null;
|
||||
if (update.hasMessage()) {
|
||||
id = update.getMessage().getChat().getId();
|
||||
} else if (update.hasCallbackQuery()) {
|
||||
id = update.getCallbackQuery().getMessage().getChat().getId();
|
||||
}
|
||||
User user = userService.getByTelegramId(id);
|
||||
if (user == null) {
|
||||
return roleService.findRoleById(3L);
|
||||
} else {
|
||||
return user.getRole();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
34
src/main/java/ru/ldeloff/servermonitorbot/init/AddUsers.java
Normal file
34
src/main/java/ru/ldeloff/servermonitorbot/init/AddUsers.java
Normal file
@@ -0,0 +1,34 @@
|
||||
package ru.ldeloff.servermonitorbot.init;
|
||||
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.boot.ApplicationArguments;
|
||||
import org.springframework.boot.ApplicationRunner;
|
||||
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;
|
||||
|
||||
@Component
|
||||
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;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void run(ApplicationArguments args) throws Exception {
|
||||
initUsers.forEach(initUserDto -> {
|
||||
User user = userService.saveOrUpdateUser(userMapper.dtoToUser(initUserDto));
|
||||
});
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,22 @@
|
||||
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;
|
||||
}
|
||||
}
|
||||
25
src/main/java/ru/ldeloff/servermonitorbot/model/Command.java
Normal file
25
src/main/java/ru/ldeloff/servermonitorbot/model/Command.java
Normal 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;
|
||||
}
|
||||
}
|
||||
17
src/main/java/ru/ldeloff/servermonitorbot/model/Role.java
Normal file
17
src/main/java/ru/ldeloff/servermonitorbot/model/Role.java
Normal 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 = "name")
|
||||
private String name;
|
||||
}
|
||||
21
src/main/java/ru/ldeloff/servermonitorbot/model/User.java
Normal file
21
src/main/java/ru/ldeloff/servermonitorbot/model/User.java
Normal 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 = "telegram_id")
|
||||
private Long telegramId;
|
||||
|
||||
@ManyToOne
|
||||
@JoinColumn(name = "role_id")
|
||||
private Role role;
|
||||
}
|
||||
@@ -0,0 +1,14 @@
|
||||
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,13 @@
|
||||
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);
|
||||
}
|
||||
@@ -12,6 +12,7 @@ import ru.ldeloff.servermonitorbot.model.SshServer;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.Objects;
|
||||
|
||||
@Slf4j
|
||||
@Repository
|
||||
@@ -65,7 +66,7 @@ public class SshRepositoryImpl implements SshRepository {
|
||||
sshServers.forEach(server -> {
|
||||
text.append(server.getName())
|
||||
.append(": ")
|
||||
.append(server.getSession().isConnected() ? "OK" : "отключён")
|
||||
.append(checkStatusServer(server))
|
||||
.append("\n");
|
||||
});
|
||||
return String.valueOf(text);
|
||||
@@ -75,4 +76,13 @@ public class SshRepositoryImpl implements SshRepository {
|
||||
public List<SshServer> getSshServers() {
|
||||
return sshServers;
|
||||
}
|
||||
|
||||
private String checkStatusServer(SshServer server) {
|
||||
if (Objects.isNull(server.getSession())) {
|
||||
return "нет соединения";
|
||||
} else {
|
||||
return server.getSession().isConnected() ? "OK" : "отключён";
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@@ -0,0 +1,10 @@
|
||||
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);
|
||||
}
|
||||
@@ -9,5 +9,6 @@ public interface TelegramBotService {
|
||||
void getStatusSessions(Update update, TelegramBotController bot);
|
||||
void sendUnameAggregate(Update update, TelegramBotController bot);
|
||||
void uname(Update update, TelegramBotController bot);
|
||||
void firstUse(Update update, TelegramBotController telegramBotController);
|
||||
void firstUse(Update update, TelegramBotController bot);
|
||||
void notAuthorization(Update update, TelegramBotController bot);
|
||||
}
|
||||
|
||||
@@ -27,10 +27,32 @@ public class TelegramBotServiceImpl implements TelegramBotService {
|
||||
SendMessage message = new SendMessage();
|
||||
long chatId = 0L;
|
||||
chatId = update.getMessage().getChatId();
|
||||
message.setText("Добро пожалвать " + update.getMessage().getChat().getUserName() + "!");
|
||||
message.setText("Добро пожаловать " + update.getMessage().getChat().getUserName() + "!");
|
||||
message.setChatId(chatId);
|
||||
sendMessage(telegramBotKeyboard.uiForm(message), bot);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void notAuthorization(Update update, TelegramBotController bot) {
|
||||
long id = 0L;
|
||||
String login = null;
|
||||
long chatId = 0L;
|
||||
if (update.hasMessage()) {
|
||||
id = update.getMessage().getChat().getId();
|
||||
login = update.getMessage().getChat().getUserName();
|
||||
chatId = update.getMessage().getChatId();
|
||||
} else if (update.hasCallbackQuery()) {
|
||||
id = update.getCallbackQuery().getMessage().getChat().getId();
|
||||
login = update.getCallbackQuery().getMessage().getChat().getUserName();
|
||||
chatId = update.getCallbackQuery().getMessage().getChatId();
|
||||
}
|
||||
SendMessage message = new SendMessage();
|
||||
message.setText("У пользователя " + login
|
||||
+ " (" + id + ") недостаточно прав для выполнения этой команды");
|
||||
message.setChatId(chatId);
|
||||
sendMessage(message, bot);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void switchToMainMenu(Update update, TelegramBotController bot) {
|
||||
SendMessage message = new SendMessage();
|
||||
|
||||
@@ -0,0 +1,9 @@
|
||||
package ru.ldeloff.servermonitorbot.service.role;
|
||||
|
||||
import ru.ldeloff.servermonitorbot.model.Role;
|
||||
|
||||
public interface RoleService {
|
||||
Role findRoleByName(String name);
|
||||
|
||||
Role findRoleById(Long id);
|
||||
}
|
||||
@@ -0,0 +1,28 @@
|
||||
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());
|
||||
}
|
||||
}
|
||||
@@ -3,6 +3,7 @@ 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;
|
||||
@@ -10,7 +11,6 @@ import java.util.List;
|
||||
public interface SshService {
|
||||
SendMessage getStatusSessions(Update update);
|
||||
List<SshServer> getSshServers();
|
||||
|
||||
String execute(String s, SshServer sshServer);
|
||||
String execute(String s, SshServer sshServer, int timeout);
|
||||
Command execute(Command command);
|
||||
List<Command> execute(List<Command> commands);
|
||||
}
|
||||
|
||||
@@ -8,10 +8,12 @@ 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
|
||||
@@ -33,34 +35,44 @@ public class SshServiceImpl implements SshService {
|
||||
}
|
||||
|
||||
@Override
|
||||
public String execute(String s, SshServer sshServer) {
|
||||
return execute(s, sshServer, 100);
|
||||
public Command execute(Command command) {
|
||||
List<Command> commands = new ArrayList<>();
|
||||
commands.add(command);
|
||||
return execute(commands).get(0);
|
||||
}
|
||||
|
||||
@Override
|
||||
public String execute(String s, SshServer sshServer, int timeout) {
|
||||
Session session = sshServer.getSession();
|
||||
public List<Command> execute(List<Command> commands) {
|
||||
Session session = commands.get(0).getSshServer().getSession(); // подразумевается, что в листе с командами сервер один и тот же
|
||||
ChannelExec channel = null;
|
||||
try {
|
||||
channel = (ChannelExec) session.openChannel("exec");
|
||||
channel.setCommand(s);
|
||||
ByteArrayOutputStream responseStream = new ByteArrayOutputStream();
|
||||
channel.setOutputStream(responseStream);
|
||||
channel.connect();
|
||||
ChannelExec finalChannel = channel;
|
||||
|
||||
while (channel.isConnected()) {
|
||||
Thread.sleep(timeout);
|
||||
}
|
||||
|
||||
String responseString = new String(responseStream.toByteArray());
|
||||
return responseString;
|
||||
} catch (JSchException | InterruptedException e) {
|
||||
log.error(e.getMessage());
|
||||
return null;
|
||||
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;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -5,8 +5,10 @@ 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;
|
||||
@@ -49,7 +51,7 @@ public class UnameServiceImpl implements UnameService {
|
||||
}
|
||||
|
||||
private String unameAllHost() {
|
||||
List<SshServer> servers = sshService.getSshServers();
|
||||
List<SshServer> servers = SshServerUtils.filterGoodServers(sshService.getSshServers());
|
||||
StringBuilder response = new StringBuilder();
|
||||
servers.forEach(server -> {
|
||||
response.append(unameSpecificHost(server.getName())).append("\n");
|
||||
@@ -63,9 +65,9 @@ public class UnameServiceImpl implements UnameService {
|
||||
.filter(x -> x.getName().equals(serverName))
|
||||
.findFirst();
|
||||
if (server.isPresent()) {
|
||||
String result = sshService.execute("uname -a", server.get());
|
||||
return server.get().getName() + ": " + (Objects.isNull(result) ? "ошибка при выполнении команды" : result);
|
||||
|
||||
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;
|
||||
|
||||
@@ -0,0 +1,9 @@
|
||||
package ru.ldeloff.servermonitorbot.service.user;
|
||||
|
||||
import ru.ldeloff.servermonitorbot.model.User;
|
||||
|
||||
public interface UserService {
|
||||
User saveOrUpdateUser(User user);
|
||||
|
||||
User getByTelegramId(Long id);
|
||||
}
|
||||
@@ -0,0 +1,29 @@
|
||||
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,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;
|
||||
}
|
||||
}
|
||||
@@ -17,8 +17,8 @@ public class TelegramBotKeyboard implements UiFormer {
|
||||
@Override
|
||||
public SendMessage uiForm(SendMessage message) {
|
||||
ReplyKeyboardMarkup replyKeyboardMarkup = new ReplyKeyboardMarkup();
|
||||
replyKeyboardMarkup.setResizeKeyboard(true); //подгоняем размер
|
||||
replyKeyboardMarkup.setOneTimeKeyboard(false); //скрываем после использования
|
||||
replyKeyboardMarkup.setResizeKeyboard(true);
|
||||
replyKeyboardMarkup.setOneTimeKeyboard(false);
|
||||
|
||||
ArrayList<KeyboardRow> keyboardRows = new ArrayList<>();
|
||||
|
||||
|
||||
@@ -12,6 +12,7 @@ import ru.ldeloff.servermonitorbot.utils.ui.UiFormer;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.Objects;
|
||||
|
||||
@Component
|
||||
@RequiredArgsConstructor
|
||||
@@ -26,15 +27,19 @@ public class UnameChatButtonAggregate implements UiFormer {
|
||||
InlineKeyboardMarkup inlineKeyboardMarkup = new InlineKeyboardMarkup();
|
||||
|
||||
sshService.getSshServers().forEach(sshServer -> {
|
||||
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 (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 (sshService.getSshServers().size() > 1) {
|
||||
if (keyboard.size() > 1) {
|
||||
List<InlineKeyboardButton> keyboardRow = new ArrayList<>();
|
||||
InlineKeyboardButton inlineKeyboardButton = new InlineKeyboardButton();
|
||||
inlineKeyboardButton.setText("Все сервера");
|
||||
|
||||
@@ -1,6 +1,27 @@
|
||||
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"
|
||||
users:
|
||||
- telegramId: 123456789
|
||||
role: admin
|
||||
ssh:
|
||||
servers:
|
||||
-
|
||||
|
||||
10
src/main/resources/db/scripts/changelog-master.xml
Normal file
10
src/main/resources/db/scripts/changelog-master.xml
Normal 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>
|
||||
16
src/main/resources/db/scripts/release_0_0_1/changelog.xml
Normal file
16
src/main/resources/db/scripts/release_0_0_1/changelog.xml
Normal file
@@ -0,0 +1,16 @@
|
||||
<?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>
|
||||
@@ -0,0 +1,17 @@
|
||||
--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;
|
||||
@@ -0,0 +1,15 @@
|
||||
--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;
|
||||
@@ -0,0 +1,2 @@
|
||||
ALTER TABLE users
|
||||
ADD COLUMN role_id INT NOT NULL REFERENCES roles(id);
|
||||
@@ -0,0 +1,2 @@
|
||||
INSERT INTO roles (id, role) VALUES (1, 'ADMIN');
|
||||
INSERT INTO roles (id, role) VALUES (2, 'USER');
|
||||
@@ -0,0 +1 @@
|
||||
INSERT INTO roles (id, role) VALUES (3, 'ANONYMOUS');
|
||||
@@ -0,0 +1,9 @@
|
||||
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;
|
||||
@@ -0,0 +1,6 @@
|
||||
ALTER TABLE roles
|
||||
DROP COLUMN role;
|
||||
|
||||
ALTER TABLE roles
|
||||
ADD COLUMN name VARCHAR(128);
|
||||
COMMENT ON COLUMN roles.name IS 'Имя роли';
|
||||
@@ -0,0 +1,4 @@
|
||||
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');
|
||||
Reference in New Issue
Block a user