MageUI.java

package mage.client.components;

import java.awt.Component;
import java.util.EnumMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;
import javax.swing.JButton;

public class MageUI {

    private final Map<MageComponents, Component> ui = new EnumMap<>(MageComponents.class);
    private final Map<MageComponents, Object> sync = new EnumMap<>(MageComponents.class);

    public JButton getButton(MageComponents name) throws InterruptedException {
        //System.out.println("request for " + name);
        Object buttonSync;
        synchronized (ui) {
            if (ui.containsKey(name)) {
                //System.out.println("clicking " + name);
                return (JButton) ui.get(name);
            } else {
                buttonSync = new Object();
                sync.put(name, buttonSync);
            }
        }

        synchronized (buttonSync) {
            //System.out.println("waiting " + name + " to be created");
            buttonSync.wait();
            //System.out.println(name + "was created");
            if (!ui.containsKey(name)) {
                throw new IllegalStateException("Component wasn't initialized. This should not happen.");
            }
            return (JButton) ui.get(name);
        }

    }

    public Component getComponent(MageComponents name) throws InterruptedException {
        Object componentSync;
        synchronized (ui) {
            if (ui.containsKey(name)) {
                return ui.get(name);
            } else {
                componentSync = new Object();
                sync.put(name, componentSync);
            }
        }

        synchronized (componentSync) {
            componentSync.wait();
            if (!ui.containsKey(name)) {
                throw new IllegalStateException("Component wasn't initialized. This should not happen.");
            }
            return ui.get(name);
        }

    }

    public void addButton(MageComponents name, JButton button) {
        synchronized (ui) {
            //System.out.println("added " + name);
            ui.put(name, button);
            if (sync.containsKey(name)) {
                synchronized (sync.get(name)) {
                    //System.out.println("notifyAll - " + name);
                    sync.get(name).notifyAll();
                }
            }
        }
    }

    public void removeButton(MageComponents name) {
        synchronized (ui) {
            ui.remove(name);
        }
    }

    public void addComponent(MageComponents name, Component component) {
        synchronized (ui) {
            ui.put(name, component);
            if (sync.containsKey(name)) {
                synchronized (sync.get(name)) {
                    sync.get(name).notifyAll();
                }
            }
        }
    }

    public void doClick(MageComponents name) throws InterruptedException {
        doClick(name, 0);
    }

    public void doClick(MageComponents name, int waitBeforeClick) throws InterruptedException {
        final JButton j = getButton(name);
        TimeUnit.MILLISECONDS.sleep(waitBeforeClick);
        while (!j.isEnabled()) {
            TimeUnit.MILLISECONDS.sleep(10);
        }
        Thread t = new Thread(() -> j.doClick());
        t.start();
    }
}