From 7e0b5ce7b82b85eef37d6f4d10a53c536baae209 Mon Sep 17 00:00:00 2001
From: Maxim <maxplusim@mail.ru>
Date: Thu, 24 Oct 2024 20:58:32 +0300
Subject: [PATCH] Update policy generator

---
 .../policy/generator/PolicyGenerator.java      | 18 +++++++++++++++++-
 .../tools/policy/generator/PolicyHelper.java   |  2 ++
 2 files changed, 19 insertions(+), 1 deletion(-)

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 beb4e43..81fd0cc 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
@@ -227,6 +227,9 @@ private void generateFailPolicy(boolean isTagged, String profileName) {
                 .replace("{fileNameToBeReplaced}", shortFileName)
                 .replace("ISSUE_NUMBER_PART", getIssueNumberPart())
                 .replace("{failedRulesToBeReplaced}", failedRulesToBeReplaced));
+        if (isTagged) {
+            content.append(PolicyHelper.PATTERN_END);
+        }
         nodeList = document.getElementsByTagName("rule");
         int size = nodeList.getLength();
         StringBuilder messageToBeReplaced = new StringBuilder();
@@ -238,8 +241,15 @@ private void generateFailPolicy(boolean isTagged, String profileName) {
                     Integer.parseInt(node.getNamedItem("testNumber").getNodeValue()),
                     Integer.parseInt(node.getNamedItem("failedChecks").getNodeValue())));
         }
+        if (isTagged) {
+            content.append('\n');
+        }
         for (Map.Entry<String, SortedSet<RuleInfo>> map : ruleInfoMap.entrySet()) {
             Iterator<RuleInfo> iterator = map.getValue().iterator();
+            if (isTagged) {
+                content.append(PolicyHelper.PATTERN_START);
+                content.append('\n');
+            }
             content.append(PolicyHelper.FAIL_RULE);
             if (isTagged) {
                 content.append("object != '").append(map.getKey()).append("' or\n            ");
@@ -265,9 +275,15 @@ private void generateFailPolicy(boolean isTagged, String profileName) {
             }
             content.append(PolicyHelper.RULE_END
                     .replace("{messageToBeReplaced}", messageToBeReplaced));
+            if (isTagged) {
+                content.append(PolicyHelper.PATTERN_END);
+                content.append('\n');
+            }
             messageToBeReplaced = new StringBuilder();
         }
-        content.append(PolicyHelper.PATTERN_END);
+        if (!isTagged) {
+            content.append(PolicyHelper.PATTERN_END);
+        }
 
         System.out.println("Policy was created. PDF file is not compliant with " + profileName + " requirements");
     }
diff --git a/policy-generator/src/main/java/org/verapdf/tools/policy/generator/PolicyHelper.java b/policy-generator/src/main/java/org/verapdf/tools/policy/generator/PolicyHelper.java
index 1acf063..e532271 100644
--- a/policy-generator/src/main/java/org/verapdf/tools/policy/generator/PolicyHelper.java
+++ b/policy-generator/src/main/java/org/verapdf/tools/policy/generator/PolicyHelper.java
@@ -41,6 +41,8 @@ public class PolicyHelper {
             "            {messageToBeReplaced}</sch:assert>\n" +
             "        </sch:rule>\n" +
             "\n";
+
+    public static final String PATTERN_START = "    <sch:pattern>\n";
     public static final String PATTERN_END = "    </sch:pattern>\n";
     public static final String LOGS_REPORT = "\n" +
             "    <sch:pattern name = \"Checking the logs\">\n" +