using debug var instead of msg

ansible-vault-tools.py

@@ -1,98 +1,261 @@
 #!/usr/bin/env python3
 """Encrypt or decrypt using Ansible-vault and Ansible"""
-
 # SPDX-FileCopyrightText: 2023 Max Mehl <https://mehl.mx>
 #
 # SPDX-License-Identifier: Apache-2.0
 
-import subprocess
-import json
+# pylint: disable=invalid-name
+
 import argparse
+import json
+import os
+import subprocess
+import sys
 
 parser = argparse.ArgumentParser(description=__doc__)
 subparsers = parser.add_subparsers(title="commands", dest="command", required=True)
 # Encrypt arguments
 parser_encrypt = subparsers.add_parser(
     "encrypt",
-    help="Encrypt a string using ansible-vault",
+    help="Encrypt a string or file using ansible-vault",
 )
-parser_encrypt.add_argument(
+encrypt_flags = parser_encrypt.add_mutually_exclusive_group(required=True)
+encrypt_flags.add_argument(
     "-s",
     "--string",
     help="String that shall be encrypted",
     dest="encrypt_string",
 )
+encrypt_flags.add_argument(
+    "-f",
+    "--file",
+    help="File that shall be encrypted",
+    dest="encrypt_file",
+)
 # Decrypt arguments
 parser_decrypt = subparsers.add_parser(
     "decrypt",
-    help="Print a variable of one or multiple hosts and decrypt it if necessary",
+    help="Decrypt a string or file using ansible-vault",
 )
-parser_decrypt.add_argument(
+decrypt_flags = parser_decrypt.add_mutually_exclusive_group(required=True)
+decrypt_flags.add_argument(
     "-H",
     "--host",
-    help="Host name from Ansible inventory for which you want to get the variable",
+    help=(
+        "Host name from Ansible inventory for which you want to get a specific variable."
+        "Also supports 'all'"
+    ),
     dest="decrypt_host",
 )
+decrypt_flags.add_argument(
+    "-f",
+    "--file",
+    help="File that shall be decrypted",
+    dest="decrypt_file",
+)
 parser_decrypt.add_argument(
     "-v",
     "--var",
     help="Variable you want to print",
     dest="decrypt_var",
 )
+# All variables
+parser_allvars = subparsers.add_parser(
+    "allvars",
+    help="Print all variables of a host",
+)
+parser_allvars.add_argument(
+    "-H",
+    "--host",
+    help=(
+        "Host name from Ansible inventory for which you want to get all variables."
+        "Also supports 'all'"
+    ),
+    dest="allvars_host",
+)
 
 
-def encrypt(password):
+def convert_ansible_errors(error: str) -> str:
+    """Convert typical Ansible errors to more user-friendly messages"""
+    if "VARIABLE IS NOT DEFINED" in error:
+        return "(undefined variable)"
+
+    # If no conversion was possible, return the original error
+    return error
+
+
+def ask_for_confirm(question: str) -> bool:
+    """Ask for confirmation.
+
+    Args:
+        question (str): The question to ask the user.
+
+    Returns:
+        bool: True if the user confirms with 'y', False otherwise.
+    """
+    while True:
+        answer = input(f"{question} [y/n]: ").lower().strip()
+        if answer in ("y", "n"):
+            break
+        print("Invalid input. Please enter 'y' or 'n'.")
+
+    return answer == "y"
+
+
+def encrypt_string(password: str) -> str:
     """Encrypt string with ansible-vault"""
     result = subprocess.run(
         ["ansible-vault", "encrypt_string"],
         input=password,
         text=True,
         capture_output=True,
+        check=False,
     )
     return result.stdout.strip()
 
 
-def decrypt(host, var):
-    """Decrypt/print a variable from one or multiple hosts"""
-    # Run ansible msg for variable
+def format_data(data: dict) -> str:
+    """Format data nicely in columns"""
+    if len(data) > 1:
+        max_key_length = max(len(key) for key in data.keys())
+
+        formatted_strings = [f"{key.ljust(max_key_length)}: {value}" for key, value in data.items()]
+    else:
+        # If only one host, return the single value
+        formatted_strings = list(data.values())
+
+    return "\n".join(formatted_strings)
+
+
+def decrypt_file(filename: str) -> str:
+    """Decrypt file with ansible-vault"""
+
+    if not os.path.exists(filename):
+        sys.exit(f"ERROR: File '{filename}' does not exist")
+
+    decrypted_content = subprocess.run(
+        ["ansible-vault", "decrypt", "--output", "-", filename], check=False, capture_output=True
+    )
+
+    if decrypted_content.returncode != 0:
+        sys.exit(
+            f"ERROR: Could not decrypt file '{filename}'. This is the error:"
+            f"\n{decrypted_content.stderr.decode()}"
+        )
+
+    print(decrypted_content.stdout.decode().strip())
+    if ask_for_confirm("Shall I write the encrypted content as seen above to the file?"):
+        decrypted_content = subprocess.run(
+            ["ansible-vault", "decrypt", filename], check=True, capture_output=True
+        )
+        return f"Decrypted '{filename}' successfully"
+
+    return f"File '{filename}' was not changed"
+
+
+def allvars(host: str) -> str:
+    """Decrypt/print all variables from one or multiple hosts"""
+    # Run ansible var for all host vars as seen from localhost
     # Send return as JSON
-    ansible_command = ["ansible", host, "-m", "debug", "-a", f"msg={{{{ {var} }}}}"]
+    ansible_command = ["ansible", "localhost", "-m", "debug", "-a", "var=hostvars"]
     ansible_env = {
         "ANSIBLE_LOAD_CALLBACK_PLUGINS": "1",
         "ANSIBLE_STDOUT_CALLBACK": "json",
     }
     result = subprocess.run(
-        ansible_command, env=ansible_env, capture_output=True, text=True
+        ansible_command, env=ansible_env, capture_output=True, text=True, check=False
     )
 
-    # Parse JSON to just get the "msg"
-    ansible_output = json.loads(result.stdout)
-    msg = [
-        host["msg"]
-        for play in ansible_output["plays"]
-        for task in play["tasks"]
-        for host in task["hosts"].values()
+    # Reduce JSON
+    ansible_output = json.loads(result.stdout)["plays"][0]["tasks"][0]["hosts"]["localhost"][
+        "hostvars"
     ]
 
-    # Pretty print the JSON
-    return json.dumps(msg, indent=2)
+    # If only a specific host was requested, reduce the output to that host
+    if host != "all":
+        try:
+            ansible_output = ansible_output[host]
+        except KeyError:
+            sys.exit(f"ERROR: Host '{host}' not found.")
+
+    return json.dumps(ansible_output, indent=2)
+
+
+def encrypt_file(filename: str) -> str:
+    """Encrypt a file with ansible-vault"""
+
+    if not os.path.exists(filename):
+        sys.exit(f"ERROR: File '{filename}' does not exist")
+
+    encrypted_return = subprocess.run(
+        ["ansible-vault", "encrypt", filename], check=False, capture_output=True
+    )
+
+    if encrypted_return.returncode != 0:
+        sys.exit(
+            f"ERROR: Could not encrypt file '{filename}'. This is the error:"
+            f"\n{encrypted_return.stderr.decode()}"
+        )
+
+    return f"Encrypted '{filename}' successfully"
+
+
+def decrypt_string(host, var) -> str:
+    """Decrypt/print a variable from one or multiple hosts"""
+    # Run ansible msg for variable
+    # Send return as JSON
+    ansible_command = ["ansible", host, "-m", "debug", "-a", f"var={var}"]
+    ansible_env = {
+        "ANSIBLE_LOAD_CALLBACK_PLUGINS": "1",
+        "ANSIBLE_STDOUT_CALLBACK": "json",
+    }
+    result = subprocess.run(
+        ansible_command, env=ansible_env, capture_output=True, text=True, check=False
+    )
+
+    # Parse JSON
+    try:
+        ansible_output = json.loads(result.stdout)["plays"][0]["tasks"][0]["hosts"]
+    except IndexError:
+        sys.exit(f"ERROR: Host '{host}' not found.")
+
+    # Attempt to create a :-separated list of host/values
+    output = {}
+    for hostname, values in ansible_output.items():
+        output[hostname] = convert_ansible_errors(values[var])
+
+    return format_data(output)
 
 
 def main():
     """Main function"""
     args = parser.parse_args()
+    output = ""
 
+    # ENCRYPTION
     if args.command == "encrypt":
-        password = (
-            input("Enter string: ") if not args.encrypt_string else args.encrypt_string
-        )
-        vaultpw = encrypt(password)
+        if args.encrypt_string:
+            password = input("Enter string: ") if not args.encrypt_string else args.encrypt_string
+            output = encrypt_string(password)
+        elif args.encrypt_file:
+            filename = input("Enter filename: ") if not args.encrypt_file else args.encrypt_file
+            output = encrypt_file(filename)
+    # DECRYPTION
     elif args.command == "decrypt":
-        host = input("Enter host: ") if not args.decrypt_host else args.decrypt_host
-        var = input("Enter variable: ") if not args.decrypt_var else args.decrypt_var
-        vaultpw = decrypt(host, var)
+        if args.decrypt_host:
+            host = input("Enter host: ") if not args.decrypt_host else args.decrypt_host
+            var = input("Enter variable: ") if not args.decrypt_var else args.decrypt_var
+            output = decrypt_string(host, var)
+        elif args.decrypt_file:
+            filename = input("Enter filename: ") if not args.decrypt_file else args.decrypt_file
+            output = decrypt_file(filename)
+    # ALLVARS
+    elif args.command == "allvars":
+        host = input("Enter host: ") if not args.allvars_host else args.allvars_host
+        output = allvars(host)
 
-    print(vaultpw)
+    if output:
+        print(output)
 
 
 if __name__ == "__main__":