#!/usr/bin/python3
import argparse
import subprocess
import textwrap

def act(cmd, dry_run=True, verbose=True):
    if dry_run:
        print(f"will execute: {cmd}")
        return
    if verbose:
        print(cmd)
    subprocess.run(cmd.split())

parser = argparse.ArgumentParser(
                    prog='leif',
                    formatter_class=argparse.RawTextHelpFormatter,
                    description=textwrap.dedent('''
                    Verktygslåda för att hantera och monitorera API Gateway-tjänstens systemd-tjänst och container.
                    '''),
                    epilog=textwrap.dedent('''
                    Kontakt: Integration, Team Flex <datautbyte@trafikverket.se>

                    För dokumentation, se wiki:
                    - wiki.trafikverket.local/IKC
                    '''))
parser.add_argument('-v', '--verbose',
                    help="Utökad utskrift vid körning",
                    action='store_true')
parser.add_argument('-d', '--dry-run',
                    help="Simulera åtgärd istället",
                    action='store_true')
subparsers = parser.add_subparsers(dest="action")

subparsers.add_parser("inspect", help="inspektera den körande containern")
logs_parser = subparsers.add_parser("logs", help="skriv ut containerns logg")
logs_parser.add_argument("-f", "--follow", action="store_true", help="svansa och följ loggen")
subparsers.add_parser("restart", help="starta om systemd-tjänsten")
subparsers.add_parser("shell", help="initiera ett bash-skal inuti den körande containern")
subparsers.add_parser("start", help="starta systemd-tjänsten")
subparsers.add_parser("status", help="skriv ut container-ID och upptid för den körande containern")
subparsers.add_parser("stop", help="stoppa systemd-tjänsten")
subparsers.add_parser("whoami", help="skriv ut containeravbildens ProGet-ID")


if __name__ == "__main__":
    args = parser.parse_args()
    match args.action:
        case "inspect":
            act("podman inspect api_gateway", args.dry_run, args.verbose)
        case "restart":
            act("systemcl --user restart container-api-gateway", args.dry_run, args.verbose)
        case "shell":
            act("podman exec -it api_gateway /bin/bash", args.dry_run, args.verbose)
        case "start":
            act("systemcl --user start container-api-gateway", args.dry_run, args.verbose)
        case "status":
            act("podman ps --format '{{.ID}}'", args.dry_run, args.verbose)
        case "stop":
            act("systemcl --user stop container-api-gateway", args.dry_run, args.verbose)
        case "whoami":
            act("podman inspect api_gateway | grep Image", args.dry_run, args.verbose)
        case "logs":
            flags = "-f --tail 1 " if args.follow else ""
            act(f"podman logs {flags}api_gateway", args.dry_run, args.verbose)