0b85bfc9d546e1fd232d8a7b86ffe30d26d2ec0f
[vamsas.git] / src / org / vamsas / test / simpleclient / VamsasArchive.java
1 package org.vamsas.test.simpleclient;
2
3 import java.io.BufferedInputStream;
4 import java.io.File;
5 import java.io.InputStream;
6 import java.io.InputStreamReader;
7 import java.io.ObjectOutputStream;
8 import java.io.PrintWriter;
9
10 import org.apache.commons.logging.Log;
11 import org.apache.commons.logging.LogFactory;
12 import org.vamsas.client.simpleclient.*;
13 import org.vamsas.objects.core.AppDataChoice;
14 import org.vamsas.objects.core.ApplicationData;
15 import org.vamsas.objects.core.User;
16 import org.vamsas.objects.core.VAMSAS;
17 import org.vamsas.objects.core.VamsasDocument;
18 import org.vamsas.test.objects.Core;
19
20 public class VamsasArchive {
21   /**
22    * test the org.vamsas.simpleclient.vamsasArchive class
23    */
24   static Log log = LogFactory.getLog(VamsasArchive.class);
25   
26   public static void main(String args[]) {
27     
28     try {
29       File av;
30       if (args.length>0)
31         av = new File(args[0]);
32       else
33         av = new File("test/vamsas.zip");
34       log.info("Opening archive "+av);
35       org.vamsas.client.simpleclient.VamsasArchive varchive = new org.vamsas.client.simpleclient.VamsasArchive(av, true);
36       
37       VAMSAS[] roots = (VAMSAS[]) varchive.getOriginalRoots();
38       
39       if (roots!=null) {
40         log.info("Report on Original roots in archive:");
41         ArchiveReports.rootReport(roots, true);
42       }
43       log.info("Getting current vamsas document.");
44       VamsasDocument doc = varchive.getVamsasDocument();
45       ArchiveReports.reportDocument(doc, varchive.getOriginalArchiveReader(), true); // not modified document so references will still be valid
46       // do some stuff
47       log.info("Retrieving backup");
48       File backup = varchive.backupFile();
49       if (backup==null)
50         log.info(av+" is a New Archive.");
51       else 
52         log.info(av+" has been backed up as "+backup);
53       if (backup!=null) {
54         log.info("Backup "+backup+" is being written to.");
55         // open another and...
56         ApplicationData appdata = new ApplicationData();
57         appdata.setName("org.vamsas.test.simpleclient.VamsasArchive");
58         appdata.setAppDataChoice(new AppDataChoice());
59         appdata.getAppDataChoice().setData(new String("this is some test data.").getBytes());
60         User apuser = new User();
61         apuser.setFullname("arnold Bugger esq");
62         apuser.setOrganization("disOrganised");
63         String appdata_ref = "vamsas:org.vamsas.test.simpleclient.VamsasArchive/"+apuser.getOrganization()+"/"+apuser.getFullname();
64         org.vamsas.client.simpleclient.VamsasArchive va = new org.vamsas.client.simpleclient.VamsasArchive(backup, true);
65         VamsasArchiveReader vread = va.getOriginalArchiveReader();
66         if (vread.getAppdataStream(appdata_ref)!=null) {
67           int i=0;
68           while (vread.getAppdataStream(appdata_ref+"/"+Integer.toString(++i))!=null)
69               ;
70           appdata_ref+="/"+Integer.toString(i);
71         }
72         log.info("Writing an apdata reference using AppDataStream interface.");
73         apuser.setAppDataChoice(new AppDataChoice());
74         apuser.getAppDataChoice().setDataReference(appdata_ref);
75         ObjectOutputStream apdos = new ObjectOutputStream(va.getAppDataStream(appdata_ref));
76         apdos.writeObject(appdata_ref);
77         apdos.writeObject(apuser); // some random content
78         apdos.close(); // this should *not* close the archive!
79         log.info("Preparing to write new document.");
80         doc.addApplicationData(appdata);
81         doc.addVAMSAS(Core.getDemoVamsas());
82         PrintWriter vxml = varchive.getDocumentOutputStream();
83         doc.marshal(vxml);
84         // TODO: verify that the vxml stream close method overridden ?
85         va.closeArchive();
86         log.info("Dump of new vamsas document :");
87         va = new org.vamsas.client.simpleclient.VamsasArchive(backup, true); // TODO - refactor Reader class to have deserializing helpers
88         ArchiveReports.reportDocument(va.getVamsasDocument(), va.getOriginalArchiveReader(), true);
89         backup.delete(); // tidy up
90       }
91       log.info("Cancelling write to original archive "+av);
92       varchive.cancelArchive();
93
94     } catch (Exception e) {
95       e.printStackTrace(System.err);
96     }
97   }
98 }