1 # This file is part of NIT ( http://www.nitlanguage.org ).
3 # Copyright 2012 Jean Privat <jean@pryen.org>
5 # Licensed under the Apache License, Version 2.0 (the "License");
6 # you may not use this file except in compliance with the License.
7 # You may obtain a copy of the License at
9 # http://www.apache.org/licenses/LICENSE-2.0
11 # Unless required by applicable law or agreed to in writing, software
12 # distributed under the License is distributed on an "AS IS" BASIS,
13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 # See the License for the specific language governing permissions and
15 # limitations under the License.
17 # Load nit source files and build the associated model
21 # FIXME split this module into submodules
22 # FIXME add missing error checks
23 module modelbuilder_base
29 private import more_collections
33 redef class ToolContext
35 # The modelbuilder 1-to-1 associated with the toolcontext
36 fun modelbuilder
: ModelBuilder do return modelbuilder_real
.as(not null)
38 private var modelbuilder_real
: nullable ModelBuilder = null
42 # A model builder knows how to load nit source files and build the associated model
44 # The model where new modules, classes and properties are added
47 # The toolcontext used to control the interaction with the user (getting options and displaying messages)
48 var toolcontext
: ToolContext
50 # Instantiate a modelbuilder for a model and a toolcontext
51 # Important, the options of the toolcontext must be correctly set (parse_option already called)
54 assert toolcontext
.modelbuilder_real
== null
55 toolcontext
.modelbuilder_real
= self
58 # Return a class named `name` visible by the module `mmodule`.
59 # Visibility in modules is correctly handled.
60 # If no such a class exists, then null is returned.
61 # If more than one class exists, then an error on `anode` is displayed and null is returned.
62 # FIXME: add a way to handle class name conflict
63 fun try_get_mclass_by_name
(anode
: ANode, mmodule
: MModule, name
: String): nullable MClass
65 var classes
= model
.get_mclasses_by_name
(name
)
66 if classes
== null then
70 var res
: nullable MClass = null
71 for mclass
in classes
do
72 if not mmodule
.in_importation
<= mclass
.intro_mmodule
then continue
73 if not mmodule
.is_visible
(mclass
.intro_mmodule
, mclass
.visibility
) then continue
77 error
(anode
, "Error: ambiguous class name `{name}`; conflict between `{mclass.full_name}` and `{res.full_name}`.")
84 # Return a class identified by `qid` visible by the module `mmodule`.
85 # Visibility in modules and qualified names are correctly handled.
87 # If more than one class exists, then null is silently returned.
88 # It is up to the caller to post-analysis the result and display a correct error message.
89 # The method `class_not_found` can be used to display such a message.
90 fun try_get_mclass_by_qid
(qid
: AQclassid, mmodule
: MModule): nullable MClass
92 var name
= qid
.n_id
.text
94 var classes
= model
.get_mclasses_by_name
(name
)
95 if classes
== null then
99 var res
: nullable MClass = null
100 for mclass
in classes
do
101 if not mmodule
.in_importation
<= mclass
.intro_mmodule
then continue
102 if not mmodule
.is_visible
(mclass
.intro_mmodule
, mclass
.visibility
) then continue
103 if not qid
.accept
(mclass
) then continue
114 # Like `try_get_mclass_by_name` but display an error message when the class is not found
115 fun get_mclass_by_name
(node
: ANode, mmodule
: MModule, name
: String): nullable MClass
117 var mclass
= try_get_mclass_by_name
(node
, mmodule
, name
)
118 if mclass
== null then
119 error
(node
, "Type Error: missing primitive class `{name}'.")
124 # Return a property named `name` on the type `mtype` visible in the module `mmodule`.
125 # Visibility in modules is correctly handled.
126 # Protected properties are returned (it is up to the caller to check and reject protected properties).
127 # If no such a property exists, then null is returned.
128 # If more than one property exists, then an error on `anode` is displayed and null is returned.
129 # FIXME: add a way to handle property name conflict
130 fun try_get_mproperty_by_name2
(anode
: ANode, mmodule
: MModule, mtype
: MType, name
: String): nullable MProperty
132 var props
= self.model
.get_mproperties_by_name
(name
)
133 if props
== null then
137 var cache
= self.try_get_mproperty_by_name2_cache
[mmodule
, mtype
, name
]
138 if cache
!= null then return cache
140 var res
: nullable MProperty = null
141 var ress
: nullable Array[MProperty] = null
142 for mprop
in props
do
143 if not mtype
.has_mproperty
(mmodule
, mprop
) then continue
144 if not mmodule
.is_visible
(mprop
.intro_mclassdef
.mmodule
, mprop
.visibility
) then continue
146 # new-factories are invisible outside of the class
147 if mprop
isa MMethod and mprop
.is_new
and (not mtype
isa MClassType or mprop
.intro_mclassdef
.mclass
!= mtype
.mclass
) then
156 # Two global properties?
157 # First, special case for init, keep the most specific ones
158 if res
isa MMethod and mprop
isa MMethod and res
.is_init
and mprop
.is_init
then
159 var restype
= res
.intro_mclassdef
.bound_mtype
160 var mproptype
= mprop
.intro_mclassdef
.bound_mtype
161 if mproptype
.is_subtype
(mmodule
, null, restype
) then
162 # found a most specific constructor, so keep it
168 # Ok, just keep all prop in the ress table
170 ress
= new Array[MProperty]
177 if ress
!= null and res
isa MMethod and res
.is_init
then
178 # special case forinit again
179 var restype
= res
.intro_mclassdef
.bound_mtype
180 var ress2
= new Array[MProperty]
182 var mproptype
= mprop
.intro_mclassdef
.bound_mtype
183 if not restype
.is_subtype
(mmodule
, null, mproptype
) then
185 else if not mprop
isa MMethod or not mprop
.is_init
then
189 if ress2
.is_empty
then
198 assert ress
.length
> 1
199 var s
= new Array[String]
200 for mprop
in ress
do s
.add mprop
.full_name
201 self.error
(anode
, "Error: ambiguous property name `{name}` for `{mtype}`; conflict between {s.join(" and ")}.")
204 self.try_get_mproperty_by_name2_cache
[mmodule
, mtype
, name
] = res
208 private var try_get_mproperty_by_name2_cache
= new HashMap3[MModule, MType, String, nullable MProperty]
211 # Alias for try_get_mproperty_by_name2(anode, mclassdef.mmodule, mclassdef.mtype, name)
212 fun try_get_mproperty_by_name
(anode
: ANode, mclassdef
: MClassDef, name
: String): nullable MProperty
214 return try_get_mproperty_by_name2
(anode
, mclassdef
.mmodule
, mclassdef
.bound_mtype
, name
)
217 # Helper function to display an error on a node.
218 # Alias for `self.toolcontext.error(n.hot_location, text)`
220 # This automatically sets `n.is_broken` to true.
221 fun error
(n
: nullable ANode, text
: String)
228 self.toolcontext
.error
(l
, text
)
231 # Helper function to display a warning on a node.
232 # Alias for: `self.toolcontext.warning(n.hot_location, text)`
233 fun warning
(n
: nullable ANode, tag
, text
: String)
236 if n
!= null then l
= n
.hot_location
237 self.toolcontext
.warning
(l
, tag
, text
)
240 # Helper function to display an advice on a node.
241 # Alias for: `self.toolcontext.advice(n.hot_location, text)`
242 fun advice
(n
: nullable ANode, tag
, text
: String)
245 if n
!= null then l
= n
.hot_location
246 self.toolcontext
.advice
(l
, tag
, text
)
249 # Force to get the primitive method named `name` on the type `recv` or do a fatal error on `n`
250 fun force_get_primitive_method
(n
: nullable ANode, name
: String, recv
: MClass, mmodule
: MModule): MMethod
252 var res
= mmodule
.try_get_primitive_method
(name
, recv
)
255 if n
!= null then l
= n
.hot_location
256 self.toolcontext
.fatal_error
(l
, "Fatal Error: `{recv}` must have a property named `{name}`.")
262 # Return the static type associated to the node `ntype`.
263 # `mmodule` and `mclassdef` is the context where the call is made (used to understand formal types)
264 # In case of problem, an error is displayed on `ntype` and null is returned.
265 # FIXME: the name "resolve_mtype" is awful
266 fun resolve_mtype_unchecked
(mmodule
: MModule, mclassdef
: nullable MClassDef, ntype
: AType, with_virtual
: Bool): nullable MType
268 var qid
= ntype
.n_qid
269 var name
= qid
.n_id
.text
273 if mclassdef
!= null and with_virtual
then
274 var prop
= try_get_mproperty_by_name
(ntype
, mclassdef
, name
).as(nullable MVirtualTypeProp)
276 if not ntype
.n_types
.is_empty
then
277 error
(ntype
, "Type Error: formal type `{name}` cannot have formal parameters.")
279 res
= prop
.mvirtualtype
280 if ntype
.n_kwnullable
!= null then res
= res
.as_nullable
286 # Check parameter type
287 if mclassdef
!= null then
288 for p
in mclassdef
.mclass
.mparameters
do
289 if p
.name
!= name
then continue
291 if not ntype
.n_types
.is_empty
then
292 error
(ntype
, "Type Error: formal type `{name}` cannot have formal parameters.")
296 if ntype
.n_kwnullable
!= null then res
= res
.as_nullable
303 var mclass
= try_get_mclass_by_qid
(qid
, mmodule
)
304 if mclass
!= null then
305 var arity
= ntype
.n_types
.length
306 if arity
!= mclass
.arity
then
308 error
(ntype
, "Type Error: `{mclass.signature_to_s}` is a generic class.")
309 else if mclass
.arity
== 0 then
310 error
(ntype
, "Type Error: `{name}` is not a generic class.")
312 error
(ntype
, "Type Error: expected {mclass.arity} formal argument(s) for `{mclass.signature_to_s}`; got {arity}.")
317 res
= mclass
.mclass_type
318 if ntype
.n_kwnullable
!= null then res
= res
.as_nullable
322 var mtypes
= new Array[MType]
323 for nt
in ntype
.n_types
do
324 var mt
= resolve_mtype_unchecked
(mmodule
, mclassdef
, nt
, with_virtual
)
325 if mt
== null then return null # Forward error
328 res
= mclass
.get_mtype
(mtypes
)
329 if ntype
.n_kwnullable
!= null then res
= res
.as_nullable
335 # If everything fail, then give up with class by proposing things.
337 # TODO Give hints on formal types (param and virtual)
338 class_not_found
(qid
, mmodule
)
339 ntype
.is_broken
= true
343 # Print an error and suggest hints when the class identified by `qid` in `mmodule` is not found.
345 # This just print error messages.
346 fun class_not_found
(qid
: AQclassid, mmodule
: MModule)
348 var name
= qid
.n_id
.text
349 var qname
= qid
.full_name
351 var all_classes
= model
.get_mclasses_by_name
(name
)
352 var hints
= new Array[String]
354 # Look for conflicting classes.
355 if all_classes
!= null then for c
in all_classes
do
356 if not mmodule
.is_visible
(c
.intro_mmodule
, c
.visibility
) then continue
357 if not qid
.accept
(c
) then continue
358 hints
.add
"`{c.full_name}`"
360 if hints
.length
> 1 then
361 error
(qid
, "Error: ambiguous class name `{qname}` in module `{mmodule}`. Conflicts are between {hints.join(",", " and ")}.")
366 # Look for imported but invisible classes.
367 if all_classes
!= null then for c
in all_classes
do
368 if not mmodule
.in_importation
<= c
.intro_mmodule
then continue
369 if mmodule
.is_visible
(c
.intro_mmodule
, c
.visibility
) then continue
370 if not qid
.accept
(c
) then continue
371 error
(qid
, "Error: class `{c.full_name}` not visible in module `{mmodule}`.")
375 # Look for not imported but known classes from importable modules
376 if all_classes
!= null then for c
in all_classes
do
377 if mmodule
.in_importation
<= c
.intro_mmodule
then continue
378 if c
.intro_mmodule
.in_importation
<= mmodule
then continue
379 if c
.visibility
<= private_visibility
then continue
380 if not qid
.accept
(c
) then continue
381 hints
.add
"`{c.intro_mmodule.full_name}`"
383 if hints
.not_empty
then
384 error
(qid
, "Error: class `{qname}` not found in module `{mmodule}`. Maybe import {hints.join(",", " or ")}?")
388 # Look for classes with an approximative name.
389 var bests
= new BestDistance[MClass](qname
.length
- name
.length
/ 2) # limit up to 50% name change
390 for c
in model
.mclasses
do
391 if not mmodule
.in_importation
<= c
.intro_mmodule
then continue
392 if not mmodule
.is_visible
(c
.intro_mmodule
, c
.visibility
) then continue
393 var d
= qname
.levenshtein_distance
(c
.name
)
395 d
= qname
.levenshtein_distance
(c
.full_name
)
398 if bests
.best_items
.not_empty
then
399 for c
in bests
.best_items
do hints
.add
"`{c.full_name}`"
400 error
(qid
, "Error: class `{qname}` not found in module `{mmodule}`. Did you mean {hints.join(",", " or ")}?")
404 error
(qid
, "Error: class `{qname}` not found in module `{mmodule}`.")
407 # Return the static type associated to the node `ntype`.
408 # `mmodule` and `mclassdef` is the context where the call is made (used to understand formal types)
409 # In case of problem, an error is displayed on `ntype` and null is returned.
410 # FIXME: the name "resolve_mtype" is awful
411 fun resolve_mtype
(mmodule
: MModule, mclassdef
: nullable MClassDef, ntype
: AType): nullable MType
413 var mtype
= ntype
.mtype
414 if mtype
== null then mtype
= resolve_mtype_unchecked
(mmodule
, mclassdef
, ntype
, true)
415 if mtype
== null then return null # Forward error
417 if ntype
.checked_mtype
then return mtype
418 if mtype
isa MGenericType then
419 var mclass
= mtype
.mclass
420 for i
in [0..mclass
.arity
[ do
421 var intro
= mclass
.try_intro
422 if intro
== null then return null # skip error
423 var bound
= intro
.bound_mtype
.arguments
[i
]
424 var nt
= ntype
.n_types
[i
]
425 var mt
= resolve_mtype
(mmodule
, mclassdef
, nt
)
426 if mt
== null then return null # forward error
428 if mclassdef
!= null then anchor
= mclassdef
.bound_mtype
else anchor
= null
429 if not check_subtype
(nt
, mmodule
, anchor
, mt
, bound
) then
430 error
(nt
, "Type Error: expected `{bound}`, got `{mt}`.")
435 ntype
.checked_mtype
= true
439 # Check that `sub` is a subtype of `sup`.
440 # Do not display an error message.
442 # This method is used a an entry point for the modelize phase to test static subtypes.
443 # Some refinements could redefine it to collect statictics.
444 fun check_subtype
(node
: ANode, mmodule
: MModule, anchor
: nullable MClassType, sub
, sup
: MType): Bool
446 return sub
.is_subtype
(mmodule
, anchor
, sup
)
449 # Check that `sub` and `sup` are equvalent types.
450 # Do not display an error message.
452 # This method is used a an entry point for the modelize phase to test static equivalent types.
453 # Some refinements could redefine it to collect statictics.
454 fun check_sametype
(node
: ANode, mmodule
: MModule, anchor
: nullable MClassType, sub
, sup
: MType): Bool
456 return sub
.is_subtype
(mmodule
, anchor
, sup
) and sup
.is_subtype
(mmodule
, anchor
, sub
)
461 # The indication that the node did not pass some semantic verifications.
463 # This simple flag is set by a given analysis to say that the node is broken and unusable in
465 # When a node status is set to broken, it is usually associated with a error message.
467 # If it is safe to do so, clients of the AST SHOULD just skip broken nodes in their processing.
468 # Clients that do not care about the executability (e.g. metrics) MAY still process the node or
469 # perform specific checks to determinate the validity of the node.
471 # Note that the broken status is not propagated to parent or children nodes.
472 # e.g. a broken expression used as argument does not make the whole call broken.
473 var is_broken
= false is writable
477 # The mtype associated to the node
478 var mtype
: nullable MType = null
480 # Is the mtype a valid one?
481 var checked_mtype
: Bool = false
484 redef class AVisibility
485 # The visibility level associated with the AST node class
486 fun mvisibility
: MVisibility is abstract
488 redef class AIntrudeVisibility
489 redef fun mvisibility
do return intrude_visibility
491 redef class APublicVisibility
492 redef fun mvisibility
do return public_visibility
494 redef class AProtectedVisibility
495 redef fun mvisibility
do return protected_visibility
497 redef class APrivateVisibility
498 redef fun mvisibility
do return private_visibility
502 private var mdoc_cache
: nullable MDoc
504 # Convert `self` to a `MDoc`
508 if res
!= null then return res
509 res
= new MDoc(location
)
510 for c
in n_comment
do
512 if text
.length
< 2 then
516 assert text
.chars
[0] == '#'
517 if text
.chars
[1] == ' ' then
518 text
= text
.substring_from
(2) # eat starting `#` and space
520 text
= text
.substring_from
(1) # eat atarting `#` only
522 if text
.chars
.last
== '\n' then text
= text
.substring
(0, text
.length-1
) # drop \n
523 res
.content
.add
(text
)
530 redef class AQclassid
531 # The name of the package part, if any
532 fun mpackname
: nullable String do
533 var nqualified
= n_qualified
534 if nqualified
== null then return null
535 var nids
= nqualified
.n_id
536 if nids
.length
<= 0 then return null
540 # The name of the module part, if any
541 fun mmodname
: nullable String do
542 var nqualified
= n_qualified
543 if nqualified
== null then return null
544 var nids
= nqualified
.n_id
545 if nids
.length
<= 1 then return null
549 # Does `mclass` match the full qualified name?
550 fun accept
(mclass
: MClass): Bool
552 if mclass
.name
!= n_id
.text
then return false
553 var mpackname
= self.mpackname
554 if mpackname
!= null then
555 var mpackage
= mclass
.intro_mmodule
.mpackage
556 if mpackage
== null then return false
557 if mpackage
.name
!= mpackname
then return false
558 var mmodname
= self.mmodname
559 if mmodname
!= null and mclass
.intro_mmodule
.name
!= mmodname
then return false
564 # The pretty name represented by self.
565 fun full_name
: String
568 var nqualified
= n_qualified
569 if nqualified
== null then return res
570 var ncid
= nqualified
.n_classid
571 if ncid
!= null then res
= ncid
.text
+ "::" + res
572 var nids
= nqualified
.n_id
573 if nids
.not_empty
then for n
in nids
.reverse_iterator
do
574 res
= n
.text
+ "::" + res