io.github.svndump_to_git.git.model.tree.utils.JGitTreeUtils Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of repository-tools-common-git Show documentation
Common Code Related to Git
The newest version!
/*
 *  Copyright 2014 The Kuali Foundation Licensed under the
 *	Educational Community License, Version 2.0 (the "License"); you may
 *	not use this file except in compliance with the License. You may
 *	obtain a copy of the License at
 *
 *	http://www.osedu.org/licenses/ECL-2.0
 *
 *	Unless required by applicable law or agreed to in writing,
 *	software distributed under the License is distributed on an "AS IS"
 *	BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 *	or implied. See the License for the specific language governing
 *	permissions and limitations under the License.
 */
package io.github.svndump_to_git.git.model.tree.utils;

import java.io.IOException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

import io.github.svndump_to_git.git.model.tree.JGitTreeData;
import org.eclipse.jgit.errors.CorruptObjectException;
import org.eclipse.jgit.errors.IncorrectObjectTypeException;
import org.eclipse.jgit.errors.MissingObjectException;
import org.eclipse.jgit.lib.Constants;
import org.eclipse.jgit.lib.FileMode;
import org.eclipse.jgit.lib.ObjectId;
import org.eclipse.jgit.lib.ObjectInserter;
import org.eclipse.jgit.lib.ObjectReader;
import org.eclipse.jgit.lib.TreeFormatter;
import org.eclipse.jgit.revwalk.RevCommit;
import org.eclipse.jgit.treewalk.TreeWalk;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * @author Kuali Student Team
 *
 */
public class JGitTreeUtils {

	private static final Logger log = LoggerFactory.getLogger(JGitTreeUtils.class);
	
	/**
	 * 
	 */
	public JGitTreeUtils() {
		// TODO Auto-generated constructor stub
	}
	
	/**
	 * Helper to load the mutable tree data (at the root level) for the indicated tree and to then insert the blob and content provided.
	 * 
	 * @param or
	 * @param inserter
	 * @param treeId
	 * @param blobName
	 * @param blobContent
	 * @return
	 * @throws MissingObjectException
	 * @throws IncorrectObjectTypeException
	 * @throws CorruptObjectException
	 * @throws IOException
	 */
	public static List insertBlob (ObjectReader or, ObjectInserter inserter, ObjectId treeId, String blobName, String blobContent) throws MissingObjectException, IncorrectObjectTypeException, CorruptObjectException, IOException {
		
		List data = extractBaseTreeLevel(or, treeId);
		
		ObjectId blobId = inserter.insert(Constants.OBJ_BLOB, blobContent.getBytes());
		
		data.add(new JGitTreeData(blobName, FileMode.REGULAR_FILE, blobId));
		
		return data;
	}
	/**
	 * Extract the list of elements in the commit tree at the base or root level.
	 * 
	 * @param or
	 * @param commit
	 * @return
	 * @throws IOException 
	 * @throws CorruptObjectException 
	 * @throws IncorrectObjectTypeException 
	 * @throws MissingObjectException 
	 */
	public static ListextractBaseTreeLevel (ObjectReader or, RevCommit commit) throws MissingObjectException, IncorrectObjectTypeException, CorruptObjectException, IOException {
		
		return extractBaseTreeLevel(or, commit.getTree().getId());
		
	}
	
	
	public static ListextractBaseTreeLevel (ObjectReader or, ObjectId treeId) throws MissingObjectException, IncorrectObjectTypeException, CorruptObjectException, IOException {
		
		ArrayList treeData = new ArrayList<>();
		
		TreeWalk tw = new TreeWalk(or);
		
		tw.addTree(treeId);
		
		tw.setRecursive(false);
		
		while (tw.next()) {
		
			String name = tw.getNameString();
			
			FileMode fileMode = tw.getFileMode(0);
			ObjectId objectId = tw.getObjectId(0);
			
			JGitTreeData jgtd = new JGitTreeData(name, fileMode, objectId);
			
			treeData.add(jgtd);
			
		}
		
		tw.close();
		
		return treeData;
	}
	
	
	public static ObjectId createTree (Listdata, ObjectInserter inserter) throws IOException {
		
		TreeFormatter tf = new TreeFormatter();
		
		Collections.sort(data, JGitTreeData.GIT_SORT_ORDERING);
		
		for (JGitTreeData treeData : data) {
			
			tf.append(treeData.getName(), treeData.getFileMode(), treeData.getObjectId());
		}
		
		ObjectId treeId = inserter.insert(tf);
		
		return treeId;
		
	}

}