129 lines
3.8 KiB
Python
Executable file
129 lines
3.8 KiB
Python
Executable file
#!/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 <caesar.kabalan@gmail.com>
|
|
# 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)
|
|
|
|
def main(xrdb_path, output_path=None):
|
|
|
|
# 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(xrdb_path)):
|
|
name = xrdb_regex.match(i).group(1)
|
|
|
|
# Read XRDB file
|
|
with open(os.path.join(xrdb_path, i)) as f:
|
|
xrdb_data = f.read()
|
|
|
|
# Open output file
|
|
output = sys.stdout
|
|
|
|
if output_path:
|
|
dest = os.path.join(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 output_path:
|
|
output.close()
|
|
|
|
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()
|
|
main(args.xrdb_path, args.output_path)
|
|
|
|
|