-- Copyright (C) 2009 Benedikt Schmidt
--
-- 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; either version 2, or (at your option)
-- any later version.
--
-- 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.
--
-- You should have received a copy of the GNU General Public License
-- along with this program; see the file COPYING.  If not, write to
-- the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
-- Boston, MA 02110-1301, USA.


{-# LANGUAGE GeneralizedNewtypeDeriving #-}
{-# LANGUAGE MultiParamTypeClasses #-}

module Darcs.Patch.Index.Monad
    ( withPatchMods
    , applyToFileMods
    , makePatchID
    ) where

import Prelude ()
import Darcs.Prelude

import Darcs.Patch.Index.Types ( PatchMod(..), PatchId(..) )
import Darcs.Patch.Info ( makePatchname, PatchInfo )
import Darcs.Patch.Apply ( Apply(..) )
import Darcs.Patch.ApplyMonad ( ApplyMonad(..), ApplyMonadTree(..) )
import Control.Monad.State
import Control.Arrow
import Darcs.Util.Path ( FileName, fn2fp, movedirfilename )
import qualified Data.Set as S
import Data.Set ( Set )
import Data.List ( isPrefixOf )
import Darcs.Util.Tree (Tree)

newtype FileModMonad a = FMM (State (Set FileName, [PatchMod FileName]) a)
  deriving (a -> FileModMonad b -> FileModMonad a
(a -> b) -> FileModMonad a -> FileModMonad b
(forall a b. (a -> b) -> FileModMonad a -> FileModMonad b)
-> (forall a b. a -> FileModMonad b -> FileModMonad a)
-> Functor FileModMonad
forall a b. a -> FileModMonad b -> FileModMonad a
forall a b. (a -> b) -> FileModMonad a -> FileModMonad b
forall (f :: * -> *).
(forall a b. (a -> b) -> f a -> f b)
-> (forall a b. a -> f b -> f a) -> Functor f
<$ :: a -> FileModMonad b -> FileModMonad a
$c<$ :: forall a b. a -> FileModMonad b -> FileModMonad a
fmap :: (a -> b) -> FileModMonad a -> FileModMonad b
$cfmap :: forall a b. (a -> b) -> FileModMonad a -> FileModMonad b
Functor, Functor FileModMonad
a -> FileModMonad a
Functor FileModMonad =>
(forall a. a -> FileModMonad a)
-> (forall a b.
    FileModMonad (a -> b) -> FileModMonad a -> FileModMonad b)
-> (forall a b c.
    (a -> b -> c)
    -> FileModMonad a -> FileModMonad b -> FileModMonad c)
-> (forall a b. FileModMonad a -> FileModMonad b -> FileModMonad b)
-> (forall a b. FileModMonad a -> FileModMonad b -> FileModMonad a)
-> Applicative FileModMonad
FileModMonad a -> FileModMonad b -> FileModMonad b
FileModMonad a -> FileModMonad b -> FileModMonad a
FileModMonad (a -> b) -> FileModMonad a -> FileModMonad b
(a -> b -> c) -> FileModMonad a -> FileModMonad b -> FileModMonad c
forall a. a -> FileModMonad a
forall a b. FileModMonad a -> FileModMonad b -> FileModMonad a
forall a b. FileModMonad a -> FileModMonad b -> FileModMonad b
forall a b.
FileModMonad (a -> b) -> FileModMonad a -> FileModMonad b
forall a b c.
(a -> b -> c) -> FileModMonad a -> FileModMonad b -> FileModMonad c
forall (f :: * -> *).
Functor f =>
(forall a. a -> f a)
-> (forall a b. f (a -> b) -> f a -> f b)
-> (forall a b c. (a -> b -> c) -> f a -> f b -> f c)
-> (forall a b. f a -> f b -> f b)
-> (forall a b. f a -> f b -> f a)
-> Applicative f
<* :: FileModMonad a -> FileModMonad b -> FileModMonad a
$c<* :: forall a b. FileModMonad a -> FileModMonad b -> FileModMonad a
*> :: FileModMonad a -> FileModMonad b -> FileModMonad b
$c*> :: forall a b. FileModMonad a -> FileModMonad b -> FileModMonad b
liftA2 :: (a -> b -> c) -> FileModMonad a -> FileModMonad b -> FileModMonad c
$cliftA2 :: forall a b c.
(a -> b -> c) -> FileModMonad a -> FileModMonad b -> FileModMonad c
<*> :: FileModMonad (a -> b) -> FileModMonad a -> FileModMonad b
$c<*> :: forall a b.
FileModMonad (a -> b) -> FileModMonad a -> FileModMonad b
pure :: a -> FileModMonad a
$cpure :: forall a. a -> FileModMonad a
$cp1Applicative :: Functor FileModMonad
Applicative, Applicative FileModMonad
a -> FileModMonad a
Applicative FileModMonad =>
(forall a b.
 FileModMonad a -> (a -> FileModMonad b) -> FileModMonad b)
-> (forall a b. FileModMonad a -> FileModMonad b -> FileModMonad b)
-> (forall a. a -> FileModMonad a)
-> Monad FileModMonad
FileModMonad a -> (a -> FileModMonad b) -> FileModMonad b
FileModMonad a -> FileModMonad b -> FileModMonad b
forall a. a -> FileModMonad a
forall a b. FileModMonad a -> FileModMonad b -> FileModMonad b
forall a b.
FileModMonad a -> (a -> FileModMonad b) -> FileModMonad b
forall (m :: * -> *).
Applicative m =>
(forall a b. m a -> (a -> m b) -> m b)
-> (forall a b. m a -> m b -> m b)
-> (forall a. a -> m a)
-> Monad m
return :: a -> FileModMonad a
$creturn :: forall a. a -> FileModMonad a
>> :: FileModMonad a -> FileModMonad b -> FileModMonad b
$c>> :: forall a b. FileModMonad a -> FileModMonad b -> FileModMonad b
>>= :: FileModMonad a -> (a -> FileModMonad b) -> FileModMonad b
$c>>= :: forall a b.
FileModMonad a -> (a -> FileModMonad b) -> FileModMonad b
$cp1Monad :: Applicative FileModMonad
Monad, MonadState (Set FileName, [PatchMod FileName]))

withPatchMods :: FileModMonad a -> Set FileName -> (Set FileName, [PatchMod FileName])
withPatchMods :: FileModMonad a
-> Set FileName -> (Set FileName, [PatchMod FileName])
withPatchMods (FMM m :: State (Set FileName, [PatchMod FileName]) a
m) fps :: Set FileName
fps = ([PatchMod FileName] -> [PatchMod FileName])
-> (Set FileName, [PatchMod FileName])
-> (Set FileName, [PatchMod FileName])
forall (a :: * -> * -> *) b c d.
Arrow a =>
a b c -> a (d, b) (d, c)
second [PatchMod FileName] -> [PatchMod FileName]
forall a. [a] -> [a]
reverse ((Set FileName, [PatchMod FileName])
 -> (Set FileName, [PatchMod FileName]))
-> (Set FileName, [PatchMod FileName])
-> (Set FileName, [PatchMod FileName])
forall a b. (a -> b) -> a -> b
$ State (Set FileName, [PatchMod FileName]) a
-> (Set FileName, [PatchMod FileName])
-> (Set FileName, [PatchMod FileName])
forall s a. State s a -> s -> s
execState State (Set FileName, [PatchMod FileName]) a
m (Set FileName
fps,[])

-- These instances are defined to be used only with
-- apply.
instance ApplyMonad Tree FileModMonad where
    type ApplyMonadBase FileModMonad = FileModMonad
    nestedApply :: FileModMonad x
-> Tree (ApplyMonadBase FileModMonad)
-> FileModMonad (x, Tree (ApplyMonadBase FileModMonad))
nestedApply _ _ = String -> FileModMonad (x, Tree FileModMonad)
forall a. String -> a
bug "nestedApply FileModMonad"
    liftApply :: (Tree (ApplyMonadBase FileModMonad)
 -> ApplyMonadBase FileModMonad x)
-> Tree (ApplyMonadBase FileModMonad)
-> FileModMonad (x, Tree (ApplyMonadBase FileModMonad))
liftApply _ _ = String -> FileModMonad (x, Tree FileModMonad)
forall a. String -> a
bug "liftApply FileModMonad"
    getApplyState :: FileModMonad (Tree (ApplyMonadBase FileModMonad))
getApplyState = String -> FileModMonad (Tree FileModMonad)
forall a. String -> a
bug "getApplyState FileModMonad"

instance ApplyMonadTree FileModMonad where
    mDoesDirectoryExist :: FileName -> FileModMonad Bool
mDoesDirectoryExist d :: FileName
d = do
      Set FileName
fps <- ((Set FileName, [PatchMod FileName]) -> Set FileName)
-> FileModMonad (Set FileName)
forall s (m :: * -> *) a. MonadState s m => (s -> a) -> m a
gets (Set FileName, [PatchMod FileName]) -> Set FileName
forall a b. (a, b) -> a
fst
      Bool -> FileModMonad Bool
forall (m :: * -> *) a. Monad m => a -> m a
return (Bool -> FileModMonad Bool) -> Bool -> FileModMonad Bool
forall a b. (a -> b) -> a -> b
$ FileName -> Set FileName -> Bool
forall a. Ord a => a -> Set a -> Bool
S.member FileName
d Set FileName
fps
    mDoesFileExist :: FileName -> FileModMonad Bool
mDoesFileExist f :: FileName
f = do
      Set FileName
fps <- ((Set FileName, [PatchMod FileName]) -> Set FileName)
-> FileModMonad (Set FileName)
forall s (m :: * -> *) a. MonadState s m => (s -> a) -> m a
gets (Set FileName, [PatchMod FileName]) -> Set FileName
forall a b. (a, b) -> a
fst
      Bool -> FileModMonad Bool
forall (m :: * -> *) a. Monad m => a -> m a
return (Bool -> FileModMonad Bool) -> Bool -> FileModMonad Bool
forall a b. (a -> b) -> a -> b
$ FileName -> Set FileName -> Bool
forall a. Ord a => a -> Set a -> Bool
S.member FileName
f Set FileName
fps
    mReadFilePS :: FileName -> FileModMonad ByteString
mReadFilePS _ = String -> FileModMonad ByteString
forall a. String -> a
bug "mReadFilePS FileModMonad"
    mCreateFile :: FileName -> FileModMonad ()
mCreateFile = FileName -> FileModMonad ()
createFile
    mCreateDirectory :: FileName -> FileModMonad ()
mCreateDirectory = FileName -> FileModMonad ()
createDir
    mRemoveFile :: FileName -> FileModMonad ()
mRemoveFile = FileName -> FileModMonad ()
remove
    mRemoveDirectory :: FileName -> FileModMonad ()
mRemoveDirectory = FileName -> FileModMonad ()
remove
    mRename :: FileName -> FileName -> FileModMonad ()
mRename a :: FileName
a b :: FileName
b = do
      Set FileName
fns <- ((Set FileName, [PatchMod FileName]) -> Set FileName)
-> FileModMonad (Set FileName)
forall s (m :: * -> *) a. MonadState s m => (s -> a) -> m a
gets (Set FileName, [PatchMod FileName]) -> Set FileName
forall a b. (a, b) -> a
fst
      if FileName -> Set FileName -> Bool
forall a. Ord a => a -> Set a -> Bool
S.notMember FileName
a Set FileName
fns then
         PatchMod FileName -> FileModMonad ()
addMod (FileName -> PatchMod FileName
forall a. a -> PatchMod a
PInvalid FileName
a)  -- works around some old repo inconsistencies
       else
        do -- we have to account for directory moves
           PatchMod FileName -> FileModMonad ()
addMod (FileName -> FileName -> PatchMod FileName
forall a. a -> a -> PatchMod a
PRename FileName
a FileName
b)
           (Set FileName -> Set FileName) -> FileModMonad ()
modifyFps (FileName -> Set FileName -> Set FileName
forall a. Ord a => a -> Set a -> Set a
S.delete FileName
a)
           FileName -> FileModMonad ()
addFile FileName
b
           [FileName] -> (FileName -> FileModMonad ()) -> FileModMonad ()
forall (t :: * -> *) (m :: * -> *) a b.
(Foldable t, Monad m) =>
t a -> (a -> m b) -> m ()
forM_ (Set FileName -> [FileName]
forall a. Set a -> [a]
S.toList Set FileName
fns) ((FileName -> FileModMonad ()) -> FileModMonad ())
-> (FileName -> FileModMonad ()) -> FileModMonad ()
forall a b. (a -> b) -> a -> b
$ \fn :: FileName
fn ->
             Bool -> FileModMonad () -> FileModMonad ()
forall (f :: * -> *). Applicative f => Bool -> f () -> f ()
when (FileName -> String
fn2fp FileName
a String -> String -> Bool
forall a. Eq a => [a] -> [a] -> Bool
`isPrefixOf` FileName -> String
fn2fp FileName
fn) (FileModMonad () -> FileModMonad ())
-> FileModMonad () -> FileModMonad ()
forall a b. (a -> b) -> a -> b
$ do
               (Set FileName -> Set FileName) -> FileModMonad ()
modifyFps (FileName -> Set FileName -> Set FileName
forall a. Ord a => a -> Set a -> Set a
S.delete FileName
fn)
               let newfn :: FileName
newfn = FileName -> FileName -> FileName -> FileName
movedirfilename FileName
a FileName
b FileName
fn
               FileName -> FileModMonad ()
addFile FileName
newfn
               PatchMod FileName -> FileModMonad ()
addMod (FileName -> FileName -> PatchMod FileName
forall a. a -> a -> PatchMod a
PRename FileName
fn FileName
newfn)

    mModifyFilePS :: FileName
-> (ByteString -> FileModMonad ByteString) -> FileModMonad ()
mModifyFilePS f :: FileName
f _ = PatchMod FileName -> FileModMonad ()
addMod (FileName -> PatchMod FileName
forall a. a -> PatchMod a
PTouch FileName
f)

-- ---------------------------------------------------------------------
-- State Handling Functions

addMod :: PatchMod FileName -> FileModMonad ()
addMod :: PatchMod FileName -> FileModMonad ()
addMod pm :: PatchMod FileName
pm = ((Set FileName, [PatchMod FileName])
 -> (Set FileName, [PatchMod FileName]))
-> FileModMonad ()
forall s (m :: * -> *). MonadState s m => (s -> s) -> m ()
modify (((Set FileName, [PatchMod FileName])
  -> (Set FileName, [PatchMod FileName]))
 -> FileModMonad ())
-> ((Set FileName, [PatchMod FileName])
    -> (Set FileName, [PatchMod FileName]))
-> FileModMonad ()
forall a b. (a -> b) -> a -> b
$ ([PatchMod FileName] -> [PatchMod FileName])
-> (Set FileName, [PatchMod FileName])
-> (Set FileName, [PatchMod FileName])
forall (a :: * -> * -> *) b c d.
Arrow a =>
a b c -> a (d, b) (d, c)
second (PatchMod FileName
pm PatchMod FileName -> [PatchMod FileName] -> [PatchMod FileName]
forall a. a -> [a] -> [a]
:)

addFile :: FileName -> FileModMonad ()
addFile :: FileName -> FileModMonad ()
addFile f :: FileName
f = (Set FileName -> Set FileName) -> FileModMonad ()
modifyFps (FileName -> Set FileName -> Set FileName
forall a. Ord a => a -> Set a -> Set a
S.insert FileName
f)

createFile :: FileName -> FileModMonad ()
createFile :: FileName -> FileModMonad ()
createFile fn :: FileName
fn = do
  FileName -> Bool -> FileModMonad ()
errorIfPresent FileName
fn Bool
True
  PatchMod FileName -> FileModMonad ()
addMod (FileName -> PatchMod FileName
forall a. a -> PatchMod a
PCreateFile FileName
fn)
  FileName -> FileModMonad ()
addFile FileName
fn

createDir :: FileName -> FileModMonad ()
createDir :: FileName -> FileModMonad ()
createDir fn :: FileName
fn = do
  FileName -> Bool -> FileModMonad ()
errorIfPresent FileName
fn Bool
False
  PatchMod FileName -> FileModMonad ()
addMod (FileName -> PatchMod FileName
forall a. a -> PatchMod a
PCreateDir FileName
fn)
  FileName -> FileModMonad ()
addFile FileName
fn

errorIfPresent :: FileName -> Bool -> FileModMonad ()
errorIfPresent :: FileName -> Bool -> FileModMonad ()
errorIfPresent fn :: FileName
fn isFile :: Bool
isFile = do
    Set FileName
fs <- ((Set FileName, [PatchMod FileName]) -> Set FileName)
-> FileModMonad (Set FileName)
forall s (m :: * -> *) a. MonadState s m => (s -> a) -> m a
gets (Set FileName, [PatchMod FileName]) -> Set FileName
forall a b. (a, b) -> a
fst
    Bool -> FileModMonad () -> FileModMonad ()
forall (f :: * -> *). Applicative f => Bool -> f () -> f ()
when (FileName -> Set FileName -> Bool
forall a. Ord a => a -> Set a -> Bool
S.member FileName
fn Set FileName
fs) (FileModMonad () -> FileModMonad ())
-> FileModMonad () -> FileModMonad ()
forall a b. (a -> b) -> a -> b
$
        String -> FileModMonad ()
forall a. HasCallStack => String -> a
error (String -> FileModMonad ()) -> String -> FileModMonad ()
forall a b. (a -> b) -> a -> b
$ [String] -> String
unwords [ "error: patch index entry for"
                        , if Bool
isFile then "file" else "directory"
                        , FileName -> String
fn2fp FileName
fn
                        , "created >1 times. Run `darcs repair` and try again."
                        ]

remove :: FileName -> FileModMonad ()
remove :: FileName -> FileModMonad ()
remove f :: FileName
f = PatchMod FileName -> FileModMonad ()
addMod (FileName -> PatchMod FileName
forall a. a -> PatchMod a
PRemove FileName
f) FileModMonad () -> FileModMonad () -> FileModMonad ()
forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> (Set FileName -> Set FileName) -> FileModMonad ()
modifyFps (FileName -> Set FileName -> Set FileName
forall a. Ord a => a -> Set a -> Set a
S.delete FileName
f)

modifyFps :: (Set FileName -> Set FileName) -> FileModMonad ()
modifyFps :: (Set FileName -> Set FileName) -> FileModMonad ()
modifyFps f :: Set FileName -> Set FileName
f = ((Set FileName, [PatchMod FileName])
 -> (Set FileName, [PatchMod FileName]))
-> FileModMonad ()
forall s (m :: * -> *). MonadState s m => (s -> s) -> m ()
modify (((Set FileName, [PatchMod FileName])
  -> (Set FileName, [PatchMod FileName]))
 -> FileModMonad ())
-> ((Set FileName, [PatchMod FileName])
    -> (Set FileName, [PatchMod FileName]))
-> FileModMonad ()
forall a b. (a -> b) -> a -> b
$ (Set FileName -> Set FileName)
-> (Set FileName, [PatchMod FileName])
-> (Set FileName, [PatchMod FileName])
forall (a :: * -> * -> *) b c d.
Arrow a =>
a b c -> a (b, d) (c, d)
first Set FileName -> Set FileName
f

makePatchID :: PatchInfo -> PatchId
makePatchID :: PatchInfo -> PatchId
makePatchID = SHA1 -> PatchId
PID (SHA1 -> PatchId) -> (PatchInfo -> SHA1) -> PatchInfo -> PatchId
forall b c a. (b -> c) -> (a -> b) -> a -> c
. PatchInfo -> SHA1
makePatchname

--------------------------------------------------------------------------------
-- | Apply a patch to set of 'FileName's, yielding the new set of 'FileName's and 'PatchMod's
applyToFileMods :: (Apply p, ApplyState p ~ Tree) => p wX wY -> Set FileName -> (Set FileName, [PatchMod FileName])
applyToFileMods :: p wX wY -> Set FileName -> (Set FileName, [PatchMod FileName])
applyToFileMods patch :: p wX wY
patch = FileModMonad ()
-> Set FileName -> (Set FileName, [PatchMod FileName])
forall a.
FileModMonad a
-> Set FileName -> (Set FileName, [PatchMod FileName])
withPatchMods (p wX wY -> FileModMonad ()
forall (p :: * -> * -> *) (m :: * -> *) wX wY.
(Apply p, ApplyMonad (ApplyState p) m) =>
p wX wY -> m ()
apply p wX wY
patch)