org.sdmlib.modelcouch.CouchDBAdapter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of SDMLib Show documentation
SDMLib is a light weight modeling library. SDMLib intentionally comes without any tool or editor.
There is a newer version: 2.3.2341
package org.sdmlib.modelcouch;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.net.HttpURLConnection;
import java.net.URL;
import java.net.URLConnection;
import java.nio.charset.Charset;
import java.util.Collection;
import java.util.LinkedList;
import java.util.Map.Entry;

import org.sdmlib.modelcouch.authentication.Authenticator;
import org.sdmlib.modelcouch.authentication.BasicAuthenticator;

import de.uniks.networkparser.json.JsonArray;
import de.uniks.networkparser.json.JsonObject;

public class CouchDBAdapter {
	private String userName = "";
	private String hostName = "localhost";
	private int port = 5984;
	
	public String getHostName() {
		return hostName;
	}

	public int getPort() {
		return port;
	}
	
	public String getUserName() {
		return userName;
	}
	
	public void setHostName(String hostName) {
		this.hostName = hostName;
	}
	
	public CouchDBAdapter withHostName(String hostName){
		this.hostName = hostName;
		return this;
	}
	
	public void setPort(int port) {
		this.port = port;
	}
	
	public CouchDBAdapter withPort(int port) {
		this.port = port;
		return this;
	}
	
	public void setUserName(String username) {
		this.userName = username;
	}
	
	public CouchDBAdapter withUserName(String username) {
		this.userName = username;
		return this;
	}
	
	public ReturnObject replicate(ModelCouch modelCouch, String source, String target) {
		RequestObject request = createRequestObject();
		request.setPath("_replicate");
		request.setRequestType(RequestType.POST);

		/*
		 * {"source":"http://example.org/example-database","target":
		 * "http://admin:[email protected]:5984/example-database"}
		 */

		JsonObject replicateObject = new JsonObject();
		replicateObject.add("create_target", true);
		replicateObject.add("source", source);
		replicateObject.add("target", target);
		replicateObject.add("retries_per_request", 2);

		request.setOutput(replicateObject.toString().getBytes());

		return send(request);
	}

	public ReturnObject setUserPrivileges(String database, Collection adminNames, Collection adminRoles,
			Collection memberNames, Collection memberRoles) {
		RequestObject request = createRequestObject();
		request.setPath(database + "/_security");
		request.setRequestType(RequestType.PUT);
		request.setShouldHandleInput(true);

		JsonObject json = new JsonObject();
		JsonObject admins = new JsonObject();
		JsonArray adminNamesJson = new JsonArray();
		adminNamesJson.addAll(adminNames);
		admins.add("names", adminNamesJson);
		JsonArray adminRolesJson = new JsonArray();
		adminRolesJson.addAll(adminRoles);
		admins.add("roles", adminRolesJson);
		json.put("admins", admins);
		JsonObject members = new JsonObject();
		JsonArray memberNamesJson = new JsonArray();
		memberNamesJson.addAll(memberNames);
		members.add("names", memberNamesJson);
		JsonArray memberRolesJson = new JsonArray();
		memberRolesJson.addAll(memberRoles);
		members.add("roles", memberRolesJson);
		json.put("members", members);

		request.setOutput(json.toString().getBytes(Charset.forName("UTF-8")));

		return send(request);
	}

	public RequestObject createRequestObject() {
		return new RequestObject(this);
	}

	public ReturnObject send(RequestObject request) {
		ReturnObject res = new ReturnObject();
		try {
			URL obj = new URL(request.getServer() + request.getPath());
			HttpURLConnection con = (HttpURLConnection) obj.openConnection();
			// try to delete database
			con.setRequestMethod(request.getRequestType().toString());
			con.setDoInput(true);
			if (request.getRequestProperties() == null) {
				con.addRequestProperty("Content-Type", "application/json");
			} else {
				for (Entry entry : request.getRequestProperties().entrySet()) {
					con.addRequestProperty(entry.getKey(), entry.getValue());
				}
			}
			authenticate(con);
			if ((request.getOutput() != null && request.getOutput().length > 0)) {
				con.setDoOutput(true);
				con.getOutputStream().write(request.getOutput());
			}
			res.responseCode = con.getResponseCode();

			if (res.responseCode >= 400 && con.getErrorStream() != null) {
				InputStream errorStream = (InputStream) con.getErrorStream();
				BufferedReader reader = new BufferedReader(new InputStreamReader(errorStream));
				try {
					LinkedList lines = new LinkedList<>();
					String readLine = null;
					do {
						readLine = reader.readLine();
						if (readLine == null) {
							break;
						}
						lines.add(readLine);
					} while (readLine != null);
					res.error = lines;
				} catch (IOException e) {
					// e.printStackTrace();
				}
			} else {
				if (request.isShouldHandleInput()) {
					InputStream content = (InputStream) con.getContent();
					BufferedReader reader = new BufferedReader(new InputStreamReader(content));
					try {
						LinkedList lines = new LinkedList<>();
						String readLine = null;
						do {
							readLine = reader.readLine();
							if (readLine == null) {
								break;
							}
							lines.add(readLine);
						} while (readLine != null);
						res.setContent(lines);
					} catch (IOException e) {
						// e.printStackTrace();
					}
				}
			}

			con.disconnect();
		} catch (Exception e) {
			res.responseCode = 400;
			// e.printStackTrace();
		}
		return res;
	}

	public boolean testConnection() {
		String urlString = "http://" + getHostName() + ":" + port;
		try {
			URL urlObj = new URL(urlString);
			URLConnection openConnection = urlObj.openConnection();
			openConnection.connect();
		} catch (IOException e) {
			return false;
		}
		return true;
	}

	public boolean testConnection(String databaseName) {
		String urlString = "http://" + hostName + ":" + port + "/" + databaseName + "/";
		try {
			// first check, if host is available..
			URL urlObj = new URL(urlString);
			URLConnection openConnection = urlObj.openConnection();
			openConnection.connect();

			// now check for existence of DataBase
			RequestObject check = createRequestObject();
			check.setRequestType(RequestType.GET);
			check.setPath(databaseName);
			check.setShouldHandleInput(true);

			ReturnObject send = send(check);
			if (send.getResponseCode() >= 400) {
				return false;
			}
		} catch (IOException e) {
			return false;
		}
		return true;
	}

	public ReturnObject createDB(String dbName) {
		RequestObject create = createRequestObject();
		create.setRequestType(RequestType.PUT);
		create.setPath(dbName);
		return send(create);
	}

	public ReturnObject deleteDatabase(String databaseName) {
		RequestObject delete = createRequestObject();
		delete.setPath(databaseName);
		delete.setRequestType(RequestType.DELETE);
		return send(delete);
	}

	private Authenticator authenticator;

	public CouchDBAdapter withAuthenticator(Authenticator autheticator) {
		this.authenticator = autheticator;
		return this;
	}

	public Authenticator getAutheticator() {
		return authenticator;
	}
	
	/**
	 * Must be called after setting credentials (and authenticator)
	 * 
	 * You can get the Return from the Server by asking the Authenticator for the loginRequest
	 * 
	 * @param password The Password
	 * @return ThisComponent
	 * @throws Exception any Errors for login
	 */
	public CouchDBAdapter login(String password) throws Exception{
		if(this.authenticator == null){
			this.authenticator = new BasicAuthenticator();
		}
		if(this.authenticator.login(getUserName(), password , this)){
			return this;
		}else{
			throw new Exception("Couldn't log in...");
		}
	}
	
	protected void authenticate(HttpURLConnection connection) {
		if(authenticator != null){
			authenticator.authenticate(connection);
		}
	}

}