Taimoor ZaeemandSteve Chavez
a688878236
When db-pre-config is accidentally set to a pg reserved word like "true", it fails with a confusing error. The function names should be properly quoted to avoid such errors. This commit resolves this by quoting the pre-config function name. Signed-off-by: Taimoor Zaeem <taimoorzaeem@gmail.com>
65 lines
1.8 KiB
Haskell
65 lines
1.8 KiB
Haskell
{-# LANGUAGE DeriveAnyClass #-}
|
|
{-# LANGUAGE DeriveGeneric #-}
|
|
|
|
module PostgREST.SchemaCache.Identifiers
|
|
( FieldName
|
|
, QualifiedIdentifier(..)
|
|
, RelIdentifier(..)
|
|
, Schema
|
|
, TableName
|
|
, dumpQi
|
|
, escapeIdent
|
|
, isAnyElement
|
|
, quoteQi
|
|
, toQi
|
|
, trimNullChars
|
|
) where
|
|
|
|
import qualified Data.Aeson as JSON
|
|
import qualified Data.Text as T
|
|
|
|
import Protolude
|
|
|
|
data RelIdentifier = RelId QualifiedIdentifier | RelAnyElement
|
|
deriving (Eq, Ord, Generic, JSON.ToJSON, JSON.ToJSONKey, Show)
|
|
instance Hashable RelIdentifier
|
|
|
|
-- | Represents a pg identifier with a prepended schema name "schema.table".
|
|
-- When qiSchema is "", the schema is defined by the pg search_path.
|
|
-- TODO: Refactor this, we also use QI for procedure names
|
|
data QualifiedIdentifier = QualifiedIdentifier
|
|
{ qiSchema :: Schema
|
|
, qiName :: TableName
|
|
}
|
|
deriving (Eq, Show, Ord, Generic, JSON.ToJSON, JSON.ToJSONKey)
|
|
|
|
instance Hashable QualifiedIdentifier
|
|
|
|
isAnyElement :: QualifiedIdentifier -> Bool
|
|
isAnyElement y = QualifiedIdentifier "pg_catalog" "anyelement" == y
|
|
|
|
dumpQi :: QualifiedIdentifier -> Text
|
|
dumpQi (QualifiedIdentifier s i) =
|
|
(if T.null s then mempty else s <> ".") <> i
|
|
|
|
quoteQi :: QualifiedIdentifier -> Text
|
|
quoteQi (QualifiedIdentifier s i) =
|
|
(if T.null s then mempty else escapeIdent s <> ".") <> escapeIdent i
|
|
|
|
-- TODO: Handle a case where the QI comes like this: "my.fav.schema"."my.identifier"
|
|
-- Right now it only handles the schema.identifier case
|
|
toQi :: Text -> QualifiedIdentifier
|
|
toQi txt = case T.drop 1 <$> T.breakOn "." txt of
|
|
(i, "") -> QualifiedIdentifier mempty i
|
|
(s, i) -> QualifiedIdentifier s i
|
|
|
|
escapeIdent :: Text -> Text
|
|
escapeIdent x = "\"" <> T.replace "\"" "\"\"" (trimNullChars x) <> "\""
|
|
|
|
trimNullChars :: Text -> Text
|
|
trimNullChars = T.takeWhile (/= '\x0')
|
|
|
|
type Schema = Text
|
|
type TableName = Text
|
|
type FieldName = Text
|