Allow input files to not have .erl extension (62753294) · Commits · Program Slicing / e-Knife for Erlang with Constrained Edges

e-Knife/src/main/resources/ast.erl

+6 −4

Original line number	Diff line number	Diff line
		@@ -28,13 +28,15 @@ getASTs(FileOrDir) ->
		end.

		getErlASTs(FileOrDir) ->
		case filename:extension(FileOrDir) of
		[] -> % It is a directory
		IsFile = filelib:is_regular(FileOrDir),
		IsDir = filelib:is_dir(FileOrDir),
		if
		IsDir -> % A directory
		{ok, Files} = file:list_dir(FileOrDir),
		lists:concat([getErlASTs(filename:absname(F, FileOrDir)) \|\| F <- Files]);
		".erl" -> % An erlang file
		IsFile -> % A file
		getFileASTs(FileOrDir);
		_ -> % It is not an erlang file
		true ->
		[]
		end.