Show values (except Lists)

[my-scheme.git] / app / Main.hs

diff --git a/app/Main.hs b/app/Main.hs
index 6fcf6675f175374d22b749097dc06142312b8400..1495605d09602a2ed23717d3b5956402874a6ef4 100644 (file)
--- a/app/Main.hs
+++ b/app/Main.hs
@@ -1,13 +1,17 @@
 module Main where
 module Main where

+import Control.Monad

 import Text.ParserCombinators.Parsec hiding (spaces)
 import System.Environment
 import Text.ParserCombinators.Parsec hiding (spaces)
 import System.Environment

+import Numeric

 
 data LispVal = Atom String
              | List [LispVal]
              | DottedList [LispVal] LispVal
              | Number Integer
              | String String
 
 data LispVal = Atom String
              | List [LispVal]
              | DottedList [LispVal] LispVal
              | Number Integer
              | String String

+             | Character Char

              | Bool Bool
              | Bool Bool

+             deriving Show

 
 symbol :: Parser Char
 symbol = oneOf "!#$%&|*+-/:<=>?@^_~"
 
 symbol :: Parser Char
 symbol = oneOf "!#$%&|*+-/:<=>?@^_~"


@@ -15,12 +19,29 @@ symbol = oneOf "!#$%&|*+-/:<=>?@^_~"
 spaces :: Parser ()
 spaces = skipMany space
 
 spaces :: Parser ()
 spaces = skipMany space
 

+nonPrintableChar :: Parser Char
+nonPrintableChar = do c <- char '\\' >> oneOf "\\nrtf"
+                      return $ case c of '\\' -> '\\'
+                                         'n'  -> '\n'
+                                         'r'  -> '\r'
+                                         't'  -> '\t'
+                                         'f'  -> '\f'
+
+parseCharacter :: Parser LispVal
+parseCharacter = do char '\''
+                    c <- noneOf ['\\', '\''] <|> try singleQuote <|> try nonPrintableChar
+                    char '\''
+                    return $ Character c
+                 where
+                    singleQuote = char '\\' >> char '\''
+

 parseString :: Parser LispVal
 parseString :: Parser LispVal

-parseString = do
-    char '"'
-    x <- many (noneOf "\"")
-    char '"'
-    return $ String x
+parseString = do char '"'
+                 x <- many innerChar
+                 char '"'
+                 return $ String x
+              where innerChar = noneOf ['\\', '"'] <|> try doubleQuote <|> try nonPrintableChar
+                    doubleQuote = char '\\' >> char '"'

 
 parseAtom :: Parser LispVal
 parseAtom = do
 
 parseAtom :: Parser LispVal
 parseAtom = do


@@ -32,12 +53,59 @@ parseAtom = do
                 "#f" -> Bool False
                 _    -> Atom atom
 
                 "#f" -> Bool False
                 _    -> Atom atom
 

+parseNumber :: Parser LispVal
+parseNumber = do toNum <- radix
+                 ds <- many1 digit
+                 let ((a,_):_) = toNum ds
+                 return $ Number a
+              where radix = do r <- try (char '#' >> oneOf "bodx") <|> return 'd'
+                               return $ case r of
+                                           'd' -> readDec
+                                           'x' -> readHex
+                                           'o' -> readOct
+                                           'b' -> readInt 2 (\x -> elem x "01") (read . (:[]))
+
+parseList :: Parser LispVal
+parseList = liftM List $ sepBy parseExpr spaces
+
+parseDottedList :: Parser LispVal
+parseDottedList = do
+    head <- endBy parseExpr spaces
+    tail <- char '.' >> spaces >> parseExpr
+    return $ DottedList head tail
+
+parseQuoted :: Parser LispVal
+parseQuoted = do
+    char '\''
+    x <- parseExpr
+    return $ List [Atom "quote", x]
+
+parseExpr :: Parser LispVal
+parseExpr = parseString
+        <|> parseNumber
+        <|> parseAtom
+        <|> try parseCharacter
+        <|> parseQuoted
+        <|> do char '('
+               x <- try parseList <|> parseDottedList
+               char ')'
+               return x
+

 readExpr :: String -> String
 readExpr :: String -> String

-readExpr input = case parse (spaces >> symbol) "lisp" input of
+readExpr input = case parse parseExpr "lisp" input of

     Left err -> "No match: " ++ show err
     Left err -> "No match: " ++ show err

-    Right val -> "Found value"
+    Right val -> "Found value: " ++ showVal val
+
+showVal :: LispVal -> String
+showVal (Atom atom)   = atom
+showVal (Number n)    = show n
+showVal (String str)  = "\"" ++ str ++ "\""
+showVal (Character c) = "'" ++ [c] ++ "'"
+showVal (Bool True)   = "#t"
+showVal (Bool False)  = "#f"

 
 main :: IO ()
 main = do
     args <- getLine
     putStrLn (readExpr args)
 
 main :: IO ()
 main = do
     args <- getLine
     putStrLn (readExpr args)

+    main