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

   1 /*
   2  *  Copyright (C) 2006  Shawn Pearce <spearce@spearce.org>
   3  *
   4  *  This library is free software; you can redistribute it and/or
   5  *  modify it under the terms of the GNU Lesser General Public
   6  *  License, version 2.1, as published by the Free Software Foundation.
   7  *
   8  *  This library is distributed in the hope that it will be useful,
   9  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  10  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11  *  Lesser General Public License for more details.
  12  *
  13  *  You should have received a copy of the GNU Lesser General Public
  14  *  License along with this library; if not, write to the Free Software
  15  *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301
  16  */
  17 package org.spearce.jgit.lib;
  18
  19 import java.io.BufferedReader;
  20 import java.io.ByteArrayInputStream;
  21 import java.io.IOException;
  22 import java.io.InputStreamReader;
  23 import java.util.ArrayList;
  24 import java.util.List;
  25
  26 import org.spearce.jgit.errors.CorruptObjectException;
  27 import org.spearce.jgit.errors.MissingObjectException;
  28
  29 public class Commit implements Treeish {
  30     private final Repository objdb;
  31
  32     private ObjectId commitId;
  33
  34     private ObjectId treeId;
  35
  36     private List parentIds;
  37
  38     private PersonIdent author;
  39
  40     private PersonIdent committer;
  41
  42     private String message;
  43
  44     private Tree treeObj;
  45
  46     private byte[] raw;
  47
  48     public Commit(final Repository db) {
  49         objdb = db;
  50         parentIds = new ArrayList(2);
  51     }
  52
  53     public Commit(final Repository db, final ObjectId id, final byte[] raw) {
  54         objdb = db;
  55         commitId = id;
  56         treeId = ObjectId.fromString(raw, 5);
  57         parentIds = new ArrayList(2);
  58         int rawPtr = 46;
  59         for (;;) {
  60             if (raw[rawPtr] != 'p')
  61                 break;
  62             parentIds.add(ObjectId.fromString(raw, rawPtr + 7));
  63             rawPtr += 48;
  64         }
  65
  66         this.raw = raw;
  67     }
  68
  69     public ObjectId getCommitId() {
  70         return commitId;
  71     }
  72
  73     public void setCommitId(final ObjectId id) {
  74         commitId = id;
  75     }
  76
  77     public ObjectId getTreeId() {
  78         return treeId;
  79     }
  80
  81     public void setTreeId(final ObjectId id) {
  82         if (!treeId.equals(id)) {
  83             treeObj = null;
  84         }
  85         treeId = id;
  86     }
  87
  88     public Tree getTree() throws IOException {
  89         if (treeObj == null) {
  90             treeObj = objdb.mapTree(getTreeId());
  91             if (treeObj == null) {
  92                 throw new MissingObjectException(getTreeId(),
  93                         Constants.TYPE_TREE);
  94             }
  95         }
  96         return treeObj;
  97     }
  98
  99     public void setTree(final Tree t) {
 100         treeId = t.getTreeId();
 101         treeObj = t;
 102     }
 103
 104     public PersonIdent getAuthor() {
 105         decode();
 106         return author;
 107     }
 108
 109     public void setAuthor(final PersonIdent a) {
 110         author = a;
 111     }
 112
 113     public PersonIdent getCommitter() {
 114         decode();
 115         return committer;
 116     }
 117
 118     public void setCommitter(final PersonIdent c) {
 119         committer = c;
 120     }
 121
 122     public List getParentIds() {
 123         return parentIds;
 124     }
 125
 126     public String getMessage() {
 127         decode();
 128         return message;
 129     }
 130
 131     private void decode() {
 132         // FIXME: handle I/O errors
 133         if (raw != null) {
 134             try {
 135                 BufferedReader br = new BufferedReader(new InputStreamReader(
 136                         new ByteArrayInputStream(raw)));
 137                 String n = br.readLine();
 138                 if (n == null || !n.startsWith("tree ")) {
 139                     throw new CorruptObjectException(commitId, "no tree");
 140                 }
 141                 while ((n = br.readLine()) != null && n.startsWith("parent ")) {
 142                     // empty body
 143                 }
 144                 if (n == null || !n.startsWith("author ")) {
 145                     throw new CorruptObjectException(commitId, "no author");
 146                 }
 147                 author = new PersonIdent(n.substring("author ".length()));
 148                 n = br.readLine();
 149                 if (n == null || !n.startsWith("committer ")) {
 150                     throw new CorruptObjectException(commitId, "no committer");
 151                 }
 152                 committer = new PersonIdent(n.substring("committer ".length()));
 153                 n = br.readLine();
 154                 if (n == null || !n.equals("")) {
 155                     throw new CorruptObjectException(commitId,
 156                             "malformed header");
 157                 }
 158                 StringBuffer tempMessage = new StringBuffer();
 159                 char[] readBuf = new char[2048];
 160                 int readLen;
 161                 while ((readLen = br.read(readBuf)) > 0) {
 162                     tempMessage.append(readBuf, 0, readLen);
 163                 }
 164                 message = tempMessage.toString();
 165             } catch (IOException e) {
 166                 e.printStackTrace();
 167             } finally {
 168                 raw = null;
 169             }
 170         }
 171     }
 172
 173     public void setMessage(final String m) {
 174         message = m;
 175     }
 176
 177     public void commit() throws IOException {
 178         if (getCommitId() != null)
 179             throw new IllegalStateException("exists " + getCommitId());
 180         setCommitId(new ObjectWriter(objdb).writeCommit(this));
 181     }
 182
 183     public String toString() {
 184         return "Commit[" + getCommitId() + " " + getAuthor() + "]";
 185     }
 186 }