/*
|
* Copyright 2007-2008, Plutext Pty Ltd.
|
*
|
* This file is part of docx4j.
|
|
docx4j is licensed under the Apache License, Version 2.0 (the "License");
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
Unless required by applicable law or agreed to in writing, software
|
distributed under the License is distributed on an "AS IS" BASIS,
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
See the License for the specific language governing permissions and
|
limitations under the License.
|
|
*/
|
|
package cn.flightfeather.supervision.docx4j.simpleDemo;
|
|
|
import org.docx4j.openpackaging.URIHelper;
|
import org.docx4j.openpackaging.exceptions.Docx4JException;
|
import org.docx4j.openpackaging.packages.WordprocessingMLPackage;
|
import org.docx4j.openpackaging.parts.Part;
|
import org.docx4j.openpackaging.parts.relationships.RelationshipsPart;
|
import org.docx4j.relationships.Relationship;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
|
import java.io.File;
|
import java.net.URI;
|
import java.util.ArrayList;
|
import java.util.List;
|
|
|
/**
|
* Remove certain parts from docx,
|
* either the specified docx, or each one in
|
* the specified directory (beware!).
|
*
|
* For this reason, this sample is shipped
|
* pointing to a non-existent directory.
|
*
|
* Which parts are deleted is configurable
|
* to some extent; please consult the code to see.
|
*
|
* @author jharrop
|
*
|
*/
|
public class PartsStrip {
|
|
|
static boolean save = true;
|
static boolean flatOpcXmlOutput = false;
|
static boolean overwriteInputFile = true;
|
|
static String dir = System.getProperty("user.dir") + "/test-strip/";
|
static String file = null; //"blagh"; // set to null to process all docx in dir
|
|
static boolean stripPropertiesParts = true;
|
static boolean keepStyles = true;
|
static boolean defaultToDelete = false;
|
|
private static Logger log = LoggerFactory.getLogger(PartsStrip.class);
|
|
/**
|
* @param args
|
*/
|
public static void main(String[] args) throws Exception {
|
|
if (file==null) {
|
|
List<File> filesToProcess = new ArrayList<File>();
|
File[] filesAndDirs = new File(dir).listFiles();
|
for (File file : filesAndDirs) {
|
if (file.isFile()
|
&& file.getName().endsWith(".docx") ) {
|
filesToProcess.add(file);
|
}
|
}
|
|
for (File file : filesToProcess) {
|
String outputfilepath = null;
|
if (save && overwriteInputFile) {
|
outputfilepath = file.getAbsolutePath();
|
}
|
|
processFile(file, outputfilepath);
|
|
}
|
|
} else {
|
String inputfilepath = dir + file + ".docx";
|
|
// If so, whereto?
|
String outputfilepath = null;
|
if (save) {
|
if (overwriteInputFile) {
|
outputfilepath = inputfilepath;
|
} else if (flatOpcXmlOutput) {
|
outputfilepath = dir + file + "_OUT.xml";
|
} else {
|
outputfilepath = dir + file + "_OUT.docx";
|
}
|
}
|
|
processFile(new File(inputfilepath), outputfilepath);
|
}
|
}
|
|
private static void processFile(File inputfile, String outputfilepath)
|
throws Docx4JException {
|
|
// Open a document from the file system
|
// 1. Load the Package - .docx or Flat OPC .xml
|
WordprocessingMLPackage wordMLPackage = WordprocessingMLPackage.load(inputfile);
|
|
// List the parts by walking the rels tree
|
RelationshipsPart rp = wordMLPackage.getRelationshipsPart();
|
StringBuilder sb = new StringBuilder();
|
printInfo(rp, sb, "");
|
traverseRelationships(wordMLPackage, rp, sb, " ");
|
|
System.out.println(sb.toString());
|
|
if (save) {
|
wordMLPackage.save(new File(outputfilepath));
|
System.out.println("Saved stripped to " + outputfilepath);
|
} else {
|
System.out.println("Stripped parts from " + inputfile.getName() );
|
}
|
}
|
|
public static void printInfo(Part p, StringBuilder sb, String indent) {
|
sb.append("\n" + indent + p.getPartName() + " [" + p.getClass().getName() + "] " );
|
}
|
|
public static void traverseRelationships(WordprocessingMLPackage wordMLPackage,
|
RelationshipsPart rp,
|
StringBuilder sb, String indent)
|
throws Docx4JException {
|
|
// Sme approach as in PartsList sample
|
|
List<Relationship> deletions = new ArrayList<Relationship>();
|
|
for ( Relationship r : rp.getRelationships().getRelationship() ) {
|
|
log.info("For Relationship Id=" + r.getId()
|
+ " Source is " + rp.getSourceP().getPartName()
|
+ ", Target is " + r.getTarget() );
|
|
if (r.getTargetMode() != null
|
&& r.getTargetMode().equals("External") ) {
|
|
sb.append("\n" + indent + "external resource " + r.getTarget()
|
+ " of type " + r.getType() );
|
continue;
|
}
|
|
try {
|
String resolvedPartUri = URIHelper.resolvePartUri(rp.getSourceURI(), new URI(r.getTarget() ) ).toString();
|
resolvedPartUri = resolvedPartUri.substring(1);
|
|
Part part = rp.getPart(r);
|
|
// Or could have done:
|
// Part part = wordMLPackage.getParts().get(new PartName("/" + resolvedPartUri));
|
|
if (part!=null) {
|
printInfo(part, sb, indent);
|
}
|
|
if (part==null) {
|
sb.append("Part " + resolvedPartUri + " not found! \n");
|
} else if ( part instanceof org.docx4j.openpackaging.parts.ThemePart
|
|| part instanceof org.docx4j.openpackaging.parts.WordprocessingML.NumberingDefinitionsPart
|
|| part instanceof org.docx4j.openpackaging.parts.WordprocessingML.FontTablePart) {
|
deletions.add(r );
|
sb.append(".. DELETED" );
|
} else if (part instanceof org.docx4j.openpackaging.parts.WordprocessingML.StyleDefinitionsPart) {
|
|
if (!keepStyles) {
|
deletions.add(r );
|
sb.append(".. DELETED" );
|
} else {
|
sb.append(".. KEEPING" );
|
}
|
|
} else {
|
if (stripPropertiesParts
|
&& ( part instanceof org.docx4j.openpackaging.parts.DocPropsExtendedPart
|
|| part instanceof org.docx4j.openpackaging.parts.DocPropsCorePart
|
|| part instanceof org.docx4j.openpackaging.parts.DocPropsCustomPart
|
|| part instanceof org.docx4j.openpackaging.parts.WordprocessingML.WebSettingsPart
|
|| part instanceof org.docx4j.openpackaging.parts.WordprocessingML.DocumentSettingsPart)) {
|
|
deletions.add(r );
|
sb.append(".. DELETED" );
|
|
} else if ( part instanceof org.docx4j.openpackaging.parts.WordprocessingML.MainDocumentPart
|
|| part instanceof org.docx4j.openpackaging.parts.WordprocessingML.StyleDefinitionsPart
|
|| part instanceof org.docx4j.openpackaging.parts.WordprocessingML.HeaderPart) {
|
sb.append(".. KEEPING" );
|
if (part.getRelationshipsPart()==null) {
|
sb.append(".. no rels" );
|
} else {
|
traverseRelationships(wordMLPackage, part.getRelationshipsPart(), sb, indent + " ");
|
}
|
}
|
else if (defaultToDelete) {
|
// Delete it
|
deletions.add(r );
|
sb.append(".. DELETED" );
|
}
|
}
|
|
} catch (Exception e) {
|
throw new Docx4JException("Failed to add parts from relationships", e);
|
}
|
|
}
|
|
for ( Relationship r : deletions) {
|
rp.removeRelationship(r);
|
}
|
|
}
|
|
|
}
|
