module Ki.Internal.ByteCount
  ( ByteCount,
    kilobytes,
    megabytes,
    byteCountToInt64,
  )
where

import Data.Coerce (coerce)
import Data.Int (Int64)
import Numeric.Natural (Natural)

-- | A number of bytes.
newtype ByteCount = ByteCount Int64
  deriving newtype (ByteCount -> ByteCount -> Bool
(ByteCount -> ByteCount -> Bool)
-> (ByteCount -> ByteCount -> Bool) -> Eq ByteCount
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
$c== :: ByteCount -> ByteCount -> Bool
== :: ByteCount -> ByteCount -> Bool
$c/= :: ByteCount -> ByteCount -> Bool
/= :: ByteCount -> ByteCount -> Bool
Eq, Eq ByteCount
Eq ByteCount =>
(ByteCount -> ByteCount -> Ordering)
-> (ByteCount -> ByteCount -> Bool)
-> (ByteCount -> ByteCount -> Bool)
-> (ByteCount -> ByteCount -> Bool)
-> (ByteCount -> ByteCount -> Bool)
-> (ByteCount -> ByteCount -> ByteCount)
-> (ByteCount -> ByteCount -> ByteCount)
-> Ord ByteCount
ByteCount -> ByteCount -> Bool
ByteCount -> ByteCount -> Ordering
ByteCount -> ByteCount -> ByteCount
forall a.
Eq a =>
(a -> a -> Ordering)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> a)
-> (a -> a -> a)
-> Ord a
$ccompare :: ByteCount -> ByteCount -> Ordering
compare :: ByteCount -> ByteCount -> Ordering
$c< :: ByteCount -> ByteCount -> Bool
< :: ByteCount -> ByteCount -> Bool
$c<= :: ByteCount -> ByteCount -> Bool
<= :: ByteCount -> ByteCount -> Bool
$c> :: ByteCount -> ByteCount -> Bool
> :: ByteCount -> ByteCount -> Bool
$c>= :: ByteCount -> ByteCount -> Bool
>= :: ByteCount -> ByteCount -> Bool
$cmax :: ByteCount -> ByteCount -> ByteCount
max :: ByteCount -> ByteCount -> ByteCount
$cmin :: ByteCount -> ByteCount -> ByteCount
min :: ByteCount -> ByteCount -> ByteCount
Ord)

instance Show ByteCount where
  show :: ByteCount -> String
show (ByteCount Int64
b)
    | (Int64
mb, Int64
0) <- Int64 -> Int64 -> (Int64, Int64)
forall a. Integral a => a -> a -> (a, a)
quotRem Int64
b Int64
1048576, Int64
mb Int64 -> Int64 -> Bool
forall a. Ord a => a -> a -> Bool
> Int64
0 = String
"megabytes " String -> ShowS
forall a. [a] -> [a] -> [a]
++ Int64 -> String
forall a. Show a => a -> String
show Int64
mb
    | (Int64
kb, Int64
0) <- Int64 -> Int64 -> (Int64, Int64)
forall a. Integral a => a -> a -> (a, a)
quotRem Int64
b Int64
1024 = String
"kilobytes " String -> ShowS
forall a. [a] -> [a] -> [a]
++ Int64 -> String
forall a. Show a => a -> String
show Int64
kb
    | Bool
otherwise = String
forall a. HasCallStack => a
undefined

-- | A number of kilobytes.
kilobytes :: Natural -> ByteCount
kilobytes :: Natural -> ByteCount
kilobytes Natural
n =
  Int64 -> ByteCount
ByteCount (Natural -> Int64
snip (Natural
n Natural -> Natural -> Natural
forall a. Num a => a -> a -> a
* Natural
1024))

-- | A number of megabytes.
megabytes :: Natural -> ByteCount
megabytes :: Natural -> ByteCount
megabytes Natural
n =
  Int64 -> ByteCount
ByteCount (Natural -> Int64
snip (Natural
n Natural -> Natural -> Natural
forall a. Num a => a -> a -> a
* Natural
1048576))

byteCountToInt64 :: ByteCount -> Int64
byteCountToInt64 :: ByteCount -> Int64
byteCountToInt64 =
  ByteCount -> Int64
forall a b. Coercible a b => a -> b
coerce

snip :: Natural -> Int64
snip :: Natural -> Int64
snip Natural
n =
  Natural -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral (Natural -> Natural -> Natural
forall a. Ord a => a -> a -> a
min (Int64 -> Natural
forall a b. (Integral a, Num b) => a -> b
fromIntegral (Int64
forall a. Bounded a => a
maxBound :: Int64)) Natural
n)