Generate Extra Attributes from Logos Links

Useful, e.g., to add lemma or Louw Nida information when exporting to
OSIS, while using the same CSV files that are used to add them to Logos.

biblemulticonverter-logos/src/main/java/biblemulticonverter/logos/LogosModuleRegistry.java

@@ -22,6 +22,7 @@ public Collection<Module<ExportFormat>> getExportFormats() {
 		List<Module<ExportFormat>> result = new ArrayList<Module<ExportFormat>>();
 		result.add(new Module<ExportFormat>("LogosVersificationDetector", "Detect what Logos versification to use best for exporting a module", new String[0], LogosVersificationDetector.class));
 		result.add(new Module<ExportFormat>("LogosRenumberedDiffable", "Renumber named verses for Logos before exporting as Diffable.", LogosRenumberedDiffable.HELP_TEXT, LogosRenumberedDiffable.class));
+		result.add(new Module<ExportFormat>("AugmentLogosLinks", "Add values of Logos links to Extra Attributes of a Bible", AugmentLogosLinks.HELP_TEXT, AugmentLogosLinks.class));
 		result.add(new Module<ExportFormat>("LogosHTML", "HTML Export format for Logos Bible Software", LogosHTML.HELP_TEXT, LogosHTML.class));
 		return result;
 	}

biblemulticonverter-logos/src/main/java/biblemulticonverter/logos/format/AugmentLogosLinks.java

@@ -0,0 +1,96 @@
+package biblemulticonverter.logos.format;
+
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+
+import biblemulticonverter.Main;
+import biblemulticonverter.ModuleRegistry.Module;
+import biblemulticonverter.data.Bible;
+import biblemulticonverter.data.Book;
+import biblemulticonverter.data.Chapter;
+import biblemulticonverter.data.FormattedText;
+import biblemulticonverter.data.FormattedText.ExtraAttributePriority;
+import biblemulticonverter.data.FormattedText.Visitor;
+import biblemulticonverter.data.Verse;
+import biblemulticonverter.data.Versification;
+import biblemulticonverter.format.ExportFormat;
+
+public class AugmentLogosLinks implements ExportFormat {
+
+	public static final String[] HELP_TEXT = {
+			"Add values of Logos links to Extra Attributes of a Bible",
+			"",
+			"Usage: AugmentLogosLinks [<prefix>=<category>:<key>:<value>...] -- <ExportFormat> [<ExportArgs>...]",
+			"",
+			"Useful, for example, to fill 'osisgrammar:lemma:lemma' or 'osisgrammar:morph:louwnida' before exporting to OSIS."
+	};
+
+	@Override
+	public void doExport(Bible bible, String... exportArgs) throws Exception {
+		int formatArg = 0;
+		LogosLinksGenerator linksGenerator = new LogosLinksGenerator();
+		List<String[]> mappingRules = new ArrayList<>();
+		while (!exportArgs[formatArg].equals("--")) {
+			int pos = exportArgs[formatArg].indexOf("=");
+			String[] parts = exportArgs[formatArg].substring(pos + 1).split(":");
+			if (parts.length != 3)
+				throw new IOException("Invalid rule: " + exportArgs[formatArg]);
+			mappingRules.add(new String[] { exportArgs[formatArg].substring(0, pos), parts[0], parts[1], parts[2] });
+			formatArg++;
+		}
+		formatArg++;
+
+		for (Book book : bible.getBooks()) {
+			int cnum = 0;
+			for (Chapter chapter : book.getChapters()) {
+				cnum++;
+				List<Verse> verses = chapter.getVerses();
+				for (int i = 0; i < verses.size(); i++) {
+					Verse v1 = verses.get(i);
+					Verse v2 = new Verse(v1.getNumber());
+					v1.accept(new AugmentLogosLinksVisitor(v2.getAppendVisitor(), new Versification.Reference(book.getId(), cnum, v1.getNumber()), mappingRules, linksGenerator));
+					v2.finished();
+					verses.set(i, v2);
+				}
+			}
+		}
+		Module<ExportFormat> exportModule = Main.exportFormats.get(exportArgs[formatArg]);
+		ExportFormat exportFormat = exportModule.getImplementationClass().newInstance();
+		exportFormat.doExport(bible, Arrays.copyOfRange(exportArgs, formatArg + 1, exportArgs.length));
+	}
+
+	private static class AugmentLogosLinksVisitor extends FormattedText.VisitorAdapter<RuntimeException> {
+
+		private final Versification.Reference reference;
+		private final List<String[]> mappingRules;
+		private final LogosLinksGenerator linksGenerator;
+
+		private AugmentLogosLinksVisitor(Visitor<RuntimeException> next, Versification.Reference reference, List<String[]> mappingRules, LogosLinksGenerator linksGenerator) throws RuntimeException {
+			super(next);
+			this.reference = reference;
+			this.mappingRules = mappingRules;
+			this.linksGenerator = linksGenerator;
+		}
+
+		@Override
+		protected Visitor<RuntimeException> wrapChildVisitor(Visitor<RuntimeException> childVisitor) throws RuntimeException {
+			return new AugmentLogosLinksVisitor(childVisitor, reference, mappingRules, linksGenerator);
+		}
+
+		@Override
+		public Visitor<RuntimeException> visitGrammarInformation(char[] strongsPrefixes, int[] strongs, String[] rmac, int[] sourceIndices) throws RuntimeException {
+			Visitor<RuntimeException> vv = getVisitor().visitGrammarInformation(strongsPrefixes, strongs, rmac, sourceIndices);
+			List<String> links = linksGenerator.generateLinks(reference.getBook().isNT(), reference, strongsPrefixes, strongs, rmac, sourceIndices);
+			for (String[] rule : mappingRules) {
+				for (String link : links) {
+					if (link.startsWith(rule[0])) {
+						vv.visitExtraAttribute(ExtraAttributePriority.SKIP, rule[1], rule[2], rule[3]).visitText(link.substring(rule[0].length()));
+					}
+				}
+			}
+			return wrapChildVisitor(vv);
+		}
+	}
+}