first commit

1 files changed, 15 insertions, 0 deletions

diff --git a/awscli_plugin_passtotp/__init__.py b/awscli_plugin_passtotp/__init__.py
new file mode 100644
index 0000000..aeb8606
--- /dev/null
+++ b/awscli_plugin_passtotp/__init__.py
@@ -0,0 +1,15 @@
+from .prompter import inject_pass_totp_prompter
+from .commands import SessionEnv
+
+
+def awscli_initialize(cli):
+    cli.register(
+        "session-initialized",
+        inject_pass_totp_prompter,
+        unique_id="inject_pass_totp_provider",
+    )
+    cli.register("building-command-table.main", awscli_register_commands)
+
+
+def awscli_register_commands(command_table, session, **kwargs):
+    command_table["session-env"] = SessionEnv(session)