public void onTimeoutExpire(User user, Server server) {
String serverId = server.getId();
TempServerConfig serverConfig = serverStorage.get(serverId);
if (serverConfig == null) {
serverConfig = new TempServerConfig(serverId);
serverStorage.put(serverId, serverConfig);
}
ServerTimeoutStorage storage = serverConfig.getServerTimeouts();
if (storage != null) {
ServerTimeout timeout = storage.getTimeouts().remove(user.getId());
if (timeout != null) {
saveServerConfig(serverConfig);
LOGGER.info(
"Expiring timeout for {} ({}) in {} ({})",
user.getUsername(),
user.getId(),
server.getName(),
server.getId());
if (apiClient.getUserById(user.getId(), server) != NO_USER) {
apiClient.sendMessage(
loc.localize("message.mod.timeout.expire", user.getId()), server.getId());
}
removeTimeoutRole(user, server, apiClient.getChannelById(server.getId()));
return;
}
}
LOGGER.warn(
"Unable to expire: find server or timeout entry for {} ({}) in {} ({})",
user.getUsername(),
user.getId(),
server.getName(),
server.getId());
}
@Test
@TestDir
public void constructorsGetters() throws Exception {
Server server =
new Server(
"server",
getAbsolutePath("/a"),
getAbsolutePath("/b"),
getAbsolutePath("/c"),
getAbsolutePath("/d"),
new Configuration(false));
assertEquals(server.getHomeDir(), getAbsolutePath("/a"));
assertEquals(server.getConfigDir(), getAbsolutePath("/b"));
assertEquals(server.getLogDir(), getAbsolutePath("/c"));
assertEquals(server.getTempDir(), getAbsolutePath("/d"));
assertEquals(server.getName(), "server");
assertEquals(server.getPrefix(), "server");
assertEquals(server.getPrefixedName("name"), "server.name");
assertNotNull(server.getConfig());
server =
new Server(
"server",
getAbsolutePath("/a"),
getAbsolutePath("/b"),
getAbsolutePath("/c"),
getAbsolutePath("/d"));
assertEquals(server.getHomeDir(), getAbsolutePath("/a"));
assertEquals(server.getConfigDir(), getAbsolutePath("/b"));
assertEquals(server.getLogDir(), getAbsolutePath("/c"));
assertEquals(server.getTempDir(), getAbsolutePath("/d"));
assertEquals(server.getName(), "server");
assertEquals(server.getPrefix(), "server");
assertEquals(server.getPrefixedName("name"), "server.name");
assertNull(server.getConfig());
server =
new Server(
"server", TestDirHelper.getTestDir().getAbsolutePath(), new Configuration(false));
assertEquals(server.getHomeDir(), TestDirHelper.getTestDir().getAbsolutePath());
assertEquals(server.getConfigDir(), TestDirHelper.getTestDir() + "/conf");
assertEquals(server.getLogDir(), TestDirHelper.getTestDir() + "/log");
assertEquals(server.getTempDir(), TestDirHelper.getTestDir() + "/temp");
assertEquals(server.getName(), "server");
assertEquals(server.getPrefix(), "server");
assertEquals(server.getPrefixedName("name"), "server.name");
assertNotNull(server.getConfig());
server = new Server("server", TestDirHelper.getTestDir().getAbsolutePath());
assertEquals(server.getHomeDir(), TestDirHelper.getTestDir().getAbsolutePath());
assertEquals(server.getConfigDir(), TestDirHelper.getTestDir() + "/conf");
assertEquals(server.getLogDir(), TestDirHelper.getTestDir() + "/log");
assertEquals(server.getTempDir(), TestDirHelper.getTestDir() + "/temp");
assertEquals(server.getName(), "server");
assertEquals(server.getPrefix(), "server");
assertEquals(server.getPrefixedName("name"), "server.name");
assertNull(server.getConfig());
}
public List<ContainerUnit> listContainers(String applicationName, boolean withModules)
throws ServiceException {
List<ContainerUnit> containers = new ArrayList<>();
try {
Application application =
findByNameAndUser(authentificationUtils.getAuthentificatedUser(), applicationName);
if (application != null) {
try {
// Serveurs
List<Server> servers = application.getServers();
// Ajout des containers de type server
for (Server server : servers) {
DockerContainer dockerContainer = new DockerContainer();
dockerContainer.setName(server.getName());
dockerContainer = DockerContainer.findOne(dockerContainer, application.getManagerIp());
server = containerMapper.mapDockerContainerToServer(dockerContainer, server);
ContainerUnit containerUnit =
new ContainerUnit(server.getName(), server.getContainerID(), "server");
containers.add(containerUnit);
}
if (withModules) {
// Ajout des containers de type module
List<Module> modules = application.getModules();
for (Module module : modules) {
// on evite de remonter les modules de type toolkit
// (git, maven...)
if (module.isTool()) {
continue;
}
DockerContainer dockerContainer = new DockerContainer();
dockerContainer.setName(module.getName());
dockerContainer =
DockerContainer.findOne(dockerContainer, application.getManagerIp());
module = containerMapper.mapDockerContainerToModule(dockerContainer, module);
ContainerUnit containerUnit =
new ContainerUnit(module.getName(), module.getContainerID(), "module");
containers.add(containerUnit);
}
}
} catch (Exception ex) {
// Si une application sort en erreur, il ne faut pas
// arrêter la suite des traitements
logger.error(application.toString(), ex);
}
}
} catch (Exception e) {
throw new ServiceException(e.getLocalizedMessage(), e);
}
return containers;
}
private void refreshTimeoutOnEvade(User user, Server server) {
ServerTimeout timeout =
SafeNav.of(serverStorage.get(server.getId()))
.next(TempServerConfig::getServerTimeouts)
.next(ServerTimeoutStorage::getTimeouts)
.next(timeouts -> timeouts.get(user.getId()))
.get();
if (timeout == null) {
LOGGER.warn(
"Attempted to refresh a timeout on a user who was not timed out! {} ({})",
user.getUsername(),
user.getId());
return;
}
LOGGER.info(
"User {} ({}) attempted to evade a timeout on {} ({})!",
user.getUsername(),
user.getId(),
server.getName(),
server.getId());
Channel channel = apiClient.getChannelById(server.getId(), server);
apiClient.sendMessage(
loc.localize(
"listener.mod.timeout.on_evasion",
user.getId(),
formatDuration(Duration.between(Instant.now(), timeout.getEndTime())),
formatInstant(timeout.getEndTime())),
channel);
applyTimeoutRole(user, server, channel);
}
private void stopTimeout(MessageContext context, String args) {
if (args.isEmpty()) {
apiClient.sendMessage(
loc.localize("commands.mod.stoptimeout.response.invalid"), context.getChannel());
return;
}
String uid = args;
if (uid.length() > 4) {
if (uid.startsWith("<@")) {
uid = uid.substring(2, uid.length() - 1);
}
Server server = context.getServer();
User user = apiClient.getUserById(uid, server);
if (user == NO_USER) {
user = new User("UNKNOWN", uid, "", null);
}
LOGGER.info(
"{} ({}) is attempting to cancel timeout for {} ({}) in {} ({})",
context.getAuthor().getUsername(),
context.getAuthor().getId(),
user.getUsername(),
user.getId(),
server.getName(),
server.getId());
cancelTimeout(user, server, context.getChannel());
} else {
apiClient.sendMessage(
loc.localize("commands.mod.stoptimeout.response.invalid"), context.getChannel());
}
}
public boolean applyTimeout(
User issuingUser, Channel noticeChannel, Server server, User user, Duration duration) {
String serverId = server.getId();
if (duration != null && !duration.isNegative() && !duration.isZero()) {
ServerTimeout timeout =
new ServerTimeout(
duration,
Instant.now(),
user.getId(),
serverId,
user.getUsername(),
issuingUser.getId());
TempServerConfig serverConfig = serverStorage.get(serverId);
if (serverConfig == null) {
serverConfig = new TempServerConfig(serverId);
serverStorage.put(serverId, serverConfig);
}
ServerTimeoutStorage storage = serverConfig.getServerTimeouts();
if (storage == null) {
storage = new ServerTimeoutStorage();
serverConfig.setServerTimeouts(storage);
}
if (applyTimeoutRole(user, server, noticeChannel)) {
storage.getTimeouts().put(user.getId(), timeout);
ScheduledFuture future =
timeoutService.schedule(
() -> onTimeoutExpire(user, server), duration.getSeconds(), TimeUnit.SECONDS);
timeout.setTimerFuture(future);
saveServerConfig(serverConfig);
String durationStr = formatDuration(duration);
String instantStr = formatInstant(timeout.getEndTime());
String msg =
loc.localize(
"commands.mod.timeout.response",
user.getUsername(),
user.getId(),
durationStr,
instantStr);
apiClient.sendMessage(msg, noticeChannel);
LOGGER.info(
"[{}] '{}': Timing out {} ({}) for {} (until {}), issued by {} ({})",
serverId,
server.getName(),
user.getUsername(),
user.getId(),
durationStr,
instantStr,
issuingUser.getUsername(),
issuingUser.getId());
}
// No else with error - applyTimeoutRole does that for us
return true;
} else {
LOGGER.warn("Invalid duration format");
}
return false;
}
public static Builder fromServer(Server in) {
return new Builder()
.uuid(in.getUuid())
.name(in.getName())
.resourceUri(in.getResourceUri())
.owner(in.getOwner())
.status(in.getStatus())
.runtime(in.getRuntime());
}
public void testLoadServer() throws Exception {
Database database = _category.getDatabase();
database.begin();
Server server = (Server) database.load(Server.class, new Integer(3));
database.commit();
assertNotNull(server);
assertEquals("ctf.jdo.tc9x.Server", server.getClass().getName());
assertEquals(3, server.getId());
assertEquals("server 3", server.getName());
database.close();
}
private void saveServers() {
File serverDir = new File(settings.getLocalFolder(), "servers");
if (!serverDir.exists()) {
serverDir.mkdirs();
}
for (Server s : model.getAllServers()) {
try {
File serverFile = new File(serverDir, s.getName() + ".server");
s.writeToFile(serverFile);
} catch (IOException ex) {
logger.log(Level.SEVERE, null, ex);
}
}
}
public void cancelTimeout(User user, Server server, Channel invocationChannel) {
String serverId = server.getId();
TempServerConfig serverConfig = serverStorage.get(serverId);
if (serverConfig == null) {
serverConfig = new TempServerConfig(serverId);
serverStorage.put(serverId, serverConfig);
}
ServerTimeoutStorage storage = serverConfig.getServerTimeouts();
removeTimeoutRole(user, server, apiClient.getChannelById(serverId));
if (storage != null) {
ServerTimeout timeout = storage.getTimeouts().remove(user.getId());
saveServerConfig(serverConfig);
if (timeout != null) {
SafeNav.of(timeout.getTimerFuture()).ifPresent(f -> f.cancel(true));
LOGGER.info(
"Cancelling timeout for {} ({}) in {} ({})",
user.getUsername(),
user.getId(),
server.getName(),
serverId);
apiClient.sendMessage(
loc.localize("commands.mod.stoptimeout.response", user.getUsername(), user.getId()),
invocationChannel);
return;
}
}
LOGGER.warn(
"Unable to cancel: cannot find server or timeout entry for {} ({}) in {} ({})",
user.getUsername(),
user.getId(),
server.getName(),
server.getId());
apiClient.sendMessage(
loc.localize(
"commands.mod.stoptimeout.response.not_found", user.getUsername(), user.getId()),
invocationChannel);
}
public void deleteKnownServer() {
Server currentServer = model.getCurrentServer();
if (currentServer != null) {
try {
model.removeServer(currentServer);
File serverDir = new File(settings.getLocalFolder(), "servers");
File serverFile = new File(serverDir, currentServer.getName() + ".server");
Files.delete(serverFile.toPath());
} catch (IOException ex) {
model.addErrorMessage("Server-Datei konnte nicht gelöscht werden");
logger.log(Level.SEVERE, null, ex);
}
model.fireChange();
}
}
/**
* Removes the timeout role from the given user. This does NOT create or manage any
* storage/persistence, it only sets the user's roles
*
* @param user The user to remove the timeout role
* @param server The server on which to remove the user from the timeout role
* @param invocationChannel The channel to send messages on error
*/
public boolean removeTimeoutRole(User user, Server server, Channel invocationChannel) {
String serverId = server.getId();
TempServerConfig serverConfig = serverStorage.get(serverId);
if (serverConfig == null) {
serverConfig = new TempServerConfig(serverId);
serverStorage.put(serverId, serverConfig);
}
ServerTimeoutStorage storage = serverConfig.getServerTimeouts();
String serverName = server.getName();
if (storage != null && storage.getTimeoutRoleId() != null) {
String timeoutRoleId = storage.getTimeoutRoleId();
Role timeoutRole = apiClient.getRole(timeoutRoleId, server);
if (timeoutRole != NO_ROLE) {
// Get roles
Set<Role> userRoles =
apiClient.getMemberRoles(apiClient.getUserMember(user, server), server);
// Delete the ban role
LinkedHashSet<String> newRoles = new LinkedHashSet<>(userRoles.size() - 1);
userRoles
.stream()
.map(Role::getId)
.filter(s -> !timeoutRoleId.equals(s))
.forEach(newRoles::add);
// Update
apiClient.updateRoles(user, server, newRoles);
return userRoles.size() == newRoles.size();
} else {
LOGGER.warn(
"Timeout role ID {} for server {} ({}) does not exist",
timeoutRoleId,
serverName,
serverId);
apiClient.sendMessage(
loc.localize("message.mod.timeout.bad_role", timeoutRoleId), invocationChannel);
}
} else {
storage = new ServerTimeoutStorage();
serverConfig.setServerTimeouts(storage);
serverStorage.put(serverId, serverConfig);
LOGGER.warn(
"Timeout role for server {} ({}) is not configured",
storage.getTimeoutRoleId(),
serverName,
serverId);
apiClient.sendMessage(loc.localize("message.mod.timeout.not_configured"), invocationChannel);
}
return false;
}
public void loadServerConfig(Path path) {
boolean purge = false;
TempServerConfig config;
ServerTimeoutStorage storage;
try (Reader reader = Files.newBufferedReader(path, UTF_8)) {
config = gson.fromJson(reader, TempServerConfig.class);
serverStorage.put(config.getServerId(), config);
storage = config.getServerTimeouts();
if (storage != null) {
Server server = apiClient.getServerByID(config.getServerId());
if (server == NO_SERVER) {
LOGGER.warn("Rejecting {} server storage file: server not found", config.getServerId());
return;
}
LOGGER.info(
"Loaded {} ({}) server storage file",
server.getName(),
server.getId(),
storage.getTimeoutRoleId());
// Prune expired entries
for (Iterator<Map.Entry<String, ServerTimeout>> iter =
storage.getTimeouts().entrySet().iterator();
iter.hasNext(); ) {
Map.Entry<String, ServerTimeout> e = iter.next();
ServerTimeout timeout = e.getValue();
String userId = timeout.getUserId();
User user = apiClient.getUserById(userId, server);
if (!isUserTimedOut(userId, server.getId())) {
// Purge!
purge = true;
if (user == NO_USER) {
LOGGER.info(
"Ending timeout for departed user {} ({}) in {} ({})",
timeout.getLastUsername(),
userId,
server.getName(),
server.getId());
//
// apiClient.sendMessage(loc.localize("message.mod.timeout.expire.not_found",
// user.getId()),
// server.getId());
// Don't need to remove the timeout role because leaving does that for us
} else {
// Duplicated from onTimeoutExpire except without remove since we're removing in an
// iter
LOGGER.info(
"Expiring timeout for {} ({}) in {} ({})",
user.getUsername(),
user.getId(),
server.getName(),
server.getId());
// Only send message if they still have the role
if (removeTimeoutRole(user, server, apiClient.getChannelById(server.getId()))) {
//
// apiClient.sendMessage(loc.localize("message.mod.timeout.expire",
// user.getId()),
// server.getId());
}
}
SafeNav.of(timeout.getTimerFuture()).ifPresent(f -> f.cancel(true));
iter.remove();
} else {
// Start our futures
Duration duration = Duration.between(Instant.now(), timeout.getEndTime());
ScheduledFuture future =
timeoutService.schedule(
() -> onTimeoutExpire(user, server), duration.getSeconds(), TimeUnit.SECONDS);
timeout.setTimerFuture(future);
}
}
}
} catch (IOException | JsonParseException e) {
LOGGER.warn("Unable to load server storage file " + path.toString(), e);
return;
}
if (purge) {
saveServerConfig(config);
}
}
public static String getName() {
return server.getName();
}
/**
* Remove an application
*
* @param application
* @param user
* @return
* @throws ServiceException
*/
@Override
@Transactional
public Application remove(Application application, User user)
throws ServiceException, CheckException {
try {
logger.info("Starting removing application " + application.getName());
// Delete all modules
List<Module> listModules = application.getModules();
for (Module module : listModules) {
try {
moduleService.remove(application, user, module, false, application.getStatus());
} catch (ServiceException | CheckException e) {
application.setStatus(Status.FAIL);
logger.error(
"ApplicationService Error : failed to remove module "
+ module.getName()
+ " for application "
+ application.getName()
+ " : "
+ e);
e.printStackTrace();
}
}
// Delete all alias
List<String> aliases = new ArrayList<>();
aliases.addAll(application.getAliases());
for (String alias : aliases) {
removeAlias(application, alias);
}
// Delete all servers
List<Server> listServers = application.getServers();
for (Server server : listServers) {
serverService.remove(server.getName());
if (listServers.indexOf(server) == listServers.size() - 1) {
hipacheRedisUtils.removeRedisAppKey(application);
applicationDAO.delete(server.getApplication());
portUtils.releaseProxyPorts(application);
}
}
logger.info("ApplicationService : Application successfully removed ");
} catch (PersistenceException e) {
setStatus(application, Status.FAIL);
logger.error(
"ApplicationService Error : failed to remove " + application.getName() + " : " + e);
throw new ServiceException(e.getLocalizedMessage(), e);
} catch (ServiceException e) {
setStatus(application, Status.FAIL);
logger.error(
"ApplicationService Error : failed to remove application "
+ application.getName()
+ " : "
+ e);
}
return application;
}