org.biins.cauchbase.builder.ViewConfigHelper Maven / Gradle / Ivy

Go to download
package org.biins.cauchbase.builder;

import com.couchbase.cbadmin.client.ViewConfigBuilder;
import org.biins.cauchbase.View;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

/**
 * @author Martin Janys
 */
public class ViewConfigHelper {

    private static final String CLASS_PATH_PREFIX = "classpath:";

    public static ViewConfigBuilder create(String designName, String name, String password) {
        ViewConfigBuilder builder = new ViewConfigBuilder(designName, name);
        builder.password(password);
        return builder;
    }

    public static void addView(ViewConfigBuilder viewConfigBuilder, View view) {
        viewConfigBuilder.view(
                view.name(),
                resolveFunction(view.map()),
                !view.reduce().isEmpty() ? resolveFunction(view.reduce()) : null);
    }

    private static String resolveFunction(String func) {
        if (func.startsWith(CLASS_PATH_PREFIX)) {
            return readClassPathResource(func);
        }
        else {
            return func;
        }
    }

    private static String readClassPathResource(String path) {
        StringBuilder sb = new StringBuilder();
        BufferedReader reader = new BufferedReader(new InputStreamReader(ClassLoader.getSystemResourceAsStream(parseClasspathPath(path))));
        String lineSeparator = String.format("%n");
        String line;
        try {
            while ((line = reader.readLine()) != null) {
                sb.append(line).append(lineSeparator);
            }
        }
        catch (IOException e) {
            throw new RuntimeException(e);
        }
        return sb.toString();
    }

    private static String parseClasspathPath(String path) {
        String prefix = CLASS_PATH_PREFIX + "/";
        if (!path.startsWith(prefix)) {
            throw new IllegalArgumentException("Set absolute classpath path starts with '/'. For example classpath:/path/to/script.js");
        }
        else {
            return path.substring(prefix.length());
        }
    }

}