#59: Add imports() in standardlib

src/rkd/standardlib/__init__.py

@@ -13,10 +13,13 @@
 from ..api.contract import ExecutionContext
 from ..api.contract import TaskDeclarationInterface
 from ..api.contract import ArgparseArgument
+from ..api.syntax import TaskDeclaration
 from ..inputoutput import SystemIO
 from ..inputoutput import clear_formatting
 from ..aliasgroups import parse_alias_groups_from_env, AliasGroup
 
+from .shell import ShellCommandTask
+
 
 class InitTask(TaskInterface):
     """
@@ -590,3 +593,14 @@ def git_add(self, path: str):
 
     def check_git_is_clean(self):
         return self.sh('git diff --stat || true', capture=True).strip() == ''
+
+
+def imports() -> list:
+    return [
+        TaskDeclaration(InitTask()),
+        TaskDeclaration(TasksListingTask()),
+        TaskDeclaration(VersionTask()),
+        TaskDeclaration(ShellCommandTask()),
+        TaskDeclaration(LineInFileTask()),
+        TaskDeclaration(CreateStructureTask())
+    ]

src/rkd/standardlib/env.py

@@ -2,8 +2,8 @@
 import os
 from typing import Optional
 from argparse import ArgumentParser
-from rkd.api.contract import TaskInterface
-from rkd.api.contract import ExecutionContext
+from ..api.syntax import TaskDeclaration
+from ..api.contract import TaskInterface, ExecutionContext
 
 
 class GetEnvTask(TaskInterface):
@@ -89,3 +89,10 @@ def parse_env_name_from_line(line: str) -> Optional[str]:
             return None
 
         return parts[0]
+
+
+def imports() -> list:
+    return [
+        TaskDeclaration(GetEnvTask()),
+        TaskDeclaration(SetEnvTask())
+    ]

src/rkd/standardlib/shell.py

@@ -2,6 +2,7 @@
 from argparse import ArgumentParser
 from subprocess import CalledProcessError
 from typing import Callable
+from ..api.syntax import TaskDeclaration
 from ..api.contract import TaskInterface
 from ..api.contract import ExecutionContext
 
@@ -110,3 +111,10 @@ def execute(self, context: ExecutionContext) -> bool:
             return False
 
         return True
+
+
+def imports() -> list:
+    return [
+        TaskDeclaration(ShellCommandTask()),
+        TaskDeclaration(ExecProcessCommand())
+    ]