Fix some problems

Author: MasloMaslane

src/sinol_make/__init__.py

@@ -59,15 +59,19 @@ def main_exn():
                 except Exception as err:
                     util.exit_with_error('`oiejq` could not be installed.\n' + err)
 
-                command.run(args)
-                exit(0)
+            command.run(args)
+            exit(0)
 
     parser.print_help()
 
 
 def main():
     try:
         main_exn()
+    except argparse.ArgumentError as err:
+        util.exit_with_error(err)
+    except SystemExit as err:
+        exit(err.code)
     except:
         print(traceback.format_exc())
         util.exit_with_error('An error occurred while running the command.\n'

src/sinol_make/helpers/compile.py

@@ -101,14 +101,15 @@ def compile_file(file_path: str, name: str, compilers: Compilers, weak_compilati
 
     extra_compilation_files = [os.path.join(os.getcwd(), "prog", file)
                                for file in config.get("extra_compilation_files", [])]
-    extra_compilation_args = config.get('extra_compilation_args', {}).get(os.path.splitext(name)[1], [])
+    extra_compilation_args = [os.path.join(os.getcwd(), "prog", file)
+                              for file in config.get('extra_compilation_args', {}).get(os.path.splitext(file_path)[1][1:], [])]
 
     output = os.path.join(executable_dir, name)
     compile_log_path = os.path.join(compile_log_dir, os.path.splitext(name)[0] + '.compile_log')
     with open(compile_log_path, 'w') as compile_log:
         try:
-            if compile(file_path, output, compilers, compile_log, weak_compilation_flags, extra_compilation_files,
-                       extra_compilation_args):
+            if compile(file_path, output, compilers, compile_log, weak_compilation_flags, extra_compilation_args,
+                       extra_compilation_files):
                 return output, compile_log_path
         except CompilationError:
             pass