org.jetbrains.kotlin.config.CompilerConfiguration Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of kotlin-compiler-embeddable Show documentation
the Kotlin compiler embeddable
There is a newer version: 2.1.0-RC
Show newest version
/*
 * Copyright 2010-2015 JetBrains s.r.o.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.jetbrains.kotlin.config;

import com.intellij.openapi.util.Key;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import java.util.*;

@SuppressWarnings("unchecked")
public class CompilerConfiguration {
    public static CompilerConfiguration EMPTY = new CompilerConfiguration();

    private final Map map = new LinkedHashMap<>();
    private boolean readOnly = false;

    static {
        EMPTY.setReadOnly(true);
    }

    @Nullable
    public  T get(@NotNull CompilerConfigurationKey key) {
        T data = (T) map.get(key.ideaKey);
        return data == null ? null : unmodifiable(data);
    }

    @NotNull
    public  T get(@NotNull CompilerConfigurationKey key, @NotNull T defaultValue) {
        T data = get(key);
        return data == null ? defaultValue : data;
    }

    @NotNull
    public  T getNotNull(@NotNull CompilerConfigurationKey key) {
        T data = get(key);
        assert data != null : "No value for configuration key: " + key;
        return data;
    }

    public boolean getBoolean(@NotNull CompilerConfigurationKey key) {
        return get(key, false);
    }

    @NotNull
    public  List getList(@NotNull CompilerConfigurationKey> key) {
        List data = get(key);
        return data == null ? Collections.emptyList() : data;
    }

    @NotNull
    public  Map getMap(@NotNull CompilerConfigurationKey> key) {
        Map data = get(key);
        return data == null ? Collections.emptyMap() : data;
    }

    public  void put(@NotNull CompilerConfigurationKey key, @NotNull T value) {
        checkReadOnly();
        map.put(key.ideaKey, value);
    }

    public  T putIfAbsent(@NotNull CompilerConfigurationKey key, @NotNull T value) {
        T data = get(key);
        if (data != null) return data;

        checkReadOnly();
        put(key, value);
        return value;
    }

    public  void putIfNotNull(@NotNull CompilerConfigurationKey key, @Nullable T value) {
        if (value != null) {
            put(key, value);
        }
    }

    public  void add(@NotNull CompilerConfigurationKey> key, @NotNull T value) {
        checkReadOnly();
        Key> ideaKey = key.ideaKey;
        map.computeIfAbsent(ideaKey, k -> new ArrayList());
        List list = (List) map.get(ideaKey);
        list.add(value);
    }

    public  void put(@NotNull CompilerConfigurationKey> configurationKey, @NotNull K key, @NotNull V value) {
        checkReadOnly();
        Key> ideaKey = configurationKey.ideaKey;
        map.computeIfAbsent(ideaKey, k -> new HashMap());
        Map data = (Map) map.get(ideaKey);
        data.put(key, value);
    }

    public  void addAll(@NotNull CompilerConfigurationKey> key, @Nullable Collection values) {
        if (values != null) {
            addAll(key, getList(key).size(), values);
        }
    }

    public  void addAll(@NotNull CompilerConfigurationKey> key, int index, @NotNull Collection values) {
        checkReadOnly();
        checkForNullElements(values);
        Key> ideaKey = key.ideaKey;
        map.computeIfAbsent(ideaKey, k -> new ArrayList());
        List list = (List) map.get(ideaKey);
        list.addAll(index, values);
    }

    public CompilerConfiguration copy() {
        CompilerConfiguration copy = new CompilerConfiguration();
        copy.map.putAll(map);
        return copy;
    }

    private void checkReadOnly() {
        if (readOnly) {
            throw new IllegalStateException("CompilerConfiguration is read-only");
        }
    }

    public void setReadOnly(boolean readOnly) {
        if (readOnly != this.readOnly) {
            this.readOnly = readOnly;
        }
    }

    public boolean isReadOnly() {
        return readOnly;
    }

    @NotNull
    private static  T unmodifiable(@NotNull T object) {
        if (object instanceof List) {
            return (T) Collections.unmodifiableList((List) object);
        }
        else if (object instanceof Map) {
            return (T) Collections.unmodifiableMap((Map) object);
        }
        else if (object instanceof Set) {
            return (T) Collections.unmodifiableSet((Set) object);
        }
        else if (object instanceof Collection) {
            return (T) Collections.unmodifiableCollection((Collection) object);
        }
        else {
            return object;
        }
    }

    @Override
    public String toString() {
        return map.toString();
    }

    private static  void checkForNullElements(Collection values) {
        int index = 0;
        for (T value : values) {
            if (value == null) {
                throw new IllegalArgumentException("Element " + index
                                                   + " is null, while null values in compiler configuration are not allowed");
            }
            index++;
        }
    }
}