diff --git a/policy-generator/src/main/java/org/verapdf/tools/policy/generator/PolicyGenerator.java b/policy-generator/src/main/java/org/verapdf/tools/policy/generator/PolicyGenerator.java index 07d3c34..a4262cc 100644 --- a/policy-generator/src/main/java/org/verapdf/tools/policy/generator/PolicyGenerator.java +++ b/policy-generator/src/main/java/org/verapdf/tools/policy/generator/PolicyGenerator.java @@ -37,7 +37,7 @@ public class PolicyGenerator { private Document document; private StringBuilder content; private String fileName; - private String shortFilePath; + private String shortFileName; private InputStream report; private ValidationProfile customProfile; private boolean isLogsEnabled = true; @@ -184,8 +184,7 @@ public void generate() { try { document = (DocumentBuilderFactory.newInstance().newDocumentBuilder()).parse(report); - String[] name = fileName.split("/"); - shortFilePath = name[name.length - 1]; + shortFileName = new File(fileName).getName(); Path policy = Paths.get(fileName.replace(".pdf", ".sch")); NodeList nodeList = document.getElementsByTagName("validationReport"); @@ -216,8 +215,8 @@ private void generateFailPolicy() { String failedRulesToBeReplaced = nodeList.item(0).getAttributes().getNamedItem("failedRules").getNodeValue(); content = new StringBuilder(PolicyHelper.FAIL - .replace("{fileNameToBeReplaced}", shortFilePath) - .replace("ISSUE_NUM", getIssueNumberPart()) + .replace("{fileNameToBeReplaced}", shortFileName) + .replace("ISSUE_NUMBER_PART", getIssueNumberPart()) .replace("{failedRulesToBeReplaced}", failedRulesToBeReplaced)); nodeList = document.getElementsByTagName("rule"); @@ -259,8 +258,8 @@ private void generateFailPolicy() { private void generatePassPolicy() { content = new StringBuilder(PolicyHelper.PASS - .replace("{fileNameToBeReplaced}", shortFilePath) - .replace("ISSUE_NUM", getIssueNumberPart())); + .replace("{fileNameToBeReplaced}", shortFileName) + .replace("ISSUE_NUMBER_PART", getIssueNumberPart())); System.out.println("Policy was created. PDF file is compliant with Validation Profile requirements"); } @@ -280,8 +279,8 @@ private void generateExceptionPolicy() { String outOfMemoryToBeReplaced = node.getNamedItem("outOfMemory").getNodeValue(); String veraExceptionsToBeReplaced = node.getNamedItem("veraExceptions").getNodeValue(); content = new StringBuilder(PolicyHelper.EXC - .replace("{fileNameToBeReplaced}", shortFilePath) - .replace("ISSUE_NUM", getIssueNumberPart()) + .replace("{fileNameToBeReplaced}", shortFileName) + .replace("ISSUE_NUMBER_PART", getIssueNumberPart()) .replace("{exceptionMessageToBeReplaced}", exceptionMessageToBeReplaced) .replace("{exceptionToBeReplaced}", exceptionToBeReplaced) .replace("{totalJobsToBeReplaced}", totalJobsToBeReplaced) @@ -323,7 +322,7 @@ private void appendLogs() { LogInfo logInfo = iterator.next(); content.append(PolicyHelper.LOG .replace("{logToBeReplaced}", logInfo.getMessage().replace("'", "'") - .replace(shortFilePath, ".pdf")) + .replace(shortFileName, ".pdf")) .replace("{occurrencesToBeReplaced}", String.valueOf(logInfo.getOccurrences())) .replace("{levelToBeReplaced}", logInfo.getLevel().getName()));