core/src/main/java/de/ids_mannheim/korap/config/KustvaktConfiguration.java

package de.ids_mannheim.korap.config;

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Properties;
import java.util.Set;
import java.util.regex.Pattern;

import de.ids_mannheim.korap.util.KrillProperties;
import de.ids_mannheim.korap.utils.TimeUtils;
import lombok.Getter;
import lombok.Setter;

/**
 * Describes configuration for Kustvakt by importing properties 
 * from kustvakt.conf file and setting default values if they are 
 * not configured.  
 * 
 * MH: if configuration class is extended, loadSubTypes method should be
 * overriden
 * 
 * @author hanl
 * @author margaretha
 */

@Setter
@Getter
public class KustvaktConfiguration {

    public static final Map<String, Object> KUSTVAKT_USER = new HashMap<>();

    private String indexDir;
    private int port;
    // todo: make exclusive so that the containg languages can really
    // only be used then
    private List<String> queryLanguages;

    private String serverHost;

    private int maxhits;
    private int returnhits;
    private String keystoreLocation;
    private String keystorePassword;
    private Properties mailProperties;
    private String host;
    private String shibUserMapping;
    private String userConfig;
    private int inactiveTime;
    private int loginAttemptTTL;
    private long loginAttemptNum;
    private boolean allowMultiLogIn;
    private int loadFactor;
    @Deprecated
    private int validationStringLength;
    @Deprecated
    private int validationEmaillength;

    private byte[] sharedSecret;
    @Deprecated
    private String adminToken;
    private int longTokenTTL;
    private int tokenTTL;
    private int shortTokenTTL;
    private String[] rewrite_strategies;
    private String passcodeSaltField;

    private String default_pos;
    private String default_morphology;
    private String default_lemma;
    private String default_orthography;
    private String default_dep;
    private String default_const;
    private String defaultStructureFoundry;
    private ArrayList<String> foundries;
    private ArrayList<String> layers;

    private String baseURL;
    private Properties properties;

    private Set<String> supportedVersions;
    private String currentVersion;

    // deprec?!
    private final BACKENDS DEFAULT_ENGINE = BACKENDS.LUCENE;
    
    // license patterns
    protected Pattern publicLicensePattern;
    protected Pattern freeLicensePattern;
    protected Pattern allLicensePattern;

    // random code generator
    private String secureRandomAlgorithm;
    private String messageDigestAlgorithm;

    
    public KustvaktConfiguration (Properties properties) throws Exception {
        load(properties);
        KrillProperties.setProp(properties);
    }

    /**
     * loading of the properties and mapping to parameter variables
     * 
     * @param properties
     * @return
     * @throws Exception
     */
    protected void load (Properties properties) throws Exception {
        currentVersion = properties.getProperty("current.api.version", "v1.0");
        String supportedVersions =
                properties.getProperty("supported.api.version", "");
        
        this.supportedVersions = new HashSet<>(Arrays.asList(supportedVersions.split(" ")));
        this.supportedVersions.add(currentVersion);

        baseURL = properties.getProperty("kustvakt.base.url", "/api/*");
        maxhits = new Integer(properties.getProperty("maxhits", "50000"));
        returnhits = new Integer(properties.getProperty("returnhits", "50000"));
        indexDir = properties.getProperty("krill.indexDir", "");
        port = new Integer(properties.getProperty("server.port", "8095"));
        // server options
        serverHost = String
                .valueOf(properties.getProperty("server.host", "localhost"));
        String queries = properties.getProperty("korap.ql", "");
        String[] qls = queries.split(",");
        queryLanguages = new ArrayList<>();
        for (String querylang : qls)
            queryLanguages.add(querylang.trim().toUpperCase());

        default_const =
                properties.getProperty("default.foundry.constituent", "corenlp");
        default_dep =
                properties.getProperty("default.foundry.dependency", "malt");
        default_lemma = properties.getProperty("default.foundry.lemma", "tt");
        default_morphology = properties.getProperty("default.foundry.morphology", "marmot");
        default_pos =
                properties.getProperty("default.foundry.partOfSpeech", "tt");
        default_orthography =
                properties.getProperty("default.foundry.orthography", "opennlp");
        defaultStructureFoundry =
                properties.getProperty("default.foundry.structure", "base");

        // security configuration
        inactiveTime = TimeUtils.convertTimeToSeconds(
                properties.getProperty("security.idleTimeoutDuration", "10M"));
        allowMultiLogIn = Boolean
                .valueOf(properties.getProperty("security.multipleLogIn"));

        loginAttemptNum = Long.parseLong(
                properties.getProperty("security.loginAttemptNum", "3"));
        loginAttemptTTL = TimeUtils.convertTimeToSeconds(
                properties.getProperty("security.authAttemptTTL", "30M"));

        loadFactor = Integer.valueOf(
                properties.getProperty("security.encryption.loadFactor", "15"));
        validationStringLength = Integer.valueOf(properties
                .getProperty("security.validation.stringLength", "150"));
        validationEmaillength = Integer.valueOf(properties
                .getProperty("security.validation.emailLength", "40"));

        sharedSecret =
                properties.getProperty("security.sharedSecret", "").getBytes();
        adminToken = properties.getProperty("security.adminToken");

        longTokenTTL = TimeUtils.convertTimeToSeconds(
                properties.getProperty("security.longTokenTTL", "100D"));
        tokenTTL = TimeUtils.convertTimeToSeconds(
                properties.getProperty("security.tokenTTL", "72H"));
        shortTokenTTL = TimeUtils.convertTimeToSeconds(
                properties.getProperty("security.shortTokenTTL", "3H"));

        // passcodeSaltField =
        // properties.getProperty("security.passcode.salt",
        // "accountCreation");
        
        setSecureRandomAlgorithm(properties
                .getProperty("security.secure.random.algorithm", "SHA1PRNG"));

        setMessageDigestAlgorithm(
                properties.getProperty("security.md.algorithm", "MD5"));

    }

    /**
     * set properties
     * 
     * @param props
     * @throws IOException
     */
    // public void setProperties (Properties props) throws IOException
    // {
    // this.load(props);
    // }

    /**
     * properties can be overloaded after spring init
     * 
     * @param stream
     * @throws Exception
     */
    public void setPropertiesAsStream (InputStream stream) throws Exception {
        try {

            Properties p = new Properties();
            p.load(stream);
            this.load(p);
        }
        catch (IOException e) {
            e.printStackTrace();
        }

    }

    public BACKENDS chooseBackend (String value) {
        if (value == null || value.equals("null"))
            return DEFAULT_ENGINE;
        else
            return Enum.valueOf(BACKENDS.class, value.toUpperCase());
    }

    public enum BACKENDS {
        NEO4J, LUCENE
    }

}