HALF WORK: 2024.1.8

1 files changed, 47 insertions, 7 deletions

diff --git a/tools/cru-py/cru/service/__main__.py b/tools/cru-py/cru/service/__main__.py
index a7add4d..923c25b 100644
--- a/tools/cru-py/cru/service/__main__.py
+++ b/tools/cru-py/cru/service/__main__.py
@@ -1,11 +1,51 @@
-import argparse
+from pathlib import Path
 
+from cru import CruException
 
-arg_parser = argparse.ArgumentParser(description="Service management")
-command_subparser = arg_parser.add_subparsers(dest="command")
+from ._base import AppBase, DATA_DIR_NAME, CommandDispatcher
+from ._config import ConfigManager
+from ._data import DataManager
+from ._template import TemplateManager
 
-template_parser = command_subparser.add_parser("template", help="Template management")
-template_subparser = template_parser.add_subparsers(dest="template_command")
 
-template_subparser.add_parser('list', description="List templates")
-template_subparser.add_parser('generate')
+class App(AppBase):
+    def __init__(self, root: str):
+        super().__init__("crupest-service", root)
+        self.add_feature(DataManager())
+        self.add_feature(ConfigManager())
+        self.add_feature(TemplateManager())
+        self.add_feature(CommandDispatcher())
+
+    def setup(self):
+        for feature in self.features:
+            feature.setup()
+
+    def run_command(self):
+        command_dispatcher = self.get_feature(CommandDispatcher)
+        command_dispatcher.run_command()
+
+
+def _find_root() -> Path:
+    cwd = Path.cwd()
+    data_dir = cwd / DATA_DIR_NAME
+    if data_dir.is_dir():
+        return data_dir
+    raise CruException(
+        "No valid data directory found. Please run 'init' to create one."
+    )
+
+
+def create_app() -> App:
+    root = _find_root()
+    app = App(str(root))
+    app.setup()
+    return app
+
+
+def main():
+    app = create_app()
+    app.run_command()
+
+
+if __name__ == "__main__":
+    main()