write_files.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
"Dimidium" color scheme for terminals
https://github.com/dofuuz/dimidium

SPDX-FileCopyrightText: (c) 2024 Myungchul Keum
SPDX-License-Identifier: Zlib

Write terminal color settings and preview from palette.
"""

from collections import OrderedDict
import json

import numpy as np

from dimidium import generate_colors, get_colors_from_tsv


rgbs = generate_colors(9)
# rgbs = get_colors_from_tsv(9).astype(np.uint8)


names = [
    'Background',
    'Black',
    'Red',
    'Green',
    'Yellow',
    'Blue',
    'Magenta',
    'Cyan',
    'White',
    'Bright Black',
    'Bright Red',
    'Bright Green',
    'Bright Yellow',
    'Bright Blue',
    'Bright Magenta',
    'Bright Cyan',
    'Bright White',
]

print('<table style="background: #000">')
for ix, (r, g, b) in enumerate(rgbs):
    hexa = f'#{r:02X}{g:02X}{b:02X}'
    rgbd = f'{r},{g},{b}'
    print(f'<tr style="color:{hexa}"><td style="background:{hexa}">　</td><td>{names[ix]}</td><td>{rgbd}</td><td>{hexa}</td></tr>')
print('</table>')

for r, g, b in rgbs:
    print(f'{r},{g},{b} | #{r:02X}{g:02X}{b:02X}')

# Write putty.reg
REG_HEADER = '''Windows Registry Editor Version 5.00

[HKEY_CURRENT_USER\Software\SimonTatham\PuTTY\Sessions\Default%20Settings]
'''
with open('config/dimidium-putty.reg', 'wt') as f:
    f.write(REG_HEADER)
    for idx, rdx in enumerate([8, 16, 0, 9, 0, 11, 1, 9, 2, 10, 3, 11, 4, 12, 5, 13, 6, 14, 7, 15, 8, 16]):
        r, g, b = rgbs[rdx]
        print(f'"Colour{idx}"="{r},{g},{b}"', file=f)

# Write mintty
mintty = OrderedDict()
mintty['ForegroundColour'] = rgbs[8]
mintty['BackgroundColour'] = rgbs[0]
mintty['CursorColour'] = rgbs[11]
mintty['Black'] = rgbs[1]
mintty['BoldBlack'] = rgbs[9]
mintty['Red'] = rgbs[2]
mintty['BoldRed'] = rgbs[10]
mintty['Green'] = rgbs[3]
mintty['BoldGreen'] = rgbs[11]
mintty['Yellow'] = rgbs[4]
mintty['BoldYellow'] = rgbs[12]
mintty['Blue'] = rgbs[5]
mintty['BoldBlue'] = rgbs[13]
mintty['Magenta'] = rgbs[6]
mintty['BoldMagenta'] = rgbs[14]
mintty['Cyan'] = rgbs[7]
mintty['BoldCyan'] = rgbs[15]
mintty['White'] = rgbs[8]
mintty['BoldWhite'] = rgbs[16]

with open('config/dimidium-mintty', 'wt') as f:
    for key, (r, g, b) in mintty.items():
        print(f'{key} = {r},{g},{b}', file=f)

# Print Windows Terminal JSON
WINTERM_KEYS = [
    'background',
    'black', 'red', 'green', 'yellow', 'blue', 'purple', 'cyan', 'white',
    'brightBlack', 'brightRed', 'brightGreen', 'brightYellow', 'brightBlue', 'brightPurple', 'brightCyan', 'brightWhite',
]

winterm_dict = {'name': 'Dimidium', 'selectionBackground': '#FFFFFF'}
for idx, key in enumerate(WINTERM_KEYS):
    r, g, b = rgbs[idx]
    winterm_dict[key] = f'#{r:02X}{g:02X}{b:02X}'

winterm_dict['foreground'] = winterm_dict['white']
winterm_dict['cursorColor'] = winterm_dict['brightGreen']

with open('config/dimidium-windowsterminal.json', 'w') as jf:
    json.dump({"schemes": [winterm_dict]}, jf, indent=4)


# Generate preveiw
mintty['BoldBoldGreen'] = mintty['BoldGreen']
mintty['BoldBackgroundColour'] = mintty['BackgroundColour']

m = {}
h = OrderedDict()
for key, rgb in mintty.items():
    r, g, b = rgb
    h[key] = f'{r:02x}{g:02x}{b:02x}'
    m[key] = f'{r}, {g}, {b}'

aha_to_ansi = [['bg', 'BackgroundColour'], ['black', 'Black'], ['dimgray', 'Black'], ['red', 'Red'], ['lime', 'Green'], ['#55FF55', 'BoldGreen'], ['yellow', 'Yellow'], ['#3333FF', 'Blue'], ['fuchsia', 'Magenta'], ['aqua', 'Cyan'], ['white', 'White']]

d = OrderedDict()
d['"color:white; background-color:black"'] = '"color:#{}; background-color:#{}"'.format(h['ForegroundColour'], h['BackgroundColour'])
d['"color:black; background-color:white"'] = '"color:#{}; background-color:#{}"'.format(h['BackgroundColour'], h['ForegroundColour'])
d.update({f'background-color:{k};': f'background-color:#{h[v]};' for k, v in aha_to_ansi})
d.update({f'font-weight:bold;color:{k};': f'font-weight:bold;color:#{h["Bold"+v]};' for k, v in aha_to_ansi})
d.update({f'color:{k};': f'color:#{h[v]};' for k, v in aha_to_ansi})
d['"font-weight:bold;"'] = '"font-weight:bold;color:#{};"'.format(h['BoldWhite'])
d['"font-weight:bold;background-color:'] = '"font-weight:bold;color:#{};background-color:'.format(h['BoldWhite'])
d.update({f' xb.{k} ': f' #{h["Bold"+v]} ' for k, v in aha_to_ansi})
d.update({f' x.{k} ': f' #{h[v]} ' for k, v in aha_to_ansi})
d.update({f' rgb.b.{k} ': f' {m["Bold"+v]:13s} ' for k, v in aha_to_ansi})
d.update({f' rgb.{k} ': f' {m[v]:13s} ' for k, v in aha_to_ansi})

with open('recipe/tty-template.html') as f:
    html = f.read()

for k, v in d.items():
    html = html.replace(k, v)

with open('preview/tty-preview.html', 'wt') as f:
    f.write(html)

html = html.replace('font-weight:bold;', '')
with open('preview/tty-preview-nobold.html', 'wt') as f:
    f.write(html)