org.graylog.plugins.pipelineprocessor.functions.urls.URL Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of graylog2-server Show documentation
Graylog
There is a newer version: 6.0.1
/*
 * Copyright (C) 2020 Graylog, Inc.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the Server Side Public License, version 1,
 * as published by MongoDB, Inc.
 *
 * 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
 * Server Side Public License for more details.
 *
 * You should have received a copy of the Server Side Public License
 * along with this program. If not, see
 * .
 */
package org.graylog.plugins.pipelineprocessor.functions.urls;

import com.google.common.base.Joiner;
import com.google.common.collect.Maps;
import okhttp3.HttpUrl;

import java.util.List;
import java.util.Map;

/**
 * This class simply delegates the safe methods to the {@link java.net.URL}.
 *
 * Specifically we want to disallow called {@link java.net.URL#getContent()} from a rule.
 */
public class URL {
    private final HttpUrl url;
    private Map queryMap;

    public URL(java.net.URL url) {
        this.url = HttpUrl.get(url);
    }

    public URL(java.net.URI uri) {
        this.url = HttpUrl.get(uri);
    }

    public URL(String urlString) {
        url = HttpUrl.parse(urlString);
    }

    public String getQuery() {
        return url.encodedQuery();
    }

    public Map getQueryParams() {
        if (queryMap == null) {
            final Map queryMap = Maps.newHashMapWithExpectedSize(url.querySize());
            for(String name : url.queryParameterNames()) {
                final List values = url.queryParameterValues(name);
                final String valueString = Joiner.on(',').join(values);
                queryMap.put(name, valueString);
            }
            this.queryMap = queryMap;
        }
        return queryMap;
    }

    public String getUserInfo() {
        final String username = url.encodedUsername();
        return username.isEmpty() ? "" : username + ':' + url.encodedPassword();
    }

    public String getHost() {
        return url.host();
    }

    public String getPath() {
        return url.encodedPath();
    }

    public String getFile() {
        return url.querySize() == 0 ? url.encodedPath() : url.encodedPath() + '?' + url.encodedQuery();
    }

    public String getProtocol() {
        return url.scheme();
    }

    public int getDefaultPort() {
        return url.port();
    }

    /**
     * alias for #getRef, fragment is more commonly used
     */
    public String getFragment() {
        return url.encodedFragment();
    }

    public String getRef() {
        return getFragment();
    }

    public String getAuthority() {
        final String userInfo = getUserInfo();
        return userInfo.isEmpty() ? getHost() + ':' + getPort() : getUserInfo() + '@' + getHost() + ':' + getPort();
    }

    public int getPort() {
        return url.port();
    }

    @SuppressWarnings("EqualsWhichDoesntCheckParameterClass")
    @Override
    public boolean equals(Object obj) {
        return url.equals(obj);
    }

    @Override
    public int hashCode() {
        return url.hashCode();
    }
}