bludgeon/ZipFileObject.cs

   1 using System;
   2 using System.Collections;
   3 using System.IO;
   4 using System.Text;
   5
   6 using ICSharpCode.SharpZipLib;
   7 using ICSharpCode.SharpZipLib.Zip;
   8
   9 using Beagle.Util;
  10 using Beagle;
  11
  12 namespace Bludgeon {
  13
  14         public class ZipFileObject : FileObject {
  15
  16                 public ZipFileObject (DirectoryObject zip_root)
  17                 {
  18                         AllowChildren ();
  19
  20                         ArrayList to_be_zipped;
  21                         to_be_zipped = new ArrayList (zip_root.Children);
  22
  23                         foreach (FileSystemObject fso in to_be_zipped) {
  24                                 zip_root.RemoveChild (fso, null);
  25                                 this.AddChild (fso, null);
  26                         }
  27                 }
  28
  29                 override protected string GetChildUri (FileSystemObject children)
  30                 {
  31                         // FIXME: What is the uri scheme for zip files?
  32                         string uri = this.Uri + "#" + children.Name;
  33                         return uri;
  34                 }
  35
  36                 override public string MimeType {
  37                         get { return "application/x-zip"; }
  38                 }
  39
  40                 override public string Extension {
  41                         get { return ".zip"; }
  42                 }
  43
  44                 private void WriteObjectToZip (ZipOutputStream  zip_out,
  45                                                FileSystemObject fso,
  46                                                EventTracker     tracker)
  47                 {
  48
  49                         MemoryStream memory = null;
  50
  51                         string name;
  52                         name = fso.FullName.Substring (this.FullName.Length + 1);
  53                         if (fso is DirectoryObject)
  54                                 name += "/";
  55
  56                         ZipEntry entry;
  57                         entry = new ZipEntry (name);
  58                         entry.DateTime = fso.Timestamp;
  59
  60                         if (fso is DirectoryObject)
  61                                 entry.Size = 0;
  62                         else {
  63                                 memory = new MemoryStream ();
  64                                 ((FileObject) fso).AddToStream (memory, tracker);
  65                                 entry.Size = memory.Length;
  66                         }
  67
  68                         zip_out.PutNextEntry (entry);
  69                         if (memory != null) {
  70                                 zip_out.Write (memory.ToArray (), 0, (int) memory.Length);
  71                                 memory.Close ();
  72                         }
  73
  74                         // If this is a directory, write out the children
  75                         if (fso is DirectoryObject)
  76                                 foreach (FileSystemObject child in fso.Children)
  77                                         WriteObjectToZip (zip_out, child, tracker);
  78                 }
  79
  80                 override public void AddToStream (Stream stream, EventTracker tracker)
  81                 {
  82                         if (tracker != null)
  83                                 tracker.ExpectingAdded (this.Uri);
  84
  85                         UnclosableStream unclosable;
  86                         unclosable = new UnclosableStream (stream);
  87
  88                         ZipOutputStream zip_out;
  89                         zip_out = new ZipOutputStream (unclosable);
  90
  91                         foreach (FileSystemObject fso in Children)
  92                                 WriteObjectToZip (zip_out, fso, tracker);
  93                         zip_out.Finish ();
  94                         zip_out.Close ();
  95                 }
  96
  97                 override protected bool MatchesQueryPart (QueryPart part)
  98                 {
  99                         // The only thing that could match is the name, and we check
 100                         // for that already in FileSystemObject.MatchesQuery.
 101                         return false;
 102                 }
 103         }
 104 }