ea_skin.py

from ea_UI import QtGui, QtWidgets, QtCore, Name_UI, Reskin_UI
from ea_utils import config, root_dir, save_config, set_style
from idaapi import *


def color_selected(i, color):
    back = "{:04x}".format(color.rgb())[2:]
    buttons[i][0].setStyleSheet("background: " + "#" + back)
    buttons[i][2] = back


def select_color(i):

    if not (i == 9 and config["match_background"]):
        global color
        color = QtWidgets.QColorDialog()
        # color.setCustomColor(0, 0x212121)
        color.setCurrentColor(QtGui.QColor.fromRgb(int(buttons[i][0].styleSheet().split("#")[1], 16)))
        color.colorSelected.connect(lambda color: color_selected(i, color))
        color.open()


def save_preset():

    global name_ui
    name_ui = QtWidgets.QFrame()
    form_2 = Name_UI()
    form_2.setupUi(name_ui)
    name_ui.show()
    form_2.pushButton.clicked.connect(lambda: save_preset_2(form_2, name_ui))


def save_preset_2(form_2, name_ui):

    name = form_2.lineEdit.text()
    form.comboBox.addItem(name)
    config["skins"].append([name] + [item[2] for item in buttons])
    save_config()
    form.comboBox.setCurrentIndex(len(config["skins"]) - 1)
    name_ui.close()


def load_preset(i):

    if i == 0:
        colors = config["current_skin"]
    else:
        colors = config["skins"][i - 1][1:]

    for i, v in enumerate(colors):
        if i == 9 and config["match_background"]:
            buttons[i][0].setStyleSheet("background: " + "#" + colors[0])
        else:
            buttons[i][0].setStyleSheet("background: " + "#" + v)
        buttons[i][2] = v


def apply_skin(init = False):

    with open(root_dir + "style_template.css", "r") as r:
        style = r.read()

    skin_values = config["current_skin"][:9] if init else (item[2] for item in buttons)

    for i, c in enumerate(item for item in skin_values):
    	style = style.replace("{%s}" % (i), "#" + c)

    for group in ("Disassembly", "Hex View", "Text input"):

        try: 
            s = QtCore.QSettings()
            s.beginGroup("Font")
            s.beginGroup(group)
            font_name = s.value("Name")
            font_size = s.value("Size")
            style = style.replace("{%s %s}" % (group, "Name"), font_name)
            style = style.replace("{%s %s}" % (group, "Size"), str(font_size))
        except:
            pass



    if not init:
        config["current_skin"] = [item[2] for item in buttons]
        save_config()

    set_style()

    QtWidgets.qApp.setStyleSheet(QtWidgets.qApp.styleSheet().split("/*IDA EA START*/")[0] + style)


def toggle_apply_on_startup(state):
    config["apply_skin_on_startup"] = True if state else False


def toggle_match_background(state):
    config["match_background"] = True if state else False
    save_config()

    if state:
        buttons[9][0].setStyleSheet("background: #" + buttons[0][0].styleSheet().split("#")[1])
    else:
        buttons[9][0].setStyleSheet("background: #" + config["current_skin"][9])


def ea_reskin():

    global a
    global form
    global buttons

    a = QtWidgets.QFrame()
    form = Reskin_UI()
    form.setupUi(a)
    a.show()

    form.comboBox.addItem("<Current skin>")
    form.comboBox.activated.connect(load_preset)

    buttons = [
        [form.pushButton, form.pushButton_2,     None],
        [form.pushButton_23, form.pushButton_24, None],
        [form.pushButton_25, form.pushButton_26, None],
        [form.pushButton_9, form.pushButton_10,  None],
        [form.pushButton_13, form.pushButton_14, None],
        [form.pushButton_27, form.pushButton_28, None],
        [form.pushButton_21, form.pushButton_22, None],
        [form.pushButton_15, form.pushButton_16, None],
        [form.pushButton_11, form.pushButton_12, None],

        [form.pushButton_43, form.pushButton_44, None],
        [form.pushButton_3, form.pushButton_4, None],
        [form.pushButton_31, form.pushButton_32, None],
        [form.pushButton_29, form.pushButton_30, None],
        [form.pushButton_33, form.pushButton_34, None],
    ]

    for x in range(len(buttons)):
        buttons[x][0].clicked.connect(lambda x=x: select_color(x))
        buttons[x][1].clicked.connect(lambda x=x: select_color(x))

    form.pushButton_18.clicked.connect(save_preset)
    form.pushButton_17.clicked.connect(lambda: process_ui_action("SetColors"))
    form.pushButton_20.clicked.connect(apply_skin)
    form.checkBox.stateChanged.connect(lambda x: toggle_apply_on_startup(x))
    form.checkBox_2.stateChanged.connect(lambda x: toggle_match_background(x))
    load_preset(0)

    for i in config["skins"]:
        form.comboBox.addItem(i[0])

    if config["apply_skin_on_startup"]:
        form.checkBox.setChecked(True)

    if config["match_background"]:
        form.checkBox_2.setChecked(True)
        buttons[9][0].setStyleSheet("background: #" + buttons[0][0].styleSheet().split("#")[1])

a = None
form = None
buttons = None
name_ui = None
color = None