import argparse
import sys

from . import image
from . import scheme
from . import term


def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument('--apply', '-a', action='store_true')
    parser.add_argument('--image', '-i')
    parser.add_argument('--256', action='store_true', dest='full')
    parser.add_argument('-n', type=int, default=8)
    parser.add_argument('colors', nargs='*')
    return parser.parse_args()


def main():
    args = parse_args()

    colors = []
    if args.image:
        colors += list(image.extract_colors(args.image, args.n))
    colors += args.colors

    if len(colors) < 6:
        sys.exit('Need at least 6 colors')

    s = scheme.colors2scheme(colors, full_256=args.full)

    print(';'.join(s))
    if args.full:
        term.palette_256(s)
    else:
        term.palette(s)
    if args.apply:
        term.apply(s)


if __name__ == '__main__':
    main()