git.iko.soy/rowtype-yoga/purescript-native
on 4efe14b help sign in
rowtype-yoga/purescript-native/src/Language/PureScript/Scope.hs
56 lines 1.6 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--
13-----------------------------------------------------------------------------
15module Language.PureScript.Scope (
16 usedNames,
17 unusedNames
18) where
20import Data.Data
21import Data.List ((\\), nub)
22import Data.Generics (extQ, mkQ, everything)
24import Language.PureScript.Values
25import Language.PureScript.Names
26import Language.PureScript.CodeGen.JS.AST
28usedNames :: (Data d) => d -> [Ident]
29usedNames val = nub $ everything (++) (mkQ [] namesV `extQ` namesS `extQ` namesB `extQ` namesJS) val
30 where
31 namesV :: Value -> [Ident]
32 namesV (Abs args _) = args
33 namesV (Var (Qualified Nothing name)) = [name]
34 namesV _ = []
35 namesS :: Statement -> [Ident]
36 namesS (VariableIntroduction name _) = [name]
37 namesS (For name _ _ _) = [name]
38 namesS _ = []
39 namesB :: Binder -> [Ident]
40 namesB (VarBinder name) = [name]
41 namesB _ = []
42 namesJS :: JS -> [Ident]
43 namesJS (JSVar name) = [name]
44 namesJS (JSFunction (Just name) args _) = name : args
45 namesJS (JSFunction Nothing args _) = args
46 namesJS (JSVariableIntroduction name _) = [name]
47 namesJS (JSFor name _ _ _) = [name]
48 namesJS _ = []
50unusedNames :: (Data d) => d -> [Ident]
51unusedNames val =
52 let
53 allNames = usedNames val
54 varNames = map (Ident . ('_' :) . show) ([1..] :: [Int])
55 in
56 varNames \\ allNames