contrib/objcwrapper: move `model` to an attribute of `CodeGenerator`

[nit.git] / contrib / objcwrapper / src / objcwrapper.nit

diff --git a/contrib/objcwrapper/src/objcwrapper.nit b/contrib/objcwrapper/src/objcwrapper.nit
index 3a941ed..0e3cb18 100644 (file)
--- a/contrib/objcwrapper/src/objcwrapper.nit
+++ b/contrib/objcwrapper/src/objcwrapper.nit
@@ -15,18 +15,34 @@
 # Generator of Nit modules to wrap Objective-C services
 module objcwrapper
 
+import nitcc_runtime
+import opts
+
 import objc_visitor
 import objc_model
 import objc_generator
-
-import nitcc_runtime
 import objc_lexer
 import objc_parser
 
-var v = new Interpretor
-var g = new CodeGenerator
+var opt_help = new OptionBool("Show this help message", "-h", "--help")
+
+var opts = new OptionContext
+opts.add_option(opt_help, opt_output, opt_init_as_methods)
+opts.parse(args)
+
+if opts.errors.not_empty or opts.rest.is_empty or opt_help.value then
+       print """
+Usage: objcwrapper [options] input_file [other_input_file [...]]
+Options:"""
+       opts.usage
+
+       if opt_help.value then exit 0
+       exit 1
+end
+
+var v = new ObjcVisitor
 
-for arg in args do
+for arg in opts.rest do
        # Read input
        var content = arg.to_path.read_all
 
@@ -47,4 +63,5 @@ for arg in args do
        v.enter_visit root
 end
 
-g.generator v.model.classes
+var g = new CodeGenerator(v.model)
+g.generate