2020-01-10 23:14:05 +03:00
|
|
|
|
{-# LANGUAGE AllowAmbiguousTypes #-}
|
2020-01-10 20:22:13 +03:00
|
|
|
|
{-# LANGUAGE OverloadedStrings #-}
|
|
|
|
|
{-# LANGUAGE ScopedTypeVariables #-}
|
|
|
|
|
{-# LANGUAGE TypeApplications #-}
|
|
|
|
|
{-# LANGUAGE TypeOperators #-}
|
2020-01-14 19:13:00 +03:00
|
|
|
|
{-# OPTIONS_GHC -Wno-incomplete-uni-patterns #-}
|
2020-01-08 21:02:53 +03:00
|
|
|
|
module Main (main) where
|
2020-01-08 20:51:37 +03:00
|
|
|
|
|
2020-01-10 23:14:05 +03:00
|
|
|
|
import Control.Algebra
|
|
|
|
|
import Control.Carrier.Sketch.Fresh
|
2020-01-10 20:22:13 +03:00
|
|
|
|
import Control.Monad
|
2020-01-13 22:05:31 +03:00
|
|
|
|
import Convert.ToScopeGraph
|
2020-01-08 21:13:48 +03:00
|
|
|
|
import qualified Data.ByteString as ByteString
|
2020-01-14 22:19:57 +03:00
|
|
|
|
import Data.Name (Name)
|
2020-01-08 21:25:43 +03:00
|
|
|
|
import qualified Data.ScopeGraph as ScopeGraph
|
|
|
|
|
import qualified Language.Python ()
|
2020-01-10 20:22:13 +03:00
|
|
|
|
import Source.Loc
|
|
|
|
|
import qualified Source.Source as Source
|
|
|
|
|
import System.Exit (die)
|
2020-01-13 18:08:07 +03:00
|
|
|
|
import qualified System.Path as Path
|
2020-01-08 21:13:48 +03:00
|
|
|
|
import qualified TreeSitter.Python as TSP
|
2020-01-10 20:22:13 +03:00
|
|
|
|
import qualified TreeSitter.Python.AST as Py
|
2020-01-08 21:13:48 +03:00
|
|
|
|
import qualified TreeSitter.Unmarshal as TS
|
2020-01-10 20:22:13 +03:00
|
|
|
|
|
|
|
|
|
{-
|
|
|
|
|
|
|
|
|
|
The Python code here is
|
|
|
|
|
|
|
|
|
|
hello = ()
|
|
|
|
|
goodbye = ()
|
|
|
|
|
|
|
|
|
|
The graph should be
|
|
|
|
|
|
|
|
|
|
🏁
|
|
|
|
|
|
|
|
|
|
|
1️⃣----"hello"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2️⃣----"goodbye"
|
|
|
|
|
|
|
|
|
|
-}
|
|
|
|
|
|
2020-01-13 18:08:07 +03:00
|
|
|
|
|
2020-01-14 22:19:57 +03:00
|
|
|
|
runScopeGraph :: ToScopeGraph t => Path.AbsRelFile -> Source.Source -> t Loc -> (ScopeGraph.ScopeGraph Name, Result)
|
|
|
|
|
runScopeGraph p _src item = run . runSketch @Name (Just p) $ scopeGraph item
|
2020-01-13 18:08:07 +03:00
|
|
|
|
|
2020-01-14 22:19:57 +03:00
|
|
|
|
sampleGraphThing :: (Has (Sketch Name) sig m) => m Result
|
2020-01-10 23:14:05 +03:00
|
|
|
|
sampleGraphThing = do
|
2020-01-14 22:19:57 +03:00
|
|
|
|
declare @Name "hello" DeclProperties
|
|
|
|
|
declare @Name "goodbye" DeclProperties
|
2020-01-14 19:13:00 +03:00
|
|
|
|
pure Complete
|
2020-01-10 23:14:05 +03:00
|
|
|
|
|
2020-01-10 20:22:13 +03:00
|
|
|
|
|
|
|
|
|
assertEqual :: (Show a, Eq a) => a -> a -> IO ()
|
|
|
|
|
assertEqual a b = unless (a == b) (die (show a <> "\ndoes not equal\n" <> show b))
|
2020-01-08 21:13:48 +03:00
|
|
|
|
|
2020-01-08 20:51:37 +03:00
|
|
|
|
main :: IO ()
|
2020-01-08 21:13:48 +03:00
|
|
|
|
main = do
|
2020-01-13 23:06:15 +03:00
|
|
|
|
let path = "semantic-python/test/fixtures/1-04-toplevel-assignment.py"
|
2020-01-13 18:08:07 +03:00
|
|
|
|
file <- ByteString.readFile path
|
2020-01-08 21:25:43 +03:00
|
|
|
|
tree <- TS.parseByteString @Py.Module @Loc TSP.tree_sitter_python file
|
|
|
|
|
pyModule <- either die pure tree
|
2020-01-14 22:19:57 +03:00
|
|
|
|
let (expecto, Complete) = run $ runSketch @Name Nothing sampleGraphThing
|
2020-01-14 19:13:00 +03:00
|
|
|
|
let (result, Complete) = runScopeGraph (Path.absRel path) (Source.fromUTF8 file) pyModule
|
2020-01-10 20:22:13 +03:00
|
|
|
|
print result
|
|
|
|
|
assertEqual expecto result
|
|
|
|
|
|