Utils.hs

{-# LANGUAGE CPP #-}

-- | Copyright: (c) 2021-2025 berberman
-- SPDX-License-Identifier: MIT
-- Maintainer: berberman <berberman@yandex.com>
-- Stability: experimental
-- Portability: portable
module NvFetcher.Utils where

import Data.Binary
import qualified Data.ByteString as BS
import qualified Data.ByteString.Lazy as LBS
import Data.Text (Text)
import qualified Data.Text as T
import NvFetcher.Types
import System.Directory.Extra (XdgDirectory (..), getXdgDirectory)
import System.FilePath.Glob (Pattern, compile)
import Text.Regex.TDFA ((=~))

#if MIN_VERSION_aeson(2,0,0)
import qualified Data.Aeson.Key as A
#endif

encode' :: (Binary a) => a -> BS.ByteString
encode' = BS.concat . LBS.toChunks . encode

decode' :: (Binary a) => BS.ByteString -> a
decode' = decode . LBS.fromChunks . pure

quote :: Text -> Text
quote x = T.concat [q, x, q]
  where
    q = T.pack "\""

quoteAndEscape :: Text -> Text
quoteAndEscape = T.pack . show

isLegalNixId :: Text -> Bool
isLegalNixId x = x =~ "^[a-zA-Z_][a-zA-Z0-9_'-]*$"

quoteIfNeeds :: Text -> Text
quoteIfNeeds x
  | isLegalNixId x = x
  | otherwise = quote x

getDataDir :: IO FilePath
getDataDir = getXdgDirectory XdgData "nvfetcher"

-- | Compile a 'Glob' pattern into a 'Pattern'.
-- See <https://hackage.haskell.org/package/Glob/docs/System-FilePath-Glob.html#v:compile> for details.
compileGlob :: Glob -> Pattern
compileGlob (Glob txt) = compile txt

#if MIN_VERSION_aeson(2,0,0)
aesonKey :: Text -> A.Key
aesonKey = A.fromText
#else
aesonKey :: Text -> Text
aesonKey = id
#endif