6c81e0c8f1
Several nice speed wins I think. At 340/633 files converted.
88 lines
2.8 KiB
Haskell
88 lines
2.8 KiB
Haskell
{- adjusted branch names
|
|
-
|
|
- Copyright 2016-2018 Joey Hess <id@joeyh.name>
|
|
-
|
|
- Licensed under the GNU AGPL version 3 or higher.
|
|
-}
|
|
|
|
{-# LANGUAGE OverloadedStrings #-}
|
|
|
|
module Annex.AdjustedBranch.Name (
|
|
originalToAdjusted,
|
|
adjustedToOriginal,
|
|
AdjBranch(..),
|
|
OrigBranch,
|
|
) where
|
|
|
|
import Types.AdjustedBranch
|
|
import Git
|
|
import qualified Git.Ref
|
|
import Utility.Misc
|
|
|
|
import Control.Applicative
|
|
import Data.Char
|
|
import qualified Data.ByteString as S
|
|
|
|
adjustedBranchPrefix :: S.ByteString
|
|
adjustedBranchPrefix = "refs/heads/adjusted/"
|
|
|
|
class SerializeAdjustment t where
|
|
serializeAdjustment :: t -> S.ByteString
|
|
deserializeAdjustment :: S.ByteString -> Maybe t
|
|
|
|
instance SerializeAdjustment Adjustment where
|
|
serializeAdjustment (LinkAdjustment l) =
|
|
serializeAdjustment l
|
|
serializeAdjustment (PresenceAdjustment p Nothing) =
|
|
serializeAdjustment p
|
|
serializeAdjustment (PresenceAdjustment p (Just l)) =
|
|
serializeAdjustment p <> "-" <> serializeAdjustment l
|
|
deserializeAdjustment s =
|
|
(LinkAdjustment <$> deserializeAdjustment s)
|
|
<|>
|
|
(PresenceAdjustment <$> deserializeAdjustment s1 <*> pure (deserializeAdjustment s2))
|
|
<|>
|
|
(PresenceAdjustment <$> deserializeAdjustment s <*> pure Nothing)
|
|
where
|
|
(s1, s2) = separate' (== (fromIntegral (ord '-'))) s
|
|
|
|
instance SerializeAdjustment LinkAdjustment where
|
|
serializeAdjustment UnlockAdjustment = "unlocked"
|
|
serializeAdjustment LockAdjustment = "locked"
|
|
serializeAdjustment FixAdjustment = "fixed"
|
|
serializeAdjustment UnFixAdjustment = "unfixed"
|
|
deserializeAdjustment "unlocked" = Just UnlockAdjustment
|
|
deserializeAdjustment "locked" = Just LockAdjustment
|
|
deserializeAdjustment "fixed" = Just FixAdjustment
|
|
deserializeAdjustment "unfixed" = Just UnFixAdjustment
|
|
deserializeAdjustment _ = Nothing
|
|
|
|
instance SerializeAdjustment PresenceAdjustment where
|
|
serializeAdjustment HideMissingAdjustment = "hidemissing"
|
|
serializeAdjustment ShowMissingAdjustment = "showmissing"
|
|
deserializeAdjustment "hidemissing" = Just HideMissingAdjustment
|
|
deserializeAdjustment "showmissing" = Just ShowMissingAdjustment
|
|
deserializeAdjustment _ = Nothing
|
|
|
|
newtype AdjBranch = AdjBranch { adjBranch :: Branch }
|
|
|
|
originalToAdjusted :: OrigBranch -> Adjustment -> AdjBranch
|
|
originalToAdjusted orig adj = AdjBranch $ Ref $
|
|
adjustedBranchPrefix <> base <> "(" <> serializeAdjustment adj <> ")"
|
|
where
|
|
base = fromRef' (Git.Ref.base orig)
|
|
|
|
type OrigBranch = Branch
|
|
|
|
adjustedToOriginal :: Branch -> Maybe (Adjustment, OrigBranch)
|
|
adjustedToOriginal b
|
|
| adjustedBranchPrefix `S.isPrefixOf` bs = do
|
|
let (base, as) = separate' (== openparen) (S.drop prefixlen bs)
|
|
adj <- deserializeAdjustment (S.takeWhile (/= closeparen) as)
|
|
Just (adj, Git.Ref.branchRef (Ref base))
|
|
| otherwise = Nothing
|
|
where
|
|
bs = fromRef' b
|
|
prefixlen = S.length adjustedBranchPrefix
|
|
openparen = fromIntegral (ord '(')
|
|
closeparen = fromIntegral (ord ')')
|