git.iko.soy/rowtype-yoga/purescript-native
on d3f368b help sign in
rowtype-yoga/purescript-native/src/Language/PureScript/Scope.hs
63 lines 1.9 kB history raw
1-----------------------------------------------------------------------------
2--
3-- Module : Language.PureScript.Scope
4-- Copyright : (c) Phil Freeman 2013
5-- License : MIT
6--
7-- Maintainer : Phil Freeman <paf31@cantab.net>
8-- Stability : experimental
9-- Portability :
10--
11-- |
12-- Utility functions for working with names in scope
13--
14-----------------------------------------------------------------------------
16module Language.PureScript.Scope (
17 usedNames,
18 unusedNames
19) where
21import Data.Data
22import Data.List ((\\), nub)
23import Data.Generics (extQ, mkQ, everything)
25import Language.PureScript.Values
26import Language.PureScript.Names
27import Language.PureScript.CodeGen.JS.AST
29-- |
30-- Gather all used names appearing inside a value
31--
32usedNames :: (Data d) => d -> [Ident]
33usedNames val = nub $ everything (++) (mkQ [] namesV `extQ` namesS `extQ` namesB `extQ` namesJS) val
34 where
35 namesV :: Value -> [Ident]
36 namesV (Abs arg _) = [arg]
37 namesV (Var (Qualified Nothing name)) = [name]
38 namesV _ = []
39 namesS :: Statement -> [Ident]
40 namesS (VariableIntroduction name _) = [name]
41 namesS (For name _ _ _) = [name]
42 namesS _ = []
43 namesB :: Binder -> [Ident]
44 namesB (VarBinder name) = [name]
45 namesB _ = []
46 namesJS :: JS -> [Ident]
47 namesJS (JSVar name) = [Ident name]
48 namesJS (JSFunction (Just name) args _) = (Ident name) : (Ident `map` args)
49 namesJS (JSFunction Nothing args _) = (Ident `map` args)
50 namesJS (JSVariableIntroduction name _) = [Ident name]
51 namesJS (JSFor name _ _ _) = [Ident name]
52 namesJS _ = []
54-- |
55-- Generate a set of names which are unused inside a value, of the form @_{n}@ for an integer @n@
56--
57unusedNames :: (Data d) => d -> [Ident]
58unusedNames val =
59 let
60 allNames = usedNames val
61 varNames = map (Ident . ('_' :) . show) ([1..] :: [Int])
62 in
63 varNames \\ allNames