Add removal methods to WindowsScheduler and StationSettings

This commit is contained in:
Nathan McRae 2025-03-03 21:10:07 -08:00
parent e74ddc99ca
commit 6c43289860
2 changed files with 109 additions and 0 deletions

View File

@ -1,5 +1,6 @@
package name.nathanmcrae.numbersstation;
import com.leakyabstractions.result.api.Result;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.nio.file.attribute.BasicFileAttributes;
@ -13,6 +14,7 @@ import java.time.LocalDate;
import java.time.LocalTime;
import java.util.ArrayList;
import java.util.logging.Logger;
import java.util.logging.Level;
import java.util.Random;
public class StationSettings {
@ -65,6 +67,44 @@ public class StationSettings {
scheduleStartTime = LocalTime.now();
}
public static void deleteStationData(String stationName) throws IOException {
StationSettings temp = new StationSettings(stationName);
temp.deleteDir();
String osName = System.getProperty("os.name").toLowerCase();
if (osName.contains("win")) {
WindowsScheduler.removeSchedule(temp);
} else if (osName.contains("nix") || osName.contains("nux") || osName.contains("aix")) {
// TODO: linux
logger.log(Level.SEVERE, "Unsupported OS " + osName);
} else {
logger.log(Level.SEVERE, "Unsupported OS " + osName);
}
}
public static boolean stationDataExists(String stationName) {
StationSettings temp = new StationSettings(stationName);
if (Files.exists(temp.stationPath())) {
return true;
}
String osName = System.getProperty("os.name").toLowerCase();
if (osName.contains("win")) {
Result<Boolean, String> result = WindowsScheduler.scheduleExists(temp);
if (result.hasSuccess() && result.getSuccess().get()) {
return true;
}
} else if (osName.contains("nix") || osName.contains("nux") || osName.contains("aix")) {
// TODO: linux
logger.log(Level.SEVERE, "Unsupported OS " + osName);
} else {
logger.log(Level.SEVERE, "Unsupported OS " + osName);
}
return false;
}
public void deleteDir() throws IOException {
Files.walkFileTree(stationPath(), new SimpleFileVisitor<Path>() {
@Override
@ -81,6 +121,7 @@ public class StationSettings {
});
}
public Path nextMessagePath() {
return stationPath().resolve("next-message.txt");
}

View File

@ -341,6 +341,52 @@ public class WindowsScheduler {
return Results.success(true);
}
public static Result<Boolean, String> removeSchedule(StationSettings settings) {
Path tempFile = null;
try {
String taskName = "numbers-station-main_" + settings.getName();
Process queryTaskProcess = new ProcessBuilder("schtasks.exe", "/query", "/tn", taskName).start();
if (!queryTaskProcess.waitFor(5, TimeUnit.SECONDS)) {
String message = "Failed to query " + taskName + " task: process timed out";
logger.log(Level.SEVERE, message);
return Results.failure(message);
}
// Remove previous instance of task if it exists
if (queryTaskProcess.exitValue() == 0) {
Process removeTaskProcess = new ProcessBuilder("schtasks.exe", "/delete", "/f", "/tn", taskName).start();
if (!removeTaskProcess.waitFor(5, TimeUnit.SECONDS)) {
String message = "Failed to remove " + taskName + " task: process timed out";
logger.log(Level.SEVERE, message);
return Results.failure(message);
}
if (removeTaskProcess.exitValue() == 0) {
logger.info(taskName + " task removed successfully.");
} else {
Pair<String, String> output = captureProcessOutput(removeTaskProcess);
String message = "Failed to remove " + taskName + " task. Exit code: " + removeTaskProcess.exitValue() + ". stdout: " + output.getKey() + "\n\tstderr: " + output.getValue();
logger.log(Level.SEVERE, message);
return Results.failure(message);
}
}
} catch (IOException | InterruptedException e) {
String message = "Exception while registering schedule";
logger.log(Level.SEVERE, message, e);
return Results.failure(message);
} finally {
if (tempFile != null) {
try {
Files.delete(tempFile);
} catch (IOException e) {
logger.log(Level.SEVERE, "Failed to delete temporary file", e);
}
}
}
return Results.success(true);
}
public static Result<Boolean, String> runSchedule(StationSettings settings) {
String taskName = "numbers-station-main_" + settings.getName();
@ -355,6 +401,28 @@ public class WindowsScheduler {
return Results.success(true);
}
public static Result<Boolean, String> scheduleExists(StationSettings settings) {
try {
String taskName = "numbers-station-main_" + settings.getName();
Process queryTaskProcess = new ProcessBuilder("schtasks.exe", "/query", "/tn", taskName).start();
if (!queryTaskProcess.waitFor(5, TimeUnit.SECONDS)) {
String message = "Failed to query " + taskName + " task: process timed out";
logger.log(Level.SEVERE, message);
return Results.failure(message);
}
return Results.success(queryTaskProcess.exitValue() == 0);
} catch (InterruptedException | IOException e) {
String message = "Exception while querying schedule";
logger.log(Level.SEVERE, message, e);
return Results.failure(message);
}
}
/**
* @return (stdout contents, stderr contents)
*/
public static Pair<String, String> captureProcessOutput(Process process) throws IOException {
StringBuilder output = new StringBuilder();
StringBuilder error = new StringBuilder();