Skip to content

Commit

Permalink
feat: added command-line options: no-init, version, help (#26)
Browse files Browse the repository at this point in the history
  • Loading branch information
idoavrah authored Aug 9, 2023
1 parent aa8930b commit 2e5fc4a
Showing 1 changed file with 41 additions and 8 deletions.
49 changes: 41 additions & 8 deletions src/tftui/__main__.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
import asyncio
import argparse
from textual import work
from textual.app import App, ComposeResult, Binding
from textual.widgets import (
Expand All @@ -12,7 +14,9 @@
)
from textual.containers import Vertical, Horizontal
from shutil import which
import asyncio
import importlib.metadata

global_no_init = False


class StateTree(Tree):
Expand Down Expand Up @@ -60,12 +64,13 @@ async def refresh_state(self) -> None:
self.selected_nodes = []
self.current_node = None
self.clear()
self.app.status.update("Executing Terraform init")

returncode, stdout = await execute_async("terraform", "init", "-no-color")
if returncode != 0:
self.app.exit(message=stdout)
return
if not global_no_init:
self.app.status.update("Executing Terraform init")
returncode, stdout = await execute_async("terraform", "init", "-no-color")
if returncode != 0:
self.app.exit(message=stdout)
return

self.app.status.update("Executing Terraform show")

Expand Down Expand Up @@ -311,10 +316,38 @@ async def execute_async(*command: str) -> tuple[str, str]:
return (proc.returncode, response)


def main() -> None:
def parse_command_line() -> None:
global global_no_init

parser = argparse.ArgumentParser(
prog="tftui", description="TFTUI - the terraform terminal UI", epilog="Enjoy!"
)
parser.add_argument(
"-n",
"--no-init",
help="do not run terraform init on startup",
action="store_true",
)
parser.add_argument(
"-v", "--version", help="show version information", action="store_true"
)

args = parser.parse_args()

if args.version:
version = importlib.metadata.version("tftui")
print(f"tftui v{version}")
exit(0)
if args.no_init:
global_no_init = True

if which("terraform") is None and which("terraform.exe") is None:
print("Terraform not found. Please install Terraform and try again.")
return
exit(1)


def main() -> None:
parse_command_line()

app = TerraformTUI()
result = app.run()
Expand Down

0 comments on commit 2e5fc4a

Please sign in to comment.