enable compiler config

tommy · tommy · commit ad0cf42f2031 · 2021-08-16T03:19:57.000+09:00
diff --git a/README.md b/README.md
@@ -234,6 +234,9 @@ code_generator_file="~/custom_code_generator.py"
 [postprocess]
 exec_on_each_problem_dir='clang-format -i ./*.cpp'
 exec_on_contest_dir='touch CMakeLists.txt'
+[compiler]
+compile_command='g++ main.cpp -o main -std=c++17'
+compile_only_when_diff_detected=true
 [tester]
 compile_before_testing=true
 compile_only_when_diff_detected=true
diff --git a/atcodertools/config/compiler_config.py b/atcodertools/config/compiler_config.py
@@ -0,0 +1,8 @@
+class CompilerConfig:
+
+    def __init__(self,
+                 compile_only_when_diff_detected: bool = False,
+                 compile_command: str = None
+                 ):
+        self.compile_only_when_diff_detected = compile_only_when_diff_detected
+        self.compile_command = compile_command
diff --git a/atcodertools/config/config.py b/atcodertools/config/config.py
@@ -8,13 +8,15 @@
 from atcodertools.config.etc_config import EtcConfig
 from atcodertools.config.postprocess_config import PostprocessConfig
 from atcodertools.config.tester_config import TesterConfig
+from atcodertools.config.compiler_config import CompilerConfig
 
 
 class ConfigType(Enum):
     CODESTYLE = "codestyle"
     POSTPROCESS = "postprocess"
     TESTER = "tester"
     ETC = "etc"
+    COMPILER = "compiler"
 
 
 def _update_config_dict(target_dic: Dict[str, Any], update_dic: Dict[str, Any]):
@@ -44,7 +46,8 @@ def __init__(self,
                  code_style_config: CodeStyleConfig = CodeStyleConfig(),
                  postprocess_config: PostprocessConfig = PostprocessConfig(),
                  tester_config: TesterConfig = TesterConfig(),
-                 etc_config: EtcConfig = EtcConfig()
+                 etc_config: EtcConfig = EtcConfig(),
+                 compiler_config: CompilerConfig = CompilerConfig()
                  ):
         self.code_style_config = code_style_config
         self.postprocess_config = postprocess_config
@@ -101,5 +104,12 @@ def load(cls, fp: TextIO, get_config_type, args: Optional[Namespace] = None, lan
                                                          save_no_session_cache=args.save_no_session_cache,
                                                          skip_existing_problems=args.skip_existing_problems))
             result.etc_config = EtcConfig(**etc_config_dic)
+        if ConfigType.COMPILER in get_config_type:
+            compiler_config_dic = get_config_dic(config_dic, ConfigType.COMPILER)
+            if args:
+                compiler_config_dic = _update_config_dict(compiler_config_dic, dict(
+                                                         compile_only_when_diff_detected=args.compile_only_when_diff_detected,
+                                                         compile_command=args.compile_command))
+            result.compiler_config = CompilerConfig(**compiler_config_dic)
 
         return result
diff --git a/atcodertools/tools/compiler.py b/atcodertools/tools/compiler.py
@@ -1,11 +1,16 @@
 #!/usr/bin/python3
 import argparse
+from os.path import expanduser
 
 from atcodertools.executils.run_command import run_command_with_returncode
 from atcodertools.tools.models.metadata import Metadata
 from atcodertools.common.language import Language
 import os
 import pathlib
+from atcodertools.config.config import Config, ConfigType
+from atcodertools.tools import get_default_config_path
+
+USER_CONFIG_PATH = os.path.join(expanduser("~"), ".atcodertools.toml")
 
 
 class BadStatusCodeException(Exception):
@@ -60,9 +65,36 @@ def main(prog, args):
                         type=bool,
                         default=False)
 
+    parser.add_argument("--config",
+                        help="File path to your config file\n{0}{1}".format("[Default (Primary)] {}\n".format(
+                            USER_CONFIG_PATH),
+                            "[Default (Secondary)] {}\n".format(
+                                get_default_config_path())),
+                        default=None)
+
     args = parser.parse_args(args)
+    if args.config is None:
+        if os.path.exists(USER_CONFIG_PATH):
+            args.config = USER_CONFIG_PATH
+        else:
+            args.config = get_default_config_path()
 
     metadata = Metadata.load_from("./metadata.json")
-    force_compile = not args.compile_only_when_diff_detected
+    lang = metadata.lang
+
+    with open(args.config, "r") as f:
+        config = Config.load(f, {ConfigType.COMPILER}, args, lang.name)
+
+    if args.compile_only_when_diff_detected:
+        force_compile = not args.compile_only_when_diff_detected
+    else:
+        force_compile = not config.compiler_config.compile_only_when_diff_detected
+
+    if args.compile_command:
+        compile_command = args.compile_command
+    else:
+        compile_command = config.compiler_config.compile_command
+    if compile_command:
+        compile_command = lang.get_compile_command("main", compile_command)
     compile_main_and_judge_programs(metadata.lang, force_compile=force_compile,
-                                    compile_command=args.compile_command)
+                                    compile_command=compile_command)
