diff --git a/tools/xrdb2putty.py b/tools/xrdb2putty.py new file mode 100755 index 0000000..05f0017 --- /dev/null +++ b/tools/xrdb2putty.py @@ -0,0 +1,125 @@ +#!/usr/bin/env python +# coding: utf-8 +# +# This script converts xrdb (X11) color scheme format to PuTTY registry +# file which creates a new session with proper colors. +# +# Usage: +# xrdb2putty.py path/to/xrdb/files -d path/to/putty/files +# +# Author: Caesar Kabalan +# Adapted from xrdb2konsole by Stéphane Travostino +# Adapted from xrdb2terminator by Xabier Larrakoetxea + +import os +import sys +import re +import argparse + +# Takes #000A0B and returns (0, 10, 11) +def hex_to_rgb(color): + return (int(color[1:3], 16), int(color[3:5], 16), int(color[5:7], 16)) + +def build_putty_color(name, r, g, b): + return "\"%s\"=\"%d,%d,%d\"\n" % (name, r, g, b) + +if __name__ == "__main__": + + parser = argparse.ArgumentParser( + description='Translate X color schemes to termiantor format') + parser.add_argument('xrdb_path', type=str, help='path to xrdb files') + parser.add_argument('-d', '--out-directory', type=str, dest='output_path', + help='path where putty config files will be' + + ' created, if not provided then will be printed') + + args = parser.parse_args() + + + # The regexes to match the colors + color_regex = re.compile("#define +Ansi_(\d+)_Color +(#[A-Fa-f0-9]{6})") + bg_regex = re.compile("#define +Background_Color +(#[A-Fa-f0-9]{6})") + fg_regex = re.compile("#define +Foreground_Color +(#[A-Fa-f0-9]{6})") + bold_regex = re.compile("#define +Bold_Color +(#[A-Fa-f0-9]{6})") + cursor_regex = re.compile("#define +Cursor_Color +(#[A-Fa-f0-9]{6})") + cursor_text_regex = re.compile("#define +Cursor_Text_Color +(#[A-Fa-f0-9]{6})") + + # File regex + xrdb_regex = re.compile("(.+)\.[xX][rR][dD][bB]") + + for i in filter(lambda x: xrdb_regex.match(x), os.listdir(args.xrdb_path)): + name = xrdb_regex.match(i).group(1) + + # Read XRDB file + with open(os.path.join(args.xrdb_path, i)) as f: + xrdb_data = f.read() + + # Open output file + output = sys.stdout + + if args.output_path: + dest = os.path.join(args.output_path, name) + output = open('{0}.reg'.format(dest), 'w+') + else: + output.write('\n%s:\n' % name) + + # Emit header + output.write("Windows Registry Editor Version 5.00\n\n[HKEY_CURRENT_USER\Software\SimonTatham\PuTTY\Sessions\%s]\n" % name) + + # Emit background color + bg_color = hex_to_rgb(bg_regex.search(xrdb_data).group(1)) + output.write(build_putty_color('Colour2', *bg_color)) + output.write(build_putty_color('Colour3', *bg_color)) + + # Emit foreground color + fg_color = hex_to_rgb(fg_regex.search(xrdb_data).group(1)) + output.write(build_putty_color('Colour0', *fg_color)) + + # Emit bold color, if any + match = bold_regex.search(xrdb_data) + if match: + bold_color = hex_to_rgb(match.group(1)) + output.write(build_putty_color('Colour1', *bold_color)) + else: + output.write(build_putty_color('Colour1', *fg_color)) + + # Emit cursor color + cursor_color = hex_to_rgb(cursor_regex.search(xrdb_data).group(1)) + output.write(build_putty_color('Colour5', *cursor_color)) + + # Emit cursor text color + cursor_text_color = hex_to_rgb(cursor_text_regex.search(xrdb_data).group(1)) + output.write(build_putty_color('Colour4', *cursor_text_color)) + + # The ANSI color list from xrdb doesn't match up with the weird order + # that the putty registry entries use. This converts Ansi_0_Color from + # .xrdb files to Colour6 which go into the PuTTY .reg files. + xrdb_to_putty_dict = { + 0 : 6, + 1 : 8, + 2 : 10, + 3 : 12, + 4 : 14, + 5 : 16, + 6 : 18, + 7 : 20, + 8 : 7, + 9 : 9, + 10 : 11, + 11 : 13, + 12 : 15, + 13 : 17, + 14 : 19, + 15 : 21, + } + + # Emit other colors + for match in color_regex.findall(xrdb_data): + color_index = int(match[0]) + color_rgb = hex_to_rgb(match[1]) + + color_name = 'Colour%d' % xrdb_to_putty_dict[color_index] + + output.write(build_putty_color(color_name, *color_rgb)) + + if args.output_path: + output.close()