src/org/spearce/jgit/lib/FileCopier.java

   1 package org.spearce.jgit.lib;
   2
   3 import java.io.File;
   4 import java.io.FileOutputStream;
   5 import java.io.IOException;
   6 import java.io.InputStream;
   7 import java.util.Iterator;
   8
   9 public class FileCopier {
  10     public void copyOut(final Tree src, final File dest) throws IOException {
  11         dest.mkdirs();
  12         copyOut(src, dest, new byte[8192]);
  13     }
  14
  15     protected void copyOut(final Tree src, final File dest,
  16             final byte[] copyBuffer) throws IOException {
  17         final Iterator i;
  18
  19         i = src.getTreeEntries().iterator();
  20         while (i.hasNext()) {
  21             final Tree.Entry e = (Tree.Entry) i.next();
  22             final File f = new File(dest, e.getName());
  23
  24             if (e.isTree()) {
  25                 f.mkdir();
  26                 copyOut(e.getTree(), f, copyBuffer);
  27             } else if (e.isSymlink()) {
  28                 // TODO: We don't handle symlinks right now.
  29             } else {
  30                 final ObjectReader or = e.openBlob();
  31                 final InputStream is;
  32                 if (or == null) {
  33                     throw new CorruptObjectException("Missing blob "
  34                             + e.getId());
  35                 }
  36                 is = or.getInputStream();
  37                 try {
  38                     final FileOutputStream fos = new FileOutputStream(f);
  39                     try {
  40                         int r;
  41                         while ((r = is.read(copyBuffer)) > 0) {
  42                             fos.write(copyBuffer, 0, r);
  43                         }
  44                     } finally {
  45                         fos.close();
  46                     }
  47                 } finally {
  48                     or.close();
  49                 }
  50             }
  51         }
  52     }
  53 }