com.github.apetrelli.scafa.config.Configuration Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of scafa-server Show documentation
Scafa: main code
There is a newer version: 1.0.0-alpha-3
/**
 * Scafa - A universal non-caching proxy for the road warrior
 * Copyright (C) 2015  Antonio Petrelli
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see .
 */
package com.github.apetrelli.scafa.config;

import java.io.File;
import java.io.IOException;
import java.util.Iterator;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.stream.Collectors;

import org.ini4j.Ini;
import org.ini4j.InvalidFileFormatException;
import org.ini4j.Profile.Section;

public class Configuration {
    private Ini ini;

    public static Configuration create(String profile) throws InvalidFileFormatException, IOException {
        if (profile == null) {
            profile = "direct";
        }
        Ini ini = new Ini();
        ini.load(new File(System.getProperty("user.home") + "/.scafa/" + profile + ".ini"));
        return new Configuration(ini);
    }

    private Configuration(Ini ini) {
        this.ini = ini;
    }

    public Section getMainConfiguration() {
        return ini.get("main");
    }

    public Section getConfigurationByHost(String host) {
        boolean found = false;
        Iterator keyIt = ini.keySet().iterator();
        Section section = null;
        while (!found && keyIt.hasNext()) {
            section = ini.get(keyIt.next());
            if (!"main".equals(section.getName())) {
                List excludeRegexp = section.getAll("excludeRegexp");
                if (excludeRegexp == null || excludeRegexp.isEmpty()) {
                    List exclude = section.getAll("exclude");
                    if (exclude == null || exclude.isEmpty()) {
                        section.put("excludeRegexp", "doesnotmatchathing"); // Just to avoid useless re-execution
                        excludeRegexp = section.getAll("excludeRegexp");
                    } else {
                        excludeRegexp = exclude.stream().map(t -> createRegexpFromWildcard(t)).collect(Collectors.toList());
                        section.putAll("excludeRegexp", excludeRegexp);
                    }
                }
                boolean excluded = excludeRegexp.stream().anyMatch(t -> host.matches(t));
                found = !excluded;
            }
        }
        return found ? section : null;
    }

    private String createRegexpFromWildcard(String subject) {
        Pattern regex = Pattern.compile("[^*]+|(\\*)");
        Matcher m = regex.matcher(subject);
        StringBuffer b= new StringBuffer();
        while (m.find()) {
            if(m.group(1) != null) m.appendReplacement(b, ".*");
            else m.appendReplacement(b, "\\\\Q" + m.group(0) + "\\\\E");
        }
        m.appendTail(b);
        return b.toString();
    }
}