Mercurial > jhg
view src/org/tmatesoft/hg/core/HgAddRemoveCommand.java @ 647:c75297c17867
Location of repository files as enumeration, use file constants instead of plain names
author | Artem Tikhomirov <tikhomirov.artem@gmail.com> |
---|---|
date | Tue, 25 Jun 2013 18:53:18 +0200 |
parents | 65c01508f002 |
children | b4242b7e7dfe |
line wrap: on
line source
/* * Copyright (c) 2013 TMate Software Ltd * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; version 2 of the License. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * For information on how to redistribute this software under * the terms of a license other than GNU General Public License * contact TMate Software at support@hg4j.com */ package org.tmatesoft.hg.core; import java.util.LinkedHashSet; import org.tmatesoft.hg.internal.COWTransaction; import org.tmatesoft.hg.internal.DirstateBuilder; import org.tmatesoft.hg.internal.DirstateReader; import org.tmatesoft.hg.internal.Internals; import org.tmatesoft.hg.internal.Transaction; import org.tmatesoft.hg.repo.HgManifest.Flags; import org.tmatesoft.hg.repo.HgRepository; import org.tmatesoft.hg.repo.HgRepositoryLock; import org.tmatesoft.hg.repo.HgRuntimeException; import org.tmatesoft.hg.util.CancelSupport; import org.tmatesoft.hg.util.CancelledException; import org.tmatesoft.hg.util.Path; import org.tmatesoft.hg.util.ProgressSupport; /** * Schedule files for addition and removal * XXX and, perhaps, forget() functionality shall be here as well? * * @since 1.1 * @author Artem Tikhomirov * @author TMate Software Ltd. */ public class HgAddRemoveCommand extends HgAbstractCommand<HgAddRemoveCommand> { private final HgRepository repo; private final LinkedHashSet<Path> toAdd, toRemove; public HgAddRemoveCommand(HgRepository hgRepo) { repo = hgRepo; toAdd = new LinkedHashSet<Path>(); toRemove = new LinkedHashSet<Path>(); } /** * Schedule specified files to get listed in dirstate as added * * @param paths files to mark as added, additive * @return <code>this</code> for convenience */ public HgAddRemoveCommand add(Path... paths) { if (paths == null) { throw new IllegalArgumentException(); } for (Path p : paths) { toRemove.remove(p); toAdd.add(p); } return this; } /** * Schedule specified files to be marked as removed * * @param paths files to mark as removed, additive * @return <code>this</code> for convenience */ public HgAddRemoveCommand remove(Path... paths) { if (paths == null) { throw new IllegalArgumentException(); } for (Path p : paths) { toAdd.remove(p); toRemove.add(p); } return this; } public HgAddRemoveCommand addAll() { throw Internals.notImplemented(); } public HgAddRemoveCommand forget(Path path) { throw Internals.notImplemented(); } /** * Perform scheduled addition/removal * * @throws HgException subclass thereof to indicate specific issue with the command arguments or repository state * @throws HgRepositoryLockException if failed to lock the repo for modifications * @throws CancelledException if execution of the command was cancelled */ public void execute() throws HgException, HgRepositoryLockException, CancelledException { final HgRepositoryLock wdLock = repo.getWorkingDirLock(); wdLock.acquire(); try { final ProgressSupport progress = getProgressSupport(null); final CancelSupport cancellation = getCancelSupport(null, true); cancellation.checkCancelled(); progress.start(2 + toAdd.size() + toRemove.size()); Internals implRepo = Internals.getInstance(repo); final DirstateBuilder dirstateBuilder = new DirstateBuilder(implRepo); dirstateBuilder.fillFrom(new DirstateReader(implRepo, new Path.SimpleSource())); progress.worked(1); cancellation.checkCancelled(); for (Path p : toAdd) { dirstateBuilder.recordAdded(p, Flags.RegularFile, -1); progress.worked(1); cancellation.checkCancelled(); } for (Path p : toRemove) { dirstateBuilder.recordRemoved(p); progress.worked(1); cancellation.checkCancelled(); } Transaction.Factory trFactory = new COWTransaction.Factory(); Transaction tr = trFactory.create(repo); try { dirstateBuilder.serialize(tr); tr.commit(); } catch (RuntimeException ex) { tr.rollback(); throw ex; } catch (HgException ex) { tr.rollback(); throw ex; } progress.worked(1); progress.done(); } catch (HgRuntimeException ex) { throw new HgLibraryFailureException(ex); } finally { wdLock.release(); } } }