{-# LANGUAGE OverloadedStrings #-}

-- Module      : Text.EDE.Internal.Syntax
-- Copyright   : (c) 2013-2020 Brendan Hay <brendan.g.hay@gmail.com>
-- License     : This Source Code Form is subject to the terms of
--               the Mozilla Public License, v. 2.0.
--               A copy of the MPL can be found in the LICENSE file or
--               you can obtain it at http://mozilla.org/MPL/2.0/.
-- Maintainer  : Brendan Hay <brendan.g.hay@gmail.com>
-- Stability   : experimental
-- Portability : non-portable (GHC extensions)

module Text.EDE.Internal.Syntax where

import Control.Lens
import Data.HashSet (HashSet)
import qualified Data.HashSet as Set
import Text.EDE.Internal.Types
import Text.Parser.Token.Style
import Text.Trifecta

-- | The default ED-E syntax.
--
-- Delimiters:
--
-- * Pragma: @{! ... !}@
--
-- * Inline: @{{ ... }}@
--
-- * Comments: @{# ... #}@
--
-- * Blocks: @{% ... %}@
defaultSyntax :: Syntax
defaultSyntax =
  Syntax
    { _delimPragma = ("{!", "!}"),
      _delimInline = ("{{", "}}"),
      _delimComment = ("{#", "#}"),
      _delimBlock = ("{%", "%}")
    }

-- | An alternate syntax (based on Play/Scala templates) designed to
-- be used when the default is potentially ambiguous due to another encountered
-- smarty based syntax.
--
-- Delimiters:
--
-- * Inline: @\<\@ ... \@>@
--
-- * Comments: @\@* ... *\@@
--
-- * Blocks: @\@( ... )\@@
alternateSyntax :: Syntax
alternateSyntax =
  Syntax
    { _delimPragma = ("@!", "!@"),
      _delimInline = ("<@", "@>"),
      _delimComment = ("@*", "*@"),
      _delimBlock = ("@(", ")@")
    }

commentStyle :: String -> String -> CommentStyle
commentStyle s e = emptyCommentStyle & commentStart .~ s & commentEnd .~ e

operatorStyle :: TokenParsing m => IdentifierStyle m
operatorStyle = haskellOps & styleLetter .~ oneOf "-+!&|=><"

variableStyle :: TokenParsing m => IdentifierStyle m
variableStyle = keywordStyle & styleName .~ "variable"

keywordStyle :: TokenParsing m => IdentifierStyle m
keywordStyle =
  haskellIdents
    & styleReserved .~ keywordSet
    & styleName .~ "keyword"

keywordSet :: HashSet String
keywordSet =
  Set.fromList
    [ "if",
      "elif",
      "else",
      "case",
      "when",
      "for",
      "include",
      "let",
      "endif",
      "endcase",
      "endfor",
      "endlet",
      "in",
      "with",
      "_",
      ".",
      "true",
      "false"
    ]

pragmaStyle :: TokenParsing m => IdentifierStyle m
pragmaStyle =
  haskellIdents
    & styleReserved .~ pragmaSet
    & styleName .~ "pragma field"

pragmaSet :: HashSet String
pragmaSet =
  Set.fromList
    [ "pragma",
      "inline",
      "comment",
      "block"
    ]