import org.gradle.internal.logging.text.StyledTextOutput;

public interface ColorMap {
    Color getColourFor(StyledTextOutput.Style style);

    Color getColourFor(Style style);

    Color getStatusBarColor();

    interface Color {
        void on(Ansi ansi);

        void off(Ansi ansi);
    }

class TestColorMap implements ColorMap {
    def Ansi.Attribute statusBarOn = Ansi.Attribute.INTENSITY_BOLD
    def Ansi.Attribute statusBarOff = Ansi.Attribute.RESET

    ColorMap.Color getStatusBarColor() {
        if (statusBarOn == Ansi.Attribute.RESET) {
            return {} as ColorMap.Color
        }
        return [on : { ansi -> ansi.a(statusBarOn) },
                off: { ansi -> ansi.a(statusBarOff) }

            colorSpec += COLOR_DIVIDER + getColorSpecForStyle(styles[i]);
        }
        addDefault(style.name().toLowerCase(Locale.ROOT), colorSpec);
    }

    @Override
    public Color getStatusBarColor() {
        return getColor(STATUS_BAR);
    }

    @Override
    public Color getColourFor(StyledTextOutput.Style style) {
        return getColor(style.name().toLowerCase(Locale.ROOT));
    }