aboutsummaryrefslogtreecommitdiff
path: root/Key.hs
blob: f52aea31b7843a6865549e2b9f7c5d8b79361a79 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
{- git-annex Key data type
 -
 - Copyright 2011 Joey Hess <joey@kitenet.net>
 -
 - Licensed under the GNU GPL version 3 or higher.
 -}

module Key (
	Key(..),
	stubKey,
	readKey,

	prop_idempotent_key_read_show
) where

import Test.QuickCheck
import Utility
import System.Posix.Types

{- A Key has a unique name, is associated with a key/value backend,
 - and may contain other optional metadata. -}
data Key = Key {
	keyName :: String,
	keyBackendName :: String,
	keySize :: Maybe Integer,
	keyMtime :: Maybe EpochTime
} deriving (Eq, Ord)

stubKey :: Key
stubKey = Key {
	keyName = "",
	keyBackendName = "",
	keySize = Nothing,
	keyMtime = Nothing
}

fieldSep :: Char
fieldSep = '-'

{- Keys show as strings that are suitable for use as filenames.
 - The name field is always shown last, separated by doubled fieldSeps,
 - and is the only field allowed to contain the fieldSep. -}
instance Show Key where
	show Key { keyBackendName = b, keySize = s, keyMtime = m, keyName = n } =
		b +++ ('s' ?: s) +++ ('m' ?: m) +++ (fieldSep : n)
		where
			"" +++ y = y
			x +++ "" = x
			x +++ y = x ++ fieldSep:y
			c ?: (Just v) = c:(show v)
			_ ?: _ = ""

readKey :: String -> Maybe Key
readKey s = if key == Just stubKey then Nothing else key
	where
		key = startbackend stubKey s

		startbackend k v = sepfield k v addbackend
		
		sepfield k v a = case span (/= fieldSep) v of
			(v', _:r) -> findfields r $ a k v'
			_ -> Nothing

		findfields (c:v) (Just k)
			| c == fieldSep = Just $ k { keyName = v }
			| otherwise = sepfield k v $ addfield c
		findfields _ v = v

		addbackend k v = Just k { keyBackendName = v }
		addfield 's' k v = Just k { keySize = readMaybe v }
		addfield 'm' k v = Just k { keyMtime = readMaybe v }
		addfield _ _ _ = Nothing

-- for quickcheck
instance Arbitrary Key where
	arbitrary = do
		n <- arbitrary
		b <- elements ['A'..'Z']
		return $ Key {
			keyName = n,
			keyBackendName = [b],
			keySize = Nothing,
			keyMtime = Nothing
		}

prop_idempotent_key_read_show :: Key -> Bool
prop_idempotent_key_read_show k = Just k == (readKey $ show k)