1 # This file is part of NIT ( http://www.nitlanguage.org ).
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
7 # http://www.apache.org/licenses/LICENSE-2.0
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
15 # Tool that produces highlighting for Nit programs
20 var toolcontext
= new ToolContext
22 # Disable `cached` because it causes issues when printing transformed AST. FIXME
23 toolcontext
.cached_phase
.disabled
= true
25 var opt_fragment
= new OptionBool("Omit document header and footer", "-f", "--fragment")
26 var opt_first_line
= new OptionInt("Start the source file at this line (default: 1)", 0, "--first-line")
27 var opt_last_line
= new OptionInt("End the source file at this line (default: to the end)", 0, "--last-line")
28 var opt_dir
= new OptionString("Output html files in a specific directory (required if more than one module)", "-d", "--dir")
29 var opt_full
= new OptionBool("Process also imported modules", "--full")
30 var opt_ast
= new OptionBool("Generate specific HTML elements for each Node of the AST", "--ast")
31 toolcontext
.option_context
.add_option
(opt_fragment
, opt_first_line
, opt_last_line
, opt_dir
, opt_full
)
32 toolcontext
.tooldescription
= "Usage: nitlight [OPTION]... <file.nit>...\nGenerates HTML of highlited code from Nit source files."
33 toolcontext
.process_options
(args
)
36 var modelbuilder
= new ModelBuilder(model
, toolcontext
)
38 var args
= toolcontext
.option_context
.rest
40 var mmodules
= modelbuilder
.parse
(args
)
41 modelbuilder
.run_phases
43 if opt_full
.value
then mmodules
= model
.mmodules
45 var dir
= opt_dir
.value
48 else if mmodules
.length
> 1 then
49 print
"More than one module to render, use option -d"
54 if dir
!= null then toolcontext
.info
("write {dir}/{mm.name}.html", 1)
56 var v
= new HighlightVisitor
58 if opt_first_line
.value
!= 0 then v
.first_line
= opt_first_line
.value
59 if opt_last_line
.value
!= 0 then v
.last_line
= opt_last_line
.value
60 if opt_ast
.value
then v
.with_ast
= true
62 var m
= modelbuilder
.mmodule2nmodule
[mm
]
63 if not opt_fragment
.value
then
64 page
= new HTMLTag("html")
65 page
.add_raw_html
"""<head>
66 <meta charset="utf-8">
67 <title>file {{{m.location.file.filename}}}</title>"""
70 <style type="text/css">
75 page
.add_raw_html
"""<link rel="stylesheet" type="text/css" href="style.css" />"""
77 page
.add_raw_html v
.head_content
78 page
.add_raw_html
"</head><body><pre class='nit_code'>"
81 if not opt_fragment
.value
then
83 page
.add_raw_html
"</pre>"
84 page
.add_raw_html v
.foot_content
85 page
.add_raw_html
"</body>"
91 page
.write_to_file
("{dir}/{mm.name}.html")
98 toolcontext
.info
("write {dir}/index.html", 1)
100 var page
= new HTMLTag("html")
101 page
.add_raw_html
"""<head>
102 <meta charset="utf-8">
105 for mm
in mmodules
do
106 var n
= new HTMLTag("li")
107 var n2
= new HTMLTag("a")
110 n2
.attr
("href", "{mm.name}.html")
113 page
.add_raw_html
"</li></body>"
114 page
.write_to_file
("{dir}/index.html")
116 var v
= new HighlightVisitor
117 toolcontext
.info
("write {dir}/style.css", 1)
118 var f
= new OFStream.open
("{dir}/style.css")
119 f
.write v
.css_content