x2term.py

#!/usr/bin/env python3

import argparse
import json
import sys
import re
from typing import Dict, List, Tuple


def parse_args():
    parser = argparse.ArgumentParser(description="")
    parser.add_argument(
        "--infile", default=sys.stdin, type=argparse.FileType("r")
    )
    parser.add_argument(
        "--outfile", default=sys.stdout, type=argparse.FileType("w")
    )
    return parser.parse_args()


def convert(infile, outfile) -> None:
    colormap: Dict[str, str] = {}
    colors: List[Tuple[str, str]] = []
    for line in infile:
        line = re.sub(r"\s+", " ", line.strip())
        if line.startswith("#define"):
            # definition
            _, name, color, *r = line.split(" ")
            colormap[name] = color
        if line.startswith("*.color"):
            # assignment
            color, ref = line.split(" ")
            color = color[2:-1]
            colors.append((color, colormap.get(ref, ref)))

    # zipped = [a for b in zip(colors[:7], colors[8:]) for a in b]
    palette = ":".join(c for _, c in colors)
    print(" | ".join(f"{a}={b}" for a, b in colors))
    outfile.write(f"palette = \"{palette}\"")


if __name__ == "__main__":
    params = parse_args()
    convert(params.infile, params.outfile)