Main.hs

-- Copyright 2014 Google Inc. All rights reserved.

-- Licensed under the Apache License, Version 2.0 (the "License");
-- you may not use this file except in compliance with the License.
-- You may obtain a copy of the License at

--     http://www.apache.org/licenses/LICENSE-2.0

-- Unless required by applicable law or agreed to in writing, software
-- distributed under the License is distributed on an "AS IS" BASIS,
-- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-- See the License for the specific language governing permissions and
-- limitations under the License.
-- Module takes a JSAST and gives each vertex a unique integer label. The label counter is simply
-- threaded through the tree. Traversal is depth first. It's all fairly straight-forward.


-- The methods in the pipeline are:
-- 		ParseJS.parseTree - NEEDS PRETTY PRINT
-- 		ParseJS.getJSASTWithSource - NEEDS PRETTY PRINT
-- 		ResolveJSASTSourceFragments.jsastListWSMakeSourceFragments
-- 		LabelJSAST.label
-- 		DeclarationGraph.getDeclarationGraph - NEEDS PRETTY PRINT
-- 		DeclarationGraph.graphGetAllRules (optional) - TYPES NEED PRETTY PRINT
-- 		DeclarationGraph.cleanFunctionRules (optional)
-- 		DeclarationGraph.cleanFunction (optional)


module Main
(
main
) where


import DeclarationGraph
import LabelJSAST
import Language.JavaScript.Parser
import ParseJS
import PrettyPrint
import ResolveSourceFragments
import System.Environment
import TypeRules


-- TODO: Reverse these so that the earlier layers come first.
-- TODO: Put these sections into functions.
main :: IO ()
main = do
	(infile:[]) <- getArgs
	pr <- readFile infile
	putStrLn ""
	-- *****************
	putStrLn ""
	putStrLn "Print the stripped parse tree"
	printParseTreeStripped $ jsnGetNode $ parseTree pr infile
	putStrLn ""
	putStrLn "Print the raw parse tree"
	putStrLn $ show $ parse pr infile
	-- *****************
	-- PRETTY PRINTED
	-- Prints declared functions and function expressions, and the identifiers
	-- that are visible to each one.
	-- putStrLn ""
	-- putStrLn "Pretty print cleaned functions and function expressions, with identifiers"
	-- putStr "Top Level:"
	-- printCleanedElementList ((makeCleanedFunctions pr infile):[]) (makeIndent "")

	-- **PRETTY PRINTED**
	-- Prints the rules, indented base on their scope, with optional source code, and an optional
	-- list of the identifiers that are visible at that each scope.
	putStrLn ""
	putStrLn "Pretty print cleaned function rules with identifiers"
	putStr "Top Level:"
	printCleanedRulesList ((makeCleanedFunctionRules pr infile):[]) (makeIndent "") False True

	-- PRETTY PRINTED (could be improved with pretty printing for types)
	-- Print all the rules, optionally with source code.
	-- putStrLn ""
	-- putStrLn "Pretty print all the rules with code fragments"
	-- mapM_ (\r -> printRule r "" True) (makeAllRules pr infile)

	-- Print the raw declaration graph with rules.
	-- WARNING: Large and more-or-less illegible.
	-- putStrLn ""
	-- putStrLn "Print the declaration graph"
	-- putStrLn $ show $ makeDeclarationGraph pr infile

	-- PRETTY PRINTED
	-- Print the cleaned ATS.
	-- putStrLn ""
	-- putStrLn "Pretty print labelled JSAST without labels or source fragments"
	-- mapPrintASTChild (makeLabelledJSAST pr infile) (makeIndent "") False False
	-- PRETTY PRINTED
	-- Print the cleaned ATS with labels.
	-- putStrLn ""
	-- putStrLn "Pretty print labelled JSAST with labels"
	-- mapPrintASTChild (makeLabelledJSAST pr infile) (makeIndent "") False True
	-- PRETTY PRINTED
	-- Print the cleaned ATS with source.
	-- putStrLn ""
	-- putStrLn "Pretty print labelled JSAST with source fragments"
	-- mapPrintASTChild (makeLabelledJSAST pr infile) (makeIndent "") True False
	-- **PRETTY PRINTED**
	-- Print the cleaned ATS with labels and source.
	putStrLn ""
	putStrLn "Pretty print labelled JSAST with labels and source fragments"
	mapPrintASTChild (makeLabelledJSAST pr infile) (makeIndent "") True True

	-- **PRETTY PRINTED**
	-- Pretty print the JSASTWithSourceFragment with source fragments
	putStrLn ""
	putStrLn "Pretty print JSASTWithSourceFragment with source fragments"
	mapPrintASTWS (makeJSASTWithSourceFragments pr infile) (makeIndent "") True
	-- **PRETTY PRINTED**
	-- Pretty print the JSASTWithSourceFragment without source fragments
	putStrLn ""
	putStrLn "Pretty print JSASTWithSourceFragment without source fragments"
	mapPrintASTWS (makeJSASTWithSourceFragments pr infile) (makeIndent "") False

	-- Prints the first JSAST (pre-labels).
	-- putStrLn ""
	-- putStrLn "Print the the original JSAST"
	-- mapM_ print (makeJSAST pr infile)

	-- Rudimentary. Prints the parse tree using Language.JavaScript's showStripped function. Prints
	-- one top-level parse tree node per line.
	-- putStrLn ""
	-- putStrLn "Print the stripped parse tree"
	-- printParseTreeStripped $ jsnGetNode $ parseTree pr infile

	-- Print the raw parse tree.
	-- putStrLn ""
	-- putStrLn "Print the raw parse tree"
	-- putStrLn $ show $ parse pr infile


-- TODO: Reverse these so that the earlier layers come first.
makeCleanedFunctions :: String -> SourceFileName -> CleanedFunction
makeCleanedFunctions input fileName = cleanFunction $ makeCleanedFunctionRules input fileName


makeCleanedFunctionRules :: String -> SourceFileName -> CleanedFunctionRules
makeCleanedFunctionRules input fileName = cleanFunctionRules $ makeDeclarationGraph input fileName


makeAllRules :: String -> SourceFileName -> [Rule]
makeAllRules input fileName = graphGetAllRules $ makeDeclarationGraph input fileName


makeDeclarationGraph :: String -> SourceFileName -> FunctionRules
makeDeclarationGraph input fileName =
	getDeclarationGraph
		(makeLabelledJSAST input fileName)
		(fileName, 1, 1, ((length $ lines input) + 1), 1)


makeLabelledJSAST :: String -> SourceFileName -> [ASTChild]
makeLabelledJSAST input fileName = label $ makeJSASTWithSourceFragments input fileName


-- FIXME: Passing the file name here might mean that we don't need to thread it through the whole
-- AST.
makeJSASTWithSourceFragments :: String -> SourceFileName -> [JSASTWithSourceFragment]
makeJSASTWithSourceFragments input fileName =
	jsastListWSMakeSourceFragments (makeJSAST input fileName) (SpanPoint fileName ((length $ lines input) + 1) 1)


makeJSAST :: String -> SourceFileName -> ([JSASTWithSourceSpan], SourceFileName)
makeJSAST input fileName = getJSASTWithSource (parseTree input fileName) fileName