Start implementing task scheduler
This commit is contained in:
parent
591d431226
commit
a68952653c
@ -397,7 +397,7 @@ public class MainSettingsController {
|
|||||||
|
|
||||||
@FXML
|
@FXML
|
||||||
private void handleTestConnectionButtonPress() {
|
private void handleTestConnectionButtonPress() {
|
||||||
// TODO
|
WindowsScheduler.registerSchedule(settings);
|
||||||
}
|
}
|
||||||
|
|
||||||
public StringProperty stationAddressProperty() {
|
public StringProperty stationAddressProperty() {
|
||||||
|
|||||||
@ -0,0 +1,238 @@
|
|||||||
|
package name.nathanmcrae.numbersstation;
|
||||||
|
|
||||||
|
import java.io.BufferedReader;
|
||||||
|
import java.io.BufferedWriter;
|
||||||
|
import java.io.File;
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.InputStreamReader;
|
||||||
|
import java.nio.file.Files;
|
||||||
|
import java.nio.file.Path;
|
||||||
|
import java.util.logging.Level;
|
||||||
|
import java.util.logging.Logger;
|
||||||
|
import javax.xml.parsers.DocumentBuilder;
|
||||||
|
import javax.xml.parsers.DocumentBuilderFactory;
|
||||||
|
import javax.xml.parsers.ParserConfigurationException;
|
||||||
|
import javax.xml.transform.OutputKeys;
|
||||||
|
import javax.xml.transform.Transformer;
|
||||||
|
import javax.xml.transform.TransformerException;
|
||||||
|
import javax.xml.transform.TransformerFactory;
|
||||||
|
import javax.xml.transform.dom.DOMSource;
|
||||||
|
import javax.xml.transform.stream.StreamResult;
|
||||||
|
import org.w3c.dom.Document;
|
||||||
|
import org.w3c.dom.Element;
|
||||||
|
|
||||||
|
public class WindowsScheduler {
|
||||||
|
private static final Logger logger = Logger.getLogger(Main.class.getName());
|
||||||
|
|
||||||
|
public static Document generateScheduleDocument(String authorName, String userId) throws ParserConfigurationException {
|
||||||
|
DocumentBuilderFactory docFactory = DocumentBuilderFactory.newInstance();
|
||||||
|
DocumentBuilder docBuilder = docFactory.newDocumentBuilder();
|
||||||
|
|
||||||
|
// Root element
|
||||||
|
Document doc = docBuilder.newDocument();
|
||||||
|
Element rootElement = doc.createElement("Task");
|
||||||
|
rootElement.setAttribute("version", "1.2");
|
||||||
|
rootElement.setAttribute("xmlns", "http://schemas.microsoft.com/windows/2004/02/mit/task");
|
||||||
|
doc.appendChild(rootElement);
|
||||||
|
|
||||||
|
// RegistrationInfo element
|
||||||
|
Element registrationInfo = doc.createElement("RegistrationInfo");
|
||||||
|
rootElement.appendChild(registrationInfo);
|
||||||
|
|
||||||
|
Element date = doc.createElement("Date");
|
||||||
|
date.appendChild(doc.createTextNode("2025-01-22T22:09:01.6765321"));
|
||||||
|
registrationInfo.appendChild(date);
|
||||||
|
|
||||||
|
Element author = doc.createElement("Author");
|
||||||
|
author.appendChild(doc.createTextNode(authorName));
|
||||||
|
registrationInfo.appendChild(author);
|
||||||
|
|
||||||
|
Element uri = doc.createElement("URI");
|
||||||
|
uri.appendChild(doc.createTextNode("\\test task"));
|
||||||
|
registrationInfo.appendChild(uri);
|
||||||
|
|
||||||
|
// Triggers element
|
||||||
|
Element triggers = doc.createElement("Triggers");
|
||||||
|
rootElement.appendChild(triggers);
|
||||||
|
|
||||||
|
Element calendarTrigger = doc.createElement("CalendarTrigger");
|
||||||
|
triggers.appendChild(calendarTrigger);
|
||||||
|
|
||||||
|
Element startBoundary = doc.createElement("StartBoundary");
|
||||||
|
startBoundary.appendChild(doc.createTextNode("2025-01-22T22:08:06"));
|
||||||
|
calendarTrigger.appendChild(startBoundary);
|
||||||
|
|
||||||
|
Element enabled = doc.createElement("Enabled");
|
||||||
|
enabled.appendChild(doc.createTextNode("true"));
|
||||||
|
calendarTrigger.appendChild(enabled);
|
||||||
|
|
||||||
|
Element randomDelay = doc.createElement("RandomDelay");
|
||||||
|
randomDelay.appendChild(doc.createTextNode("PT1H"));
|
||||||
|
calendarTrigger.appendChild(randomDelay);
|
||||||
|
|
||||||
|
Element scheduleByDay = doc.createElement("ScheduleByDay");
|
||||||
|
calendarTrigger.appendChild(scheduleByDay);
|
||||||
|
|
||||||
|
Element daysInterval = doc.createElement("DaysInterval");
|
||||||
|
daysInterval.appendChild(doc.createTextNode("1"));
|
||||||
|
scheduleByDay.appendChild(daysInterval);
|
||||||
|
|
||||||
|
// Principals element
|
||||||
|
Element principals = doc.createElement("Principals");
|
||||||
|
rootElement.appendChild(principals);
|
||||||
|
|
||||||
|
Element principal = doc.createElement("Principal");
|
||||||
|
principal.setAttribute("id", "Author");
|
||||||
|
principals.appendChild(principal);
|
||||||
|
|
||||||
|
Element userIdElement = doc.createElement("UserId");
|
||||||
|
userIdElement.appendChild(doc.createTextNode(userId));
|
||||||
|
principal.appendChild(userIdElement);
|
||||||
|
|
||||||
|
Element logonType = doc.createElement("LogonType");
|
||||||
|
logonType.appendChild(doc.createTextNode("InteractiveToken"));
|
||||||
|
principal.appendChild(logonType);
|
||||||
|
|
||||||
|
Element runLevel = doc.createElement("RunLevel");
|
||||||
|
runLevel.appendChild(doc.createTextNode("LeastPrivilege"));
|
||||||
|
principal.appendChild(runLevel);
|
||||||
|
|
||||||
|
// Settings element
|
||||||
|
Element settings = doc.createElement("Settings");
|
||||||
|
rootElement.appendChild(settings);
|
||||||
|
|
||||||
|
Element multipleInstancesPolicy = doc.createElement("MultipleInstancesPolicy");
|
||||||
|
multipleInstancesPolicy.appendChild(doc.createTextNode("IgnoreNew"));
|
||||||
|
settings.appendChild(multipleInstancesPolicy);
|
||||||
|
|
||||||
|
Element disallowStartIfOnBatteries = doc.createElement("DisallowStartIfOnBatteries");
|
||||||
|
disallowStartIfOnBatteries.appendChild(doc.createTextNode("true"));
|
||||||
|
settings.appendChild(disallowStartIfOnBatteries);
|
||||||
|
|
||||||
|
Element stopIfGoingOnBatteries = doc.createElement("StopIfGoingOnBatteries");
|
||||||
|
stopIfGoingOnBatteries.appendChild(doc.createTextNode("true"));
|
||||||
|
settings.appendChild(stopIfGoingOnBatteries);
|
||||||
|
|
||||||
|
Element allowHardTerminate = doc.createElement("AllowHardTerminate");
|
||||||
|
allowHardTerminate.appendChild(doc.createTextNode("true"));
|
||||||
|
settings.appendChild(allowHardTerminate);
|
||||||
|
|
||||||
|
Element startWhenAvailable = doc.createElement("StartWhenAvailable");
|
||||||
|
startWhenAvailable.appendChild(doc.createTextNode("false"));
|
||||||
|
settings.appendChild(startWhenAvailable);
|
||||||
|
|
||||||
|
Element runOnlyIfNetworkAvailable = doc.createElement("RunOnlyIfNetworkAvailable");
|
||||||
|
runOnlyIfNetworkAvailable.appendChild(doc.createTextNode("false"));
|
||||||
|
settings.appendChild(runOnlyIfNetworkAvailable);
|
||||||
|
|
||||||
|
Element idleSettings = doc.createElement("IdleSettings");
|
||||||
|
settings.appendChild(idleSettings);
|
||||||
|
|
||||||
|
Element stopOnIdleEnd = doc.createElement("StopOnIdleEnd");
|
||||||
|
stopOnIdleEnd.appendChild(doc.createTextNode("true"));
|
||||||
|
idleSettings.appendChild(stopOnIdleEnd);
|
||||||
|
|
||||||
|
Element restartOnIdle = doc.createElement("RestartOnIdle");
|
||||||
|
restartOnIdle.appendChild(doc.createTextNode("false"));
|
||||||
|
idleSettings.appendChild(restartOnIdle);
|
||||||
|
|
||||||
|
Element allowStartOnDemand = doc.createElement("AllowStartOnDemand");
|
||||||
|
allowStartOnDemand.appendChild(doc.createTextNode("true"));
|
||||||
|
settings.appendChild(allowStartOnDemand);
|
||||||
|
|
||||||
|
Element enabledSetting = doc.createElement("Enabled");
|
||||||
|
enabledSetting.appendChild(doc.createTextNode("true"));
|
||||||
|
settings.appendChild(enabledSetting);
|
||||||
|
|
||||||
|
Element hidden = doc.createElement("Hidden");
|
||||||
|
hidden.appendChild(doc.createTextNode("false"));
|
||||||
|
settings.appendChild(hidden);
|
||||||
|
|
||||||
|
Element runOnlyIfIdle = doc.createElement("RunOnlyIfIdle");
|
||||||
|
runOnlyIfIdle.appendChild(doc.createTextNode("false"));
|
||||||
|
settings.appendChild(runOnlyIfIdle);
|
||||||
|
|
||||||
|
Element wakeToRun = doc.createElement("WakeToRun");
|
||||||
|
wakeToRun.appendChild(doc.createTextNode("false"));
|
||||||
|
settings.appendChild(wakeToRun);
|
||||||
|
|
||||||
|
Element executionTimeLimit = doc.createElement("ExecutionTimeLimit");
|
||||||
|
executionTimeLimit.appendChild(doc.createTextNode("PT72H"));
|
||||||
|
settings.appendChild(executionTimeLimit);
|
||||||
|
|
||||||
|
Element priority = doc.createElement("Priority");
|
||||||
|
priority.appendChild(doc.createTextNode("7"));
|
||||||
|
settings.appendChild(priority);
|
||||||
|
|
||||||
|
// Actions element
|
||||||
|
Element actions = doc.createElement("Actions");
|
||||||
|
actions.setAttribute("Context", "Author");
|
||||||
|
rootElement.appendChild(actions);
|
||||||
|
|
||||||
|
Element exec = doc.createElement("Exec");
|
||||||
|
actions.appendChild(exec);
|
||||||
|
|
||||||
|
Element command = doc.createElement("Command");
|
||||||
|
command.appendChild(doc.createTextNode("notepad.exe"));
|
||||||
|
exec.appendChild(command);
|
||||||
|
|
||||||
|
return doc;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static String getUserId() throws IOException, InterruptedException {
|
||||||
|
Process process = new ProcessBuilder("whoami", "/user").start();
|
||||||
|
process.waitFor();
|
||||||
|
|
||||||
|
try (BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()))) {
|
||||||
|
String line;
|
||||||
|
while ((line = reader.readLine()) != null) {
|
||||||
|
if (line.contains("S-")) {
|
||||||
|
return line.split("\\s+")[1];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static void registerSchedule(StationSettings settings) {
|
||||||
|
Path tempFile = null;
|
||||||
|
try {
|
||||||
|
String taskName = "numbers-station-main_" + settings.getName();
|
||||||
|
|
||||||
|
Document doc = generateScheduleDocument(System.getProperty("user.name"), getUserId());
|
||||||
|
|
||||||
|
TransformerFactory transformerFactory = TransformerFactory.newInstance();
|
||||||
|
Transformer transformer = transformerFactory.newTransformer();
|
||||||
|
transformer.setOutputProperty(OutputKeys.INDENT, "yes");
|
||||||
|
transformer.setOutputProperty(OutputKeys.OMIT_XML_DECLARATION, "yes");
|
||||||
|
DOMSource source = new DOMSource(doc);
|
||||||
|
|
||||||
|
tempFile = Files.createTempFile("task", ".xml");
|
||||||
|
StreamResult result = new StreamResult(tempFile.toFile());
|
||||||
|
|
||||||
|
try (BufferedWriter writer = Files.newBufferedWriter(tempFile)) {
|
||||||
|
writer.write("<?xml version=\"1.0\"?>\n");
|
||||||
|
transformer.transform(source, new StreamResult(writer));
|
||||||
|
}
|
||||||
|
|
||||||
|
Process process = new ProcessBuilder("schtasks.exe", "/create", "/tn", taskName, "/xml", tempFile.toString()).start();
|
||||||
|
int exitCode = process.waitFor();
|
||||||
|
|
||||||
|
if (exitCode == 0) {
|
||||||
|
logger.info(taskName + " task registered successfully.");
|
||||||
|
} else {
|
||||||
|
logger.info("Failed to register " + taskName + " task. Exit code: " + exitCode);
|
||||||
|
}
|
||||||
|
} catch (ParserConfigurationException | TransformerException | IOException | InterruptedException e) {
|
||||||
|
logger.log(Level.SEVERE, "Exception while registering schedule", e);
|
||||||
|
} finally {
|
||||||
|
if (tempFile != null) {
|
||||||
|
try {
|
||||||
|
Files.delete(tempFile);
|
||||||
|
} catch (IOException e) {
|
||||||
|
logger.log(Level.SEVERE, "Failed to delete temporary file", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
x
Reference in New Issue
Block a user