blob: 01e57bb781cea53bd982f8b6331e6c7e6d17fad5 (
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
|
-- Copyright 2018 Google LLC
--
-- Licensed under the Apache License, Version 2.0 (the "License"); you may not
-- use this file except in compliance with the License. You may obtain a copy of
-- the License at
--
-- https://www.apache.org/licenses/LICENSE-2.0
--
-- Unless required by applicable law or agreed to in writing, software
-- distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
-- WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
-- License for the specific language governing permissions and limitations under
-- the License.
{-# LANGUAGE OverloadedStrings #-}
module BTLS.TestUtilities
( abbreviate
, hex
) where
import qualified Data.ByteString.Base16 as Base16
import qualified Data.ByteString.Base16.Lazy as L.Base16
import Data.ByteString.Char8 (ByteString, unpack)
import qualified Data.ByteString.Lazy.Char8 as Lazy (ByteString)
import qualified Data.ByteString.Lazy.Char8 as L
import Data.Char (isAscii, isPrint)
import Data.Int (Int64)
abbreviate :: Lazy.ByteString -> String
abbreviate input =
let maxLen = 22 in
if L.all isShowable (L.take (maxLen - 2) input)
then show (addEllipsisIfNecessary (maxLen - 2) input)
else L.unpack (addEllipsisIfNecessary maxLen (L.Base16.encode input))
where isShowable c = isAscii c && isPrint c
addEllipsisIfNecessary :: Int64 -> Lazy.ByteString -> Lazy.ByteString
addEllipsisIfNecessary maxLen s =
let ellipsis = "..."
ellipsisLen = L.length ellipsis
(x, y) = L.splitAt (maxLen - ellipsisLen) s in
x `L.append` if L.length y <= ellipsisLen then y else ellipsis
hex :: ByteString -> ByteString
hex s =
case Base16.decode s of
(r, "") -> r
_ -> error $ "invalid hex string " ++ unpack s
|
