#!/usr/bin/env python3

import argparse
import subprocess

LOGO_PATH = "/usr/share/aosc-logos/"
CODENAMES = ['fsck',
             'gumblex',
             'hotfix',
             'icenowy',
             'jinkela',
             'koshki',
             'localhost']
CURRENT = CODENAMES[len(CODENAMES) - 1]

parser = argparse.ArgumentParser(description='Display AOSC OS logo ASCII arts.')
parser.add_argument('--codename', '-c', type=str, help='Select a specific version of AOSC OS logo', nargs='?', default=CURRENT)

args = parser.parse_args()

for code in CODENAMES:
    if args.codename == code or args.codename == code[:1]:
        subprocess.run(LOGO_PATH + code)
        exit()

subprocess.run(LOGO_PATH + CURRENT)
