Improved .xlsm support - still not quite there though

git-svn-id: https://svn.apache.org/repos/asf/poi/branches/ooxml@680857 13f79535-47bb-0310-9956-ffa450edef68

src/ooxml/java/org/apache/poi/xssf/usermodel/XSSFWorkbook.java

@@ -80,6 +80,12 @@ public class XSSFWorkbook extends POIXMLDocument implements Workbook {
 			"/xl/workbook.xml",
 			null
 	);
+	public static final XSSFRelation MACROS_WORKBOOK = new XSSFRelation(
+			"application/vnd.ms-excel.sheet.macroEnabled.main+xml",
+			"http://schemas.openxmlformats.org/officeDocument/2006/relationships/officeDocument",
+			"/xl/workbook.xml",
+			null
+	);
 	public static final XSSFRelation WORKSHEET = new XSSFRelation(
 			"application/vnd.openxmlformats-officedocument.spreadsheetml.worksheet+xml",
 			"http://schemas.openxmlformats.org/officeDocument/2006/relationships/worksheet",
@@ -255,6 +261,9 @@ public class XSSFWorkbook extends POIXMLDocument implements Workbook {
 		}
 	}
 
+	/** Are we a normal workbook, or a macro enabled one? */
+	private boolean isMacroEnabled = false;
+	
     private CTWorkbook workbook;
     
     private List<XSSFSheet> sheets = new LinkedList<XSSFSheet>();
@@ -288,6 +297,10 @@ public class XSSFWorkbook extends POIXMLDocument implements Workbook {
             WorkbookDocument doc = WorkbookDocument.Factory.parse(getCorePart().getInputStream());
             this.workbook = doc.getWorkbook();
             
+            // Are we macro enabled, or just normal?
+            isMacroEnabled = 
+            		getCorePart().getContentType().equals(MACROS_WORKBOOK.getContentType());
+            
             try {
 	            // Load shared strings
 	            this.sharedStringSource = (SharedStringSource)
@@ -660,6 +673,14 @@ public class XSSFWorkbook extends POIXMLDocument implements Workbook {
     	return sr;
 	}
     
+    /**
+     * Are we a normal workbook (.xlsx), or a 
+     *  macro enabled workbook (.xlsm)?
+     */
+    public boolean isMacroEnabled() {
+    	return isMacroEnabled;
+    }
+
 	public void insertChartRecord() {
         // TODO Auto-generated method stub
 
@@ -764,16 +785,21 @@ public class XSSFWorkbook extends POIXMLDocument implements Workbook {
     }
 
     public void write(OutputStream stream) throws IOException {
+    	// What kind of workbook are we?
+    	XSSFRelation workbookRelation = WORKBOOK;
+    	if(isMacroEnabled) {
+    		workbookRelation = MACROS_WORKBOOK;
+    	}
 
         try {
             // Create a package referring the temp file.
             Package pkg = Package.create(stream);
             // Main part
-            PackagePartName corePartName = PackagingURIHelper.createPartName("/xl/workbook.xml");
+            PackagePartName corePartName = PackagingURIHelper.createPartName(workbookRelation.getDefaultFileName());
             // Create main part relationship
             pkg.addRelationship(corePartName, TargetMode.INTERNAL, PackageRelationshipTypes.CORE_DOCUMENT, "rId1");
             // Create main document part
-            PackagePart corePart = pkg.createPart(corePartName, WORKBOOK.getContentType());
+            PackagePart corePart = pkg.createPart(corePartName, workbookRelation.getContentType());
             OutputStream out;
 
             XmlOptions xmlOptions = new XmlOptions();
@@ -840,9 +866,10 @@ public class XSSFWorkbook extends POIXMLDocument implements Workbook {
             	}
             }
             
-            // VBA Macros
+            // Macro related bits
+            if(isMacroEnabled) {
+	            // Copy VBA Macros if present
 	            if(VBA_MACROS.exists( getCorePart() )) {
-            	// Copy over
 	            	try {
 		            	XSSFModel vba = VBA_MACROS.load(getCorePart());
 		            	VBA_MACROS.save(vba, corePart);
@@ -850,6 +877,7 @@ public class XSSFWorkbook extends POIXMLDocument implements Workbook {
 	            		throw new RuntimeException("Unable to copy vba macros over", e);
 	            	}
 	            }
+            }
 
             // Now we can write out the main Workbook, with
             //  the correct references to the other parts

src/ooxml/testcases/org/apache/poi/xssf/usermodel/TestXSSFBugs.java

@@ -20,6 +20,7 @@ package org.apache.poi.xssf.usermodel;
 import java.io.ByteArrayInputStream;
 import java.io.ByteArrayOutputStream;
 import java.io.File;
+import java.io.FileOutputStream;
 
 import junit.framework.TestCase;
 
@@ -85,6 +86,7 @@ public class TestXSSFBugs extends TestCase {
 		Package pkg = Package.open(getFilePath("45431.xlsm"));
 		XSSFWorkbook wb = new XSSFWorkbook(pkg);
 		
+		// Check the various macro related bits can be found
 		PackagePart vba = pkg.getPart(
 				PackagingURIHelper.createPartName("/xl/vbaProject.bin")
 		);
@@ -106,6 +108,10 @@ public class TestXSSFBugs extends TestCase {
 		);
 		assertNotNull(vba);
 		
+		FileOutputStream fout = new FileOutputStream("/tmp/foo.xlsm");
+		nwb.write(fout);
+		fout.close();
+		
 		// For testing with excel
 //		FileOutputStream fout = new FileOutputStream("/tmp/foo.xlsm");
 //		nwb.write(fout);