Skip to content

Commit

Permalink
Clean up, add error reporting and command line support
Browse files Browse the repository at this point in the history
  • Loading branch information
@FloBoJa
FloBoJa committed Apr 15, 2024
1 parent 3b27b84 commit 14cf0dc
Show file tree
Hide file tree
Showing 2 changed files with 101 additions and 60 deletions.
28 changes: 20 additions & 8 deletions ...tor.retriever.core/src/org/palladiosimulator/retriever/core/cli/RetrieverApplication.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
import java.nio.file.Paths;
import java.util.Arrays;
import java.util.Locale;
import java.util.Optional;
import java.util.Set;
import java.util.stream.Collectors;

Expand Down Expand Up @@ -36,7 +37,8 @@ private static Options createOptions(final Set<String> availableRuleIDs) {
.addRequiredOption("r", "rules", true,
"Supported rules for reverse engineering: " + String.join(", ", availableRuleIDs));

options.addOption("x", "rules-directory", true, "Path to the directory with additional project specific rules.");
options.addOption("x", "rules-directory", true,
"Path to the directory with additional project specific rules.");

options.addOption("h", "help", false, "Print this help message.");

Expand Down Expand Up @@ -101,7 +103,7 @@ public Object start(final IApplicationContext context) throws Exception {
return -1;
}

if(cmd.hasOption("rules-directory")) {
if (cmd.hasOption("rules-directory")) {
try {
configuration.setRulesFolder(URI.createFileURI(URI.decode(Paths.get(cmd.getOptionValue("x"))
.toAbsolutePath()
Expand All @@ -111,13 +113,12 @@ public Object start(final IApplicationContext context) throws Exception {
System.err.println("Invalid rules path: " + e.getMessage());
return -1;
}
}

// Enable all discoverers, in case a selected rule depends on them.
// TODO: This is unnecessary once rule dependencies are in place
final ServiceConfiguration<Discoverer> discovererConfig = configuration.getConfig(Discoverer.class);
for (final Discoverer discoverer : new DiscovererCollection().getServices()) {
discovererConfig.select(discoverer);
// Enable all discoverers, in case the project-specific rules depend on them.
final ServiceConfiguration<Discoverer> discovererConfig = configuration.getConfig(Discoverer.class);
for (final Discoverer discoverer : new DiscovererCollection().getServices()) {
discovererConfig.select(discoverer);
}
}

final ServiceConfiguration<Rule> ruleConfig = configuration.getConfig(Rule.class);
Expand All @@ -137,6 +138,17 @@ public Object start(final IApplicationContext context) throws Exception {
ruleConfig.select(rule);
}

if (cmd.hasOption("rules-directory")) {
final Optional<Rule> projectSpecificRulesProxy = availableRules.stream()
.filter(x -> "org.palladiosimulator.retriever.extraction.rules.project_specific".equals(x.getID()))
.findAny();
if (projectSpecificRulesProxy.isEmpty()) {
System.err.println("Internal error: could not find project-specific rules proxy");
return -1;
}
ruleConfig.select(projectSpecificRulesProxy.get());
}

new RetrieverJob(configuration).execute(new NullProgressMonitor());

return 0;
Expand Down
133 changes: 81 additions & 52 deletions ...ules/src/org/palladiosimulator/retriever/extraction/rules/ProjectSpecificRulesProxy.xtend
View file
Original file line number Diff line number Diff line change
Expand Up @@ -12,12 +12,9 @@ import java.io.File
import java.util.HashSet
import org.eclipse.xtend.core.XtendInjectorSingleton
import javax.tools.ToolProvider
import javax.tools.ForwardingJavaFileManager
import javax.tools.JavaFileManager
import org.eclipse.core.runtime.Platform
import org.eclipse.emf.common.CommonPlugin
import org.eclipse.emf.ecore.plugin.EcorePlugin
import org.eclipse.core.runtime.FileLocator
import java.io.IOException

class ProjectSpecificRulesProxy implements Rule {

Expand All @@ -28,68 +25,100 @@ class ProjectSpecificRulesProxy implements Rule {
Optional<Rule> innerRule = Optional.empty;

override create(RetrieverConfiguration config, RetrieverBlackboard blackboard) {
val rulePath = new File(config.getConfig(Rule).getConfig(RULE_ID, RULE_PATH_KEY))
val outputPath = new File(rulePath.parentFile.toPath.resolve("xtend-gen").toString)
outputPath.mkdirs

val rulesDirectory = getConfiguredRulesDirectory(config)

val xtendGenDirectory = new File(rulesDirectory.parentFile.toPath.resolve("xtend-gen").toString)
if (!xtendGenDirectory.mkdirs) {
throw new IOException("Could not create intermediate compilation directory at " + xtendGenDirectory);
}

compileXtend(rulesDirectory, xtendGenDirectory)
compileJava(xtendGenDirectory)

val classDirectoryURL = xtendGenDirectory.toURI.toURL
val classLoader = new URLClassLoader(#[classDirectoryURL], class.classLoader)
val ruleInstance = try {
val loadedClass = classLoader.loadClass(LOADED_CLASS_NAME)
loadedClass.getConstructor().newInstance()
} catch (ClassNotFoundException exception) {
throw new IllegalArgumentException(
"Could not find project-specific rule. It must have the fully qualified name " + LOADED_CLASS_NAME,
exception)
}

innerRule = Optional.of(ruleInstance as Rule)
innerRule.get().create(config, blackboard)
}

def getConfiguredRulesDirectory(RetrieverConfiguration config) {
val configuredValue = config.getConfig(Rule).getConfig(RULE_ID, RULE_PATH_KEY)
if (configuredValue !== null) {
return new File(configuredValue)
} else if (config.getOutputFolder() !== null) {
val localURI = CommonPlugin.asLocalURI(config.getOutputFolder())
return new File(localURI.devicePath())
}
throw new IllegalArgumentException("No path for project-specific rules is specified");
}

def compileXtend(File inputDirectory, File outputDirectory) {
val compiler = XtendInjectorSingleton.INJECTOR.getInstance(XtendBatchCompiler)
compiler.sourcePath = rulePath.toString
compiler.outputPath = outputPath.toString
compiler.sourcePath = inputDirectory.toString
compiler.outputPath = outputDirectory.toString
compiler.currentClassLoader = class.classLoader
compiler.useCurrentClassLoaderAsParent = true
if (compiler.compile()) {
val javaFile = outputPath.toPath.resolve("org").resolve("palladiosimulator").resolve("retriever").resolve(
"extraction").resolve("rules").resolve("ProjectSpecificRules.java")
val javaCompiler = ToolProvider.systemJavaCompiler
val standardFileManager = javaCompiler.getStandardFileManager(null, null, null)
val workingDirectory = new File(Platform.installLocation.URL.path).toPath
val jarpath = workingDirectory.resolve("plugins").resolve("*")
val classpath = buildClassPath(jarpath.toString)
val javaCompilerOptions = #["-classpath", classpath]
val javaFileObject = standardFileManager.getJavaFileObjects(javaFile)
val compilationTask = javaCompiler.getTask(null, standardFileManager, null, javaCompilerOptions, null, javaFileObject)
compilationTask.call
val outputUrl = outputPath.toURI.toURL
val classLoader = new URLClassLoader(#[outputUrl], class.classLoader)
val loadedClass = classLoader.loadClass(LOADED_CLASS_NAME)
val ruleInstance = loadedClass.getConstructor().newInstance()
// TODO: handle failures
innerRule = Optional.of(Rule.cast(ruleInstance))
if (!compiler.compile()) {
throw new IllegalArgumentException("Could not compile xtend files located in " + inputDirectory)
}
}

def compileJava(File inOutDirectory) {
val sourcePath = inOutDirectory.toPath.resolve("org").resolve("palladiosimulator").resolve("retriever").resolve(
"extraction").resolve("rules").resolve("ProjectSpecificRules.java")

if (innerRule.isPresent()) {
innerRule.get().create(config, blackboard)
} else {
Rule.super.create(config, blackboard)
// TODO: Fail or log error
val workingDirectory = new File(Platform.installLocation.URL.path).toPath
val jarpath = workingDirectory.resolve("plugins").resolve("*")
val classpath = buildClassPath(jarpath.toString)

val compiler = ToolProvider.systemJavaCompiler
val fileManager = compiler.getStandardFileManager(null, null, null)
val compilerOptions = #["-classpath", classpath]
val sourceFile = fileManager.getJavaFileObjects(sourcePath)
val compilationTask = compiler.getTask(null, fileManager, null, compilerOptions, null, sourceFile)

// This call saves the compiled class files into the source directory.
if (!compilationTask.call()) {
throw new IllegalArgumentException("Could not compile java files generated from xtend files located in " +
inOutDirectory)
}
}

/**
* This function builds a classpath from the passed Strings
*
* @param paths classpath elements
* @return returns the complete classpath with wildcards expanded
*/
def buildClassPath(String... paths) {
val sb = new StringBuilder()
for (path : paths) {
if (path.endsWith("*")) {
val subPath = path.substring(0, path.length - 1)
val pathFile = new File(subPath)
for (file : pathFile.listFiles) {
if (file.isFile && file.name.endsWith(".jar")) {
sb.append(subPath)
sb.append(file.name)
sb.append(System.getProperty("path.separator"))
}
}
} else {
sb.append(path)
sb.append(System.getProperty("path.separator"))
}
}
return sb.toString
// Adapted from https://stackoverflow.com/a/22989029
val sb = new StringBuilder()
for (path : paths) {
if (path.endsWith("*")) {
val subPath = path.substring(0, path.length - 1)
val pathFile = new File(subPath)
for (file : pathFile.listFiles) {
if (file.isFile && file.name.endsWith(".jar")) {
sb.append(subPath)
sb.append(file.name)
sb.append(System.getProperty("path.separator"))
}
}
} else {
sb.append(path)
sb.append(System.getProperty("path.separator"))
}
}
return sb.toString
}

override isBuildRule() {
Expand Down

0 comments on commit 14cf0dc

Please sign in to comment.