pep8analysis: remove old-style accessors in the AST

[nit.git] / src / toolcontext.nit

diff --git a/src/toolcontext.nit b/src/toolcontext.nit
index b91fdad..5b9d020 100644 (file)
--- a/src/toolcontext.nit
+++ b/src/toolcontext.nit
@@ -21,6 +21,7 @@ module toolcontext
 
 import opts
 import location
+import version
 
 class Message
        super Comparable
@@ -89,9 +90,9 @@ class ToolContext
 
                        for m in messages do
                                if opt_no_color.value then
-                                       stderr.write("{m}\n")
+                                       sys.stderr.write("{m}\n")
                                else
-                                       stderr.write("{m.to_color_string}\n")
+                                       sys.stderr.write("{m.to_color_string}\n")
                                end
                        end
 
@@ -133,6 +134,33 @@ class ToolContext
                end
        end
 
+       # Executes a program while checking if it's available and if the execution ended correctly
+       #
+       # Stops execution and prints errors if the program isn't available or didn't end correctly
+       fun exec_and_check(args: Array[String], error: String)
+        do
+                var prog = args.first
+                args.remove_at 0
+
+                # Is the wanted program available?
+                var proc_which = new IProcess.from_a("which", [prog])
+                proc_which.wait
+                var res = proc_which.status
+                if res != 0 then
+                        print "{error}: executable \"{prog}\" not found"
+                        exit 1
+                end
+
+                # Execute the wanted program
+                var proc = new Process.from_a(prog, args)
+                proc.wait
+                res = proc.status
+                if res != 0 then
+                        print "{error}: execution of \"{prog} {args.join(" ")}\" failed"
+                        exit 1
+                end
+        end
+
        # Global OptionContext
        var option_context: OptionContext = new OptionContext
 
@@ -179,7 +207,15 @@ class ToolContext
        # eg. `"Usage: tool [OPTION]... [FILE]...\nDo some things."`
        var tooldescription: String writable = "Usage: [OPTION]... [ARG]..."
 
+       # Does `process_options` should accept an empty sequence of arguments.
+       # ie. nothing except options.
+       # Is `false` by default.
+       #
+       # If required, if should be set by the client before calling `process_options`
+       var accept_no_arguments writable = false
+
        # print the full usage of the tool.
+       # Is called by `process_option` on `--help`.
        # It also could be called by the client.
        fun usage
        do
@@ -188,13 +224,37 @@ class ToolContext
        end
 
        # Parse and process the options given on the command line
-       fun process_options
+       fun process_options(args: Sequence[String])
        do
                self.opt_warn.value = 1
 
                # init options
                option_context.parse(args)
 
+               if opt_help.value then
+                       usage
+                       exit 0
+               end
+
+               if opt_version.value then
+                       print nit_version
+                       exit 0
+               end
+
+               var errors = option_context.get_errors
+               if not errors.is_empty then
+                       for e in errors do print "Error: {e}"
+                       print tooldescription
+                       print "Use --help for help"
+                       exit 1
+               end
+
+               if option_context.rest.is_empty and not accept_no_arguments then
+                       print tooldescription
+                       print "Use --help for help"
+                       exit 1
+               end
+
                # Set verbose level
                verbose_level = opt_verbose.value
 
@@ -205,5 +265,31 @@ class ToolContext
                        # Make sure the output directory exists
                        log_directory.mkdir
                end
+
+               nit_dir = compute_nit_dir
+       end
+
+       # The identified root directory of the Nit project
+       var nit_dir: nullable String
+
+       private fun compute_nit_dir: nullable String
+       do
+               # a environ variable has precedence
+               var res = "NIT_DIR".environ
+               if not res.is_empty then return res
+
+               # find the runpath of the program from argv[0]
+               res = "{sys.program_name.dirname}/.."
+               if res.file_exists and "{res}/src/nit.nit".file_exists then return res.simplify_path
+
+               # find the runpath of the process from /proc
+               var exe = "/proc/self/exe"
+               if exe.file_exists then
+                       res = exe.realpath
+                       res = res.dirname.join_path("..")
+                       if res.file_exists and "{res}/src/nit.nit".file_exists then return res.simplify_path
+               end
+
+               return null
        end
 end