#!/usr/bin/env python

from __future__ import absolute_import

import os
import subprocess
import sys
import sqlite3
import argparse
from ansible_toolset import consts as C
from ansible_toolset.utils import *
from ansible_toolset.ansible.auto import AnsibleAdapter
from ansible_toolset.vault import VaultManager
from ansible_toolset.models import init_models

init_models(C.ANSIBLE_TOOLSET_DB)
ansible = AnsibleAdapter(vault_password=get_ansible_vault_password())

parser = argparse.ArgumentParser()
parser.add_argument("action", choices=["open", "close"])
args = parser.parse_args()

vaultManager = VaultManager(ansible=ansible)

if args.action == "open":
    vaultManager.open()
elif args.action == "close":
    vaultManager.close()
