import argparse
import json
import OnePakRequests as opr
import re

with open("config.json") as configFile:
    config = json.load(configFile)


parser = argparse.ArgumentParser(description='Use the OnePak API.')

parser.add_argument("--search", action=argparse.BooleanOptionalAction)
parser.add_argument("--cancel", action=argparse.BooleanOptionalAction)
parser.add_argument("--lease", type=str, required=True)

args = parser.parse_args()

assert re.match("\d{3}-\d{7}-\d{3}", args.lease) != None , f"Invalid lease number: '{args.lease}'"

if args.search:
    searchResult = opr.retrieve_return_data(args.lease, config["APIKey"])
    print(searchResult)
elif args.cancel:
    cancelResult = opr.search_and_cancel(args.lease, config["APIKey"])
    print(cancelResult)