aboutsummaryrefslogtreecommitdiff
path: root/src/main
diff options
context:
space:
mode:
authorMichael Williamson <mike@zwobble.org>2016-04-17 18:57:29 +0100
committerMichael Williamson <mike@zwobble.org>2016-04-17 18:58:12 +0100
commit52140d52210eda0f25ecab9f89b5c73473e9d8e0 (patch)
treecb383edb71e44b8ae7b2b0d0890f13574d1922f0 /src/main
parent0f9ef005f063198ea2e16ca36beaead4bd5e3824 (diff)
Add CLI for .NET version
Diffstat (limited to 'src/main')
-rw-r--r--src/main/java/org/zwobble/mammoth/DocumentConverter.java83
-rw-r--r--src/main/java/org/zwobble/mammoth/internal/InternalDocumentConverter.java110
-rw-r--r--src/main/java/org/zwobble/mammoth/internal/docx/ZippedDocxFile.java5
3 files changed, 120 insertions, 78 deletions
diff --git a/src/main/java/org/zwobble/mammoth/DocumentConverter.java b/src/main/java/org/zwobble/mammoth/DocumentConverter.java
index f7112f4..3a21fed 100644
--- a/src/main/java/org/zwobble/mammoth/DocumentConverter.java
+++ b/src/main/java/org/zwobble/mammoth/DocumentConverter.java
@@ -1,31 +1,11 @@
package org.zwobble.mammoth;
-import org.zwobble.mammoth.internal.conversion.DocumentToHtml;
+import org.zwobble.mammoth.internal.InternalDocumentConverter;
import org.zwobble.mammoth.internal.conversion.DocumentToHtmlOptions;
-import org.zwobble.mammoth.internal.documents.DocumentElement;
-import org.zwobble.mammoth.internal.documents.HasChildren;
-import org.zwobble.mammoth.internal.documents.Paragraph;
-import org.zwobble.mammoth.internal.documents.Text;
-import org.zwobble.mammoth.internal.docx.DocxFile;
-import org.zwobble.mammoth.internal.docx.InMemoryDocxFile;
-import org.zwobble.mammoth.internal.docx.ZippedDocxFile;
-import org.zwobble.mammoth.internal.html.Html;
-import org.zwobble.mammoth.internal.results.InternalResult;
-import org.zwobble.mammoth.internal.util.Casts;
-import org.zwobble.mammoth.internal.util.PassThroughException;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
-import java.nio.file.Path;
-import java.util.List;
-import java.util.Optional;
-import java.util.function.Function;
-import java.util.zip.ZipFile;
-
-import static org.zwobble.mammoth.internal.docx.DocumentReader.readDocument;
-import static org.zwobble.mammoth.internal.util.Iterables.lazyMap;
-import static org.zwobble.mammoth.internal.util.Lists.list;
public class DocumentConverter {
private final DocumentToHtmlOptions options;
@@ -77,26 +57,14 @@ public class DocumentConverter {
* means that relative paths to other files, such as images, cannot be resolved.
*/
public Result<String> convertToHtml(InputStream stream) throws IOException {
- return PassThroughException.unwrap(() ->
- withDocxFile(stream, zipFile ->
- convertToHtml(Optional.empty(), zipFile)).toResult());
+ return new InternalDocumentConverter(options).convertToHtml(stream).toResult();
}
/**
* Converts {@code file} into an HTML string.
*/
public Result<String> convertToHtml(File file) throws IOException {
- return PassThroughException.unwrap(() ->
- withDocxFile(file, zipFile ->
- convertToHtml(Optional.of(file.toPath()), zipFile)).toResult());
- }
-
- private InternalResult<String> convertToHtml(Optional<Path> path, DocxFile zipFile) {
- return readDocument(path, zipFile)
- .flatMap(nodes -> DocumentToHtml.convertToHtml(nodes, options))
- .map(Html::stripEmpty)
- .map(Html::collapse)
- .map(Html::write);
+ return new InternalDocumentConverter(options).convertToHtml(file).toResult();
}
/**
@@ -105,9 +73,7 @@ public class DocumentConverter {
* Each paragraph is followed by two newlines.
*/
public Result<String> extractRawText(InputStream stream) throws IOException {
- return PassThroughException.unwrap(() ->
- withDocxFile(stream, zipFile ->
- extractRawText(Optional.empty(), zipFile)).toResult());
+ return new InternalDocumentConverter(options).extractRawText(stream).toResult();
}
/**
@@ -116,45 +82,6 @@ public class DocumentConverter {
* Each paragraph is followed by two newlines.
*/
public Result<String> extractRawText(File file) throws IOException {
- return PassThroughException.unwrap(() ->
- withDocxFile(file, zipFile ->
- extractRawText(Optional.of(file.toPath()), zipFile)).toResult());
- }
-
- private InternalResult<String> extractRawText(Optional<Path> path, DocxFile zipFile) {
- return readDocument(path, zipFile)
- .map(DocumentConverter::extractRawTextOfChildren);
- }
-
- private static <T> T withDocxFile(File file, Function<DocxFile, T> function) throws IOException {
- try (DocxFile zipFile = new ZippedDocxFile(new ZipFile(file))) {
- return function.apply(zipFile);
- }
- }
-
- private static <T> T withDocxFile(InputStream stream, Function<DocxFile, T> function) throws IOException {
- try (DocxFile zipFile = InMemoryDocxFile.fromStream(stream)) {
- return function.apply(zipFile);
- }
- }
-
- private static String extractRawTextOfChildren(HasChildren parent) {
- return extractRawText(parent.getChildren());
- }
-
- private static String extractRawText(List<DocumentElement> nodes) {
- return String.join("", lazyMap(nodes, node -> extractRawText(node)));
- }
-
- private static String extractRawText(DocumentElement node) {
- return Casts.tryCast(Text.class, node)
- .map(Text::getValue)
- .orElseGet(() -> {
- List<DocumentElement> children = Casts.tryCast(HasChildren.class, node)
- .map(HasChildren::getChildren)
- .orElse(list());
- String suffix = node instanceof Paragraph ? "\n\n" : "";
- return extractRawText(children) + suffix;
- });
+ return new InternalDocumentConverter(options).extractRawText(file).toResult();
}
}
diff --git a/src/main/java/org/zwobble/mammoth/internal/InternalDocumentConverter.java b/src/main/java/org/zwobble/mammoth/internal/InternalDocumentConverter.java
new file mode 100644
index 0000000..4cc0e28
--- /dev/null
+++ b/src/main/java/org/zwobble/mammoth/internal/InternalDocumentConverter.java
@@ -0,0 +1,110 @@
+package org.zwobble.mammoth.internal;
+
+import org.zwobble.mammoth.internal.conversion.DocumentToHtml;
+import org.zwobble.mammoth.internal.conversion.DocumentToHtmlOptions;
+import org.zwobble.mammoth.internal.documents.DocumentElement;
+import org.zwobble.mammoth.internal.documents.HasChildren;
+import org.zwobble.mammoth.internal.documents.Paragraph;
+import org.zwobble.mammoth.internal.documents.Text;
+import org.zwobble.mammoth.internal.docx.DocxFile;
+import org.zwobble.mammoth.internal.docx.InMemoryDocxFile;
+import org.zwobble.mammoth.internal.docx.ZippedDocxFile;
+import org.zwobble.mammoth.internal.html.Html;
+import org.zwobble.mammoth.internal.results.InternalResult;
+import org.zwobble.mammoth.internal.util.PassThroughException;
+
+import java.io.File;
+import java.io.IOException;
+import java.io.InputStream;
+import java.nio.file.Path;
+import java.util.List;
+import java.util.Optional;
+import java.util.function.Function;
+
+import static org.zwobble.mammoth.internal.docx.DocumentReader.readDocument;
+import static org.zwobble.mammoth.internal.util.Casts.tryCast;
+import static org.zwobble.mammoth.internal.util.Iterables.lazyMap;
+import static org.zwobble.mammoth.internal.util.Lists.list;
+
+public class InternalDocumentConverter {
+ private final DocumentToHtmlOptions options;
+
+ public InternalDocumentConverter(DocumentToHtmlOptions options) {
+ this.options = options;
+ }
+
+ public InternalResult<String> convertToHtml(InputStream stream) throws IOException {
+ return PassThroughException.unwrap(() ->
+ withDocxFile(stream, zipFile ->
+ convertToHtml(Optional.empty(), zipFile)));
+ }
+
+ public InternalResult<String> convertToHtml(File file) throws IOException {
+ return PassThroughException.unwrap(() ->
+ withDocxFile(file, zipFile ->
+ convertToHtml(Optional.of(file.toPath()), zipFile)));
+ }
+
+ private InternalResult<String> convertToHtml(Optional<Path> path, DocxFile zipFile) {
+ return readDocument(path, zipFile)
+ .flatMap(nodes -> DocumentToHtml.convertToHtml(nodes, options))
+ .map(Html::stripEmpty)
+ .map(Html::collapse)
+ .map(Html::write);
+ }
+
+ public InternalResult<String> extractRawText(InputStream stream) throws IOException {
+ return PassThroughException.unwrap(() ->
+ withDocxFile(stream, zipFile ->
+ extractRawText(Optional.empty(), zipFile)));
+ }
+
+ public InternalResult<String> extractRawText(File file) throws IOException {
+ return PassThroughException.unwrap(() ->
+ withDocxFile(file, zipFile ->
+ extractRawText(Optional.of(file.toPath()), zipFile)));
+ }
+
+ private InternalResult<String> extractRawText(Optional<Path> path, DocxFile zipFile) {
+ return readDocument(path, zipFile)
+ .map(InternalDocumentConverter::extractRawTextOfChildren);
+ }
+
+ private static <T> T withDocxFile(File file, Function<DocxFile, T> function) throws IOException {
+ DocxFile zipFile = new ZippedDocxFile(file);
+ try {
+ return function.apply(zipFile);
+ } finally {
+ zipFile.close();
+ }
+ }
+
+ private static <T> T withDocxFile(InputStream stream, Function<DocxFile, T> function) throws IOException {
+ DocxFile zipFile = InMemoryDocxFile.fromStream(stream);
+ try {
+ return function.apply(zipFile);
+ } finally {
+ zipFile.close();
+ }
+ }
+
+ private static String extractRawTextOfChildren(HasChildren parent) {
+ return extractRawText(parent.getChildren());
+ }
+
+ private static String extractRawText(List<DocumentElement> nodes) {
+ return String.join("", lazyMap(nodes, node -> extractRawText(node)));
+ }
+
+ private static String extractRawText(DocumentElement node) {
+ return tryCast(Text.class, node)
+ .map(Text::getValue)
+ .orElseGet(() -> {
+ List<DocumentElement> children = tryCast(HasChildren.class, node)
+ .map(HasChildren::getChildren)
+ .orElse(list());
+ String suffix = tryCast(Paragraph.class, node).map(paragraph -> "\n\n").orElse("");
+ return extractRawText(children) + suffix;
+ });
+ }
+}
diff --git a/src/main/java/org/zwobble/mammoth/internal/docx/ZippedDocxFile.java b/src/main/java/org/zwobble/mammoth/internal/docx/ZippedDocxFile.java
index d263578..559be1b 100644
--- a/src/main/java/org/zwobble/mammoth/internal/docx/ZippedDocxFile.java
+++ b/src/main/java/org/zwobble/mammoth/internal/docx/ZippedDocxFile.java
@@ -1,5 +1,6 @@
package org.zwobble.mammoth.internal.docx;
+import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.util.Optional;
@@ -13,6 +14,10 @@ public class ZippedDocxFile implements DocxFile {
this.zipFile = zipFile;
}
+ public ZippedDocxFile(File file) throws IOException {
+ this(new ZipFile(file));
+ }
+
@Override
public Optional<InputStream> tryGetInputStream(String name) throws IOException {
ZipEntry entry = zipFile.getEntry(name);