import sys
import time
import base64
import random


def generate_str(length):
    string = ""
    chars = "AaBbCcDdEeFfGgHhIiJjKkLlMmNnOoPpQqRrSsTtUuVvWwXxYyZz0123456789"
    chars_len = len(chars) - 1
    for i in range(length):
        string += chars[random.randint(0, chars_len)]
    return string


env_type_list = ("dev", "staging", "prod")

if __name__ == "__main__":
    if len(sys.argv) != 2:
        print("Usage: script service_name")
        print("Example: ./main.py aipower")
        exit(0)
        pass
    service_name = sys.argv[1]
    data_dict = {}
    cur_time = str(time.time())
    for env_type in env_type_list:
        final_str = (
            service_name + ":" + env_type + ":" + cur_time + ":" + generate_str(8)
        )
        print("origin key: " + final_str)
        encoded_str = base64.b64encode(final_str.encode("utf-8"))
        print("encoded key: " + encoded_str)
        decoded_str = base64.b64decode(encoded_str)
        assert final_str == decoded_str
        data_dict[env_type] = encoded_str
        pass
    print("\n\n=================Key Info ==============\n\n")
    for k, v in data_dict.items():
        print(k + ":" + v)