apoc.util.UriResolver Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of apoc Show documentation
A collection of useful Neo4j Procedures
There is a newer version: 4.4.0.32
/*
 * Copyright (c) "Neo4j"
 * Neo4j Sweden AB [http://neo4j.com]
 *
 * This file is part of Neo4j.
 *
 * 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 apoc.util;

import static apoc.ApocConfig.apocConfig;

import java.net.URI;
import java.net.URISyntaxException;
import org.neo4j.driver.AuthToken;
import org.neo4j.driver.AuthTokens;

/**
 * @author AgileLARUS
 * @since 03.09.17
 */
public class UriResolver {

    private URI uri;
    private String url;
    private String prefix;
    private AuthToken token;

    public UriResolver(String url, String prefix) throws URISyntaxException {
        this.url = url;
        this.prefix = prefix;
    }

    public URI getConfiguredUri() {
        return uri;
    }

    public AuthToken getToken() {
        return token;
    }

    private String getConfiguredUri(String key) {
        String keyUrl = "apoc." + this.prefix + "." + key + ".url";
        if (apocConfig().containsKey("apoc.bolt.url")) {
            key = apocConfig().getString("apoc.bolt.url");
        } else if (apocConfig().containsKey(keyUrl)) {
            key = apocConfig().getString(keyUrl, key);
        }
        return key;
    }

    public void initialize() throws URISyntaxException {
        this.url = getConfiguredUri(this.url);
        URI uri;
        try {
            uri = new URI(this.url);
        } catch (URISyntaxException e) {
            throw new URISyntaxException(e.getInput(), e.getMessage());
        }
        this.uri = uri;
        String[] userInfoArray =
                uri.getUserInfo() == null ? new String[2] : uri.getUserInfo().split(":");
        String user = userInfoArray[0];
        String password = userInfoArray[1];
        if (user != null && password == null || user == null && password != null)
            throw new RuntimeException(
                    "user and password don't defined check your URL or if you use a key the property in your apoc.conf file");
        this.token = (user != null && password != null) ? AuthTokens.basic(user, password) : AuthTokens.none();
    }
}