1{-# LANGUAGE GeneralizedNewtypeDeriving, ConstraintKinds, PatternGuards #-}
3module Idris.Parser(module Idris.Parser,
4 module Idris.ParseExpr,
5 module Idris.ParseData,
6 module Idris.ParseHelpers,
7 module Idris.ParseOps) where
9import Prelude hiding (pi)
11import Text.Trifecta.Delta
12import Text.Trifecta hiding (span, stringLiteral, charLiteral, natural, symbol, char, string, whiteSpace, Err)
13import Text.Parser.LookAhead
14import Text.Parser.Expression
15import qualified Text.Parser.Token as Tok
16import qualified Text.Parser.Char as Chr
17import qualified Text.Parser.Token.Highlight as Hi
19import Text.PrettyPrint.ANSI.Leijen (Doc, plain)
20import qualified Text.PrettyPrint.ANSI.Leijen as ANSI
22import Idris.AbsSyntax hiding (namespace, params)
25import Idris.Delaborate
27import Idris.Elab.Value
36import Idris.ParseHelpers
41import Idris.Docstrings hiding (Unchecked)
45import Util.DynamicLinker
46import qualified Util.Pretty as P
49import Idris.Core.Evaluate
51import Control.Applicative hiding (Const)
53import Control.Monad.State.Strict
57import qualified Data.List.Split as Spl
62import qualified Data.Map as M
63import qualified Data.HashSet as HS
64import qualified Data.Text as T
65import qualified Data.ByteString.UTF8 as UTF8
66import qualified Data.Set as S
75 grammar shortcut notation:
76 ~CHARSEQ = complement of char sequence (i.e. any character except CHARSEQ)
77 RULE? = optional rule (i.e. RULE or nothing)
78 RULE* = repeated rule (i.e. RULE zero or more times)
79 RULE+ = repeated rule with at least one match (i.e. RULE one or more times)
80 RULE! = invalid rule (i.e. rule that is not valid in context, report meaningful error in case)
81 RULE{n} = rule repeated n times
87{- | Parses module definition
90 ModuleHeader ::= DocComment_t? 'module' Identifier_t ';'?;
93moduleHeader :: IdrisParser (Maybe (Docstring ()), [String])
94moduleHeader = try (do docs <- optional docComment
98 option ';' (lchar ';')
99 return (fmap fst docs, moduleName i))
100 <|> try (do lchar '%'; reserved "unqualified"
101 return (Nothing, []))
102 <|> return (Nothing, moduleName "Main")
103 where moduleName x = case span (/='.') x of
105 (x, '.':y) -> x : moduleName y
106 noArgs (Just (_, args)) | not (null args) = fail "Modules do not take arguments"
109{- | Parses an import statement
112 Import ::= 'import' Identifier_t ';'?;
115import_ :: IdrisParser (Bool, String, Maybe String, FC)
116import_ = do fc <- getFC
118 reexport <- option False (do reserved "public"; return True)
120 newName <- optional (reserved "as" *> identifier)
121 option ';' (lchar ';')
122 return (reexport, toPath id, toPath <$> newName, fc)
123 <?> "import statement"
124 where toPath = foldl1' (</>) . Spl.splitOn "."
126{- | Parses program source
132prog :: SyntaxInfo -> IdrisParser [PDecl]
133prog syn = do whiteSpace
134 decls <- many (decl syn)
135 let c = (concat decls)
137 Nothing -> do notOpenBraces; eof
141 put ist { idris_parsedSpan = Just (FC (fc_fname fc) (0,0) (fc_end fc)),
142 ibc_write = IBCParsedRegion fc : ibc_write ist }
145{-| Parses a top-level declaration
164decl :: SyntaxInfo -> IdrisParser [PDecl]
165decl syn = do fc <- getFC
166 -- if we're after maxline, stop here
167 let continue = case maxline syn of
169 Just l -> if fst (fc_end fc) > l
170 then mut_nesting syn /= 0
172 if continue then do notEndBlock
174 else fail "End of readable input"
175 where declBody :: IdrisParser [PDecl]
183 <|> do d <- dsl syn; return [d]
187 <|> do import_; fail "imports must be at top of file"
189 declBody' :: IdrisParser [PDecl]
190 declBody' = do d <- decl' syn
192 let d' = fmap (debindApp syn . (desugar syn i)) d
195{- | Parses a top-level declaration with possible syntax sugar
207decl' :: SyntaxInfo -> IdrisParser PDecl
215{- | Parses a syntax extension declaration (and adds the rule to parser state)
218 SyntaxDecl ::= SyntaxRule;
221syntaxDecl :: SyntaxInfo -> IdrisParser PDecl
222syntaxDecl syn = do s <- syntaxRule syn
224 put (i `addSyntax` s)
226 return (PSyntax fc s)
228-- | Extend an 'IState' with a new syntax extension. See also 'addReplSyntax'.
229addSyntax :: IState -> Syntax -> IState
230addSyntax i s = i { syntax_rules = updateSyntaxRules [s] rs,
231 syntax_keywords = ks ++ ns,
232 ibc_write = IBCSyntax s : map IBCKeyword ks ++ ibc }
233 where rs = syntax_rules i
234 ns = syntax_keywords i
236 ks = map show (syntaxNames s)
238-- | Like 'addSyntax', but no effect on the IBC.
239addReplSyntax :: IState -> Syntax -> IState
240addReplSyntax i s = i { syntax_rules = updateSyntaxRules [s] rs,
241 syntax_keywords = ks ++ ns }
242 where rs = syntax_rules i
243 ns = syntax_keywords i
244 ks = map show (syntaxNames s)
246{- | Parses a syntax extension declaration
249SyntaxRuleOpts ::= 'term' | 'pattern';
254 SyntaxRuleOpts? 'syntax' SyntaxSym+ '=' TypeExpr Terminator;
258SyntaxSym ::= '[' Name_t ']'
265syntaxRule :: SyntaxInfo -> IdrisParser Syntax
269 sty <- option AnySyntax (do reserved "term"; return TermSyntax
270 <|> do reserved "pattern"; return PatternSyntax)
273 syms <- some syntaxSym
274 when (all isExpr syms) $ unexpected "missing keywords in syntax rule"
275 let ns = mapMaybe getName syms
276 when (length ns /= length (nub ns))
277 $ unexpected "repeated variable in syntax rule"
279 tm <- typeExpr (allowImp syn)
281 return (Rule (mkSimple syms) tm sty)
283 isExpr (Expr _) = True
285 getName (Expr n) = Just n
287 -- Can't parse two full expressions (i.e. expressions with application) in a row
288 -- so change them both to a simple expression
289 mkSimple (Expr e : es) = SimpleExpr e : mkSimple' es
290 mkSimple xs = mkSimple' xs
292 mkSimple' (Expr e : Expr e1 : es) = SimpleExpr e : SimpleExpr e1 :
294 -- Can't parse a full expression followed by operator like characters due to ambiguity
295 mkSimple' (Expr e : Symbol s : es)
296 | takeWhile (`elem` opChars) ts /= "" && ts `notElem` invalidOperators = SimpleExpr e : Symbol s : mkSimple' es
297 where ts = dropWhile isSpace . dropWhileEnd isSpace $ s
298 mkSimple' (e : es) = e : mkSimple' es
302{- | Parses a syntax symbol (either binding variable, keyword or expression)
305SyntaxSym ::= '[' Name_t ']'
312syntaxSym :: IdrisParser SSymbol
313syntaxSym = try (do lchar '['; n <- name; lchar ']'
315 <|> try (do lchar '{'; n <- name; lchar '}'
319 <|> do sym <- stringLiteral
323{- | Parses a function declaration with possible syntax sugar
326 FunDecl ::= FunDecl';
329fnDecl :: SyntaxInfo -> IdrisParser [PDecl]
330fnDecl syn = try (do notEndBlock
333 let d' = fmap (desugar syn i) d
334 return [d']) <?> "function declaration"
336{-| Parses a function declaration
340 DocComment_t? FnOpts* Accessibility? FnOpts* FnName TypeSig Terminator
347fnDecl' :: SyntaxInfo -> IdrisParser PDecl
348fnDecl' syn = checkFixity $
349 do (doc, argDocs, fc, opts', n, acc) <- try (do
351 (doc, argDocs) <- docstring syn
353 let initOpts = if default_total ist
356 opts <- fnOpts initOpts
357 acc <- optional accessibility
360 let n = expandNS syn n_in
363 return (doc, argDocs, fc, opts', n, acc))
364 ty <- typeExpr (allowImp syn)
367 return (PTy doc argDocs syn fc opts' n ty)
371 <?> "function declaration"
372 where checkFixity :: IdrisParser PDecl -> IdrisParser PDecl
373 checkFixity p = do decl <- p
375 Nothing -> return decl
376 Just n -> do fOk <- fixityOK n
378 "Missing fixity declaration for " ++ show n
380 getName (PTy _ _ _ _ _ n _) = Just n
382 fixityOK (NS n _) = fixityOK n
383 fixityOK (UN n) | all (flip elem opChars) (str n) =
384 do fixities <- fmap idris_infixes get
385 return . elem (str n) . map (\ (Fix _ op) -> op) $ fixities
386 | otherwise = return True
387 fixityOK _ = return True
389{-| Parses function options given initial options
397 | '%' 'error_handler'
399 | '%' 'specialise' '[' NameTimesList? ']'
404NameTimes ::= FnName Natural?;
410 | NameTimes ',' NameTimesList
414-- FIXME: Check compatability for function options (i.e. partal/total)
416-- Issue #1574 on the issue tracker.
417-- https://github.com/idris-lang/Idris-dev/issues/1574
418fnOpts :: [FnOpt] -> IdrisParser [FnOpt]
420 = do reserved "total"; fnOpts (TotalFn : opts)
421 <|> do reserved "partial"; fnOpts (PartialFn : (opts \\ [TotalFn]))
422 <|> do reserved "covering"; fnOpts (CoveringFn : (opts \\ [TotalFn]))
423 <|> do try (lchar '%' *> reserved "export"); c <- stringLiteral;
424 fnOpts (CExport c : opts)
425 <|> do try (lchar '%' *> reserved "no_implicit");
426 fnOpts (NoImplicit : opts)
427 <|> do try (lchar '%' *> reserved "inline");
428 fnOpts (Inlinable : opts)
429 <|> do try (lchar '%' *> reserved "assert_total");
430 fnOpts (AssertTotal : opts)
431 <|> do try (lchar '%' *> reserved "error_handler");
432 fnOpts (ErrorHandler : opts)
433 <|> do try (lchar '%' *> reserved "error_reverse");
434 fnOpts (ErrorReverse : opts)
435 <|> do try (lchar '%' *> reserved "reflection");
436 fnOpts (Reflection : opts)
437 <|> do lchar '%'; reserved "specialise";
438 lchar '['; ns <- sepBy nameTimes (lchar ','); lchar ']'
439 fnOpts (Specialise ns : opts)
440 <|> do reserved "implicit"; fnOpts (Implicit : opts)
442 <?> "function modifier"
443 where nameTimes :: IdrisParser (Name, Maybe Int)
444 nameTimes = do n <- fnName
445 t <- option Nothing (do reds <- natural
446 return (Just (fromInteger reds)))
449{- | Parses a postulate
453 DocComment_t? 'postulate' FnOpts* Accesibility? FnOpts* FnName TypeSig Terminator
457postulate :: SyntaxInfo -> IdrisParser PDecl
458postulate syn = do doc <- try $ do (doc, _) <- docstring syn
463 let initOpts = if default_total ist
466 opts <- fnOpts initOpts
467 acc <- optional accessibility
470 let n = expandNS syn n_in
472 ty <- typeExpr (allowImp syn)
476 return (PPostulate doc syn fc opts' n ty)
479{- | Parses a using declaration
483 'using' '(' UsingDeclList ')' OpenBlock Decl* CloseBlock
487using_ :: SyntaxInfo -> IdrisParser [PDecl]
489 do reserved "using"; lchar '('; ns <- usingDeclList syn; lchar ')'
491 let uvars = using syn
492 ds <- many (decl (syn { using = uvars ++ ns }))
495 <?> "using declaration"
497{- | Parses a parameters declaration
501 'parameters' '(' TypeDeclList ')' OpenBlock Decl* CloseBlock
505params :: SyntaxInfo -> IdrisParser [PDecl]
507 do reserved "parameters"; lchar '('; ns <- typeDeclList syn; lchar ')'
509 let pvars = syn_params syn
510 ds <- many (decl syn { syn_params = pvars ++ ns })
513 return [PParams fc ns (concat ds)]
514 <?> "parameters declaration"
516{- | Parses a mutual declaration (for mutually recursive functions)
520 'mutual' OpenBlock Decl* CloseBlock
524mutual :: SyntaxInfo -> IdrisParser [PDecl]
528 let pvars = syn_params syn
529 ds <- many (decl (syn { mut_nesting = mut_nesting syn + 1 } ))
532 return [PMutual fc (concat ds)]
535{-| Parses a namespace declaration
539 'namespace' identifier OpenBlock Decl+ CloseBlock
543namespace :: SyntaxInfo -> IdrisParser [PDecl]
545 do reserved "namespace"; n <- identifier;
547 ds <- some (decl syn { syn_namespace = n : syn_namespace syn })
549 return [PNamespace n (concat ds)]
550 <?> "namespace declaration"
552{- | Parses a methods block (for instances)
555 InstanceBlock ::= 'where' OpenBlock FnDecl* CloseBlock
558instanceBlock :: SyntaxInfo -> IdrisParser [PDecl]
559instanceBlock syn = do reserved "where"
561 ds <- many (fnDecl syn)
566{- | Parses a methods and instances block (for type classes)
577 'where' OpenBlock MethodOrInstance* CloseBlock
581classBlock :: SyntaxInfo -> IdrisParser [PDecl]
582classBlock syn = do reserved "where"
584 ds <- many (notEndBlock >> instance_ syn <|> fnDecl syn)
589{-| Parses a type class declaration
594 | '(' Name ':' Expr ')'
600 DocComment_t? Accessibility? 'class' ConstraintList? Name ClassArgument* ClassBlock?
604class_ :: SyntaxInfo -> IdrisParser [PDecl]
605class_ syn = do (doc, argDocs, acc)
606 <- try (do (doc, argDocs) <- docstring syn
607 acc <- optional accessibility
609 return (doc, argDocs, acc))
611 cons <- constraintList syn
613 let n = expandNS syn n_in
615 fds <- option (map fst cs) fundeps
616 ds <- option [] (classBlock syn)
617 accData acc n (concatMap declared ds)
618 return [PClass doc syn fc cons n cs argDocs fds ds]
619 <?> "type-class declaration"
621 fundeps :: IdrisParser [Name]
622 fundeps = do lchar '|'; sepBy name (lchar ',')
624 carg :: IdrisParser (Name, PTerm)
625 carg = do lchar '('; i <- name; lchar ':'; ty <- expr syn; lchar ')'
630{- | Parses a type class instance declaration
634 DocComment_t? 'instance' InstanceName? ConstraintList? Name SimpleExpr* InstanceBlock?
639InstanceName ::= '[' Name ']';
642instance_ :: SyntaxInfo -> IdrisParser [PDecl]
643instance_ syn = do (doc, argDocs)
644 <- try (docstring syn <* reserved "instance")
646 en <- optional instanceName
647 cs <- constraintList syn
649 args <- many (simpleExpr syn)
650 let sc = PApp fc (PRef fc cn) (map pexp args)
651 let t = bindList (PPi constraint) cs sc
652 ds <- option [] (instanceBlock syn)
653 return [PInstance doc argDocs syn fc cs cn args t en ds]
654 <?> "instance declaration"
655 where instanceName :: IdrisParser Name
656 instanceName = do lchar '['; n_in <- fnName; lchar ']'
657 let n = expandNS syn n_in
662-- | Parse a docstring
663docstring :: SyntaxInfo
664 -> IdrisParser (Docstring (Either Err PTerm),
665 [(Name,Docstring (Either Err PTerm))])
666docstring syn = do (doc, argDocs) <- option noDocs docComment
668 let doc' = annotCode (tryFullExpr syn ist) doc
669 argDocs' = [ (n, annotCode (tryFullExpr syn ist) d)
670 | (n, d) <- argDocs ]
671 return (doc', argDocs')
674{- | Parses a using declaration list
686 | UsingDecl ',' UsingDeclList'
697usingDeclList :: SyntaxInfo -> IdrisParser [Using]
699 = try (sepBy1 (usingDecl syn) (lchar ','))
700 <|> do ns <- sepBy1 name (lchar ',')
702 t <- typeExpr (disallowImp syn)
703 return (map (\x -> UImplicit x t) ns)
704 <?> "using declaration list"
706{- |Parses a using declaration
715usingDecl :: SyntaxInfo -> IdrisParser Using
716usingDecl syn = try (do x <- fnName
718 t <- typeExpr (disallowImp syn)
719 return (UImplicit x t))
722 return (UConstraint c xs)
723 <?> "using declaration"
725{- | Parse a clause with patterns
731pattern :: SyntaxInfo -> IdrisParser PDecl
732pattern syn = do fc <- getFC
734 return (PClauses fc [] (sMN 2 "_") [clause]) -- collect together later
737{- | Parse a constant applicative form declaration
740CAF ::= 'let' FnName '=' Expr Terminator;
743caf :: SyntaxInfo -> IdrisParser PDecl
744caf syn = do reserved "let"
745 n_in <- fnName; let n = expandNS syn n_in
751 <?> "constant applicative form declaration"
753{- | Parse an argument expression
756ArgExpr ::= HSimpleExpr | {- In Pattern External (User-defined) Expression -};
759argExpr :: SyntaxInfo -> IdrisParser PTerm
760argExpr syn = let syn' = syn { inPattern = True } in
761 try (hsimpleExpr syn') <|> simpleExternalExpr syn'
762 <?> "argument expression"
764{- | Parse a right hand side of a function
774RHSName ::= '{' FnName '}';
777rhs :: SyntaxInfo -> Name -> IdrisParser PTerm
778rhs syn n = do lchar '='; expr syn
781 name <- option n' (do symbol "{"; n <- fnName; symbol "}";
784 return (addLet fc name r)
785 <|> do reserved "impossible"; return PImpossible
786 <?> "function right hand side"
787 where mkN :: Name -> Name
788 mkN (UN x) = if (tnull x || not (isAlpha (thead x)))
789 then sUN "infix_op_lemma_1"
790 else sUN (str x++"_lemma_1")
791 mkN (NS x n) = NS (mkN x) n
794 addLet :: FC -> Name -> PTerm -> PTerm
795 addLet fc nm (PLet fc' n ty val r) = PLet fc' n ty val (addLet fc nm r)
796 addLet fc nm (PCase fc' t cs) = PCase fc' t (map addLetC cs)
797 where addLetC (l, r) = (l, addLet fc nm r)
798 addLet fc nm r = (PLet fc (sUN "value") Placeholder r (PMetavar nm))
800{- |Parses a function clause
803RHSOrWithBlock ::= RHS WhereOrTerminator
804 | 'with' SimpleExpr OpenBlock FnDecl+ CloseBlock
809Clause ::= WExpr+ RHSOrWithBlock
810 | SimpleExpr '<==' FnName RHS WhereOrTerminator
811 | ArgExpr Operator ArgExpr WExpr* RHSOrWithBlock {- Except "=" and "?=" operators to avoid ambiguity -}
812 | FnName ConstraintArg* ImplicitOrArgExpr* WExpr* RHSOrWithBlock
817ImplicitOrArgExpr ::= ImplicitArg | ArgExpr;
821WhereOrTerminator ::= WhereBlock | Terminator;
824clause :: SyntaxInfo -> IdrisParser PClause
826 = do wargs <- try (do pushIndent; some (wExpr syn))
829 n <- case lastParse ist of
831 Nothing -> fail "Invalid clause"
833 let ctxt = tt_ctxt ist
834 let wsyn = syn { syn_namespace = [] }
835 (wheres, nmap) <- choice [do x <- whereBlock n wsyn
840 return $ PClauseR fc wargs r wheres) <|> (do
843 wval <- simpleExpr syn
846 ds <- some $ fnDecl syn
847 let withs = concat ds
849 return $ PWithR fc wargs wval pn withs)
850 <|> do ty <- try (do pushIndent
855 n_in <- fnName; let n = expandNS syn n_in
858 let ctxt = tt_ctxt ist
859 let wsyn = syn { syn_namespace = [] }
860 (wheres, nmap) <- choice [do x <- whereBlock n wsyn
865 let capp = PLet fc (sMN 0 "match")
868 (PRef fc (sMN 0 "match"))
870 put (ist { lastParse = Just n })
871 return $ PClause fc n capp [] r wheres
872 <|> do (l, op) <- try (do
876 when (op == "=" || op == "?=" ) $
877 fail "infix clause definition with \"=\" and \"?=\" not supported "
879 let n = expandNS syn (sUN op)
882 wargs <- many (wExpr syn)
884 let wsyn = syn { syn_namespace = [] }
885 (wheres, nmap) <- choice [do x <- whereBlock n wsyn
891 let capp = PApp fc (PRef fc n) [pexp l, pexp r]
892 put (ist { lastParse = Just n })
893 return $ PClause fc n capp wargs rs wheres) <|> (do
896 wval <- bracketed syn
899 ds <- some $ fnDecl syn
902 let capp = PApp fc (PRef fc n) [pexp l, pexp r]
903 let withs = map (fillLHSD n capp wargs) $ concat ds
904 put (ist { lastParse = Just n })
905 return $ PWith fc n capp wargs wval pn withs)
907 n_in <- fnName; let n = expandNS syn n_in
908 cargs <- many (constraintArg syn)
910 args <- many (try (implicitArg (syn { inPattern = True } ))
911 <|> (fmap pexp (argExpr syn)))
912 wargs <- many (wExpr syn)
913 let capp = PApp fc (PRef fc n)
917 let ctxt = tt_ctxt ist
918 let wsyn = syn { syn_namespace = [] }
919 (wheres, nmap) <- choice [do x <- whereBlock n wsyn
925 put (ist { lastParse = Just n })
926 return $ PClause fc n capp wargs r wheres) <|> (do
929 put (ist { lastParse = Just n })
930 wval <- bracketed syn
933 ds <- some $ fnDecl syn
934 let withs = map (fillLHSD n capp wargs) $ concat ds
937 return $ PWith fc n capp wargs wval pn withs)
938 <?> "function clause"
940 optProof = option Nothing (do reserved "proof"
944 fillLHS :: Name -> PTerm -> [PTerm] -> PClause -> PClause
945 fillLHS n capp owargs (PClauseR fc wargs v ws)
946 = PClause fc n capp (owargs ++ wargs) v ws
947 fillLHS n capp owargs (PWithR fc wargs v pn ws)
948 = PWith fc n capp (owargs ++ wargs) v pn
949 (map (fillLHSD n capp (owargs ++ wargs)) ws)
952 fillLHSD :: Name -> PTerm -> [PTerm] -> PDecl -> PDecl
953 fillLHSD n c a (PClauses fc o fn cs) = PClauses fc o fn (map (fillLHS n c a) cs)
956{-| Parses with pattern
962wExpr :: SyntaxInfo -> IdrisParser PTerm
963wExpr syn = do lchar '|'
964 expr' (syn { inPattern = True })
967{- | Parses a where block
970WhereBlock ::= 'where' OpenBlock Decl+ CloseBlock;
973whereBlock :: Name -> SyntaxInfo -> IdrisParser ([PDecl], [(Name, Name)])
975 = do reserved "where"
976 ds <- indentedBlock1 (decl syn)
977 let dns = concatMap (concatMap declared) ds
978 return (concat ds, map (\x -> (x, decoration syn x)) dns)
981{- |Parses a code generation target language name
993codegen_ :: IdrisParser Codegen
994codegen_ = do n <- identifier; return (Via (map toLower n))
995 <|> do reserved "Bytecode"; return Bytecode
996 <?> "code generation language"
998{- |Parses a compiler directive
1002 | String ',' StringList
1007Directive ::= '%' Directive';
1011Directive' ::= 'lib' CodeGen String_t
1012 | 'link' CodeGen String_t
1013 | 'flag' CodeGen String_t
1014 | 'include' CodeGen String_t
1017 | 'access' Accessibility
1018 | 'default' Totality
1020 | 'dynamic' StringList
1021 | 'name' Name NameList
1022 | 'error_handlers' Name NameList
1023 | 'language' 'TypeProviders'
1024 | 'language' 'ErrorReflection'
1028directive :: SyntaxInfo -> IdrisParser [PDecl]
1029directive syn = do try (lchar '%' *> reserved "lib"); cgn <- codegen_; lib <- stringLiteral;
1030 return [PDirective (do addLib cgn lib
1031 addIBC (IBCLib cgn lib))]
1032 <|> do try (lchar '%' *> reserved "link"); cgn <- codegen_; obj <- stringLiteral;
1033 return [PDirective (do dirs <- allImportDirs
1034 o <- liftIO $ findInPath dirs obj
1035 addIBC (IBCObj cgn obj) -- just name, search on loading ibc
1036 addObjectFile cgn o)]
1037 <|> do try (lchar '%' *> reserved "flag"); cgn <- codegen_;
1038 flag <- stringLiteral
1039 return [PDirective (do addIBC (IBCCGFlag cgn flag)
1041 <|> do try (lchar '%' *> reserved "include"); cgn <- codegen_; hdr <- stringLiteral;
1042 return [PDirective (do addHdr cgn hdr
1043 addIBC (IBCHeader cgn hdr))]
1044 <|> do try (lchar '%' *> reserved "hide"); n <- fnName
1045 return [PDirective (do setAccessibility n Hidden
1046 addIBC (IBCAccess n Hidden))]
1047 <|> do try (lchar '%' *> reserved "freeze"); n <- iName []
1048 return [PDirective (do setAccessibility n Frozen
1049 addIBC (IBCAccess n Frozen))]
1050 <|> do try (lchar '%' *> reserved "access"); acc <- accessibility
1051 return [PDirective (do i <- get
1052 put(i { default_access = acc }))]
1053 <|> do try (lchar '%' *> reserved "default"); tot <- totality
1055 put (i { default_total = tot } )
1056 return [PDirective (do i <- get
1057 put(i { default_total = tot }))]
1058 <|> do try (lchar '%' *> reserved "logging"); i <- natural;
1059 return [PDirective (setLogLevel (fromInteger i))]
1060 <|> do try (lchar '%' *> reserved "dynamic"); libs <- sepBy1 stringLiteral (lchar ',');
1061 return [PDirective (do added <- addDyLib libs
1063 Left lib -> addIBC (IBCDyLib (lib_name lib))
1066 <|> do try (lchar '%' *> reserved "name")
1068 ns <- sepBy1 name (lchar ',')
1069 return [PDirective (do ty' <- disambiguate ty
1070 mapM_ (addNameHint ty') ns
1071 mapM_ (\n -> addIBC (IBCNameHint (ty', n))) ns)]
1072 <|> do try (lchar '%' *> reserved "error_handlers")
1075 ns <- sepBy1 name (lchar ',')
1076 return [PDirective (do fn' <- disambiguate fn
1077 ns' <- mapM disambiguate ns
1078 addFunctionErrorHandlers fn' arg ns'
1079 mapM_ (addIBC . IBCFunctionErrorHandler fn' arg) ns')]
1080 <|> do try (lchar '%' *> reserved "language"); ext <- pLangExt;
1081 return [PDirective (addLangExt ext)]
1083 try (lchar '%' *> reserved "used")
1086 return [PDirective (addUsedName fc fn arg)]
1089 where disambiguate :: Name -> Idris Name
1090 disambiguate n = do i <- getIState
1091 case lookupCtxtName n (idris_implicits i) of
1092 [(n', _)] -> return n'
1093 [] -> throwError (NoSuchVariable n)
1094 more -> throwError (CantResolveAlts (map fst more))
1096pLangExt :: IdrisParser LanguageExt
1097pLangExt = (reserved "TypeProviders" >> return TypeProviders)
1098 <|> (reserved "ErrorReflection" >> return ErrorReflection)
1100{- | Parses a totality
1103Totality ::= 'partial' | 'total'
1107totality :: IdrisParser Bool
1109 = do reserved "total"; return True
1110 <|> do reserved "partial"; return False
1112{- | Parses a type provider
1115Provider ::= DocComment_t? '%' 'provide' Provider_What? '(' FnName TypeSig ')' 'with' Expr;
1116ProviderWhat ::= 'proof' | 'term' | 'type' | 'postulate'
1119provider :: SyntaxInfo -> IdrisParser [PDecl]
1120provider syn = do doc <- try (do (doc, _) <- docstring syn
1124 provideTerm doc <|> providePostulate doc
1126 where provideTerm doc =
1127 do lchar '('; n <- fnName; lchar ':'; t <- typeExpr syn; lchar ')'
1130 e <- expr syn <?> "provider expression"
1131 return [PProvider doc syn fc (ProvTerm t e) n]
1132 providePostulate doc =
1133 do reserved "postulate"
1137 e <- expr syn <?> "provider expression"
1138 return [PProvider doc syn fc (ProvPostulate e) n]
1140{- | Parses a transform
1143Transform ::= '%' 'transform' Expr '==>' Expr
1146transform :: SyntaxInfo -> IdrisParser [PDecl]
1147transform syn = do try (lchar '%' *> reserved "transform")
1148 -- leave it unchecked, until we work out what this should
1150-- safety <- option True (do reserved "unsafe"
1156 return [PTransform fc False l r]
1160{- * Loading and parsing -}
1161{- | Parses an expression from input -}
1162parseExpr :: IState -> String -> Result PTerm
1163parseExpr st = runparser (fullExpr defaultSyntax) st "(input)"
1165{- | Parses a constant form input -}
1166parseConst :: IState -> String -> Result Const
1167parseConst st = runparser constant st "(input)"
1169{- | Parses a tactic from input -}
1170parseTactic :: IState -> String -> Result PTactic
1171parseTactic st = runparser (fullTactic defaultSyntax) st "(input)"
1173-- | Parse module header and imports
1174parseImports :: FilePath -> String -> Idris (Maybe (Docstring ()), [String], [(Bool, String, Maybe String, FC)], Maybe Delta)
1175parseImports fname input
1177 case parseString (runInnerParser (evalStateT imports i)) (Directed (UTF8.fromString fname) 0 0 0 0) input of
1178 Failure err -> fail (show err)
1179 Success (x, i) -> do putIState i
1181 where imports :: IdrisParser ((Maybe (Docstring ()), [String], [(Bool, String, Maybe String, FC)], Maybe Delta), IState)
1182 imports = do whiteSpace
1183 (mdoc, mname) <- moduleHeader
1186 isEof <- lookAheadMatches eof
1191 return ((mdoc, mname, ps, mrk'), i)
1193-- | There should be a better way of doing this...
1194findFC :: Doc -> (FC, String)
1195findFC x = let s = show (plain x) in
1196 case span (/= ':') s of
1197 (failname, ':':rest) -> case span isDigit rest of
1198 (line, ':':rest') -> case span isDigit rest' of
1199 (col, ':':msg) -> let pos = (read line, read col) in
1200 (FC failname pos pos, msg)
1202-- | Check if the coloring matches the options and corrects if necessary
1203fixColour :: Bool -> ANSI.Doc -> ANSI.Doc
1204fixColour False doc = ANSI.plain doc
1205fixColour True doc = doc
1207-- | A program is a list of declarations, possibly with associated
1208-- documentation strings.
1209parseProg :: SyntaxInfo -> FilePath -> String -> Maybe Delta ->
1211parseProg syn fname input mrk
1213 case runparser mainProg i fname input of
1214 Failure doc -> do -- FIXME: Get error location from trifecta
1215 -- this can't be the solution!
1216 -- Issue #1575 on the issue tracker.
1217 -- https://github.com/idris-lang/Idris-dev/issues/1575
1218 let (fc, msg) = findFC doc
1220 case idris_outputmode i of
1221 RawOutput h -> iputStrLn (show $ fixColour (idris_colourRepl i) doc)
1222 IdeMode n h -> iWarn fc (P.text msg)
1223 putIState (i { errSpan = Just fc })
1225 Success (x, i) -> do putIState i
1226 reportParserWarnings
1228 where mainProg :: IdrisParser ([PDecl], IState)
1229 mainProg = case mrk of
1230 Nothing -> do i <- get; return ([], i)
1237{- | Load idris module and show error if something wrong happens -}
1238loadModule :: FilePath -> Idris (Maybe String)
1240 = idrisCatch (loadModule' f)
1241 (\e -> do setErrSpan (getErrSpan e)
1243 iWarn (getErrSpan e) $ pprintErr ist e
1246{- | Load idris module -}
1247loadModule' :: FilePath -> Idris (Maybe String)
1250 let file = takeWhile (/= ' ') f
1251 ibcsd <- valIBCSubDir i
1252 ids <- allImportDirs
1253 fp <- findImport ids ibcsd file
1254 if file `elem` imported i
1255 then do iLOG $ "Already read " ++ file
1257 else do putIState (i { imported = file : imported i })
1259 IDR fn -> loadSource False fn Nothing
1260 LIDR fn -> loadSource True fn Nothing
1262 idrisCatch (loadIBC True fn)
1263 (\c -> do iLOG $ fn ++ " failed " ++ pshow i c
1265 IDR sfn -> loadSource False sfn Nothing
1266 LIDR sfn -> loadSource True sfn Nothing)
1269{- | Load idris code from file -}
1270loadFromIFile :: Bool -> IFileType -> Maybe Int -> Idris ()
1271loadFromIFile reexp i@(IBC fn src) maxline
1272 = do iLOG $ "Skipping " ++ getSrcFile i
1273 idrisCatch (loadIBC reexp fn)
1274 (\err -> ierror $ LoadingFailed fn err)
1276 getSrcFile (IDR fn) = fn
1277 getSrcFile (LIDR fn) = fn
1278 getSrcFile (IBC f src) = getSrcFile src
1280loadFromIFile _ (IDR fn) maxline = loadSource' False fn maxline
1281loadFromIFile _ (LIDR fn) maxline = loadSource' True fn maxline
1283{-| Load idris source code and show error if something wrong happens -}
1284loadSource' :: Bool -> FilePath -> Maybe Int -> Idris ()
1285loadSource' lidr r maxline
1286 = idrisCatch (loadSource lidr r maxline)
1287 (\e -> do setErrSpan (getErrSpan e)
1290 At f e' -> iWarn f (pprintErr ist e')
1291 _ -> iWarn (getErrSpan e) (pprintErr ist e))
1293{- | Load Idris source code-}
1294loadSource :: Bool -> FilePath -> Maybe Int -> Idris ()
1295loadSource lidr f toline
1296 = do iLOG ("Reading " ++ f)
1298 let def_total = default_total i
1299 file_in <- runIO $ readFile f
1300 file <- if lidr then tclift $ unlit f file_in else return file_in
1301 (mdocs, mname, imports_in, pos) <- parseImports f file
1302 ai <- getAutoImports
1303 let imports = map (\n -> (True, n, Just n, emptyFC)) ai ++ imports_in
1304 ids <- allImportDirs
1305 ibcsd <- valIBCSubDir i
1307 do fp <- findImport ids ibcsd f
1309 LIDR fn -> ifail $ "No ibc for " ++ f
1310 IDR fn -> ifail $ "No ibc for " ++ f
1311 IBC fn src -> loadIBC True fn)
1312 [(re, fn) | (re, fn, _, _) <- imports]
1313 reportParserWarnings
1315 -- process and check module aliases
1316 let modAliases = M.fromList
1317 [(prep alias, prep realName) | (reexport, realName, Just alias, fc) <- imports]
1318 prep = map T.pack . reverse . Spl.splitOn "/"
1319 aliasNames = [(alias, fc) | (_, _, Just alias, fc) <- imports]
1320 histogram = groupBy ((==) `on` fst) . sortBy (comparing fst) $ aliasNames
1321 case map head . filter ((/= 1) . length) $ histogram of
1322 [] -> logLvl 3 $ "Module aliases: " ++ show (M.toList modAliases)
1323 (n,fc):_ -> throwError . At fc . Msg $ "import alias not unique: " ++ show n
1326 putIState (i { default_access = Hidden, module_aliases = modAliases })
1327 clearIBC -- start a new .ibc file
1328 -- record package info in .ibc
1329 imps <- allImportDirs
1330 mapM_ addIBC (map IBCImportDir imps)
1331 mapM_ (addIBC . IBCImport) [(reexport, realName) | (reexport, realName, alias, fc) <- imports]
1332 let syntax = defaultSyntax{ syn_namespace = reverse mname,
1335 -- Save the span from parsing the module header, because
1336 -- an empty program parse might obliterate it.
1337 let oldSpan = idris_parsedSpan ist
1338 ds' <- parseProg syntax f file pos
1340 case (ds', oldSpan) of
1342 -- If no program elements were parsed, we dind't
1343 -- get a loaded region in the IBC file. That
1344 -- means we need to add it back.
1346 putIState ist { idris_parsedSpan = oldSpan
1347 , ibc_write = IBCParsedRegion fc :
1352 -- Parsing done, now process declarations
1354 let ds = namespaces mname ds'
1355 logLvl 3 (show $ showDecls verbosePPOption ds)
1357 logLvl 10 (show (toAlist (idris_implicits i)))
1358 logLvl 3 (show (idris_infixes i))
1359 -- Now add all the declarations to the context
1361 when v $ iputStrLn $ "Type checking " ++ f
1362 -- we totality check after every Mutual block, so if
1363 -- anything is a single definition, wrap it in a
1364 -- mutual block on its own
1365 elabDecls toplevel (map toMutual ds)
1367 -- simplify every definition do give the totality checker
1369 mapM_ (\n -> do logLvl 5 $ "Simplifying " ++ show n
1370 updateContext (simplifyCasedef n $ getErasureInfo i))
1371 (map snd (idris_totcheck i))
1372 -- build size change graph from simplified definitions
1373 iLOG "Totality checking"
1375 mapM_ buildSCG (idris_totcheck i)
1376 mapM_ checkDeclTotality (idris_totcheck i)
1378 -- Redo totality check for deferred names
1379 let deftots = idris_defertotcheck i
1380 iLOG $ "Totality checking " ++ show deftots
1381 mapM_ (\x -> do tot <- getTotality x
1383 Total _ -> setTotality x Unchecked
1384 _ -> return ()) (map snd deftots)
1385 mapM_ buildSCG deftots
1386 mapM_ checkDeclTotality deftots
1388 iLOG ("Finished " ++ f)
1389 ibcsd <- valIBCSubDir i
1390 iLOG "Universe checking"
1392 let ibc = ibcPathNoFallback ibcsd f
1394 addHides (hide_list i)
1396 -- Save module documentation if applicable
1399 Nothing -> return ()
1400 Just docs -> addModDoc syntax mname docs
1403 -- Finally, write an ibc if checking was successful
1406 idrisCatch (do writeIBC f ibc; clearIBC)
1407 (\c -> return ()) -- failure is harmless
1409 putIState (i { default_total = def_total,
1413 namespaces :: [String] -> [PDecl] -> [PDecl]
1414 namespaces [] ds = ds
1415 namespaces (x:xs) ds = [PNamespace x (namespaces xs ds)]
1417 toMutual :: PDecl -> PDecl
1418 toMutual m@(PMutual _ d) = m
1419 toMutual (PNamespace x ds) = PNamespace x (map toMutual ds)
1420 toMutual x = let r = PMutual (fileFC "single mutual") [x] in
1427 addModDoc :: SyntaxInfo -> [String] -> Docstring () -> Idris ()
1428 addModDoc syn mname docs =
1430 docs' <- elabDocTerms recinfo (parsedDocs ist)
1431 let modDocs' = addDef docName docs' (idris_moduledocs ist)
1432 putIState ist { idris_moduledocs = modDocs' }
1433 addIBC (IBCModDocs docName)
1435 docName = NS modDocName (map T.pack (reverse mname))
1436 parsedDocs ist = annotCode (tryFullExpr syn ist) docs
1438{- | Adds names to hide list -}
1439addHides :: [(Name, Maybe Accessibility)] -> Idris ()
1440addHides xs = do i <- getIState
1441 let defh = default_access i
1442 let (hs, as) = partition isNothing xs
1445 (map (\ (n, _) -> (n, defh)) hs ++
1446 map (\ (n, Just a) -> (n, a)) as)
1447 where isNothing (_, Nothing) = True
1450 doHide (n, a) = do setAccessibility n a
1451 addIBC (IBCAccess n a)
keysesc to close
- j k
- move down, up
- enter
- open
- esc
- back, or up a level
- 1 2 3 4
- readme, files, commits, refs
- /
- filter
- y
- copy the clone command
- c
- compare the chosen commit with another (or shift-click both)
- ?
- show or hide this list