packages feed

postgresql-syntax-0.5.0.2: library-internal/PostgresqlSyntax/Ast/OverrideKind.hs

module PostgresqlSyntax.Ast.OverrideKind where

import PostgresqlSyntax.Algebra
import qualified PostgresqlSyntax.Helpers.Parsers as Parsers
import PostgresqlSyntax.Prelude
import qualified Test.QuickCheck as Qc

-- |
-- ==== References
-- @
-- override_kind:
--   | USER
--   | SYSTEM_P
-- @
data OverrideKind = UserOverrideKind | SystemOverrideKind
  deriving (Show, Generic, Eq, Ord, Data, Enum, Bounded)

instance IsAst OverrideKind where
  toTextBuilder _settings = \case
    UserOverrideKind -> "USER"
    SystemOverrideKind -> "SYSTEM"
  parser _settings =
    asum
      [ UserOverrideKind <$ Parsers.keyword "user",
        SystemOverrideKind <$ Parsers.keyword "system"
      ]

instance Qc.Arbitrary OverrideKind where
  shrink = Qc.genericShrink
  arbitrary = Qc.elements [minBound .. maxBound]