Compare commits
No commits in common. "be5e487e3154fdf2e8a083e3f33b007933752c3d" and "fbb0fb27fbeaa394aa9dc01b1fe8c7fbc4606d34" have entirely different histories.
be5e487e31
...
fbb0fb27fb
@ -1,71 +0,0 @@
|
||||
module Datalog.Parser
|
||||
( parseTerm
|
||||
, parseAtom
|
||||
, parseRule
|
||||
, parseQuery
|
||||
, parseProgram
|
||||
)
|
||||
where
|
||||
|
||||
import Datalog.Syntax
|
||||
import Data.Text (Text)
|
||||
import Data.Text qualified as T
|
||||
import Data.Void (Void)
|
||||
import Text.Megaparsec
|
||||
import Text.Megaparsec.Char
|
||||
import Text.Megaparsec.Char.Lexer qualified as L
|
||||
|
||||
type Parser = Parsec Void Text
|
||||
|
||||
lexeme :: (MonadParsec e Text m) => m a -> m a
|
||||
lexeme = L.lexeme whitespace
|
||||
|
||||
symbol :: (MonadParsec e Text m) => Text -> m Text
|
||||
symbol = L.symbol whitespace
|
||||
|
||||
whitespace :: (MonadParsec e Text m) => m ()
|
||||
whitespace = L.space
|
||||
space1
|
||||
(L.skipLineComment "--")
|
||||
(L.skipBlockComment "{-" "-}")
|
||||
|
||||
parens :: (MonadParsec e Text m) => m a -> m a
|
||||
parens = between (symbol "(") (symbol ")")
|
||||
|
||||
comma, period :: (MonadParsec e Text m) => m ()
|
||||
comma = () <$ symbol ","
|
||||
period = () <$ symbol "."
|
||||
|
||||
parseCon :: (MonadParsec e Text m) => m Term
|
||||
parseCon = Con . ConId . T.pack <$> lexeme (liftA2 (:) (numberChar <|> lowerChar) (many alphaNumChar))
|
||||
|
||||
parseVar :: (MonadParsec e Text m) => m Term
|
||||
parseVar = Var . VarId . T.pack <$> lexeme (liftA2 (:) upperChar (many alphaNumChar))
|
||||
|
||||
parseTerm :: Parser Term
|
||||
parseTerm = parseVar <|> parseCon
|
||||
|
||||
parseAtom :: Parser Atom
|
||||
parseAtom = do
|
||||
rel <- RelId . T.pack <$> lexeme (liftA2 (:) lowerChar (many alphaNumChar))
|
||||
args <- parens (parseTerm `sepBy` comma)
|
||||
return (Atom rel args)
|
||||
|
||||
parseQuery :: Parser [Atom]
|
||||
parseQuery = parseAtom `sepBy` comma
|
||||
|
||||
parseFact :: Parser Rule
|
||||
parseFact = do
|
||||
headAtom <- parseAtom
|
||||
period
|
||||
return (headAtom :- [])
|
||||
|
||||
parseRule :: Parser Rule
|
||||
parseRule = try parseFact <|> do
|
||||
headAtom <- parseAtom <* symbol ":-"
|
||||
bodyAtoms <- parseQuery
|
||||
period
|
||||
return (headAtom :- bodyAtoms)
|
||||
|
||||
parseProgram :: Parser Program
|
||||
parseProgram = Program <$> many parseRule
|
||||
@ -1,86 +0,0 @@
|
||||
module Datalog.Syntax where
|
||||
|
||||
import Data.Char (isUpper)
|
||||
import Data.Text (Text)
|
||||
import Data.Text qualified as T
|
||||
import Data.Set qualified as Set
|
||||
import Data.Set (Set)
|
||||
|
||||
newtype ConId = ConId Text
|
||||
deriving (Eq, Ord, Show)
|
||||
|
||||
newtype VarId = VarId Text
|
||||
deriving (Eq, Ord, Show)
|
||||
|
||||
newtype RelId = RelId Text
|
||||
deriving (Eq, Ord, Show)
|
||||
|
||||
data Term = Con ConId | Var VarId
|
||||
deriving (Eq, Ord, Show)
|
||||
|
||||
con :: Text -> Term
|
||||
con = Con . ConId
|
||||
|
||||
var :: Text -> Term
|
||||
var = Var . VarId
|
||||
|
||||
term :: Text -> Term
|
||||
term t = if not (T.null t) && isUpper (T.head t) then var t else con t
|
||||
|
||||
data Atom = Atom RelId [Term]
|
||||
deriving (Eq, Ord, Show)
|
||||
|
||||
atom :: Text -> [Text] -> Atom
|
||||
atom relName args = Atom (RelId relName) (map term args)
|
||||
|
||||
data Rule = Atom :- [Atom]
|
||||
deriving (Eq, Ord, Show)
|
||||
|
||||
data Program = Program [Rule]
|
||||
deriving (Eq, Ord, Show)
|
||||
|
||||
class HasConstants a where
|
||||
constants :: a -> Set ConId
|
||||
|
||||
instance HasConstants Term where
|
||||
constants t = case t of
|
||||
Con x -> Set.singleton x
|
||||
Var _ -> Set.empty
|
||||
|
||||
instance HasConstants a => HasConstants [a] where
|
||||
constants xs = Set.unions (map constants xs)
|
||||
|
||||
instance HasConstants Atom where
|
||||
constants (Atom _ ts) = constants ts
|
||||
|
||||
instance HasConstants Rule where
|
||||
constants (h :- b) = Set.union (constants h) (constants b)
|
||||
|
||||
instance HasConstants Program where
|
||||
constants (Program rs) = constants rs
|
||||
|
||||
class Pretty t where
|
||||
pretty :: t -> Text
|
||||
|
||||
instance Pretty ConId where
|
||||
pretty (ConId t) = t
|
||||
|
||||
instance Pretty VarId where
|
||||
pretty (VarId t) = t
|
||||
|
||||
instance Pretty RelId where
|
||||
pretty (RelId t) = t
|
||||
|
||||
instance Pretty Term where
|
||||
pretty (Con conId) = pretty conId
|
||||
pretty (Var varId) = pretty varId
|
||||
|
||||
instance Pretty Atom where
|
||||
pretty (Atom relId terms) = pretty relId <> "(" <> T.intercalate "," (map pretty terms) <> ")"
|
||||
|
||||
instance Pretty Rule where
|
||||
pretty (h :- []) = pretty h <> "."
|
||||
pretty (h :- ts) = pretty h <> " :- " <> T.intercalate ", " (map pretty ts) <> "."
|
||||
|
||||
instance Pretty Program where
|
||||
pretty (Program rs) = T.unlines (map pretty rs)
|
||||
@ -15,32 +15,13 @@ extra-doc-files: CHANGELOG.md
|
||||
common warnings
|
||||
ghc-options: -Wall
|
||||
|
||||
common extensions
|
||||
default-extensions:
|
||||
OverloadedRecordDot
|
||||
OverloadedStrings
|
||||
|
||||
library datalog-parser
|
||||
import: warnings, extensions
|
||||
build-depends:
|
||||
base,
|
||||
text,
|
||||
containers,
|
||||
megaparsec
|
||||
exposed-modules:
|
||||
Datalog.Parser,
|
||||
Datalog.Syntax
|
||||
hs-source-dirs: datalog/src
|
||||
default-language: GHC2024
|
||||
|
||||
executable datalog-lsp
|
||||
import: warnings, extensions
|
||||
executable geolog-lsp
|
||||
import: warnings
|
||||
main-is: Main.hs
|
||||
build-depends:
|
||||
base,
|
||||
-- other-modules:
|
||||
-- other-extensions:
|
||||
build-depends: base,
|
||||
lsp,
|
||||
text,
|
||||
containers,
|
||||
lens
|
||||
hs-source-dirs: datalog-lsp/src
|
||||
text
|
||||
hs-source-dirs: src
|
||||
default-language: GHC2024
|
||||
|
||||
@ -1,41 +1,39 @@
|
||||
{-# LANGUAGE DuplicateRecordFields #-}
|
||||
{-# LANGUAGE OverloadedStrings #-}
|
||||
|
||||
import Control.Lens ((^.))
|
||||
import Control.Monad.IO.Class
|
||||
import Data.Either (fromRight)
|
||||
import Data.Maybe (fromJust)
|
||||
import Language.LSP.Protocol.Lens (params, textDocument, uri)
|
||||
import Language.LSP.Protocol.Message
|
||||
import Language.LSP.Protocol.Types
|
||||
import Language.LSP.Server
|
||||
import Language.LSP.VFS (virtualFileText)
|
||||
import Data.Either (fromRight)
|
||||
|
||||
handlers :: Handlers (LspM ())
|
||||
handlers =
|
||||
mconcat
|
||||
[ notificationHandler SMethod_Initialized $ \_ -> pure ()
|
||||
, requestHandler SMethod_TextDocumentHover $ \req responder -> do
|
||||
let
|
||||
docUri = toNormalizedUri $ req ^. (params . textDocument . uri)
|
||||
TRequestMessage _ _ _ (HoverParams _doc pos _workDone) = req
|
||||
let TRequestMessage _ _ _ (HoverParams _doc pos _workDone) = req
|
||||
Position _l _c' = pos
|
||||
rsp txt = Hover (InL . mkMarkdown $ txt) (Just range)
|
||||
rsp = Hover (InL ms) (Just range)
|
||||
ms = mkMarkdown "Hello world"
|
||||
range = Range pos pos
|
||||
c <- fromJust <$> getVirtualFile docUri
|
||||
responder (Right . InL . rsp . virtualFileText $ c)
|
||||
responder (Right $ InL rsp)
|
||||
, requestHandler SMethod_TextDocumentSemanticTokensFull $ \req responder -> do
|
||||
let TRequestMessage _ _ _ (SemanticTokensParams _doc _workDone _partial) = req
|
||||
docUri = toNormalizedUri $ req ^. (params . textDocument . uri)
|
||||
token =
|
||||
SemanticTokenAbsolute
|
||||
{ _line = 0
|
||||
, _startChar = 0
|
||||
, _length = 5
|
||||
, _tokenType = SemanticTokenTypes_Keyword
|
||||
, _tokenModifiers = []
|
||||
}
|
||||
|
||||
-- Each token is encoded as 5 uints:
|
||||
-- [deltaLine, deltaStartChar, length, tokenTypeIndex, tokenModifiersBitset]
|
||||
-- This example returns a single token at (0,0) of length 5.
|
||||
let -- tokenTypeIndex=0 is "whatever the legend's 0 is"
|
||||
token = SemanticTokenAbsolute {
|
||||
_line = 0,
|
||||
_startChar = 0,
|
||||
_length = 5,
|
||||
_tokenType = SemanticTokenTypes_Keyword,
|
||||
_tokenModifiers = []
|
||||
}
|
||||
tokens = fromRight (error "failed") $ makeSemanticTokens defaultSemanticTokensLegend [token]
|
||||
c <- fromJust <$> getVirtualFile docUri
|
||||
|
||||
responder (Right $ InL tokens)
|
||||
]
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user