module Darcs.UI.Commands.Rollback ( rollback ) where
import Prelude ()
import Darcs.Prelude
import Control.Exception ( catch, IOException )
import Control.Monad ( when )
import Data.List ( sort )
import Darcs.Util.Tree( Tree )
import System.Exit ( exitSuccess )
import Darcs.Patch.Apply( ApplyState )
import Darcs.Patch.Match ( firstMatch )
import Darcs.Patch.PatchInfoAnd ( n2pia )
import Darcs.Patch ( IsRepoType, RepoPatch, invert, effect, fromPrims, sortCoalesceFL,
canonize, PrimOf )
import Darcs.Patch.Named.Wrapped ( anonymous )
import Darcs.Patch.Set ( PatchSet(..), patchSet2FL )
import Darcs.Patch.Split ( reversePrimSplitter )
import Darcs.Patch.Witnesses.Ordered ( FL(..), (:>)(..), RL(..), concatFL,
nullFL, mapFL_FL )
import Darcs.Patch.Witnesses.Sealed ( Sealed(..) )
import Darcs.Repository.Flags ( AllowConflicts (..), UseIndex(..), Reorder(..),
ScanKnown(..), UpdateWorking(..), DryRun(NoDryRun))
import Darcs.Repository ( Repository, withRepoLock, RepoJob(..),
applyToWorking, readRepo,
finalizeRepositoryChanges, tentativelyAddToPending,
considerMergeToWorking )
import Darcs.UI.Commands ( DarcsCommand(..), withStdOpts, nodefaults, setEnvDarcsPatches,
amInHashedRepository, putInfo )
import Darcs.UI.Commands.Unrecord ( getLastPatches )
import Darcs.UI.Commands.Util ( announceFiles )
import Darcs.UI.Completion ( knownFileArgs )
import Darcs.UI.Flags ( DarcsFlag, verbosity, umask, useCache,
compress, externalMerge, wantGuiPause,
diffAlgorithm, fixSubPaths, isInteractive )
import Darcs.UI.Options
( (^), odesc, ocheck, onormalise
, defaultFlags, parseFlags, (?)
)
import qualified Darcs.UI.Options.All as O
import Darcs.UI.SelectChanges ( WhichChanges(..),
selectionContext, selectionContextPrim,
runSelection )
import qualified Darcs.UI.SelectChanges as S ( PatchSelectionOptions(..) )
import Darcs.Util.Path ( toFilePath, AbsolutePath )
import Darcs.Util.Printer ( text )
import Darcs.Util.Progress ( debugMessage )
rollbackDescription :: String
rollbackDescription =
"Apply the inverse of recorded changes to the working tree."
rollbackHelp :: String
rollbackHelp = unlines
[ "Rollback is used to undo the effects of some changes from patches"
, "in the repository. The selected changes are undone in your working"
, "tree, but the repository is left unchanged. First you are offered a"
, "choice of which patches to undo, then which changes within the"
, "patches to undo."
, ""
, "Before doing `rollback`, you may want to temporarily undo the changes"
, "of your working tree (if there are) and save them for later use."
, "To do so, you can run `revert`, then run `rollback`, record a patch,"
, "and run `unrevert` to restore the saved changes into your working tree."
]
patchSelOpts :: [DarcsFlag] -> S.PatchSelectionOptions
patchSelOpts flags = S.PatchSelectionOptions
{ S.verbosity = verbosity ? flags
, S.matchFlags = parseFlags O.matchSeveralOrLast flags
, S.interactive = isInteractive True flags
, S.selectDeps = O.PromptDeps
, S.summary = O.NoSummary
, S.withContext = O.NoContext
}
rollback :: DarcsCommand [DarcsFlag]
rollback = DarcsCommand
{ commandProgramName = "darcs"
, commandName = "rollback"
, commandHelp = rollbackHelp
, commandDescription = rollbackDescription
, commandExtraArgs = 1
, commandExtraArgHelp = ["[FILE or DIRECTORY]..."]
, commandCommand = rollbackCmd
, commandPrereq = amInHashedRepository
, commandCompleteArgs = knownFileArgs
, commandArgdefaults = nodefaults
, commandAdvancedOptions = odesc rollbackAdvancedOpts
, commandBasicOptions = odesc rollbackBasicOpts
, commandDefaults = defaultFlags rollbackOpts
, commandCheckOptions = ocheck rollbackOpts
, commandParseOptions = onormalise rollbackOpts
}
where
rollbackBasicOpts
= O.matchSeveralOrLast
^ O.interactive
^ O.repoDir
^ O.diffAlgorithm
rollbackAdvancedOpts = O.umask
rollbackOpts = rollbackBasicOpts `withStdOpts` rollbackAdvancedOpts
exitIfNothingSelected :: FL p wX wY -> String -> IO ()
exitIfNothingSelected ps what =
when (nullFL ps) $ putStrLn ("No " ++ what ++ " selected!") >> exitSuccess
rollbackCmd :: (AbsolutePath, AbsolutePath) -> [DarcsFlag] -> [String] -> IO ()
rollbackCmd fps opts args = withRepoLock NoDryRun (useCache ? opts)
YesUpdateWorking (umask ? opts) $ RepoJob $ \repository -> do
files <- if null args
then return Nothing
else Just . sort <$> fixSubPaths fps args
when (files == Just []) $
fail "No valid arguments were given."
announceFiles (verbosity ? opts) files "Rolling back changes in"
allpatches <- readRepo repository
let matchFlags = parseFlags O.matchSeveralOrLast opts
(_ :> patches) <- return $
if firstMatch matchFlags
then getLastPatches matchFlags allpatches
else PatchSet NilRL NilRL :> patchSet2FL allpatches
let filesFps = map toFilePath <$> files
patchCtx = selectionContext LastReversed "rollback" (patchSelOpts opts) Nothing filesFps
(_ :> ps) <-
runSelection patches patchCtx
exitIfNothingSelected ps "patches"
setEnvDarcsPatches ps
let hunkContext = selectionContextPrim Last "rollback" (patchSelOpts opts)
(Just (reversePrimSplitter (diffAlgorithm ? opts)))
filesFps Nothing
hunks = concatFL . mapFL_FL (canonize $ diffAlgorithm ? opts) . sortCoalesceFL . effect $ ps
whatToUndo <- runSelection hunks hunkContext
undoItNow opts repository whatToUndo
undoItNow :: (IsRepoType rt, RepoPatch p, ApplyState p ~ Tree)
=> [DarcsFlag] -> Repository rt p wR wU wT
-> (q :> FL (PrimOf p)) wA wT -> IO ()
undoItNow opts repo (_ :> prims) = do
exitIfNothingSelected prims "changes"
rbp <- n2pia `fmap` anonymous (fromPrims $ invert prims)
Sealed pw <- considerMergeToWorking repo "rollback"
YesAllowConflictsAndMark YesUpdateWorking
(externalMerge ? opts) (wantGuiPause opts)
(compress ? opts) (verbosity ? opts) NoReorder
(UseIndex, ScanKnown, diffAlgorithm ? opts)
NilFL (rbp :>: NilFL)
tentativelyAddToPending repo YesUpdateWorking pw
finalizeRepositoryChanges repo YesUpdateWorking
(compress ? opts)
_ <- applyToWorking repo (verbosity ? opts) pw
`catch`
\(e :: IOException) -> fail $
"error applying rolled back patch to working directory\n"
++ show e
debugMessage "Finished applying unrecorded rollback patch"
putInfo opts $ text "Changes rolled back in working directory"