tools: add entrypoint information to the program
[nit.git] / src / program.nit
1 # This file is part of NIT ( http://www.nitlanguage.org ).
2 #
3 # Copyright 2009 Jean-Sebastien Gelinas <calestar@gmail.com>
4 #
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
8 #
9 # http://www.apache.org/licenses/LICENSE-2.0
10 #
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.
16
17 # Common things to represent a Nit program/library
18 package program
19
20 import metamodel
21
22 # Instances of this class represent a program/library that will
23 # be analyzed/compiled by nitc
24 class Program
25 # This module is the 'main' module, the one where we find the 'main' method
26 readable var _module: MMModule
27
28 # This method is the entry point of this program
29 # There might be no entry point (if in fact we are compiling a library)
30 readable var _main_method: nullable MMMethod = null
31
32 # This is the class that contains the main method.
33 # Would be null if there is no main method
34 readable var _main_class: nullable MMLocalClass = null
35
36 fun compute_main_method do
37 # Check for the 'Sys' class
38 var sysname = once "Sys".to_symbol
39 if not module.has_global_class_named(sysname) then return
40 var sys = module.class_by_name(sysname)
41
42 # Check for 'Sys::main' method
43 var entryname = once "main".to_symbol
44 if not sys.has_global_property_by_name(entryname) then return
45
46 _main_method = sys.select_method(entryname)
47 _main_class = sys
48 end
49
50 init(m: MMModule) do
51 _module = m
52 end
53 end