@@ -58,12 +58,12 @@ isReassigned var1 = everything (||) (mkQ False check)
5858 check _ = False
5959
6060isRebound :: (Data d) => JS -> d -> Bool
61−isRebound (JSVar var1) = everything (||) (mkQ False check)
61+isRebound js d = any (\var -> isReassigned var d) (variablesOf js)
6262 where
63− check :: JS -> Bool
64− check (JSFunction _ args _) | var1 `elem` args = True
65− check _ = False
66−isRebound _ = const False
63+ variablesOf (JSVar var) = [var]
64+ variablesOf (JSAccessor _ val) = variablesOf val
65+ variablesOf (JSIndexer index val) = variablesOf index ++ variablesOf val
66+ variablesOf _ = []
6767
6868isUsed :: (Data d) => Ident -> d -> Bool
6969isUsed var1 = everything (||) (mkQ False check)
@@ -72,9 +72,17 @@ isUsed var1 = everything (||) (mkQ False check)
7272 check (JSVar var2) | var1 == var2 = True
7373 check (JSAssignment target _) | var1 == targetVariable target = True
7474 check _ = False
75− targetVariable :: JSAssignment -> Ident
76− targetVariable (JSAssignVariable var) = var
77− targetVariable (JSAssignProperty _ tgt) = targetVariable tgt
75+
76+targetVariable :: JSAssignment -> Ident
77+targetVariable (JSAssignVariable var) = var
78+targetVariable (JSAssignProperty _ tgt) = targetVariable tgt
79+
80+isUpdated :: (Data d) => Ident -> d -> Bool
81+isUpdated var1 = everything (||) (mkQ False check)
82+ where
83+ check :: JS -> Bool
84+ check (JSAssignment target _) | var1 == targetVariable target = True
85+ check _ = False
7886
7987shouldInline :: JS -> Bool
8088shouldInline (JSVar _) = True
@@ -93,7 +101,9 @@ inlineVariables = everywhere (mkT removeFromBlock)
93101 removeFromBlock js = js
94102 go :: [JS] -> [JS]
95103 go [] = []
96− go (JSVariableIntroduction var (Just js) : sts) | shouldInline js && not (isReassigned var sts) && not (isRebound js sts) = go (replaceIdent var js sts)
104+ go (s@(JSVariableIntroduction var (Just js)) : sts)
105+ | shouldInline js && not (isReassigned var sts) && not (isRebound js sts) && not (isUpdated var sts) =
106+ go (replaceIdent var js sts)
97107 go (s:sts) = s : go sts
98108
99109removeUnusedVariables :: JS -> JS