mirror of
https://github.com/github/semantic.git
synced 2024-12-21 13:51:44 +03:00
75 lines
2.3 KiB
Haskell
75 lines
2.3 KiB
Haskell
{-# LANGUAGE TypeApplications #-}
|
|
module Main (main) where
|
|
|
|
import System.Environment
|
|
import TreeSitter.Unmarshal
|
|
import TreeSitter.Python.AST
|
|
import TreeSitter.Python
|
|
import Source.Range
|
|
import Source.Span
|
|
import Data.ByteString.Char8
|
|
import Data.ByteString (pack, readFile, ByteString)
|
|
import System.IO (FilePath)
|
|
import Options.Applicative hiding (style)
|
|
import Data.Semigroup ((<>))
|
|
import Text.Pretty.Simple (pPrint, pPrintNoColor)
|
|
|
|
data SemanticAST = SemanticAST
|
|
{ format :: Format
|
|
, color :: Bool
|
|
, source :: Either FilePath Prelude.String
|
|
}
|
|
|
|
parseAST :: Parser SemanticAST
|
|
parseAST = SemanticAST
|
|
<$> option auto
|
|
( long "format"
|
|
<> help "Specify desired output: show, json, sexpression" )
|
|
<*> switch
|
|
( long "no-color"
|
|
<> help "Print with color: --color"
|
|
)
|
|
<*> (Left <$> strOption
|
|
( long "sourceFile"
|
|
<> metavar "FILEPATH"
|
|
<> help "Specify filepath containing source code to parse" )
|
|
<|> Right <$> strOption
|
|
( long "sourceString"
|
|
<> metavar "STRING"
|
|
<> help "Specify source input to parse"
|
|
))
|
|
|
|
main :: IO ()
|
|
main = generateAST =<< execParser opts
|
|
|
|
generateAST :: SemanticAST -> IO ()
|
|
generateAST (SemanticAST format color source) = do
|
|
bytestring <- case source of
|
|
Left filePath -> do
|
|
Data.ByteString.readFile filePath
|
|
Right source -> do
|
|
pure $ Data.ByteString.Char8.pack source
|
|
ast <- parseByteString @TreeSitter.Python.AST.Module @(Range, Span) tree_sitter_python bytestring
|
|
case format of
|
|
Show -> print ast
|
|
Pretty -> pPrint ast
|
|
|
|
opts :: ParserInfo SemanticAST
|
|
opts = info (parseAST <**> helper)
|
|
( fullDesc
|
|
<> progDesc "Parse source code and produce an AST"
|
|
<> header "semantic-ast is a package used to parse source code" )
|
|
|
|
-- TODO: Define formats for json, sexpression, etc.
|
|
data Format = Show | Pretty
|
|
deriving (Read)
|
|
|
|
-- CLI feature request -> implementation
|
|
-- add a boolean flag that represents "print with color" or not
|
|
-- `semantic-ast --format=Pretty --no-color` won't show those pretty colors
|
|
-- adding a boolean field to represent whether color is on or not
|
|
-- and use an appropriate construct in optparse-applicative to indicate that `--color` means True
|
|
-- and -- no-color means false, and not provided means True
|
|
|
|
-- The difference between auto and string option
|