PasswordManager.java

package com.thesecretserver;

import java.io.IOException;
import java.lang.reflect.Field;
import java.net.URL;

import play.Logger;

import com.thesecretserver.service.AddSecretResult;
import com.thesecretserver.service.ArrayOfInt;
import com.thesecretserver.service.ArrayOfString;
import com.thesecretserver.service.GetSecretResult;
import com.thesecretserver.service.SSWebService;
import com.thesecretserver.service.SSWebServiceSoap;
import com.thesecretserver.service.SecretItem;

public class PasswordManager {
	
	private String username;
	private String password;
	private String organization;
	private String domain;
	private SSWebServiceSoap ssWebServiceSoap;
	private static final int URL_FIELD_ID = 38;
	private static final int USERNAME_FIELD_ID = 39;
	private static final int PASSWORD_FIELD_ID = 40;
	private static final int NOTES_FIELD_ID = 41;
	private static final int SECRET_TYPE_ID = 9;
	private static final int FOLDER_ID = 12;
	
	public PasswordManager(String username, String password, String organization, String domain) {
		this.username = username;
		this.password = password;
		this.organization = organization;
		this.domain = domain;
	}

	public int addLoginCredentials(String secretName, LoginCredentials loginCredentials) throws IOException {
		String token = authenticate();
		if (token == null || token.isEmpty()) throw new IOException();
		AddSecretResult addSecretResult = ssWebServiceSoap.addSecret(token,
				SECRET_TYPE_ID, secretName,
				getSecretFieldIds(), getSecretItemValues(loginCredentials),
				FOLDER_ID);
		for (String error : addSecretResult.getErrors().getString()) {
			Logger.error("addSecret: " + error);
		}
		return addSecretResult.getSecret().getId();
	}
	
	public LoginCredentials getLoginCredentials(int secretId) throws IOException {
		String token = authenticate();
		if (token == null || token.isEmpty()) throw new IOException();
		GetSecretResult getSecretResult = ssWebServiceSoap.getSecret(token, secretId, null, null);
		LoginCredentials loginCredentials = new LoginCredentials();
		for (SecretItem secretItem : getSecretResult.getSecret().getItems().getSecretItem()) {
			if (secretItem.getFieldId().equals(URL_FIELD_ID))
				loginCredentials.url = secretItem.getValue();
			else if (secretItem.getFieldId().equals(USERNAME_FIELD_ID))
				loginCredentials.username = secretItem.getValue();
			else if (secretItem.getFieldId().equals(PASSWORD_FIELD_ID))
				loginCredentials.password = secretItem.getValue();
		}
		return loginCredentials;
	}
	
	public static String versionGet() {
		SSWebServiceSoap ssWebServiceSoap = new SSWebService().getSSWebServiceSoap();
		return ssWebServiceSoap.versionGet().getVersion();
	}
	
	public static URL getSecretServerEndpoint() {
		try {
			Field f = SSWebService.class.getDeclaredField("SSWEBSERVICE_WSDL_LOCATION");
			f.setAccessible(true);
			return (URL) f.get(null);
		} catch (Exception e) {
			Logger.warn("Could not inspect WSDL location. ", e);
		}
		return null;
	}
	
	private String authenticate() {
		ssWebServiceSoap = new SSWebService().getSSWebServiceSoap();
		return ssWebServiceSoap.authenticate(username, password, organization, domain).getToken();
	}
	
	private ArrayOfInt getSecretFieldIds() {
		ArrayOfInt secretFieldIds = new ArrayOfInt();
		secretFieldIds.getInt().add(URL_FIELD_ID);
		secretFieldIds.getInt().add(USERNAME_FIELD_ID);
		secretFieldIds.getInt().add(PASSWORD_FIELD_ID);
		secretFieldIds.getInt().add(NOTES_FIELD_ID);
		return secretFieldIds;
	}
	
	private ArrayOfString getSecretItemValues(LoginCredentials loginCredentials) {
		ArrayOfString secretItemValues = new ArrayOfString();
		secretItemValues.getString().add(loginCredentials.url);
		secretItemValues.getString().add(loginCredentials.username);
		secretItemValues.getString().add(loginCredentials.password);
		secretItemValues.getString().add("");
		return secretItemValues;
	}
}