рефактор

feature/task-38-roles
L_DelOff 2024-11-16 16:35:00 +03:00
parent a03a6abdfc
commit 93e04a7e56
58 changed files with 677 additions and 675 deletions

View File

@ -3,10 +3,12 @@ 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 org.springframework.boot.context.properties.EnableConfigurationProperties;
import ru.ldeloff.servermonitorbot.config.SshConfig; import org.springframework.scheduling.annotation.EnableScheduling;
import ru.ldeloff.servermonitorbot.config.SshServerList;
@SpringBootApplication @SpringBootApplication
@EnableConfigurationProperties(SshConfig.class) @EnableScheduling
@EnableConfigurationProperties(SshServerList.class)
public class ServerMonitorBotApplication { public class ServerMonitorBotApplication {
public static void main(String[] args) { public static void main(String[] args) {

View File

@ -1,11 +1,10 @@
package ru.ldeloff.servermonitorbot.config; package ru.ldeloff.servermonitorbot.config;
import lombok.Getter; import lombok.Getter;
import lombok.RequiredArgsConstructor;
import lombok.Setter; import lombok.Setter;
import org.springframework.boot.context.properties.ConfigurationProperties; import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.Configuration;
import ru.ldeloff.servermonitorbot.model.SshServer; import ru.ldeloff.servermonitorbot.internal.data.model.SshServer;
import java.util.List; import java.util.List;
@ -13,6 +12,8 @@ import java.util.List;
@ConfigurationProperties(prefix = "ssh") @ConfigurationProperties(prefix = "ssh")
@Getter @Getter
@Setter @Setter
public class SshConfig { public class SshServerList {
private List<SshServer> servers; private List<SshServer> servers;
} }

View File

@ -2,16 +2,37 @@ package ru.ldeloff.servermonitorbot.config;
import lombok.Getter; import lombok.Getter;
import lombok.Setter; import lombok.Setter;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.context.properties.ConfigurationProperties; import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Configuration; import org.springframework.stereotype.Component;
import ru.ldeloff.servermonitorbot.model.dto.InitUserDto; import ru.ldeloff.servermonitorbot.internal.data.model.User;
import ru.ldeloff.servermonitorbot.internal.data.service.RoleService;
import java.util.List; import java.util.List;
@Configuration @Component
@ConfigurationProperties(prefix = "bot") @ConfigurationProperties(prefix = "bot")
@Getter @Getter
@Setter @Setter
public class UserConfig { 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();
}
} }

View File

@ -1,7 +0,0 @@
package ru.ldeloff.servermonitorbot.init;
public class AddCommands {
public static void fillCommandTable() {
}
}

View File

@ -1,34 +1,24 @@
package ru.ldeloff.servermonitorbot.init; 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.ApplicationArguments;
import org.springframework.boot.ApplicationRunner; import org.springframework.boot.ApplicationRunner;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
import ru.ldeloff.servermonitorbot.config.UserConfig; import ru.ldeloff.servermonitorbot.config.UserConfig;
import ru.ldeloff.servermonitorbot.mapper.UserMapper; import ru.ldeloff.servermonitorbot.internal.data.service.UserService;
import ru.ldeloff.servermonitorbot.model.User;
import ru.ldeloff.servermonitorbot.model.dto.InitUserDto;
import ru.ldeloff.servermonitorbot.service.user.UserService;
import java.util.List;
@Slf4j
@Component @Component
@RequiredArgsConstructor
public class AddUsers implements ApplicationRunner { public class AddUsers implements ApplicationRunner {
final List<InitUserDto> initUsers;
final UserService userService;
final UserMapper userMapper;
@Autowired final UserConfig userConfig;
public AddUsers(UserConfig userConfig, UserService userService, UserMapper userMapper) { final UserService userService;
this.initUsers = userConfig.getUsers();
this.userService = userService;
this.userMapper = userMapper;
}
@Override @Override
public void run(ApplicationArguments args) throws Exception { public void run(ApplicationArguments args) {
initUsers.forEach(initUserDto -> { userConfig.getUsers().forEach(userService::saveOrUpdateUser);
User user = userService.saveOrUpdateUser(userMapper.dtoToUser(initUserDto)); log.info("Загружены и обновлены пользователи");
});
} }
} }

View File

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

View File

@ -1,4 +1,4 @@
package ru.ldeloff.servermonitorbot.init; package ru.ldeloff.servermonitorbot.integration.in;
import lombok.NoArgsConstructor; import lombok.NoArgsConstructor;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
@ -10,13 +10,11 @@ import org.telegram.telegrambots.meta.TelegramBotsApi;
import org.telegram.telegrambots.meta.api.methods.send.SendMessage; import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
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.controller.TelegramBotController; import ru.ldeloff.servermonitorbot.internal.data.model.User;
import ru.ldeloff.servermonitorbot.model.User; import ru.ldeloff.servermonitorbot.internal.template.CommandTemplate;
import ru.ldeloff.servermonitorbot.service.command.CommandTemplate; import ru.ldeloff.servermonitorbot.internal.data.service.UserService;
import ru.ldeloff.servermonitorbot.service.user.UserService; import ru.ldeloff.servermonitorbot.internal.ui.TelegramBotKeyboard;
import ru.ldeloff.servermonitorbot.utils.ui.TelegramBotKeyboard;
import static ru.ldeloff.servermonitorbot.init.AddCommands.fillCommandTable;
@Slf4j @Slf4j
@Component @Component
@ -35,7 +33,6 @@ public class StartBot implements ApplicationRunner {
@Override @Override
public void run(ApplicationArguments args) { public void run(ApplicationArguments args) {
try { try {
fillCommandTable();
TelegramBotsApi botsApi = new TelegramBotsApi(DefaultBotSession.class); TelegramBotsApi botsApi = new TelegramBotsApi(DefaultBotSession.class);
botsApi.registerBot(telegramBot); botsApi.registerBot(telegramBot);
userService.getAllUsers().forEach(this::SendInitMessage); userService.getAllUsers().forEach(this::SendInitMessage);

View File

@ -1,24 +1,26 @@
package ru.ldeloff.servermonitorbot.controller; package ru.ldeloff.servermonitorbot.integration.in;
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import org.telegram.telegrambots.bots.TelegramLongPollingBot; import org.telegram.telegrambots.bots.TelegramLongPollingBot;
import org.telegram.telegrambots.meta.api.objects.Message; import org.telegram.telegrambots.meta.api.objects.Message;
import org.telegram.telegrambots.meta.api.objects.Update; import org.telegram.telegrambots.meta.api.objects.Update;
import ru.ldeloff.servermonitorbot.model.TelegramBot; import ru.ldeloff.servermonitorbot.integration.in.model.TelegramBot;
import ru.ldeloff.servermonitorbot.repository.SshRepository; import ru.ldeloff.servermonitorbot.integration.out.service.SshService;
import ru.ldeloff.servermonitorbot.service.command.CommandTemplate; import ru.ldeloff.servermonitorbot.internal.template.CommandTemplate;
import ru.ldeloff.servermonitorbot.service.command.SwitchToMainMenu; import ru.ldeloff.servermonitorbot.internal.mainmenu.service.SwitchToMainMenu;
import java.util.Map; import java.util.Map;
@Service @Service
@RequiredArgsConstructor @RequiredArgsConstructor
public class TelegramBotController extends TelegramLongPollingBot { public class TelegramBotController extends TelegramLongPollingBot {
private final Map<String, CommandTemplate> commands; private final Map<String, CommandTemplate> commands;
final TelegramBot telegramBot; private final TelegramBot telegramBot;
final SshRepository sshRepository; private final SwitchToMainMenu switchToMainMenu;
final SwitchToMainMenu switchToMainMenu; private final SshService sshService;
@Override @Override
public void onUpdateReceived(Update update) { public void onUpdateReceived(Update update) {
boolean result = false; boolean result = false;
@ -37,17 +39,20 @@ public class TelegramBotController extends TelegramLongPollingBot {
switchToMainMenu.execute(update, this); switchToMainMenu.execute(update, this);
} }
} }
@Override @Override
public String getBotUsername() { public String getBotUsername() {
return telegramBot.getBotUsername(); return telegramBot.getBotUsername();
} }
@Override @Override
public String getBotToken() { public String getBotToken() {
return telegramBot.getBotToken(); return telegramBot.getBotToken();
} }
@Override @Override
public void onClosing() { public void onClosing() {
super.onClosing(); super.onClosing();
sshRepository.disconnectSessions(); sshService.disconnectAllConnections();
} }
} }

View File

@ -1,4 +1,4 @@
package ru.ldeloff.servermonitorbot.model; package ru.ldeloff.servermonitorbot.integration.in.model;
import lombok.Getter; import lombok.Getter;
import org.springframework.beans.factory.annotation.Value; import org.springframework.beans.factory.annotation.Value;

View File

@ -0,0 +1,12 @@
package ru.ldeloff.servermonitorbot.integration.out.repository;
import com.jcraft.jsch.JSchException;
import ru.ldeloff.servermonitorbot.internal.data.model.SshServer;
public interface SshRepository {
void connect(SshServer sshServer) throws JSchException;
void disconnect(SshServer sshServer);
}

View File

@ -0,0 +1,37 @@
package ru.ldeloff.servermonitorbot.integration.out.repository;
import com.jcraft.jsch.JSch;
import com.jcraft.jsch.JSchException;
import com.jcraft.jsch.Session;
import lombok.RequiredArgsConstructor;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Repository;
import ru.ldeloff.servermonitorbot.internal.data.model.SshServer;
@Repository
@RequiredArgsConstructor
public class SshRepositoryImpl implements SshRepository {
@Value("${ssh.timeout:5000}")
private int TIMEOUT;
@Override
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();
sshServer.setSession(session);
}
@Override
public void disconnect(SshServer sshServer) {
sshServer.getSession().disconnect();
}
}

View File

@ -0,0 +1,25 @@
package ru.ldeloff.servermonitorbot.integration.out.service;
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
import ru.ldeloff.servermonitorbot.internal.data.model.SshCommand;
import ru.ldeloff.servermonitorbot.internal.data.model.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);
}

View File

@ -1,4 +1,4 @@
package ru.ldeloff.servermonitorbot.service.ssh; package ru.ldeloff.servermonitorbot.integration.out.service;
import com.jcraft.jsch.ChannelExec; import com.jcraft.jsch.ChannelExec;
import com.jcraft.jsch.JSchException; import com.jcraft.jsch.JSchException;
@ -7,27 +7,83 @@ import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import org.telegram.telegrambots.meta.api.methods.send.SendMessage; import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
import ru.ldeloff.servermonitorbot.model.SshCommand; import ru.ldeloff.servermonitorbot.config.SshServerList;
import ru.ldeloff.servermonitorbot.model.SshServer; import ru.ldeloff.servermonitorbot.internal.data.model.SshCommand;
import ru.ldeloff.servermonitorbot.repository.SshRepository; import ru.ldeloff.servermonitorbot.internal.data.model.SshServer;
import ru.ldeloff.servermonitorbot.integration.out.repository.SshRepository;
import java.io.ByteArrayOutputStream; import java.io.ByteArrayOutputStream;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.List; import java.util.List;
import java.util.Objects;
@Service @Service
@Slf4j @Slf4j
@RequiredArgsConstructor @RequiredArgsConstructor
public class SshServiceImpl implements SshService { public class SshServiceImpl implements SshService {
final SshRepository sshRepository;
private final SshServerList sshServerList;
private final SshRepository sshRepository;
@Override
public void updateConnection() {
sshServerList.getServers().forEach(sshServer -> {
try {
sshRepository.connect(sshServer);
log.info("Успешно подключён к {}", sshServer.getHost());
} catch (JSchException e) {
log.warn("Не удалось соединиться с {}: {}", sshServer.getHost(), e.getMessage());
}
});
}
@Override
public void disconnectAllConnections() {
sshServerList.getServers().forEach(sshRepository::disconnect);
}
@Override @Override
public SendMessage getStatusSessions(SendMessage message) { public SendMessage getStatusSessions(SendMessage message) {
message.setText(sshRepository.getStatusSessions()); StringBuilder text = new StringBuilder("Статус соединения (может выполняться долго): \n");
sshServerList.getServers().forEach(server -> {
text.append(server.getName())
.append(": ")
.append(checkStatusServer(server))
.append("\n");
});
message.setText(text.toString());
return message; 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 @Override
public List<SshServer> getSshServers() { public List<SshServer> getAllSshServers() {
return sshRepository.getSshServers(); return sshServerList.getServers();
}
@Override
public List<SshServer> getHealthSshServers() {
return sshServerList.getServers().stream().filter(this::checkHealth).toList();
} }
@Override @Override

View File

@ -1,30 +1,29 @@
package ru.ldeloff.servermonitorbot.service.command; package ru.ldeloff.servermonitorbot.internal.cpu.temp.service;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import ru.ldeloff.servermonitorbot.model.SshCommand; import ru.ldeloff.servermonitorbot.internal.template.CommandTemplate;
import ru.ldeloff.servermonitorbot.model.SshServer; import ru.ldeloff.servermonitorbot.internal.template.util.UpdateUtil;
import ru.ldeloff.servermonitorbot.service.role.RoleService; import ru.ldeloff.servermonitorbot.internal.data.model.SshCommand;
import ru.ldeloff.servermonitorbot.service.ssh.SshService; import ru.ldeloff.servermonitorbot.internal.data.model.SshServer;
import ru.ldeloff.servermonitorbot.service.user.UserService; import ru.ldeloff.servermonitorbot.integration.out.service.SshService;
import ru.ldeloff.servermonitorbot.utils.ui.ServerListButtons; import ru.ldeloff.servermonitorbot.internal.ui.ServerListButtons;
import java.util.Objects; import java.util.Objects;
import java.util.Optional; import java.util.Optional;
@Service @Service
@Slf4j @Slf4j
public class CpuTempCommand extends CommandTemplate { public class CpuTemp extends CommandTemplate {
final SshService sshService;
public CpuTempCommand(UserService userService, RoleService roleService, SshService sshService, ServerListButtons serverListButtons) { public CpuTemp(UpdateUtil updateUtil, ServerListButtons serverListButtons, SshService sshService) {
super(userService, roleService, serverListButtons, sshService); super(updateUtil, serverListButtons, sshService);
this.sshService = sshService;
setName("CPU.temp"); setName("CPU.temp");
} }
@Override @Override
public String executeCommandSpecificHost(String serverName) { public String executeCommandSpecificHost(String serverName) {
Optional<SshServer> server = sshService.getSshServers() Optional<SshServer> server = sshService.getAllSshServers()
.stream() .stream()
.filter(x -> x.getName().equals(serverName)) .filter(x -> x.getName().equals(serverName))
.findFirst(); .findFirst();
@ -36,7 +35,7 @@ public class CpuTempCommand extends CommandTemplate {
Double.parseDouble(result.getResponse().replaceAll("\n", ""))/1000)) Double.parseDouble(result.getResponse().replaceAll("\n", ""))/1000))
+ "°C"; + "°C";
} else { } else {
log.error("Ошибка при выполнении команды 'CPUtemp'. Искомый сервер (" + serverName + ") не найден"); log.error("Ошибка при выполнении команды 'CPUtemp'. Искомый сервер ({}) не найден", serverName);
return null; return null;
} }
} }

View File

@ -0,0 +1,27 @@
package ru.ldeloff.servermonitorbot.internal.data.mapper;
import org.springframework.stereotype.Component;
import ru.ldeloff.servermonitorbot.internal.data.model.User;
import ru.ldeloff.servermonitorbot.internal.data.model.UserEntity;
import ru.ldeloff.servermonitorbot.internal.data.service.RoleService;
@Component
public class UserMapper {
final RoleService roleService;
public UserMapper(RoleService roleService) {
this.roleService = roleService;
}
public UserEntity toEntity(User user) {
return new UserEntity()
.setTelegramId(user.getTelegramId())
.setRole(user.getRole());
}
public User toModel(UserEntity userEntity) {
return new User()
.setTelegramId(userEntity.getTelegramId())
.setRole(userEntity.getRole());
}
}

View File

@ -1,4 +1,4 @@
package ru.ldeloff.servermonitorbot.model; package ru.ldeloff.servermonitorbot.internal.data.model;
import jakarta.persistence.*; import jakarta.persistence.*;
import lombok.Data; import lombok.Data;

View File

@ -1,4 +1,4 @@
package ru.ldeloff.servermonitorbot.model; package ru.ldeloff.servermonitorbot.internal.data.model;
import jakarta.validation.constraints.Min; import jakarta.validation.constraints.Min;
import jakarta.validation.constraints.NotNull; import jakarta.validation.constraints.NotNull;

View File

@ -1,4 +1,4 @@
package ru.ldeloff.servermonitorbot.model; package ru.ldeloff.servermonitorbot.internal.data.model;
import com.jcraft.jsch.Session; import com.jcraft.jsch.Session;
import jakarta.validation.constraints.NotNull; import jakarta.validation.constraints.NotNull;

View File

@ -0,0 +1,12 @@
package ru.ldeloff.servermonitorbot.internal.data.model;
import lombok.Data;
import lombok.experimental.Accessors;
@Data
@Accessors(chain = true)
public class User {
private Long telegramId;
private Role role;
private String login;
}

View File

@ -1,12 +1,14 @@
package ru.ldeloff.servermonitorbot.model; package ru.ldeloff.servermonitorbot.internal.data.model;
import jakarta.persistence.*; import jakarta.persistence.*;
import lombok.Data; import lombok.Data;
import lombok.experimental.Accessors;
@Entity @Entity
@Data @Data
@Table(name = "users") @Table(name = "users")
public class User { @Accessors(chain = true)
public class UserEntity {
@Id @Id
@Column(name = "id") @Column(name = "id")
@GeneratedValue(strategy = GenerationType.IDENTITY) @GeneratedValue(strategy = GenerationType.IDENTITY)
@ -18,7 +20,4 @@ public class User {
@ManyToOne @ManyToOne
@JoinColumn(name = "role_id") @JoinColumn(name = "role_id")
private Role role; private Role role;
@Transient
private String login;
} }

View File

@ -1,8 +1,8 @@
package ru.ldeloff.servermonitorbot.repository; package ru.ldeloff.servermonitorbot.internal.data.repository;
import org.springframework.data.jpa.repository.JpaRepository; import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.stereotype.Repository; import org.springframework.stereotype.Repository;
import ru.ldeloff.servermonitorbot.model.Role; import ru.ldeloff.servermonitorbot.internal.data.model.Role;
import java.util.Optional; import java.util.Optional;

View File

@ -0,0 +1,17 @@
package ru.ldeloff.servermonitorbot.internal.data.repository;
import jakarta.validation.constraints.NotNull;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.stereotype.Repository;
import ru.ldeloff.servermonitorbot.internal.data.model.UserEntity;
import java.util.List;
@Repository
public interface UserRepository extends JpaRepository<UserEntity, Long> {
UserEntity getByTelegramId(long id);
@NotNull List<UserEntity> findAll();
}

View File

@ -0,0 +1,9 @@
package ru.ldeloff.servermonitorbot.internal.data.service;
import ru.ldeloff.servermonitorbot.internal.data.model.Role;
public interface RoleService {
Role findRoleByName(String name);
Role findRoleById(Long id);
}

View File

@ -1,9 +1,9 @@
package ru.ldeloff.servermonitorbot.service.role; package ru.ldeloff.servermonitorbot.internal.data.service;
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import ru.ldeloff.servermonitorbot.model.Role; import ru.ldeloff.servermonitorbot.internal.data.model.Role;
import ru.ldeloff.servermonitorbot.repository.RoleRepository; import ru.ldeloff.servermonitorbot.internal.data.repository.RoleRepository;
import java.util.Locale; import java.util.Locale;
import java.util.Optional; import java.util.Optional;
@ -11,6 +11,7 @@ import java.util.Optional;
@Service @Service
@RequiredArgsConstructor @RequiredArgsConstructor
public class RoleServiceImpl implements RoleService { public class RoleServiceImpl implements RoleService {
private final RoleRepository roleRepository; private final RoleRepository roleRepository;
@Override @Override

View File

@ -0,0 +1,19 @@
package ru.ldeloff.servermonitorbot.internal.data.service;
import ru.ldeloff.servermonitorbot.internal.data.model.User;
import ru.ldeloff.servermonitorbot.internal.data.model.UserEntity;
import java.util.List;
public interface UserService {
void saveOrUpdateUser(User user);
UserEntity getEntityByTelegramId(Long id);
User getByTelegramId(Long id);
List<UserEntity> getAllEntity();
List<User> getAllUsers();
}

View File

@ -0,0 +1,47 @@
package ru.ldeloff.servermonitorbot.internal.data.service;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import ru.ldeloff.servermonitorbot.internal.data.mapper.UserMapper;
import ru.ldeloff.servermonitorbot.internal.data.model.User;
import ru.ldeloff.servermonitorbot.internal.data.model.UserEntity;
import ru.ldeloff.servermonitorbot.internal.data.repository.UserRepository;
import java.util.List;
@Service
@RequiredArgsConstructor
@Slf4j
public class UserServiceImpl implements UserService {
final UserRepository userRepository;
final UserMapper userMapper;
@Override
public void saveOrUpdateUser(User user) {
UserEntity userEntity = getEntityByTelegramId(user.getTelegramId());
if (userEntity != null) {
userEntity.setRole(user.getRole());
} else {
userEntity = userMapper.toEntity(user);
}
userRepository.save(userEntity);
}
@Override
public UserEntity getEntityByTelegramId(Long id) {
return userRepository.getByTelegramId(id);
}
@Override
public User getByTelegramId(Long id) {
return userMapper.toModel(userRepository.getByTelegramId(id));
}
@Override
public List<UserEntity> getAllEntity() {return userRepository.findAll(); }
@Override
public List<User> getAllUsers() {return userRepository.findAll().stream().map(userMapper::toModel).toList(); }
}

View File

@ -1,28 +1,28 @@
package ru.ldeloff.servermonitorbot.service.command; package ru.ldeloff.servermonitorbot.internal.hdd.mdadm.service;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import ru.ldeloff.servermonitorbot.model.SshCommand; import ru.ldeloff.servermonitorbot.internal.template.CommandTemplate;
import ru.ldeloff.servermonitorbot.model.SshServer; import ru.ldeloff.servermonitorbot.internal.template.util.UpdateUtil;
import ru.ldeloff.servermonitorbot.service.role.RoleService; import ru.ldeloff.servermonitorbot.internal.data.model.SshCommand;
import ru.ldeloff.servermonitorbot.service.ssh.SshService; import ru.ldeloff.servermonitorbot.internal.data.model.SshServer;
import ru.ldeloff.servermonitorbot.service.user.UserService; import ru.ldeloff.servermonitorbot.integration.out.service.SshService;
import ru.ldeloff.servermonitorbot.utils.ui.ServerListButtons; import ru.ldeloff.servermonitorbot.internal.ui.ServerListButtons;
import java.util.*; import java.util.*;
@Service @Service
@Slf4j @Slf4j
public class MdadmStatusCommand extends CommandTemplate { public class MdadmStatus extends CommandTemplate {
final SshService sshService;
public MdadmStatusCommand(UserService userService, RoleService roleService, SshService sshService, ServerListButtons serverListButtons) { public MdadmStatus(UpdateUtil updateUtil, ServerListButtons serverListButtons, SshService sshService) {
super(userService, roleService, serverListButtons, sshService); super(updateUtil, serverListButtons, sshService);
this.sshService = sshService;
setName("MDADM"); setName("MDADM");
} }
@Override @Override
public String executeCommandSpecificHost(String serverName) { public String executeCommandSpecificHost(String serverName) {
Optional<SshServer> server = sshService.getSshServers() Optional<SshServer> server = sshService.getAllSshServers()
.stream() .stream()
.filter(x -> x.getName().equals(serverName)) .filter(x -> x.getName().equals(serverName))
.findFirst(); .findFirst();
@ -40,10 +40,11 @@ public class MdadmStatusCommand extends CommandTemplate {
}); });
return server.get().getName() + ": \n" + result; return server.get().getName() + ": \n" + result;
} else { } else {
log.error("Ошибка при выполнении команды 'mdadm'. Искомый сервер (" + serverName + ") не найден"); log.error("Ошибка при выполнении команды 'mdadm'. Искомый сервер ({}) не найден", serverName);
return null; return null;
} }
} }
private List<String> parseRaidNames(SshCommand result) { private List<String> parseRaidNames(SshCommand result) {
List<String> names = new ArrayList<>(); List<String> names = new ArrayList<>();
Arrays.stream(result.getResponse().split("\n")) Arrays.stream(result.getResponse().split("\n"))
@ -53,6 +54,7 @@ public class MdadmStatusCommand extends CommandTemplate {
.forEach(names::add); .forEach(names::add);
return names; return names;
} }
private List<String> parseHddState(SshCommand result) { private List<String> parseHddState(SshCommand result) {
List<String> names = new ArrayList<>(); List<String> names = new ArrayList<>();
names.add("Number Name State"); names.add("Number Name State");

View File

@ -1,29 +1,28 @@
package ru.ldeloff.servermonitorbot.service.command; package ru.ldeloff.servermonitorbot.internal.hdd.temp.service;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import ru.ldeloff.servermonitorbot.model.SshCommand; import ru.ldeloff.servermonitorbot.internal.template.CommandTemplate;
import ru.ldeloff.servermonitorbot.model.SshServer; import ru.ldeloff.servermonitorbot.internal.template.util.UpdateUtil;
import ru.ldeloff.servermonitorbot.service.role.RoleService; import ru.ldeloff.servermonitorbot.internal.data.model.SshCommand;
import ru.ldeloff.servermonitorbot.service.ssh.SshService; import ru.ldeloff.servermonitorbot.internal.data.model.SshServer;
import ru.ldeloff.servermonitorbot.service.user.UserService; import ru.ldeloff.servermonitorbot.integration.out.service.SshService;
import ru.ldeloff.servermonitorbot.utils.ui.ServerListButtons; import ru.ldeloff.servermonitorbot.internal.ui.ServerListButtons;
import java.util.Objects; import java.util.Objects;
import java.util.Optional; import java.util.Optional;
@Service @Service
@Slf4j @Slf4j
public class HddTempCommand extends CommandTemplate { public class HddTemp extends CommandTemplate {
final SshService sshService; public HddTemp(UpdateUtil updateUtil, ServerListButtons serverListButtons, SshService sshService) {
public HddTempCommand(UserService userService, RoleService roleService, SshService sshService, ServerListButtons serverListButtons) { super(updateUtil, serverListButtons, sshService);
super(userService, roleService, serverListButtons, sshService);
this.sshService = sshService;
setName("HDD.temp"); setName("HDD.temp");
} }
@Override @Override
public String executeCommandSpecificHost(String serverName) { public String executeCommandSpecificHost(String serverName) {
Optional<SshServer> server = sshService.getSshServers() Optional<SshServer> server = sshService.getAllSshServers()
.stream() .stream()
.filter(x -> x.getName().equals(serverName)) .filter(x -> x.getName().equals(serverName))
.findFirst(); .findFirst();

View File

@ -0,0 +1,34 @@
package ru.ldeloff.servermonitorbot.internal.mainmenu.service;
import org.springframework.stereotype.Service;
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
import ru.ldeloff.servermonitorbot.internal.template.CommandTemplate;
import ru.ldeloff.servermonitorbot.internal.template.util.UpdateUtil;
import ru.ldeloff.servermonitorbot.internal.data.model.User;
import ru.ldeloff.servermonitorbot.integration.out.service.SshService;
import ru.ldeloff.servermonitorbot.internal.ui.ServerListButtons;
import ru.ldeloff.servermonitorbot.internal.ui.TelegramBotKeyboard;
@Service
public class FirstUseCommand extends CommandTemplate {
private final TelegramBotKeyboard telegramBotKeyboard;
public FirstUseCommand(UpdateUtil updateUtil, ServerListButtons serverListButtons, SshService sshService, TelegramBotKeyboard telegramBotKeyboard) {
super(updateUtil, serverListButtons, sshService);
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;
}
}

View File

@ -0,0 +1,38 @@
package ru.ldeloff.servermonitorbot.internal.mainmenu.service;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
import ru.ldeloff.servermonitorbot.internal.template.CommandTemplate;
import ru.ldeloff.servermonitorbot.internal.template.util.UpdateUtil;
import ru.ldeloff.servermonitorbot.internal.data.model.User;
import ru.ldeloff.servermonitorbot.integration.out.service.SshService;
import ru.ldeloff.servermonitorbot.internal.ui.ServerListButtons;
import ru.ldeloff.servermonitorbot.internal.ui.TelegramBotKeyboard;
@Service
@Slf4j
public class SwitchToMainMenu extends CommandTemplate {
private final TelegramBotKeyboard telegramBotKeyboard;
public SwitchToMainMenu(UpdateUtil updateUtil, ServerListButtons serverListButtons, SshService sshService, TelegramBotKeyboard telegramBotKeyboard) {
super(updateUtil, serverListButtons, sshService);
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);
}
}

View File

@ -0,0 +1,18 @@
package ru.ldeloff.servermonitorbot.internal.statuschecker.scheduler;
import lombok.RequiredArgsConstructor;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Service;
import ru.ldeloff.servermonitorbot.integration.out.service.SshService;
@Service
@RequiredArgsConstructor
public class CheckSessionStatusScheduler {
private final SshService sshService;
@Scheduled(fixedRateString = "${ssh.checkConnectionDelay}")
public void updateSessionStatusCheck() {
sshService.updateConnection();
}
}

View File

@ -0,0 +1,27 @@
package ru.ldeloff.servermonitorbot.internal.statuschecker.service;
import org.springframework.stereotype.Service;
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
import ru.ldeloff.servermonitorbot.internal.template.CommandTemplate;
import ru.ldeloff.servermonitorbot.internal.template.util.UpdateUtil;
import ru.ldeloff.servermonitorbot.internal.data.model.User;
import ru.ldeloff.servermonitorbot.integration.out.service.SshService;
import ru.ldeloff.servermonitorbot.internal.ui.ServerListButtons;
@Service
public class CheckSessionStatus extends CommandTemplate {
public CheckSessionStatus(UpdateUtil updateUtil, ServerListButtons serverListButtons, SshService sshService) {
super(updateUtil, serverListButtons, sshService);
setName("Статус");
}
public SendMessage executeAggregate(User user, SendMessage message) {
return super.sshService.getStatusSessions(message);
}
@Override
public String executeCommandSpecificHost(String serverName) {
return null;
}
}

View File

@ -1,4 +1,4 @@
package ru.ldeloff.servermonitorbot.service.command; package ru.ldeloff.servermonitorbot.internal.template;
import lombok.Getter; import lombok.Getter;
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
@ -7,14 +7,12 @@ import lombok.extern.slf4j.Slf4j;
import org.telegram.telegrambots.meta.api.methods.send.SendMessage; import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
import org.telegram.telegrambots.meta.api.objects.Update; import org.telegram.telegrambots.meta.api.objects.Update;
import org.telegram.telegrambots.meta.exceptions.TelegramApiException; import org.telegram.telegrambots.meta.exceptions.TelegramApiException;
import ru.ldeloff.servermonitorbot.controller.TelegramBotController; import ru.ldeloff.servermonitorbot.integration.in.TelegramBotController;
import ru.ldeloff.servermonitorbot.model.SshServer; import ru.ldeloff.servermonitorbot.internal.data.model.SshServer;
import ru.ldeloff.servermonitorbot.model.User; import ru.ldeloff.servermonitorbot.internal.template.util.UpdateUtil;
import ru.ldeloff.servermonitorbot.service.role.RoleService; import ru.ldeloff.servermonitorbot.internal.data.model.User;
import ru.ldeloff.servermonitorbot.service.ssh.SshService; import ru.ldeloff.servermonitorbot.integration.out.service.SshService;
import ru.ldeloff.servermonitorbot.service.user.UserService; import ru.ldeloff.servermonitorbot.internal.ui.ServerListButtons;
import ru.ldeloff.servermonitorbot.utils.SshServerUtils;
import ru.ldeloff.servermonitorbot.utils.ui.ServerListButtons;
import java.util.Arrays; import java.util.Arrays;
import java.util.List; import java.util.List;
@ -23,10 +21,10 @@ import java.util.Objects;
@Slf4j @Slf4j
@RequiredArgsConstructor @RequiredArgsConstructor
public abstract class CommandTemplate { public abstract class CommandTemplate {
private final UserService userService;
private final RoleService roleService; private final UpdateUtil updateUtil;
private final ServerListButtons serverListButtons; private final ServerListButtons serverListButtons;
private final SshService sshService; public final SshService sshService;
@Getter @Getter
@Setter @Setter
private long expectedRole = 1L; private long expectedRole = 1L;
@ -39,13 +37,13 @@ public abstract class CommandTemplate {
return false; return false;
} }
User user = getUser(update); User user = updateUtil.getUser(update);
String message = getMessage(update); String message = updateUtil.getMessage(update);
SendMessage answer = new SendMessage(); SendMessage answer = new SendMessage();
answer.setChatId(user.getTelegramId()); answer.setChatId(user.getTelegramId());
answer.setText(message); answer.setText(message);
if (!isAuth(update)) { if (!updateUtil.roleChecker(update, expectedRole)) {
logNotAuth(user, message); logNotAuth(user, message);
sendMessage(actionForNotAuth(user, answer), bot); sendMessage(actionForNotAuth(user, answer), bot);
return true; return true;
@ -72,7 +70,7 @@ public abstract class CommandTemplate {
if (update.getMessage().hasText()) { if (update.getMessage().hasText()) {
String messageText = update.getMessage().getText(); String messageText = update.getMessage().getText();
if (Objects.equals(messageText, name)) { if (Objects.equals(messageText, name)) {
log.debug("Команда " + messageText + " будет выполнена"); log.debug("Команда {} будет выполнена", messageText);
return true; return true;
} }
} }
@ -80,7 +78,7 @@ public abstract class CommandTemplate {
String [] tags = update.getCallbackQuery().getData().split(":"); String [] tags = update.getCallbackQuery().getData().split(":");
if (tags.length > 1) { if (tags.length > 1) {
if (Objects.equals(tags[0], name)) { if (Objects.equals(tags[0], name)) {
log.debug("Команда " + tags[0] + " будет выполнена"); log.debug("Команда {} будет выполнена", tags[0]);
return true; return true;
} }
} else { } else {
@ -106,19 +104,17 @@ public abstract class CommandTemplate {
} }
return "error"; return "error";
} }
private boolean isAuth(Update update) {
User user = getUser(update);
return user.getRole().getId() <= expectedRole;
}
// Методы при верных условиях // Методы при верных условиях
void logSuccess(User user, String message) { void logSuccess(User user, String message) {
log.info("Получена команда '" + message + "' от " + user.getLogin() log.info("Получена команда '{}' от {} ({}). OK", message, user.getLogin(), user.getTelegramId());
+ " (" + user.getTelegramId() + "). OK");
} }
SendMessage executeAggregate(User user, SendMessage message) {
public SendMessage executeAggregate(User user, SendMessage message) {
return serverListButtons.uiForm(message); return serverListButtons.uiForm(message);
} }
SendMessage executeCommand(User user, SendMessage message) { SendMessage executeCommand(User user, SendMessage message) {
String [] tags = Arrays.stream(message.getText().split(":")) String [] tags = Arrays.stream(message.getText().split(":"))
.map(String::trim) .map(String::trim)
@ -132,7 +128,7 @@ public abstract class CommandTemplate {
return message; return message;
} }
private String executeCommandAllHost() { private String executeCommandAllHost() {
List<SshServer> servers = SshServerUtils.filterGoodServers(sshService.getSshServers()); List<SshServer> servers = sshService.getHealthSshServers();
StringBuilder response = new StringBuilder(); StringBuilder response = new StringBuilder();
servers.forEach(server -> response.append(executeCommandSpecificHost(server.getName())).append("\n")); servers.forEach(server -> response.append(executeCommandSpecificHost(server.getName())).append("\n"));
return response.toString(); return response.toString();
@ -151,35 +147,6 @@ public abstract class CommandTemplate {
} }
// Общее // Общее
// TODO: вынести в UpdateUtils
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;
}
String getMessage(Update update) {
if (update.hasMessage()) {
return update.getMessage().getText();
} else if (update.hasCallbackQuery()) {
return update.getCallbackQuery().getData();
}
return "";
}
// TODO: вынести // TODO: вынести
public static void sendMessage(SendMessage message, TelegramBotController bot) { public static void sendMessage(SendMessage message, TelegramBotController bot) {
try { try {

View File

@ -0,0 +1,4 @@
package ru.ldeloff.servermonitorbot.internal.template.model;
public enum CommandType {
}

View File

@ -0,0 +1,55 @@
package ru.ldeloff.servermonitorbot.internal.template.util;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
import org.telegram.telegrambots.meta.api.objects.Update;
import ru.ldeloff.servermonitorbot.internal.data.model.Role;
import ru.ldeloff.servermonitorbot.internal.data.model.User;
import ru.ldeloff.servermonitorbot.internal.data.service.RoleService;
import ru.ldeloff.servermonitorbot.internal.data.service.UserService;
@Service
@RequiredArgsConstructor
public class UpdateUtil {
private final UserService userService;
private final RoleService roleService;
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().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;
}
public Role getRole(Update update) {
return getUser(update).getRole();
}
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, long expectedRole) {
User user = getUser(update);
return user.getRole().getId() <= expectedRole;
}
}

View File

@ -1,14 +1,11 @@
package ru.ldeloff.servermonitorbot.utils.ui; package ru.ldeloff.servermonitorbot.internal.ui;
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
import org.telegram.telegrambots.meta.api.methods.send.SendMessage; 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.InlineKeyboardMarkup;
import org.telegram.telegrambots.meta.api.objects.replykeyboard.buttons.InlineKeyboardButton; import org.telegram.telegrambots.meta.api.objects.replykeyboard.buttons.InlineKeyboardButton;
import ru.ldeloff.servermonitorbot.model.Role; import ru.ldeloff.servermonitorbot.integration.out.service.SshService;
import ru.ldeloff.servermonitorbot.service.ssh.SshService;
import ru.ldeloff.servermonitorbot.utils.RoleChecker;
import ru.ldeloff.servermonitorbot.utils.ui.UiFormer;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.List; import java.util.List;
@ -25,7 +22,7 @@ public class ServerListButtons implements UiFormer {
List<List<InlineKeyboardButton>> keyboard = new ArrayList<>(); List<List<InlineKeyboardButton>> keyboard = new ArrayList<>();
InlineKeyboardMarkup inlineKeyboardMarkup = new InlineKeyboardMarkup(); InlineKeyboardMarkup inlineKeyboardMarkup = new InlineKeyboardMarkup();
sshService.getSshServers().forEach(sshServer -> { sshService.getAllSshServers().forEach(sshServer -> {
if (Objects.nonNull(sshServer.getSession())) { if (Objects.nonNull(sshServer.getSession())) {
if (sshServer.getSession().isConnected()) { if (sshServer.getSession().isConnected()) {
List<InlineKeyboardButton> keyboardRow = new ArrayList<>(); List<InlineKeyboardButton> keyboardRow = new ArrayList<>();

View File

@ -1,16 +1,18 @@
package ru.ldeloff.servermonitorbot.utils.ui; package ru.ldeloff.servermonitorbot.internal.ui;
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
import org.springframework.context.ApplicationContext;
import org.springframework.context.annotation.Lazy;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
import org.telegram.telegrambots.meta.api.methods.send.SendMessage; 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.ReplyKeyboardMarkup;
import org.telegram.telegrambots.meta.api.objects.replykeyboard.buttons.KeyboardButton; import org.telegram.telegrambots.meta.api.objects.replykeyboard.buttons.KeyboardButton;
import org.telegram.telegrambots.meta.api.objects.replykeyboard.buttons.KeyboardRow; import org.telegram.telegrambots.meta.api.objects.replykeyboard.buttons.KeyboardRow;
import ru.ldeloff.servermonitorbot.model.CommandList; import ru.ldeloff.servermonitorbot.internal.data.model.Role;
import ru.ldeloff.servermonitorbot.model.Role; import ru.ldeloff.servermonitorbot.internal.data.model.User;
import ru.ldeloff.servermonitorbot.service.command.CommandTemplate; import ru.ldeloff.servermonitorbot.internal.data.service.UserService;
import ru.ldeloff.servermonitorbot.utils.CommandUtil; import ru.ldeloff.servermonitorbot.internal.template.CommandTemplate;
import ru.ldeloff.servermonitorbot.utils.RoleChecker; import ru.ldeloff.servermonitorbot.internal.ui.util.CommandUtil;
import java.util.ArrayList; import java.util.ArrayList;
@ -20,23 +22,28 @@ import java.util.Map;
@Component @Component
@RequiredArgsConstructor @RequiredArgsConstructor
public class TelegramBotKeyboard implements UiFormer { public class TelegramBotKeyboard implements UiFormer {
private final RoleChecker roleChecker;
private final UserService userService;
private final ApplicationContext applicationContext;
@Override @Override
public SendMessage uiForm(SendMessage message) { public SendMessage uiForm(SendMessage message) {
ReplyKeyboardMarkup replyKeyboardMarkup = new ReplyKeyboardMarkup(); ReplyKeyboardMarkup replyKeyboardMarkup = new ReplyKeyboardMarkup();
replyKeyboardMarkup.setResizeKeyboard(true); replyKeyboardMarkup.setResizeKeyboard(true);
replyKeyboardMarkup.setOneTimeKeyboard(false); replyKeyboardMarkup.setOneTimeKeyboard(false);
ArrayList<KeyboardRow> keyboardRows = formKeyboard( String chatId = message.getChatId();
roleChecker.getRole(message).getId()); User user = userService.getByTelegramId(Long.parseLong(chatId));
Role role = user.getRole();
ArrayList<KeyboardRow> keyboardRows = formKeyboard(role.getId());
replyKeyboardMarkup.setKeyboard(keyboardRows); replyKeyboardMarkup.setKeyboard(keyboardRows);
message.setReplyMarkup(replyKeyboardMarkup); message.setReplyMarkup(replyKeyboardMarkup);
return message; return message;
} }
private ArrayList<KeyboardRow> formKeyboard(long role) { private ArrayList<KeyboardRow> formKeyboard(long role) {
HashMap<String, Long> commands = CommandUtil.getCommands(); Map<String, CommandTemplate> commands = applicationContext.getBeansOfType(CommandTemplate.class);
if (role < 3) { if (role < 3) {
commands.remove("/start"); commands.remove("/start");
commands.remove("Not found"); commands.remove("Not found");
@ -46,16 +53,16 @@ public class TelegramBotKeyboard implements UiFormer {
ArrayList<KeyboardRow> keyboardRows = new ArrayList<>(); ArrayList<KeyboardRow> keyboardRows = new ArrayList<>();
KeyboardRow keyboardRow = new KeyboardRow(); KeyboardRow keyboardRow = new KeyboardRow();
for (Map.Entry<String, Long> entry : commands.entrySet()) { for (Map.Entry<String, CommandTemplate> entry : commands.entrySet()) {
if (role <= entry.getValue()) { if (role <= entry.getValue().getExpectedRole()) {
keyboardRow.add(new KeyboardButton(entry.getKey())); keyboardRow.add(new KeyboardButton(entry.getValue().getName()));
} }
if (keyboardRow.size()>=buttonOnLine) { if (keyboardRow.size()>=buttonOnLine) {
keyboardRows.add(keyboardRow); keyboardRows.add(keyboardRow);
keyboardRow = new KeyboardRow(); keyboardRow = new KeyboardRow();
} }
} }
if (keyboardRow.size() > 0) { if (!keyboardRow.isEmpty()) {
keyboardRows.add(keyboardRow); keyboardRows.add(keyboardRow);
} }
return keyboardRows; return keyboardRows;

View File

@ -1,4 +1,4 @@
package ru.ldeloff.servermonitorbot.utils.ui; package ru.ldeloff.servermonitorbot.internal.ui;
import org.telegram.telegrambots.meta.api.methods.send.SendMessage; import org.telegram.telegrambots.meta.api.methods.send.SendMessage;

View File

@ -0,0 +1,21 @@
package ru.ldeloff.servermonitorbot.internal.ui.util;
import lombok.RequiredArgsConstructor;
import ru.ldeloff.servermonitorbot.internal.template.CommandTemplate;
import java.util.HashMap;
import java.util.Map;
@RequiredArgsConstructor
public class CommandUtil {
private final Map<String, CommandTemplate> commandsMap;
public HashMap<String, Long> getCommands() {
HashMap<String, Long> commands = new HashMap<>();
for (Map.Entry<String, CommandTemplate> entry : commandsMap.entrySet()) {
commands.put(entry.getValue().getName(), entry.getValue().getExpectedRole());
}
return commands;
}
}

View File

@ -0,0 +1,39 @@
package ru.ldeloff.servermonitorbot.internal.uname.service;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import ru.ldeloff.servermonitorbot.internal.template.CommandTemplate;
import ru.ldeloff.servermonitorbot.internal.template.util.UpdateUtil;
import ru.ldeloff.servermonitorbot.internal.data.model.SshCommand;
import ru.ldeloff.servermonitorbot.internal.data.model.SshServer;
import ru.ldeloff.servermonitorbot.integration.out.service.SshService;
import ru.ldeloff.servermonitorbot.internal.ui.ServerListButtons;
import java.util.Objects;
import java.util.Optional;
@Service
@Slf4j
public class Uname extends CommandTemplate {
public Uname(UpdateUtil updateUtil, ServerListButtons serverListButtons, SshService sshService) {
super(updateUtil, serverListButtons, sshService);
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;
}
}
}

View File

@ -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;
}
}

View File

@ -1,28 +0,0 @@
package ru.ldeloff.servermonitorbot.model;
import jakarta.annotation.PostConstruct;
import lombok.Getter;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Component;
import ru.ldeloff.servermonitorbot.service.command.CommandTemplate;
import ru.ldeloff.servermonitorbot.utils.CommandUtil;
import java.util.HashMap;
import java.util.Map;
@Component
@RequiredArgsConstructor
public class CommandList {
private final Map<String, CommandTemplate> services;
@Getter
private HashMap<String, Long> commands;
@PostConstruct
private void setCommands() {
commands = new HashMap<>();
for (Map.Entry<String, CommandTemplate> entry : services.entrySet()) {
commands.put(entry.getValue().getName(), entry.getValue().getExpectedRole());
}
CommandUtil.setCommands(commands);
}
}

View File

@ -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;
}

View File

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

@ -1,108 +0,0 @@
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.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 java.util.*;
@Slf4j
@Repository
public class SshRepositoryImpl implements SshRepository {
private final List<SshServer> sshServers;
@Value("${ssh.timeout:5000}")
private int TIMEOUT;
@Autowired
public SshRepositoryImpl(SshConfig sshConfig) {
this.sshServers = sshConfig.getServers();
}
private Set<Session> sessions = new HashSet<>();
@Override
public void connectToAllServer() {
sshServers.forEach(sshServer -> {
try {
Session session = connectToServer(sshServer);
sshServer.setSession(session);
log.info("Успешно подключён к " + sshServer.getHost());
} 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.setTimeout(TIMEOUT);
session.connect();
sessions.add(session);
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())) {
try {
connectToServer(server);
return "OK";
} catch (JSchException e) {
log.warn("Не удалось соединиться с " + server.getHost() + ": " + e.getMessage());
return "нет соединения";
}
}
if (server.getSession().isConnected()) {
return "OK";
} else {
try {
connectToServer(server);
return "OK";
} catch (JSchException e) {
log.warn("Не удалось соединиться с " + server.getHost() + ": " + e.getMessage());
return "потеряно соединение";
}
}
}
}

View File

@ -1,15 +0,0 @@
package ru.ldeloff.servermonitorbot.repository;
import jakarta.validation.constraints.NotNull;
import org.springframework.data.domain.Example;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.stereotype.Repository;
import ru.ldeloff.servermonitorbot.model.User;
import java.util.List;
@Repository
public interface UserRepository extends JpaRepository<User, Long> {
User getByTelegramId(long id);
@NotNull List<User> findAll();
}

View File

@ -1,37 +0,0 @@
package ru.ldeloff.servermonitorbot.service.command;
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.User;
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.ui.ServerListButtons;
import ru.ldeloff.servermonitorbot.utils.ui.TelegramBotKeyboard;
@Service
public class FirstUseCommand extends CommandTemplate {
final TelegramBotKeyboard telegramBotKeyboard;
public FirstUseCommand(UserService userService,
RoleService roleService,
SshService sshService,
ServerListButtons serverListButtons,
TelegramBotKeyboard telegramBotKeyboard) {
super(userService, roleService, serverListButtons, sshService);
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;
}
}

View File

@ -1,27 +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;
import ru.ldeloff.servermonitorbot.utils.ui.ServerListButtons;
@Service
public class GetStatusSessions extends CommandTemplate {
final SshService sshService;
public GetStatusSessions(UserService userService, RoleService roleService, SshService sshService, ServerListButtons serverListButtons) {
super(userService, roleService, serverListButtons, sshService);
this.sshService = sshService;
setName("Статус");
}
@Override
public SendMessage executeAggregate(User user, SendMessage message) {
return sshService.getStatusSessions(message);
}
@Override
public String executeCommandSpecificHost(String serverName) {
return null;
}
}

View File

@ -1,47 +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 org.telegram.telegrambots.meta.api.objects.Update;
import ru.ldeloff.servermonitorbot.controller.TelegramBotController;
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;
import ru.ldeloff.servermonitorbot.utils.ui.ServerListButtons;
import ru.ldeloff.servermonitorbot.utils.ui.TelegramBotKeyboard;
@Service
@Slf4j
public class SwitchToMainMenu extends CommandTemplate {
final TelegramBotKeyboard telegramBotKeyboard;
public SwitchToMainMenu(UserService userService,
RoleService roleService,
SshService sshService,
ServerListButtons serverListButtons,
TelegramBotKeyboard telegramBotKeyboard) {
super(userService, roleService, serverListButtons, sshService);
this.telegramBotKeyboard = telegramBotKeyboard;
setName("Not found");
}
@Override
public String executeCommandSpecificHost(String serverName) {
return null;
}
@Override
void logSuccess(User user, String message) {
log.warn("Получена несуществующая команда '" + message + "' от " + user.getLogin()
+ " (" + user.getTelegramId() + "). OK");
}
@Override
SendMessage executeAggregate(User user, SendMessage message) {
SendMessage answer = new SendMessage();
answer.setChatId(user.getTelegramId());
answer.setText("Неверная команда. Попробуем сначала.");
return telegramBotKeyboard.uiForm(answer);
}
}

View File

@ -1,39 +0,0 @@
package ru.ldeloff.servermonitorbot.service.command;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import ru.ldeloff.servermonitorbot.model.SshCommand;
import ru.ldeloff.servermonitorbot.model.SshServer;
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.ui.ServerListButtons;
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, ServerListButtons serverListButtons) {
super(userService, roleService, serverListButtons, sshService);
this.sshService = sshService;
setName("uname");
}
@Override
public String executeCommandSpecificHost(String serverName) {
Optional<SshServer> server = sshService.getSshServers()
.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;
}
}
}

View File

@ -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);
}

View File

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

View File

@ -1,13 +0,0 @@
package ru.ldeloff.servermonitorbot.service.user;
import ru.ldeloff.servermonitorbot.model.User;
import java.util.List;
public interface UserService {
User saveOrUpdateUser(User user);
User getByTelegramId(Long id);
List<User> getAllUsers();
}

View File

@ -1,34 +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;
import java.util.List;
@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);
}
@Override
public List<User> getAllUsers() {return userRepository.findAll(); }
}

View File

@ -1,15 +0,0 @@
package ru.ldeloff.servermonitorbot.utils;
import lombok.Getter;
import lombok.Setter;
import lombok.experimental.UtilityClass;
import java.util.HashMap;
@UtilityClass
public class CommandUtil {
@Setter
@Getter
private HashMap<String, Long> commands;
}

View File

@ -1,19 +0,0 @@
package ru.ldeloff.servermonitorbot.utils;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Component;
import org.telegram.telegrambots.meta.api.methods.send.SendMessage;
import ru.ldeloff.servermonitorbot.model.Role;
import ru.ldeloff.servermonitorbot.model.User;
import ru.ldeloff.servermonitorbot.service.user.UserService;
@RequiredArgsConstructor
@Component
public class RoleChecker {
final UserService userService;
public Role getRole(SendMessage sendMessage) {
String chatId = sendMessage.getChatId();
User user = userService.getByTelegramId(Long.parseLong(chatId));
return user.getRole();
}
}

View File

@ -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;
}
}

View File

@ -17,6 +17,7 @@ bot:
- telegramId: 123456789 - telegramId: 123456789
role: admin role: admin
ssh: ssh:
checkConnectionDelay: 60000
timeout: 5000 timeout: 5000
servers: servers:
- -